WebSVN - Projects - Blame - Rev 627 - /VibrationTest/branches/v1.2/VibrationTest/VibrationTestGui.py

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
606	FredericG	6	import os
612	FredericG	7	import time
		8	import thread
		9	import ConfigParser
		10
584	FredericG	11	import wx
585	FredericG	12	import wx.lib
		13	import wx.lib.plot
612	FredericG	14	import wx.lib.newevent
623	FredericG	15	import wx.lib.agw.speedmeter as speedmeter
584	FredericG	16
613	FredericG	17	import mkProto
		18
		19
		20
		21	CHANNEL_NAMES = ["GyroYaw", "GyroRoll", "GyroNick", "Pressure", "Batt", "AccTop", "AccRoll", "AccNick"]
615	FredericG	22	FS = 11111
623	FredericG	23	pi = 3.14
613	FredericG	24
626	FredericG	25	rootPath = ""
		26
585	FredericG	27	# Needs Numeric or numarray or NumPy
		28	try:
		29	import numpy.oldnumeric as _Numeric
		30	except:
		31	try:
		32	import numarray as _Numeric #if numarray is used it is renamed Numeric
		33	except:
		34	try:
		35	import Numeric as _Numeric
		36	except:
		37	msg= """
		38	This module requires the Numeric/numarray or NumPy module,
		39	which could not be imported. It probably is not installed
		40	(it's not part of the standard Python distribution). See the
		41	Numeric Python site (http://numpy.scipy.org) for information on
		42	downloading source or binaries."""
		43	raise ImportError, "Numeric,numarray or NumPy not found. \n" + msg
		44
584	FredericG	45	# begin wxGlade: extracode
		46	# end wxGlade
		47
		48
612	FredericG	49	# This creates a new Event class and a EVT binder function
613	FredericG	50	(MeasStatusUpdateEvent, EVT_MEAS_STATUS_UPDATE) = wx.lib.newevent.NewEvent()
		51	(MeasDataEvent, EVT_MEAS_DATA) = wx.lib.newevent.NewEvent()
584	FredericG	52
612	FredericG	53	class MeasureDialog(wx.Dialog):
		54	def __init__(self, args, *kwds):
		55	# begin wxGlade: MeasureDialog.__init__
626	FredericG	56	kwds["style"] = wx.CAPTION\|wx.RESIZE_BORDER\|wx.THICK_FRAME
612	FredericG	57	wx.Dialog.__init__(self, args, *kwds)
		58	self.text_ctrl_1 = wx.TextCtrl(self, -1, "", style=wx.TE_MULTILINE\|wx.TE_READONLY\|wx.HSCROLL)
		59	self.button = wx.Button(self, -1, "STOP")
623	FredericG	60	self.voltageCtrl = speedmeter.SpeedMeter(self, extrastyle=speedmeter.SM_DRAW_HAND \| speedmeter.SM_DRAW_PARTIAL_SECTORS \| speedmeter.SM_DRAW_MIDDLE_ICON )
		61	self.speedCtrl = speedmeter.SpeedMeter(self, extrastyle=speedmeter.SM_DRAW_HAND \| speedmeter.SM_DRAW_PARTIAL_SECTORS \| speedmeter.SM_DRAW_MIDDLE_TEXT \| speedmeter.SM_DRAW_SECONDARY_TICKS)
626	FredericG	62
612	FredericG	63	self.__set_properties()
		64	self.__do_layout()
		65
		66	self.Bind(wx.EVT_BUTTON, self.onButton, self.button)
		67	# end wxGlade
		68
626	FredericG	69	self.button.SetFocus()
		70
623	FredericG	71	YELLOW = wx.Colour(255, 240, 0)
		72	BG = wx.Colour(0x80, 0x80, 0x80)
		73	# Configure Voltage Ctrl
		74	self.voltageCtrl.SetAngleRange(0,pi)
		75	intervals = range(0, 5)
		76	self.voltageCtrl.SetIntervals(intervals)
		77	colours = [wx.RED, wx.GREEN, wx.GREEN, YELLOW]
		78	self.voltageCtrl.SetIntervalColours(colours)
		79	ticks = ["", "", "", "", ""]
		80	self.voltageCtrl.SetTicks(ticks)
		81	self.voltageCtrl.SetTicksColour(wx.WHITE)
		82	self.voltageCtrl.SetHandColour(YELLOW)
		83
626	FredericG	84	icon = wx.Icon("%s/Resources/fuel.ico" % rootPath, wx.BITMAP_TYPE_ICO)
623	FredericG	85	icon.SetWidth(24)
		86	icon.SetHeight(24)
		87
		88	self.voltageCtrl.SetMiddleIcon(icon)
		89	self.voltageCtrl.SetSpeedBackground(BG)
		90	self.voltageCtrl.SetArcColour(wx.WHITE)
		91	self.voltageCtrl.SetSpeedValue(2)
		92
		93
		94	# Configure Speed Ctr;
		95	self.speedCtrl.SetAngleRange(0,pi)
		96	intervals = range(0, 261, 20)
		97	self.speedCtrl.SetIntervals(intervals)
		98
		99	colours = [BG]*(len(intervals)-1)
		100	for i in range(5,10):
		101	colours[i] = wx.GREEN
		102	self.speedCtrl.SetIntervalColours(colours)
		103	ticks = [str(interval) for interval in intervals]
		104	self.speedCtrl.SetTicks(ticks)
		105	self.speedCtrl.SetTicksColour(wx.WHITE)
		106	self.speedCtrl.SetNumberOfSecondaryTicks(1)
		107	self.speedCtrl.SetTicksFont(wx.Font(7, wx.SWISS, wx.NORMAL, wx.NORMAL))
		108	self.speedCtrl.SetMiddleText("Speed")
		109	self.speedCtrl.SetMiddleTextColour(wx.WHITE)
		110	self.speedCtrl.SetMiddleTextFont(wx.Font(8, wx.SWISS, wx.NORMAL, wx.BOLD))
		111	self.speedCtrl.SetHandColour(YELLOW)
		112	self.speedCtrl.SetSpeedBackground(BG)
		113	self.speedCtrl.SetArcColour(wx.WHITE)
		114	self.speedCtrl.SetSpeedValue(0)
		115
		116
612	FredericG	117	self.running = True
		118	self.Bind(EVT_MEAS_STATUS_UPDATE, self.OnUpdate)
613	FredericG	119	self.Bind(EVT_MEAS_DATA, self.OnData)
		120	# The first argument that is passed to the constructor is the parent
		121	self.app = args[0].app
		122	self.error = False
623	FredericG	123	self.firstVoltage = True
612	FredericG	124
623	FredericG	125
612	FredericG	126	def __set_properties(self):
		127	# begin wxGlade: MeasureDialog.__set_properties
		128	self.SetTitle("Measuring Status")
		129	self.text_ctrl_1.SetMinSize((400,300))
