1from __future__ import print_function
2
3template = """\
4/* Copyright (C) 2015 Broadcom
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the "Software"),
8 * to deal in the Software without restriction, including without limitation
9 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10 * and/or sell copies of the Software, and to permit persons to whom the
11 * Software is furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice (including the next
14 * paragraph) shall be included in all copies or substantial portions of the
15 * Software.
16 *
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
20 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
22 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
23 * IN THE SOFTWARE.
24 */
25
26#ifndef _NIR_BUILDER_OPCODES_
27#define _NIR_BUILDER_OPCODES_
28
29<%
30def src_decl_list(num_srcs):
31   return ', '.join('nir_ssa_def *src' + str(i) for i in range(num_srcs))
32
33def src_list(num_srcs):
34   return ', '.join('src' + str(i) if i < num_srcs else 'NULL' for i in range(4))
35%>
36
37% for name, opcode in sorted(opcodes.items()):
38static inline nir_ssa_def *
39nir_${name}(nir_builder *build, ${src_decl_list(opcode.num_inputs)})
40{
41   return nir_build_alu(build, nir_op_${name}, ${src_list(opcode.num_inputs)});
42}
43% endfor
44
45/* Generic builder for system values. */
46static inline nir_ssa_def *
47nir_load_system_value(nir_builder *build, nir_intrinsic_op op, int index,
48                      unsigned bit_size)
49{
50   nir_intrinsic_instr *load = nir_intrinsic_instr_create(build->shader, op);
51   load->num_components = nir_intrinsic_infos[op].dest_components;
52   load->const_index[0] = index;
53   nir_ssa_dest_init(&load->instr, &load->dest,
54                     nir_intrinsic_infos[op].dest_components, bit_size, NULL);
55   nir_builder_instr_insert(build, &load->instr);
56   return &load->dest.ssa;
57}
58
59<%
60def sysval_decl_list(opcode):
61   res = ''
62   if opcode.indices:
63      res += ', unsigned ' + opcode.indices[0].lower()
64   if len(opcode.bit_sizes) != 1:
65      res += ', unsigned bit_size'
66   return res
67
68def sysval_arg_list(opcode):
69   args = []
70   if opcode.indices:
71      args.append(opcode.indices[0].lower())
72   else:
73      args.append('0')
74
75   if len(opcode.bit_sizes) == 1:
76      bit_size = opcode.bit_sizes[0]
77      args.append(str(bit_size))
78   else:
79      args.append('bit_size')
80
81   return ', '.join(args)
82%>
83
84% for name, opcode in filter(lambda v: v[1].sysval, sorted(INTR_OPCODES.items())):
85<% assert len(opcode.bit_sizes) > 0 %>
86static inline nir_ssa_def *
87nir_${name}(nir_builder *build${sysval_decl_list(opcode)})
88{
89   return nir_load_system_value(build, nir_intrinsic_${name},
90                                ${sysval_arg_list(opcode)});
91}
92% endfor
93
94#endif /* _NIR_BUILDER_OPCODES_ */"""
95
96from nir_opcodes import opcodes
97from nir_intrinsics import INTR_OPCODES
98from mako.template import Template
99
100print(Template(template).render(opcodes=opcodes, INTR_OPCODES=INTR_OPCODES))
101