tools/mkarray.py

   1 #!/usr/bin/env python
   2 #
   3 # Copyright (c) 2011 Martin Decky
   4 # All rights reserved.
   5 #
   6 # Redistribution and use in source and binary forms, with or without
   7 # modification, are permitted provided that the following conditions
   8 # are met:
   9 #
  10 # - Redistributions of source code must retain the above copyright
  11 #   notice, this list of conditions and the following disclaimer.
  12 # - Redistributions in binary form must reproduce the above copyright
  13 #   notice, this list of conditions and the following disclaimer in the
  14 #   documentation and/or other materials provided with the distribution.
  15 # - The name of the author may not be used to endorse or promote products
  16 #   derived from this software without specific prior written permission.
  17 #
  18 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  19 # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  20 # OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  21 # IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  22 # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  23 # NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27 # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28 #
  29 """
  30 C structure creator
  31 """
  32
  33 import sys
  34 import os
  35 import struct
  36
  37 def usage(prname):
  38         "Print usage syntax"
  39         print("%s <DESTINATION> <LABEL> [SOURCE ...]" % prname)
  40
  41 def main():
  42         if (len(sys.argv) < 3):
  43                 usage(sys.argv[0])
  44                 return
  45
  46         dest = sys.argv[1]
  47         label = sys.argv[2]
  48
  49         header_ctx = []
  50         data_ctx = []
  51         for src in sys.argv[3:]:
  52                 basename = os.path.basename(src)
  53                 symbol = basename.replace(".", "_")
  54
  55                 print("%s -> %s" % (src, symbol))
  56
  57                 src_in = open(src, "rb")
  58                 src_data = src_in.read()
  59                 src_in.close()
  60
  61                 header_rec = "extern uint8_t %s[];" % symbol
  62                 header_ctx.append(header_rec)
  63
  64                 data_rec = "uint8_t %s[] = {\n\t" % symbol
  65
  66                 fmt = 'B'
  67                 item_size = struct.calcsize(fmt)
  68                 offset = 0
  69                 cnt = 0
  70
  71                 while (len(src_data[offset:]) >= item_size):
  72                         byte = struct.unpack_from(fmt, src_data, offset)
  73
  74                         if (offset > 0):
  75                                 if ((cnt % 15) == 0):
  76                                         data_rec += ",\n\t"
  77                                 else:
  78                                         data_rec += ", "
  79
  80                         data_rec += "0x%x" % byte
  81                         offset += item_size
  82                         cnt += 1
  83
  84                 data_rec += "\n};\n"
  85                 data_ctx.append(data_rec)
  86
  87                 header_rec = "extern size_t %s_size;" % symbol
  88                 header_ctx.append(header_rec)
  89
  90                 data_rec = "size_t %s_size = %u;\n" % (symbol, offset)
  91                 data_ctx.append(data_rec)
  92
  93         header = open("%s.h" % dest, "w")
  94
  95         header.write('/***************************************\n')
  96         header.write(' * AUTO-GENERATED FILE, DO NOT EDIT!!! *\n')
  97         header.write(' * Generated by: tools/mkarray.py      *\n')
  98         header.write(' ***************************************/\n\n')
  99         header.write("#ifndef %s_H_\n" % label)
 100         header.write("#define %s_H_\n\n" % label)
 101         header.write("#include <sys/types.h>\n\n")
 102         header.write("\n".join(header_ctx))
 103         header.write("\n\n")
 104         header.write("#endif\n")
 105
 106         header.close()
 107
 108         data = open("%s.c" % dest, "w")
 109
 110         data.write('/***************************************\n')
 111         data.write(' * AUTO-GENERATED FILE, DO NOT EDIT!!! *\n')
 112         data.write(' * Generated by: tools/mkarray.py      *\n')
 113         data.write(' ***************************************/\n\n')
 114         data.write("#include \"%s.h\"\n\n" % dest)
 115         data.write("\n".join(data_ctx))
 116
 117         data.close()
 118
 119 if __name__ == '__main__':
 120         main()