src/intel/genxml/gen_zipped_file.py

   1 #encoding=utf-8
   2 #
   3 # Copyright © 2017 Intel Corporation
   4 #
   5 # Permission is hereby granted, free of charge, to any person obtaining a
   6 # copy of this software and associated documentation files (the "Software"),
   7 # to deal in the Software without restriction, including without limitation
   8 # the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9 # and/or sell copies of the Software, and to permit persons to whom the
  10 # Software is furnished to do so, subject to the following conditions:
  11 #
  12 # The above copyright notice and this permission notice (including the next
  13 # paragraph) shall be included in all copies or substantial portions of the
  14 # Software.
  15 #
  16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  19 # THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  21 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  22 # IN THE SOFTWARE.
  23 #
  24
  25 from __future__ import print_function
  26 import sys
  27 import zlib
  28 import xml.etree.ElementTree as et
  29
  30 def main():
  31     if len(sys.argv) < 2:
  32         print("No input xml file specified")
  33         sys.exit(1)
  34
  35     compress = zlib.compressobj()
  36
  37     print("static const struct {")
  38     print("   uint32_t gen_10;")
  39     print("   uint32_t offset;")
  40     print("   uint32_t length;")
  41     print("} genxml_files_table[] = {")
  42
  43     xml_offset = 0
  44     compressed_data = b''
  45     for i in range(1, len(sys.argv)):
  46         filename = sys.argv[i]
  47         xml = open(filename, "rb").read()
  48         xml_length = len(xml)
  49         root = et.fromstring(xml)
  50
  51         print("   { %i, %i, %i }," %
  52               (int(float(root.attrib['gen']) * 10), xml_offset, xml_length))
  53
  54         compressed_data += compress.compress(xml)
  55         xml_offset += xml_length
  56
  57     print("};")
  58
  59     compressed_data += compress.flush()
  60
  61     print("")
  62     print("static const uint8_t compress_genxmls[] = {")
  63     print("   ", end='')
  64     for i, c in enumerate(bytearray(compressed_data), start=1):
  65         print("0x%.2x, " % c, end='\n   ' if not i % 12 else '')
  66     print('\n};')
  67
  68
  69 if __name__ == '__main__':
  70     main()