[mlir][dxsa] Add dcl_output_topology instruction

Example:
  dxsa.dcl_output_topology trianglestrip

Signed-off-by: Vladimir Shiryaev <tagolog@users.noreply.github.com>

Author: tagolog

mlir/include/mlir/Dialect/DXSA/IR/DXSAOps.td

@@ -62,6 +62,26 @@ def DXSA_GlobalFlagsAttr :
   let assemblyFormat = "`<` $value `>`";
 }
 
+def DXSA_OutputPrimitiveTopology_PointList     : I32EnumAttrCase<"pointlist",     0>;
+def DXSA_OutputPrimitiveTopology_LineStrip     : I32EnumAttrCase<"linestrip",     1>;
+def DXSA_OutputPrimitiveTopology_TriangleStrip : I32EnumAttrCase<"trianglestrip", 2>;
+
+def DXSA_OutputPrimitiveTopology : I32EnumAttr<
+    "OutputPrimitiveTopology", "shader output primitive topology", [
+      DXSA_OutputPrimitiveTopology_PointList,
+      DXSA_OutputPrimitiveTopology_LineStrip,
+      DXSA_OutputPrimitiveTopology_TriangleStrip
+    ]> {
+  let cppNamespace = "::mlir::dxsa";
+  let genSpecializedAttr = 0;
+}
+
+def DXSA_OutputPrimitiveTopologyAttr :
+    EnumAttr<DXSADialect, DXSA_OutputPrimitiveTopology,
+             "output_primitive_topology"> {
+  let assemblyFormat = "$value";
+}
+
 //===----------------------------------------------------------------------===//
 // DXSA op definitions
 //===----------------------------------------------------------------------===//
@@ -158,4 +178,20 @@ def DXSA_DclTemps : DXSA_Op<"dcl_temps"> {
   let assemblyFormat = [{ $count attr-dict }];
 }
 
+def DXSA_DclOutputTopology : DXSA_Op<"dcl_output_topology"> {
+  let summary = "declare what primitive topology the shader generates as output";
+  let description = [{
+    The `dxsa.dcl_output_topology` operation declares the primitive topology
+    that the shader generates as output.
+
+    Example:
+
+    ```mlir
+    dxsa.dcl_output_topology trianglestrip
+    ```
+  }];
+  let arguments = (ins DXSA_OutputPrimitiveTopologyAttr:$topology);
+  let assemblyFormat = "$topology attr-dict";
+}
+
 #endif // DXSA_OPS

mlir/lib/Target/DXSA/BinaryParser.cpp

@@ -516,6 +516,13 @@ class DXBuilder {
                                   builder.getI32IntegerAttr(count));
   }
 
+  Instruction buildDclOutputTopology(dxsa::OutputPrimitiveTopology topology,
+                                     Location loc) {
+    auto topologyAttr =
+        dxsa::OutputPrimitiveTopologyAttr::get(builder.getContext(), topology);
+    return dxsa::DclOutputTopology::create(builder, loc, topologyAttr);
+  }
+
 private:
   MLIRContext *context;
   ModuleOp module;
@@ -831,6 +838,24 @@ class Parser {
     return builder.buildDclTemps(count, loc);
   }
 
+  FailureOr<Instruction> parseDclOutputTopology(uint32_t opcodeToken,
+                                                Location loc) {
+    auto topology = DECODE_D3D10_SB_GS_OUTPUT_PRIMITIVE_TOPOLOGY(opcodeToken);
+    switch (topology) {
+    case D3D10_SB_PRIMITIVE_TOPOLOGY_POINTLIST:
+      return builder.buildDclOutputTopology(
+          dxsa::OutputPrimitiveTopology::pointlist, loc);
+    case D3D10_SB_PRIMITIVE_TOPOLOGY_LINESTRIP:
+      return builder.buildDclOutputTopology(
+          dxsa::OutputPrimitiveTopology::linestrip, loc);
+    case D3D10_SB_PRIMITIVE_TOPOLOGY_TRIANGLESTRIP:
+      return builder.buildDclOutputTopology(
+          dxsa::OutputPrimitiveTopology::trianglestrip, loc);
+    default:
+      return emitError(loc, "invalid output primitive topology");
+    }
+  }
+
   OptionalParseResult parseDclInstruction(uint32_t opcodeToken, Location loc,
                                           Instruction &out) {
     FailureOr<Instruction> result;
@@ -841,6 +866,9 @@ class Parser {
     case D3D10_SB_OPCODE_DCL_TEMPS:
       result = parseDclTemps(loc);
       break;
+    case D3D10_SB_OPCODE_DCL_GS_OUTPUT_PRIMITIVE_TOPOLOGY:
+      result = parseDclOutputTopology(opcodeToken, loc);
+      break;
     default:
       return std::nullopt;
     }

mlir/test/Target/DXSA/dcl_output_topology_linestrip.mlir

@@ -0,0 +1,3 @@
+// RUN: mlir-translate --import-dxsa-bin %S/inputs/dcl_output_topology_linestrip.bin | FileCheck %s
+
+// CHECK: dxsa.dcl_output_topology linestrip

mlir/test/Target/DXSA/dcl_output_topology_pointlist.mlir

@@ -0,0 +1,3 @@
+// RUN: mlir-translate --import-dxsa-bin %S/inputs/dcl_output_topology_pointlist.bin | FileCheck %s
+
+// CHECK: dxsa.dcl_output_topology pointlist

mlir/test/Target/DXSA/dcl_output_topology_trianglestrip.mlir

@@ -0,0 +1,3 @@
+// RUN: mlir-translate --import-dxsa-bin %S/inputs/dcl_output_topology_trianglestrip.bin | FileCheck %s
+
+// CHECK: dxsa.dcl_output_topology trianglestrip