[NFC] add utils for PSV. (microsoft#6869)

Add init functions to help build
PSVSignatureElement0/PSVRuntimeInfo/PSVResourceInfo.

Add print function for DxilPipelineStateValidation.

Add new option -dumppsv to dxa for testing.

For microsoft#6817.

Author: python3kgae

include/dxc/DxilContainer/DxilContainer.h

@@ -166,6 +166,9 @@ enum class DxilProgramSigSemantic : uint32_t {
   InnerCoverage = 70,
 };
 
+DxilProgramSigSemantic SemanticKindToSystemValue(DXIL::SemanticKind,
+                                                 DXIL::TessellatorDomain);
+
 enum class DxilProgramSigCompType : uint32_t {
   Unknown = 0,
   UInt32 = 1,
@@ -179,6 +182,9 @@ enum class DxilProgramSigCompType : uint32_t {
   Float64 = 9,
 };
 
+DxilProgramSigCompType CompTypeToSigCompType(DXIL::ComponentType,
+                                             bool i1ToUnknownCompat);
+
 struct DxilProgramSignatureElement {
   uint32_t Stream; // Stream index (parameters must appear in non-decreasing
                    // stream order)

include/dxc/DxilContainer/DxilPipelineStateValidation.h

@@ -13,9 +13,14 @@
 #define __DXIL_PIPELINE_STATE_VALIDATION__H__
 
 #include "dxc/WinAdapter.h"
+#include <assert.h>
 #include <cstring>
 #include <stdint.h>
 
+namespace llvm {
+class raw_ostream;
+}
+
 // Don't include assert.h here.
 // Since this header is included from multiple environments,
 // it is necessary to define assert before this header is included.
@@ -229,11 +234,13 @@ struct PSVResourceBindInfo0 {
   uint32_t Space;
   uint32_t LowerBound;
   uint32_t UpperBound;
+  void Print(llvm::raw_ostream &O) const;
 };
 
 struct PSVResourceBindInfo1 : public PSVResourceBindInfo0 {
   uint32_t ResKind;  // PSVResourceKind
   uint32_t ResFlags; // special characteristics of the resource
+  void Print(llvm::raw_ostream &O) const;
 };
 
 // Helpers for output dependencies (ViewID and Input-Output tables)
@@ -266,7 +273,8 @@ struct PSVComponentMask {
     if (ComponentIndex < NumVectors * 4)
       Mask[ComponentIndex >> 5] &= ~(1 << (ComponentIndex & 0x1F));
   }
-  bool IsValid() { return Mask != nullptr; }
+  bool IsValid() const { return Mask != nullptr; }
+  void Print(llvm::raw_ostream &, const char *, const char *) const;
 };
 
 struct PSVDependencyTable {
@@ -282,14 +290,23 @@ struct PSVDependencyTable {
       : Table(pTable), InputVectors(inputVectors),
         OutputVectors(outputVectors) {}
   PSVComponentMask GetMaskForInput(uint32_t inputComponentIndex) {
+    return getMaskForInput(inputComponentIndex);
+  }
+  const PSVComponentMask GetMaskForInput(uint32_t inputComponentIndex) const {
+    return getMaskForInput(inputComponentIndex);
+  }
+  bool IsValid() const { return Table != nullptr; }
+  void Print(llvm::raw_ostream &, const char *, const char *) const;
+
+private:
+  PSVComponentMask getMaskForInput(uint32_t inputComponentIndex) const {
     if (!Table || !InputVectors || !OutputVectors)
       return PSVComponentMask();
     return PSVComponentMask(
         Table + (PSVComputeMaskDwordsFromVectors(OutputVectors) *
                  inputComponentIndex),
         OutputVectors);
   }
-  bool IsValid() { return Table != nullptr; }
 };
 
 struct PSVString {
@@ -431,6 +448,7 @@ class PSVSignatureElement {
   uint32_t GetDynamicIndexMask() const {
     return !m_pElement0 ? 0 : (uint32_t)m_pElement0->DynamicMaskAndStream & 0xF;
   }
+  void Print(llvm::raw_ostream &O) const;
 };
 
 #define MAX_PSV_VERSION 3
@@ -732,6 +750,9 @@ class DxilPipelineStateValidation {
                ? m_StringTable.Get(m_pPSVRuntimeInfo3->EntryFunctionName)
                : "";
   }
+  void PrintPSVRuntimeInfo(llvm::raw_ostream &O, uint8_t ShaderKind,
+                           const char *Comment) const;
+  void Print(llvm::raw_ostream &O, uint8_t ShaderKind) const;
 };
 
 // Return true if size fits in remaing buffer.
@@ -1036,6 +1057,10 @@ DxilPipelineStateValidation::ReadOrWrite(const void *pBits, uint32_t *pSize,
 
 namespace hlsl {
 
+class DxilResourceBase;
+class DxilSignatureElement;
+class DxilModule;
+
 class ViewIDValidator {
 public:
   enum class Result {
@@ -1059,6 +1084,21 @@ class ViewIDValidator {
 ViewIDValidator *NewViewIDValidator(unsigned viewIDCount,
                                     unsigned gsRastStreamIndex);
 
+void InitPSVResourceBinding(PSVResourceBindInfo0 *, PSVResourceBindInfo1 *,
+                            DxilResourceBase *);
+
+// Setup PSVSignatureElement0 with DxilSignatureElement.
+// Note that the SemanticName and SemanticIndexes are not done.
+void InitPSVSignatureElement(PSVSignatureElement0 &E,
+                             const DxilSignatureElement &SE,
+                             bool i1ToUnknownCompat);
+
+// Setup PSVRuntimeInfo* with DxilModule.
+// Note that the EntryFunctionName is not done.
+void InitPSVRuntimeInfo(PSVRuntimeInfo0 *pInfo, PSVRuntimeInfo1 *pInfo1,
+                        PSVRuntimeInfo2 *pInfo2, PSVRuntimeInfo3 *pInfo3,
+                        const DxilModule &DM);
+
 } // namespace hlsl
 
 #undef PSV_RETB

lib/DxilContainer/CMakeLists.txt

@@ -7,6 +7,7 @@ add_llvm_library(LLVMDxilContainer
   DxilContainerAssembler.cpp
   DxilContainerReader.cpp
   DxcContainerBuilder.cpp
+  DxilPipelineStateValidation.cpp
   DxilRDATBuilder.cpp
   DxilRuntimeReflection.cpp
   RDATDumper.cpp