Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 552 | fredericg | 1 | #! /usr/bin/env python |
| 2 | |||
| 3 | # |
||
| 4 | # Mikrokopter Serial protocol |
||
| 5 | # |
||
| 6 | # Author: FredericG |
||
| 7 | # |
||
| 8 | |||
| 9 | import os |
||
| 10 | import glob |
||
| 11 | import serial |
||
| 12 | import time |
||
| 13 | import traceback |
||
| 14 | |||
| 15 | class MkException(Exception): |
||
| 16 | pass |
||
| 17 | |||
| 18 | class InvalidMsg(MkException): |
||
| 19 | def __str__(self): |
||
| 20 | return "Invalid message" |
||
| 21 | |||
| 22 | class CrcError(MkException): |
||
| 23 | def __str__(self): |
||
| 24 | return "CRC error" |
||
| 25 | |||
| 26 | class InvalidArguments(MkException): |
||
| 27 | def __str__(self): |
||
| 28 | return "Invalid Arguments" |
||
| 29 | |||
| 30 | class InvalidMsgType(MkException): |
||
| 31 | def __str__(self): |
||
| 32 | return "Invalid Message type" |
||
| 33 | |||
| 34 | class NoResponse(MkException): |
||
| 35 | def __init__(self, cmd): |
||
| 36 | self.cmd = cmd |
||
| 37 | |||
| 38 | def __str__(self): |
||
| 39 | return "No Reponse. Waiting for \"%s\" message" % self.cmd |
||
| 40 | |||
| 41 | pass |
||
| 42 | |||
| 43 | def calcCrcBytes(str): |
||
| 44 | crc = 0 |
||
| 45 | for c in str: |
||
| 46 | crc += ord(c) |
||
| 47 | crc &= 0xfff |
||
| 48 | return (chr(crc/64+ord('=')), chr(crc%64+ord('='))) |
||
| 49 | |||
| 50 | |||
| 51 | class MkMsg: |
||
| 52 | def __init__(self, msg=None, address=None, cmd=None, data=None): |
||
| 53 | if (msg != None): |
||
| 54 | # Create instance based on received message |
||
| 55 | self.parse(msg) |
||
| 56 | elif (address != None and cmd != None and data != None): |
||
| 57 | # Create instance based on address, command and data |
||
| 58 | self.address = address |
||
| 59 | self.cmd = cmd |
||
| 60 | self.data = data |
||
| 61 | else: |
||
| 62 | # Cannot create instance |
||
| 63 | raise InvalidArguments |
||
| 64 | |||
| 65 | def generateMsg(self): |
||
| 66 | msg = "" |
||
| 67 | |||
| 68 | # make header |
||
| 69 | msg += '#' |
||
| 70 | msg += chr(self.address+ord('a')) |
||
| 71 | msg += self.cmd |
||
| 72 | |||
| 73 | # add data |
||
| 74 | done = False |
||
| 75 | i = 0 |
||
| 76 | while (i<len(self.data)) and not done: |
||
| 77 | a = 0 |
||
| 78 | b = 0 |
||
| 79 | c = 0 |
||
| 80 | try: |
||
| 81 | a = self.data[i] |
||
| 82 | b = self.data[i+1] |
||
| 83 | c = self.data[i+2] |
||
| 84 | i = i + 3 |
||
| 85 | except IndexError: |
||
| 86 | done = True |
||
| 87 | msg += chr(ord('=') + (a >> 2)) |
||
| 88 | msg += chr(ord('=') + (((a & 0x03) << 4) | ((b & 0xf0) >> 4))) |
||
| 89 | msg += chr(ord('=') + (((b & 0x0f) << 2) | ((c & 0xc0) >> 6))) |
||
| 90 | msg += chr(ord('=') + ( c & 0x3f)) |
||
| 91 | |||
| 92 | # add crc and NL |
||
| 93 | crc1,crc2 = calcCrcBytes(msg) |
||
| 94 | msg += crc1 + crc2 |
||
| 95 | msg += '\r' |
||
| 96 | return msg |
||
| 97 | |||
| 98 | |||
| 99 | def parse(self, msg): |
||
| 100 | if len(msg)<6: |
||
| 101 | raise InvalidMsg() |
||
| 102 | if (msg[0] != '#'): |
||
| 103 | raise InvalidMsg() |
||
| 104 | if (msg[-1] != '\r'): |
||
| 105 | raise InvalidMsg() |
||
| 106 | |||
| 107 | self.address = ord(msg[1]) |
||
| 108 | self.cmd = msg[2] |
||
| 109 | |||
| 110 | data64 = map(ord, msg[3:-3]) # last 3 bytes are CRC and \n |
||
| 111 | |||
| 112 | done = False |
||
| 113 | i = 0 |
||
| 114 | self.data = [] |
||
| 115 | while (i<len(data64)) and not done: |
||
| 116 | a = 0 |
||
| 117 | b = 0 |
||
| 118 | c = 0 |
||
| 119 | d = 0 |
||
| 120 | try: |
||
| 121 | a = data64[i] - ord('=') |
||
| 122 | b = data64[i+1] - ord('=') |
||
| 123 | c = data64[i+2] - ord('=') |
||
| 124 | d = data64[i+3] - ord('=') |
||
| 125 | i = i + 4 |
||
| 126 | except IndexError: |
||
| 127 | done = True |
||
| 128 | |||
| 129 | self.data.append((a << 2)&0xFF | (b >> 4)) |
||
| 130 | self.data.append(((b & 0x0f) << 4)&0xFF | (c >> 2)); |
||
| 131 | self.data.append(((c & 0x03) << 6)&0xFF | d); |
||
| 132 | |||
| 133 | crc1,crc2 = calcCrcBytes(msg[:-3]) |
||
| 134 | if (crc1 != msg[-3] or crc2 != msg[-2]): |
||
| 135 | #print msg |
||
| 136 | raise CrcError |
||
| 137 | |||
| 138 | #print "crc= %x %x %x %x" % ( crc1, crc2, (ord(msg[-3])-ord('=')), (ord(msg[-2])-ord('='))) |
||
| 139 | |||
| 140 | |||
| 141 | def data2SignedInt(self, index): |
||
| 142 | int = self.data[index]+self.data[index+1]*256 |
||
| 143 | if (int > 0xFFFF/2): |
||
| 144 | int -= 0xFFFF |
||
| 145 | return int |
||
| 146 | |||
| 147 | |||
| 148 | class DebugDataMsg: |
||
| 149 | IDX_ANALOG_ACCNICK = 2+2*2 |
||
| 150 | IDX_ANALOG_ACCROLL = 2+2*3 |
||
| 151 | IDX_ANALOG_COMPASS = 2+2*3 |
||
| 152 | IDX_ANALOG_VOLTAGE = 2+2*9 |
||
| 153 | |||
| 154 | def __init__(self, msg): |
||
| 155 | if (msg.cmd != 'D'): |
||
| 156 | raise InvalidMsgType |
||
| 157 | self.msg = msg |
||
| 158 | |||
| 159 | def getAccNick(self): |
||
| 160 | return self.msg.data2SignedInt(DebugDataMsg.IDX_ANALOG_ACCNICK); |
||
| 161 | |||
| 162 | def getAccRoll(self): |
||
| 163 | return self.msg.data2SignedInt(DebugDataMsg.IDX_ANALOG_ACCROLL); |
||
| 164 | |||
| 165 | def getCompassHeading(self): |
||
| 166 | return self.msg.data2SignedInt(DebugDataMsg.IDX_ANALOG_COMPASS); |
||
| 167 | |||
| 168 | def getVoltage(self): |
||
| 169 | return float(self.msg.data2SignedInt(DebugDataMsg.IDX_ANALOG_VOLTAGE))/10; |
||
| 170 | |||
| 171 | class VibrationDataMsg: |
||
| 172 | def __init__(self, msg): |
||
| 173 | if (msg.cmd != 'F'): |
||
| 174 | raise InvalidMsgType |
||
| 175 | self.msg = msg |
||
| 176 | |||
| 177 | def getData(self): |
||
| 178 | data = [] |
||
| 179 | for i in range(0,25): |
||
| 180 | data.append(self.msg.data2SignedInt(2*i)) |
||
| 181 | return data |
||
| 182 | |||
| 183 | class VersionMsg: |
||
| 184 | def __init__(self, msg): |
||
| 185 | if (msg.cmd != 'V'): |
||
| 186 | raise InvalidMsgType |
||
| 187 | self.msg = msg |
||
| 188 | |||
| 189 | def getVersion(self): |
||
| 190 | return (self.msg.data[0], self.msg.data[1]) |
||
| 191 | |||
| 192 | class MkComm: |
||
| 193 | FCADDRESS = 1 |
||
| 194 | |||
| 195 | def __init__(self, printDebugMsg=False): |
||
| 196 | #self.logfile = open('mklog.txt', "rbU") |
||
| 197 | |||
| 198 | self.serPort = None |
||
| 199 | self.printDebugMsg = printDebugMsg |
||
| 200 | |||
| 201 | msg = MkMsg(address=0, cmd='v', data=[]) |
||
| 202 | self.getVersionMsgLn = msg.generateMsg() |
||
| 203 | msg = MkMsg(address=0, cmd='d', data=[500]) |
||
| 204 | self.getDebugMsgLn = msg.generateMsg() |
||
| 205 | |||
| 206 | |||
| 207 | |||
| 208 | def open(self, comPort): |
||
| 209 | self.serPort = serial.Serial(comPort, 57600, timeout=0.5) |
||
| 210 | if not self.serPort.isOpen(): |
||
| 211 | raise IOError("Failed to open serial port") |
||
| 212 | |||
| 213 | def isOpen(self): |
||
| 214 | return self.serPort != None |
||
| 215 | |||
| 216 | def sendLn(self, ln): |
||
| 217 | self.serPort.write(ln) |
||
| 218 | |||
| 219 | def waitForLn(self): |
||
| 220 | return self.serPort.readline(eol='\r') |
||
| 221 | |||
| 222 | def waitForMsg(self, cmd2wait4): |
||
| 223 | msg = None |
||
| 224 | done = False |
||
| 225 | while (not done): |
||
| 226 | line = self.waitForLn() |
||
| 227 | if len(line) == 0: |
||
| 228 | raise NoResponse(cmd2wait4) |
||
| 229 | try: |
||
| 230 | msg = MkMsg(msg=line) |
||
| 231 | if (msg.cmd == cmd2wait4): |
||
| 232 | done = True |
||
| 233 | except InvalidMsg: |
||
| 234 | if self.printDebugMsg: |
||
| 235 | print "DebugMsg: \"%s\"" % line[:-1] |
||
| 236 | pass |
||
| 237 | return msg |
||
| 238 | |||
| 239 | def getDebugMsg(self): |
||
| 240 | self.serPort.flushInput() |
||
| 241 | self.sendLn(self.getDebugMsgLn) |
||
| 242 | msg = self.waitForMsg('D') |
||
| 243 | msg = DebugDataMsg(msg) |
||
| 244 | return msg |
||
| 245 | |||
| 246 | def getVersionMsg(self): |
||
| 247 | self.sendLn(self.getVersionMsgLn) |
||
| 248 | msg = self.waitForMsg('V') |
||
| 249 | msg = VersionMsg(msg) |
||
| 250 | return msg |
||
| 251 | |||
| 252 | def setMotorTest(self, motorSpeeds): |
||
| 253 | msg = MkMsg(address=MkComm.FCADDRESS, cmd='t', data=motorSpeeds) |
||
| 254 | self.sendLn(msg.generateMsg()) |
||
| 255 | |||
| 256 | def doVibrationTest(self, motorSpeeds, nbSamples, channel): |
||
| 257 | data = [] |
||
| 258 | for i in range(0,min(nbSamples,1000)/50): |
||
| 259 | self.setMotorTest(motorSpeeds) |
||
| 260 | time.sleep(0.05) |
||
| 261 | #print i |
||
| 262 | msg = MkMsg(address=MkComm.FCADDRESS, cmd='f', data=[channel, i]) |
||
| 263 | self.sendLn(msg.generateMsg()) |
||
| 264 | msg = self.waitForMsg('F') |
||
| 265 | msg = VibrationDataMsg(msg) |
||
| 266 | data += msg.getData() |
||
| 267 | |||
| 268 | #print data |
||
| 269 | return data |
||
| 270 | |||
| 271 | |||
| 272 | |||
| 273 | |||
| 274 | |||
| 275 | |||
| 276 | if __name__ == '__main__': |
||
| 277 | try: |
||
| 278 | |||
| 279 | ## file = open('mklog.txt', "rbU") |
||
| 280 | ## |
||
| 281 | ## for line in file: |
||
| 282 | ## try: |
||
| 283 | ## msg = MkMsg(msg=line) |
||
| 284 | ## #print "Rec Addr:%02d cmd:%c data:" % (msg.address, msg.cmd), msg.data |
||
| 285 | ## if (msg.cmd == 'D'): |
||
| 286 | ## #print len(msg.data) |
||
| 287 | ## msg = DebugDataMsg(msg) |
||
| 288 | ## |
||
| 289 | ## #print "%d %d" % (msg.getAccNick(), msg.getAccRoll()) |
||
| 290 | ## |
||
| 291 | ## #val = abs(msg.getAccNick()) |
||
| 292 | ## #print "*************************************************************"[:val/10] |
||
| 293 | ## |
||
| 294 | ## print msg.getCompassHeading() |
||
| 295 | ## |
||
| 296 | ## except InvalidMsg: |
||
| 297 | ## pass |
||
| 298 | ## |
||
| 299 | ## file.close() |
||
| 300 | ## print "OK" |
||
| 301 | |||
| 302 | ## msg = MkMsg(address=1, cmd='a', data=[255, 254, 10, 11, 12]) |
||
| 303 | ## print msg.generateMsg() |
||
| 304 | ## msg2 = MkMsg(msg.generateMsg()) |
||
| 305 | ## print msg2.address, msg2.cmd, msg2.data |
||
| 306 | |||
| 307 | comm = MkComm() |
||
| 308 | comm.open(comPort="COM5") |
||
| 309 | |||
| 310 | msg = comm.getVersionMsg() |
||
| 311 | print "Version: %d.%d" % msg.getVersion() |
||
| 312 | |||
| 313 | comm.doVibrationTest() |
||
| 314 | |||
| 315 | # msg = comm.getVersionMsg() |
||
| 316 | # print "Version: %d.%d" % msg.getVersion() |
||
| 317 | # |
||
| 318 | # comm.setMotorTest([0,0,80,0]) |
||
| 319 | # time.sleep(10) |
||
| 320 | # |
||
| 321 | # minN = 0 |
||
| 322 | # maxN = 0 |
||
| 323 | # minR = 0 |
||
| 324 | # maxR = 0 |
||
| 325 | # |
||
| 326 | # time.clock() |
||
| 327 | # for i in range(0,1000): |
||
| 328 | # try: |
||
| 329 | # msg = comm.getDebugMsg() |
||
| 330 | # n = msg.getAccNick() |
||
| 331 | # r = msg.getAccRoll() |
||
| 332 | # print "%d,%d" % (n, r) |
||
| 333 | # minN = min(minN, n) |
||
| 334 | # maxN = max(maxN, n) |
||
| 335 | # minR = min(minR, r) |
||
| 336 | # maxR = max(maxR, r) |
||
| 337 | # except Exception: |
||
| 338 | # pass |
||
| 339 | # |
||
| 340 | # comm.setMotorTest([0,0,0,0]) |
||
| 341 | # print time.clock() |
||
| 342 | # print maxN-minN,maxR-minR |
||
| 343 | |||
| 344 | |||
| 345 | |||
| 346 | |||
| 347 | except Exception,e: |
||
| 348 | print |
||
| 349 | print "An error occured: ", e |
||
| 350 | print |
||
| 351 | traceback.print_exc() |
||
| 352 | raw_input("Press ENTER, the application will close") |