class Window(QtGui.QMainWindow):
OPPOSITE_TEXT = {"On": "Off", "Off": "On",
"Play": "Pause", "Pause": "Play"}
def __init__(self):
super(Window, self).__init__()
self.setGeometry(50, 50, 500, 500)
self.setWindowTitle("Sound editor")
self.view_model = ViewModel()
self.view_model.call_after_change.append(self.repaint)
self.tracks = []
self.statusBar()
main_menu = self.menuBar()
self.file_menu = main_menu.addMenu('&File')
self.edit_menu = main_menu.addMenu('&Edit')
self.add_menu_item(self.file_menu, "&Open",
self.open_file, QtGui.QKeySequence.Open)
self.add_menu_item(self.edit_menu, "&To end", self.view_model.to_end,
QtGui.QKeySequence.MoveToNextWord)
self.add_menu_item(self.edit_menu, "&To begin",
self.view_model.to_begin,
QtGui.QKeySequence.MoveToPreviousWord)
self.add_menu_item(self.edit_menu, "&Copy", self.view_model.copy,
QtGui.QKeySequence.Copy)
self.add_menu_item(self.edit_menu, "&Paste", self.view_model.paste,
QtGui.QKeySequence.Paste)
self.add_menu_item(self.edit_menu, "&Delete", self.view_model.delete,
QtGui.QKeySequence.Delete)
self.add_menu_item(self.edit_menu, "&Reverse",
self.view_model.reverse,
QtGui.QKeySequence("Ctrl+R"))
self.add_menu_item(self.edit_menu, "&Select all",
self.view_model.select_all,
QtGui.QKeySequence.SelectAll)
self.add_menu_item(self.edit_menu, "&Paste to new track",
self.paste_to_new_track,
QtGui.QKeySequence.New)
self.add_menu_item(self.edit_menu, "&Sum active tracks",
self.add_sum, QtGui.QKeySequence("Alt+S"))
self.scroll_layout = QtGui.QFormLayout()
self.scroll_widget = QtGui.QWidget()
self.scroll_widget.setLayout(self.scroll_layout)
self.scroll_area = QtGui.QScrollArea()
self.scroll_area.setWidgetResizable(True)
self.scroll_area.setWidget(self.scroll_widget)
self.setCentralWidget(self.scroll_area)
self.show()
def create_menu_item_action(self, name, func, shortcut):
menu_item_action = QtGui.QAction(name, self)
menu_item_action.triggered.connect(func)
menu_item_action.setShortcut(shortcut)
return menu_item_action
def add_menu_item(self, menu, name, func, shortcut):
action = self.create_menu_item_action(name, func, shortcut)
menu.addAction(action)
def save_master_to_file(self):
pass
def open_file(self):
file_name = QtGui.QFileDialog.getOpenFileName()
if file_name != "":
try:
wave_state = WaveState.read_from_file(file_name)
except wave.Error:
message_box = QtGui.QMessageBox()
message_box.warning(self, "Incorrect file",
"Selected file is incorrect")
else:
track_model = TrackModel(wave_state, self.view_model)
self.add_track_model(track_model)
def paste_to_new_track(self):
track_model = self.view_model.get_pasted_to_new_track_model()
self.add_track_model(track_model)
def add_track_model(self, track_model):
self.view_model.add_track_model(track_model)
track = Track(track_model, self)
self.tracks.append(track)
add_effect = track.actions_panel.add_effect
add_effect("Fade in", FadeInParamsWidget(track_model))
add_effect("Fade out", FadeOutParamsWidget(track_model))
add_effect("Change speed", ChangeSpeedParamsWidget(track_model))
add_effect("Compress sound", CompressSoundParamsWidget(track_model))
add_button = track.actions_panel.add_button
add_button("Delete", lambda: self.delete_track(track))
add_button("Save", lambda: self.save_to_file(track_model.wave_state))
add_button("Adjust loudness",
lambda: self.adjust_loudness(track_model))
on_off_button = QtGui.QPushButton("On")
on_off_button.clicked.connect(
lambda: self.switch_on_off(track_model, on_off_button))
track.actions_panel.add_action_view(on_off_button)
play_pause_button = QtGui.QPushButton("Play")
play_pause_button.clicked.connect(
lambda: self.switch_play_pause(track))
track.set_play_pause_button(play_pause_button)
track.actions_panel.add_action_view(play_pause_button)
self.scroll_layout.addRow(track)
def delete_track(self, track):
self.scroll_layout.removeWidget(track)
self.view_model.delete_track_model(track.track_model)
self.tracks.remove(track)
track.close()
def adjust_loudness(self, track_model):
dialog = QtGui.QInputDialog(self)
ratio_and_bool = dialog.getDouble(self, "Adjust loudness",
"Input ratio:", 1)
if ratio_and_bool[1]:
track_model.adjust_loudness(ratio_and_bool[0])
def switch_on_off(self, track_model, button):
track_model.is_active = not track_model.is_active
button.setText(self.OPPOSITE_TEXT[button.text()])
def switch_play_pause(self, track):
button = track.play_pause_button
if button.text() == "Play":
self.stop_all_sounds()
track.play()
else:
track.stop()
button.setText(self.OPPOSITE_TEXT[button.text()])
def stop_all_sounds(self):
for track in self.tracks:
button = track.play_pause_button
if button.text() == "Pause":
track.stop()
button.setText(self.OPPOSITE_TEXT[button.text()])
def add_sum(self):
wave_state = self.view_model.get_sum()
track_model = TrackModel(wave_state, self.view_model)
self.add_track_model(track_model)
@staticmethod
def save_to_file(wave_state):
file_name = QtGui.QFileDialog.getSaveFileName()
if file_name != "":
wave_state.save_to_file(file_name)
