|
12 | 12 | //===----------------------------------------------------------------------===//
|
13 | 13 |
|
14 | 14 | #include "llvm-objdump.h"
|
15 |
| -#include "llvm/BinaryFormat/DXContainer.h" |
16 | 15 | #include "llvm/Object/DXContainer.h"
|
17 |
| -#include "llvm/Support/ScopedPrinter.h" |
18 | 16 |
|
19 | 17 | using namespace llvm;
|
20 |
| -using namespace llvm::object; |
21 |
| - |
22 |
| -static llvm::SmallString<4> maskToString(uint8_t Mask) { |
23 |
| - llvm::SmallString<4> Result(" "); |
24 |
| - if (Mask & 1) |
25 |
| - Result[0] = 'x'; |
26 |
| - if (Mask & 2) |
27 |
| - Result[1] = 'y'; |
28 |
| - if (Mask & 4) |
29 |
| - Result[2] = 'z'; |
30 |
| - if (Mask & 8) |
31 |
| - Result[3] = 'w'; |
32 |
| - return Result; |
33 |
| -} |
34 |
| - |
35 |
| -static void printColumnHeader(raw_ostream &OS, size_t Length) { |
36 |
| - for (size_t I = 0; I < Length; ++I) |
37 |
| - OS << "-"; |
38 |
| -} |
39 |
| - |
40 |
| -static void printColumnHeaders(raw_ostream &OS, ArrayRef<size_t> Lengths) { |
41 |
| - for (auto L : Lengths) { |
42 |
| - printColumnHeader(OS, L); |
43 |
| - OS << " "; |
44 |
| - } |
45 |
| - OS << "\n"; |
46 |
| -} |
47 |
| - |
48 |
| -static size_t digitsForNumber(size_t N) { |
49 |
| - return static_cast<size_t>(log10(static_cast<double>(N))) + 1; |
50 |
| -} |
51 | 18 |
|
52 | 19 | namespace {
|
53 | 20 | class DXContainerDumper : public objdump::Dumper {
|
54 |
| - const DXContainerObjectFile &Obj; |
55 |
| - |
56 | 21 | public:
|
57 |
| - DXContainerDumper(const DXContainerObjectFile &O) |
58 |
| - : objdump::Dumper(O), Obj(O) {} |
59 |
| - |
60 |
| - void printPrivateHeaders() override; |
61 |
| - void printSignature(const DirectX::Signature &S); |
| 22 | + DXContainerDumper(const object::DXContainerObjectFile &Obj) |
| 23 | + : objdump::Dumper(Obj) {} |
62 | 24 | };
|
63 |
| - |
64 |
| -void DXContainerDumper::printSignature(const DirectX::Signature &S) { |
65 |
| - // DXC prints a table like this as part of the shader disassembly: |
66 |
| - //; Name Index Mask Register SysValue Format Used |
67 |
| - //; -------------------- ----- ------ -------- -------- ------- ------ |
68 |
| - //; NORMAL 0 xyz 0 NONE float xyz |
69 |
| - //; TEXCOORD 0 xy 1 NONE float xy |
70 |
| - |
71 |
| - // DXC's implementation doesn't scale columns entirely completely for the |
72 |
| - // provided input, so this implementation is a bit more complicated in |
73 |
| - // formatting logic to scale with the size of the printed text. |
74 |
| - |
75 |
| - // DXC gives names 21 characters for some unknown reason, I arbitrarily chose |
76 |
| - // to start at 24 so that we're not going shorter but are using a round |
77 |
| - // number. |
78 |
| - size_t LongestName = 24; |
79 |
| - size_t LongestSV = 10; |
80 |
| - size_t LongestIndex = strlen("Index"); |
81 |
| - size_t LongestRegister = strlen("Register"); |
82 |
| - size_t LongestFormat = strlen("Format"); |
83 |
| - const size_t MaskWidth = 5; |
84 |
| - // Compute the column widths. Skip calculating the "Mask" and "Used" columns |
85 |
| - // since they both have widths of 4. |
86 |
| - for (auto El : S) { |
87 |
| - LongestName = std::max(LongestName, S.getName(El.NameOffset).size()); |
88 |
| - LongestSV = std::max( |
89 |
| - LongestSV, |
90 |
| - enumToStringRef(El.SystemValue, dxbc::getD3DSystemValues()).size()); |
91 |
| - LongestIndex = std::max(LongestIndex, digitsForNumber(El.Index)); |
92 |
| - LongestRegister = std::max(LongestRegister, digitsForNumber(El.Register)); |
93 |
| - LongestFormat = std::max( |
94 |
| - LongestFormat, |
95 |
| - enumToStringRef(El.CompType, dxbc::getSigComponentTypes()).size()); |
96 |
| - } |
97 |
| - |
98 |
| - // Print Column headers. |
99 |
| - OS << "; "; |
100 |
| - OS << left_justify("Name", LongestName) << " "; |
101 |
| - OS << right_justify("Index", LongestIndex) << " "; |
102 |
| - OS << right_justify("Mask", MaskWidth) << " "; |
103 |
| - OS << right_justify("Register", LongestRegister) << " "; |
104 |
| - OS << right_justify("SysValue", LongestSV) << " "; |
105 |
| - OS << right_justify("Format", LongestFormat) << " "; |
106 |
| - OS << right_justify("Used", MaskWidth) << "\n"; |
107 |
| - OS << "; "; |
108 |
| - printColumnHeaders(OS, {LongestName, LongestIndex, MaskWidth, LongestRegister, |
109 |
| - LongestSV, LongestFormat, MaskWidth}); |
110 |
| - |
111 |
| - for (auto El : S) { |
112 |
| - OS << "; " << left_justify(S.getName(El.NameOffset), LongestName) << " "; |
113 |
| - OS << right_justify(std::to_string(El.Index), LongestIndex) << " "; |
114 |
| - OS << right_justify(maskToString(El.Mask), MaskWidth) << " "; |
115 |
| - OS << right_justify(std::to_string(El.Register), LongestRegister) << " "; |
116 |
| - OS << right_justify( |
117 |
| - enumToStringRef(El.SystemValue, dxbc::getD3DSystemValues()), |
118 |
| - LongestSV) |
119 |
| - << " "; |
120 |
| - OS << right_justify( |
121 |
| - enumToStringRef(El.CompType, dxbc::getSigComponentTypes()), |
122 |
| - LongestFormat) |
123 |
| - << " "; |
124 |
| - OS << right_justify(maskToString(El.ExclusiveMask), MaskWidth) << "\n"; |
125 |
| - } |
126 |
| -} |
127 |
| - |
128 |
| -void DXContainerDumper::printPrivateHeaders() { |
129 |
| - const DXContainer &C = |
130 |
| - cast<object::DXContainerObjectFile>(Obj).getDXContainer(); |
131 |
| - |
132 |
| - if (!C.getInputSignature().isEmpty()) { |
133 |
| - OS << "; Input signature:\n;\n"; |
134 |
| - printSignature(C.getInputSignature()); |
135 |
| - OS << ";\n"; |
136 |
| - } |
137 |
| - |
138 |
| - if (!C.getOutputSignature().isEmpty()) { |
139 |
| - OS << "; Output signature:\n;\n"; |
140 |
| - printSignature(C.getOutputSignature()); |
141 |
| - OS << ";\n"; |
142 |
| - } |
143 |
| - |
144 |
| - if (!C.getPatchConstantSignature().isEmpty()) { |
145 |
| - OS << "; Patch Constant signature:\n;\n"; |
146 |
| - printSignature(C.getPatchConstantSignature()); |
147 |
| - OS << ";\n"; |
148 |
| - } |
149 |
| -} |
150 | 25 | } // namespace
|
151 | 26 |
|
152 | 27 | std::unique_ptr<objdump::Dumper> llvm::objdump::createDXContainerDumper(
|
|
0 commit comments