Rev 593 | Rev 596 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
584 | FredericG | 1 | #!/usr/bin/env python |
2 | # -*- coding: iso-8859-15 -*- |
||
3 | # generated by wxGlade 0.6.3 on Thu Sep 24 15:46:36 2009 |
||
4 | |||
586 | FredericG | 5 | import sys |
584 | FredericG | 6 | import wx |
585 | FredericG | 7 | import wx.lib |
8 | import wx.lib.plot |
||
584 | FredericG | 9 | |
585 | FredericG | 10 | # Needs Numeric or numarray or NumPy |
11 | try: |
||
12 | import numpy.oldnumeric as _Numeric |
||
13 | except: |
||
14 | try: |
||
15 | import numarray as _Numeric #if numarray is used it is renamed Numeric |
||
16 | except: |
||
17 | try: |
||
18 | import Numeric as _Numeric |
||
19 | except: |
||
20 | msg= """ |
||
21 | This module requires the Numeric/numarray or NumPy module, |
||
22 | which could not be imported. It probably is not installed |
||
23 | (it's not part of the standard Python distribution). See the |
||
24 | Numeric Python site (http://numpy.scipy.org) for information on |
||
25 | downloading source or binaries.""" |
||
26 | raise ImportError, "Numeric,numarray or NumPy not found. \n" + msg |
||
27 | |||
584 | FredericG | 28 | # begin wxGlade: extracode |
29 | # end wxGlade |
||
30 | |||
31 | |||
32 | |||
587 | FredericG | 33 | class MenuBar(wx.MenuBar): |
34 | def __init__(self, *args, **kwds): |
||
35 | # content of this block not found: did you rename this class? |
||
36 | pass |
||
37 | |||
38 | def __set_properties(self): |
||
39 | # content of this block not found: did you rename this class? |
||
40 | pass |
||
41 | |||
42 | def __do_layout(self): |
||
43 | # content of this block not found: did you rename this class? |
||
44 | pass |
||
45 | |||
46 | # end of class MenuBar |
||
47 | |||
48 | |||
585 | FredericG | 49 | class wxFrame(wx.Panel): |
50 | def __init__(self, *args, **kwds): |
||
586 | FredericG | 51 | # content of this block not found: did you rename this class? |
52 | pass |
||
585 | FredericG | 53 | |
54 | def __set_properties(self): |
||
586 | FredericG | 55 | # content of this block not found: did you rename this class? |
585 | FredericG | 56 | pass |
57 | |||
58 | def __do_layout(self): |
||
586 | FredericG | 59 | # content of this block not found: did you rename this class? |
585 | FredericG | 60 | pass |
61 | |||
62 | # end of class wxFrame |
||
63 | |||
64 | |||
65 | class MyFrame(wx.Frame): |
||
66 | def __init__(self, *args, **kwds): |
||
67 | # content of this block not found: did you rename this class? |
||
68 | pass |
||
69 | |||
70 | def __set_properties(self): |
||
71 | # content of this block not found: did you rename this class? |
||
72 | pass |
||
73 | |||
74 | def __do_layout(self): |
||
75 | # content of this block not found: did you rename this class? |
||
76 | pass |
||
77 | |||
78 | # end of class MyFrame |
||
79 | |||
80 | |||
584 | FredericG | 81 | class MainFrame(wx.Frame): |
82 | def __init__(self, *args, **kwds): |
||
83 | # begin wxGlade: MainFrame.__init__ |
||
84 | kwds["style"] = wx.DEFAULT_FRAME_STYLE |
||
85 | wx.Frame.__init__(self, *args, **kwds) |
||
587 | FredericG | 86 | |
87 | # Menu Bar |
||
88 | self.frame_1_menubar = wx.MenuBar() |
||
89 | wxglade_tmp_menu = wx.Menu() |
||
90 | wxglade_tmp_menu.Append(wx.NewId(), "Exit", "", wx.ITEM_NORMAL) |
||
91 | self.frame_1_menubar.Append(wxglade_tmp_menu, "File") |
||
92 | wxglade_tmp_menu = wx.Menu() |
||
93 | wxglade_tmp_menu.Append(wx.NewId(), "Clear", "", wx.ITEM_NORMAL) |
||
94 | wxglade_tmp_menu.Append(wx.NewId(), "Import", "", wx.ITEM_NORMAL) |
||
95 | self.frame_1_menubar.Append(wxglade_tmp_menu, "TestSet") |
||
96 | wxglade_tmp_menu = wx.Menu() |
||
97 | self.frame_1_menubar.Append(wxglade_tmp_menu, "MK") |
||
98 | wxglade_tmp_menu = wx.Menu() |
||
99 | self.frame_1_menubar.Append(wxglade_tmp_menu, "Help") |
||
100 | self.SetMenuBar(self.frame_1_menubar) |
||
101 | # Menu Bar end |
||
584 | FredericG | 102 | self.Port = wx.StaticText(self, -1, "Port") |
103 | self.text_ctrl_1 = wx.TextCtrl(self, -1, "") |
||
104 | self.label_5 = wx.StaticText(self, -1, "label_5") |
||
105 | self.text_ctrl_5 = wx.TextCtrl(self, -1, "") |
||
106 | self.label_2 = wx.StaticText(self, -1, "Motor(s)") |
||
107 | self.text_ctrl_2 = wx.TextCtrl(self, -1, "") |
||
108 | self.label_6 = wx.StaticText(self, -1, "label_6") |
||
109 | self.text_ctrl_6 = wx.TextCtrl(self, -1, "") |
||
110 | self.label_3 = wx.StaticText(self, -1, "Channel(s)") |
||
111 | self.text_ctrl_3 = wx.TextCtrl(self, -1, "") |
||
112 | self.label_7 = wx.StaticText(self, -1, "label_7") |
||
113 | self.text_ctrl_7 = wx.TextCtrl(self, -1, "") |
||
114 | self.label_4 = wx.StaticText(self, -1, "Speed(s)") |
||
115 | self.text_ctrl_4 = wx.TextCtrl(self, -1, "") |
||
116 | self.label_8 = wx.StaticText(self, -1, "label_8") |
||
117 | self.text_ctrl_8 = wx.TextCtrl(self, -1, "") |
||
118 | self.button_1 = wx.Button(self, -1, "button_1") |
||
591 | FredericG | 119 | self.GraphPanel = wx.Panel(self, -1, style=wx.DOUBLE_BORDER|wx.TAB_TRAVERSAL) |
592 | FredericG | 120 | self.yAxesChoice = wx.ComboBox(self, -1, choices=["10", "25", "50", "100", "250"], style=wx.CB_DROPDOWN) |
121 | self.graphTypeChoise = wx.ComboBox(self, -1, choices=["Raw Data", "Filtered Data", "Spectrum"], style=wx.CB_DROPDOWN) |
||
586 | FredericG | 122 | self.TestListCtrl = wx.ListCtrl(self, -1, style=wx.LC_REPORT|wx.SUNKEN_BORDER) |
584 | FredericG | 123 | |
124 | self.__set_properties() |
||
125 | self.__do_layout() |
||
587 | FredericG | 126 | |
127 | self.Bind(wx.EVT_MENU, self.OnImport, id=-1) |
||
592 | FredericG | 128 | self.Bind(wx.EVT_COMBOBOX, self.onYAxesChange, self.yAxesChoice) |
584 | FredericG | 129 | # end wxGlade |
130 | |||
587 | FredericG | 131 | def setApp(self, app): |
132 | self.app = app |
||
585 | FredericG | 133 | |
584 | FredericG | 134 | def __set_properties(self): |
135 | # begin wxGlade: MainFrame.__set_properties |
||
587 | FredericG | 136 | self.SetTitle("VibrationTest") |
591 | FredericG | 137 | self.GraphPanel.SetMinSize((800,300)) |
592 | FredericG | 138 | self.yAxesChoice.SetSelection(2) |
139 | self.graphTypeChoise.SetSelection(0) |
||
584 | FredericG | 140 | # end wxGlade |
141 | |||
142 | def __do_layout(self): |
||
143 | # begin wxGlade: MainFrame.__do_layout |
||
144 | sizer_1 = wx.BoxSizer(wx.VERTICAL) |
||
586 | FredericG | 145 | grid_sizer_2 = wx.GridSizer(2, 1, 4, 4) |
591 | FredericG | 146 | sizer_2 = wx.BoxSizer(wx.HORIZONTAL) |
584 | FredericG | 147 | grid_sizer_1 = wx.GridSizer(5, 4, 4, 4) |
591 | FredericG | 148 | sizer_1.Add((20, 20), 0, 0, 0) |
584 | FredericG | 149 | grid_sizer_1.Add(self.Port, 0, wx.ALIGN_RIGHT, 0) |
150 | grid_sizer_1.Add(self.text_ctrl_1, 0, 0, 0) |
||
151 | grid_sizer_1.Add(self.label_5, 0, wx.ALIGN_RIGHT, 0) |
||
152 | grid_sizer_1.Add(self.text_ctrl_5, 0, 0, 0) |
||
153 | grid_sizer_1.Add(self.label_2, 0, wx.ALIGN_RIGHT, 0) |
||
154 | grid_sizer_1.Add(self.text_ctrl_2, 0, 0, 0) |
||
155 | grid_sizer_1.Add(self.label_6, 0, wx.ALIGN_RIGHT, 0) |
||
156 | grid_sizer_1.Add(self.text_ctrl_6, 0, 0, 0) |
||
157 | grid_sizer_1.Add(self.label_3, 0, wx.ALIGN_RIGHT, 0) |
||
158 | grid_sizer_1.Add(self.text_ctrl_3, 0, 0, 0) |
||
159 | grid_sizer_1.Add(self.label_7, 0, wx.ALIGN_RIGHT, 0) |
||
160 | grid_sizer_1.Add(self.text_ctrl_7, 0, 0, 0) |
||
161 | grid_sizer_1.Add(self.label_4, 0, wx.ALIGN_RIGHT, 0) |
||
162 | grid_sizer_1.Add(self.text_ctrl_4, 0, 0, 0) |
||
163 | grid_sizer_1.Add(self.label_8, 0, wx.ALIGN_RIGHT, 0) |
||
164 | grid_sizer_1.Add(self.text_ctrl_8, 0, 0, 0) |
||
165 | grid_sizer_1.Add(self.button_1, 0, 0, 0) |
||
166 | sizer_1.Add(grid_sizer_1, 0, 0, 2) |
||
591 | FredericG | 167 | sizer_1.Add((20, 20), 0, 0, 0) |
592 | FredericG | 168 | sizer_1.Add(self.GraphPanel, 1, 0, 0) |
169 | sizer_2.Add(self.yAxesChoice, 0, 0, 0) |
||
170 | sizer_2.Add(self.graphTypeChoise, 0, 0, 0) |
||
591 | FredericG | 171 | sizer_1.Add(sizer_2, 0, wx.EXPAND, 0) |
172 | sizer_1.Add((20, 20), 0, 0, 0) |
||
586 | FredericG | 173 | grid_sizer_2.Add(self.TestListCtrl, 1, wx.ALL|wx.EXPAND, 0) |
174 | sizer_1.Add(grid_sizer_2, 1, wx.EXPAND, 0) |
||
584 | FredericG | 175 | self.SetSizer(sizer_1) |
176 | sizer_1.Fit(self) |
||
177 | self.Layout() |
||
178 | # end wxGlade |
||
179 | |||
590 | FredericG | 180 | # List events |
181 | self.TestListCtrl.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnTestSelected, self.TestListCtrl) |
||
182 | |||
586 | FredericG | 183 | # Configure Graph |
593 | FredericG | 184 | self.client = wx.lib.plot.PlotCanvas(self.GraphPanel, size=(800,300)) |
185 | |||
585 | FredericG | 186 | self.client.SetPointLabelFunc(self.DrawPointLabel) |
187 | |||
188 | self.client.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.NORMAL)) |
||
189 | self.client.SetFontSizeAxis(10) |
||
190 | self.client.SetFontSizeLegend(7) |
||
191 | self.client.setLogScale((False,False)) |
||
192 | |||
586 | FredericG | 193 | |
194 | # Configure TestListCtrl |
||
594 | FredericG | 195 | self.TestListCtrl.InsertColumn(0, "Description") |
196 | self.TestListCtrl.InsertColumn(1, "Speed") |
||
197 | self.TestListCtrl.InsertColumn(2, "Channel") |
||
198 | self.TestListCtrl.InsertColumn(3, "Result") |
||
586 | FredericG | 199 | |
585 | FredericG | 200 | def DrawPointLabel(self, dc, mDataDict): |
201 | """This is the fuction that defines how the pointLabels are plotted |
||
202 | dc - DC that will be passed |
||
203 | mDataDict - Dictionary of data that you want to use for the pointLabel |
||
204 | |||
205 | As an example I have decided I want a box at the curve point |
||
206 | with some text information about the curve plotted below. |
||
207 | Any wxDC method can be used. |
||
208 | """ |
||
209 | # ---------- |
||
210 | dc.SetPen(wx.Pen(wx.BLACK)) |
||
211 | dc.SetBrush(wx.Brush( wx.BLACK, wx.SOLID ) ) |
||
212 | |||
213 | sx, sy = mDataDict["scaledXY"] #scaled x,y of closest point |
||
214 | dc.DrawRectangle( sx-5,sy-5, 10, 10) #10by10 square centered on point |
||
215 | px,py = mDataDict["pointXY"] |
||
216 | cNum = mDataDict["curveNum"] |
||
217 | pntIn = mDataDict["pIndex"] |
||
218 | legend = mDataDict["legend"] |
||
219 | #make a string to display |
||
220 | s = "Crv# %i, '%s', Pt. (%.2f,%.2f), PtInd %i" %(cNum, legend, px, py, pntIn) |
||
221 | dc.DrawText(s, sx , sy+1) |
||
222 | # ----------- |
||
223 | |||
590 | FredericG | 224 | def OnTestSelected(self, event): |
225 | testId = event.m_itemIndex |
||
226 | print "Test Selected id=%d" % (testId) |
||
592 | FredericG | 227 | self.activeTestId = testId |
228 | self.drawGraph() |
||
590 | FredericG | 229 | |
592 | FredericG | 230 | def drawGraph(self): |
231 | vibTest = self.app.getTest(self.activeTestId) |
||
590 | FredericG | 232 | nb = vibTest.getDataLen() |
592 | FredericG | 233 | y = int(self.yAxesChoice.GetStringSelection()) |
590 | FredericG | 234 | |
593 | FredericG | 235 | #self.client.Clear() |
236 | xydata = _Numeric.linspace(0,0.09*nb,2*nb) |
||
590 | FredericG | 237 | xydata.shape = (nb, 2) |
238 | xydata[:,1] = vibTest.getRawData() |
||
239 | line = wx.lib.plot.PolyLine(xydata, legend= 'Raw Data', colour='red') |
||
593 | FredericG | 240 | self.client.Draw(wx.lib.plot.PlotGraphics([line],"Graph Title", "Time (ms)", "Acc"), yAxis= (-y,y)) |
592 | FredericG | 241 | self.client.SetEnableGrid('Horizontal') |
590 | FredericG | 242 | |
243 | |||
587 | FredericG | 244 | def OnImport(self, event): # wxGlade: MainFrame.<event_handler> |
245 | self.app.Import() |
||
246 | |||
592 | FredericG | 247 | def onYAxesChange(self, event): # wxGlade: MainFrame.<event_handler> |
248 | self.drawGraph() |
||
249 | |||
584 | FredericG | 250 | # end of class MainFrame |
251 | |||
252 | |||
586 | FredericG | 253 | class VibTest: |
594 | FredericG | 254 | def __init__(self, descr, motor, speed, rawData): |
255 | self.descr = descr |
||
256 | self.motor = motor |
||
257 | self.speed = speed |
||
590 | FredericG | 258 | self.rawData = _Numeric.array(rawData) |
259 | self.dc = self.rawData.mean() |
||
260 | self.rawData -= self.dc |
||
261 | self.dataLen = len(rawData) |
||
586 | FredericG | 262 | |
594 | FredericG | 263 | def getDescr(self): |
264 | return self.Descr |
||
587 | FredericG | 265 | |
590 | FredericG | 266 | def getRawData(self): |
267 | return self.rawData |
||
268 | |||
269 | def getDataLen(self): |
||
270 | return self.dataLen |
||
271 | |||
272 | |||
584 | FredericG | 273 | class App(wx.App): |
586 | FredericG | 274 | def __init__(self, par): |
275 | self.VibTests = [] |
||
276 | |||
277 | wx.App.__init__(self, par) |
||
278 | |||
279 | |||
594 | FredericG | 280 | def AddTest(self, descr, motor, speed, rawData): |
281 | test = VibTest(descr, motor, speed, rawData) |
||
586 | FredericG | 282 | self.VibTests.append(test) |
283 | |||
594 | FredericG | 284 | index = self.frame_1.TestListCtrl.InsertStringItem(sys.maxint, test.descr) |
285 | self.frame_1.TestListCtrl.SetStringItem(index, 3, "OK") |
||
586 | FredericG | 286 | |
590 | FredericG | 287 | def getTest(self, testId): |
288 | return self.VibTests[testId] |
||
289 | |||
584 | FredericG | 290 | def OnInit(self): |
291 | wx.InitAllImageHandlers() |
||
586 | FredericG | 292 | self.frame_1 = MainFrame(None, -1, "") |
587 | FredericG | 293 | self.frame_1.setApp(self); |
586 | FredericG | 294 | self.SetTopWindow(self.frame_1) |
295 | |||
587 | FredericG | 296 | self.frame_1.Show() |
297 | return 1 |
||
298 | |||
299 | def Import(self): |
||
300 | print "Import" |
||
594 | FredericG | 301 | logfile = open("./unbal200.txt", "r") |
590 | FredericG | 302 | data = None |
303 | for line in logfile: |
||
304 | values = line.split(',') |
||
305 | if data == None: |
||
306 | nbCols = len(values) |
||
307 | print "%d cols" % nbCols |
||
308 | data = [] |
||
309 | for i in range(nbCols): |
||
310 | data.append([]) |
||
594 | FredericG | 311 | else: |
312 | for i in range(nbCols): |
||
313 | data[i].append(int(values[i])) |
||
590 | FredericG | 314 | logfile.close() |
315 | for i in range(nbCols): |
||
594 | FredericG | 316 | self.AddTest("Imported %d"%i, 0,0, data[i]) |
590 | FredericG | 317 | |
318 | |||
319 | |||
586 | FredericG | 320 | |
590 | FredericG | 321 | |
584 | FredericG | 322 | # end of class App |
323 | |||
324 | if __name__ == "__main__": |
||
325 | VibrationTestGui = App(0) |
||
326 | VibrationTestGui.MainLoop() |