src/gallium/drivers/swr/rasterizer/scripts/gen_knobs.py

   1 # Copyright (C) 2014-2016 Intel Corporation.   All Rights Reserved.
   2 #
   3 # Permission is hereby granted, free of charge, to any person obtaining a
   4 # copy of this software and associated documentation files (the "Software"),
   5 # to deal in the Software without restriction, including without limitation
   6 # the rights to use, copy, modify, merge, publish, distribute, sublicense,
   7 # and/or sell copies of the Software, and to permit persons to whom the
   8 # Software is furnished to do so, subject to the following conditions:
   9 #
  10 # The above copyright notice and this permission notice (including the next
  11 # paragraph) shall be included in all copies or substantial portions of the
  12 # Software.
  13 #
  14 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  17 # THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  19 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  20 # IN THE SOFTWARE.
  21
  22 # Python source
  23 from __future__ import print_function
  24 import os
  25 import sys
  26 import argparse
  27 import knob_defs
  28 from mako.template import Template
  29 from mako.exceptions import RichTraceback
  30
  31 def write_template_to_string(template_filename, **kwargs):
  32     try:
  33         template = Template(filename=os.path.abspath(template_filename))
  34         # Split + Join fixes line-endings for whatever platform you are using
  35         return '\n'.join(template.render(**kwargs).splitlines())
  36     except:
  37         traceback = RichTraceback()
  38         for (filename, lineno, function, line) in traceback.traceback:
  39             print("File %s, line %s, in %s" % (filename, lineno, function))
  40             print(line, "\n")
  41         print("%s: %s" % (str(traceback.error.__class__.__name__), traceback.error))
  42
  43 def write_template_to_file(template_filename, output_filename, **kwargs):
  44     output_dirname = os.path.dirname(output_filename)
  45     if not os.path.exists(output_dirname):
  46         os.makedirs(output_dirname)
  47     with open(output_filename, "w") as outfile:
  48         print(write_template_to_string(template_filename, **kwargs), file=outfile)
  49
  50 def main(args=sys.argv[1:]):
  51
  52     # parse args
  53     parser = argparse.ArgumentParser()
  54     parser.add_argument("--input", "-i", help="Path to knobs.template", required=True)
  55     parser.add_argument("--output", "-o", help="Path to output file", required=True)
  56     parser.add_argument("--gen_h", "-gen_h", help="Generate gen_knobs.h", action="store_true", default=False)
  57     parser.add_argument("--gen_cpp", "-gen_cpp", help="Generate gen_knobs.cpp", action="store_true", required=False)
  58
  59     args = parser.parse_args()
  60
  61     if args.input:
  62         if args.gen_h:
  63             write_template_to_file(args.input,
  64                                    args.output,
  65                                    filename='gen_knobs',
  66                                    knobs=knob_defs.KNOBS,
  67                                    includes=['core/knobs_init.h', 'common/os.h', 'sstream', 'iomanip'],
  68                                    gen_header=True)
  69
  70         if args.gen_cpp:
  71             write_template_to_file(args.input,
  72                                    args.output,
  73                                    filename='gen_knobs',
  74                                    knobs=knob_defs.KNOBS,
  75                                    includes=['core/knobs_init.h', 'common/os.h', 'sstream', 'iomanip'],
  76                                    gen_header=False)
  77
  78     return 0
  79
  80 if __name__ == '__main__':
  81     sys.exit(main())
  82