Subversion Repositories Projects

Rev

Go to most recent revision | Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
552 fredericg 1
#! /usr/bin/env python
2
 
3
#
4
# Mikrokopter Vibration Test
5
#
6
# Author: FredericG
7
# 
8
 
9
 
10
import mkProto
11
import time
12
import traceback
13
import getopt
14
import sys
15
 
16
 
17
 
18
CHANNEL_NAMES = ["GyroYaw", "GyroRoll", "GyroNick", "Pressure", "Batt", "AccTop", "AccRoll", "AccNick"]
19
 
20
def usage():
21
    print
22
    print "VibrationTest.py COMPORT MOTORS SPEEDS CHANNELS [-m MINSPEED] [-s NBSAMPLES] [-n NAME] [-d FILENAME] [-v]"
23
    print " COMPORT         Serial port to use. e.g. COM4"
24
    print " MOTORS          Motors to activate during test. Multiple motors can be used at the same time. e.g. 1,2,3,4"
554 FredericG 25
    print " SPEEDS          Indicates at what speeds the motors need to be tested. "
26
    print "                   Format 1:  e.g. 50,110,140            Tests at speeds 50, 110 and 140"
27
    print "                   Format 2:  e.g. 100-200:50            Tests at speeds 100, 150 and 200"
552 fredericg 28
    print " CHANNELS        Channels to monitor. e.g. 5,6,7"
554 FredericG 29
    for i,channelName in enumerate(CHANNEL_NAMES):
552 fredericg 30
      print "                   Channel %d = %s" % (i, channelName)
31
    print " -m MINSPEED     Minimum speed of the motor(s)"
32
    print " -s NBSAMPLES    Number of samples"
33
    print " -n NAME         Name of the test"
34
    print " -d FILENAME     File to which the measured values will be logged in"
35
    print " -v              Verbose"          
36
    print
554 FredericG 37
 
38
def exit_usage():
39
    usage()
40
    sys.exit(2)
552 fredericg 41
 
42
if __name__ == '__main__':
43
 
44
    #print sys.argv
45
    if len(sys.argv)<5:
554 FredericG 46
        exit_usage()
552 fredericg 47
 
48
    parComPort = sys.argv[1]
49
    parMotors = sys.argv[2].split(',')
50
    parChannels = sys.argv[4].split(',')
554 FredericG 51
 
52
    par = sys.argv[3]
53
    if par.count("-") == 1:
54
        # assume from-to:step format
55
        par = par.split("-")
56
        if len(par) != 2:
57
            exit_usage()
58
 
59
        par[1] = par[1].split(":")
60
        if len(par[1]) != 2:
61
            exit_usage()
552 fredericg 62
 
554 FredericG 63
        parSpeeds = range(int(par[0]),int(par[1][0])+int(par[1][1]),int(par[1][1]))
64
 
65
    else:
66
        parSpeeds = par.split(',')
67
 
552 fredericg 68
    try:
69
        opts, args = getopt.getopt(sys.argv[5:], "s:n:d:m:v")
70
    except Exception, err:
71
        print str(err) # will print something like "option -a not recognized"
72
        usage()
73
        sys.exit(2)
74
 
75
    parVerbose = False
76
    parFileName = None
77
    parNbSamples = 400
78
    parMinSpeed = 25
79
    parTestName = ""
80
 
81
    for o, a in opts:
82
        if o == "-v":
83
            parVerbose = True
84
        elif o == "-d":
85
            parFileName = a
86
        elif o == "-s":
87
            parNbSamples = int(a)
88
        elif o == "-m":
89
            parMinSpeed = int(a)
90
        elif o == "-n":
91
            parTestName = a
92
 
93
        else:
94
            assert False, "unhandled option %s" % (o)
95
 
96
    if parVerbose:
97
        print "comPort  =", parComPort
98
        print "motors   =", parMotors
99
        print "minSpeed =", parMinSpeed
100
        print "speeds   =", parSpeeds
101
        print "channels =", parChannels
102
        print "nbSamples=", parNbSamples
103
        print "fileName =", parFileName
104
        print "testName =", parTestName
105
 
106
 