623	FredericG	130	self.voltageCtrl.SetMinSize((50,-1))
		131	self.speedCtrl.SetMinSize((50,-1))
612	FredericG	132	# end wxGlade
		133
		134	def __do_layout(self):
		135	# begin wxGlade: MeasureDialog.__do_layout
		136	sizer_1 = wx.BoxSizer(wx.HORIZONTAL)
623	FredericG	137	sizer_4 = wx.BoxSizer(wx.VERTICAL)
612	FredericG	138	sizer_2 = wx.BoxSizer(wx.VERTICAL)
		139	sizer_1.Add((20, 20), 0, 0, 0)
		140	sizer_2.Add((20, 20), 0, 0, 0)
		141	sizer_2.Add(self.text_ctrl_1, 1, wx.EXPAND, 0)
		142	sizer_2.Add((20, 20), 0, 0, 0)
		143	sizer_2.Add(self.button, 0, wx.ALIGN_CENTER_HORIZONTAL, 0)
		144	sizer_2.Add((20, 20), 0, 0, 0)
		145	sizer_1.Add(sizer_2, 1, wx.EXPAND, 0)
		146	sizer_1.Add((20, 20), 0, 0, 0)
623	FredericG	147	sizer_4.Add(self.voltageCtrl, 1, wx.EXPAND, 0)
		148	sizer_4.Add(self.speedCtrl, 1, wx.EXPAND, 0)
		149	sizer_1.Add(sizer_4, 1, wx.EXPAND, 0)
		150	sizer_1.Add((20, 20), 0, 0, 0)
612	FredericG	151	self.SetSizer(sizer_1)
		152	sizer_1.Fit(self)
		153	self.Layout()
		154	# end wxGlade
613	FredericG	155
		156	def OnData(self, evt):
		157	print "Received Data"
		158	self.app.AddTest2(evt.vibTest)
623	FredericG	159
612	FredericG	160	def OnUpdate(self, evt):
613	FredericG	161	print "Status update"
612	FredericG	162	self.running = evt.running
613	FredericG	163	if evt.error:
		164	self.error = True;
		165	self.text_ctrl_1.WriteText("ERROR: ")
		166	self.text_ctrl_1.SetBackgroundColour("Red")
612	FredericG	167	self.text_ctrl_1.WriteText("%s\n"%evt.msg)
		168	if (not self.running):
613	FredericG	169	if (not self.error):
		170	self.text_ctrl_1.SetBackgroundColour("Green")
		171	self.text_ctrl_1.write(" ") # so that the background is redrawn
623	FredericG	172	self.button.SetLabel("Close")
613	FredericG	173
623	FredericG	174	if evt.speed != None:
		175	self.speedCtrl.SetSpeedValue(evt.speed)
		176
		177	if evt.voltage != None:
		178	vmin,vmax,v = evt.voltage
		179	if self.firstVoltage:
		180	ticks = ["", "%.1f V"%vmin, "", "%.1f V"%vmax, ""]
		181	self.voltageCtrl.SetTicks(ticks)
		182	self.firstVoltage = False
		183	i = (v-vmin)/(vmax-vmin) # 0..1
		184	i *= 2
		185	i = i+1
		186	i = min(max(i,0),5)
		187	self.voltageCtrl.SetSpeedValue(i)
		188
		189
612	FredericG	190
		191	def onButton(self, event): # wxGlade: MeasureDialog.<event_handler>
		192	if (not self.running):
		193	self.Destroy()
613	FredericG	194	else:
		195	self.app.cancelMeasurement()
612	FredericG	196
		197	# end of class MeasureDialog
		198
		199
607	FredericG	200	class SettingsDialog(wx.Dialog):
601	FredericG	201	def __init__(self, args, *kwds):
607	FredericG	202	# begin wxGlade: SettingsDialog.__init__
		203	kwds["style"] = wx.DEFAULT_DIALOG_STYLE
		204	wx.Dialog.__init__(self, args, *kwds)
608	FredericG	205	self.button_5 = wx.Button(self, wx.ID_CANCEL, "")
		206	self.button_6 = wx.Button(self, wx.ID_OK, "")
601	FredericG	207
		208	self.__set_properties()
		209	self.__do_layout()
611	FredericG	210
		211	self.Bind(wx.EVT_BUTTON, self.onOK, self.button_6)
601	FredericG	212	# end wxGlade
		213
608	FredericG	214	# The first argument that is passed to the constructor is the parent
611	FredericG	215	self.settings = args[0].app.settings
608	FredericG	216	# Add text-boxes for all settings
611	FredericG	217	self.tb = []
		218	self.grid_sizer_2.SetRows(len(self.settings))
		219	for setting in self.settings.iteritems():
608	FredericG	220	lb = wx.StaticText(self, -1, setting[1].descr, style=wx.ALIGN_RIGHT)
		221	tb = wx.TextCtrl(self, -1, str(setting[1].value))
		222	self.grid_sizer_2.Add(lb, 0, wx.ALIGN_RIGHT\|wx.ALIGN_CENTER_VERTICAL, 0)
		223	self.grid_sizer_2.Add(tb, 0, 0, 0)
611	FredericG	224	self.tb.append(tb)
608	FredericG	225	self.sizer_5.Fit(self)
		226	self.Layout()
		227
601	FredericG	228	def __set_properties(self):
607	FredericG	229	# begin wxGlade: SettingsDialog.__set_properties
601	FredericG	230	self.SetTitle("Settings")
		231	# end wxGlade
		232
		233	def __do_layout(self):
607	FredericG	234	# begin wxGlade: SettingsDialog.__do_layout
608	FredericG	235	sizer_5 = wx.BoxSizer(wx.VERTICAL)
		236	grid_sizer_3 = wx.GridSizer(1, 2, 0, 0)
		237	sizer_6 = wx.BoxSizer(wx.HORIZONTAL)
		238	grid_sizer_2 = wx.GridSizer(1, 2, 4, 4)
		239	sizer_5.Add((20, 20), 0, 0, 0)
		240	sizer_6.Add((20, 20), 0, 0, 0)
		241	sizer_6.Add(grid_sizer_2, 0, 0, 0)
		242	sizer_6.Add((20, 20), 0, 0, 0)
		243	sizer_5.Add(sizer_6, 1, wx.EXPAND, 0)
		244	sizer_5.Add((20, 20), 0, 0, 0)
		245	grid_sizer_3.Add(self.button_5, 0, wx.ALIGN_CENTER_HORIZONTAL\|wx.ALIGN_CENTER_VERTICAL, 0)
		246	grid_sizer_3.Add(self.button_6, 0, wx.ALIGN_CENTER_HORIZONTAL\|wx.ALIGN_CENTER_VERTICAL, 0)
		247	sizer_5.Add(grid_sizer_3, 0, wx.EXPAND, 0)
		248	sizer_5.Add((20, 20), 0, 0, 0)
		249	self.SetSizer(sizer_5)
		250	sizer_5.Fit(self)
