Package ais :: Module sls_header
[hide private]
[frames] | no frames]

Source Code for Module ais.sls_header

  1  #!/usr/bin/env python 
  2   
  3  __version__ = '$Revision: 4791 $'.split()[1] 
  4  __date__ = '$Date: 2007-02-05 $'.split()[1] 
  5  __author__ = 'xmlbinmsg' 
  6   
  7  __doc__=''' 
  8   
  9  Autogenerated python functions to serialize/deserialize binary messages. 
 10   
 11  Generated by: ./aisxmlbinmsg2py.py 
 12   
 13  Need to then wrap these functions with the outer AIS packet and then 
 14  convert the whole binary blob to a NMEA string.  Those functions are 
 15  not currently provided in this file. 
 16   
 17  serialize: python to ais binary 
 18  deserialize: ais binary to python 
 19   
 20  The generated code uses translators.py, binary.py, and aisstring.py 
 21  which should be packaged with the resulting files. 
 22   
 23   
 24  @requires: U{epydoc<http://epydoc.sourceforge.net/>} > 3.0alpha3 
 25  @requires: U{BitVector<http://cheeseshop.python.org/pypi/BitVector>} 
 26   
 27  @author: '''+__author__+''' 
 28  @version: ''' + __version__ +''' 
 29  @var __date__: Date of last svn commit 
 30  @undocumented: __version__ __author__ __doc__ parser 
 31  @status: under development 
 32  @license: Generated code has no license 
 33  ''' 
 34   
 35  import sys 
 36  from decimal import Decimal 
 37  from BitVector import BitVector 
 38   
 39  import binary, aisstring 
 40   
 41  # FIX: check to see if these will be needed 
 42  TrueBV  = BitVector(bitstring="1") 
 43  "Why always rebuild the True bit?  This should speed things up a bunch" 
 44  FalseBV = BitVector(bitstring="0") 
 45  "Why always rebuild the False bit?  This should speed things up a bunch" 
 46   
 47   
 48  fieldList = [ 
 49          'dac', 
 50          'fid', 
 51          'reserved', 
 52          'MessageID', 
 53          'BinaryData', 
 54  ] 
 55   
