def run():
# Simplify TaurusGui
TaurusGui.TAURUS_MENU_ENABLED = False
TaurusGui.TOOLS_MENU_ENABLED = False
TaurusGui.HELP_MENU_ENABLED = False
TaurusGui.FULLSCREEN_TOOLBAR_ENABLED = False
TaurusGui.PANELS_MENU_ENABLED = False
TaurusGui.APPLETS_TOOLBAR_ENABLED = False
TaurusGui.QUICK_ACCESS_TOOLBAR_ENABLED = False
app = TaurusApplication(app_name='tpgArchiving')
gui = TaurusGui()
gui.setWindowTitle('Taurus Archiving Plot')
plot = TaurusPlot()
#class MyPlot(TaurusPlot):
#pass
##def dropEvent(self, event):
##print('In dropEvent(%s)' % str(dir(event)))
##def dragMoveEvent(self,event):
##print('Dragging ...')
##event.acceptProposedAction()
##def dragEnterEvent(self,event):
##print('Entering ...')
##event.acceptProposedAction()
#plot = MyPlot()
#plot.setAcceptDrops(True)
#plot = fandango.qt.Dropable(TaurusPlot)()
#plot.setDropLogger(fandango.printf)
#plot.setSupportedMimeTypes(fandango.qt.MIMETYPES)
plot.setBackgroundBrush(Qt.QColor('white'))
axis = DateAxisItem(orientation='bottom')
plot_items = plot.getPlotItem()
axis.attachToPlotItem(plot_items)
# TODO (cleanup menu actions)
if plot_items.legend is not None:
plot_items.legend.hide()
msi = TangoArchivingModelSelectorItem()
# hide time stamp selector
msi.ui.ts_checkBox.setVisible(False)
# hide add single Y model
msi._addSelectedAction.setVisible(False)
msi.setModel(msi.default_model)
# progress bar
pb = Qt.QProgressBar()
pb.setGeometry(0, 0, 300, 25)
def updateProgressBar(stop=True):
if stop is True:
final = 1
else:
final = 0
pb.setRange(0, final)
updateProgressBar()
###########################################################################
# Update t0 and t1 based on sigXRangeChanged
###########################################################################
def onUpdateXViewRange():
x, _ = plot.viewRange()
t0, t1 = x
t0s = str2localtime(t0)
t1s = str2localtime(t1)
msi.time_selector.ui.comboBox_begin.setItemText(5, t0s)
msi.time_selector.ui.comboBox_end.setItemText(7, t1s)
msi.time_selector.ui.comboBox_begin.setItemText(5, t0s)
msi.time_selector.ui.comboBox_end.setItemText(7, t1s)
msi.time_selector.ui.comboBox_begin.setCurrentIndex(5)
msi.time_selector.ui.comboBox_end.setCurrentIndex(7)
vb = plot.getPlotItem().getViewBox()
vb.sigXRangeChanged.connect(onUpdateXViewRange)
###########################################################################
# Legend
###########################################################################
gv = Qt.QGraphicsView(Qt.QGraphicsScene())
gv.setBackgroundBrush(Qt.QBrush(Qt.QColor('white')))
l = pg.LegendItem(None, offset=(0, 0))
gv.scene().addItem(l)
def updateExternalLegend():
for dataitem in plot_items.listDataItems():
l.removeItem(dataitem.name())
for dataitem in plot_items.listDataItems():
if dataitem.name():
l.addItem(dataitem, dataitem.name())
###########################################################################
# Connect CurvesAppearanceChooser to external legend
###########################################################################
from taurus_pyqtgraph.curveproperties import (CurvesAppearanceChooser,
CurveAppearanceProperties)
def onApply(self):
names = self.getSelectedCurveNames()
prop = self.getShownProperties()
# Update self.curvePropDict for selected properties
for n in names:
self.curvePropDict[n] = CurveAppearanceProperties.merge(
[self.curvePropDict[n], prop],
conflict=CurveAppearanceProperties.inConflict_update_a)
# emit a (PyQt) signal telling what properties (first argument) need to
# be applied to which curves (second argument)
# self.curveAppearanceChanged.emit(prop, names)
# return both values
self.curvePropAdapter.setCurveProperties(self.curvePropDict, names)
# Update legend
updateExternalLegend()
return prop, names
# Override CurvesAppearanceChooser.onApply
CurvesAppearanceChooser.onApply = onApply
###########################################################################
# Helper
###########################################################################
def updateAll(legend=True):
# Update legend
if legend is True:
updateExternalLegend()
# run plot auto range
time.sleep(0.2) # Wait till models are loading
plot_items.getViewBox().menu.autoRange()
# Stop progress bar
updateProgressBar()
###########################################################################
# onAddXYModel
###########################################################################
def onAddXYModel(models=None):
# Update progress bar
# updateProgressBar(False)
print('onAddXYModel(%s)' % models)
if not isSequenceType(models):
print('Overriding models ...')
models = msi.getSelectedModels()
c = msi.cursor()
msi.setCursor(Qt.Qt.WaitCursor)
current = plot._model_chooser_tool.getModelNames()
print('current: %s' % str(current))
models = [m for m in models if m not in current]
print('new models: %s' % str(models))
plot.addModels(models)
updateAll()
msi.setCursor(c)
# Connect button
msi.modelsAdded.connect(onAddXYModel)
#plot.setDropEventCallback(onAddXYModel)
###########################################################################
# Override TaurusGui close event
###########################################################################
def closeEvent(self, event):
try:
self.__macroBroker.removeTemporaryPanels()
except:
pass
TaurusMainWindow.closeEvent(self, event)
for n, panel in self._TaurusGui__panels.items():
panel.closeEvent(event)
panel.widget().closeEvent(event)
if not event.isAccepted():
result = Qt.QMessageBox.question(
self, 'Closing error',
"Panel '%s' cannot be closed. Proceed closing?" % n,
Qt.QMessageBox.Yes | Qt.QMessageBox.No)
if result == Qt.QMessageBox.Yes:
event.accept()
else:
break
for curve in plot_items.listDataItems():
plot.removeItem(curve)
l.removeItem(curve.name())
gui.saveSettings()
# Override TaurusGui close event
TaurusGui.closeEvent = closeEvent
###########################################################################
# Create tgarch tool bar
###########################################################################
def _onRefresh():
t0, t1 = msi.time_selector.getTimes()
# Validate models
v = TangoArchivingAttributeNameValidator()
query = "{0};t0={1};t1={2}"
for curve in plot.getPlotItem().listDataItems():
if isinstance(curve, TaurusPlotDataItem):
ymodel = curve.getModel()
# tgarch attr
if v.getUriGroups(ymodel).get('scheme') != 'tgarch':
continue
fullname, _, _ = v.getNames(ymodel)
bmodel, current_query = fullname.split('?')
db = current_query.split(';')[0]
q = query.format(db, t0, t1)
model = "{0}?{1}".format(bmodel, q)
xmodel = "{};ts".format(model)
curve.setModel(None)
curve.setXModel(None)
curve.setModel(model)
curve.setXModel(xmodel)
updateAll(legend=False)
def onRefresh():
# Update progress bar
updateProgressBar(False)
t1 = threading.Thread(target=_onRefresh)
t1.start()
tgarchToolBar = gui.addToolBar("TgArch")
tgarchToolBar.addAction(Qt.QIcon.fromTheme("view-refresh"),
"refresh tgarch curves", onRefresh)
###########################################################################
# Create panels
###########################################################################
gui.createPanel(plot, 'Plot')
gui.createPanel(msi, 'Chooser')
gui.createPanel(pb, 'Progress bar')
gui.createPanel(gv, 'Legend')
# Load configuration
gui.loadSettings()
gui.show()
app.exec_()
