def createDockWindows(self):
#dock = QDockWidget("Tokens", self)
#dock.setAllowedAreas(Qt.LeftDockWidgetArea | Qt.RightDockWidgetArea)
#dock.setWidget(self.tokenTable)
#self.addDockWidget(Qt.RightDockWidgetArea, dock)
#self.viewMenu.addAction(dock.toggleViewAction())
dock = QDockWidget("Spectrogram", self)
self.specgramWindow = SpecgramWidget(parent=dock)
dock.setWidget(self.specgramWindow)
self.addDockWidget(Qt.RightDockWidgetArea, dock)
self.viewMenu.addAction(dock.toggleViewAction())
dock = QDockWidget("Auditory Representation", self)
self.representationWindow = RepresentationWidget(parent=dock)
dock.setWidget(self.representationWindow)
self.addDockWidget(Qt.RightDockWidgetArea, dock)
self.viewMenu.addAction(dock.toggleViewAction())
dock = QDockWidget("Distance", self)
self.distanceWindow = DistanceWidget(parent=dock)
dock.setWidget(self.distanceWindow)
self.addDockWidget(Qt.RightDockWidgetArea, dock)
self.viewMenu.addAction(dock.toggleViewAction())
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.dirty = False
# Models
# - Filesystem Model
self.fileSystemModel = QFileSystemModel()
# rootPath = QDesktopServices.storageLocation(
# QDesktopServices.HomeLocation)
# fileSystemRoot = self.fileSystemModel.setRootPath(rootPath)
fileSystemRoot = self.fileSystemModel.setRootPath(
"/home/ryan/Programming/Python/projects/r3tagger/r3tagger/tests"
)
# Views
# - Filesystem View
self.fileSystemView = QTreeView()
self.fileSystemView.setModel(self.fileSystemModel)
self.fileSystemView.setRootIndex(fileSystemRoot)
self.fileSystemView.doubleClicked.connect(self.updateAlbumModel)
self.fileSystemView.expanded.connect(self.fixFileSystemColumns)
self.fileSystemView.collapsed.connect(self.fixFileSystemColumns)
# - Album View
self.albumView = albumcollection.MusicCollectionView()
self.albumView.setSelectionMode(QAbstractItemView.MultiSelection)
self.albumView.clicked.connect(self.updateEditing)
self.albumView.expanded.connect(self.fixAlbumViewColumns)
self.albumView.collapsed.connect(self.fixAlbumViewColumns)
self.albumView.model().dataChanged.connect(self.fixAlbumViewColumns)
self.albumView.model().dataChanged.connect(self._setDirty)
model = self.albumView.model()
model.dataChanged.connect(self.updateEditing)
# Editing Group
self.editingGroup = QFormLayout()
self.lineArtist = QLineEdit()
self.lineAlbum = QLineEdit()
self.lineTitle = QLineEdit()
self.lineTrack = QLineEdit()
self.lineDate = QLineEdit()
self.lineGenre = QLineEdit()
self.editingGroup.addRow("Artist:", self.lineArtist)
self.editingGroup.addRow("Album:", self.lineAlbum)
self.editingGroup.addRow("Title:", self.lineTitle)
self.editingGroup.addRow("Track:", self.lineTrack)
self.editingGroup.addRow("Date:", self.lineDate)
self.editingGroup.addRow("Genre:", self.lineGenre)
self.tagsToAttribs = {
"artist": self.lineArtist,
"album": self.lineAlbum,
"title": self.lineTitle,
"tracknumber": self.lineTrack,
"date": self.lineDate,
"genre": self.lineGenre,
}
# Confirm / Cancel / Clear Group
self.buttonGroup = QHBoxLayout()
self.buttonGroup.addStretch()
confirm = QPushButton("Confirm")
confirm.clicked.connect(self.confirmChanges)
self.buttonGroup.addWidget(confirm)
cancel = QPushButton("Cancel")
cancel.clicked.connect(self.cancelChanges)
self.buttonGroup.addWidget(cancel)
clear = QPushButton("Clear")
clear.clicked.connect(self.clearAlbumView)
self.buttonGroup.addWidget(clear)
self.buttonGroup.addStretch()
# Statusbar
# status = self.statusBar()
# status.setSizeGripEnabled(False)
# status.showMessage("Ready", 5000)
# Docks
dockAllowed = Qt.LeftDockWidgetArea | Qt.RightDockWidgetArea
# - Filesystem Dock
fileSystemDock = QDockWidget("Navigate", self)
fileSystemDock.setObjectName("fileSystemDock")
fileSystemDock.setAllowedAreas(dockAllowed)
fileSystemDock.setWidget(self.fileSystemView)
self.addDockWidget(Qt.LeftDockWidgetArea, fileSystemDock)
# - Editing Dock
editingWidget = QWidget()
editingWidget.setLayout(self.editingGroup)
editingDock = QDockWidget("Editing", self)
editingDock.setObjectName("editingDock")
editingDock.setAllowedAreas(dockAllowed)
editingDock.setWidget(editingWidget)
self.addDockWidget(Qt.RightDockWidgetArea, editingDock)
# Actions
fileAddSongAction = self._createAction(
text="Add &Songs",
slot=self.fileAddSong,
shortcut=QKeySequence.Open,
icon="fileOpen",
tip="Add files (songs)",
)
fileAddAlbumAction = self._createAction(
text="Add &Album",
slot=self.fileAddAlbum,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_O),
icon="fileOpen",
tip="Add directory (album)",
)
fileSaveAction = self._createAction(
text="&Save Changes",
slot=self.confirmChanges,
shortcut=QKeySequence.Save,
icon="fileSave",
tip="Save Changes",
)
fileQuitAction = self._createAction(
text="&Quit", slot=self.close, shortcut=QKeySequence.Quit, icon="fileQuit", tip="Quit Program"
)
editRecognizeAction = self._createAction(
text="&Recognize",
slot=self.editRecognize,
shortcut=QKeySequence(Qt.CTRL + Qt.Key_R),
icon="editRecognize",
tip="Recognize music",
)
editReorganizeAction = self._createAction(
text="Reorganize",
slot=self.editReorganize,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_R),
icon="editReorganize",
tip="Reorganize music",
)
editSettingsAction = self._createAction(
text="Settings",
slot=self.editSettings,
shortcut=QKeySequence.Preferences,
icon="editSettings",
tip="Edit settings",
)
helpDocsAction = self._createAction(
text="Documentation",
slot=self.helpDocs,
shortcut=QKeySequence.HelpContents,
icon="helpDocs",
tip="Documentation",
)
helpAboutAction = self._createAction(
text="About",
slot=self.helpAbout,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_F1),
icon="helpAbout",
tip="About",
)
toggleEditing = editingDock.toggleViewAction()
toggleEditing.setIcon(QIcon(":/toggleEditing.png"))
toggleFileNav = fileSystemDock.toggleViewAction()
toggleFileNav.setIcon(QIcon(":/toggleFileNav.png"))
# Menus
fileMenu = self.menuBar().addMenu("&File")
self._addActions(fileMenu, (fileAddSongAction, fileAddAlbumAction, fileSaveAction, fileQuitAction))
editMenu = self.menuBar().addMenu("&Edit")
self._addActions(editMenu, (editReorganizeAction, editRecognizeAction, editSettingsAction))
helpMenu = self.menuBar().addMenu("&Help")
self._addActions(helpMenu, (helpDocsAction, helpAboutAction))
# Toolbars
editToolbar = self.addToolBar("EditToolbar")
editToolbar.setObjectName("editToolbar")
self._addActions(editToolbar, (editRecognizeAction, editReorganizeAction))
toggleToolbar = self.addToolBar("ToggleToolbar")
toggleToolbar.setObjectName("toggleToolbar")
self._addActions(toggleToolbar, (toggleFileNav, toggleEditing))
# Settings
settings = QSettings()
if settings.contains("MainWindow/Geometry"):
self.restoreGeometry(settings.value("MainWindow/Geometry"))
if settings.contains("MainWindow/State"):
self.restoreState(settings.value("MainWindow/State"))
self.setWindowTitle("r3tagger")
# Final Layout
centralWidget = QWidget()
centralLayout = QVBoxLayout()
centralLayout.addWidget(self.albumView)
centralLayout.addLayout(self.buttonGroup)
centralWidget.setLayout(centralLayout)
self.setCentralWidget(centralWidget)