56 -def encode(params, validate=False):
57 '''Create a sls_header binary message payload to pack into an AIS Msg sls_header. 58 59 Fields in params: 60 - dac(uint): Designated Area Code 366 for US 61 - fid(uint): Functional Id (field automatically set to "1") 62 - reserved(uint): say what? (field automatically set to "0") 63 - MessageID(uint): Binary message indentifier 64 - BinaryData(binary): FIX: make this consume the rest! 65 @param params: Dictionary of field names/values. Throws a ValueError exception if required is missing 66 @param validate: Set to true to cause checking to occur. Runs slower. FIX: not implemented. 67 @rtype: BitVector 68 @return: encoded binary message (for binary messages, this needs to be wrapped in a msg 8 69 @note: The returned bits may not be 6 bit aligned. It is up to you to pad out the bits. 70 ''' 71 72 bvList = [] 73 bvList.append(binary.setBitVectorSize(BitVector(intVal=params['dac']),10)) 74 bvList.append(binary.setBitVectorSize(BitVector(intVal=1),6)) 75 bvList.append(binary.setBitVectorSize(BitVector(intVal=0),2)) 76 bvList.append(binary.setBitVectorSize(BitVector(intVal=params['MessageID']),6)) 77 bvList.append(params['BinaryData']) 78 79 return binary.joinBV(bvList)
80
81 -def decode(bv, validate=False):
82 '''Unpack a sls_header message 83 84 Fields in params: 85 - dac(uint): Designated Area Code 366 for US 86 - fid(uint): Functional Id (field automatically set to "1") 87 - reserved(uint): say what? (field automatically set to "0") 88 - MessageID(uint): Binary message indentifier 89 - BinaryData(binary): FIX: make this consume the rest! 90 @type bv: BitVector 91 @param bv: Bits defining a message 92 @param validate: Set to true to cause checking to occur. Runs slower. FIX: not implemented. 93 @rtype: dict 94 @return: params 95 ''' 96 97 #Would be nice to check the bit count here.. 98 #if validate: 99 # assert (len(bv)==FIX: SOME NUMBER) 100 r = {} 101 r['dac']=int(bv[0:10]) 102 r['fid']=1 103 r['reserved']=0 104 r['MessageID']=int(bv[18:24]) 105 r['BinaryData']=bv[24:] 106 return r
107
108 -def decodedac(bv, validate=False):
109 return int(bv[0:10])
110
111 -def decodefid(bv, validate=False):
112 return 1
113
114 -def decodereserved(bv, validate=False):
115 return 0
116
117 -def decodeMessageID(bv, validate=False):
118 return int(bv[18:24])
119
120 -def decodeBinaryData(bv, validate=False):
121 return bv[24:]
122 123
124 -def printHtml(params, out=sys.stdout):
125 out.write("<h3>sls_header<h3>\n") 126 out.write("<table border=\"1\">\n") 127 out.write("<tr bgcolor=\"orange\">\n") 128 out.write("<th align=\"left\">Field Name</th>\n") 129 out.write("<th align=\"left\">Type</th>\n") 130 out.write("<th align=\"left\">Value</th>\n") 131 out.write("<th align=\"left\">Value in Lookup Table</th>\n") 132 out.write("<th align=\"left\">Units</th>\n") 133 out.write("\n") 134 out.write("<tr>\n") 135 out.write("<td>dac</td>\n") 136 out.write("<td>uint</td>\n") 137 if 'dac' in params: 138 out.write(" <td>"+str(params['dac'])+"</td>\n") 139 out.write(" <td>"+str(params['dac'])+"</td>\n") 140 out.write("</tr>\n") 141 out.write("\n") 142 out.write("<tr>\n") 143 out.write("<td>fid</td>\n") 144 out.write("<td>uint</td>\n") 145 if 'fid' in params: 146 out.write(" <td>"+str(params['fid'])+"</td>\n") 147 out.write(" <td>"+str(params['fid'])+"</td>\n") 148 out.write("</tr>\n") 149 out.write("\n") 150 out.write("<tr>\n") 151 out.write("<td>reserved</td>\n") 152 out.write("<td>uint</td>\n") 153 if 'reserved' in params: 154 out.write(" <td>"+str(params['reserved'])+"</td>\n") 155 out.write(" <td>"+str(params['reserved'])+"</td>\n") 156 out.write("</tr>\n") 157 out.write("\n") 158 out.write("<tr>\n") 159 out.write("<td>MessageID</td>\n") 160 out.write("<td>uint</td>\n") 161 if 'MessageID' in params: 162 out.write(" <td>"+str(params['MessageID'])+"</td>\n") 163 out.write(" <td>"+str(params['MessageID'])+"</td>\n") 164 out.write("</tr>\n") 165 out.write("\n") 166 out.write("<tr>\n") 167 out.write("<td>BinaryData</td>\n") 168 out.write("<td>binary</td>\n") 169 if 'BinaryData' in params: 170 out.write(" <td>"+str(params['BinaryData'])+"</td>\n") 171 out.write(" <td>"+str(params['BinaryData'])+"</td>\n") 172 out.write("</tr>\n") 173 out.write("</table>\n")
174
175 -def printFields(params, out=sys.stdout, format='std', fieldList=None):
176 '''Print a BinaryData message to stdout. 177 178 Fields in params: 179 - dac(uint): Designated Area Code 366 for US 180 - fid(uint): Functional Id (field automatically set to "1") 181 - reserved(uint): say what? (field automatically set to "0") 182 - MessageID(uint): Binary message indentifier 183 - BinaryData(binary): FIX: make this consume the rest! 184 @param params: Dictionary of field names/values. 185 @param out: File like object to write to 186 @rtype: stdout 187 @return: text to out 188 ''' 189 190 if 'std'==format: 191 out.write("BinaryData:\n") 192 if 'dac' in params: out.write(" dac: "+str(params['dac'])+"\n") 193 if 'fid' in params: out.write(" fid: "+str(params['fid'])+"\n") 194 if 'reserved' in params: out.write(" reserved: "+str(params['reserved'])+"\n") 195 if 'MessageID' in params: out.write(" MessageID: "+str(params['MessageID'])+"\n") 196 if 'BinaryData' in params: out.write(" BinaryData: "+str(params['BinaryData'])+"\n") 197 elif 'csv'==format: 198 if None == options.fieldList: 199 options.fieldList = fieldList 200 needComma = False; 201 for field in fieldList: 202 if needComma: out.write(',') 203 needComma = True 204 if field in params: 205 out.write(str(params[field])) 206 # else: leave it empty 207 out.write("\n") 208 elif 'html'==format: 209 printHtml(params,out) 210 elif 'sql'==format: 211 sqlInsertStr(params,out) 212 else: 213 print "ERROR: unknown format:",format 214 assert False 215 216 return # Nothing to return
217 218 ###################################################################### 219 # SQL SUPPORT 220 ###################################################################### 221
222 -def sqlCreateStr(outfile=sys.stdout, fields=None, extraFields=None):
223 ''' 224 Return the SQL CREATE command for this message type 225 @param outfile: file like object to print to. 226 @param fields: which fields to put in the create. Defaults to all. 227 @param extraFields: A sequence of tuples containing (name,sql type) for additional fields 228 @return: sql create string 229 @rtype: str 230 231 @see: sqlCreate 232 ''' 233 outfile.write(str(sqlCreate(fields,extraFields)))
234
235 -def sqlCreate(fields=None, extraFields=None):
236 ''' 237 Return the sqlhelp object to create the table. 238 239 @param fields: which fields to put in the create. Defaults to all. 240 @param extraFields: A sequence of tuples containing (name,sql type) for additional fields 241 @return: An object that can be used to generate a return 242 @rtype: sqlhelp.create 243 ''' 244 if None == fields: fields = fieldList 245 import sqlhelp 246 c = sqlhelp.create('sls_header') 247 if 'dac' in fields: c.addInt ('dac') 248 if 'fid' in fields: c.addInt ('fid') 249 if 'reserved' in fields: c.addInt ('reserved') 250 if 'MessageID' in fields: c.addInt ('MessageID') 251 if 'BinaryData' in fields: c.addBitVarying('BinaryData',1024) 252 253 return c
254
255 -def sqlInsertStr(params, outfile=sys.stdout, extraParams=None):
256 ''' 257 Return the SQL CREATE command for this message type 258 @param params: dictionary of values keyed by field name 259 @param outfile: file like object to print to. 260 @param extraParams: A sequence of tuples containing (name,sql type) for additional fields 261 @return: sql create string 262 @rtype: str 263 264 @see: sqlCreate 265 ''' 266 outfile.write(str(sqlInsert(params,extraParams)))
267 268
269 -def sqlInsert(params,extraParams=None):
270 ''' 271 Give the SQL insert statement 272 @param params: dict keyed by field name of values 273 @param extraParams: any extra fields that you have created beyond the normal ais message fields 274 @rtype: sqlhelp.insert 275 @return: insert class instance 276 @todo: allow optional type checking of params? 277 @warning: this will take invalid keys happily and do what??? 278 ''' 279 import sqlhelp 280 i = sqlhelp.insert('sls_header') 281 for key in params: i.add(key,params[key]) 282 if None != extraParams: 283 for key in extraParams: 284 i.add(key,extraParams[key]) 285 286 return i
287 288 289 ###################################################################### 290 # UNIT TESTING 291 ###################################################################### 292 import unittest
293 -def testParams():
294 '''Return a params file base on the testvalue tags. 295 @rtype: dict 296 @return: params based on testvalue tags 297 ''' 298 params = {} 299 params['dac'] = 366 300 params['fid'] = 1 301 params['reserved'] = 0 302 params['MessageID'] = 3 303 params['BinaryData'] = BitVector(bitstring='0') 304 305 return params
306
307 -class Testsls_header(unittest.TestCase):
308 '''Use testvalue tag text from each type to build test case the sls_header message'''
309 - def testEncodeDecode(self):
310 311 params = testParams() 312 bits = encode(params) 313 r = decode(bits) 314 315 # Check that each parameter came through ok. 316 self.failUnlessEqual(r['dac'],params['dac']) 317 self.failUnlessEqual(r['fid'],params['fid']) 318 self.failUnlessEqual(r['reserved'],params['reserved']) 319 self.failUnlessEqual(r['MessageID'],params['MessageID']) 320 self.failUnlessEqual(r['BinaryData'],params['BinaryData'])
321
322 -def addMsgOptions(parser):
323 parser.add_option('-d','--decode',dest='doDecode',default=False,action='store_true', 324 help='decode a "sls_header" AIS message') 325 parser.add_option('-e','--encode',dest='doEncode',default=False,action='store_true', 326 help='encode a "sls_header" AIS message') 327 parser.add_option('--dac-field', dest='dacField',metavar='uint',type='int' 328 ,help='Field parameter value [default: %default]') 329 parser.add_option('--MessageID-field', dest='MessageIDField',metavar='uint',type='int' 330 ,help='Field parameter value [default: %default]') 331 parser.add_option('--BinaryData-field', dest='BinaryDataField',metavar='binary',type='string' 332 ,help='Field parameter value [default: %default]')
333 334 ############################################################ 335 if __name__=='__main__': 336 337 from optparse import OptionParser 338 parser = OptionParser(usage="%prog [options]", 339 version="%prog "+__version__) 340 341 parser.add_option('--doc-test',dest='doctest',default=False,action='store_true', 342 help='run the documentation tests') 343 parser.add_option('--unit-test',dest='unittest',default=False,action='store_true', 344 help='run the unit tests') 345 parser.add_option('-v','--verbose',dest='verbose',default=False,action='store_true', 346 help='Make the test output verbose') 347 348 # FIX: remove nmea from binary messages. No way to build the whole packet? 349 # FIX: or build the surrounding msg 8 for a broadcast? 350 typeChoices = ('binary','nmeapayload','nmea') # FIX: what about a USCG type message? 351 parser.add_option('-t','--type',choices=typeChoices,type='choice',dest='ioType' 352 ,default='nmeapayload' 353 ,help='What kind of string to expect ('+', '.join(typeChoices)+') [default: %default]') 354 355 356 outputChoices = ('std','html','csv','sql' ) 357 parser.add_option('-T','--output-type',choices=outputChoices,type='choice',dest='outputType' 358 ,default='std' 359 ,help='What kind of string to output ('+', '.join(outputChoices)+') [default: %default]') 360 361 parser.add_option('-o','--output',dest='outputFileName',default=None, 362 help='Name of the python file to write [default: stdout]') 363 364 parser.add_option('-f','--fields',dest='fieldList',default=None, action='append', 365 choices=fieldList, 366 help='Which fields to include in the output. Currently only for csv output [default: all]') 367 368 parser.add_option('-p','--print-csv-field-list',dest='printCsvfieldList',default=False,action='store_true', 369 help='Print the field name for csv') 370 371 parser.add_option('-c','--sql-create',dest='sqlCreate',default=False,action='store_true', 372 help='Print out an sql create command for the table.') 373 374 addMsgOptions(parser) 375 376 (options,args) = parser.parse_args() 377 success=True 378 379 if options.doctest: 380 import os; print os.path.basename(sys.argv[0]), 'doctests ...', 381 sys.argv= [sys.argv[0]] 382 if options.verbose: sys.argv.append('-v') 383 import doctest 384 numfail,numtests=doctest.testmod() 385 if numfail==0: print 'ok' 386 else: 387 print 'FAILED' 388 success=False 389 390 if not success: sys.exit('Something Failed') 391 del success # Hide success from epydoc 392 393 if options.unittest: 394 sys.argv = [sys.argv[0]] 395 if options.verbose: sys.argv.append('-v') 396 unittest.main() 397 398 outfile = sys.stdout 399 if None!=options.outputFileName: 400 outfile = file(options.outputFileName,'w') 401 402 403 if options.doEncode: 404 # First make sure all non required options are specified 405 if None==options.dacField: parser.error("missing value for dacField") 406 if None==options.MessageIDField: parser.error("missing value for MessageIDField") 407 if None==options.BinaryDataField: parser.error("missing value for BinaryDataField") 408 msgDict={ 409 'dac': options.dacField, 410 'fid': '1', 411 'reserved': '0', 412 'MessageID': options.MessageIDField, 413 'BinaryData': options.BinaryDataField, 414 } 415 416 bits = encode(msgDict) 417 if 'binary'==options.ioType: print str(bits) 418 elif 'nmeapayload'==options.ioType: 419 # FIX: figure out if this might be necessary at compile time 420 print "bitLen",len(bits) 421 bitLen=len(bits) 422 if bitLen%6!=0: 423 bits = bits + BitVector(size=(6 - (bitLen%6))) # Pad out to multiple of 6 424 print "result:",binary.bitvectoais6(bits)[0] 425 426 427 # FIX: Do not emit this option for the binary message payloads. Does not make sense. 428 elif 'nmea'==options.ioType: sys.exit("FIX: need to implement this capability") 429 else: sys.exit('ERROR: unknown ioType. Help!') 430 431 432 if options.sqlCreate: 433 sqlCreateStr(outfile,options.fieldList) 434 435 if options.printCsvfieldList: 436 # Make a csv separated list of fields that will be displayed for csv 437 if None == options.fieldList: options.fieldList = fieldList 438 import StringIO 439 buf = StringIO.StringIO() 440 for field in options.fieldList: 441 buf.write(field+',') 442 result = buf.getvalue() 443 if result[-1] == ',': print result[:-1] 444 else: print result 445 446 if options.doDecode: 447 for msg in args: 448 bv = None 449 if 'binary' == options.ioType: bv = BitVector(bitstring=msg) 450 elif 'nmeapayload'== options.ioType: bv = binary.ais6tobitvec(msg) 451 elif 'nmea' == options.ioType: bv = binary.ais6tobitvec(msg.split(',')[5]) 452 else: sys.exit('ERROR: unknown ioType. Help!') 453 454 printFields(decode(bv),out=outfile,format=options.outputType,fieldList=options.fieldList) 455