def __init__(self, controller, parent = None): QMainWindow.__init__(self, parent) self.controller = controller self.modelsList = QTreeWidget() self.modelsList.setColumnCount(1) self.modelsList.setHeaderHidden(True) self.modelsList.currentItemChanged.connect(self._setModelFromItem) self.fieldsList = QTreeWidget() self.fieldsList.setColumnCount(1) self.fieldsList.setHeaderHidden(True) self.fieldsList.currentItemChanged.connect(self._setFieldFromItem) self.timesList = QTreeWidget() self.timesList.setColumnCount(1) self.timesList.setHeaderHidden(True) self.timesList.currentItemChanged.connect(self._setPlotTimeFromItem) modelsDock = QDockWidget(self.tr("Models")) modelsDock.setWidget(self.modelsList) self.addDockWidget(Qt.LeftDockWidgetArea, modelsDock) fieldsDock = QDockWidget(self.tr("Fields")) fieldsDock.setWidget(self.fieldsList) self.addDockWidget(Qt.RightDockWidgetArea, fieldsDock) timesDock = QDockWidget(self.tr("Times")) timesDock.setWidget(self.timesList) self.addDockWidget(Qt.LeftDockWidgetArea, timesDock) fileMenu = self.menuBar().addMenu(self.tr("&File")) quitAction = fileMenu.addAction(self.tr("E&xit")) quitAction.setShortcut(QKeySequence.Quit) quitAction.triggered.connect(self.close) dockMenu = self.menuBar().addMenu(self.tr("&Docks")) dockMenu.addAction(modelsDock.toggleViewAction()) dockMenu.addAction(fieldsDock.toggleViewAction()) dockMenu.addAction(timesDock.toggleViewAction()) work_area = WorkArea(controller) self.setCentralWidget(work_area)
def add_widget_into_main(parent): """add a widget into the main window of LabGuiMain create a QDock widget and store a reference to the widget """ mywidget = SimpleConnectWidget(parent = parent) #create a QDockWidget simpleconnectDockWidget = QDockWidget("Simple instrument console", parent) simpleconnectDockWidget.setObjectName("simpleConnectWidgetDockWidget") simpleconnectDockWidget.setAllowedAreas( Qt.LeftDockWidgetArea | Qt.RightDockWidgetArea) #fill the dictionnary with the widgets added into LabGuiMain parent.widgets['ConnectTerminalWidget'] = mywidget simpleconnectDockWidget.setWidget(mywidget) parent.addDockWidget(Qt.RightDockWidgetArea, simpleconnectDockWidget) #Enable the toggle view action parent.windowMenu.addAction(simpleconnectDockWidget.toggleViewAction()) simpleconnectDockWidget.hide()
def addDock(self, name, widget, area=Qt.RightDockWidgetArea, allowed_areas=Qt.AllDockWidgetAreas): dock_widget = QDockWidget(name) dock_widget.setObjectName("%sDock" % name) dock_widget.setWidget(widget) dock_widget.setAllowedAreas(allowed_areas) self.addDockWidget(area, dock_widget) self.__view_menu.addAction(dock_widget.toggleViewAction()) return dock_widget
def insertDockWidget(self, widget, name, area, addToggleViewAction): d = QDockWidget(name, self) d.setObjectName(name) d.setWidget(widget) self.addDockWidget(area, d) if addToggleViewAction: self.ui.menuShow_View.addAction(d.toggleViewAction()) return d
def __init__(self): super(MainWindow, self).__init__() self.setStyleSheet("QMainWindow::separator {background: lightGray; width: 2px}") self.tree_widget = SetupsView() self.eqn_widget = ResizableImage("latex/eqn.png", 100, .5, 2) self.outputs_dock_area = DockArea() file_menu = self.menuBar().addMenu("File") save_action = QAction("Save", self) save_action.triggered.connect(self.save_configuration) file_menu.addAction(save_action) load_action = QAction("Load", self) load_action.triggered.connect(self.load_configuration) file_menu.addAction(load_action) docks = [] view_menu = self.menuBar().addMenu("View") for dock_name in ["Project Manager", "Properties", "Equation"]: dock = QDockWidget(dock_name) dock.setObjectName(method_style(dock_name)) view_menu.addAction(dock.toggleViewAction()) docks.append(dock) self.tree_dock, self.props_dock, self.eqn_dock = docks self.tree_dock.setWidget(self.tree_widget) self.eqn_dock.setWidget(self.eqn_widget) placeholder = QLabel("No Item Selected") placeholder.setAlignment(Qt.AlignCenter) placeholder.setMinimumSize(200, 100) self.props_dock.setWidget(placeholder) self.setCentralWidget(self.outputs_dock_area) self.setCorner(Qt.BottomLeftCorner, Qt.LeftDockWidgetArea) self.setCorner(Qt.TopLeftCorner, Qt.LeftDockWidgetArea) self.addDockWidget(Qt.LeftDockWidgetArea, self.tree_dock) self.addDockWidget(Qt.LeftDockWidgetArea, self.props_dock) self.addDockWidget(Qt.TopDockWidgetArea, self.eqn_dock) self.tree_widget.add_setup() self.tree_widget.clicked.connect(self.set_props_widget) self.status_label = QLabel("") self.progress_bar = QProgressBar() self.statusBar().addWidget(self.status_label) self.statusBar().addWidget(self.progress_bar, 1) time_toolbar = self.addToolBar("Time") time_toolbar.addWidget(QLabel("Time")) self.time_slider = QSlider(Qt.Horizontal) time_toolbar.addWidget(self.time_slider) self.restoreGeometry(settings.value("geometry").toByteArray()) self.restoreState(settings.value("state").toByteArray(), __ui_version__)
def lauchGUI(self, WorkSpace, aCase, sobjXML, Args): """ mw.dockWidgetBrowser is the Browser of the CFD MainView """ log.debug("lauchGUI") from cs_gui import process_cmd_line if CFD_Code() == CFD_Saturne: from cs_package import package from code_saturne.Base.MainView import MainView elif CFD_Code() == CFD_Neptune: from nc_package import package from neptune_cfd.core.MainView import MainView if sobjXML == None: Title = "unnamed" else: Title = sobjXML.GetName() self.Workspace = WorkSpace pkg = package() case, splash = process_cmd_line(Args) mw = MainView(pkg, case, aCase) # Put the standard panel of the MainView inside a QDockWidget # in the SALOME Desktop aTitle = self.setWindowTitle_CFD(mw, aCase, Title) dsk = sgPyQt.getDesktop() dock = QDockWidget(aTitle) dock.setWidget(mw.frame) dock.setMinimumWidth(520) dsk.addDockWidget(Qt.RightDockWidgetArea, dock) dock.setVisible(True) dock.show() # Put the QTreeView of the MainView which is already inside a QDockWidget # in the SALOME Desktop BrowserTitle = aTitle + " Browser" mw.dockWidgetBrowser.setWindowTitle(BrowserTitle) dsk.addDockWidget(Qt.LeftDockWidgetArea, mw.dockWidgetBrowser) mw.dockWidgetBrowser.setVisible(True) mw.dockWidgetBrowser.show() mw.dockWidgetBrowser.raise_() dock.raise_() #Add Dock windows are managed by CFDGUI_Management class studyId = sgPyQt.getStudyId() aStudyCFD = aCase.GetFather() aCaseCFD = aCase xmlFileName = str(Title) _c_CFDGUI.set_d_CfdCases(studyId, dock, mw.dockWidgetBrowser, mw, aStudyCFD, aCaseCFD, xmlFileName, sobjXML) self.connect(dock, SIGNAL("visibilityChanged(bool)"), self.setdockWindowBrowserActivated) self.connect(mw.dockWidgetBrowser, SIGNAL("visibilityChanged(bool)"),self.setdockWindowActivated) self.connect(dock.toggleViewAction(), SIGNAL("toggled(bool)"), self.setdockWB) self.connect(mw.dockWidgetBrowser.toggleViewAction(), SIGNAL("toggled(bool)"), self.setdock) _c_CFDGUI.tabifyDockWindows(dsk, studyId) self.showDockWindows(studyId, xmlFileName, aCaseCFD.GetName(), aStudyCFD.GetName()) updateObjectBrowser() return mw
class OrbitPlotMainWindow(QMainWindow): """ the main window has three major widgets: current, orbit tabs and element editor. """ def __init__(self, parent=None, machines=[], **kwargs): QMainWindow.__init__(self, parent) self.iqtApp = kwargs.get("iqt", None) self.setIconSize(QSize(32, 32)) self.error_bar = True self._dlgOrbitCor = None # logging self.logdock = QDockWidget("Log") self.logdock.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum) textedit = QPlainTextEdit(self.logdock) self.logger = logging.getLogger(__name__) self.guilogger = logging.getLogger("aphla.gui") # the "aphla" include lib part logging. When the lib is inside # QThread, logging message will be sent to TextEdit which is cross # thread. # self.guilogger = logging.getLogger("aphla") handler = QTextEditLoggingHandler(textedit) self.guilogger.addHandler(handler) self.guilogger.setLevel(logging.INFO) self.logdock.setWidget(textedit) self.logdock.setAllowedAreas(Qt.BottomDockWidgetArea) self.logdock.setFeatures(QDockWidget.DockWidgetMovable | QDockWidget.DockWidgetClosable) self.logdock.setFloating(False) self.logdock.setMinimumHeight(20) self.logdock.setMaximumHeight(100) self.logdock.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum) self.logdock.resize(200, 60) #print self.logdock.sizeHint() self.addDockWidget(Qt.BottomDockWidgetArea, self.logdock) #print self.logdock.sizeHint() #print self.logdock.minimumSize() #print self.logdock.maximumSize() #self.logger.info("INFO") #self.logdock.setMinimumHeight(40) #self.logdock.setMaximumHeight(160) for msg in kwargs.get("infos", []): self.logger.info(msg) # dict of (machine, (lattice dict, default_lat, pvm)) self._mach = dict([(v[0], (v[1], v[2], v[3])) for v in machines]) for m, (lats, lat0, pvm) in self._mach.items(): self.logger.info( "machine '%s' initialized: [%s]" % (m, ", ".join([lat.name for k, lat in lats.items()]))) if pvm: for pv in pvm.dead(): self.logger.warn("'{0}' is disconnected.".format(pv)) ## DCCT current plot #self.dcct = DcctCurrentPlot() #self.dcct.setMinimumHeight(100) #self.dcct.setMaximumHeight(150) #t0 = time.time() #t = np.linspace(t0 - 8*3600*24, t0, 100) #self.dcct.curve.t = t #v = 500*np.exp((t[0] - t[:50])/(4*3600*24)) #self.dcct.curve.v = v.tolist()+v.tolist() #self.dcct.updatePlot() ## MDI area self.mdiarea = QMdiArea() self.connect(self.mdiarea, SIGNAL("subWindowActivated(QMdiSubWindow)"), self.updateMachineLatticeNames) self.mdiarea.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded) self.physics = ApOrbitPhysics(self.mdiarea, iqt=self.iqtApp) self.live_orbit = True self.setCentralWidget(self.mdiarea) #self._elemed = ElementPropertyTabs() #self.elemeditor = ElementEditorDock(parent=self) #self.elemeditor.setAllowedAreas(Qt.RightDockWidgetArea) #self.elemeditor.setFeatures(QDockWidget.DockWidgetMovable| # QDockWidget.DockWidgetClosable) #self.elemeditor.setFloating(False) #self.elemeditor.setEnabled(False) #self.elemeditor.setMinimumWidth(400) #self.elemeditor.setWidget(self._elemed) #self.elemeditor.show() #self.elemeditor.hide() #self.connect(self.elemeditor, # SIGNAL("elementChecked(PyQt_PyObject, bool)"), # self.physics.elementChecked) #self.addDockWidget(Qt.RightDockWidgetArea, self.elemeditor) self.createMenuToolBar() # the first machine is the default self.machBox.addItems([v for v in self._mach.keys()]) self.reloadLatticeNames(self.machBox.currentText()) self.connect(self.machBox, SIGNAL("currentIndexChanged(QString)"), self.reloadLatticeNames) # update at 1/2Hz self.dt, self.itimer = 1500, 0 #self.timerId = None self.timerId = self.startTimer(self.dt) self.vbpm = None self.statusBar().showMessage("Welcome") #self.initMachine("nsls2v2") #self._newVelemPlot("V2SR", aphla.machines.HLA_VBPM, 'x', # "H Orbit", c = None) #print "Thread started", self.machinit.isRunning() #self.newElementPlots("BPM", "x, y") #self.newElementPlot("BPM", "y") #self.newElementPlot("HCOR", "x") #self.newElementPlot("VCOR", "y") #self.newElementPlot("QUAD", "b1") #self.newElementPlot("SEXT", "b2") def updateMachineLatticeNames(self, wsub): i = self.machBox.findText(wsub.machlat[0]) self.machBox.setCurrentIndex(i) self.reloadLatticeNames(wsub.machlat[0]) def reloadLatticeNames(self, mach): self.latBox.clear() cur_mach = str(self.machBox.currentText()) lats, lat0, pvm = self._mach.get(cur_mach, ({}, None, None)) self.latBox.addItems([lat for lat in lats.keys()]) if lat0: i = self.latBox.findText(lat0.name) self.latBox.setCurrentIndex(i) def closeEvent(self, event): self.physics.close() self.mdiarea.closeAllSubWindows() event.accept() def createMenuToolBar(self): # # file menu # #self.machMenu = self.menuBar().addMenu("&Machines") #self.connect(self.machMenu, SIGNAL("aboutToShow()"), # self.updateMachMenu) self.openMenu = self.menuBar().addMenu("&Open") self.openMenu.addAction("New Plot ...", self.openNewPlot) self.openMenu.addAction("New Tune Plot", self.openTunePlot) self.openMenu.addAction("New BPM Plot", partial(self.newElementPlots, "BPM", "x,y")) self.openMenu.addAction("New HCOR Plot", partial(self.newElementPlots, "HCOR", "x")) self.openMenu.addAction("New VCOR Plot", partial(self.newElementPlots, "VCOR", "y")) self.openMenu.addSeparator() self.openMenu.addAction("Open ORM", self.loadOrm) self.openMenu.addSeparator() self.openMenu.addAction("Save Lattice ...", self.saveSnapshot) fileQuitAction = QAction(QIcon(":/file_quit.png"), "&Quit", self) fileQuitAction.setShortcut("Ctrl+Q") fileQuitAction.setToolTip("Quit the application") fileQuitAction.setStatusTip("Quit the application") #fileQuitAction.setIcon(Qt.QIcon(":/filequit.png")) self.connect(fileQuitAction, SIGNAL("triggered()"), self.close) self.openMenu.addAction(fileQuitAction) # view self.viewMenu = self.menuBar().addMenu("&View") mkmenu = QMenu("&Mark", self.viewMenu) for fam in ["BPM", "COR", "QUAD", "SEXT", "INSERTION"]: famAct = QAction(fam, self) famAct.setCheckable(True) self.connect(famAct, SIGNAL("toggled(bool)"), self.click_markfam) mkmenu.addAction(famAct) # # errorbar #viewErrorBarAction = QAction(QIcon(":/view_errorbar.png"), # "Errorbar", self) #viewErrorBarAction.setCheckable(True) #viewErrorBarAction.setChecked(True) #self.connect(viewErrorBarAction, SIGNAL("toggled(bool)"), # self.errorBar) # #zoomM = QMenu("Zoom", self.viewMenu) # # #drift_from_now = QAction("Drift from Now", self) #drift_from_now.setCheckable(True) #drift_from_now.setShortcut("Ctrl+N") #drift_from_golden = QAction("Drift from Golden", self) #drift_from_golden.setCheckable(True) #drift_from_none = QAction("None", self) #drift_from_none.setCheckable(True) #self.viewMenu.addAction(viewLiveAction) #self.viewMenu.addAction(viewSingleShotAction) #self.viewMenu.addSeparator() #self.viewMenu.addAction(drift_from_now) #self.viewMenu.addAction(drift_from_golden) #self.viewMenu.addAction(drift_from_none) #self.viewMenu.addAction(viewAutoScale) #self.viewMenu.addAction(viewErrorBarAction) #self.viewMenu.addSeparator() self.viewMenu.addMenu(mkmenu) #drift_group = QActionGroup(self) #drift_group.addAction(drift_from_none) #drift_group.addAction(drift_from_now) #drift_group.addAction(drift_from_golden) #drift_from_none.setChecked(True) sep = self.viewMenu.addSeparator() #sep.setText("Drift") #self.connect(drift_from_now, SIGNAL("triggered()"), self.setDriftNow) #self.connect(drift_from_none, SIGNAL("triggered()"), self.setDriftNone) #self.connect(drift_from_golden, SIGNAL("triggered()"), # self.setDriftGolden) #viewStyle = QMenu("Line Style", self.viewMenu) #for act in ["Increase Point Size", "Decrease Point Size", None, # "NoCurve", "Lines", "Sticks", None, # "Solid Line", "Dashed Line", "Dotted Line", None, # "Increase Line Width", "Decrease Line Width", None, # "NoSymbol", "Ellipse", "Rect", "Diamond", "Triangle", # "Cross", "XCross", "HLine", "VLine", # "Star1", "Star2", "Hexagon", None, # "Red", "Blue", "Green"]: # if act is None: # viewStyle.addSeparator() # else: # viewStyle.addAction(act, self.setPlotStyle) #self.viewMenu.addMenu(viewStyle) #self.viewMenu.addSeparator() #self.viewMenu.addAction(viewZoomOut15Action) #self.viewMenu.addAction(viewZoomIn15Action) #self.viewMenu.addAction(viewZoomAutoAction) #self.viewMenu.addSeparator() self.viewMenu.addAction("ORM SV", self.plotSVD) # a bug in PyQwt5 for datetime x-axis, waiting for Debian 7 #self.viewMenu.addAction(viewDcct) #for ac in self.viewMenu.actions(): ac.setDisabled(True) # self.controlMenu = self.menuBar().addMenu("&Tools") self.controlMenu.addAction(QIcon(":/control_choosebpm.png"), "En-/Disable BPM", partial(chooseElement, 'BPM')) self.controlMenu.addAction("En-/Disable COR", partial(chooseElement, 'COR')) #self.controlMenu.addAction(controlResetPvDataAction) self.controlMenu.addSeparator() self.controlMenu.addAction("Lattice Snapshot ...", self.openSnapshot) #self.controlMenu.addAction(controlZoomInPlot1Action) #self.controlMenu.addAction(controlZoomOutPlot1Action) #self.controlMenu.addAction(controlZoomInPlot2Action) #self.controlMenu.addAction(controlZoomOutPlot2Action) self.controlMenu.addSeparator() self.controlMenu.addAction("Correct Hor. orbit", partial(aphla.correctOrbit, plane="H")) self.controlMenu.addAction("Correct Vert. orbit", partial(aphla.correctOrbit, plane="V")) self.controlMenu.addAction(QIcon(":/control_corrorbit.png"), "Correct orbit", partial(aphla.correctOrbit, plane="HV")) #steer_orbit.setDisabled(True) self.controlMenu.addAction("Local Bump ...", self.createLocalBump) self.controlMenu.addAction("Element Editor ...", self.showElementEditor) self.controlMenu.addSeparator() self.controlMenu.addAction("meas Beta", self.physics.measBeta) self.controlMenu.addAction("meas Dispersion", self.physics.measDispersion) self.controlMenu.addAction("beam based alignment", self.runBba) #for ac in self.controlMenu.actions(): ac.setDisabled(True) # Window self.windowMenu = self.menuBar().addMenu("&Windows") #self.windowMenu.addAction(self.elemeditor.toggleViewAction()) self.windowMenu.addAction(self.logdock.toggleViewAction()) #viewDcct = QAction("Beam Current", self) #viewDcct.setCheckable(True) #viewDcct.setChecked(True) #self.connect(viewDcct, SIGNAL("toggled(bool)"), self.dcct.setVisible) #self.windowMenu.addAction(viewDcct) self.windowMenu.addSeparator() self.windowMenu.addAction("Cascade", self.mdiarea.cascadeSubWindows) self.windowMenu.addAction("Tile", self.mdiarea.tileSubWindows) self.windowMenu.addAction("Tile Horizontally", self.tileSubWindowsHorizontally) # "ctrl+page up", "ctrl+page down" self.windowMenu.addAction("Previous", self.mdiarea.activatePreviousSubWindow, "Ctrl+Left") self.windowMenu.addAction("Next", self.mdiarea.activateNextSubWindow, "Ctrl+Right") self.windowMenu.addSeparator() # debug self.debugMenu = self.menuBar().addMenu("&Debug") self.debugMenu.addAction("_Reset Correctors_", self._reset_correctors) self.debugMenu.addAction("_Reset Quadrupoles_", self._reset_quadrupoles) self.debugMenu.addAction("_Random V Kick_", self._random_vkick) self.debugMenu.addAction("_Random H Kick_", self._random_hkick) #for ac in self.debugMenu.actions(): ac.setDisabled(True) # help self.helpMenu = self.menuBar().addMenu("&Help") self.helpMenu.addAction("About mleap", self.showAbout) #toolbar machToolBar = self.addToolBar("Machines") self.machBox = QtGui.QComboBox() self.latBox = QtGui.QComboBox() #self.connect(self.latBox, SIGNAL("currentIndexChanged(QString)"), # self.__setLattice) machToolBar.addWidget(self.machBox) machToolBar.addWidget(self.latBox) #toolbar = QToolBar(self) #self.addToolBar(toolbar) #fileToolBar = self.addToolBar("File") #fileToolBar.setObjectName("FileToolBar") #fileToolBar.addAction(fileQuitAction) # viewToolBar1 = self.addToolBar("Live View") #viewToolBar.setObjectName("ViewToolBar") #viewToolBar.addAction(viewZoomOut15Action) #viewToolBar.addAction(viewZoomIn15Action) #viewToolBar.addAction(viewZoomAutoAction) #viewToolBar1.addAction(viewLiveAction) #viewToolBar1.addAction(viewSingleShotAction) #viewToolBar1.addSeparator() viewToolBar1.addAction(QIcon(":/new_bpm.png"), "Orbits", partial(self.newElementPlots, "BPM", "x,y")) viewToolBar1.addAction(QIcon(":/new_cor.png"), "Correctors", partial(self.newElementPlots, "COR", "x,y")) viewToolBar1.addAction(QIcon(":/new_quad.png"), "Quadrupoles", partial(self.newElementPlots, "QUAD", "b1")) viewToolBar1.addAction(QIcon(":/new_sext.png"), "Sextupoles", partial(self.newElementPlots, "SEXT", "b2")) #viewToolBar.addAction(viewErrorBarAction) #viewToolBar.addAction(QWhatsThis.createAction(self)) #viewToolBar2 = self.addToolBar("Scale Plot") #zoomActions = [(":/view_zoom_xy.png", "Fit", self.scalePlot), # (None, None, None), # (":/view_zoom_y.png", "Fit In Y", self.scalePlot), # (":/view_zoomin_y.png", "Zoom In Y", self.scalePlot), # (":/view_zoomout_y.png", "Zoom Out Y", self.scalePlot), # (":/view_move_up.png", "Move Up", self.scalePlot), # (":/view_move_down.png", "Move Down", self.scalePlot), # (None, None, None), # (":/view_zoom_x.png", "Fit In X", self.scalePlot), # (":/view_zoomin_x.png", "Zoom In X", self.scalePlot), # (":/view_zoomout_x.png", "Zoom Out X", self.scalePlot), # (":/view_move_left.png", "Move Left", self.scalePlot), # (":/view_move_right.png", "Move Right", self.scalePlot), # ] #for ico,name,hdl in zoomActions: # if hdl is None: continue # viewToolBar2.addAction(QIcon(ico), name, hdl) controlToolBar = self.addToolBar("Control") controlToolBar.addAction(QIcon(":/control_orbitcor.png"), "Correct Orbit", aphla.correctOrbit) controlToolBar.addAction(QIcon(":/control_localbump.png"), "Local Bump ...", self.createLocalBump) #controlToolBar.addAction(controlResetPvDataAction) def showAbout(self): QMessageBox.about( self, self.tr("mleap"), (self.tr("""<b>Machine/Lattice Editor And Plotter</b> v %1 <p>Copyright © Lingyun Yang, BNL, 2013-2014. All rights reserved. <p>This application can be used to perform high level accelerator controls. <p>Python %2 - Qt %3 - PyQt %4 on %5""").arg(aphla.version.version).arg( platform.python_version()).arg(QtCore.QT_VERSION_STR).arg( QtCore.PYQT_VERSION_STR).arg(platform.system()))) def showElementEditor(self): mach, lat = self.getCurrentMachLattice() ed = ElementEditor(lat, parent=self) ed.setWindowFlags(Qt.Window) ed.setAttribute(Qt.WA_DeleteOnClose) ed.show() def getCurrentMachLattice(self, cadata=False): """return the current machine name and lattice object""" mach = str(self.machBox.currentText()) latname = str(self.latBox.currentText()) lat_dict, lat0, pvm = self._mach[mach] if not cadata: return mach, lat_dict[latname] else: return mach, lat_dict[latname], pvm def newElementPlots(self, elem, fields, **kw): self.logger.info("new plots: %s %s" % (elem, fields)) _mach, _lat, _pvm = self.getCurrentMachLattice(cadata=True) mach, lat = kw.get("machlat", (_mach, _lat)) handle = kw.get("handle", "readback") elems = lat.getElementList(elem) x, pvs = [], [] field_list = re.findall(r'[^ ,]+', fields) for fld in field_list: si, pvsi = [], [] for e in elems: if not e.isEnabled(): continue epv = e.pv(field=fld, handle=handle) if not epv: continue pvsi.append(epv[0]) si.append(e.sb) x.append(si) pvs.append(pvsi) if not pvs: self.logger.error("no data found for elements '{0}' " "and field '{1}'".format(elem, field)) return p = ApMdiSubPlot(pvs=pvs, x=x, labels=["%s.%s" % (elem, fld) for fld in field_list], magprof=lat.getBeamlineProfile(), iqt=self.iqtApp, **kw) #QObject.installEventFilter(p.aplot) #p.data = ManagedPvData(pvm, s, pvs, element=elemnames, # label="{0}.{1}".format(elem,field)) p.setAttribute(Qt.WA_DeleteOnClose) str_elem = "{0}".format(elem) if len(str_elem) > 12: str_elem = str_elem[:9] + "..." str_field = "{0}".format(fields) if len(str_field) > 12: str_field = str_field[:9] + "..." p.setWindowTitle("[%s.%s] %s %s" % (mach, lat.name, str_elem, str_field)) self.connect(p, SIGNAL("elementSelected(PyQt_PyObject)"), self.elementSelected) self.connect(p, SIGNAL("destroyed()"), self.subPlotDestroyed) #p.updatePlot() # set the zoom stack #p.aplot.setErrorBar(self.error_bar) #p.wid.autoScaleXY() #p.aplot.replot() self.mdiarea.addSubWindow(p) #print "Show" p.show() ##print "Enable the buttons" #if len(self.mdiarea.subWindowList()) > 0: # self.elemeditor.setEnabled(True) def subPlotDestroyed(self): #if len(self.mdiarea.subWindowList()) == 0: # self.elemeditor.setEnabled(False) pass def loadOrm(self): fileName = QtGui.QFileDialog.getOpenFileName( self, "Open Orbit Response Matrix", "", "ORM Files (*.h5 *.hdf5);;Text File (*.txt);;All Files(*)") fileName = str(fileName) try: m = np.loadtxt(fileName) except: QMessageBox.critical(self, "Abort", "Invalid matrix data") return mach, lat = self.getCurrentMachLattice() # assuming we already have the PV, name, field but just want to # replace the matrix elements. assert np.shape(m) == np.shape(lat.ormdata.m) nx, ny = np.shape(lat.ormdata.m) for i in range(nx): for j in range(ny): lat.ormdata.m[i, j] = m[i, j] def saveSnapshot(self): latdict = dict([(k, v[0]) for k, v in self._mach.items()]) mach, lat = self.getCurrentMachLattice() snapdlg = SaveSnapshotDialog(latdict, mach) snapdlg.exec_() def saveLatSnapshot(self): mach, lat = self.getCurrentMachLattice() dpath = self._prepare_parent_dirs(mach) if not dpath: QMessageBox.warning(self, "Abort", "Aborted") return dt = datetime.datetime.now() fname = os.path.join( dpath, dt.strftime("snapshot_%d_%H%M%S_") + lat.name + ".hdf5") fileName = QtGui.QFileDialog.getSaveFileName( self, "Save Lattice Snapshot Data", fname, "Data Files (*.h5 *.hdf5);;All Files(*)") fileName = str(fileName) if not fileName: return aphla.catools.save_lat_epics(fileName, lat, mode='a') self.logger.info("snapshot created '%s'" % fileName) def saveMachSnapshot(self): mach, lat = self.getCurrentMachLattice() dpath = self._prepare_parent_dirs(mach) if not dpath: QMessageBox.warning(self, "Abort", "Aborted") return dt = datetime.datetime.now() fname = os.path.join(dpath, dt.strftime("snapshot_%d_%H%M%S.hdf5")) fileName = QtGui.QFileDialog.getSaveFileName( self, "Save Lattice Snapshot Data", fname, "Data Files (*.h5 *.hdf5);;All Files(*)") if not fileName: return fileName = str(fileName) import h5py f = h5py.File(str(fileName), 'w') f.close() self.logger.info("clean snapshot file created: '%s'" % fileName) for k, lat in self._mach[mach][0].items(): aphla.catools.save_lat_epics(fileName, lat, mode='a') self.logger.info("lattice snapshot appended for '%s'" % lat.name) def openSnapshot(self): #self.logger.info("loading snapshot?") latdict = dict([(k, v[0]) for k, v in self._mach.items()]) mach, lat = self.getCurrentMachLattice() lv = LatSnapshotMain(self, latdict, mach, self.logger) lv.setWindowFlags(Qt.Window) #self.logger.info("initialized") #lv.loadLatSnapshotH5() lv.exec_() def openTunePlot(self): mach, lat = self.getCurrentMachLattice() nu = lat.getElementList('tune') pvs = [(e.pv(field="x", handle="readback")[0], e.pv(field="y", handle="readback")[0]) for e in nu] labels = [e.name for e in nu] twiss = lat.getElementList("VA") pvs.extend([(e.pv(field="nux", handle="readback")[0], e.pv(field="nuy", handle="readback")[0]) for e in twiss]) labels.extend([e.name for e in twiss]) p = ApMdiSubPlot(pvs=pvs, labels=labels, dtype="Tunes") #QObject.installEventFilter(p.aplot) #p.data = ManagedPvData(pvm, s, pvs, element=elemnames, # label="{0}.{1}".format(elem,field)) p.setAttribute(Qt.WA_DeleteOnClose) p.setWindowTitle("[%s.%s] Tunes" % (mach, lat.name)) self.connect(p, SIGNAL("elementSelected(PyQt_PyObject)"), self.elementSelected) self.connect(p, SIGNAL("destroyed()"), self.subPlotDestroyed) #p.updatePlot() # set the zoom stack #p.aplot.setErrorBar(self.error_bar) #p.wid.autoScaleXY() #p.aplot.replot() self.mdiarea.addSubWindow(p) #print "Show" p.show() def openNewPlot(self): mach, lat = self.getCurrentMachLattice() fl = QtGui.QFormLayout() fl.addRow("Machine", QtGui.QLabel("%s" % mach)) fl.addRow("Lattice", QtGui.QLabel("%s" % lat.name)) elem, fld = QtGui.QLineEdit(), QtGui.QLineEdit() fl.addRow("Elements", elem) fl.addRow("Field", fld) dtype = QtGui.QComboBox() for tx in ["Array", "Waveform", "Time Series"]: dtype.addItem(tx) fl.addRow("Data Type", dtype) dlg = QtGui.QDialog() bx = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Ok | QtGui.QDialogButtonBox.Cancel) self.connect(bx, SIGNAL("accepted()"), dlg.accept) self.connect(bx, SIGNAL("rejected()"), dlg.reject) h1 = QtGui.QHBoxLayout() h1.addStretch() h1.addWidget(bx) v1 = QtGui.QVBoxLayout() v1.addLayout(fl) v1.addLayout(h1) dlg.setLayout(v1) if dlg.exec_(): self.newElementPlots(str(elem.text()), str(fld.text()), machlat=(mach, lat), dtype=str(dtype.currentText())) def click_markfam(self, on): famname = self.sender().text() mks = [] mach, lat = self.getCurrentMachLattice() # need to convert to python str for elem in lat.getElementList(str(famname)): if elem.family != famname: continue if elem.virtual: continue mks.append([elem.name, 0.5 * (elem.sb + elem.se)]) for w in self.mdiarea.subWindowList(): w.setMarkers(mks, on) #print self._machlat.keys() def _reset_correctors(self): self.logger.info("reset correctors") aphla.hlalib._reset_trims() def _reset_quadrupoles(self): self.logger.info("reset quadrupoles") aphla.hlalib._reset_quad() def _random_hkick(self): mach, lat = self.getCurrentMachLattice() hcors = lat.getElementList('HCOR') for k in range(3): i = np.random.randint(len(hcors)) self.logger.info("Setting {0}/{1} HCOR".format(i, len(hcors))) hcors[i].x += np.random.rand() * 2e-6 def _random_vkick(self): mach, lat = self.getCurrentMachLattice() cors = lat.getElementList('VCOR') for k in range(3): i = np.random.randint(len(cors)) cors[i].y += np.random.rand() * 1e-6 self.logger.info("increased kicker '{0}' by 1e-7 ({1} {2})".format( cors[i].name, cors[i].y, cors[i].getUnit('y', None))) def viewDcctPlot(self, on): self.dcct.setVisible(on) def liveData(self, on): """Switch on/off live data taking""" self.live_orbit = on def scalePlot(self): w = self.mdiarea.currentSubWindow() if not w: return st, p = self.sender().text(), w.aplot if st == "Fit": p.scaleXBottom() p.scaleYLeft() # a hack bound = p.curvesBound() p.zoomer1.setZoomStack([bound]) elif st == "Fit In Y": p.scaleYLeft() elif st == "Zoom In Y": p.scaleYLeft(1. / 1.5) elif st == "Zoom Out Y": p.scaleYLeft(1.5) elif st == "Move Up": p.moveCurves(Qwt.QwtPlot.yLeft, 0.8) elif st == "Move Down": p.moveCurves(Qwt.QwtPlot.yLeft, -0.8) elif st == "Fit In X": p.scaleXBottom() elif st == "Zoom In X": p.scaleXBottom(1.0 / 1.5) elif st == "Zoom Out X": p.scaleXBottom(1.5) elif st == "Move Left": p.moveCurves(Qwt.QwtPlot.xBottom, 0.8) elif st == "Move Right": p.moveCurves(Qwt.QwtPlot.xBottom, -0.8) else: self.logger.error("unknow action '{0}'".format(st)) def getVisibleRange(self): w = self.mdiarea.currentSubWindow() if not w: mach, lat = self.getCurrentMachLattice() self.logger.warn( "no active plot, use full range of {0}.{1}".format( mach, lat.name)) return lat.getLocationRange() else: return w.currentXlim() def getVisibleElements(self, elemname, sb=None, se=None): w = self.mdiarea.currentSubWindow() mach, lat = self.getCurrentMachLattice() elems = lat.getElementList(elemname) if sb is not None: elems = [e for e in elems if e.sb >= sb] if se is not None: elems = [e for e in elems if e.se <= se] self.logger.info("searching for '{0}' in range [{1}, {2}]".format( elemname, sb, se)) return elems def timerEvent(self, e): if e.timerId() != self.timerId: return #if not self.elemeditor.isHidden(): # self.elemeditor.updateModelData() #if self.live_orbit: # self.itimer += 1 # #self.updatePlots() # #self.updateStatus() # for w in self.mdiarea.subWindowList(): # if not isinstance(w, ApMdiSubPlot): continue # if not w.live: continue # w.updatePlot() # self.statusBar().showMessage("plot updated: {0}".format( # time.strftime("%F %T"))) #else: # self.statusBar().showMessage("live update disabled") def singleShot(self): for w in self.mdiarea.subWindowList(): if not isinstance(w, ApMdiSubPlot): continue w.updatePlot() self.statusBar().showMessage("plot updated: {0}".format( time.strftime("%F %T"))) def elementSelected(self, elems): """this action is ignored""" mach, lat, elemnames = elems #_lat = self._machlat[mach][lat] self.logger.info("element selected") #elemobjs = _lat.getElementList(elemnames) #self._elemed.addElements(elemobjs) def activeOrbitPlot(self, field): mach = str(self.machBox.currentText()) lat = str(self.latBox.currentText()) for w in self.mdiarea.subWindowList(): #print w.machine(), w.lattice(), w.data.yfield if not isinstance(w, ApMdiSubPlot): continue if w.machine() != mach: continue if w.lattice() != lat: continue if w.data.yfield != field: continue return w return None def createLocalBump_(self): """create local bump""" if self._dlgOrbitCor is None: bpms = ap.getElements("BPM") cors = ap.getElements("COR") self._dlgOrbitCor = OrbitCorrDlg(bpms, cors) #corbitdlg.resize(600, 500) self._dlgOrbitCor.setWindowTitle("Create Local Bump") self._dlgOrbitCor.show() self._dlgOrbitCor.raise_() self._dlgOrbitCor.activateWindow() def createLocalBump(self): """create local bump""" bpms = ap.getElements("BPM") cors = ap.getElements("COR") dlgOrbitCor = OrbitCorrDlg(bpms, cors, parent=self) #corbitdlg.resize(600, 500) dlgOrbitCor.setWindowTitle("Create Local Bump") dlgOrbitCor.show() dlgOrbitCor.raise_() dlgOrbitCor.activateWindow() dlgOrbitCor.setAttribute(Qt.WA_DeleteOnClose) def runBba(self): mach, lat = self.getCurrentMachLattice() bpms = [ e for e in lat.getElementList('BPM') if e not in self.physics.deadelems ] self.physics.runBba(bpms) def plotSVD(self): mach, lat = self.getCurrentMachLattice() if not lat.ormdata: QMessageBox.critical(self, "ORM SVD", "machine '%s' ORM data is not available" % \ mach, QMessageBox.Ok) return m, brec, trec = lat.ormdata.getMatrix(None, None, full=False, ignore=self.getDeadElements()) U, s, V = np.linalg.svd(m, full_matrices=True) #print np.shape(s), s self.sp = ApSvdPlot(s) self.sp.show() def tileSubWindowsHorizontally(self): pos = QtCore.QPoint(0, 0) subwins = self.mdiarea.subWindowList() for w in subwins: height = self.mdiarea.height() / len(subwins) rect = QtCore.QRect(0, 0, self.mdiarea.width(), height) w.setGeometry(rect) w.move(pos) pos.setY(pos.y() + w.height())
class MikiWindow(QMainWindow): def __init__(self, settings, parent=None): super(MikiWindow, self).__init__(parent) self.setObjectName("mikiWindow") self.settings = settings self.notePath = settings.notePath lockPath = os.path.join(settings.notebookPath, '.mikidown_lock') if not os.path.exists(lockPath): self.lockPathFH = os.open(lockPath, os.O_CREAT | os.O_EXCL | os.O_RDWR) ################ Setup core components ################ self.notesTree = MikiTree(self) self.quickNoteNav = QLineEdit() self.notesTab = QWidget() self.completer = SlashPleter() self.completer.setModel(self.notesTree.model()) self.quickNoteNav.setCompleter(self.completer) self.notesTree.setObjectName("notesTree") self.initTree(self.notePath, self.notesTree) self.notesTree.sortItems(0, Qt.AscendingOrder) self.ix = None self.setupWhoosh() self.viewedList = QToolBar(self.tr('Recently Viewed'), self) self.viewedList.setIconSize(QSize(16, 16)) self.viewedList.setToolButtonStyle(Qt.ToolButtonTextBesideIcon) self.viewedListActions = [] self.noteSplitter = QSplitter(Qt.Horizontal) self.dockIndex = QDockWidget(self.tr("Index")) self.dockSearch = QDockWidget(self.tr("Search")) self.searchEdit = QLineEdit() self.searchView = MikiSearch(self) self.searchTab = QWidget() self.dockToc = QDockWidget(self.tr("TOC")) self.tocTree = TocTree() self.dockAttachment = QDockWidget(self.tr("Attachment")) self.attachmentView = AttachmentView(self) self.notesEdit = MikiEdit(self) self.notesEdit.setObjectName(self.tr("notesEdit")) self.loadHighlighter() self.notesView = MikiView(self) self.findBar = QToolBar(self.tr('Find'), self) self.findBar.setFixedHeight(30) self.findEdit = QLineEdit(self.findBar) self.checkBox = QCheckBox(self.tr('Match case'), self.findBar) self.statusBar = QStatusBar(self) self.statusLabel = QLabel(self) self.altPressed = False ################ Setup actions ################ self.actions = dict() self.setupActions() ################ Setup mainwindow ################ self.setupMainWindow() # show changelogs after upgrade mikidown if self.settings.version < __version__ or Mikibook.settings.value( "version", defaultValue="0") < __version__: self.changelogHelp() self.settings.qsettings.setValue("version", __version__) Mikibook.settings.setValue("version", __version__) def loadHighlighter(self): fnt = Mikibook.settings.value('editorFont', defaultValue=None) fntsize = Mikibook.settings.value('editorFontSize', type=int, defaultValue=12) header_scales_font = Mikibook.settings.value('headerScaleFont', type=bool, defaultValue=True) if fnt is not None: self.notesEdit.setFontFamily(fnt) self.notesEdit.setFontPointSize(fntsize) h = MikiHighlighter(parent=self.notesEdit, scale_font_sizes=header_scales_font) tw = Mikibook.settings.value('tabWidth', type=int, defaultValue=4) qfm = QFontMetrics(h.patterns[0][1].font()) self.notesEdit.setTabStopWidth(tw * qfm.width(' ')) def setupActions(self): # Global Actions actTabIndex = self.act(self.tr('Switch to Index Tab'), lambda: self.raiseDock(self.dockIndex), self.tr('Ctrl+Shift+I')) actTabSearch = self.act(self.tr('Switch to Search Tab'), lambda: self.raiseDock(self.dockSearch), self.tr('Ctrl+Shift+F')) self.addAction(actTabIndex) self.addAction(actTabSearch) ################ Menu Actions ################ # actions in menuFile actionNewPage = self.act(self.tr('&New Page...'), self.notesTree.newPage, QKeySequence.New) self.actions.update(newPage=actionNewPage) actionNewSubpage = self.act(self.tr('New Sub&page...'), self.notesTree.newSubpage, self.tr('Ctrl+Shift+N')) self.actions.update(newSubpage=actionNewSubpage) actionImportPage = self.act(self.tr('&Import Page...'), self.importPage) self.actions.update(importPage=actionImportPage) actionNBSettings = self.act(self.tr('Notebook Set&tings...'), self.notebookSettings) self.actions.update(NBSettings=actionNBSettings) actionMDSettings = self.act(self.tr('&Mikidown Settings...'), self.mikidownSettings) self.actions.update(MDSettings=actionMDSettings) actionOpenNotebook = self.act(self.tr('&Open Notebook...'), self.openNotebook, QKeySequence.Open) self.actions.update(openNotebook=actionOpenNotebook) actionReIndex = self.act(self.tr('Re-index'), self.reIndex) self.actions.update(reIndex=actionReIndex) actionSave = self.act(self.tr('&Save'), self.saveCurrentNote, QKeySequence.Save) actionSave.setEnabled(False) self.actions.update(save=actionSave) actionSaveAs = self.act(self.tr('Save &As...'), self.saveNoteAs, QKeySequence.SaveAs) self.actions.update(saveAs=actionSaveAs) actionHtml = self.act(self.tr('to &HTML'), self.notesEdit.saveAsHtml) self.actions.update(html=actionHtml) actionPrint = self.act(self.tr('&Print'), self.printNote, QKeySequence.Print) self.actions.update(print_=actionPrint) actionRenamePage = self.act(self.tr('&Rename Page...'), self.notesTree.renamePage, 'F2') self.actions.update(renamePage=actionRenamePage) actionDelPage = self.act(self.tr('&Delete Page'), self.notesTree.delPageWrapper, QKeySequence.Delete) self.actions.update(delPage=actionDelPage) actionQuit = self.act(self.tr('&Quit'), self.close, QKeySequence.Quit) actionQuit.setMenuRole(QAction.QuitRole) self.actions.update(quit=actionQuit) # actions in menuEdit actionUndo = self.act(self.tr('&Undo'), lambda: self.notesEdit.undo(), QKeySequence.Undo) actionUndo.setEnabled(False) self.notesEdit.undoAvailable.connect(actionUndo.setEnabled) self.actions.update(undo=actionUndo) actionRedo = self.act(self.tr('&Redo'), lambda: self.notesEdit.redo(), QKeySequence.Redo) actionRedo.setEnabled(False) self.notesEdit.redoAvailable.connect(actionRedo.setEnabled) self.actions.update(redo=actionRedo) actionFindText = self.act(self.tr('&Find Text'), self.findBar.setVisible, QKeySequence.Find, True) self.actions.update(findText=actionFindText) actionFindRepl = self.act(self.tr('Find and Replace'), FindReplaceDialog(self.notesEdit).open, QKeySequence.Replace) self.actions.update(findRepl=actionFindRepl) actionFind = self.act(self.tr('Next'), self.findText, QKeySequence.FindNext) self.actions.update(find=actionFind) actionFindPrev = self.act(self.tr('Previous'), lambda: self.findText(back=True), QKeySequence.FindPrevious) self.actions.update(findPrev=actionFindPrev) actionSortLines = self.act(self.tr('&Sort Lines'), self.sortLines) self.actions.update(sortLines=actionSortLines) actionQuickNav = self.act(self.tr("&Quick Open Note"), self.quickNoteNav.setFocus, self.tr('Ctrl+G')) self.addAction(actionQuickNav) actionInsertImage = self.act(self.tr('&Insert Attachment'), self.notesEdit.insertAttachmentWrapper, self.tr('Ctrl+I')) actionInsertImage.setEnabled(False) self.actions.update(insertImage=actionInsertImage) # actions in menuView QIcon.setThemeName( Mikibook.settings.value('iconTheme', QIcon.themeName())) #print(QIcon.themeName()) actionEdit = self.act(self.tr('Edit'), self.edit, self.tr('Ctrl+E'), True, QIcon.fromTheme('document-edit'), self.tr('Edit mode (Ctrl+E)')) self.actions.update(edit=actionEdit) actionSplit = self.act(self.tr('Split'), self.liveView, self.tr('Ctrl+R'), True, QIcon.fromTheme('view-split-left-right'), self.tr('Split mode (Ctrl+R)')) self.actions.update(split=actionSplit) actionFlipEditAndView = self.act(self.tr('Flip Edit and View'), self.flipEditAndView) actionFlipEditAndView.setEnabled(False) self.actions.update(flipEditAndView=actionFlipEditAndView) #actionLeftAndRight = self.act( # self.tr('Split into Left and Right'), trig=self.leftAndRight) #actionUpAndDown = self.act( # self.tr('Split into Up and Down'), trig=self.upAndDown) # self.actionLeftAndRight.setEnabled(False) # self.actionUpAndDown.setEnabled(False) # actions in menuHelp actionReadme = self.act(self.tr('README'), self.readmeHelp) self.actions.update(readme=actionReadme) actionChangelog = self.act(self.tr('Changelog'), self.changelogHelp) self.actions.update(changelog=actionChangelog) actionAboutQt = self.act(self.tr('About Qt'), qApp.aboutQt) self.actions.update(aboutQt=actionAboutQt) def setupMainWindow(self): self.resize(800, 600) screen = QDesktopWidget().screenGeometry() size = self.geometry() self.move((screen.width() - size.width()) / 2, (screen.height() - size.height()) / 2) self.setWindowTitle('{} - {}'.format(self.settings.notebookName, __appname__)) self.viewedList.setFixedHeight(25) self.noteSplitter.addWidget(self.notesEdit) self.noteSplitter.addWidget(self.notesView) mainSplitter = QSplitter(Qt.Vertical) mainSplitter.setChildrenCollapsible(False) mainSplitter.addWidget(self.viewedList) mainSplitter.addWidget(self.noteSplitter) mainSplitter.addWidget(self.findBar) self.setCentralWidget(mainSplitter) self.searchEdit.returnPressed.connect(self.searchNote) self.quickNoteNav.returnPressed.connect(self.openFuncWrapper) searchLayout = QVBoxLayout() searchLayout.addWidget(self.searchEdit) searchLayout.addWidget(self.searchView) self.searchTab.setLayout(searchLayout) indexLayout = QVBoxLayout(self.notesTab) indexLayout.addWidget(self.quickNoteNav) indexLayout.addWidget(self.notesTree) self.dockIndex.setObjectName("Index") self.dockIndex.setWidget(self.notesTab) self.dockSearch.setObjectName("Search") self.dockSearch.setWidget(self.searchTab) self.dockToc.setObjectName("TOC") self.dockToc.setWidget(self.tocTree) self.dockAttachment.setObjectName("Attachment") self.dockAttachment.setWidget(self.attachmentView) self.setDockOptions(QMainWindow.VerticalTabs) self.addDockWidget(Qt.LeftDockWidgetArea, self.dockIndex) self.addDockWidget(Qt.LeftDockWidgetArea, self.dockSearch) self.addDockWidget(Qt.LeftDockWidgetArea, self.dockToc) self.addDockWidget(Qt.LeftDockWidgetArea, self.dockAttachment) self.tabifyDockWidget(self.dockIndex, self.dockSearch) self.tabifyDockWidget(self.dockSearch, self.dockToc) self.tabifyDockWidget(self.dockToc, self.dockAttachment) self.setTabPosition(Qt.LeftDockWidgetArea, QTabWidget.North) self.dockIndex.raise_() # Put dockIndex on top of the tab stack menuBar = QMenuBar(self) self.setMenuBar(menuBar) menuFile = menuBar.addMenu(self.tr('&File')) menuEdit = menuBar.addMenu(self.tr('&Edit')) menuView = menuBar.addMenu(self.tr('&View')) menuHelp = menuBar.addMenu(self.tr('&Help')) # menuFile menuFile.addAction(self.actions['newPage']) menuFile.addAction(self.actions['newSubpage']) menuFile.addAction(self.actions['NBSettings']) menuFile.addAction(self.actions['MDSettings']) menuFile.addAction(self.actions['importPage']) menuFile.addAction(self.actions['openNotebook']) menuFile.addAction(self.actions['reIndex']) menuFile.addSeparator() menuFile.addAction(self.actions['save']) menuFile.addAction(self.actions['saveAs']) menuFile.addAction(self.actions['print_']) menuExport = menuFile.addMenu(self.tr('&Export')) menuExport.addAction(self.actions['html']) menuFile.addSeparator() menuFile.addAction(self.actions['renamePage']) menuFile.addAction(self.actions['delPage']) menuFile.addSeparator() menuFile.addAction(self.actions['quit']) # menuEdit menuEdit.addAction(self.actions['undo']) menuEdit.addAction(self.actions['redo']) menuEdit.addAction(self.actions['findText']) menuEdit.addAction(self.actions['findRepl']) menuEdit.addSeparator() menuEdit.addAction(self.actions['sortLines']) menuEdit.addAction(self.actions['insertImage']) # menuView menuView.addAction(self.actions['edit']) menuView.addAction(self.actions['split']) menuView.addAction(self.actions['flipEditAndView']) menuShowHide = menuView.addMenu(self.tr('Show/Hide')) menuShowHide.addAction(self.dockIndex.toggleViewAction()) menuShowHide.addAction(self.dockSearch.toggleViewAction()) menuShowHide.addAction(self.dockToc.toggleViewAction()) menuShowHide.addAction(self.dockAttachment.toggleViewAction()) #menuMode = menuView.addMenu(self.tr('Mode')) #menuMode.addAction(self.actionLeftAndRight) #menuMode.addAction(self.actionUpAndDown) # menuHelp menuHelp.addAction(self.actions['readme']) menuHelp.addAction(self.actions['changelog']) menuHelp.addAction(self.actions['aboutQt']) toolBar = QToolBar(self.tr("toolbar"), self) toolBar.setObjectName("toolbar") # needed in saveState() #toolBar.setIconSize(QSize(16, 16)) toolBar.setToolButtonStyle(Qt.ToolButtonTextBesideIcon) self.addToolBar(Qt.TopToolBarArea, toolBar) toolBar.addAction(self.actions['edit']) toolBar.addAction(self.actions['split']) self.findEdit.returnPressed.connect(self.findText) self.findBar.addWidget(self.findEdit) self.findBar.addWidget(self.checkBox) self.findBar.addAction(self.actions['findPrev']) self.findBar.addAction(self.actions['find']) self.findBar.setVisible(False) self.findBar.visibilityChanged.connect(self.findBarVisibilityChanged) self.setStatusBar(self.statusBar) self.statusBar.addWidget(self.statusLabel, 1) self.notesTree.currentItemChanged.connect( self.currentItemChangedWrapper) self.notesTree.nvwCallback = self.newNoteDisplay self.notesTree.nvwtCallback = self.newPlainTextNoteDisplay self.tocTree.itemClicked.connect(self.tocNavigate) self.notesEdit.textChanged.connect(self.noteEditted) self.notesEdit.document().modificationChanged.connect( self.modificationChanged) self.updateRecentViewedNotes() notes = self.settings.recentViewedNotes() if len(notes) != 0: item = self.notesTree.pageToItem(notes[0]) self.notesTree.setCurrentItem(item) def newNoteDisplay(self, item, anchor=None): msn = MikiSepNote(self.settings, item.text(0), self.notesTree.itemToFile(item), plain_text=False, parent=self) if anchor: msn.note_view.page().mainFrame().scrollToAnchor(anchor) msn.show() def newPlainTextNoteDisplay(self, item, anchor=None): msn = MikiSepNote(self.settings, item.text(0), self.notesTree.itemToFile(item), plain_text=True, parent=self) if anchor: item = msn.findItemByAnchor(anchor)[0] msn.tocNavigate(item) msn.show() def openFuncWrapper(self): self.openFunction(self.quickNoteNav.text())() def setupWhoosh(self): # Initialize whoosh index, make sure notePath/.indexdir exists indexdir = self.settings.indexdir try: self.ix = open_dir(indexdir) except: QDir().mkpath(indexdir) self.ix = create_in(indexdir, self.settings.schema) # Fork a process to update index, which benefit responsiveness. p = Thread(target=self.whoosh_index, args=()) p.start() def restore(self): """ Restore saved geometry and state. Set the status of side panels in View Menu correspondently. """ if self.settings.geometry: self.restoreGeometry(self.settings.geometry) if self.settings.windowstate: self.restoreState(self.settings.windowstate) def initTree(self, notePath, parent): ''' When there exist foo.md, foo.mkd, foo.markdown, only one item will be shown in notesTree. ''' if not QDir(notePath).exists(): return notebookDir = QDir(notePath) notesList = notebookDir.entryInfoList(['*.md', '*.mkd', '*.markdown'], QDir.NoFilter, QDir.Name | QDir.IgnoreCase) nl = [note.completeBaseName() for note in notesList] noduplicate = list(set(nl)) for name in noduplicate: item = QTreeWidgetItem(parent, [name]) path = notePath + '/' + name self.initTree(path, item) def updateToc(self): ''' TOC is updated in `updateView` tocTree fields: [hdrText, hdrPosition, hdrAnchor] ''' root = self.notesTree.currentPage() strip_math_for_header_parsing = False strip_fence_for_header_parsing = False if 'asciimathml' in self.settings.extensions: strip_math_for_header_parsing = True if 'fenced_code' in self.settings.extensions or 'extra' in self.settings.extensions: strip_fence_for_header_parsing = True self.tocTree.updateToc( root, parseHeaders(self.notesEdit.toPlainText(), strip_fenced_block=strip_fence_for_header_parsing, strip_ascii_math=strip_math_for_header_parsing)) def updateAttachmentView(self): # Update attachmentView to show corresponding attachments. item = self.notesTree.currentItem() index = self.attachmentView.model.index( self.notesTree.itemToAttachmentDir(item)) self.attachmentView.setRootIndex(index) def openFile(self, filename): fh = QFile(filename) try: if not fh.open(QIODevice.ReadOnly): raise IOError(fh.errorString()) except IOError as e: QMessageBox.warning( self, self.tr('Read Error'), self.tr('Failed to open %s: %s') % (filename, e)) finally: if fh is not None: noteBody = QTextStream(fh).readAll() fh.close() self.notesEdit.setPlainText(noteBody) self.notesView.scrollPosition = QPoint(0, 0) # self.actionSave.setEnabled(False) self.notesEdit.document().setModified(False) self.notesView.updateView() self.setCurrentNote() self.updateRecentViewedNotes() #self.statusLabel.setText(noteFullName) def currentItemChangedWrapper(self, current, previous): if current is None: return #if previous != None and self.notesTree.pageExists(previous): prev = self.notesTree.itemToPage(previous) if self.notesTree.pageExists(prev): self.saveNote(previous) currentFile = self.notesTree.itemToFile(current) self.openFile(currentFile) # Update attachmentView to show corresponding attachments. index = self.attachmentView.model.index( self.notesTree.itemToAttachmentDir(current)) self.attachmentView.setRootIndex(index) def tocNavigate(self, current): ''' works for notesEdit now ''' if current is None: return pos = int(current.text(1)) link = "file://" + self.notePath + "/#" + current.text(2) # Move cursor to END first will ensure # header is positioned at the top of visual area. self.notesEdit.moveCursor(QTextCursor.End) cur = self.notesEdit.textCursor() cur.setPosition(pos, QTextCursor.MoveAnchor) self.notesEdit.setTextCursor(cur) self.notesView.load(QUrl(link)) def switchNote(self, num): if num < len(self.viewedListActions): self.viewedListActions[num].trigger() def saveCurrentNote(self): item = self.notesTree.currentItem() self.saveNote(item) def saveNote(self, item): if self.notesEdit.document().isModified(): self.notesEdit.document().setModified(False) else: return self.notesEdit.save(item) def saveNoteAs(self): self.saveCurrentNote() fileName = QFileDialog.getSaveFileName( self, self.tr('Save as'), '', '(*.md *.mkd *.markdown);;' + self.tr('All files(*)')) if fileName == '': return if not QFileInfo(fileName).suffix(): fileName += '.md' fh = QFile(fileName) fh.open(QIODevice.WriteOnly) savestream = QTextStream(fh) savestream << self.notesEdit.toPlainText() fh.close() def printNote(self): printer = QPrinter(QPrinter.HighResolution) printer.setCreator(__appname__ + ' ' + __version__) printer.setDocName(self.notesTree.currentItem().text(0)) printdialog = QPrintDialog(printer, self) if printdialog.exec() == QDialog.Accepted: self.notesView.print_(printer) def noteEditted(self): """ Continuously get fired while editing""" self.updateToc() self.notesView.updateLiveView() def modificationChanged(self, changed): """ Fired one time: modified or not """ self.actions['save'].setEnabled(changed) name = self.notesTree.currentPage() self.statusBar.clearMessage() if changed: self.statusLabel.setText(name + '*') else: self.statusLabel.setText(name) def importPage(self): filename = QFileDialog.getOpenFileName( self, self.tr('Import file'), '', '(*.md *.mkd *.markdown *.txt);;' + self.tr('All files(*)')) if filename == '': return self.importPageCore(filename) def importPageCore(self, filename): fh = QFile(filename) fh.open(QIODevice.ReadOnly) fileBody = QTextStream(fh).readAll() fh.close() page = QFileInfo(filename).completeBaseName() fh = QFile(self.notesTree.pageToFile(page)) if fh.exists(): QMessageBox.warning(self, self.tr("Import Error"), self.tr("Page already exists: %s") % page) dialog = LineEditDialog(self.notePath, self) if dialog.exec_(): page = dialog.editor.text() fh.close() fh = QFile(self.notesTree.pageToFile(page)) else: return fh.open(QIODevice.WriteOnly) savestream = QTextStream(fh) savestream << fileBody fh.close() item = QTreeWidgetItem(self.notesTree, [page]) self.notesTree.sortItems(0, Qt.AscendingOrder) self.notesTree.setCurrentItem(item) def openNotebook(self): dialog = NotebookListDialog(self) if dialog.exec_(): pass def notebookSettings(self): dialog = NotebookSettingsDialog(self) if dialog.exec_(): pass def mikidownSettings(self): dialog = MikidownCfgDialog(self) if dialog.exec_(): pass def reIndex(self): """ Whoosh index breaks for unknown reasons (sometimes) """ shutil.rmtree(self.settings.indexdir) self.setupWhoosh() def act(self, name, trig, shortcut=None, checkable=False, icon=None, tooltip=None): """ A wrapper to several QAction methods """ if icon: action = QAction(icon, name, self) else: action = QAction(name, self) if shortcut: action.setShortcut(QKeySequence(shortcut)) action.setCheckable(checkable) if tooltip: action.setToolTip(tooltip) action.triggered.connect(trig) return action def edit(self, viewmode): """ Switch between EDIT and VIEW mode. """ if self.actions['split'].isChecked(): self.actions['split'].setChecked(False) self.notesView.setVisible(not viewmode) self.notesEdit.setVisible(viewmode) # Gives the keyboard input focus to notesEdit/notesView. # Without this, keyboard input may change note text even when # notesEdit is invisible. if viewmode: self.notesEdit.setFocus() else: self.notesView.setFocus() self.saveCurrentNote() self.actions['insertImage'].setEnabled(viewmode) #self.actionLeftAndRight.setEnabled(True) #self.actionUpAndDown.setEnabled(True) # Render the note text as it is. self.notesView.updateView() def liveView(self, viewmode): """ Switch between VIEW and LIVE VIEW mode. """ self.actions['split'].setChecked(viewmode) sizes = self.noteSplitter.sizes() if self.actions['edit'].isChecked(): self.actions['edit'].setChecked(False) self.notesView.setVisible(viewmode) splitSize = [sizes[0] * 0.45, sizes[0] * 0.55] else: self.notesEdit.setVisible(viewmode) splitSize = [sizes[1] * 0.45, sizes[1] * 0.55] # setFocus for the same reason as in edit(self, viewmode) if viewmode: self.notesEdit.setFocus() else: self.notesView.setFocus() self.actions['flipEditAndView'].setEnabled(viewmode) #self.actionUpAndDown.setEnabled(viewmode) self.actions['insertImage'].setEnabled(viewmode) self.noteSplitter.setSizes(splitSize) self.saveCurrentNote() # Render the note text as it is. self.notesView.updateView() def findBarVisibilityChanged(self, visible): self.actions['findText'].setChecked(visible) if visible: self.findEdit.setFocus(Qt.ShortcutFocusReason) def findText(self, back=False): flags = 0 if back: flags = QTextDocument.FindBackward if self.checkBox.isChecked(): flags = flags | QTextDocument.FindCaseSensitively text = self.findEdit.text() if not self.findMain(text, flags): if text in self.notesEdit.toPlainText(): cursor = self.notesEdit.textCursor() if back: cursor.movePosition(QTextCursor.End) else: cursor.movePosition(QTextCursor.Start) self.notesEdit.setTextCursor(cursor) self.findMain(text, flags) # self.notesView.findText(text, flags) def findMain(self, text, flags): viewFlags = QWebPage.FindFlags( flags) | QWebPage.FindWrapsAroundDocument if flags: self.notesView.findText(text, viewFlags) return self.notesEdit.find(text, flags) else: self.notesView.findText(text) return self.notesEdit.find(text) def sortLines(self): ''' sort selected lines TODO: second sort reverse the order ''' cursor = self.notesEdit.textCursor() start = cursor.selectionStart() end = cursor.selectionEnd() cursor.setPosition(start) cursor.movePosition(QTextCursor.StartOfLine) cursor.setPosition(end, mode=QTextCursor.KeepAnchor) cursor.movePosition(QTextCursor.EndOfLine, mode=QTextCursor.KeepAnchor) text = cursor.selectedText() lines = text.split('\u2029') # '\u2029' is the line break sortedLines = sorted(lines) cursor.insertText('\n'.join(sortedLines)) def notesEditInFocus(self, e): if e.gotFocus: self.actions['insertImage'].setEnabled(True) # if e.lostFocus: # self.actionInsertImage.setEnabled(False) # QWidget.focusInEvent(self,f) def searchNote(self): """ Sorting criteria: "title > path > content" Search matches are organized into html source. """ pattern = self.searchEdit.text() if not pattern: return results = [] print("Searching using", pattern) with self.ix.searcher() as searcher: matches = [] queryp = QueryParser("content", self.ix.schema) #allow escaped qutoes when regex searching queryp.add_plugin( RegexPlugin(expr=r'r"(?P<text>[^"\\]*(\\.[^"\\]*)*)"')) # ~~r"pattern" is the desired regex term format~~ Don't autoforce regexing query = queryp.parse(pattern) #print("durp durp", query) ms = searcher.search(query, limit=None) # default limit is 10! for m in ms: #if not m in matches: matches.append(m) for r in matches: title = r['title'] path = r['path'] term = r.highlights("content") results.append([title, path, term]) html = "" for title, path, hi in results: html += ("<p><a href='" + path + "'>" + title + "</a><br/><span class='path'>" + path + "</span><br/>" + hi + "</p>") self.searchView.setHtml(html) print("Finished searching", pattern) def whoosh_index(self): it = QTreeWidgetItemIterator(self.notesTree, QTreeWidgetItemIterator.All) print("Starting complete indexing.") #writer = self.ix.writer() writer = AsyncWriter(self.ix) while it.value(): treeItem = it.value() name = self.notesTree.itemToPage(treeItem) path = os.path.join(self.notesTree.pageToFile(name)).replace( os.sep, '/') print(path) fileobj = open(path, 'r', encoding='utf-8') content = fileobj.read() fileobj.close() if METADATA_CHECKER.match( content) and 'meta' in self.settings.extensions: no_metadata_content = METADATA_CHECKER.sub("", content, count=1).lstrip() self.settings.md.reset().convert(content) writer.update_document( path=name, title=parseTitle(content, name), content=no_metadata_content, tags=','.join(self.settings.md.Meta.get('tags', [])).strip()) else: writer.add_document(path=name, title=parseTitle(content, name), content=content, tags='') it += 1 writer.commit() print("Finished completely reindexing.") def listItemChanged(self, row): if row != -1: item = self.searchList.currentItem().data(Qt.UserRole) self.notesTree.setCurrentItem(item) flags = QWebPage.HighlightAllOccurrences self.notesView.findText(self.searchEdit.text(), flags) def setCurrentNote(self): item = self.notesTree.currentItem() name = self.notesTree.itemToPage(item) # Current note is inserted to head of list. notes = self.settings.recentViewedNotes() for f in notes: if f == name: notes.remove(f) notes.insert(0, name) recent_notes_n = Mikibook.settings.value('recentNotesNumber', type=int, defaultValue=20) if len(notes) > recent_notes_n: del notes[recent_notes_n:] self.settings.updateRecentViewedNotes(notes) def updateRecentViewedNotes(self): """ Switching notes will trigger this. When Alt pressed, show note number. """ self.viewedList.clear() self.viewedListActions = [] # Check notes exists. viewedNotes = self.settings.recentViewedNotes() existedNotes = [] i = 0 for f in viewedNotes: if self.notesTree.pageExists(f): existedNotes.append(f) names = f.split('/') if self.altPressed and i in range(1, 10): action = self.act(names[-1], self.openFunction(f), 'Alt+' + str(i), True, ViewedNoteIcon(i), 'Alt+' + str(i)) else: action = self.act(names[-1], self.openFunction(f), None, True) self.viewedListActions.append(action) i += 1 if not self.altPressed: self.settings.updateRecentViewedNotes(existedNotes) for action in self.viewedListActions: self.viewedList.addAction(action) if len(self.viewedListActions): self.viewedListActions[0].setChecked(True) def openFunction(self, name): item = self.notesTree.pageToItem(name) return lambda: self.notesTree.setCurrentItem(item) def raiseDock(self, widget): if not widget.isVisible(): widget.show() if widget == self.dockSearch: self.searchEdit.setFocus() widget.raise_() def flipEditAndView(self): index = self.noteSplitter.indexOf(self.notesEdit) if index == 0: self.noteSplitter.insertWidget(1, self.notesEdit) else: self.noteSplitter.insertWidget(0, self.notesEdit) def leftAndRight(self): self.liveView(True) self.noteSplitter.setOrientation(Qt.Horizontal) #self.actionLeftAndRight.setEnabled(False) #self.actionUpAndDown.setEnabled(True) def upAndDown(self): self.liveView(True) self.noteSplitter.setOrientation(Qt.Vertical) #self.actionUpAndDown.setEnabled(False) #self.actionLeftAndRight.setEnabled(True) def readmeHelp(self): readmeFile = '/usr/share/mikidown/README.mkd' if not os.path.exists(readmeFile): readmeFile = os.path.join( os.path.dirname(os.path.dirname(__file__)), 'README.mkd').replace(os.sep, '/') self.importPageCore(readmeFile) def changelogHelp(self): changeLog = "/usr/share/mikidown/Changelog.md" if not os.path.exists(changeLog): changeLog = os.path.join( os.path.dirname(os.path.dirname(__file__)), 'Changelog.md').replace(os.sep, '/') self.importPageCore(changeLog) def keyPressEvent(self, event): """ When Alt pressed, note number will be shown in viewedList. """ if event.key() == Qt.Key_Alt: self.altPressed = True self.updateRecentViewedNotes() else: QMainWindow.keyPressEvent(self, event) def keyReleaseEvent(self, event): if event.key() == Qt.Key_Alt: self.altPressed = False self.updateRecentViewedNotes() else: QMainWindow.keyPressEvent(self, event) def closeEvent(self, event): """ saveGeometry: Saves the current geometry and state for top-level widgets saveState: Restores the state of this mainwindow's toolbars and dockwidgets """ self.saveCurrentNote() self.ix.close() self.notesEdit.ix.close() if hasattr(self.notesTree, 'ix'): self.notesTree.ix.close() self.settings.saveGeometry(self.saveGeometry()) self.settings.saveWindowState(self.saveState()) event.accept() os.close(self.lockPathFH) lockPath = os.path.join(self.settings.notebookPath, '.mikidown_lock') os.remove(lockPath)
class MikiWindow(QMainWindow): def __init__(self, settings, parent=None): super(MikiWindow, self).__init__(parent) self.setObjectName("mikiWindow") self.settings = settings self.notePath = settings.notePath ################ Setup core components ################ self.notesTree = MikiTree(self) self.quickNoteNav = QLineEdit() self.notesTab = QWidget() self.completer = SlashPleter() self.completer.setModel(self.notesTree.model()) self.quickNoteNav.setCompleter(self.completer) self.notesTree.setObjectName("notesTree") self.initTree(self.notePath, self.notesTree) self.notesTree.sortItems(0, Qt.AscendingOrder) self.ix = None self.setupWhoosh() self.viewedList = QToolBar(self.tr('Recently Viewed'), self) self.viewedList.setIconSize(QSize(16, 16)) self.viewedList.setToolButtonStyle(Qt.ToolButtonTextBesideIcon) self.viewedListActions = [] self.noteSplitter = QSplitter(Qt.Horizontal) self.dockIndex = QDockWidget("Index") self.dockSearch = QDockWidget("Search") self.searchEdit = QLineEdit() self.searchView = MikiSearch(self) self.searchTab = QWidget() self.dockToc = QDockWidget("TOC") self.tocTree = TocTree() self.dockAttachment = QDockWidget("Attachment") self.attachmentView = AttachmentView(self) self.notesEdit = MikiEdit(self) self.notesEdit.setObjectName("notesEdit") MikiHighlighter(self.notesEdit) self.notesView = MikiView(self) self.findBar = QToolBar(self.tr('Find'), self) self.findBar.setFixedHeight(30) self.findEdit = QLineEdit(self.findBar) self.checkBox = QCheckBox(self.tr('Match case'), self.findBar) self.statusBar = QStatusBar(self) self.statusLabel = QLabel(self) self.altPressed = False ################ Setup actions ################ self.actions = dict() self.setupActions() ################ Setup mainwindow ################ self.setupMainWindow() # show changelogs after upgrade mikidown if self.settings.version < __version__: self.changelogHelp() self.settings.qsettings.setValue("version", __version__) def setupActions(self): # Global Actions actTabIndex = self.act(self.tr('Switch to Index Tab'), lambda: self.raiseDock(self.dockIndex), 'Ctrl+Shift+I') actTabSearch = self.act(self.tr('Switch to Search Tab'), lambda: self.raiseDock(self.dockSearch), 'Ctrl+Shift+F') self.addAction(actTabIndex) self.addAction(actTabSearch) ################ Menu Actions ################ # actions in menuFile actionNewPage = self.act(self.tr('&New Page...'), self.notesTree.newPage, QKeySequence.New) self.actions.update(newPage=actionNewPage) actionNewSubpage = self.act(self.tr('New Sub&page...'), self.notesTree.newSubpage, 'Ctrl+Shift+N') self.actions.update(newSubpage=actionNewSubpage) actionImportPage = self.act(self.tr('&Import Page...'), self.importPage) self.actions.update(importPage=actionImportPage) actionNBSettings = self.act(self.tr('Notebook Set&tings...'), self.notebookSettings) self.actions.update(NBSettings=actionNBSettings) actionMDSettings = self.act(self.tr('&Mikidown Settings...'), self.mikidownSettings) self.actions.update(MDSettings=actionMDSettings) actionOpenNotebook = self.act(self.tr('&Open Notebook...'), self.openNotebook, QKeySequence.Open) self.actions.update(openNotebook=actionOpenNotebook) actionReIndex = self.act(self.tr('Re-index'), self.reIndex) self.actions.update(reIndex=actionReIndex) actionSave = self.act(self.tr('&Save'), self.saveCurrentNote, QKeySequence.Save) actionSave.setEnabled(False) self.actions.update(save=actionSave) actionSaveAs = self.act(self.tr('Save &As...'), self.saveNoteAs, QKeySequence.SaveAs) self.actions.update(saveAs=actionSaveAs) actionHtml = self.act(self.tr('to &HTML'), self.notesEdit.saveAsHtml) self.actions.update(html=actionHtml) actionPrint = self.act(self.tr('&Print'), self.printNote, QKeySequence.Print) self.actions.update(print_=actionPrint) actionRenamePage = self.act(self.tr('&Rename Page...'), self.notesTree.renamePage, 'F2') self.actions.update(renamePage=actionRenamePage) actionDelPage = self.act(self.tr('&Delete Page'), self.notesTree.delPageWrapper, QKeySequence.Delete) self.actions.update(delPage=actionDelPage) actionQuit = self.act(self.tr('&Quit'), self.close, QKeySequence.Quit) actionQuit.setMenuRole(QAction.QuitRole) self.actions.update(quit=actionQuit) # actions in menuEdit actionUndo = self.act(self.tr('&Undo'), lambda: self.notesEdit.undo(), QKeySequence.Undo) actionUndo.setEnabled(False) self.notesEdit.undoAvailable.connect(actionUndo.setEnabled) self.actions.update(undo=actionUndo) actionRedo = self.act(self.tr('&Redo'), lambda: self.notesEdit.redo(), QKeySequence.Redo) actionRedo.setEnabled(False) self.notesEdit.redoAvailable.connect(actionRedo.setEnabled) self.actions.update(redo=actionRedo) actionFindText = self.act(self.tr('&Find Text'), self.findBar.setVisible, QKeySequence.Find, True) self.actions.update(findText=actionFindText) actionFind = self.act(self.tr('Next'), self.findText, QKeySequence.FindNext) self.actions.update(find=actionFind) actionFindPrev = self.act(self.tr('Previous'), lambda: self.findText(back=True), QKeySequence.FindPrevious) self.actions.update(findPrev=actionFindPrev) actionSortLines = self.act(self.tr('&Sort Lines'), self.sortLines) self.actions.update(sortLines=actionSortLines) actionQuickNav = self.act(self.tr("&Quick Open Note"), self.quickNoteNav.setFocus, 'Ctrl+G') self.addAction(actionQuickNav) actionInsertImage = self.act(self.tr('&Insert Attachment'), self.notesEdit.insertAttachmentWrapper, 'Ctrl+I') actionInsertImage.setEnabled(False) self.actions.update(insertImage=actionInsertImage) # actions in menuView actionEdit = self.act(self.tr('Edit'), self.edit, 'Ctrl+E', True, QIcon(':/icons/edit.svg'), 'Edit mode (Ctrl+E)') self.actions.update(edit=actionEdit) actionSplit = self.act(self.tr('Split'), self.liveView, 'Ctrl+R', True, QIcon(':/icons/split.svg'), 'Split mode (Ctrl+R)') self.actions.update(split=actionSplit) actionFlipEditAndView = self.act(self.tr('Flip Edit and View'), self.flipEditAndView) actionFlipEditAndView.setEnabled(False) self.actions.update(flipEditAndView=actionFlipEditAndView) #actionLeftAndRight = self.act( # self.tr('Split into Left and Right'), trig=self.leftAndRight) #actionUpAndDown = self.act( # self.tr('Split into Up and Down'), trig=self.upAndDown) # self.actionLeftAndRight.setEnabled(False) # self.actionUpAndDown.setEnabled(False) # actions in menuHelp actionReadme = self.act(self.tr('README'), self.readmeHelp) self.actions.update(readme=actionReadme) actionChangelog = self.act(self.tr('Changelog'), self.changelogHelp) self.actions.update(changelog=actionChangelog) actionAboutQt = self.act(self.tr('About Qt'), qApp.aboutQt) self.actions.update(aboutQt=actionAboutQt) def setupMainWindow(self): self.resize(800, 600) screen = QDesktopWidget().screenGeometry() size = self.geometry() self.move(( screen.width()-size.width())/2, (screen.height()-size.height())/2) self.setWindowTitle( '{} - {}'.format(self.settings.notebookName, __appname__)) self.viewedList.setFixedHeight(25) self.noteSplitter.addWidget(self.notesEdit) self.noteSplitter.addWidget(self.notesView) mainSplitter = QSplitter(Qt.Vertical) mainSplitter.setChildrenCollapsible(False) mainSplitter.addWidget(self.viewedList) mainSplitter.addWidget(self.noteSplitter) mainSplitter.addWidget(self.findBar) self.setCentralWidget(mainSplitter) self.searchEdit.returnPressed.connect(self.searchNote) self.quickNoteNav.returnPressed.connect(self.openFuncWrapper) searchLayout = QVBoxLayout() searchLayout.addWidget(self.searchEdit) searchLayout.addWidget(self.searchView) self.searchTab.setLayout(searchLayout) self.tocTree.header().close() indexLayout = QVBoxLayout(self.notesTab) indexLayout.addWidget(self.quickNoteNav) indexLayout.addWidget(self.notesTree) self.dockIndex.setObjectName("Index") self.dockIndex.setWidget(self.notesTab) self.dockSearch.setObjectName("Search") self.dockSearch.setWidget(self.searchTab) self.dockToc.setObjectName("TOC") self.dockToc.setWidget(self.tocTree) self.dockAttachment.setObjectName("Attachment") self.dockAttachment.setWidget(self.attachmentView) self.setDockOptions(QMainWindow.VerticalTabs) self.addDockWidget(Qt.LeftDockWidgetArea, self.dockIndex) self.addDockWidget(Qt.LeftDockWidgetArea, self.dockSearch) self.addDockWidget(Qt.LeftDockWidgetArea, self.dockToc) self.addDockWidget(Qt.LeftDockWidgetArea, self.dockAttachment) self.tabifyDockWidget(self.dockIndex, self.dockSearch) self.tabifyDockWidget(self.dockSearch, self.dockToc) self.tabifyDockWidget(self.dockToc, self.dockAttachment) self.setTabPosition(Qt.LeftDockWidgetArea, QTabWidget.North) self.dockIndex.raise_() # Put dockIndex on top of the tab stack menuBar = QMenuBar(self) self.setMenuBar(menuBar) menuFile = menuBar.addMenu(self.tr('&File')) menuEdit = menuBar.addMenu(self.tr('&Edit')) menuView = menuBar.addMenu(self.tr('&View')) menuHelp = menuBar.addMenu(self.tr('&Help')) # menuFile menuFile.addAction(self.actions['newPage']) menuFile.addAction(self.actions['newSubpage']) menuFile.addAction(self.actions['NBSettings']) menuFile.addAction(self.actions['MDSettings']) menuFile.addAction(self.actions['importPage']) menuFile.addAction(self.actions['openNotebook']) menuFile.addAction(self.actions['reIndex']) menuFile.addSeparator() menuFile.addAction(self.actions['save']) menuFile.addAction(self.actions['saveAs']) menuFile.addAction(self.actions['print_']) menuExport = menuFile.addMenu(self.tr('&Export')) menuExport.addAction(self.actions['html']) menuFile.addSeparator() menuFile.addAction(self.actions['renamePage']) menuFile.addAction(self.actions['delPage']) menuFile.addSeparator() menuFile.addAction(self.actions['quit']) # menuEdit menuEdit.addAction(self.actions['undo']) menuEdit.addAction(self.actions['redo']) menuEdit.addAction(self.actions['findText']) menuEdit.addSeparator() menuEdit.addAction(self.actions['sortLines']) menuEdit.addAction(self.actions['insertImage']) # menuView menuView.addAction(self.actions['edit']) menuView.addAction(self.actions['split']) menuView.addAction(self.actions['flipEditAndView']) menuShowHide = menuView.addMenu(self.tr('Show/Hide')) menuShowHide.addAction(self.dockIndex.toggleViewAction()) menuShowHide.addAction(self.dockSearch.toggleViewAction()) menuShowHide.addAction(self.dockToc.toggleViewAction()) menuShowHide.addAction(self.dockAttachment.toggleViewAction()) #menuMode = menuView.addMenu(self.tr('Mode')) #menuMode.addAction(self.actionLeftAndRight) #menuMode.addAction(self.actionUpAndDown) # menuHelp menuHelp.addAction(self.actions['readme']) menuHelp.addAction(self.actions['changelog']) menuHelp.addAction(self.actions['aboutQt']) toolBar = QToolBar(self.tr("toolbar"), self) toolBar.setObjectName("toolbar") # needed in saveState() toolBar.setIconSize(QSize(16, 16)) toolBar.setToolButtonStyle(Qt.ToolButtonTextBesideIcon) self.addToolBar(Qt.TopToolBarArea, toolBar) toolBar.addAction(self.actions['edit']) toolBar.addAction(self.actions['split']) self.findEdit.returnPressed.connect(self.findText) self.findBar.addWidget(self.findEdit) self.findBar.addWidget(self.checkBox) self.findBar.addAction(self.actions['findPrev']) self.findBar.addAction(self.actions['find']) self.findBar.setVisible(False) self.findBar.visibilityChanged.connect(self.findBarVisibilityChanged) self.setStatusBar(self.statusBar) self.statusBar.addWidget(self.statusLabel, 1) self.notesTree.currentItemChanged.connect( self.currentItemChangedWrapper) self.tocTree.itemClicked.connect(self.tocNavigate) self.notesEdit.textChanged.connect(self.noteEditted) self.notesEdit.document( ).modificationChanged.connect(self.modificationChanged) self.updateRecentViewedNotes() notes = self.settings.recentViewedNotes() if len(notes) != 0: item = self.notesTree.pageToItem(notes[0]) self.notesTree.setCurrentItem(item) def openFuncWrapper(self): self.openFunction(self.quickNoteNav.text())() def setupWhoosh(self): # Initialize whoosh index, make sure notePath/.indexdir exists indexdir = self.settings.indexdir try: self.ix = open_dir(indexdir) except: QDir().mkpath(indexdir) self.ix = create_in(indexdir, self.settings.schema) # Fork a process to update index, which benefit responsiveness. p = Thread(target=self.whoosh_index, args=()) p.start() def restore(self): """ Restore saved geometry and state. Set the status of side panels in View Menu correspondently. """ if self.settings.geometry: self.restoreGeometry(self.settings.geometry) if self.settings.windowstate: self.restoreState(self.settings.windowstate) def initTree(self, notePath, parent): ''' When there exist foo.md, foo.mkd, foo.markdown, only one item will be shown in notesTree. ''' if not QDir(notePath).exists(): return notebookDir = QDir(notePath) notesList = notebookDir.entryInfoList(['*.md', '*.mkd', '*.markdown'], QDir.NoFilter, QDir.Name|QDir.IgnoreCase) nl = [note.completeBaseName() for note in notesList] noduplicate = list(set(nl)) for name in noduplicate: item = QTreeWidgetItem(parent, [name]) path = notePath + '/' + name self.initTree(path, item) def updateToc(self): ''' TOC is updated in `updateView` tocTree fields: [hdrText, hdrPosition, hdrAnchor] ''' root = self.notesTree.currentPage() self.tocTree.clear() item = QTreeWidgetItem(self.tocTree, [root, '0']) curLevel = 0 for (level, h, p, a) in parseHeaders(self.notesEdit.toPlainText()): val = [h, str(p), a] if level == curLevel: item = QTreeWidgetItem(item.parent(), val) elif level < curLevel: item = QTreeWidgetItem(item.parent().parent(), val) curLevel = level else: item = QTreeWidgetItem(item, val) curLevel = level self.tocTree.expandAll() def updateAttachmentView(self): # Update attachmentView to show corresponding attachments. item = self.notesTree.currentItem() index = self.attachmentView.model.index( self.notesTree.itemToAttachmentDir(item)) self.attachmentView.setRootIndex(index) def openFile(self, filename): fh = QFile(filename) try: if not fh.open(QIODevice.ReadOnly): raise IOError(fh.errorString()) except IOError as e: QMessageBox.warning(self, 'Read Error', 'Failed to open %s: %s' % (filename, e)) finally: if fh is not None: noteBody = QTextStream(fh).readAll() fh.close() self.notesEdit.setPlainText(noteBody) self.notesView.scrollPosition = QPoint(0, 0) # self.actionSave.setEnabled(False) self.notesEdit.document().setModified(False) self.notesView.updateView() self.setCurrentNote() self.updateRecentViewedNotes() #self.statusLabel.setText(noteFullName) def currentItemChangedWrapper(self, current, previous): if current is None: return #if previous != None and self.notesTree.pageExists(previous): prev = self.notesTree.itemToPage(previous) if self.notesTree.pageExists(prev): self.saveNote(previous) currentFile = self.notesTree.itemToFile(current) self.openFile(currentFile) # Update attachmentView to show corresponding attachments. index = self.attachmentView.model.index( self.notesTree.itemToAttachmentDir(current)) self.attachmentView.setRootIndex(index) def tocNavigate(self, current): ''' works for notesEdit now ''' if current is None: return pos = int(current.text(1)) link = "file://" + self.notePath + "/#" + current.text(2) # Move cursor to END first will ensure # header is positioned at the top of visual area. self.notesEdit.moveCursor(QTextCursor.End) cur = self.notesEdit.textCursor() cur.setPosition(pos, QTextCursor.MoveAnchor) self.notesEdit.setTextCursor(cur) self.notesView.load(QUrl(link)) def switchNote(self, num): if num < len(self.viewedListActions): self.viewedListActions[num].trigger() def saveCurrentNote(self): item = self.notesTree.currentItem() self.saveNote(item) def saveNote(self, item): if self.notesEdit.document().isModified(): self.notesEdit.document().setModified(False) else: return self.notesEdit.save(item) def saveNoteAs(self): self.saveCurrentNote() fileName = QFileDialog.getSaveFileName(self, self.tr('Save as'), '', '(*.md *.mkd *.markdown);;'+self.tr('All files(*)')) if fileName == '': return if not QFileInfo(fileName).suffix(): fileName += '.md' fh = QFile(fileName) fh.open(QIODevice.WriteOnly) savestream = QTextStream(fh) savestream << self.notesEdit.toPlainText() fh.close() def printNote(self): printer = QPrinter(QPrinter.HighResolution) printer.setCreator(__appname__ + ' ' + __version__) printer.setDocName(self.notesTree.currentItem().text(0)) printdialog = QPrintDialog(printer, self) if printdialog.exec() == QDialog.Accepted: self.notesView.print_(printer) def noteEditted(self): """ Continuously get fired while editing""" self.updateToc() self.notesView.updateLiveView() def modificationChanged(self, changed): """ Fired one time: modified or not """ self.actions['save'].setEnabled(changed) name = self.notesTree.currentPage() self.statusBar.clearMessage() if changed: self.statusLabel.setText(name + '*') else: self.statusLabel.setText(name) def importPage(self): filename = QFileDialog.getOpenFileName( self, self.tr('Import file'), '', '(*.md *.mkd *.markdown *.txt);;'+self.tr('All files(*)')) if filename == '': return self.importPageCore(filename) def importPageCore(self, filename): fh = QFile(filename) fh.open(QIODevice.ReadOnly) fileBody = QTextStream(fh).readAll() fh.close() page = QFileInfo(filename).completeBaseName() fh = QFile(self.notesTree.pageToFile(page)) if fh.exists(): QMessageBox.warning(self, 'Import Error', 'Page already exists: %s' % page) dialog = LineEditDialog(self.notePath, self) if dialog.exec_(): page = dialog.editor.text() fh.close() fh = QFile(self.notesTree.pageToFile(page)) else: return fh.open(QIODevice.WriteOnly) savestream = QTextStream(fh) savestream << fileBody fh.close() item = QTreeWidgetItem(self.notesTree, [page]) self.notesTree.sortItems(0, Qt.AscendingOrder) self.notesTree.setCurrentItem(item) def openNotebook(self): dialog = NotebookListDialog(self) if dialog.exec_(): pass def notebookSettings(self): dialog = NotebookSettingsDialog(self) if dialog.exec_(): pass def mikidownSettings(self): dialog = MikidownCfgDialog(self) if dialog.exec_(): pass def reIndex(self): """ Whoosh index breaks for unknown reasons (sometimes) """ shutil.rmtree(self.settings.indexdir) self.setupWhoosh() def act(self, name, trig, shortcut=None, checkable=False, icon=None, tooltip=None): """ A wrapper to several QAction methods """ if icon: action = QAction(icon, name, self) else: action = QAction(name, self) if shortcut: action.setShortcut(QKeySequence(shortcut)) action.setCheckable(checkable) if tooltip: action.setToolTip(tooltip) action.triggered.connect(trig) return action def edit(self, viewmode): """ Switch between EDIT and VIEW mode. """ if self.actions['split'].isChecked(): self.actions['split'].setChecked(False) self.notesView.setVisible(not viewmode) self.notesEdit.setVisible(viewmode) # Gives the keyboard input focus to notesEdit/notesView. # Without this, keyboard input may change note text even when # notesEdit is invisible. if viewmode: self.notesEdit.setFocus() else: self.notesView.setFocus() self.saveCurrentNote() self.actions['insertImage'].setEnabled(viewmode) #self.actionLeftAndRight.setEnabled(True) #self.actionUpAndDown.setEnabled(True) # Render the note text as it is. self.notesView.updateView() def liveView(self, viewmode): """ Switch between VIEW and LIVE VIEW mode. """ self.actions['split'].setChecked(viewmode) sizes = self.noteSplitter.sizes() if self.actions['edit'].isChecked(): self.actions['edit'].setChecked(False) self.notesView.setVisible(viewmode) splitSize = [sizes[0]*0.45, sizes[0]*0.55] else: self.notesEdit.setVisible(viewmode) splitSize = [sizes[1]*0.45, sizes[1]*0.55] # setFocus for the same reason as in edit(self, viewmode) if viewmode: self.notesEdit.setFocus() else: self.notesView.setFocus() self.actions['flipEditAndView'].setEnabled(viewmode) #self.actionUpAndDown.setEnabled(viewmode) self.actions['insertImage'].setEnabled(viewmode) self.noteSplitter.setSizes(splitSize) self.saveCurrentNote() # Render the note text as it is. self.notesView.updateView() def findBarVisibilityChanged(self, visible): self.actions['findText'].setChecked(visible) if visible: self.findEdit.setFocus(Qt.ShortcutFocusReason) def findText(self, back=False): flags = 0 if back: flags = QTextDocument.FindBackward if self.checkBox.isChecked(): flags = flags | QTextDocument.FindCaseSensitively text = self.findEdit.text() if not self.findMain(text, flags): if text in self.notesEdit.toPlainText(): cursor = self.notesEdit.textCursor() if back: cursor.movePosition(QTextCursor.End) else: cursor.movePosition(QTextCursor.Start) self.notesEdit.setTextCursor(cursor) self.findMain(text, flags) # self.notesView.findText(text, flags) def findMain(self, text, flags): viewFlags = QWebPage.FindFlags( flags) | QWebPage.FindWrapsAroundDocument if flags: self.notesView.findText(text, viewFlags) return self.notesEdit.find(text, flags) else: self.notesView.findText(text) return self.notesEdit.find(text) def sortLines(self): ''' sort selected lines TODO: second sort reverse the order ''' cursor = self.notesEdit.textCursor() start = cursor.selectionStart() end = cursor.selectionEnd() cursor.setPosition(start) cursor.movePosition(QTextCursor.StartOfLine) cursor.setPosition(end, mode=QTextCursor.KeepAnchor) cursor.movePosition(QTextCursor.EndOfLine, mode=QTextCursor.KeepAnchor) text = cursor.selectedText() lines = text.split('\u2029') # '\u2029' is the line break sortedLines = sorted(lines) cursor.insertText('\n'.join(sortedLines)) def notesEditInFocus(self, e): if e.gotFocus: self.actions['insertImage'].setEnabled(True) # if e.lostFocus: # self.actionInsertImage.setEnabled(False) # QWidget.focusInEvent(self,f) def searchNote(self): """ Sorting criteria: "title > path > content" Search matches are organized into html source. """ pattern = self.searchEdit.text() if not pattern: return results = [] print("Searching using", pattern) with self.ix.searcher() as searcher: matches = [] for f in ["title", "path", "content"]: queryp = QueryParser(f, self.ix.schema) queryp.add_plugin(RegexPlugin()) # r"pattern" is the desired regex term format query = queryp.parse('r"' + pattern + '"') ms = searcher.search(query, limit=None) # default limit is 10! for m in ms: if not m in matches: matches.append(m) for r in matches: title = r['title'] path = r['path'] term = r.highlights("content") results.append([title, path, term]) html = "" for title, path, hi in results: html += ("<p><a href='" + path + "'>" + title + "</a><br/><span class='path'>" + path + "</span><br/>" + hi + "</p>") self.searchView.setHtml(html) print("Finished searching", pattern) def whoosh_index(self): it = QTreeWidgetItemIterator( self.notesTree, QTreeWidgetItemIterator.All) print("Starting complete indexing.") writer = self.ix.writer() while it.value(): treeItem = it.value() name = self.notesTree.itemToPage(treeItem) path = os.path.join(self.notesTree.pageToFile(name)).replace(os.sep, '/') print(path) fileobj = open(path, 'r') content = fileobj.read() fileobj.close() writer.add_document( path=name, title=parseTitle(content, name), content=content) it += 1 writer.commit() print("Finished completely reindexing.") def listItemChanged(self, row): if row != -1: item = self.searchList.currentItem().data(Qt.UserRole) self.notesTree.setCurrentItem(item) flags = QWebPage.HighlightAllOccurrences self.notesView.findText(self.searchEdit.text(), flags) def setCurrentNote(self): item = self.notesTree.currentItem() name = self.notesTree.itemToPage(item) # Current note is inserted to head of list. notes = self.settings.recentViewedNotes() for f in notes: if f == name: notes.remove(f) notes.insert(0, name) recent_notes_n = Mikibook.settings.value('recentNotesNumber',type=int, defaultValue=20) if len(notes) > recent_notes_n: del notes[recent_notes_n:] self.settings.updateRecentViewedNotes(notes) def updateRecentViewedNotes(self): """ Switching notes will trigger this. When Alt pressed, show note number. """ self.viewedList.clear() self.viewedListActions = [] # Check notes exists. viewedNotes = self.settings.recentViewedNotes() existedNotes = [] i = 0 for f in viewedNotes: if self.notesTree.pageExists(f): existedNotes.append(f) names = f.split('/') if self.altPressed and i in range(1, 10): action = self.act(names[-1], self.openFunction(f), 'Alt+'+str(i), True, ViewedNoteIcon(i), 'Alt+'+str(i)) else: action = self.act(names[-1], self.openFunction(f), None, True) self.viewedListActions.append(action) i += 1 if not self.altPressed: self.settings.updateRecentViewedNotes(existedNotes) for action in self.viewedListActions: self.viewedList.addAction(action) if len(self.viewedListActions): self.viewedListActions[0].setChecked(True) def openFunction(self, name): item = self.notesTree.pageToItem(name) return lambda: self.notesTree.setCurrentItem(item) def raiseDock(self, widget): if not widget.isVisible(): widget.show() if widget == self.dockSearch: self.searchEdit.setFocus() widget.raise_() def flipEditAndView(self): index = self.noteSplitter.indexOf(self.notesEdit) if index == 0: self.noteSplitter.insertWidget(1, self.notesEdit) else: self.noteSplitter.insertWidget(0, self.notesEdit) def leftAndRight(self): self.liveView(True) self.noteSplitter.setOrientation(Qt.Horizontal) #self.actionLeftAndRight.setEnabled(False) #self.actionUpAndDown.setEnabled(True) def upAndDown(self): self.liveView(True) self.noteSplitter.setOrientation(Qt.Vertical) #self.actionUpAndDown.setEnabled(False) #self.actionLeftAndRight.setEnabled(True) def readmeHelp(self): readmeFile = '/usr/share/mikidown/README.mkd' if not os.path.exists(readmeFile): readmeFile = os.path.join( os.path.dirname(os.path.dirname(__file__)), 'README.mkd').replace(os.sep, '/') self.importPageCore(readmeFile) def changelogHelp(self): changeLog = "/usr/share/mikidown/Changelog.md" if not os.path.exists(changeLog): changeLog = os.path.join( os.path.dirname(os.path.dirname(__file__)), 'Changelog.md').replace(os.sep, '/') self.importPageCore(changeLog) def keyPressEvent(self, event): """ When Alt pressed, note number will be shown in viewedList. """ if event.key() == Qt.Key_Alt: self.altPressed = True self.updateRecentViewedNotes() else: QMainWindow.keyPressEvent(self, event) def keyReleaseEvent(self, event): if event.key() == Qt.Key_Alt: self.altPressed = False self.updateRecentViewedNotes() else: QMainWindow.keyPressEvent(self, event) def closeEvent(self, event): """ saveGeometry: Saves the current geometry and state for top-level widgets saveState: Restores the state of this mainwindow's toolbars and dockwidgets """ self.saveCurrentNote() self.settings.saveGeometry(self.saveGeometry()) self.settings.saveWindowState(self.saveState()) event.accept()
class OrbitPlotMainWindow(QMainWindow): """ the main window has three major widgets: current, orbit tabs and element editor. """ def __init__(self, parent = None, machines=[], **kwargs): QMainWindow.__init__(self, parent) self.iqtApp = kwargs.get("iqt", None) self.setIconSize(QSize(32, 32)) self.error_bar = True self._dlgOrbitCor = None # logging self.logdock = QDockWidget("Log") self.logdock.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum) textedit = QPlainTextEdit(self.logdock) self.logger = logging.getLogger(__name__) self.guilogger = logging.getLogger("aphla.gui") # the "aphla" include lib part logging. When the lib is inside # QThread, logging message will be sent to TextEdit which is cross # thread. # self.guilogger = logging.getLogger("aphla") handler = QTextEditLoggingHandler(textedit) self.guilogger.addHandler(handler) self.guilogger.setLevel(logging.INFO) self.logdock.setWidget(textedit) self.logdock.setAllowedAreas(Qt.BottomDockWidgetArea) self.logdock.setFeatures(QDockWidget.DockWidgetMovable| QDockWidget.DockWidgetClosable) self.logdock.setFloating(False) self.logdock.setMinimumHeight(20) self.logdock.setMaximumHeight(100) self.logdock.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum) self.logdock.resize(200, 60) #print self.logdock.sizeHint() self.addDockWidget(Qt.BottomDockWidgetArea, self.logdock) #print self.logdock.sizeHint() #print self.logdock.minimumSize() #print self.logdock.maximumSize() #self.logger.info("INFO") #self.logdock.setMinimumHeight(40) #self.logdock.setMaximumHeight(160) for msg in kwargs.get("infos", []): self.logger.info(msg) # dict of (machine, (lattice dict, default_lat, pvm)) self._mach = dict([(v[0], (v[1],v[2],v[3])) for v in machines]) for m,(lats,lat0,pvm) in self._mach.items(): self.logger.info("machine '%s' initialized: [%s]" % ( m, ", ".join([lat.name for k,lat in lats.items()]))) if pvm: for pv in pvm.dead(): self.logger.warn("'{0}' is disconnected.".format(pv)) ## DCCT current plot #self.dcct = DcctCurrentPlot() #self.dcct.setMinimumHeight(100) #self.dcct.setMaximumHeight(150) #t0 = time.time() #t = np.linspace(t0 - 8*3600*24, t0, 100) #self.dcct.curve.t = t #v = 500*np.exp((t[0] - t[:50])/(4*3600*24)) #self.dcct.curve.v = v.tolist()+v.tolist() #self.dcct.updatePlot() ## MDI area self.mdiarea = QMdiArea() self.connect(self.mdiarea, SIGNAL("subWindowActivated(QMdiSubWindow)"), self.updateMachineLatticeNames) self.mdiarea.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded) self.physics = ApOrbitPhysics(self.mdiarea, iqt=self.iqtApp) self.live_orbit = True self.setCentralWidget(self.mdiarea) #self._elemed = ElementPropertyTabs() #self.elemeditor = ElementEditorDock(parent=self) #self.elemeditor.setAllowedAreas(Qt.RightDockWidgetArea) #self.elemeditor.setFeatures(QDockWidget.DockWidgetMovable| # QDockWidget.DockWidgetClosable) #self.elemeditor.setFloating(False) #self.elemeditor.setEnabled(False) #self.elemeditor.setMinimumWidth(400) #self.elemeditor.setWidget(self._elemed) #self.elemeditor.show() #self.elemeditor.hide() #self.connect(self.elemeditor, # SIGNAL("elementChecked(PyQt_PyObject, bool)"), # self.physics.elementChecked) #self.addDockWidget(Qt.RightDockWidgetArea, self.elemeditor) self.createMenuToolBar() # the first machine is the default self.machBox.addItems([v for v in self._mach.keys()]) self.reloadLatticeNames(self.machBox.currentText()) self.connect(self.machBox, SIGNAL("currentIndexChanged(QString)"), self.reloadLatticeNames) # update at 1/2Hz self.dt, self.itimer = 1500, 0 #self.timerId = None self.timerId = self.startTimer(self.dt) self.vbpm = None self.statusBar().showMessage("Welcome") #self.initMachine("nsls2v2") #self._newVelemPlot("V2SR", aphla.machines.HLA_VBPM, 'x', # "H Orbit", c = None) #print "Thread started", self.machinit.isRunning() #self.newElementPlots("BPM", "x, y") #self.newElementPlot("BPM", "y") #self.newElementPlot("HCOR", "x") #self.newElementPlot("VCOR", "y") #self.newElementPlot("QUAD", "b1") #self.newElementPlot("SEXT", "b2") def updateMachineLatticeNames(self, wsub): i = self.machBox.findText(wsub.machlat[0]) self.machBox.setCurrentIndex(i) self.reloadLatticeNames(wsub.machlat[0]) def reloadLatticeNames(self, mach): self.latBox.clear() cur_mach = str(self.machBox.currentText()) lats, lat0, pvm = self._mach.get(cur_mach, ({}, None, None)) self.latBox.addItems([lat for lat in lats.keys()]) if lat0: i = self.latBox.findText(lat0.name) self.latBox.setCurrentIndex(i) def closeEvent(self, event): self.physics.close() self.mdiarea.closeAllSubWindows() event.accept() def createMenuToolBar(self): # # file menu # #self.machMenu = self.menuBar().addMenu("&Machines") #self.connect(self.machMenu, SIGNAL("aboutToShow()"), # self.updateMachMenu) self.openMenu = self.menuBar().addMenu("&Open") self.openMenu.addAction("New Plot ...", self.openNewPlot) self.openMenu.addAction("New Tune Plot", self.openTunePlot) self.openMenu.addAction("New BPM Plot", partial( self.newElementPlots, "BPM", "x,y")) self.openMenu.addAction("New HCOR Plot", partial( self.newElementPlots, "HCOR", "x")) self.openMenu.addAction("New VCOR Plot", partial( self.newElementPlots, "VCOR", "y")) self.openMenu.addSeparator() self.openMenu.addAction("Open ORM", self.loadOrm) self.openMenu.addSeparator() self.openMenu.addAction("Save Lattice ...", self.saveSnapshot) fileQuitAction = QAction(QIcon(":/file_quit.png"), "&Quit", self) fileQuitAction.setShortcut("Ctrl+Q") fileQuitAction.setToolTip("Quit the application") fileQuitAction.setStatusTip("Quit the application") #fileQuitAction.setIcon(Qt.QIcon(":/filequit.png")) self.connect(fileQuitAction, SIGNAL("triggered()"), self.close) self.openMenu.addAction(fileQuitAction) # view self.viewMenu = self.menuBar().addMenu("&View") mkmenu = QMenu("&Mark", self.viewMenu) for fam in ["BPM", "COR", "QUAD", "SEXT", "INSERTION"]: famAct = QAction(fam, self) famAct.setCheckable(True) self.connect(famAct, SIGNAL("toggled(bool)"), self.click_markfam) mkmenu.addAction(famAct) # # errorbar #viewErrorBarAction = QAction(QIcon(":/view_errorbar.png"), # "Errorbar", self) #viewErrorBarAction.setCheckable(True) #viewErrorBarAction.setChecked(True) #self.connect(viewErrorBarAction, SIGNAL("toggled(bool)"), # self.errorBar) # #zoomM = QMenu("Zoom", self.viewMenu) # # #drift_from_now = QAction("Drift from Now", self) #drift_from_now.setCheckable(True) #drift_from_now.setShortcut("Ctrl+N") #drift_from_golden = QAction("Drift from Golden", self) #drift_from_golden.setCheckable(True) #drift_from_none = QAction("None", self) #drift_from_none.setCheckable(True) #self.viewMenu.addAction(viewLiveAction) #self.viewMenu.addAction(viewSingleShotAction) #self.viewMenu.addSeparator() #self.viewMenu.addAction(drift_from_now) #self.viewMenu.addAction(drift_from_golden) #self.viewMenu.addAction(drift_from_none) #self.viewMenu.addAction(viewAutoScale) #self.viewMenu.addAction(viewErrorBarAction) #self.viewMenu.addSeparator() self.viewMenu.addMenu(mkmenu) #drift_group = QActionGroup(self) #drift_group.addAction(drift_from_none) #drift_group.addAction(drift_from_now) #drift_group.addAction(drift_from_golden) #drift_from_none.setChecked(True) sep = self.viewMenu.addSeparator() #sep.setText("Drift") #self.connect(drift_from_now, SIGNAL("triggered()"), self.setDriftNow) #self.connect(drift_from_none, SIGNAL("triggered()"), self.setDriftNone) #self.connect(drift_from_golden, SIGNAL("triggered()"), # self.setDriftGolden) #viewStyle = QMenu("Line Style", self.viewMenu) #for act in ["Increase Point Size", "Decrease Point Size", None, # "NoCurve", "Lines", "Sticks", None, # "Solid Line", "Dashed Line", "Dotted Line", None, # "Increase Line Width", "Decrease Line Width", None, # "NoSymbol", "Ellipse", "Rect", "Diamond", "Triangle", # "Cross", "XCross", "HLine", "VLine", # "Star1", "Star2", "Hexagon", None, # "Red", "Blue", "Green"]: # if act is None: # viewStyle.addSeparator() # else: # viewStyle.addAction(act, self.setPlotStyle) #self.viewMenu.addMenu(viewStyle) #self.viewMenu.addSeparator() #self.viewMenu.addAction(viewZoomOut15Action) #self.viewMenu.addAction(viewZoomIn15Action) #self.viewMenu.addAction(viewZoomAutoAction) #self.viewMenu.addSeparator() self.viewMenu.addAction("ORM SV", self.plotSVD) # a bug in PyQwt5 for datetime x-axis, waiting for Debian 7 #self.viewMenu.addAction(viewDcct) #for ac in self.viewMenu.actions(): ac.setDisabled(True) # self.controlMenu = self.menuBar().addMenu("&Tools") self.controlMenu.addAction( QIcon(":/control_choosebpm.png"), "En-/Disable BPM", partial(chooseElement, 'BPM')) self.controlMenu.addAction( "En-/Disable COR", partial(chooseElement, 'COR')) #self.controlMenu.addAction(controlResetPvDataAction) self.controlMenu.addSeparator() self.controlMenu.addAction("Lattice Snapshot ...", self.openSnapshot) #self.controlMenu.addAction(controlZoomInPlot1Action) #self.controlMenu.addAction(controlZoomOutPlot1Action) #self.controlMenu.addAction(controlZoomInPlot2Action) #self.controlMenu.addAction(controlZoomOutPlot2Action) self.controlMenu.addSeparator() self.controlMenu.addAction("Correct Hor. orbit", partial(aphla.correctOrbit, plane="H")) self.controlMenu.addAction("Correct Vert. orbit", partial(aphla.correctOrbit, plane="V")) self.controlMenu.addAction( QIcon(":/control_corrorbit.png"), "Correct orbit", partial(aphla.correctOrbit, plane="HV")) #steer_orbit.setDisabled(True) self.controlMenu.addAction("Local Bump ...", self.createLocalBump) self.controlMenu.addAction("Element Editor ...", self.showElementEditor) self.controlMenu.addSeparator() self.controlMenu.addAction("meas Beta", self.physics.measBeta) self.controlMenu.addAction("meas Dispersion", self.physics.measDispersion) self.controlMenu.addAction("beam based alignment", self.runBba) #for ac in self.controlMenu.actions(): ac.setDisabled(True) # Window self.windowMenu = self.menuBar().addMenu("&Windows") #self.windowMenu.addAction(self.elemeditor.toggleViewAction()) self.windowMenu.addAction(self.logdock.toggleViewAction()) #viewDcct = QAction("Beam Current", self) #viewDcct.setCheckable(True) #viewDcct.setChecked(True) #self.connect(viewDcct, SIGNAL("toggled(bool)"), self.dcct.setVisible) #self.windowMenu.addAction(viewDcct) self.windowMenu.addSeparator() self.windowMenu.addAction("Cascade", self.mdiarea.cascadeSubWindows) self.windowMenu.addAction("Tile", self.mdiarea.tileSubWindows) self.windowMenu.addAction("Tile Horizontally", self.tileSubWindowsHorizontally) # "ctrl+page up", "ctrl+page down" self.windowMenu.addAction("Previous", self.mdiarea.activatePreviousSubWindow, "Ctrl+Left") self.windowMenu.addAction("Next", self.mdiarea.activateNextSubWindow, "Ctrl+Right") self.windowMenu.addSeparator() # debug self.debugMenu = self.menuBar().addMenu("&Debug") self.debugMenu.addAction("_Reset Correctors_", self._reset_correctors) self.debugMenu.addAction("_Reset Quadrupoles_", self._reset_quadrupoles) self.debugMenu.addAction("_Random V Kick_", self._random_vkick) self.debugMenu.addAction("_Random H Kick_", self._random_hkick) #for ac in self.debugMenu.actions(): ac.setDisabled(True) # help self.helpMenu = self.menuBar().addMenu("&Help") self.helpMenu.addAction("About mleap", self.showAbout) #toolbar machToolBar = self.addToolBar("Machines") self.machBox = QtGui.QComboBox() self.latBox = QtGui.QComboBox() #self.connect(self.latBox, SIGNAL("currentIndexChanged(QString)"), # self.__setLattice) machToolBar.addWidget(self.machBox) machToolBar.addWidget(self.latBox) #toolbar = QToolBar(self) #self.addToolBar(toolbar) #fileToolBar = self.addToolBar("File") #fileToolBar.setObjectName("FileToolBar") #fileToolBar.addAction(fileQuitAction) # viewToolBar1 = self.addToolBar("Live View") #viewToolBar.setObjectName("ViewToolBar") #viewToolBar.addAction(viewZoomOut15Action) #viewToolBar.addAction(viewZoomIn15Action) #viewToolBar.addAction(viewZoomAutoAction) #viewToolBar1.addAction(viewLiveAction) #viewToolBar1.addAction(viewSingleShotAction) #viewToolBar1.addSeparator() viewToolBar1.addAction( QIcon(":/new_bpm.png"), "Orbits", partial(self.newElementPlots, "BPM", "x,y")) viewToolBar1.addAction( QIcon(":/new_cor.png"), "Correctors", partial(self.newElementPlots, "COR", "x,y")) viewToolBar1.addAction( QIcon(":/new_quad.png"), "Quadrupoles", partial(self.newElementPlots, "QUAD", "b1")) viewToolBar1.addAction( QIcon(":/new_sext.png"), "Sextupoles", partial(self.newElementPlots, "SEXT", "b2")) #viewToolBar.addAction(viewErrorBarAction) #viewToolBar.addAction(QWhatsThis.createAction(self)) #viewToolBar2 = self.addToolBar("Scale Plot") #zoomActions = [(":/view_zoom_xy.png", "Fit", self.scalePlot), # (None, None, None), # (":/view_zoom_y.png", "Fit In Y", self.scalePlot), # (":/view_zoomin_y.png", "Zoom In Y", self.scalePlot), # (":/view_zoomout_y.png", "Zoom Out Y", self.scalePlot), # (":/view_move_up.png", "Move Up", self.scalePlot), # (":/view_move_down.png", "Move Down", self.scalePlot), # (None, None, None), # (":/view_zoom_x.png", "Fit In X", self.scalePlot), # (":/view_zoomin_x.png", "Zoom In X", self.scalePlot), # (":/view_zoomout_x.png", "Zoom Out X", self.scalePlot), # (":/view_move_left.png", "Move Left", self.scalePlot), # (":/view_move_right.png", "Move Right", self.scalePlot), # ] #for ico,name,hdl in zoomActions: # if hdl is None: continue # viewToolBar2.addAction(QIcon(ico), name, hdl) controlToolBar = self.addToolBar("Control") controlToolBar.addAction( QIcon(":/control_orbitcor.png"), "Correct Orbit", aphla.correctOrbit) controlToolBar.addAction( QIcon(":/control_localbump.png"), "Local Bump ...", self.createLocalBump) #controlToolBar.addAction(controlResetPvDataAction) def showAbout(self): QMessageBox.about( self, self.tr("mleap"), (self.tr("""<b>Machine/Lattice Editor And Plotter</b> v %1 <p>Copyright © Lingyun Yang, BNL, 2013-2014. All rights reserved. <p>This application can be used to perform high level accelerator controls. <p>Python %2 - Qt %3 - PyQt %4 on %5""").arg(aphla.version.version) .arg(platform.python_version()).arg(QtCore.QT_VERSION_STR) .arg(QtCore.PYQT_VERSION_STR).arg(platform.system()))) def showElementEditor(self): mach, lat = self.getCurrentMachLattice() ed = ElementEditor(lat, parent=self) ed.setWindowFlags(Qt.Window) ed.setAttribute(Qt.WA_DeleteOnClose) ed.show() def getCurrentMachLattice(self, cadata = False): """return the current machine name and lattice object""" mach = str(self.machBox.currentText()) latname = str(self.latBox.currentText()) lat_dict, lat0, pvm = self._mach[mach] if not cadata: return mach, lat_dict[latname] else: return mach, lat_dict[latname], pvm def newElementPlots(self, elem, fields, **kw): self.logger.info("new plots: %s %s" % (elem, fields)) _mach, _lat, _pvm = self.getCurrentMachLattice(cadata=True) mach, lat = kw.get("machlat", (_mach, _lat)) handle = kw.get("handle", "readback") elems = lat.getElementList(elem) x, pvs = [], [] field_list = re.findall(r'[^ ,]+', fields) for fld in field_list: si, pvsi = [], [] for e in elems: if not e.isEnabled(): continue epv = e.pv(field=fld, handle=handle) if not epv: continue pvsi.append(epv[0]) si.append(e.sb) x.append(si) pvs.append(pvsi) if not pvs: self.logger.error("no data found for elements '{0}' " "and field '{1}'".format(elem, field)) return p = ApMdiSubPlot(pvs=pvs, x = x, labels=["%s.%s" % (elem,fld) for fld in field_list], magprof = lat.getBeamlineProfile(), iqt = self.iqtApp, **kw) #QObject.installEventFilter(p.aplot) #p.data = ManagedPvData(pvm, s, pvs, element=elemnames, # label="{0}.{1}".format(elem,field)) p.setAttribute(Qt.WA_DeleteOnClose) str_elem = "{0}".format(elem) if len(str_elem) > 12: str_elem = str_elem[:9] + "..." str_field = "{0}".format(fields) if len(str_field) > 12: str_field = str_field[:9] + "..." p.setWindowTitle("[%s.%s] %s %s" % ( mach, lat.name, str_elem, str_field)) self.connect(p, SIGNAL("elementSelected(PyQt_PyObject)"), self.elementSelected) self.connect(p, SIGNAL("destroyed()"), self.subPlotDestroyed) #p.updatePlot() # set the zoom stack #p.aplot.setErrorBar(self.error_bar) #p.wid.autoScaleXY() #p.aplot.replot() self.mdiarea.addSubWindow(p) #print "Show" p.show() ##print "Enable the buttons" #if len(self.mdiarea.subWindowList()) > 0: # self.elemeditor.setEnabled(True) def subPlotDestroyed(self): #if len(self.mdiarea.subWindowList()) == 0: # self.elemeditor.setEnabled(False) pass def loadOrm(self): fileName = QtGui.QFileDialog.getOpenFileName( self, "Open Orbit Response Matrix", "", "ORM Files (*.h5 *.hdf5);;Text File (*.txt);;All Files(*)") fileName = str(fileName) try: m = np.loadtxt(fileName) except: QMessageBox.critical(self, "Abort", "Invalid matrix data") return mach, lat = self.getCurrentMachLattice() # assuming we already have the PV, name, field but just want to # replace the matrix elements. assert np.shape(m) == np.shape(lat.ormdata.m) nx, ny = np.shape(lat.ormdata.m) for i in range(nx): for j in range(ny): lat.ormdata.m[i,j] = m[i,j] def saveSnapshot(self): latdict = dict([(k,v[0]) for k,v in self._mach.items()]) mach, lat = self.getCurrentMachLattice() snapdlg = SaveSnapshotDialog(latdict, mach) snapdlg.exec_() def saveLatSnapshot(self): mach, lat = self.getCurrentMachLattice() dpath = self._prepare_parent_dirs(mach) if not dpath: QMessageBox.warning(self, "Abort", "Aborted") return dt = datetime.datetime.now() fname = os.path.join(dpath, dt.strftime("snapshot_%d_%H%M%S_") + lat.name + ".hdf5") fileName = QtGui.QFileDialog.getSaveFileName( self, "Save Lattice Snapshot Data", fname, "Data Files (*.h5 *.hdf5);;All Files(*)") fileName = str(fileName) if not fileName: return aphla.catools.save_lat_epics(fileName, lat, mode='a') self.logger.info("snapshot created '%s'" % fileName) def saveMachSnapshot(self): mach, lat = self.getCurrentMachLattice() dpath = self._prepare_parent_dirs(mach) if not dpath: QMessageBox.warning(self, "Abort", "Aborted") return dt = datetime.datetime.now() fname = os.path.join(dpath, dt.strftime("snapshot_%d_%H%M%S.hdf5")) fileName = QtGui.QFileDialog.getSaveFileName( self, "Save Lattice Snapshot Data", fname, "Data Files (*.h5 *.hdf5);;All Files(*)") if not fileName: return fileName = str(fileName) import h5py f = h5py.File(str(fileName), 'w') f.close() self.logger.info("clean snapshot file created: '%s'" % fileName) for k,lat in self._mach[mach][0].items(): aphla.catools.save_lat_epics(fileName, lat, mode='a') self.logger.info("lattice snapshot appended for '%s'" % lat.name) def openSnapshot(self): #self.logger.info("loading snapshot?") latdict = dict([(k,v[0]) for k,v in self._mach.items()]) mach, lat = self.getCurrentMachLattice() lv = LatSnapshotMain(self, latdict, mach, self.logger) lv.setWindowFlags(Qt.Window) #self.logger.info("initialized") #lv.loadLatSnapshotH5() lv.exec_() def openTunePlot(self): mach, lat = self.getCurrentMachLattice() nu = lat.getElementList('tune') pvs = [(e.pv(field="x", handle="readback")[0], e.pv(field="y", handle="readback")[0]) for e in nu] labels = [e.name for e in nu] twiss = lat.getElementList("VA") pvs.extend([(e.pv(field="nux", handle="readback")[0], e.pv(field="nuy", handle="readback")[0]) for e in twiss]) labels.extend([e.name for e in twiss]) p = ApMdiSubPlot(pvs=pvs, labels=labels, dtype = "Tunes") #QObject.installEventFilter(p.aplot) #p.data = ManagedPvData(pvm, s, pvs, element=elemnames, # label="{0}.{1}".format(elem,field)) p.setAttribute(Qt.WA_DeleteOnClose) p.setWindowTitle("[%s.%s] Tunes" % (mach, lat.name)) self.connect(p, SIGNAL("elementSelected(PyQt_PyObject)"), self.elementSelected) self.connect(p, SIGNAL("destroyed()"), self.subPlotDestroyed) #p.updatePlot() # set the zoom stack #p.aplot.setErrorBar(self.error_bar) #p.wid.autoScaleXY() #p.aplot.replot() self.mdiarea.addSubWindow(p) #print "Show" p.show() def openNewPlot(self): mach, lat = self.getCurrentMachLattice() fl = QtGui.QFormLayout() fl.addRow("Machine", QtGui.QLabel("%s" % mach)) fl.addRow("Lattice", QtGui.QLabel("%s" % lat.name)) elem, fld = QtGui.QLineEdit(), QtGui.QLineEdit() fl.addRow("Elements", elem) fl.addRow("Field", fld) dtype = QtGui.QComboBox() for tx in ["Array", "Waveform", "Time Series"]: dtype.addItem(tx) fl.addRow("Data Type", dtype) dlg = QtGui.QDialog() bx = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Ok | QtGui.QDialogButtonBox.Cancel) self.connect(bx, SIGNAL("accepted()"), dlg.accept) self.connect(bx, SIGNAL("rejected()"), dlg.reject) h1 = QtGui.QHBoxLayout() h1.addStretch() h1.addWidget(bx) v1 = QtGui.QVBoxLayout() v1.addLayout(fl) v1.addLayout(h1) dlg.setLayout(v1) if dlg.exec_(): self.newElementPlots(str(elem.text()), str(fld.text()), machlat=(mach, lat), dtype = str(dtype.currentText())) def click_markfam(self, on): famname = self.sender().text() mks = [] mach, lat = self.getCurrentMachLattice() # need to convert to python str for elem in lat.getElementList(str(famname)): if elem.family != famname: continue if elem.virtual: continue mks.append([elem.name, 0.5*(elem.sb+elem.se)]) for w in self.mdiarea.subWindowList(): w.setMarkers(mks, on) #print self._machlat.keys() def _reset_correctors(self): self.logger.info("reset correctors") aphla.hlalib._reset_trims() def _reset_quadrupoles(self): self.logger.info("reset quadrupoles") aphla.hlalib._reset_quad() def _random_hkick(self): mach, lat = self.getCurrentMachLattice() hcors = lat.getElementList('HCOR') for k in range(3): i = np.random.randint(len(hcors)) self.logger.info("Setting {0}/{1} HCOR".format(i, len(hcors))) hcors[i].x += np.random.rand() * 2e-6 def _random_vkick(self): mach, lat = self.getCurrentMachLattice() cors = lat.getElementList('VCOR') for k in range(3): i = np.random.randint(len(cors)) cors[i].y += np.random.rand() * 1e-6 self.logger.info("increased kicker '{0}' by 1e-7 ({1} {2})".format( cors[i].name, cors[i].y, cors[i].getUnit('y', None))) def viewDcctPlot(self, on): self.dcct.setVisible(on) def liveData(self, on): """Switch on/off live data taking""" self.live_orbit = on def scalePlot(self): w = self.mdiarea.currentSubWindow() if not w: return st, p = self.sender().text(), w.aplot if st == "Fit": p.scaleXBottom() p.scaleYLeft() # a hack bound = p.curvesBound() p.zoomer1.setZoomStack([bound]) elif st == "Fit In Y": p.scaleYLeft() elif st == "Zoom In Y": p.scaleYLeft(1./1.5) elif st == "Zoom Out Y": p.scaleYLeft(1.5) elif st == "Move Up": p.moveCurves(Qwt.QwtPlot.yLeft, 0.8) elif st == "Move Down": p.moveCurves(Qwt.QwtPlot.yLeft, -0.8) elif st == "Fit In X": p.scaleXBottom() elif st == "Zoom In X": p.scaleXBottom(1.0/1.5) elif st == "Zoom Out X": p.scaleXBottom(1.5) elif st == "Move Left": p.moveCurves(Qwt.QwtPlot.xBottom, 0.8) elif st == "Move Right": p.moveCurves(Qwt.QwtPlot.xBottom, -0.8) else: self.logger.error("unknow action '{0}'".format(st)) def getVisibleRange(self): w = self.mdiarea.currentSubWindow() if not w: mach, lat = self.getCurrentMachLattice() self.logger.warn("no active plot, use full range of {0}.{1}".format( mach, lat.name)) return lat.getLocationRange() else: return w.currentXlim() def getVisibleElements(self, elemname, sb = None, se = None): w = self.mdiarea.currentSubWindow() mach, lat = self.getCurrentMachLattice() elems = lat.getElementList(elemname) if sb is not None: elems = [e for e in elems if e.sb >= sb] if se is not None: elems = [e for e in elems if e.se <= se] self.logger.info("searching for '{0}' in range [{1}, {2}]".format( elemname, sb, se)) return elems def timerEvent(self, e): if e.timerId() != self.timerId: return #if not self.elemeditor.isHidden(): # self.elemeditor.updateModelData() #if self.live_orbit: # self.itimer += 1 # #self.updatePlots() # #self.updateStatus() # for w in self.mdiarea.subWindowList(): # if not isinstance(w, ApMdiSubPlot): continue # if not w.live: continue # w.updatePlot() # self.statusBar().showMessage("plot updated: {0}".format( # time.strftime("%F %T"))) #else: # self.statusBar().showMessage("live update disabled") def singleShot(self): for w in self.mdiarea.subWindowList(): if not isinstance(w, ApMdiSubPlot): continue w.updatePlot() self.statusBar().showMessage("plot updated: {0}".format( time.strftime("%F %T"))) def elementSelected(self, elems): """this action is ignored""" mach, lat, elemnames = elems #_lat = self._machlat[mach][lat] self.logger.info("element selected") #elemobjs = _lat.getElementList(elemnames) #self._elemed.addElements(elemobjs) def activeOrbitPlot(self, field): mach = str(self.machBox.currentText()) lat = str(self.latBox.currentText()) for w in self.mdiarea.subWindowList(): #print w.machine(), w.lattice(), w.data.yfield if not isinstance(w, ApMdiSubPlot): continue if w.machine() != mach: continue if w.lattice() != lat: continue if w.data.yfield != field: continue return w return None def createLocalBump_(self): """create local bump""" if self._dlgOrbitCor is None: bpms = ap.getElements("BPM") cors = ap.getElements("COR") self._dlgOrbitCor = OrbitCorrDlg(bpms, cors) #corbitdlg.resize(600, 500) self._dlgOrbitCor.setWindowTitle("Create Local Bump") self._dlgOrbitCor.show() self._dlgOrbitCor.raise_() self._dlgOrbitCor.activateWindow() def createLocalBump(self): """create local bump""" bpms = ap.getElements("BPM") cors = ap.getElements("COR") dlgOrbitCor = OrbitCorrDlg(bpms, cors, parent=self) #corbitdlg.resize(600, 500) dlgOrbitCor.setWindowTitle("Create Local Bump") dlgOrbitCor.show() dlgOrbitCor.raise_() dlgOrbitCor.activateWindow() dlgOrbitCor.setAttribute(Qt.WA_DeleteOnClose) def runBba(self): mach, lat = self.getCurrentMachLattice() bpms = [e for e in lat.getElementList('BPM') if e not in self.physics.deadelems] self.physics.runBba(bpms) def plotSVD(self): mach, lat = self.getCurrentMachLattice() if not lat.ormdata: QMessageBox.critical(self, "ORM SVD", "machine '%s' ORM data is not available" % \ mach, QMessageBox.Ok) return m, brec, trec = lat.ormdata.getMatrix(None, None, full=False, ignore=self.getDeadElements()) U, s, V = np.linalg.svd(m, full_matrices=True) #print np.shape(s), s self.sp = ApSvdPlot(s) self.sp.show() def tileSubWindowsHorizontally(self): pos = QtCore.QPoint(0, 0) subwins = self.mdiarea.subWindowList() for w in subwins: height = self.mdiarea.height()/len(subwins) rect = QtCore.QRect(0, 0, self.mdiarea.width(), height) w.setGeometry(rect) w.move(pos) pos.setY(pos.y() + w.height())
def lauchGUI(self, WorkSpace, aCase, sobjXML, Args): """ mw.dockWidgetBrowser is the Browser of the CFD MainView """ log.debug("lauchGUI") from cs_gui import process_cmd_line if CFD_Code() == CFD_Saturne: from cs_package import package from code_saturne.Base.MainView import MainView elif CFD_Code() == CFD_Neptune: from nc_package import package from neptune_cfd.core.MainView import MainView if sobjXML == None: Title = "unnamed" else: Title = sobjXML.GetName() self.Workspace = WorkSpace pkg = package() case, splash = process_cmd_line(Args) mw = MainView(pkg, case, aCase) # Put the standard panel of the MainView inside a QDockWidget # in the SALOME Desktop aTitle = self.setWindowTitle_CFD(mw, aCase, Title) dsk = sgPyQt.getDesktop() dock = QDockWidget(aTitle) dock.setWidget(mw.frame) dock.setMinimumWidth(520) dsk.addDockWidget(Qt.RightDockWidgetArea, dock) dock.setVisible(True) dock.show() # Put the QTreeView of the MainView which is already inside a QDockWidget # in the SALOME Desktop BrowserTitle = aTitle + " Browser" mw.dockWidgetBrowser.setWindowTitle(BrowserTitle) dsk.addDockWidget(Qt.LeftDockWidgetArea, mw.dockWidgetBrowser) mw.dockWidgetBrowser.setVisible(True) mw.dockWidgetBrowser.show() mw.dockWidgetBrowser.raise_() dock.raise_() #Add Dock windows are managed by CFDGUI_Management class studyId = sgPyQt.getStudyId() aStudyCFD = aCase.GetFather() aCaseCFD = aCase xmlFileName = str(Title) _c_CFDGUI.set_d_CfdCases(studyId, dock, mw.dockWidgetBrowser, mw, aStudyCFD, aCaseCFD, xmlFileName, sobjXML) self.connect(dock, SIGNAL("visibilityChanged(bool)"), self.setdockWindowBrowserActivated) self.connect(mw.dockWidgetBrowser, SIGNAL("visibilityChanged(bool)"), self.setdockWindowActivated) self.connect(dock.toggleViewAction(), SIGNAL("toggled(bool)"), self.setdockWB) self.connect(mw.dockWidgetBrowser.toggleViewAction(), SIGNAL("toggled(bool)"), self.setdock) _c_CFDGUI.tabifyDockWindows(dsk, studyId) self.showDockWindows(studyId, xmlFileName, aCaseCFD.GetName(), aStudyCFD.GetName()) updateObjectBrowser() return mw