1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
__author__ = 'Tonn Rueter'
try:
from PyMca import Plugin1DBase
except ImportError:
from . import Plugin1DBase
try:
from PyMca.PyMcaPlugins import MotorInfoWindow
except ImportError:
try:
# Frozen version
from PyMcaPlugins import MotorInfoWindow
except:
print("MotorInfoPlugin importing from somewhere else")
import MotorInfoWindow
DEBUG = 0
class MotorInfo(Plugin1DBase.Plugin1DBase):
def __init__(self, plotWindow, **kw):
Plugin1DBase.Plugin1DBase.__init__(self, plotWindow, **kw)
self.methodDict = {}
text = 'Show values of various motors.'
function = self.showMotorInfo
icon = None
info = text
self.methodDict["Show Motor Info"] =[function, info, icon]
self.widget = None
def getMethods(self, plottype=None):
names = list(self.methodDict.keys())
names.sort()
return names
def getMethodToolTip(self, name):
return self.methodDict[name][1]
def getMethodPixmap(self, name):
return self.methodDict[name][2]
def applyMethod(self, name):
self.methodDict[name][0]()
return
def showMotorInfo(self):
legendList, motorValuesList = self._getLists()
if self.widget is None:
self._createWidget(legendList, motorValuesList)
else:
self.widget.table.updateTable(legendList, motorValuesList)
self.widget.show()
self.widget.raise_()
def _getLists(self):
curves = self.getAllCurves()
nCurves = len(curves)
if DEBUG:
print ("Received %d curve(s).." % nCurves)
legendList = [leg for (xvals, yvals, leg, info) in curves]
infoList = [info for (xvals, yvals, leg, info) in curves]
motorValuesList = self._convertInfoDictionary( infoList )
return legendList, motorValuesList
def _convertInfoDictionary(self, infosList):
ret = []
for info in infosList :
motorNames = info.get('MotorNames', None)
if motorNames is not None:
if type(motorNames) == str:
namesList = motorNames.split()
elif type(motorNames) == list:
namesList = motorNames
else:
namesList = []
else:
namesList = []
motorValues = info.get('MotorValues', None)
if motorNames is not None:
if type(motorValues) == str:
valuesList = motorValues.split()
elif type(motorValues) == list:
valuesList = motorValues
else:
valuesList = []
else:
valuesList = []
if len(namesList) == len(valuesList):
ret.append( dict( zip( namesList, valuesList ) ) )
else:
print("Number of motors and values does not match!")
return ret
def _createWidget(self, legendList, motorValuesList):
parent = None
self.widget = MotorInfoWindow.MotorInfoDialog(parent,
legendList,
motorValuesList)
self.widget.buttonUpdate.clicked.connect(self.showMotorInfo)
MENU_TEXT = "Motor Info"
def getPlugin1DInstance(plotWindow, **kw):
ob = MotorInfo(plotWindow)
return ob
if __name__ == "__main__":
# Basic test setup
import numpy
from PyMca import Plot1D
from PyMca import PyMcaQt as qt
app = qt.QApplication([])
x = numpy.arange(100.)
y = numpy.arange(100.)
plot = Plot1D.Plot1D()
plot.addCurve(x, y, "Curve1", {'MotorNames': "foo bar", 'MotorValues': "3.14 2.97"})
plot.addCurve(x+100, y, "Curve2", {'MotorNames': "baz", 'MotorValues': "6.28"})
plugin = getPlugin1DInstance(plot)
plugin.applyMethod(plugin.getMethods()[0])
app.exec_()
|
