[HLSL][RootSignature] Add serialization for RootFlags

Author: inbelic

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

@@ -46,6 +46,8 @@ enum class RootFlags : uint32_t {
   ValidFlags = 0x00000fff
 };
 
+raw_ostream &operator<<(raw_ostream &OS, const RootFlags &Flags);
+
 enum class DescriptorRangeFlags : unsigned {
   None = 0,
   DescriptorsVolatile = 0x1,

llvm/lib/Frontend/HLSL/HLSLRootSignature.cpp

@@ -132,6 +132,71 @@ static raw_ostream &operator<<(raw_ostream &OS,
   return OS;
 }
 
+raw_ostream &operator<<(raw_ostream &OS, const RootFlags &Flags) {
+  OS << "RootFlags(";
+  bool FlagSet = false;
+  unsigned Remaining = llvm::to_underlying(Flags);
+  while (Remaining) {
+    unsigned Bit = 1u << llvm::countr_zero(Remaining);
+    if (Remaining & Bit) {
+      if (FlagSet)
+        OS << " | ";
+
+      switch (static_cast<RootFlags>(Bit)) {
+      case RootFlags::AllowInputAssemblerInputLayout:
+        OS << "AllowInputAssemblerInputLayout";
+        break;
+      case RootFlags::DenyVertexShaderRootAccess:
+        OS << "DenyVertexShaderRootAccess";
+        break;
+      case RootFlags::DenyHullShaderRootAccess:
+        OS << "DenyHullShaderRootAccess";
+        break;
+      case RootFlags::DenyDomainShaderRootAccess:
+        OS << "DenyDomainShaderRootAccess";
+        break;
+      case RootFlags::DenyGeometryShaderRootAccess:
+        OS << "DenyGeometryShaderRootAccess";
+        break;
+      case RootFlags::DenyPixelShaderRootAccess:
+        OS << "DenyPixelShaderRootAccess";
+        break;
+      case RootFlags::AllowStreamOutput:
+        OS << "AllowStreamOutput";
+        break;
+      case RootFlags::LocalRootSignature:
+        OS << "LocalRootSignature";
+        break;
+      case RootFlags::DenyAmplificationShaderRootAccess:
+        OS << "DenyAmplificationShaderRootAccess";
+        break;
+      case RootFlags::DenyMeshShaderRootAccess:
+        OS << "DenyMeshShaderRootAccess";
+        break;
+      case RootFlags::CBVSRVUAVHeapDirectlyIndexed:
+        OS << "CBVSRVUAVHeapDirectlyIndexed";
+        break;
+      case RootFlags::SamplerHeapDirectlyIndexed:
+        OS << "SamplerHeapDirectlyIndexed";
+        break;
+      default:
+        OS << "invalid: " << Bit;
+        break;
+      }
+
+      FlagSet = true;
+    }
+    Remaining &= ~Bit;
+  }
+
+  if (!FlagSet)
+    OS << "None";
+
+  OS << ")";
+
+  return OS;
+}
+
 raw_ostream &operator<<(raw_ostream &OS, const RootConstants &Constants) {
   OS << "RootConstants(num32BitConstants = " << Constants.Num32BitConstants
      << ", " << Constants.Reg << ", space = " << Constants.Space

llvm/unittests/Frontend/HLSLRootSignatureDumpTest.cpp

@@ -140,4 +140,41 @@ TEST(HLSLRootSignatureTest, SetRootConstantsDump) {
   EXPECT_EQ(Out, Expected);
 }
 
+TEST(HLSLRootSignatureTest, NoneRootFlagsDump) {
+  RootFlags Flags = RootFlags::None;
+
+  std::string Out;
+  llvm::raw_string_ostream OS(Out);
+  OS << Flags;
+  OS.flush();
+
+  std::string Expected = "RootFlags(None)";
+  EXPECT_EQ(Out, Expected);
+}
+
+TEST(HLSLRootSignatureTest, AllRootFlagsDump) {
+  RootFlags Flags = RootFlags::ValidFlags;
+
+  std::string Out;
+  llvm::raw_string_ostream OS(Out);
+  OS << Flags;
+  OS.flush();
+
+  std::string Expected = "RootFlags("
+                         "AllowInputAssemblerInputLayout | "
+                         "DenyVertexShaderRootAccess | "
+                         "DenyHullShaderRootAccess | "
+                         "DenyDomainShaderRootAccess | "
+                         "DenyGeometryShaderRootAccess | "
+                         "DenyPixelShaderRootAccess | "
+                         "AllowStreamOutput | "
+                         "LocalRootSignature | "
+                         "DenyAmplificationShaderRootAccess | "
+                         "DenyMeshShaderRootAccess | "
+                         "CBVSRVUAVHeapDirectlyIndexed | "
+                         "SamplerHeapDirectlyIndexed)";
+
+  EXPECT_EQ(Out, Expected);
+}
+
 } // namespace