107
    try:
108
        parMotors = map(int, parMotors)
109
        parSpeeds = map(int, parSpeeds)
110
        parChannels = map(int, parChannels)
111
 
112
 
113
        if parVerbose:
114
          print "Opening comPort... "
115
        mk = mkProto.MkComm()
116
        mk.open(comPort=parComPort)
117
 
118
        msg = mk.getVersionMsg()
119
        voltage = 0
120
        if parVerbose:
121
          print "Version: %d.%d" % msg.getVersion()
122
 
123
        msg = mk.getDebugMsg()
124
        if (voltage > 4.2*3):
125
          minVoltage = 4*3.5
126
        else:
127
          minVoltage = 3*3.5
128
 
129
        if parVerbose:
130
          print "Voltage: %2.1fV" % msg.getVoltage()
131
          print "Minimum Voltage: %2.1fV" % minVoltage
132
 
133
        motorStartSpeeds = [0,0,0,0]
134
 
135
        if parVerbose:
136
          print "Starting motor(s) (speed=%d)... " % (parMinSpeed),
137
        for i in range(0,parMinSpeed):
138
          time.sleep(.2)
139
          for motor in parMotors:
140
            motorStartSpeeds[motor-1] = i
141
          mk.setMotorTest(motorStartSpeeds)
142
 
143
        if parVerbose:
144
          print "OK"
145
 
146
        for speed in parSpeeds:
147
          if len(parChannels)>1:
148
            print
149
 
150
          if parVerbose:
151
            print "Setting speed to %3d ..." % speed
152
          motorSpeeds = [0,0,0,0]  
153
          for motor in parMotors:
154
            motorSpeeds[motor-1] = speed
155
          for i in range(0,10):
156
            time.sleep(.1)
157
            mk.setMotorTest(motorSpeeds)
158
 
159
          for channel in parChannels:
160
            channelName = CHANNEL_NAMES[channel]
161
            if parVerbose:
162
              print "Getting data... " ,
163
            data = mk.doVibrationTest(motorSpeeds, parNbSamples, channel)
553 FredericG 164
            minval = min(data[1:])
165
            maxval = max(data[1:])
552 fredericg 166
            if (maxval-minval)>100: print data
167
            msg = mk.getDebugMsg()
168
            voltage = msg.getVoltage()
169
 
170
            if voltage<minVoltage:
171
              print "VOLTAGE TOO LOW, TEST ABORTED"
172
              sys.exit(2)
173
 
174
            print "%s Speed=%3d U=%2.1fV Channel=%-10s Min=%3d Max=%3d pp=%3d" % (parTestName,  speed, voltage, channelName, minval, maxval, maxval-minval)
175
 
176
            if parFileName != None:
177
              try:
178
                logfile = open(parFileName, "r")
179
                newFile = False
180
              except:
181
                newFile = True
182
 
183
              if newFile:
184
                logfile = open(parFileName, "w")
185
                if parVerbose:
186
                  print "Writing result to %s ..." % parFileName,
187
                logfile.write("%s %d %s\n" % (parTestName, speed, channelName))
188
                for value in data[1:]:
189
                  logfile.write("%d\n" % value)
190
                logfile.close()
191
                if parVerbose:  
192
                  print "OK"
193
              else:
194
                if parVerbose:
195
                  print "Appending result to %s ..." % parFileName,
196
                prevData = []
197
                for line in logfile:
198
                  prevData.append(line[:-1])
199
                logfile.close()
200
                logfile = open(parFileName, "w")
201
                logfile.write("%s,%s %d %s\n" % (prevData[0], parTestName, speed, channelName))
202
                i = 1
203
                for value in data[1:]:
204
                  logfile.write("%s,%d\n" % (prevData[i], value))
205
                  i += 1
206
                logfile.close()
207
                if parVerbose:
208
                  print "OK"
209
 
210
    except Exception,e:
211
      print
212
      print "== ERROR ==: \"%s\"" % e
213
      if parVerbose:
214
        print
215
        print "Traceback:"
216
        traceback.print_exc()
217
        print
218
      raw_input("Press ENTER, the application will close")
219
      print