WebSVN - Projects - Diff - Rev 612 and 613 - /VibrationTest/trunk/VibrationTest/VibrationTestGui.py/

 import wx
 import wx.lib
 import wx.lib.plot
 import wx.lib.newevent
+import mkProto
+CHANNEL_NAMES = ["GyroYaw", "GyroRoll", "GyroNick", "Pressure", "Batt", "AccTop", "AccRoll", "AccNick"]
 # Needs Numeric or numarray or NumPy
 try:
     import numpy.oldnumeric as _Numeric
 except:
 # begin wxGlade: extracode
 # end wxGlade
+# This creates a new Event class and a EVT binder function
-# This creates a new Event class and a EVT binder function
+(MeasStatusUpdateEvent, EVT_MEAS_STATUS_UPDATE) = wx.lib.newevent.NewEvent()
-(MeasStatusUpdateEvent, EVT_MEAS_STATUS_UPDATE) = wx.lib.newevent.NewEvent()
+(MeasDataEvent, EVT_MEAS_DATA) = wx.lib.newevent.NewEvent()
 class MeasureDialog(wx.Dialog):
         self.Bind(wx.EVT_BUTTON, self.onButton, self.button)
         # end wxGlade
         self.running = True
+        self.Bind(EVT_MEAS_STATUS_UPDATE, self.OnUpdate)
+        self.Bind(EVT_MEAS_DATA, self.OnData)
+        # The first argument that is passed to the constructor is the parent
+        self.app = args[0].app
-        self.Bind(EVT_MEAS_STATUS_UPDATE, self.OnUpdate)
+        self.error = False
     def __set_properties(self):
         # begin wxGlade: MeasureDialog.__set_properties
         sizer_1.Add((20, 20), 0, 0, 0)
         self.SetSizer(sizer_1)
         sizer_1.Fit(self)
         self.Layout()
         # end wxGlade
+    def OnData(self, evt):
+        print "Received Data"
+        self.app.AddTest2(evt.vibTest)
     def OnUpdate(self, evt):
-        print "Event received"
+        print "Status update"
+        self.running = evt.running
+        if evt.error:
+            self.error = True;
+            self.text_ctrl_1.WriteText("ERROR: ")
-        self.running = evt.running
+            self.text_ctrl_1.SetBackgroundColour("Red")
         self.text_ctrl_1.WriteText("%s\n"%evt.msg)
+        if (not self.running):
+            if (not self.error):
+                self.text_ctrl_1.SetBackgroundColour("Green")
-        if (not self.running):
+                self.text_ctrl_1.write(" ") # so that the background is redrawn
+            self.button.SetLabel("Close")
             self.button.SetLabel("Close")
     def onButton(self, event): # wxGlade: MeasureDialog.<event_handler>
+        if (not self.running):
+            self.Destroy()
-        if (not self.running):
+        else:
-            self.Destroy()
+            self.app.cancelMeasurement()
         wxglade_tmp_menu = wx.Menu()
         self.frame_1_menubar.Append(wxglade_tmp_menu, "Help")
         self.SetMenuBar(self.frame_1_menubar)
         # Menu Bar end
         self.Description = wx.StaticText(self, -1, "Description")
-        self.text_ctrl_6 = wx.TextCtrl(self, -1, "Test")
+        self.tcDescr = wx.TextCtrl(self, -1, "Test")
         self.label_37 = wx.StaticText(self, -1, "Speed(s)")
-        self.text_ctrl_9 = wx.TextCtrl(self, -1, "100-200:10")
+        self.tcSpeeds = wx.TextCtrl(self, -1, "100-200:10")
         self.label_35 = wx.StaticText(self, -1, "Motor(s)")
-        self.text_ctrl_7 = wx.TextCtrl(self, -1, "1")
+        self.tcMotors = wx.TextCtrl(self, -1, "1")
         self.label_38 = wx.StaticText(self, -1, "")
         self.text_ctrl_10 = wx.TextCtrl(self, -1, "")
-        self.label_36 = wx.StaticText(self, -1, "Channel")
+        self.label_36 = wx.StaticText(self, -1, "Channel(s)")
-        self.text_ctrl_8 = wx.TextCtrl(self, -1, "6")
+        self.tcChannels = wx.TextCtrl(self, -1, "6")
         self.label_39 = wx.StaticText(self, -1, "")
         self.text_ctrl_11 = wx.TextCtrl(self, -1, "")
         self.button_4 = wx.Button(self, -1, "Start")
         self.GraphPanel = wx.Panel(self, -1)
         self.label_40 = wx.StaticText(self, -1, "Graph Type ")
         sizer_10 = wx.BoxSizer(wx.HORIZONTAL)
         grid_sizer_1 = wx.GridSizer(3, 4, 4, 5)
         sizer_3.Add((20, 20), 0, 0, 0)
         sizer_8.Add((20, 20), 0, 0, 0)
         grid_sizer_1.Add(self.Description, 0, wx.ALIGN_RIGHT|wx.ALIGN_CENTER_VERTICAL, 0)