601	FredericG	251	self.Layout()
		252	# end wxGlade
		253
608	FredericG	254	# Store some of the items, we will need them later
		255	self.grid_sizer_2 = grid_sizer_2
		256	self.sizer_5 = sizer_5
		257
		258
611	FredericG	259	def onOK(self, event): # wxGlade: SettingsDialog.<event_handler>
		260	print "Updating parameters"
		261	try:
		262	i=0
		263	for setting in self.settings.iteritems():
		264	print setting[0], self.tb[i].GetValue()
		265	setting[1].set(self.tb[i].GetValue())
		266	i += 1
		267	event.Skip()
		268	except:
		269	wx.MessageBox("Invalid format for \"%s\" setting." % setting[1].descr)
		270
607	FredericG	271	# end of class SettingsDialog
		272
		273
		274
584	FredericG	275	class MainFrame(wx.Frame):
		276	def __init__(self, args, *kwds):
		277	# begin wxGlade: MainFrame.__init__
		278	kwds["style"] = wx.DEFAULT_FRAME_STYLE
		279	wx.Frame.__init__(self, args, *kwds)
587	FredericG	280
		281	# Menu Bar
		282	self.frame_1_menubar = wx.MenuBar()
		283	wxglade_tmp_menu = wx.Menu()
607	FredericG	284	wxglade_tmp_menu.Append(101, "Settings", "", wx.ITEM_NORMAL)
616	FredericG	285	wxglade_tmp_menu.AppendSeparator()
607	FredericG	286	wxglade_tmp_menu.Append(150, "Exit", "", wx.ITEM_NORMAL)
587	FredericG	287	self.frame_1_menubar.Append(wxglade_tmp_menu, "File")
		288	wxglade_tmp_menu = wx.Menu()
616	FredericG	289	wxglade_tmp_menu.Append(301, "Clear All", "", wx.ITEM_NORMAL)
		290	wxglade_tmp_menu.Append(302, "Clear Selected", "", wx.ITEM_NORMAL)
		291	wxglade_tmp_menu.AppendSeparator()
627	FredericG	292	wxglade_tmp_menu.Append(310, "Select All", "", wx.ITEM_NORMAL)
		293	wxglade_tmp_menu.AppendSeparator()
616	FredericG	294	wxglade_tmp_menu.Append(303, "Load", "", wx.ITEM_NORMAL)
		295	wxglade_tmp_menu.Append(304, "Save", "", wx.ITEM_NORMAL)
587	FredericG	296	self.frame_1_menubar.Append(wxglade_tmp_menu, "TestSet")
		297	wxglade_tmp_menu = wx.Menu()
626	FredericG	298	wxglade_tmp_menu.Append(401, "Flash VibTest FC software", "", wx.ITEM_NORMAL)
		299	wxglade_tmp_menu.Append(402, "Restore original FC software", "", wx.ITEM_NORMAL)
587	FredericG	300	self.frame_1_menubar.Append(wxglade_tmp_menu, "MK")
		301	wxglade_tmp_menu = wx.Menu()
626	FredericG	302	wxglade_tmp_menu.Append(1099, "About", "", wx.ITEM_NORMAL)
587	FredericG	303	self.frame_1_menubar.Append(wxglade_tmp_menu, "Help")
		304	self.SetMenuBar(self.frame_1_menubar)
		305	# Menu Bar end
600	FredericG	306	self.Description = wx.StaticText(self, -1, "Description")
613	FredericG	307	self.tcDescr = wx.TextCtrl(self, -1, "Test")
600	FredericG	308	self.label_37 = wx.StaticText(self, -1, "Speed(s)")
613	FredericG	309	self.tcSpeeds = wx.TextCtrl(self, -1, "100-200:10")
600	FredericG	310	self.label_35 = wx.StaticText(self, -1, "Motor(s)")
613	FredericG	311	self.tcMotors = wx.TextCtrl(self, -1, "1")
600	FredericG	312	self.label_38 = wx.StaticText(self, -1, "")
626	FredericG	313	self.panel_2 = wx.Panel(self, -1)
613	FredericG	314	self.label_36 = wx.StaticText(self, -1, "Channel(s)")
		315	self.tcChannels = wx.TextCtrl(self, -1, "6")
600	FredericG	316	self.label_39 = wx.StaticText(self, -1, "")
626	FredericG	317	self.panel_3 = wx.Panel(self, -1)
		318	self.button_4 = wx.Button(self, -1, "Measure")
		319	self.graphCtrl = wx.lib.plot.PlotCanvas(self, size=(800,300))
600	FredericG	320	self.label_40 = wx.StaticText(self, -1, "Graph Type ")
		321	self.graphTypeChoice = wx.Choice(self, -1, choices=["Raw Signal", "Filtered Signal", "Spectrum"])
		322	self.label_41 = wx.StaticText(self, -1, "Y Axis Range ")
		323	self.yAxisChoice = wx.Choice(self, -1, choices=["25", "50", "100", "200"])
627	FredericG	324	self.copyGraphButton = wx.Button(self, -1, "Copy Graph Data")
586	FredericG	325	self.TestListCtrl = wx.ListCtrl(self, -1, style=wx.LC_REPORT\|wx.SUNKEN_BORDER)
584	FredericG	326
		327	self.__set_properties()
		328	self.__do_layout()
587	FredericG	329
607	FredericG	330	self.Bind(wx.EVT_MENU, self.OnSettings, id=101)
626	FredericG	331	self.Bind(wx.EVT_MENU, self.onExit, id=150)
607	FredericG	332	self.Bind(wx.EVT_MENU, self.onClear, id=301)
616	FredericG	333	self.Bind(wx.EVT_MENU, self.onClearSelected, id=302)
627	FredericG	334	self.Bind(wx.EVT_MENU, self.onSelectAll, id=310)
616	FredericG	335	self.Bind(wx.EVT_MENU, self.OnImport, id=303)
		336	self.Bind(wx.EVT_MENU, self.onExport, id=304)
626	FredericG	337	self.Bind(wx.EVT_MENU, self.onAbout, id=1099)
612	FredericG	338	self.Bind(wx.EVT_BUTTON, self.onStartMeasure, self.button_4)
600	FredericG	339	self.Bind(wx.EVT_CHOICE, self.onGraphTypeChange, self.graphTypeChoice)
		340	self.Bind(wx.EVT_CHOICE, self.onYAxisChange, self.yAxisChoice)
627	FredericG	341	self.Bind(wx.EVT_BUTTON, self.onCopyGraphData, self.copyGraphButton)
584	FredericG	342	# end wxGlade
626	FredericG	343	favicon = wx.Icon('%s/Resources/60px-Procman.ico' % rootPath, wx.BITMAP_TYPE_ICO, 32, 32)
		344	wx.Frame.SetIcon(self, favicon)
