Subversion Repositories Projects

Rev

Rev 591 | Rev 593 | 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
591 FredericG 184
        self.client = wx.lib.plot.PlotCanvas(self.GraphPanel)
585 FredericG 185
        self.client.SetPointLabelFunc(self.DrawPointLabel)
186
 
187
        self.client.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.NORMAL))
188
        self.client.SetFontSizeAxis(10)
189
        self.client.SetFontSizeLegend(7)
190
        self.client.setLogScale((False,False))
191
        self.client.SetXSpec('auto')
192
        self.client.SetYSpec('auto')
193
 
194
        data1 = 2.*_Numeric.pi*_Numeric.arange(200)/200.
195
        data1.shape = (100, 2)
196
        data1[:,1] = _Numeric.sin(data1[:,0])
197
        markers1 = wx.lib.plot.PolyMarker(data1, legend='Green Markers', colour='green', marker='circle',size=1)
198
        pi = 3.1415
199
        markers2 = wx.lib.plot.PolyMarker([(0., 0.), (pi/4., 1.), (pi/2, 0.),
200
                          (3.*pi/4., -1)], legend='Cross Legend', colour='blue',
201
                          marker='cross')
590 FredericG 202
        #self.client.Draw(wx.lib.plot.PlotGraphics([markers1,markers2],"Graph Title", "X Axis", "Y Axis"))
586 FredericG 203
 
204
        # Configure TestListCtrl
205
        self.TestListCtrl.InsertColumn(0, "Name")
206
        self.TestListCtrl.InsertColumn(1, "Result")
207
 
585 FredericG 208
    def DrawPointLabel(self, dc, mDataDict):
209
        """This is the fuction that defines how the pointLabels are plotted
210
            dc - DC that will be passed
211
            mDataDict - Dictionary of data that you want to use for the pointLabel
212
 
213
            As an example I have decided I want a box at the curve point
214
            with some text information about the curve plotted below.
215
            Any wxDC method can be used.
216
        """
217
        # ----------
218
        dc.SetPen(wx.Pen(wx.BLACK))
219
        dc.SetBrush(wx.Brush( wx.BLACK, wx.SOLID ) )
220
 
221
        sx, sy = mDataDict["scaledXY"] #scaled x,y of closest point
222
        dc.DrawRectangle( sx-5,sy-5, 10, 10)  #10by10 square centered on point
223
        px,py = mDataDict["pointXY"]
224
        cNum = mDataDict["curveNum"]
225
        pntIn = mDataDict["pIndex"]
226
        legend = mDataDict["legend"]
227
        #make a string to display
228
        s = "Crv# %i, '%s', Pt. (%.2f,%.2f), PtInd %i" %(cNum, legend, px, py, pntIn)
229
        dc.DrawText(s, sx , sy+1)
230
        # -----------
231
 
590 FredericG 232
    def OnTestSelected(self, event):
233
         testId = event.m_itemIndex
234
         print "Test Selected id=%d" % (testId)
592 FredericG 235
         self.activeTestId = testId
236
         self.drawGraph()
590 FredericG 237
 
592 FredericG 238
    def drawGraph(self):
239
         vibTest = self.app.getTest(self.activeTestId)
590 FredericG 240
         nb = vibTest.getDataLen()
592 FredericG 241
         y = int(self.yAxesChoice.GetStringSelection())
590 FredericG 242
 
243
         self.client.Clear()
244
         xydata = _Numeric.arange(nb*2)
245
         xydata.shape = (nb, 2)
246
         xydata[:,1] = vibTest.getRawData()
247
         line = wx.lib.plot.PolyLine(xydata, legend= 'Raw Data', colour='red')
592 FredericG 248
         self.client.Draw(wx.lib.plot.PlotGraphics([line],"Graph Title", "Time", "Acc"), yAxis= (-y,y))
249
         self.client.SetEnableGrid('Horizontal')
590 FredericG 250
 
251
 
587 FredericG 252
    def OnImport(self, event): # wxGlade: MainFrame.<event_handler>
253
        self.app.Import()
254
 
592 FredericG 255
    def onYAxesChange(self, event): # wxGlade: MainFrame.<event_handler>
256
        self.drawGraph()
257
 
584 FredericG 258
# end of class MainFrame
259
 
260
 
586 FredericG 261
class VibTest:
262
    def __init__(self, name, rawData):
263
        self.name = name
590 FredericG 264
        self.rawData = _Numeric.array(rawData)
265
        self.dc = self.rawData.mean()
266
        self.rawData -= self.dc
267
        self.dataLen = len(rawData)
586 FredericG 268
 
590 FredericG 269
    def getName(self):
270
        return self.name
587 FredericG 271
 
590 FredericG 272
    def getRawData(self):
273
        return self.rawData
274
 
275
    def getDataLen(self):
276
        return self.dataLen
277
 
278
 
584 FredericG 279
class App(wx.App):
586 FredericG 280
    def __init__(self, par):
281
        self.VibTests = []
282
 
283
        wx.App.__init__(self, par)
284
 
285
 
286
    def AddTest(self, name, rawData):
287
        test = VibTest(name, rawData)
288
        self.VibTests.append(test)
289
 
290
        index = self.frame_1.TestListCtrl.InsertStringItem(sys.maxint, test.name)
291
        self.frame_1.TestListCtrl.SetStringItem(index, 1, "OK")
292
 
590 FredericG 293
    def getTest(self, testId):
294
        return self.VibTests[testId]
295
 
584 FredericG 296
    def OnInit(self):
297
        wx.InitAllImageHandlers()
586 FredericG 298
        self.frame_1 = MainFrame(None, -1, "")
587 FredericG 299
        self.frame_1.setApp(self);
586 FredericG 300
        self.SetTopWindow(self.frame_1)
301
 
587 FredericG 302
        self.frame_1.Show()
303
        return 1
304
 
305
    def Import(self):
306
        print "Import"
590 FredericG 307
        logfile = open("../Misc/Matlab/Data/unbal200.txt", "r")
308
        data = None
309
        for line in logfile:
310
            values = line.split(',')
311
            if data == None:
312
                nbCols = len(values)
313
                print "%d cols" % nbCols
314
                data = []
315
                for i in range(nbCols):
316
                    data.append([])
317
            for i in range(nbCols):
318
                data[i].append(int(values[i]))
319
        logfile.close()
320
        for i in range(nbCols):
321
            self.AddTest("Imported %d"%i, data[i])
322
 
323
 
324
 
586 FredericG 325
 
590 FredericG 326
 
584 FredericG 327
# end of class App
328
 
329
if __name__ == "__main__":
330
    VibrationTestGui = App(0)
331
    VibrationTestGui.MainLoop()