-        grid_sizer_1.Add(self.text_ctrl_6, 0, 0, 0)
+        grid_sizer_1.Add(self.tcDescr, 0, 0, 0)
         grid_sizer_1.Add(self.label_37, 0, wx.ALIGN_RIGHT|wx.ALIGN_CENTER_VERTICAL, 0)
-        grid_sizer_1.Add(self.text_ctrl_9, 0, 0, 0)
+        grid_sizer_1.Add(self.tcSpeeds, 0, 0, 0)
         grid_sizer_1.Add(self.label_35, 0, wx.ALIGN_RIGHT|wx.ALIGN_CENTER_VERTICAL, 0)
-        grid_sizer_1.Add(self.text_ctrl_7, 0, 0, 0)
+        grid_sizer_1.Add(self.tcMotors, 0, 0, 0)
         grid_sizer_1.Add(self.label_38, 0, wx.ALIGN_RIGHT|wx.ALIGN_CENTER_VERTICAL, 0)
         grid_sizer_1.Add(self.text_ctrl_10, 0, 0, 0)
         grid_sizer_1.Add(self.label_36, 0, wx.ALIGN_RIGHT|wx.ALIGN_CENTER_VERTICAL, 0)
-        grid_sizer_1.Add(self.text_ctrl_8, 0, 0, 0)
+        grid_sizer_1.Add(self.tcChannels, 0, 0, 0)
         grid_sizer_1.Add(self.label_39, 0, wx.ALIGN_RIGHT|wx.ALIGN_CENTER_VERTICAL, 0)
         grid_sizer_1.Add(self.text_ctrl_11, 0, 0, 0)
         sizer_9.Add(grid_sizer_1, 0, 0, 0)
         sizer_10.Add((50, 20), 0, 0, 0)
         sizer_10.Add(self.button_4, 0, wx.ALIGN_CENTER_VERTICAL, 0)
         self.TestListCtrl.SetStringItem(index, 2, test.channel)
         vv = int(test.getVibValue())
         vvs = "|%s| (%d)" % ("----------------------------------------------------------------------------------------------------"[0:min(vv,100)], vv)
-        self.TestListCtrl.SetStringItem(index, 3, vvs)
-        if (index == 0):
+        self.TestListCtrl.SetStringItem(index, 3, vvs)
-            self.TestListCtrl.Select(index)
+        self.TestListCtrl.Select(index)
         val = dlg.ShowModal()  # this does not return until the dialog is closed.
         dlg.Destroy()
         self.app.onSettingsChanged()
+    def onStartMeasure(self, event): # wxGlade: MainFrame.<event_handler>
+        # Collect measure parameters
+        mp = MeasureParameters()
+        mp.descr = self.tcDescr.GetValue()
+        mp.motors = map(int,self.tcMotors.GetValue().split(','))
+        mp.channels = map(int,self.tcChannels.GetValue().split(','))
+        s = self.tcSpeeds.GetValue()
+        if s.count("-") == 1:
+            # assume from-to:step format
+            s = s.split("-")
+            if len(s) != 2: raise Exception("Invalid format")
+            s[1] = s[1].split(":")
+            if len(s[1]) != 2: raise Exception("Invalid format")
+            mp.speeds = range(int(s[0]),int(s[1][0])+int(s[1][1]),int(s[1][1]))
+        else:
-    def onStartMeasure(self, event): # wxGlade: MainFrame.<event_handler>
+            mp.speeds = s.split(',')
         # create the dialog that will show the satus
         dlg = MeasureDialog(self)
+        dlg.CenterOnScreen()
-        dlg.CenterOnScreen()
-        # Signal that application
+        # Signal the application
+        self.app.startMeasure(mp, dlg)
         self.app.startMeasure(dlg)
         # Show the dialog
         val = dlg.ShowModal()  # this does not return until the dialog is closed.
         dlg.Destroy()
         if isinstance(self.value, int):
             self.value = int(newValue)
         else:
             self.value = str(newValue)
+class MeasureParameters:
       pass
+class MeasureThread:
+    def __init__(self, measureParameters, evtConsumer):
-class MeasureThread:
+        self.mk = mkProto.MkComm()
+        self.param = measureParameters
+        self.evtConsumer = evtConsumer
-    def __init__(self, evtConsumer):
+        self.cancel = False
-        self.evtConsumer = evtConsumer
+        self.running = False
     def start(self):