584	FredericG	345
587	FredericG	346	def setApp(self, app):
		347	self.app = app
585	FredericG	348
584	FredericG	349	def __set_properties(self):
		350	# begin wxGlade: MainFrame.__set_properties
587	FredericG	351	self.SetTitle("VibrationTest")
600	FredericG	352	self.SetSize((850, 700))
		353	self.Description.SetMinSize((53, 13))
601	FredericG	354	self.button_4.SetMinSize((80, 80))
626	FredericG	355	self.graphCtrl.SetMinSize((800,300))
600	FredericG	356	self.graphTypeChoice.SetSelection(0)
		357	self.yAxisChoice.SetSelection(1)
		358	self.TestListCtrl.SetMinSize((800,300))
584	FredericG	359	# end wxGlade
		360
		361	def __do_layout(self):
		362	# begin wxGlade: MainFrame.__do_layout
600	FredericG	363	sizer_3 = wx.BoxSizer(wx.HORIZONTAL)
		364	sizer_8 = wx.BoxSizer(wx.VERTICAL)
		365	sizer_11 = wx.BoxSizer(wx.VERTICAL)
		366	sizer_12 = wx.BoxSizer(wx.HORIZONTAL)
		367	sizer_9 = wx.BoxSizer(wx.HORIZONTAL)
		368	sizer_10 = wx.BoxSizer(wx.HORIZONTAL)
		369	grid_sizer_1 = wx.GridSizer(3, 4, 4, 5)
		370	sizer_3.Add((20, 20), 0, 0, 0)
		371	sizer_8.Add((20, 20), 0, 0, 0)
		372	grid_sizer_1.Add(self.Description, 0, wx.ALIGN_RIGHT\|wx.ALIGN_CENTER_VERTICAL, 0)
613	FredericG	373	grid_sizer_1.Add(self.tcDescr, 0, 0, 0)
600	FredericG	374	grid_sizer_1.Add(self.label_37, 0, wx.ALIGN_RIGHT\|wx.ALIGN_CENTER_VERTICAL, 0)
613	FredericG	375	grid_sizer_1.Add(self.tcSpeeds, 0, 0, 0)
600	FredericG	376	grid_sizer_1.Add(self.label_35, 0, wx.ALIGN_RIGHT\|wx.ALIGN_CENTER_VERTICAL, 0)
613	FredericG	377	grid_sizer_1.Add(self.tcMotors, 0, 0, 0)
600	FredericG	378	grid_sizer_1.Add(self.label_38, 0, wx.ALIGN_RIGHT\|wx.ALIGN_CENTER_VERTICAL, 0)
626	FredericG	379	grid_sizer_1.Add(self.panel_2, 1, wx.EXPAND, 0)
600	FredericG	380	grid_sizer_1.Add(self.label_36, 0, wx.ALIGN_RIGHT\|wx.ALIGN_CENTER_VERTICAL, 0)
613	FredericG	381	grid_sizer_1.Add(self.tcChannels, 0, 0, 0)
600	FredericG	382	grid_sizer_1.Add(self.label_39, 0, wx.ALIGN_RIGHT\|wx.ALIGN_CENTER_VERTICAL, 0)
626	FredericG	383	grid_sizer_1.Add(self.panel_3, 1, wx.EXPAND, 0)
600	FredericG	384	sizer_9.Add(grid_sizer_1, 0, 0, 0)
		385	sizer_10.Add((50, 20), 0, 0, 0)
		386	sizer_10.Add(self.button_4, 0, wx.ALIGN_CENTER_VERTICAL, 0)
		387	sizer_9.Add(sizer_10, 1, wx.EXPAND, 0)
		388	sizer_8.Add(sizer_9, 0, 0, 0)
		389	sizer_8.Add((20, 30), 0, 0, 0)
626	FredericG	390	sizer_11.Add(self.graphCtrl, 1, wx.EXPAND, 0)
600	FredericG	391	sizer_11.Add((20, 5), 0, 0, 0)
		392	sizer_12.Add(self.label_40, 0, wx.ALIGN_CENTER_VERTICAL, 0)
		393	sizer_12.Add(self.graphTypeChoice, 0, 0, 0)
		394	sizer_12.Add((40, 20), 0, 0, 0)
		395	sizer_12.Add(self.label_41, 0, wx.ALIGN_CENTER_VERTICAL, 0)
		396	sizer_12.Add(self.yAxisChoice, 0, 0, 0)
627	FredericG	397	sizer_12.Add((80, 20), 0, 0, 0)
		398	sizer_12.Add(self.copyGraphButton, 0, 0, 0)
600	FredericG	399	sizer_11.Add(sizer_12, 0, 0, 0)
		400	sizer_8.Add(sizer_11, 0, 0, 0)
		401	sizer_8.Add((20, 30), 0, 0, 0)
		402	sizer_8.Add(self.TestListCtrl, 1, 0, 0)
		403	sizer_8.Add((20, 20), 0, 0, 0)
		404	sizer_3.Add(sizer_8, 1, wx.EXPAND, 0)
		405	self.SetSizer(sizer_3)
584	FredericG	406	self.Layout()
600	FredericG	407	self.SetSize((850, 700))
584	FredericG	408	# end wxGlade
		409
590	FredericG	410	# List events
		411	self.TestListCtrl.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnTestSelected, self.TestListCtrl)
		412
586	FredericG	413	# Configure Graph
626	FredericG	414	#self.graphCtrl = wx.lib.plot.PlotCanvas(self.GraphPanel, size=(800,300))
593	FredericG	415
626	FredericG	416	self.graphCtrl.SetPointLabelFunc(self.DrawPointLabel)
585	FredericG	417
626	FredericG	418	self.graphCtrl.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.NORMAL))
		419	self.graphCtrl.SetFontSizeAxis(10)
		420	self.graphCtrl.SetFontSizeLegend(7)
		421	self.graphCtrl.setLogScale((False,False))
585	FredericG	422
586	FredericG	423
		424	# Configure TestListCtrl
594	FredericG	425	self.TestListCtrl.InsertColumn(0, "Description")
619	FredericG	426	self.TestListCtrl.InsertColumn(1, "Voltage")
		427	self.TestListCtrl.InsertColumn(2, "Speed")
		428	self.TestListCtrl.InsertColumn(3, "Channel")
		429	self.TestListCtrl.InsertColumn(4, "Vibration Value")
		430	self.TestListCtrl.SetColumnWidth(4, 500)
