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-01-18 $'.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 else: 211 print "ERROR: unknown format:",format 212 assert False 213 214 return # Nothing to return
215 216 217 218 ###################################################################### 219 # UNIT TESTING 220 ###################################################################### 221 import unittest
222 -def testParams():
223 '''Return a params file base on the testvalue tags. 224 @rtype: dict 225 @return: params based on testvalue tags 226 ''' 227 params = {} 228 params['dac'] = 366 229 params['fid'] = 1 230 params['reserved'] = 0 231 params['MessageID'] = 3 232 params['BinaryData'] = BitVector(bitstring='0') 233 234 return params
235
236 -class Testsls_header(unittest.TestCase):
237 '''Use testvalue tag text from each type to build test case the sls_header message'''
238 - def testEncodeDecode(self):
239 240 params = testParams() 241 bits = encode(params) 242 r = decode(bits) 243 244 # Check that each parameter came through ok. 245 self.failUnlessEqual(r['dac'],params['dac']) 246 self.failUnlessEqual(r['fid'],params['fid']) 247 self.failUnlessEqual(r['reserved'],params['reserved']) 248 self.failUnlessEqual(r['MessageID'],params['MessageID']) 249 self.failUnlessEqual(r['BinaryData'],params['BinaryData'])
250
251 -def addMsgOptions(parser):
252 parser.add_option('-d','--decode',dest='doDecode',default=False,action='store_true', 253 help='decode a "sls_header" AIS message') 254 parser.add_option('-e','--encode',dest='doEncode',default=False,action='store_true', 255 help='encode a "sls_header" AIS message') 256 parser.add_option('--dac-field', dest='dacField',metavar='uint',type='int' 257 ,help='Field parameter value [default: %default]') 258 parser.add_option('--MessageID-field', dest='MessageIDField',metavar='uint',type='int' 259 ,help='Field parameter value [default: %default]') 260 parser.add_option('--BinaryData-field', dest='BinaryDataField',metavar='binary',type='string' 261 ,help='Field parameter value [default: %default]')
262 263 ############################################################ 264 if __name__=='__main__': 265 266 from optparse import OptionParser 267 parser = OptionParser(usage="%prog [options]", 268 version="%prog "+__version__) 269 270 parser.add_option('--doc-test',dest='doctest',default=False,action='store_true', 271 help='run the documentation tests') 272 parser.add_option('--unit-test',dest='unittest',default=False,action='store_true', 273 help='run the unit tests') 274 parser.add_option('-v','--verbose',dest='verbose',default=False,action='store_true', 275 help='Make the test output verbose') 276 277 # FIX: remove nmea from binary messages. No way to build the whole packet? 278 # FIX: or build the surrounding msg 8 for a broadcast? 279 typeChoices = ('binary','nmeapayload','nmea') # FIX: what about a USCG type message? 280 parser.add_option('-t','--type',choices=typeChoices,type='choice',dest='ioType' 281 ,default='nmeapayload' 282 ,help='What kind of string to expect ('+', '.join(typeChoices)+') [default: %default]') 283 284 285 outputChoices = ('std','html','csv' ) 286 parser.add_option('-T','--output-type',choices=outputChoices,type='choice',dest='outputType' 287 ,default='std' 288 ,help='What kind of string to output ('+', '.join(outputChoices)+') [default: %default]') 289 290 parser.add_option('-o','--output',dest='outputFileName',default=None, 291 help='Name of the python file to write [default: stdout]') 292 293 parser.add_option('-f','--fields',dest='fieldList',default=None, action='append', 294 choices=fieldList, 295 help='Which fields to include in the output. Currently only for csv output [default: all]') 296 297 parser.add_option('-p','--print-csv-field-list',dest='printCsvfieldList',default=False,action='store_true', 298 help='Print the field name for csv') 299 300 addMsgOptions(parser) 301 302 (options,args) = parser.parse_args() 303 success=True 304 305 if options.doctest: 306 import os; print os.path.basename(sys.argv[0]), 'doctests ...', 307 sys.argv= [sys.argv[0]] 308 if options.verbose: sys.argv.append('-v') 309 import doctest 310 numfail,numtests=doctest.testmod() 311 if numfail==0: print 'ok' 312 else: 313 print 'FAILED' 314 success=False 315 316 if not success: sys.exit('Something Failed') 317 del success # Hide success from epydoc 318 319 if options.unittest: 320 sys.argv = [sys.argv[0]] 321 if options.verbose: sys.argv.append('-v') 322 unittest.main() 323 324 outfile = sys.stdout 325 if None!=options.outputFileName: 326 outfile = file(options.outputFileName,'w') 327 328 329 if options.doEncode: 330 # First make sure all non required options are specified 331 if None==options.dacField: parser.error("missing value for dacField") 332 if None==options.MessageIDField: parser.error("missing value for MessageIDField") 333 if None==options.BinaryDataField: parser.error("missing value for BinaryDataField") 334 msgDict={ 335 'dac': options.dacField, 336 'fid': '1', 337 'reserved': '0', 338 'MessageID': options.MessageIDField, 339 'BinaryData': options.BinaryDataField, 340 } 341 342 bits = encode(msgDict) 343 if 'binary'==options.ioType: print str(bits) 344 elif 'nmeapayload'==options.ioType: 345 # FIX: figure out if this might be necessary at compile time 346 print "bitLen",len(bits) 347 bitLen=len(bits) 348 if bitLen%6!=0: 349 bits = bits + BitVector(size=(6 - (bitLen%6))) # Pad out to multiple of 6 350 print "result:",binary.bitvectoais6(bits)[0] 351 352 353 # FIX: Do not emit this option for the binary message payloads. Does not make sense. 354 elif 'nmea'==options.ioType: sys.exit("FIX: need to implement this capability") 355 else: sys.exit('ERROR: unknown ioType. Help!') 356 357 if options.printCsvfieldList: 358 # Make a csv separated list of fields that will be displayed for csv 359 if None == options.fieldList: options.fieldList = fieldList 360 import StringIO 361 buf = StringIO.StringIO() 362 for field in options.fieldList: 363 buf.write(field+',') 364 result = buf.getvalue() 365 if result[-1] == ',': print result[:-1] 366 else: print result 367 368 if options.doDecode: 369 for msg in args: 370 bv = None 371 if 'binary' == options.ioType: bv = BitVector(bitstring=msg) 372 elif 'nmeapayload'== options.ioType: bv = binary.ais6tobitvec(msg) 373 elif 'nmea' == options.ioType: bv = binary.ais6tobitvec(msg.split(',')[5]) 374 else: sys.exit('ERROR: unknown ioType. Help!') 375 376 printFields(decode(bv),out=outfile,format=options.outputType,fieldList=options.fieldList) 377