dri: Use DRM_FORMAT_* instead of defining our own copy.
[mesa.git] / src / compiler / nir / nir_builder_opcodes_h.py
index e27206ea8fc143dbf3278d3016a346b63e4cb081..53fb23ca2b351cba113854c3cf14d4973d811629 100644 (file)
@@ -1,4 +1,4 @@
-#! /usr/bin/env python
+from __future__ import print_function
 
 template = """\
 /* Copyright (C) 2015 Broadcom
@@ -26,13 +26,75 @@ template = """\
 #ifndef _NIR_BUILDER_OPCODES_
 #define _NIR_BUILDER_OPCODES_
 
-% for name, opcode in sorted(opcodes.iteritems()):
-ALU${opcode.num_inputs}(${name});
+<%
+def src_decl_list(num_srcs):
+   return ', '.join('nir_ssa_def *src' + str(i) for i in range(num_srcs))
+
+def src_list(num_srcs):
+   return ', '.join('src' + str(i) if i < num_srcs else 'NULL' for i in range(4))
+%>
+
+% for name, opcode in sorted(opcodes.items()):
+static inline nir_ssa_def *
+nir_${name}(nir_builder *build, ${src_decl_list(opcode.num_inputs)})
+{
+   return nir_build_alu(build, nir_op_${name}, ${src_list(opcode.num_inputs)});
+}
+% endfor
+
+/* Generic builder for system values. */
+static inline nir_ssa_def *
+nir_load_system_value(nir_builder *build, nir_intrinsic_op op, int index,
+                      unsigned bit_size)
+{
+   nir_intrinsic_instr *load = nir_intrinsic_instr_create(build->shader, op);
+   load->num_components = nir_intrinsic_infos[op].dest_components;
+   load->const_index[0] = index;
+   nir_ssa_dest_init(&load->instr, &load->dest,
+                     nir_intrinsic_infos[op].dest_components, bit_size, NULL);
+   nir_builder_instr_insert(build, &load->instr);
+   return &load->dest.ssa;
+}
+
+<%
+def sysval_decl_list(opcode):
+   res = ''
+   if opcode.indices:
+      res += ', unsigned ' + opcode.indices[0].lower()
+   if len(opcode.bit_sizes) != 1:
+      res += ', unsigned bit_size'
+   return res
+
+def sysval_arg_list(opcode):
+   args = []
+   if opcode.indices:
+      args.append(opcode.indices[0].lower())
+   else:
+      args.append('0')
+
+   if len(opcode.bit_sizes) == 1:
+      bit_size = opcode.bit_sizes[0]
+      args.append(str(bit_size))
+   else:
+      args.append('bit_size')
+
+   return ', '.join(args)
+%>
+
+% for name, opcode in filter(lambda v: v[1].sysval, sorted(INTR_OPCODES.items())):
+<% assert len(opcode.bit_sizes) > 0 %>
+static inline nir_ssa_def *
+nir_${name}(nir_builder *build${sysval_decl_list(opcode)})
+{
+   return nir_load_system_value(build, nir_intrinsic_${name},
+                                ${sysval_arg_list(opcode)});
+}
 % endfor
 
 #endif /* _NIR_BUILDER_OPCODES_ */"""
 
 from nir_opcodes import opcodes
+from nir_intrinsics import INTR_OPCODES
 from mako.template import Template
 
-print Template(template).render(opcodes=opcodes)
+print(Template(template).render(opcodes=opcodes, INTR_OPCODES=INTR_OPCODES))