586	FredericG	431
585	FredericG	432	def DrawPointLabel(self, dc, mDataDict):
		433	"""This is the fuction that defines how the pointLabels are plotted
		434	dc - DC that will be passed
		435	mDataDict - Dictionary of data that you want to use for the pointLabel
		436
		437	As an example I have decided I want a box at the curve point
		438	with some text information about the curve plotted below.
		439	Any wxDC method can be used.
		440	"""
		441	# ----------
		442	dc.SetPen(wx.Pen(wx.BLACK))
		443	dc.SetBrush(wx.Brush( wx.BLACK, wx.SOLID ) )
		444
		445	sx, sy = mDataDict["scaledXY"] #scaled x,y of closest point
		446	dc.DrawRectangle( sx-5,sy-5, 10, 10) #10by10 square centered on point
		447	px,py = mDataDict["pointXY"]
		448	cNum = mDataDict["curveNum"]
		449	pntIn = mDataDict["pIndex"]
		450	legend = mDataDict["legend"]
		451	#make a string to display
		452	s = "Crv# %i, '%s', Pt. (%.2f,%.2f), PtInd %i" %(cNum, legend, px, py, pntIn)
		453	dc.DrawText(s, sx , sy+1)
		454	# -----------
		455
596	FredericG	456
		457	def onNewTest(self, test):
		458	index = self.TestListCtrl.InsertStringItem(sys.maxint, test.descr)
619	FredericG	459	self.TestListCtrl.SetStringItem(index, 1, "%.1f V" %test.voltage)
		460	self.TestListCtrl.SetStringItem(index, 2, str(test.speed))
		461	self.TestListCtrl.SetStringItem(index, 3, test.channel)
607	FredericG	462
615	FredericG	463	vv = int(test.getVibValue(self.app.settings["hpf"].value, self.app.settings["lpf"].value))
607	FredericG	464	vvs = "\|%s\| (%d)" % ("----------------------------------------------------------------------------------------------------"[0:min(vv,100)], vv)
619	FredericG	465	self.TestListCtrl.SetStringItem(index, 4, vvs)
613	FredericG	466	self.TestListCtrl.Select(index)
596	FredericG	467
		468
590	FredericG	469	def OnTestSelected(self, event):
		470	testId = event.m_itemIndex
		471	print "Test Selected id=%d" % (testId)
592	FredericG	472	self.activeTestId = testId
		473	self.drawGraph()
590	FredericG	474
592	FredericG	475	def drawGraph(self):
607	FredericG	476
600	FredericG	477	y = int(self.yAxisChoice.GetStringSelection())
590	FredericG	478
607	FredericG	479	nbSelected = self.TestListCtrl.SelectedItemCount
596	FredericG	480
616	FredericG	481	if nbSelected == 0:
626	FredericG	482	self.graphCtrl.Clear()
616	FredericG	483
		484	elif nbSelected > 1:
607	FredericG	485	self.graphTypeChoice.Disable()
627	FredericG	486	self.copyGraphButton.Enable()
607	FredericG	487	x = 1
		488	data = []
		489	idx = self.TestListCtrl.GetFirstSelected()
		490	while idx != -1:
615	FredericG	491	data.append([x,self.app.getTest(idx).getVibValue(self.app.settings["hpf"].value, self.app.settings["lpf"].value)])
607	FredericG	492	x += 1
		493	idx = self.TestListCtrl.GetNextSelected(idx)
		494	line = wx.lib.plot.PolyLine(data, legend= 'Vibrations', colour='red', width=2)
		495	markers = wx.lib.plot.PolyMarker(data, legend= '', colour='red', marker='circle',size=2)
		496	title = "Comparing tests"
626	FredericG	497	self.graphCtrl.Draw(wx.lib.plot.PlotGraphics([line, markers], title, "Test", "Vibration Value"), xAxis=(1,max(x,10)), yAxis=(0,y))
		498	self.graphCtrl.SetEnableGrid('Horizontal')
600	FredericG	499
607	FredericG	500	else:
		501	self.graphTypeChoice.Enable()
627	FredericG	502	self.copyGraphButton.Disable()
607	FredericG	503	vibTest = self.app.getTest(self.activeTestId)
		504	nb = vibTest.getDataLen()
590	FredericG	505
607	FredericG	506	if self.graphTypeChoice.GetSelection() == 0:
		507	xydata = _Numeric.linspace(0,0.09nb,2nb)
		508	xydata.shape = (nb, 2)
		509	xydata[:,1] = vibTest.getRawData()
		510	line = wx.lib.plot.PolyLine(xydata, legend= 'Raw Data', colour='red', width=2)
		511
		512	title = "Raw Signal: %s %s %d" %(vibTest.descr, vibTest.channel, vibTest.speed)
626	FredericG	513	self.graphCtrl.Draw(wx.lib.plot.PlotGraphics([line], title, "Time (ms)", "Acc"), yAxis= (-y,y))
		514	self.graphCtrl.SetEnableGrid('Horizontal')
599	FredericG	515
607	FredericG	516	if self.graphTypeChoice.GetSelection() == 1:
		517	xydata = _Numeric.linspace(0,0.09nb,2nb)
		518	xydata.shape = (nb, 2)
615	FredericG	519	xydata[:,1] = vibTest.getFilteredData(self.app.settings["hpf"].value, self.app.settings["lpf"].value)
607	FredericG	520	line = wx.lib.plot.PolyLine(xydata, legend= 'Raw Data', colour='red', width=2)
		521
		522	title = "Filtered Signal: %s %s %d" %(vibTest.descr, vibTest.channel, vibTest.speed)
626	FredericG	523	self.graphCtrl.Draw(wx.lib.plot.PlotGraphics([line], title, "Time (ms)", "Acc"), yAxis= (-y,y))
		524	self.graphCtrl.SetEnableGrid('Horizontal')
607	FredericG	525
		526	elif self.graphTypeChoice.GetSelection() == 2:
615	FredericG	527	xydata = _Numeric.linspace(0,FS/2,nb)
607	FredericG	528	xydata.shape = (nb/2, 2)
		529
		530	xydata[:,1] = vibTest.getSpectrum()
		531
		532	line = wx.lib.plot.PolyLine(xydata, legend= 'Spectrum', colour='red')
		533	markers = wx.lib.plot.PolyMarker(xydata, legend= '', colour='red', marker='circle',size=2)
615	FredericG	534
		535	fc = self.app.settings["hpf"].value
		536	filterLine1 = wx.lib.plot.PolyLine(((fc,0),(fc,y)), legend='HP Filter', colour='Black', width=4)
		537	fc = self.app.settings["lpf"].value
		538	filterLine2 = wx.lib.plot.PolyLine(((fc,0),(fc,y)), legend='HP Filter', colour='Black', width=4)
607	FredericG	539
		540	title = "Spectrum: %s %s %d" %(vibTest.descr, vibTest.channel, vibTest.speed)
626	FredericG	541	self.graphCtrl.Draw(wx.lib.plot.PlotGraphics([line,markers, filterLine1, filterLine2], title, "Freq (Hz)", "Acc"), xAxis=(0,200), yAxis= (-0,y))
		542	self.graphCtrl.SetEnableGrid(True)