+        thread.start_new_thread(self._run, ())
-        thread.start_new_thread(self._run, ())
+    def stop(self):
-    def _run(self):
+        self.cancel = True
-        i=0
+    def _testCancel(self):
-        for i in range(20):
+        if self.cancel:
-            time.sleep(0.5)
+            raise Exception("Operation cancelled")
+    def _sendEvent(self, msg, error=False):
+        evt = MeasStatusUpdateEvent(running=self.running, msg=msg, error=error)
+        wx.PostEvent(self.evtConsumer, evt)
+    def _setMotorSpeed(self, speed, settlingTime):
+        speeds = [0,0,0,0]
+        for motor in self.param.motors:
+            speeds[motor-1] = speed
+        for i in range(int(settlingTime*10)):
+          self._testCancel()
+          self.mk.setMotorTest(speeds)
+          time.sleep(.1)
+    def _run(self):
+        self.running = True
+        self._sendEvent("Starting test \"%s\"" % self.param.descr)
+        try:
+            self._sendEvent("Opening SerialPort \"%s\"" % self.param.serialPort)
+            self.mk.open(comPort=self.param.serialPort)
+            msg = self.mk.getVersionMsg()
+            version = msg.getVersion()
+            self._sendEvent("Version: %d.%d" % version)
+            msg = self.mk.getDebugMsg()
+            voltage = msg.getVoltage()
+            if (voltage == 0):
+              minVoltage = 0
+            else:
+              if (voltage > 4.2*3):
+                minVoltage = 4*3.5
+              else:
+                minVoltage = 3*3.5
+            self._sendEvent("Voltage: %2.1fV" % voltage)
+            self._sendEvent("Minimum Voltage: %2.1fV" % minVoltage)
+            self._sendEvent("Starting motor(s) (speed=%d)... " % self.param.motorStartupSpeed)
+            self._setMotorSpeed(self.param.motorStartupSpeed, self.param.motorStartupSettlingTime)
+            for speed in self.param.speeds:
+                self._sendEvent("Changing motor speed to %d... " % speed)
+                self._setMotorSpeed(speed, 1)
+                for channel in self.param.channels:
+                    self._setMotorSpeed(speed, .1)
+                    self._sendEvent("Getting data from channel %s" % CHANNEL_NAMES[channel])
+                    data = self.mk.doVibrationTest(1000, channel)
+                    vt = VibTest(self.param.descr, self.param.motors, speed, CHANNEL_NAMES[channel], data)
+                    evt = MeasDataEvent(vibTest = vt)
+                    wx.PostEvent(self.evtConsumer, evt)
+            self._sendEvent("Done !")
-            evt = MeasStatusUpdateEvent(running=True, msg="Hello %d" % i)
+        except Exception, e:
-            wx.PostEvent(self.evtConsumer, evt)
+            self._sendEvent("Exception \"%s\"" % e, error=True)
         evt = MeasStatusUpdateEvent(running=False, msg="Done")
         wx.App.__init__(self, par)
         # Init settings
         self.settings={}
+        self.settings["serialport"] = Setting("Serial Port", "COM1")
+        self.settings["startupspeed"] = Setting("Motor Startup Speed", 25)
+        self.settings["startupsettling"] = Setting("Motor Startup Setting time (s)", 3)
         self.settings["serialport"] = Setting("Serial Port", "COM1")
         self.settings["hpf"] = Setting("HP Filter cutoff (Hz)", 50)
         self.settings["lpf"] = Setting("LP Filter cutoff (Hz)", 400)
         try:
             cp.read(App.SETTINGSFILE)
             for setting in cp.items("DEFAULT"):
                 print " ",setting
                 try:
-                    self.settings[setting[0]].value = setting[1]
+                    self.settings[setting[0]].set(setting[1])
                 except:
                     print "WARNING, unknown setting"
         except:
             print "ERROR reading settingsfile"
         file.close()
     def onSettingsChanged(self):
+        self.storeSettings()
+    def AddTest2(self, vibTest):
-        self.storeSettings()
+        self.VibTests.append(vibTest)
         self.frame_1.onNewTest(vibTest)
-    def AddTest(self, descr, motor, speed, channel, rawData):
-        test = VibTest(descr, motor, speed, channel, rawData)
+    def AddTest(self, descr, motor, speed, channel, rawData):
-        self.VibTests.append(test)
+        test = VibTest(descr, motor, speed, channel, rawData)
-        self.frame_1.onNewTest(test)
+        self.AddTest2(test)
         for c in range(nbCols):
             if (len(data[c]) % 2) != 0:
                 data[c].append(data[c][-1])
             self.AddTest(descr[c], 0, int(speed[c]), channel[c], data[c])
-    def startMeasure(self, dialog):
+    def startMeasure(self, measureParams, dialog):
+        print "Start measuring"
+        measureParams.serialPort = self.settings["serialport"].value
+        measureParams.motorStartupSpeed = self.settings["startupspeed"].value
-        print "Start measuring"
+        measureParams.motorStartupSettlingTime = self.settings["startupsettling"].value
+        self.measureThread = MeasureThread(measureParams, dialog)
+        self.measureThread.start()
+    def cancelMeasurement(self):
+        print "Measuring CANCEL"

Subversion Repositories Projects

(root)/VibrationTest/trunk/VibrationTest/VibrationTestGui.py/ - Rev 612 → 613