[Mesa-dev] [PATCH v2 27/31] nvir/nir: implement geometry shader nir_intrinsics

[Karol Herbst]

Signed-off-by: Karol Herbst <kherbst at redhat.com>
---
 .../drivers/nouveau/codegen/nv50_ir_from_nir.cpp   | 31 ++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/src/gallium/drivers/nouveau/codegen/nv50_ir_from_nir.cpp b/src/gallium/drivers/nouveau/codegen/nv50_ir_from_nir.cpp
index 1ac5436153..252092a391 100644
--- a/src/gallium/drivers/nouveau/codegen/nv50_ir_from_nir.cpp
+++ b/src/gallium/drivers/nouveau/codegen/nv50_ir_from_nir.cpp
@@ -396,6 +396,10 @@ operation
 Converter::getOperation(nir_intrinsic_op op)
 {
    switch (op) {
+   case nir_intrinsic_emit_vertex:
+      return OP_EMIT;
+   case nir_intrinsic_end_primitive:
+      return OP_RESTART;
    default:
       ERROR("couldn't get operation for nir_intrinsic_op %u\n", op);
       assert(false);
@@ -1608,6 +1612,33 @@ Converter::visit(nir_intrinsic_instr *insn)
       mkOp2(OP_MERGE, TYPE_U64, newDefs[0], ballot->getDef(0), loadImm(getSSA(), 0));
       break;
    }
+   case nir_intrinsic_load_per_vertex_input: {
+      const DataType dType = getDType(insn);
+      LValues &newDefs = convert(&insn->dest);
+      Value *indirectVertex;
+      Value *indirectOffset;
+      auto baseVertex = getIndirect(&insn->src[0], 0, &indirectVertex);
+      auto baseOffset = getIndirect(&insn->src[1], 0, &indirectOffset);
+
+      auto idx = nir_intrinsic_base(insn) + baseOffset;
+      Value *vtxBase = mkOp2v(OP_PFETCH, TYPE_U32, getSSA(4, FILE_ADDRESS), mkImm(baseVertex), indirectVertex);
+      if (indirectOffset)
+         indirectOffset = mkOp2v(OP_SHL, TYPE_U32, getSSA(4, FILE_ADDRESS), indirectOffset, mkImm(4));
+      for (auto i = 0u; i < insn->num_components; ++i) {
+         uint32_t address = info->in[idx].slot[nir_intrinsic_component(insn) + i] * 4;
+         Symbol *sym = mkSymbol(FILE_SHADER_INPUT, 0, dType, address);
+         Instruction *ld = mkLoad(dType, newDefs[i], sym, indirectOffset);
+         ld->setIndirect(0, 1, vtxBase);
+         ld->perPatch = info->in[idx].patch;
+      }
+      break;
+   }
+   case nir_intrinsic_emit_vertex:
+   case nir_intrinsic_end_primitive: {
+      auto idx = nir_intrinsic_stream_id(insn);
+      mkOp1(getOperation(insn->intrinsic), TYPE_U32, NULL, mkImm(idx))->fixed = 1;
+      break;
+   }
    default:
       ERROR("unknown nir_intrinsic_op %s\n", nir_intrinsic_infos[insn->intrinsic].name);
       return false;
-- 
2.14.3