WebSVN - Projects - Diff - Rev 592 and 593 - /VibrationTest/trunk/VibrationTest/VibrationTestGui.py


#!/usr/bin/env python
# -*- coding: iso-8859-15 -*-
# generated by wxGlade 0.6.3 on Thu Sep 24 15:46:36 2009
 
import sys
import wx
import wx.lib
import wx.lib.plot
 
# Needs Numeric or numarray or NumPy
try:
    import numpy.oldnumeric as _Numeric
except:
    try:
        import numarray as _Numeric  #if numarray is used it is renamed Numeric
    except:
        try:
            import Numeric as _Numeric
        except:
            msg= """
            This module requires the Numeric/numarray or NumPy module,
            which could not be imported.  It probably is not installed
            (it's not part of the standard Python distribution). See the
            Numeric Python site (http://numpy.scipy.org) for information on
            downloading source or binaries."""
            raise ImportError, "Numeric,numarray or NumPy not found. \n" + msg
 
# begin wxGlade: extracode
# end wxGlade
 
 
 
class MenuBar(wx.MenuBar):
    def __init__(self, *args, **kwds):
        # content of this block not found: did you rename this class?
        pass
 
    def __set_properties(self):
        # content of this block not found: did you rename this class?
        pass
 
    def __do_layout(self):
        # content of this block not found: did you rename this class?
        pass
 
# end of class MenuBar
 
 
class wxFrame(wx.Panel):
    def __init__(self, *args, **kwds):
        # content of this block not found: did you rename this class?
        pass
 
    def __set_properties(self):
        # content of this block not found: did you rename this class?
        pass
 
    def __do_layout(self):
        # content of this block not found: did you rename this class?
        pass
 
# end of class wxFrame
 
 
class MyFrame(wx.Frame):
    def __init__(self, *args, **kwds):
        # content of this block not found: did you rename this class?
        pass
 
    def __set_properties(self):
        # content of this block not found: did you rename this class?
        pass
 
    def __do_layout(self):
        # content of this block not found: did you rename this class?
        pass
 
