implement support for dumping descriptor table struct

Author: inbelic

llvm/include/llvm/Frontend/HLSL/HLSLRootSignature.h

@@ -61,6 +61,8 @@ struct Register {
 struct DescriptorTable {
   ShaderVisibility Visibility = ShaderVisibility::All;
   uint32_t NumClauses = 0; // The number of clauses in the table
+
+  void dump(raw_ostream &OS) const;
 };
 
 static const uint32_t NumDescriptorsUnbounded = 0xffffffff;

llvm/lib/Frontend/HLSL/HLSLRootSignature.cpp

@@ -39,6 +39,42 @@ void Register::dump(raw_ostream &OS) const {
   OS << Number;
 }
 
+static void dumpVisibility(raw_ostream &OS, ShaderVisibility Visibility) {
+  switch (Visibility) {
+    case ShaderVisibility::All:
+      OS << "All";
+      break;
+    case ShaderVisibility::Vertex:
+      OS << "Vertex";
+      break;
+    case ShaderVisibility::Hull:
+      OS << "Hull";
+      break;
+    case ShaderVisibility::Domain:
+      OS << "Domain";
+      break;
+    case ShaderVisibility::Geometry:
+      OS << "Geometry";
+      break;
+    case ShaderVisibility::Pixel:
+      OS << "Pixel";
+      break;
+    case ShaderVisibility::Amplification:
+      OS << "Amplification";
+      break;
+    case ShaderVisibility::Mesh:
+      OS << "Mesh";
+      break;
+  }
+}
+
+void DescriptorTable::dump(raw_ostream &OS) const {
+  OS << "DescriptorTable(numClauses = " << NumClauses;
+  OS << ", visibility = ";
+  dumpVisibility(OS, Visibility);
+  OS << ")";
+}
+
 static void dumpClauseType(raw_ostream& OS, ClauseType Type) {
   switch (Type) {
   case ClauseType::CBuffer:

llvm/unittests/Frontend/HLSLRootSignatureDumpTest.cpp

@@ -95,4 +95,19 @@ TEST(HLSLRootSignatureTest, DescriptorSamplerClauseDump) {
   EXPECT_EQ(Out, Expected);
 }
 
+TEST(HLSLRootSignatureTest, DescriptorTableDump) {
+  DescriptorTable Table;
+  Table.NumClauses = 4;
+  Table.Visibility = ShaderVisibility::Geometry;
+
+  std::string Out;
+  llvm::raw_string_ostream OS(Out);
+  Table.dump(OS);
+  OS.flush();
+
+  std::string Expected =
+    "DescriptorTable(numClauses = 4, visibility = Geometry)";
+  EXPECT_EQ(Out, Expected);
+}
+
 } // namespace