src/compiler/nir/nir_intrinsics_c.py

   1 from functools import reduce
   2 import operator
   3
   4 template = """\
   5 /* Copyright (C) 2018 Red Hat
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a
   8  * copy of this software and associated documentation files (the "Software"),
   9  * to deal in the Software without restriction, including without limitation
  10  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  11  * and/or sell copies of the Software, and to permit persons to whom the
  12  * Software is furnished to do so, subject to the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the next
  15  * paragraph) shall be included in all copies or substantial portions of the
  16  * Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  21  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  23  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  24  * IN THE SOFTWARE.
  25  */
  26
  27 #include "nir.h"
  28
  29 const nir_intrinsic_info nir_intrinsic_infos[nir_num_intrinsics] = {
  30 % for name, opcode in sorted(INTR_OPCODES.items()):
  31 {
  32    .name = "${name}",
  33    .num_srcs = ${opcode.num_srcs},
  34 % if opcode.src_components:
  35    .src_components = {
  36       ${", ".join(str(comp) for comp in opcode.src_components)}
  37    },
  38 % endif
  39    .has_dest = ${"true" if opcode.has_dest else "false"},
  40    .dest_components = ${max(opcode.dest_components, 0)},
  41    .dest_bit_sizes = ${hex(reduce(operator.or_, opcode.bit_sizes, 0))},
  42    .num_indices = ${opcode.num_indices},
  43 % if opcode.indices:
  44    .index_map = {
  45 % for i in range(len(opcode.indices)):
  46       [${opcode.indices[i]}] = ${i + 1},
  47 % endfor
  48     },
  49 % endif
  50    .flags = ${"0" if len(opcode.flags) == 0 else " | ".join(opcode.flags)},
  51 },
  52 % endfor
  53 };
  54 """
  55
  56 from nir_intrinsics import INTR_OPCODES
  57 from mako.template import Template
  58 import argparse
  59 import os
  60
  61 def main():
  62     parser = argparse.ArgumentParser()
  63     parser.add_argument('--outdir', required=True,
  64                         help='Directory to put the generated files in')
  65
  66     args = parser.parse_args()
  67
  68     path = os.path.join(args.outdir, 'nir_intrinsics.c')
  69     with open(path, 'wb') as f:
  70         f.write(Template(template, output_encoding='utf-8').render(INTR_OPCODES=INTR_OPCODES, reduce=reduce, operator=operator))
  71
  72 if __name__ == '__main__':
  73     main()
  74