# end of class MyFrame
 
 
class MainFrame(wx.Frame):
    def __init__(self, *args, **kwds):
        # begin wxGlade: MainFrame.__init__
        kwds["style"] = wx.DEFAULT_FRAME_STYLE
        wx.Frame.__init__(self, *args, **kwds)
       
        # Menu Bar
        self.frame_1_menubar = wx.MenuBar()
        wxglade_tmp_menu = wx.Menu()
        wxglade_tmp_menu.Append(wx.NewId(), "Exit", "", wx.ITEM_NORMAL)
        self.frame_1_menubar.Append(wxglade_tmp_menu, "File")
        wxglade_tmp_menu = wx.Menu()
        wxglade_tmp_menu.Append(wx.NewId(), "Clear", "", wx.ITEM_NORMAL)
        wxglade_tmp_menu.Append(wx.NewId(), "Import", "", wx.ITEM_NORMAL)
        self.frame_1_menubar.Append(wxglade_tmp_menu, "TestSet")
        wxglade_tmp_menu = wx.Menu()
        self.frame_1_menubar.Append(wxglade_tmp_menu, "MK")
        wxglade_tmp_menu = wx.Menu()
        self.frame_1_menubar.Append(wxglade_tmp_menu, "Help")
        self.SetMenuBar(self.frame_1_menubar)
        # Menu Bar end
        self.Port = wx.StaticText(self, -1, "Port")
        self.text_ctrl_1 = wx.TextCtrl(self, -1, "")
        self.label_5 = wx.StaticText(self, -1, "label_5")
        self.text_ctrl_5 = wx.TextCtrl(self, -1, "")
        self.label_2 = wx.StaticText(self, -1, "Motor(s)")
        self.text_ctrl_2 = wx.TextCtrl(self, -1, "")
        self.label_6 = wx.StaticText(self, -1, "label_6")
        self.text_ctrl_6 = wx.TextCtrl(self, -1, "")
        self.label_3 = wx.StaticText(self, -1, "Channel(s)")
        self.text_ctrl_3 = wx.TextCtrl(self, -1, "")
        self.label_7 = wx.StaticText(self, -1, "label_7")
        self.text_ctrl_7 = wx.TextCtrl(self, -1, "")
        self.label_4 = wx.StaticText(self, -1, "Speed(s)")
        self.text_ctrl_4 = wx.TextCtrl(self, -1, "")
        self.label_8 = wx.StaticText(self, -1, "label_8")
        self.text_ctrl_8 = wx.TextCtrl(self, -1, "")
        self.button_1 = wx.Button(self, -1, "button_1")
        self.GraphPanel = wx.Panel(self, -1, style=wx.DOUBLE_BORDER|wx.TAB_TRAVERSAL)
        self.yAxesChoice = wx.ComboBox(self, -1, choices=["10", "25", "50", "100", "250"], style=wx.CB_DROPDOWN)
        self.graphTypeChoise = wx.ComboBox(self, -1, choices=["Raw Data", "Filtered Data", "Spectrum"], style=wx.CB_DROPDOWN)
        self.TestListCtrl = wx.ListCtrl(self, -1, style=wx.LC_REPORT|wx.SUNKEN_BORDER)
 
        self.__set_properties()
        self.__do_layout()
 
        self.Bind(wx.EVT_MENU, self.OnImport, id=-1)
        self.Bind(wx.EVT_COMBOBOX, self.onYAxesChange, self.yAxesChoice)
        # end wxGlade
 
    def setApp(self, app):
        self.app = app
 
    def __set_properties(self):
        # begin wxGlade: MainFrame.__set_properties
        self.SetTitle("VibrationTest")
        self.GraphPanel.SetMinSize((800,300))
        self.yAxesChoice.SetSelection(2)
        self.graphTypeChoise.SetSelection(0)
        # end wxGlade
 
    def __do_layout(self):
        # begin wxGlade: MainFrame.__do_layout
        sizer_1 = wx.BoxSizer(wx.VERTICAL)
        grid_sizer_2 = wx.GridSizer(2, 1, 4, 4)
        sizer_2 = wx.BoxSizer(wx.HORIZONTAL)
        grid_sizer_1 = wx.GridSizer(5, 4, 4, 4)
        sizer_1.Add((20, 20), 0, 0, 0)
        grid_sizer_1.Add(self.Port, 0, wx.ALIGN_RIGHT, 0)
        grid_sizer_1.Add(self.text_ctrl_1, 0, 0, 0)
        grid_sizer_1.Add(self.label_5, 0, wx.ALIGN_RIGHT, 0)
        grid_sizer_1.Add(self.text_ctrl_5, 0, 0, 0)
        grid_sizer_1.Add(self.label_2, 0, wx.ALIGN_RIGHT, 0)
        grid_sizer_1.Add(self.text_ctrl_2, 0, 0, 0)
        grid_sizer_1.Add(self.label_6, 0, wx.ALIGN_RIGHT, 0)
        grid_sizer_1.Add(self.text_ctrl_6, 0, 0, 0)
        grid_sizer_1.Add(self.label_3, 0, wx.ALIGN_RIGHT, 0)
        grid_sizer_1.Add(self.text_ctrl_3, 0, 0, 0)
        grid_sizer_1.Add(self.label_7, 0, wx.ALIGN_RIGHT, 0)
        grid_sizer_1.Add(self.text_ctrl_7, 0, 0, 0)
        grid_sizer_1.Add(self.label_4, 0, wx.ALIGN_RIGHT, 0)
        grid_sizer_1.Add(self.text_ctrl_4, 0, 0, 0)
        grid_sizer_1.Add(self.label_8, 0, wx.ALIGN_RIGHT, 0)
        grid_sizer_1.Add(self.text_ctrl_8, 0, 0, 0)
        grid_sizer_1.Add(self.button_1, 0, 0, 0)
        sizer_1.Add(grid_sizer_1, 0, 0, 2)
        sizer_1.Add((20, 20), 0, 0, 0)
        sizer_1.Add(self.GraphPanel, 1, 0, 0)
        sizer_2.Add(self.yAxesChoice, 0, 0, 0)
        sizer_2.Add(self.graphTypeChoise, 0, 0, 0)
        sizer_1.Add(sizer_2, 0, wx.EXPAND, 0)
        sizer_1.Add((20, 20), 0, 0, 0)
        grid_sizer_2.Add(self.TestListCtrl, 1, wx.ALL|wx.EXPAND, 0)
        sizer_1.Add(grid_sizer_2, 1, wx.EXPAND, 0)
        self.SetSizer(sizer_1)
        sizer_1.Fit(self)
        self.Layout()
        # end wxGlade
 
        # List events
        self.TestListCtrl.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnTestSelected, self.TestListCtrl)
 
        # Configure Graph
        self.client = wx.lib.plot.PlotCanvas(self.GraphPanel, size=(800,300))
       
        self.client.SetPointLabelFunc(self.DrawPointLabel)
       
        self.client.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.NORMAL))
        self.client.SetFontSizeAxis(10)
        self.client.SetFontSizeLegend(7)
        self.client.setLogScale((False,False))
 
 
 
 
 
 
 
 
 
 
 
 
 
        # Configure TestListCtrl
        self.TestListCtrl.InsertColumn(0, "Name")
        self.TestListCtrl.InsertColumn(1, "Result")
 
    def DrawPointLabel(self, dc, mDataDict):
        """This is the fuction that defines how the pointLabels are plotted
            dc - DC that will be passed
            mDataDict - Dictionary of data that you want to use for the pointLabel
 
            As an example I have decided I want a box at the curve point
            with some text information about the curve plotted below.
            Any wxDC method can be used.
        """
        # ----------
        dc.SetPen(wx.Pen(wx.BLACK))
        dc.SetBrush(wx.Brush( wx.BLACK, wx.SOLID ) )
       
        sx, sy = mDataDict["scaledXY"] #scaled x,y of closest point
        dc.DrawRectangle( sx-5,sy-5, 10, 10)  #10by10 square centered on point
        px,py = mDataDict["pointXY"]
        cNum = mDataDict["curveNum"]
        pntIn = mDataDict["pIndex"]
        legend = mDataDict["legend"]
        #make a string to display
        s = "Crv# %i, '%s', Pt. (%.2f,%.2f), PtInd %i" %(cNum, legend, px, py, pntIn)
        dc.DrawText(s, sx , sy+1)
        # -----------
 
    def OnTestSelected(self, event):
         testId = event.m_itemIndex
         print "Test Selected id=%d" % (testId)
         self.activeTestId = testId
         self.drawGraph()
 
    def drawGraph(self):
         vibTest = self.app.getTest(self.activeTestId)
         nb = vibTest.getDataLen()
         y = int(self.yAxesChoice.GetStringSelection())
 
         #self.client.Clear()
         xydata = _Numeric.linspace(0,0.09*nb,2*nb)
         xydata.shape = (nb, 2)
         xydata[:,1] = vibTest.getRawData()
         line = wx.lib.plot.PolyLine(xydata, legend= 'Raw Data', colour='red')
         self.client.Draw(wx.lib.plot.PlotGraphics([line],"Graph Title", "Time (ms)", "Acc"), yAxis= (-y,y))
         self.client.SetEnableGrid('Horizontal')
 
 
    def OnImport(self, event): # wxGlade: MainFrame.<event_handler>
        self.app.Import()
 
    def onYAxesChange(self, event): # wxGlade: MainFrame.<event_handler>
        self.drawGraph()
 
