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

Subversion Repositories Projects

(root)/VibrationTest/branches/ParameterTest/VibrationTest/VibrationTestGui.py @ 718 - Rev 647