xref: /aosp_15_r20/external/mesa3d/src/compiler/nir/nir_intrinsics_c.py (revision 6104692788411f58d303aa86923a9ff6ecaded22)
1from functools import reduce
2import operator
3
4template = """\
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
29const 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   .bit_size_src = ${opcode.bit_size_src},
43   .num_indices = ${opcode.num_indices},
44% if opcode.indices:
45   .indices = {
46% for i in range(len(opcode.indices)):
47      NIR_INTRINSIC_${opcode.indices[i].name.upper()},
48% endfor
49   },
50   .index_map = {
51% for i in range(len(opcode.indices)):
52      [NIR_INTRINSIC_${opcode.indices[i].name.upper()}] = ${i + 1},
53% endfor
54    },
55% endif
56   .flags = ${"0" if len(opcode.flags) == 0 else " | ".join(opcode.flags)},
57},
58% endfor
59};
60
61const char *nir_intrinsic_index_names[NIR_INTRINSIC_NUM_INDEX_FLAGS] = {
62% for index in INTR_INDICES:
63   "${index.name}",
64% endfor
65};
66"""
67
68from nir_intrinsics import INTR_OPCODES, INTR_INDICES
69from mako.template import Template
70import argparse
71import os
72
73def main():
74    parser = argparse.ArgumentParser()
75    parser.add_argument('--outdir', required=True,
76                        help='Directory to put the generated files in')
77
78    args = parser.parse_args()
79
80    path = os.path.join(args.outdir, 'nir_intrinsics.c')
81    with open(path, 'w', encoding='utf-8') as f:
82        f.write(Template(template).render(
83            INTR_OPCODES=INTR_OPCODES, INTR_INDICES=INTR_INDICES,
84            reduce=reduce, operator=operator))
85
86if __name__ == '__main__':
87    main()
88
89