# end of class MainFrame
 
 
class VibTest:
    def __init__(self, name, rawData):
        self.name = name
        self.rawData = _Numeric.array(rawData)
        self.dc = self.rawData.mean()
        self.rawData -= self.dc
        self.dataLen = len(rawData)
 
    def getName(self):
        return self.name
 
    def getRawData(self):
        return self.rawData
 
    def getDataLen(self):
        return self.dataLen
 
 
class App(wx.App):
    def __init__(self, par):
        self.VibTests = []
 
        wx.App.__init__(self, par)
 
 
    def AddTest(self, name, rawData):
        test = VibTest(name, rawData)
        self.VibTests.append(test)
 
        index = self.frame_1.TestListCtrl.InsertStringItem(sys.maxint, test.name)
        self.frame_1.TestListCtrl.SetStringItem(index, 1, "OK")
 
    def getTest(self, testId):
        return self.VibTests[testId]
 
    def OnInit(self):
        wx.InitAllImageHandlers()
        self.frame_1 = MainFrame(None, -1, "")
        self.frame_1.setApp(self);
        self.SetTopWindow(self.frame_1)
 
        self.frame_1.Show()
        return 1
 
    def Import(self):
        print "Import"
        logfile = open("../Misc/Matlab/Data/unbal200.txt", "r")
        data = None
        for line in logfile:
            values = line.split(',')
            if data == None:
                nbCols = len(values)
                print "%d cols" % nbCols
                data = []
                for i in range(nbCols):
                    data.append([])
            for i in range(nbCols):
                data[i].append(int(values[i]))
        logfile.close()
        for i in range(nbCols):
            self.AddTest("Imported %d"%i, data[i])
       
       
           
 
       
# end of class App
 
if __name__ == "__main__":
    VibrationTestGui = App(0)
    VibrationTestGui.MainLoop()
 

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

Subversion Repositories Projects

(root)/VibrationTest/trunk/VibrationTest/VibrationTestGui.py @ 700 - Rev 592 → 593