590	FredericG	543
599	FredericG	544
587	FredericG	545	def OnImport(self, event): # wxGlade: MainFrame.<event_handler>
606	FredericG	546	dlg = wx.FileDialog(
		547	self, message="Choose a file",
619	FredericG	548	defaultDir="%s/Data/" % os.getcwd(),
606	FredericG	549	defaultFile="*.txt",
619	FredericG	550	wildcard="*.txt",
606	FredericG	551	style=wx.OPEN \| wx.CHANGE_DIR
		552	)
		553	if dlg.ShowModal() == wx.ID_OK:
		554	paths = dlg.GetPaths();
619	FredericG	555	self.app.loadTests(paths[0])
606	FredericG	556	dlg.Destroy()
587	FredericG	557
619	FredericG	558	def onExport(self, event): # wxGlade: MainFrame.<event_handler>
		559	dlg = wx.FileDialog(
		560	self, message="Save file as ...",
		561	defaultDir="%s/Data/" % os.getcwd(),
		562	defaultFile="*.txt",
		563	wildcard="",
		564	style=wx.SAVE
		565	)
		566	if dlg.ShowModal() == wx.ID_OK:
		567	paths = dlg.GetPaths();
		568	self.app.saveTests(paths[0])
		569	dlg.Destroy()
		570
599	FredericG	571	def onYAxisChange(self, event): # wxGlade: MainFrame.<event_handler>
592	FredericG	572	self.drawGraph()
		573
600	FredericG	574	def onGraphTypeChange(self, event): # wxGlade: MainFrame.<event_handler>
599	FredericG	575	self.drawGraph()
		576
607	FredericG	577	def OnSettings(self, event): # wxGlade: MainFrame.<event_handler>
		578	dlg = SettingsDialog(self, -1, "Sample Dialog", size=(350, 200),
		579	#style=wx.CAPTION \| wx.SYSTEM_MENU \| wx.THICK_FRAME,
		580	style=wx.DEFAULT_DIALOG_STYLE, # & ~wx.CLOSE_BOX
		581	)
		582	dlg.CenterOnScreen()
611	FredericG	583	val = dlg.ShowModal() # this does not return until the dialog is closed.
607	FredericG	584	dlg.Destroy()
611	FredericG	585	self.app.onSettingsChanged()
607	FredericG	586
612	FredericG	587	def onStartMeasure(self, event): # wxGlade: MainFrame.<event_handler>
613	FredericG	588	# Collect measure parameters
		589	mp = MeasureParameters()
		590	mp.descr = self.tcDescr.GetValue()
		591	mp.motors = map(int,self.tcMotors.GetValue().split(','))
		592	mp.channels = map(int,self.tcChannels.GetValue().split(','))
		593	s = self.tcSpeeds.GetValue()
619	FredericG	594	if s=="test":
		595	mp.speeds = (100,100,100,100,100, 150,150,150,150,150, 200,200,200,200,200, 100,150,200, 100,150,200, 100,150,200, 100,150,200)
		596	elif s.count("-") == 1:
613	FredericG	597	# assume from-to:step format
		598	s = s.split("-")
		599	if len(s) != 2: raise Exception("Invalid format")
		600	s[1] = s[1].split(":")
		601	if len(s[1]) != 2: raise Exception("Invalid format")
		602	mp.speeds = range(int(s[0]),int(s[1][0])+int(s[1][1]),int(s[1][1]))
		603	else:
614	FredericG	604	mp.speeds = map(int,s.split(','))
		605
613	FredericG	606
612	FredericG	607	# create the dialog that will show the satus
		608	dlg = MeasureDialog(self)
		609	dlg.CenterOnScreen()
613	FredericG	610
		611	# Signal the application
		612	self.app.startMeasure(mp, dlg)
		613
612	FredericG	614	# Show the dialog
		615	val = dlg.ShowModal() # this does not return until the dialog is closed.
		616	dlg.Destroy()
		617
616	FredericG	618	def _removeTest(self, idx):
		619	print "Deleting test %d" % idx
		620	self.app.removeTest(idx)
		621	self.TestListCtrl.DeleteItem(idx)
		622
		623
		624	def onClear(self, event): # wxGlade: MainFrame.<event_handler>
		625	print "Clear all tests"
		626	for i in range(len(self.app.VibTests)-1, -1, -1):
		627	self._removeTest(i)
		628	self.drawGraph()
		629
		630
		631	def onClearSelected(self, event): # wxGlade: MainFrame.<event_handler>
		632	while True:
		633	idx = self.TestListCtrl.GetFirstSelected()
		634	if idx == -1: break
		635	self._removeTest(idx)
		636
627	FredericG	637
626	FredericG	638
		639	def onAbout(self, event): # wxGlade: MainFrame.<event_handler>
		640	# First we create and fill the info object
		641	print "about"
		642	info = wx.AboutDialogInfo()
		643	info.Name = "MK Vibration Test"
		644	info.Version = "0.2"
		645	info.Copyright = ""
		646	info.Developers=["Frederic Goddeeris"]
		647	info.Description = "Please consult the WIKI page for a complete description of the tool:"
		648	info.WebSite = ("http://www.mikrokopter.de/ucwiki/en/VibrationTest", "VibrationTest WIKI page")
		649	wx.AboutBox(info)
		650
		651
		652	def onExit(self, event): # wxGlade: MainFrame.<event_handler>
		653	self.Close(True)
		654
		655
627	FredericG	656	def onSelectAll(self, event): # wxGlade: MainFrame.<event_handler>
		657	for i in xrange(self.TestListCtrl.GetItemCount()):
		658	self.TestListCtrl.Select(i)
		659
		660	def onCopyGraphData(self, event): # wxGlade: MainFrame.<event_handler>
		661	clipdata = wx.TextDataObject()
		662	txt = ""
		663	idx = self.TestListCtrl.GetFirstSelected()
		664	while idx != -1:
		665	txt += ("%d\n" % self.app.getTest(idx).getVibValue(self.app.settings["hpf"].value, self.app.settings["lpf"].value))
		666	idx = self.TestListCtrl.GetNextSelected(idx)
		667	clipdata.SetText(txt)
		668	wx.TheClipboard.Open()
		669	wx.TheClipboard.SetData(clipdata)
		670	wx.TheClipboard.Close()
		671
584	FredericG	672	# end of class MainFrame
		673
607	FredericG	674	class Setting:
		675	def __init__(self, descr, defaultValue):
		676	self.descr = descr
		677	self.value = defaultValue
611	FredericG	678
		679	def set(self, newValue):
		680	if isinstance(self.value, int):
		681	self.value = int(newValue)
		682	else:
		683	self.value = str(newValue)
584	FredericG	684
613	FredericG	685	class MeasureParameters:
		686	pass
612	FredericG	687
		688	class MeasureThread:
613	FredericG	689	def __init__(self, measureParameters, evtConsumer):
		690	self.mk = mkProto.MkComm()
		691	self.param = measureParameters
