1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
#pragma once
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif
//===- llvm/MC/DXContainerPSVInfo.h - DXContainer PSVInfo -*- C++ -------*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_MC_DXCONTAINERPSVINFO_H
#define LLVM_MC_DXCONTAINERPSVINFO_H
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/BinaryFormat/DXContainer.h"
#include "llvm/TargetParser/Triple.h"
#include <array>
#include <numeric>
#include <stdint.h>
namespace llvm {
class raw_ostream;
namespace mcdxbc {
struct PSVSignatureElement {
StringRef Name;
SmallVector<uint32_t> Indices;
uint8_t StartRow;
uint8_t Cols;
uint8_t StartCol;
bool Allocated;
dxbc::PSV::SemanticKind Kind;
dxbc::PSV::ComponentType Type;
dxbc::PSV::InterpolationMode Mode;
uint8_t DynamicMask;
uint8_t Stream;
};
// This data structure is a helper for reading and writing PSV RuntimeInfo data.
// It is implemented in the BinaryFormat library so that it can be used by both
// the MC layer and Object tools.
// This structure is used to represent the extracted data in an inspectable and
// modifiable format, and can be used to serialize the data back into valid PSV
// RuntimeInfo.
struct PSVRuntimeInfo {
bool IsFinalized = false;
dxbc::PSV::v2::RuntimeInfo BaseData;
SmallVector<dxbc::PSV::v2::ResourceBindInfo> Resources;
SmallVector<PSVSignatureElement> InputElements;
SmallVector<PSVSignatureElement> OutputElements;
SmallVector<PSVSignatureElement> PatchOrPrimElements;
// TODO: Make this interface user-friendly.
// The interface here is bad, and we'll want to change this in the future. We
// probably will want to build out these mask vectors as vectors of bools and
// have this utility object convert them to the bit masks. I don't want to
// over-engineer this API now since we don't know what the data coming in to
// feed it will look like, so I kept it extremely simple for the immediate use
// case.
std::array<SmallVector<uint32_t>, 4> OutputVectorMasks;
SmallVector<uint32_t> PatchOrPrimMasks;
std::array<SmallVector<uint32_t>, 4> InputOutputMap;
SmallVector<uint32_t> InputPatchMap;
SmallVector<uint32_t> PatchOutputMap;
// Serialize PSVInfo into the provided raw_ostream. The version field
// specifies the data version to encode, the default value specifies encoding
// the highest supported version.
void write(raw_ostream &OS,
uint32_t Version = std::numeric_limits<uint32_t>::max()) const;
void finalize(Triple::EnvironmentType Stage) {
IsFinalized = true;
BaseData.SigInputElements = static_cast<uint32_t>(InputElements.size());
BaseData.SigOutputElements = static_cast<uint32_t>(OutputElements.size());
BaseData.SigPatchOrPrimElements =
static_cast<uint32_t>(PatchOrPrimElements.size());
if (!sys::IsBigEndianHost)
return;
BaseData.swapBytes();
BaseData.swapBytes(Stage);
for (auto &Res : Resources)
Res.swapBytes();
}
};
class Signature {
struct Parameter {
uint32_t Stream;
StringRef Name;
uint32_t Index;
dxbc::D3DSystemValue SystemValue;
dxbc::SigComponentType CompType;
uint32_t Register;
uint8_t Mask;
uint8_t ExclusiveMask;
dxbc::SigMinPrecision MinPrecision;
};
SmallVector<Parameter> Params;
public:
void addParam(uint32_t Stream, StringRef Name, uint32_t Index,
dxbc::D3DSystemValue SystemValue,
dxbc::SigComponentType CompType, uint32_t Register,
uint8_t Mask, uint8_t ExclusiveMask,
dxbc::SigMinPrecision MinPrecision) {
Params.push_back(Parameter{Stream, Name, Index, SystemValue, CompType,
Register, Mask, ExclusiveMask, MinPrecision});
}
void write(raw_ostream &OS);
};
} // namespace mcdxbc
} // namespace llvm
#endif // LLVM_MC_DXCONTAINERPSVINFO_H
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
|
