def __init__(self, parent = None): QMainWindow.__init__(self, parent) QCoreApplication.setOrganizationName('SpykeUtils') QCoreApplication.setApplicationName('Spyke Viewer') self.setupUi(self) self.dir = os.getcwd() self.console = None self.progress = ProgressIndicatorDialog(self) self.provider_factory = DataProvider self.selections = [] self.provider = None self.plugin_paths = [] # Drag and Drop for selections menu self.menuSelections.setAcceptDrops(True) self.menuSelections.paintEvent =\ self.on_menuSelections_paint self.menuSelections.mousePressEvent =\ self.on_menuSelections_mousePressed self.menuSelections.mouseMoveEvent =\ self.on_menuSelections_mouseMoved self.menuSelections.dragEnterEvent =\ self.on_menuSelections_dragEnter self.menuSelections.dragMoveEvent =\ self.on_menuSelections_dragMoved self.menuSelections.dropEvent =\ self.on_menuSelections_drop self.seldrag_start_pos = None self.seldrag_selection = None self.seldrag_target = None self.seldrag_target_upper = False # Docks self.setCentralWidget(None) # A lot of docks are only needed for the internal database mode self.navigationDBDock.setVisible(False) self.filterDBDock.setVisible(False) self.spikeSortingDock.setVisible(False) self.waveformsDock.setVisible(False) self.spikeTrainDock.setVisible(False) if not hasattr(self, 'internal_database_mode'): self.removeDockWidget(self.navigationDBDock) self.removeDockWidget(self.filterDBDock) self.removeDockWidget(self.spikeSortingDock) self.removeDockWidget(self.waveformsDock) self.removeDockWidget(self.spikeTrainDock) del self.navigationDBDock del self.filterDBDock del self.spikeSortingDock del self.waveformsDock del self.spikeTrainDock self.update_view_menu() if not hasattr(self, 'internal_database_mode'): self.restore_state() self.mode_switch()
class MainWindow(QMainWindow, Ui_MainWindow): """ The main window of SpikeViewer """ def __init__(self, parent = None): QMainWindow.__init__(self, parent) QCoreApplication.setOrganizationName('SpykeUtils') QCoreApplication.setApplicationName('Spyke Viewer') self.setupUi(self) self.dir = os.getcwd() self.console = None self.progress = ProgressIndicatorDialog(self) self.provider_factory = DataProvider self.selections = [] self.provider = None self.plugin_paths = [] # Drag and Drop for selections menu self.menuSelections.setAcceptDrops(True) self.menuSelections.paintEvent =\ self.on_menuSelections_paint self.menuSelections.mousePressEvent =\ self.on_menuSelections_mousePressed self.menuSelections.mouseMoveEvent =\ self.on_menuSelections_mouseMoved self.menuSelections.dragEnterEvent =\ self.on_menuSelections_dragEnter self.menuSelections.dragMoveEvent =\ self.on_menuSelections_dragMoved self.menuSelections.dropEvent =\ self.on_menuSelections_drop self.seldrag_start_pos = None self.seldrag_selection = None self.seldrag_target = None self.seldrag_target_upper = False # Docks self.setCentralWidget(None) # A lot of docks are only needed for the internal database mode self.navigationDBDock.setVisible(False) self.filterDBDock.setVisible(False) self.spikeSortingDock.setVisible(False) self.waveformsDock.setVisible(False) self.spikeTrainDock.setVisible(False) if not hasattr(self, 'internal_database_mode'): self.removeDockWidget(self.navigationDBDock) self.removeDockWidget(self.filterDBDock) self.removeDockWidget(self.spikeSortingDock) self.removeDockWidget(self.waveformsDock) self.removeDockWidget(self.spikeTrainDock) del self.navigationDBDock del self.filterDBDock del self.spikeSortingDock del self.waveformsDock del self.spikeTrainDock self.update_view_menu() if not hasattr(self, 'internal_database_mode'): self.restore_state() self.mode_switch() def update_view_menu(self): self.menuBar.clear() self.menuBar.addMenu(self.menuFile) self.menuBar.addMenu(self.menuSelections) self.menuView = self.createPopupMenu() self.menuView.setTitle('View') self.menuBar.addMenu(self.menuView) self.menuBar.addMenu(self.menuHelp) def restore_state(self): settings = QSettings() if not settings.contains('windowGeometry') or\ not settings.contains('windowState'): self.set_initial_layout() else: self.restoreGeometry(settings.value('windowGeometry')) self.restoreState(settings.value('windowState')) if not settings.contains('pluginPaths'): if hasattr(sys, 'frozen'): module_path = os.path.dirname(sys.executable) else: file_path = os.path.abspath(os.path.dirname(__file__)) module_path = os.path.dirname(os.path.dirname(file_path)) plugin_path = os.path.join(module_path, 'plugins') if os.path.isdir(plugin_path): self.plugin_paths.append(plugin_path) else: paths = settings.value('pluginPaths') if paths is None: self.plugin_paths = [] else: self.plugin_paths = paths if not settings.contains('selectionPath'): data_path = QDesktopServices.storageLocation( QDesktopServices.DataLocation) self.selection_path = os.path.join(data_path, 'selections') else: self.selection_path = settings.value('selectionPath') if not settings.contains('dataPath'): data_path = QDesktopServices.storageLocation( QDesktopServices.DataLocation) AnalysisPlugin.data_dir = os.path.join(data_path, 'data') else: AnalysisPlugin.data_dir = settings.value('dataPath') if not settings.contains('remoteScript'): if hasattr(sys, 'frozen'): path = os.path.dirname(sys.executable) else: file_path = os.path.abspath(os.path.dirname(__file__)) path = os.path.dirname(os.path.dirname(file_path)) path = os.path.join(path, 'bin') self.remote_script = os.path.join(path, 'spykeview_analysis') else: self.remote_script = settings.value('remoteScript') def set_initial_layout(self): self.setGeometry(self.x(), self.y(), 800, 750) self.navigationNeoDock.setVisible(True) self.removeDockWidget(self.filterDock) self.removeDockWidget(self.analysisDock) self.addDockWidget(Qt.RightDockWidgetArea, self.filterDock) self.addDockWidget(Qt.RightDockWidgetArea, self.analysisDock) self.tabifyDockWidget(self.filterDock, self.analysisDock) self.filterDock.setVisible(True) self.analysisDock.setVisible(True) self.consoleDock.setVisible(False) self.variableExplorerDock.setVisible(False) self.historyDock.setVisible(False) self.tabifyDockWidget(self.consoleDock, self.variableExplorerDock) self.tabifyDockWidget(self.variableExplorerDock, self.historyDock) def mode_switch(self): if self.is_db_mode(): self.switch_to_db_mode() if self.is_neo_mode(): self.switch_to_neo_mode() if self.console: self.console.interpreter.locals['current'] = self.provider def switch_to_db_mode(self): raise NotImplementedError() def switch_to_neo_mode(self): raise NotImplementedError() def _init_python_tab(self): # Console ns = {'current': self.provider, 'selections': self.selections} cmds = """ from __future__ import division import scipy as sp import numpy as np import matplotlib.pyplot as plt import guiqwt import guiqwt.pyplot as guiplt import guidata import spykeutils import spykeviewer plt.ion() """.split('\n') self.console = InternalShell(self.consoleDock, namespace=ns, multithreaded=False, commands=cmds, max_line_count=1000) self.consoleDock.setWidget(self.console) self.console.clear_terminal() # Variable browser self.browser = NamespaceBrowser(self.variableExplorerDock) self.browser.set_shellwidget(self.console) self.browser.setup(check_all=True, exclude_private=True, exclude_uppercase=False, exclude_capitalized=False, exclude_unsupported=True, excluded_names=[], truncate=False, minmax=False, collvalue=False, remote_editing=False, inplace=False, autorefresh=False) self.variableExplorerDock.setWidget(self.browser) # History self.history = CodeEditor(self.historyDock) self.history.setup_editor(linenumbers=False, language='py', scrollflagarea=False) self.history.setReadOnly(True) self.history.set_text('\n'.join(self.console.history)) self.history.set_cursor_position('eof') self.historyDock.setWidget(self.history) self.console.connect(self.console, SIGNAL("refresh()"), self._append_python_history) def _append_python_history(self): self.browser.refresh_table() self.history.append('\n' + self.console.history[-1]) self.history.set_cursor_position('eof') def is_neo_mode(self): return True def is_db_mode(self): return False @pyqtSignature("") def on_actionExit_triggered(self): self.close() def on_menuHelp_triggered(self): QMessageBox.about(self, 'How to navigate plots', 'Zoom:\tHold right mouse button' + '\nTranslate:\tHold middle mouse button' + '\nHome:\tClick middle mouse button' + '\n\nAxis synchronization does not work with some actions. ' + 'Zoom or translate changed plot to synchronize.') def on_menuSelections_mousePressed(self, event): if event.button() == Qt.LeftButton: action = self.menuSelections.actionAt(event.pos()) if action: selection = action.data() if selection: self.seldrag_start_pos = event.pos() self.seldrag_selection = selection else: self.seldrag_start_pos = None self.seldrag_selection = None self.seldrag_target = None QMenu.mousePressEvent(self.menuSelections, event) def on_menuSelections_mouseMoved(self, event): if event.buttons() & Qt.LeftButton and self.seldrag_start_pos: if ((event.pos() - self.seldrag_start_pos).manhattanLength() >= QApplication.startDragDistance()): drag = QDrag(self.menuSelections) data = QMimeData() data.setText(self.seldrag_selection.name) drag.setMimeData(data) drag.exec_() self.seldrag_start_pos = None self.seldrag_selection = None self.seldrag_target = None QMenu.mouseMoveEvent(self.menuSelections, event) def on_menuSelections_paint(self, event): QMenu.paintEvent(self.menuSelections, event) if self.seldrag_target: # Paint line where selection will be dropped p = QPainter() color = QPalette().color(self.menuSelections.foregroundRole()) pen = QPen(color, 2, Qt.SolidLine) p.begin(self.menuSelections) p.setPen(pen) rect = self.menuSelections.actionGeometry(self.seldrag_target) if self.seldrag_target_upper: p.drawLine(rect.topLeft(), rect.topRight()) else: p.drawLine(rect.bottomLeft(), rect.bottomRight()) p.end() def _menuSelections_pos_is_drop_target(self, pos): """ Return if selection can be dropped at this position and prepare information needed for drawing and dropping """ action = self.menuSelections.actionAt(pos) if not action or not action.data(): self.seldrag_target = None return False self.seldrag_target = action rect = self.menuSelections.actionGeometry(action) if pos.y() < rect.top() + rect.height() / 2: self.seldrag_target_upper = True else: self.seldrag_target_upper = False return True def on_menuSelections_dragEnter(self, event): event.setDropAction(Qt.MoveAction) if self._menuSelections_pos_is_drop_target(event.pos()): event.accept() else: event.ignore() QMenu.dragEnterEvent(self.menuSelections, event) def on_menuSelections_dragMoved(self, event): event.setDropAction(Qt.MoveAction) if self._menuSelections_pos_is_drop_target(event.pos()): event.accept() self.menuSelections.update() else: event.ignore() QMenu.dragMoveEvent(self.menuSelections, event) def on_menuSelections_drop(self, event): source = self.seldrag_selection target = self.seldrag_target.data() if source != target: self.selections.remove(source) target_index = self.selections.index(target) if not self.seldrag_target_upper: target_index += 1 self.selections.insert(target_index, source) self.populate_selection_menu() QMenu.dropEvent(self.menuSelections, event) def populate_selection_menu(self): self.menuSelections.clear() a = self.menuSelections.addAction('New') a.triggered.connect(self.on_selection_new) a = self.menuSelections.addAction('Clear') a.triggered.connect(self.on_selection_clear) self.menuSelections.addSeparator() for i, s in enumerate(self.selections): m = self.menuSelections.addMenu(s.name) m.menuAction().setData(s) a = m.addAction('Load') self.connect(a, SIGNAL('triggered()'), lambda sel=s:self.on_selection_load(sel)) a = m.addAction('Save') self.connect(a, SIGNAL('triggered()'), lambda sel=s:self.on_selection_save(sel)) a = m.addAction('Rename') self.connect(a, SIGNAL('triggered()'), lambda sel=s:self.on_selection_rename(sel)) a = m.addAction('Remove') self.connect(a, SIGNAL('triggered()'), lambda sel=s:self.on_selection_remove(sel)) def on_selection_load(self, selection): self.set_current_selection(selection.data_dict()) def on_selection_save(self, selection): i = self.selections.index(selection) self.selections[i] = self.provider_factory( self.selections[i].name, self) self.populate_selection_menu() def on_selection_clear(self): if QMessageBox.question(self, 'Confirmation', 'Do you really want to remove all selections?', QMessageBox.Yes | QMessageBox.No) == QMessageBox.No: return del self.selections[:] self.populate_selection_menu() def on_selection_rename(self, selection): (name, ok) = QInputDialog.getText(self, 'Edit selection name', 'New name:', QLineEdit.Normal, selection.name) if ok and name: selection.name = name self.populate_selection_menu() def on_selection_remove(self, selection): if QMessageBox.question(self, 'Confirmation', 'Do you really want to remove the selection "%s"?' % selection.name, QMessageBox.Yes | QMessageBox.No) == QMessageBox.No: return self.selections.remove(selection) self.populate_selection_menu() def on_selection_new(self): self.selections.append(self.provider_factory('Selection %d' % (len(self.selections) + 1), self)) self.populate_selection_menu() def serialize_selections(self): sl = list() # Selection list, current selection as first item sl.append(self.provider_factory('current', self).data_dict()) for s in self.selections: sl.append(s.data_dict()) return json.dumps(sl, sort_keys=True, indent=2) def save_selections_to_file(self, filename): f = open(filename, 'w') f.write(self.serialize_selections()) f.close() def load_selections_from_file(self, filename): try: f = open(filename, 'r') p = json.load(f) # First selection in file is current selection self.set_current_selection(p[0]) for s in p[1:]: self.add_selection(s) f.close() self.populate_selection_menu() except Exception, e: self.progress.done() QMessageBox.critical(self, 'Error loading selection', str(e).decode('utf8'))