612	FredericG	692	self.evtConsumer = evtConsumer
613	FredericG	693	self.cancel = False
		694	self.running = False
612	FredericG	695
		696	def start(self):
		697	thread.start_new_thread(self._run, ())
		698
613	FredericG	699	def stop(self):
		700	self.cancel = True
		701
		702	def _testCancel(self):
		703	if self.cancel:
		704	raise Exception("Operation cancelled")
		705
623	FredericG	706	def _sendEvent(self, msg=None, error=False, parVoltage=None, speed=None):
		707	evt = MeasStatusUpdateEvent(running=self.running, msg=msg, error=error, voltage=parVoltage, speed=speed)
613	FredericG	708	wx.PostEvent(self.evtConsumer, evt)
		709
		710	def _setMotorSpeed(self, speed, settlingTime):
		711	speeds = [0,0,0,0]
		712	for motor in self.param.motors:
		713	speeds[motor-1] = speed
		714	for i in range(int(settlingTime*10)):
		715	self._testCancel()
		716	self.mk.setMotorTest(speeds)
		717	time.sleep(.1)
		718
		719
612	FredericG	720	def _run(self):
613	FredericG	721	self.running = True
		722	self._sendEvent("Starting test \"%s\"" % self.param.descr)
		723
		724	try:
		725	self._sendEvent("Opening SerialPort \"%s\"" % self.param.serialPort)
		726	self.mk.open(comPort=self.param.serialPort)
612	FredericG	727
613	FredericG	728	msg = self.mk.getVersionMsg()
		729	version = msg.getVersion()
		730	self._sendEvent("Version: %d.%d" % version)
		731
		732	msg = self.mk.getDebugMsg()
		733	voltage = msg.getVoltage()
		734	if (voltage == 0):
		735	minVoltage = 0
623	FredericG	736	maxVoltage = 1
613	FredericG	737	else:
		738	if (voltage > 4.2*3):
623	FredericG	739	nbCells = 4
613	FredericG	740	else:
623	FredericG	741	nbCells = 3
		742	minVoltage = nbCells*3.5
		743	maxVoltage = nbCells*3.9
613	FredericG	744	self._sendEvent("Voltage: %2.1fV" % voltage)
623	FredericG	745	self._sendEvent("Min/Max Voltage: %2.1fV-%2.1fV" % (minVoltage, maxVoltage), parVoltage=(minVoltage, maxVoltage, voltage))
		746
613	FredericG	747
623	FredericG	748	self._sendEvent("Starting motor(s) (speed=%d)... " % self.param.motorStartupSpeed, speed=self.param.motorStartupSpeed)
613	FredericG	749	self._setMotorSpeed(self.param.motorStartupSpeed, self.param.motorStartupSettlingTime)
		750
		751	for speed in self.param.speeds:
623	FredericG	752	self._sendEvent("Changing motor speed to %d... " % speed, speed=speed)
613	FredericG	753	self._setMotorSpeed(speed, 1)
		754
		755	for channel in self.param.channels:
		756	self._setMotorSpeed(speed, .1)
619	FredericG	757	msg = self.mk.getDebugMsg()
		758	voltage = msg.getVoltage()
		759
623	FredericG	760	self._sendEvent("Getting data from channel %s" % CHANNEL_NAMES[channel], parVoltage=(minVoltage, maxVoltage, voltage))
613	FredericG	761	data = self.mk.doVibrationTest(1000, channel)
		762
619	FredericG	763	vt = VibTest(self.param.descr, voltage, self.param.motors, speed, CHANNEL_NAMES[channel], data)
613	FredericG	764	evt = MeasDataEvent(vibTest = vt)
		765	wx.PostEvent(self.evtConsumer, evt)
		766
619	FredericG	767	if voltage<minVoltage:
		768	raise Exception("Voltage too low")
		769
613	FredericG	770	self._sendEvent("Done !")
612	FredericG	771
613	FredericG	772	except Exception, e:
		773	self._sendEvent("Exception \"%s\"" % e, error=True)
		774
		775	self.running = False
		776	self._sendEvent("")
		777
612	FredericG	778
586	FredericG	779	class VibTest:
619	FredericG	780	def __init__(self, descr, voltage, motor, speed, channel, rawData):
594	FredericG	781	self.descr = descr
619	FredericG	782	self.voltage = voltage
594	FredericG	783	self.motor = motor
		784	self.speed = speed
596	FredericG	785	self.channel = channel
599	FredericG	786
		787	self.dataLen = len(rawData)
		788
590	FredericG	789	self.rawData = _Numeric.array(rawData)
		790	self.dc = self.rawData.mean()
		791	self.rawData -= self.dc
586	FredericG	792
600	FredericG	793	self.fft = _Numeric.fft.rfft(self.rawData)
		794
		795	self.spectrum = None
		796	self.filteredData = None
615	FredericG	797	self.fc1 = None
		798	self.fc2 = None
599	FredericG	799
602	FredericG	800	self.vibValue = None
		801
594	FredericG	802	def getDescr(self):
		803	return self.Descr
587	FredericG	804
590	FredericG	805	def getRawData(self):
		806	return self.rawData
		807
		808	def getDataLen(self):
		809	return self.dataLen
		810
599	FredericG	811	def getSpectrum(self):
600	FredericG	812	if self.spectrum == None:
		813	self.spectrum = _Numeric.absolute(self.fft[1:self.dataLen/2+1]) / (self.dataLen/2)
		814	return self.spectrum
590	FredericG	815
615	FredericG	816	def getFilteredData(self, fc1, fc2):
		817	if self.fc1 != fc1 or self.fc2 != fc2:
		818	self.filteredData = None
		819
600	FredericG	820	if self.filteredData == None:
		821	tmpfft = self.fft.copy()
615	FredericG	822	fc = (float(fc1))/(FS/2)*len(tmpfft)
		823	print "fc1=%d => fc=%d" % (fc1,fc)
		824	for i in range(0,int(fc)):
600	FredericG	825	tmpfft[i] = 0
615	FredericG	826	fc = (float(fc2))/(FS/2)*len(tmpfft)
		827	print "fc2=%d => fc=%d" % (fc2,fc)
		828	for i in range(int(fc), len(tmpfft)):
600	FredericG	829	tmpfft[i] = 0
		830	self.filteredData = _Numeric.fft.irfft(tmpfft)
615	FredericG	831	self.fc1 = fc1
		832	self.fc2 = fc2
		833
600	FredericG	834	return self.filteredData
602	FredericG	835
615	FredericG	836	def getVibValue(self, fc1, fc2):
		837	if self.fc1 != fc1 or self.fc2 != fc2:
		838	self.vibValue = None
602	FredericG	839	if self.vibValue == None:
615	FredericG	840	fd = self.getFilteredData(fc1, fc2)[100:-100];
602	FredericG	841	self.vibValue = max(fd)-min(fd)
		842	return self.vibValue
