src/gallium/drivers/swr/rasterizer/codegen/gen_backends.py

   1 # Copyright (C) 2017 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 # Compatible with Python2.X and Python3.X
  24
  25 from __future__ import print_function
  26 import itertools
  27 import os
  28 import sys
  29 from gen_common import ArgumentParser, MakoTemplateWriter
  30
  31
  32 def main(args=sys.argv[1:]):
  33     thisDir = os.path.dirname(os.path.realpath(__file__))
  34     parser = ArgumentParser("Generate files and initialization functions for all permutuations of BackendPixelRate.")
  35     parser.add_argument('--dim', help="gBackendPixelRateTable array dimensions", nargs='+', type=int, required=True)
  36     parser.add_argument('--outdir', help="output directory", nargs='?', type=str, default=thisDir)
  37     parser.add_argument('--split', help="how many lines of initialization per file [0=no split]", nargs='?', type=int, default='512')
  38     parser.add_argument('--cpp', help="Generate cpp file(s)", action='store_true', default=False)
  39     parser.add_argument('--cmake', help="Generate cmake file", action='store_true', default=False)
  40
  41     args = parser.parse_args(args);
  42
  43     class backendStrs :
  44         def __init__(self) :
  45             self.outFileName = 'gen_BackendPixelRate%s.cpp'
  46             self.functionTableName = 'gBackendPixelRateTable'
  47             self.funcInstanceHeader = ' = BackendPixelRate<SwrBackendTraits<'
  48             self.template = 'gen_backend.cpp'
  49             self.cmakeFileName = 'gen_backends.cmake'
  50             self.cmakeSrcVar = 'GEN_BACKEND_SOURCES'
  51
  52     backend = backendStrs()
  53
  54     output_list = []
  55     for x in args.dim:
  56         output_list.append(list(range(x)))
  57
  58     # generate all permutations possible for template parameter inputs
  59     output_combinations = list(itertools.product(*output_list))
  60     output_list = []
  61
  62     # for each permutation
  63     for x in range(len(output_combinations)):
  64         # separate each template peram into its own list member
  65         new_list = [output_combinations[x][i] for i in range(len(output_combinations[x]))]
  66         tempStr = backend.functionTableName
  67         #print each list member as an index in the multidimensional array
  68         for i in new_list:
  69             tempStr += '[' + str(i) + ']'
  70         #map each entry in the permuation as its own string member, store as the template instantiation string
  71         tempStr += backend.funcInstanceHeader + ','.join(map(str, output_combinations[x])) + '>>;'
  72         #append the line of c++ code in the list of output lines
  73         output_list.append(tempStr)
  74
  75     # how many files should we split the global template initialization into?
  76     if (args.split == 0):
  77         numFiles = 1
  78     else:
  79         numFiles = (len(output_list) + args.split - 1) // args.split
  80     linesPerFile = (len(output_list) + numFiles - 1) // numFiles
  81     chunkedList = [output_list[x:x+linesPerFile] for x in range(0, len(output_list), linesPerFile)]
  82
  83     # generate .cpp files
  84     if args.cpp:
  85         baseCppName = os.path.join(args.outdir, backend.outFileName)
  86         templateCpp = os.path.join(thisDir, 'templates', backend.template)
  87
  88         for fileNum in range(numFiles):
  89             filename = baseCppName % str(fileNum)
  90             #print('Generating', filename)
  91             MakoTemplateWriter.to_file(
  92                 templateCpp,
  93                 baseCppName % str(fileNum),
  94                 cmdline=sys.argv,
  95                 fileNum=fileNum,
  96                 funcList=chunkedList[fileNum])
  97
  98     # generate gen_backend.cmake file
  99     if args.cmake:
 100         templateCmake = os.path.join(thisDir, 'templates', 'gen_backend.cmake')
 101         cmakeFile = os.path.join(args.outdir, backend.cmakeFileName)
 102         #print('Generating', cmakeFile)
 103         MakoTemplateWriter.to_file(
 104             templateCmake,
 105             cmakeFile,
 106             cmdline=sys.argv,
 107             srcVar=backend.cmakeSrcVar,
 108             numFiles=numFiles,
 109             baseCppName='${RASTY_GEN_SRC_DIR}/backends/' + os.path.basename(baseCppName))
 110
 111     #print("Generated %d template instantiations in %d files" % (len(output_list), numFiles))
 112
 113     return 0
 114
 115 if __name__ == '__main__':
 116     sys.exit(main())