599	FredericG	843
607	FredericG	844
		845
584	FredericG	846	class App(wx.App):
607	FredericG	847
		848	SETTINGSFILE = "settings.cfg"
		849
586	FredericG	850	def __init__(self, par):
		851	self.VibTests = []
		852	wx.App.__init__(self, par)
		853
607	FredericG	854	# Init settings
		855	self.settings={}
611	FredericG	856	self.settings["serialport"] = Setting("Serial Port", "COM1")
613	FredericG	857	self.settings["startupspeed"] = Setting("Motor Startup Speed", 25)
		858	self.settings["startupsettling"] = Setting("Motor Startup Setting time (s)", 3)
		859	self.settings["serialport"] = Setting("Serial Port", "COM1")
607	FredericG	860	self.settings["hpf"] = Setting("HP Filter cutoff (Hz)", 50)
615	FredericG	861	self.settings["lpf"] = Setting("LP Filter cutoff (Hz)", 180)
607	FredericG	862
		863	self.readSettings()
		864
606	FredericG	865	if len(sys.argv)>1:
619	FredericG	866	self.loadTests(sys.argv[1])
586	FredericG	867
606	FredericG	868
607	FredericG	869	def readSettings(self):
		870	print "Reading settings"
		871	cp = ConfigParser.ConfigParser()
		872
		873	try:
		874	cp.read(App.SETTINGSFILE)
		875	for setting in cp.items("DEFAULT"):
		876	print " ",setting
		877	try:
613	FredericG	878	self.settings[setting[0]].set(setting[1])
607	FredericG	879	except:
		880	print "WARNING, unknown setting"
		881	except:
		882	print "ERROR reading settingsfile"
		883
		884
		885	def storeSettings(self):
		886	print "Storing settings"
		887
		888	cp = ConfigParser.ConfigParser()
		889	for setting in self.settings.iteritems():
		890	cp.set("", setting[0], setting[1].value)
		891
		892	file = open(App.SETTINGSFILE, "w")
		893	cp.write(file)
		894	file.close()
		895
611	FredericG	896
		897	def onSettingsChanged(self):
		898	self.storeSettings()
613	FredericG	899
		900	def AddTest2(self, vibTest):
		901	self.VibTests.append(vibTest)
		902	self.frame_1.onNewTest(vibTest)
611	FredericG	903
619	FredericG	904	def AddTest(self, descr, voltage, motor, speed, channel, rawData):
		905	test = VibTest(descr, voltage, motor, speed, channel, rawData)
613	FredericG	906	self.AddTest2(test)
586	FredericG	907
616	FredericG	908	def removeTest(self, idx):
		909	del(self.VibTests[idx])
		910
590	FredericG	911	def getTest(self, testId):
		912	return self.VibTests[testId]
		913
584	FredericG	914	def OnInit(self):
		915	wx.InitAllImageHandlers()
586	FredericG	916	self.frame_1 = MainFrame(None, -1, "")
587	FredericG	917	self.frame_1.setApp(self);
586	FredericG	918	self.SetTopWindow(self.frame_1)
		919
607	FredericG	920	self.frame_1.CenterOnScreen()
587	FredericG	921	self.frame_1.Show()
		922	return 1
		923
619	FredericG	924	def saveTests(self, filePath):
		925	try:
		926	logfile = open(filePath, "r")
		927	newFile = False
		928	logfile.close()
		929	except:
		930	newFile = True
		931
		932	for test in self.VibTests:
		933	if newFile:
		934	logfile = open(filePath, "w")
		935	print "Writing result to %s ..." % filePath,
		936	logfile.write("%s %d %s\n" % (test.descr, test.speed, test.channel))
		937	for value in test.rawData:
		938	logfile.write("%d\n" % value)
		939	logfile.close()
		940	print "OK"
		941	else:
		942	print "Appending result to %s ..." % filePath,
		943	logfile = open(filePath, "r")
		944	prevData = []
		945	for line in logfile:
		946	prevData.append(line[:-1])
		947	logfile.close()
		948	logfile = open(filePath, "w")
		949	logfile.write("%s,%s %d %s\n" % (prevData[0], test.descr, test.speed, test.channel))
		950	i = 1
		951	for value in test.rawData:
		952	logfile.write("%s,%d\n" % (prevData[i], value))
		953	i += 1
		954	logfile.close()
		955	print "OK"
		956	newFile = False
		957
		958
		959	def loadTests(self, filePath):
596	FredericG	960
606	FredericG	961	print "Importing file \"%s\"" % filePath
596	FredericG	962
		963	logfile = open(filePath, "r")
590	FredericG	964	data = None
596	FredericG	965
619	FredericG	966	headers = (logfile.readline()[:-1]).split(',')
596	FredericG	967	nbCols = len(headers)
		968	print "NbCols =", nbCols
		969
		970	data = []
		971	descr = []
		972	speed = []
		973	channel = []
		974	for c in range(nbCols):
		975	data.append([])
		976	h = headers[c].split(' ')
		977	descr.append(h[0]);
		978	speed.append(h[1]);
		979	channel.append(h[2]);
		980
590	FredericG	981	for line in logfile:
		982	values = line.split(',')
596	FredericG	983	for i in range(nbCols):
		984	data[i].append(int(values[i]))
590	FredericG	985	logfile.close()
596	FredericG	986
		987	for c in range(nbCols):
599	FredericG	988	if (len(data[c]) % 2) != 0:
		989	data[c].append(data[c][-1])
619	FredericG	990	self.AddTest(descr[c], 0, 0, int(speed[c]), channel[c], data[c])
612	FredericG	991
613	FredericG	992	def startMeasure(self, measureParams, dialog):
612	FredericG	993	print "Start measuring"
590	FredericG	994
613	FredericG	995	measureParams.serialPort = self.settings["serialport"].value
		996	measureParams.motorStartupSpeed = self.settings["startupspeed"].value
		997	measureParams.motorStartupSettlingTime = self.settings["startupsettling"].value
		998
		999	self.measureThread = MeasureThread(measureParams, dialog)
612	FredericG	1000	self.measureThread.start()
590	FredericG	1001
613	FredericG	1002	def cancelMeasurement(self):
		1003	print "Measuring CANCEL"
		1004
		1005	self.measureThread.stop()
		1006
612	FredericG	1007
		1008
		1009
		1010
590	FredericG	1011
586	FredericG	1012
590	FredericG	1013
584	FredericG	1014	# end of class App
		1015
		1016	if __name__ == "__main__":
626	FredericG	1017
		1018	rootPath = os.path.abspath(os.path.dirname(sys.argv[0]))
		1019
		1020	print rootPath
		1021
584	FredericG	1022	VibrationTestGui = App(0)
		1023	VibrationTestGui.MainLoop()

Subversion Repositories Projects

(root)/VibrationTest/branches/v1.2/VibrationTest/VibrationTestGui.py - Rev 627