ShaderOpTest.cpp 86 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144
  1. ///////////////////////////////////////////////////////////////////////////////
  2. // //
  3. // ShaderOpTest.cpp //
  4. // Copyright (C) Microsoft Corporation. All rights reserved. //
  5. // This file is distributed under the University of Illinois Open Source //
  6. // License. See LICENSE.TXT for details. //
  7. // //
  8. // Provides the implementation to run tests based on descriptions. //
  9. // //
  10. ///////////////////////////////////////////////////////////////////////////////
  11. #include <windows.h>
  12. #include <d3d12.h>
  13. #include <dxgi1_4.h>
  14. #include "dxc/Support/d3dx12.h"
  15. #include <d3dcompiler.h>
  16. #include <atlbase.h>
  17. #include <atlenc.h>
  18. #include "ShaderOpTest.h"
  19. #include "dxc/dxcapi.h" // IDxcCompiler
  20. #include "dxc/Support/Global.h" // OutputDebugBytes
  21. #include "dxc/Support/Unicode.h" // IsStarMatchUTF16
  22. #include "dxc/Support/dxcapi.use.h" // DxcDllSupport
  23. #include "dxc/DXIL/DxilConstants.h" // ComponentType
  24. #include "WexTestClass.h" // TAEF
  25. #include "dxc/Test/HLSLTestUtils.h" // LogCommentFmt
  26. #include <stdlib.h>
  27. #include <DirectXMath.h>
  28. #include <intsafe.h>
  29. #include <strsafe.h>
  30. #include <xmllite.h>
  31. #pragma comment(lib, "xmllite.lib")
  32. ///////////////////////////////////////////////////////////////////////////////
  33. // Useful helper functions.
  34. static st::OutputStringFn g_OutputStrFn;
  35. static void * g_OutputStrFnCtx;
  36. void st::SetOutputFn(void *pCtx, OutputStringFn F) {
  37. g_OutputStrFnCtx = pCtx;
  38. g_OutputStrFn = F;
  39. }
  40. static void ShaderOpLogFmt(_In_z_ _Printf_format_string_ const wchar_t *fmt, ...) {
  41. va_list args;
  42. va_start(args, fmt);
  43. std::wstring buf(hlsl_test::vFormatToWString(fmt, args));
  44. va_end(args);
  45. if (g_OutputStrFn == nullptr)
  46. WEX::Logging::Log::Comment(buf.data());
  47. else
  48. g_OutputStrFn(g_OutputStrFnCtx, buf.data());
  49. }
  50. // Rely on TAEF Verifier helpers.
  51. #define CHECK_HR(x) { \
  52. if (!g_OutputStrFn) VERIFY_SUCCEEDED(x); else { \
  53. HRESULT _check_hr = (x); \
  54. if (FAILED(_check_hr)) AtlThrow(x); } \
  55. }
  56. // Check the specified HRESULT and return the success value.
  57. static HRESULT CHECK_HR_RET(HRESULT hr) {
  58. CHECK_HR(hr);
  59. return hr;
  60. }
  61. HRESULT LogIfLost(HRESULT hr, ID3D12Device *pDevice) {
  62. if (hr == DXGI_ERROR_DEVICE_REMOVED) {
  63. HRESULT reason = pDevice->GetDeviceRemovedReason();
  64. LPCWSTR reasonText = L"?";
  65. if (reason == DXGI_ERROR_DEVICE_HUNG) reasonText = L"DXGI_ERROR_DEVICE_HUNG";
  66. if (reason == DXGI_ERROR_DEVICE_REMOVED) reasonText = L"DXGI_ERROR_DEVICE_REMOVED";
  67. if (reason == DXGI_ERROR_DEVICE_RESET) reasonText = L"DXGI_ERROR_DEVICE_RESET";
  68. if (reason == DXGI_ERROR_DRIVER_INTERNAL_ERROR) reasonText = L"DXGI_ERROR_DRIVER_INTERNAL_ERROR";
  69. if (reason == DXGI_ERROR_INVALID_CALL) reasonText = L"DXGI_ERROR_INVALID_CALL";
  70. ShaderOpLogFmt(L"Device lost: 0x%08x (%s)", reason, reasonText);
  71. }
  72. return hr;
  73. }
  74. HRESULT LogIfLost(HRESULT hr, ID3D12Resource *pResource) {
  75. if (hr == DXGI_ERROR_DEVICE_REMOVED) {
  76. CComPtr<ID3D12Device> pDevice;
  77. pResource->GetDevice(__uuidof(ID3D12Device), (void**)&pDevice);
  78. LogIfLost(hr, pDevice);
  79. }
  80. return hr;
  81. }
  82. bool UseHardwareDevice(const DXGI_ADAPTER_DESC1 &desc, LPCWSTR AdapterName) {
  83. if (desc.Flags & DXGI_ADAPTER_FLAG_SOFTWARE) {
  84. // Don't select the Basic Render Driver adapter.
  85. return false;
  86. }
  87. if (!AdapterName)
  88. return true;
  89. return Unicode::IsStarMatchUTF16(AdapterName, wcslen(AdapterName),
  90. desc.Description, wcslen(desc.Description));
  91. }
  92. void GetHardwareAdapter(IDXGIFactory2 *pFactory, LPCWSTR AdapterName,
  93. IDXGIAdapter1 **ppAdapter) {
  94. CComPtr<IDXGIAdapter1> adapter;
  95. *ppAdapter = nullptr;
  96. for (UINT adapterIndex = 0;
  97. DXGI_ERROR_NOT_FOUND != pFactory->EnumAdapters1(adapterIndex, &adapter);
  98. ++adapterIndex) {
  99. DXGI_ADAPTER_DESC1 desc;
  100. adapter->GetDesc1(&desc);
  101. if (!UseHardwareDevice(desc, AdapterName)) {
  102. adapter.Release();
  103. continue;
  104. }
  105. // Check to see if the adapter supports Direct3D 12, but don't create the
  106. // actual device yet.
  107. if (SUCCEEDED(D3D12CreateDevice(adapter, D3D_FEATURE_LEVEL_11_0,
  108. _uuidof(ID3D12Device), nullptr))) {
  109. break;
  110. }
  111. adapter.Release();
  112. }
  113. *ppAdapter = adapter.Detach();
  114. }
  115. void RecordTransitionBarrier(ID3D12GraphicsCommandList *pCommandList,
  116. ID3D12Resource *pResource,
  117. D3D12_RESOURCE_STATES before,
  118. D3D12_RESOURCE_STATES after) {
  119. CD3DX12_RESOURCE_BARRIER barrier(
  120. CD3DX12_RESOURCE_BARRIER::Transition(pResource, before, after));
  121. pCommandList->ResourceBarrier(1, &barrier);
  122. }
  123. void ExecuteCommandList(ID3D12CommandQueue *pQueue, ID3D12CommandList *pList) {
  124. ID3D12CommandList *ppCommandLists[] = { pList };
  125. pQueue->ExecuteCommandLists(1, ppCommandLists);
  126. }
  127. HRESULT SetObjectName(ID3D12Object *pObject, LPCSTR pName) {
  128. if (pObject && pName) {
  129. CA2W WideName(pName);
  130. return pObject->SetName(WideName);
  131. }
  132. return S_FALSE;
  133. }
  134. void WaitForSignal(ID3D12CommandQueue *pCQ, ID3D12Fence *pFence,
  135. HANDLE hFence, UINT64 fenceValue) {
  136. // Signal and increment the fence value.
  137. const UINT64 fence = fenceValue;
  138. CHECK_HR(pCQ->Signal(pFence, fence));
  139. if (pFence->GetCompletedValue() < fenceValue) {
  140. CHECK_HR(pFence->SetEventOnCompletion(fenceValue, hFence));
  141. WaitForSingleObject(hFence, INFINITE);
  142. //CHECK_HR(pCQ->Wait(pFence, fenceValue));
  143. }
  144. }
  145. static void SetupComputeValuePattern(std::vector<uint32_t> &values, size_t count) {
  146. values.resize(count); // one element per dispatch group, in bytes
  147. for (size_t i = 0; i < count; ++i) {
  148. values[i] = (uint32_t)i;
  149. }
  150. }
  151. void MappedData::dump() const {
  152. OutputDebugBytes(m_pData, m_size);
  153. }
  154. void MappedData::reset() {
  155. if (m_pResource != nullptr) {
  156. m_pResource->Unmap(0, nullptr);
  157. m_pResource.Release();
  158. }
  159. m_pData = nullptr;
  160. }
  161. ///////////////////////////////////////////////////////////////////////////////
  162. // Helper class for mapped data.
  163. void MappedData::reset(ID3D12Resource *pResource, UINT32 sizeInBytes) {
  164. reset();
  165. D3D12_RANGE r;
  166. r.Begin = 0;
  167. r.End = sizeInBytes;
  168. CHECK_HR(LogIfLost(pResource->Map(0, &r, &m_pData), pResource));
  169. m_pResource = pResource;
  170. m_size = sizeInBytes;
  171. }
  172. ///////////////////////////////////////////////////////////////////////////////
  173. // ShaderOpTest library implementation.
  174. namespace st {
  175. LPCSTR string_table::insert(LPCSTR pValue) {
  176. std::unordered_set<LPCSTR, HashStr, PredStr>::iterator i = m_values.find(pValue);
  177. if (i == m_values.end()) {
  178. size_t bufSize = strlen(pValue) + 1;
  179. std::vector<char> s;
  180. s.resize(bufSize);
  181. strcpy_s(s.data(), bufSize, pValue);
  182. LPCSTR result = s.data();
  183. m_values.insert(result);
  184. m_strings.push_back(std::move(s));
  185. return result;
  186. }
  187. else {
  188. return *i;
  189. }
  190. }
  191. LPCSTR string_table::insert(LPCWSTR pValue) {
  192. CW2A pValueAnsi(pValue);
  193. return insert(pValueAnsi.m_psz);
  194. }
  195. void CommandListRefs::CreateForDevice(ID3D12Device *pDevice, bool compute) {
  196. D3D12_COMMAND_LIST_TYPE T = compute ? D3D12_COMMAND_LIST_TYPE_COMPUTE
  197. : D3D12_COMMAND_LIST_TYPE_DIRECT;
  198. D3D12_COMMAND_QUEUE_DESC queueDesc = {};
  199. queueDesc.Flags = D3D12_COMMAND_QUEUE_FLAG_NONE;
  200. queueDesc.Type = T;
  201. if (Queue == nullptr) {
  202. CHECK_HR(pDevice->CreateCommandQueue(&queueDesc, IID_PPV_ARGS(&Queue)));
  203. }
  204. CHECK_HR(pDevice->CreateCommandAllocator(T, IID_PPV_ARGS(&Allocator)));
  205. CHECK_HR(pDevice->CreateCommandList(0, T, Allocator, nullptr,
  206. IID_PPV_ARGS(&List)));
  207. }
  208. void ShaderOpTest::CopyBackResources() {
  209. CommandListRefs ResCommandList;
  210. ResCommandList.CreateForDevice(m_pDevice, m_pShaderOp->IsCompute());
  211. ID3D12GraphicsCommandList *pList = ResCommandList.List;
  212. pList->SetName(L"ShaderOpTest Resource ReadBack CommandList");
  213. for (ShaderOpResource &R : m_pShaderOp->Resources) {
  214. if (!R.ReadBack)
  215. continue;
  216. ShaderOpResourceData &D = m_ResourceData[R.Name];
  217. RecordTransitionBarrier(pList, D.Resource, D.ResourceState,
  218. D3D12_RESOURCE_STATE_COPY_SOURCE);
  219. D.ResourceState = D3D12_RESOURCE_STATE_COPY_SOURCE;
  220. D3D12_RESOURCE_DESC &Desc = D.ShaderOpRes->Desc;
  221. if (Desc.Dimension == D3D12_RESOURCE_DIMENSION_BUFFER) {
  222. pList->CopyResource(D.ReadBack, D.Resource);
  223. }
  224. else {
  225. UINT64 rowPitch = Desc.Width * GetByteSizeForFormat(Desc.Format);
  226. if (rowPitch % D3D12_TEXTURE_DATA_PITCH_ALIGNMENT)
  227. rowPitch += D3D12_TEXTURE_DATA_PITCH_ALIGNMENT - (rowPitch % D3D12_TEXTURE_DATA_PITCH_ALIGNMENT);
  228. D3D12_PLACED_SUBRESOURCE_FOOTPRINT Footprint;
  229. Footprint.Offset = 0;
  230. Footprint.Footprint = CD3DX12_SUBRESOURCE_FOOTPRINT(Desc.Format, (UINT)Desc.Width, Desc.Height, 1, (UINT)rowPitch);
  231. CD3DX12_TEXTURE_COPY_LOCATION DstLoc(D.ReadBack, Footprint);
  232. CD3DX12_TEXTURE_COPY_LOCATION SrcLoc(D.Resource, 0);
  233. pList->CopyTextureRegion(&DstLoc, 0, 0, 0, &SrcLoc, nullptr);
  234. }
  235. }
  236. pList->Close();
  237. ExecuteCommandList(ResCommandList.Queue, pList);
  238. WaitForSignal(ResCommandList.Queue, m_pFence, m_hFence, m_FenceValue++);
  239. }
  240. void ShaderOpTest::CreateCommandList() {
  241. bool priorQueue = m_CommandList.Queue != nullptr;
  242. m_CommandList.CreateForDevice(m_pDevice, m_pShaderOp->IsCompute());
  243. m_CommandList.Allocator->SetName(L"ShaderOpTest Allocator");
  244. m_CommandList.List->SetName(L"ShaderOpTest CommandList");
  245. if (!priorQueue)
  246. m_CommandList.Queue->SetName(L"ShaderOpTest CommandList");
  247. }
  248. void ShaderOpTest::CreateDescriptorHeaps() {
  249. for (ShaderOpDescriptorHeap &H : m_pShaderOp->DescriptorHeaps) {
  250. CComPtr<ID3D12DescriptorHeap> pHeap;
  251. if (H.Desc.NumDescriptors == 0) {
  252. H.Desc.NumDescriptors = (UINT)H.Descriptors.size();
  253. }
  254. CHECK_HR(m_pDevice->CreateDescriptorHeap(&H.Desc, IID_PPV_ARGS(&pHeap)));
  255. m_DescriptorHeaps.push_back(pHeap);
  256. m_DescriptorHeapsByName[H.Name] = pHeap;
  257. SetObjectName(pHeap, H.Name);
  258. const UINT descriptorSize = m_pDevice->GetDescriptorHandleIncrementSize(H.Desc.Type);
  259. CD3DX12_CPU_DESCRIPTOR_HANDLE cpuHandle(pHeap->GetCPUDescriptorHandleForHeapStart());
  260. CD3DX12_GPU_DESCRIPTOR_HANDLE gpuHandle(pHeap->GetGPUDescriptorHandleForHeapStart());
  261. for (ShaderOpDescriptor &D : H.Descriptors) {
  262. ShaderOpResource *R = m_pShaderOp->GetResourceByName(D.ResName);
  263. if (R == nullptr) {
  264. LPCSTR DescName = D.Name ? D.Name : "[unnamed descriptor]";
  265. ShaderOpLogFmt(L"Descriptor '%S' references missing resource '%S'", DescName, D.ResName);
  266. CHECK_HR(E_INVALIDARG);
  267. }
  268. ShaderOpResourceData &Data = m_ResourceData[D.ResName];
  269. ShaderOpDescriptorData DData;
  270. DData.Descriptor = &D;
  271. DData.ResData = &Data;
  272. if (0 == _stricmp(D.Kind, "UAV")) {
  273. ID3D12Resource *pCounterResource = nullptr;
  274. if (D.CounterName && *D.CounterName) {
  275. ShaderOpResourceData &CounterData = m_ResourceData[D.CounterName];
  276. pCounterResource = CounterData.Resource;
  277. }
  278. m_pDevice->CreateUnorderedAccessView(Data.Resource, pCounterResource,
  279. &D.UavDesc, cpuHandle);
  280. }
  281. else if (0 == _stricmp(D.Kind, "SRV")) {
  282. D3D12_SHADER_RESOURCE_VIEW_DESC *pSrvDesc = nullptr;
  283. if (D.SrvDescPresent) {
  284. pSrvDesc = &D.SrvDesc;
  285. }
  286. m_pDevice->CreateShaderResourceView(Data.Resource, pSrvDesc, cpuHandle);
  287. }
  288. else if (0 == _stricmp(D.Kind, "RTV")) {
  289. m_pDevice->CreateRenderTargetView(Data.Resource, nullptr, cpuHandle);
  290. }
  291. else if (0 == _stricmp(D.Kind, "CBV")) {
  292. D3D12_CONSTANT_BUFFER_VIEW_DESC cbvDesc;
  293. cbvDesc.BufferLocation = Data.Resource->GetGPUVirtualAddress();
  294. cbvDesc.SizeInBytes = (UINT)Data.Resource->GetDesc().Width;
  295. m_pDevice->CreateConstantBufferView(&cbvDesc, cpuHandle);
  296. }
  297. DData.GPUHandle = gpuHandle;
  298. DData.CPUHandle = cpuHandle;
  299. m_DescriptorData[R->Name] = DData;
  300. cpuHandle = cpuHandle.Offset(descriptorSize);
  301. gpuHandle = gpuHandle.Offset(descriptorSize);
  302. }
  303. }
  304. // Create query descriptor heap.
  305. D3D12_QUERY_HEAP_DESC queryHeapDesc;
  306. ZeroMemory(&queryHeapDesc, sizeof(queryHeapDesc));
  307. queryHeapDesc.Count = 1;
  308. queryHeapDesc.Type = D3D12_QUERY_HEAP_TYPE_PIPELINE_STATISTICS;
  309. CHECK_HR(m_pDevice->CreateQueryHeap(&queryHeapDesc, IID_PPV_ARGS(&m_pQueryHeap)));
  310. }
  311. void ShaderOpTest::CreateDevice() {
  312. if (m_pDevice == nullptr) {
  313. const D3D_FEATURE_LEVEL FeatureLevelRequired = D3D_FEATURE_LEVEL_11_0;
  314. CComPtr<IDXGIFactory4> factory;
  315. CComPtr<ID3D12Device> pDevice;
  316. CHECK_HR(CreateDXGIFactory1(IID_PPV_ARGS(&factory)));
  317. if (m_pShaderOp->UseWarpDevice) {
  318. CComPtr<IDXGIAdapter> warpAdapter;
  319. CHECK_HR(factory->EnumWarpAdapter(IID_PPV_ARGS(&warpAdapter)));
  320. CHECK_HR(D3D12CreateDevice(warpAdapter, FeatureLevelRequired,
  321. IID_PPV_ARGS(&pDevice)));
  322. } else {
  323. CComPtr<IDXGIAdapter1> hardwareAdapter;
  324. GetHardwareAdapter(factory, m_pShaderOp->AdapterName, &hardwareAdapter);
  325. if (hardwareAdapter == nullptr) {
  326. CHECK_HR(HRESULT_FROM_WIN32(ERROR_NOT_FOUND));
  327. }
  328. CHECK_HR(D3D12CreateDevice(hardwareAdapter, FeatureLevelRequired,
  329. IID_PPV_ARGS(&pDevice)));
  330. }
  331. m_pDevice.Attach(pDevice.Detach());
  332. m_pDevice->SetName(L"ShaderOpTest Device");
  333. }
  334. m_FenceValue = 1;
  335. CHECK_HR(m_pDevice->CreateFence(0, D3D12_FENCE_FLAG_NONE,
  336. __uuidof(ID3D12Fence), (void **)&m_pFence));
  337. m_pFence->SetName(L"ShaderOpTest Fence");
  338. m_hFence = CreateEvent(nullptr, FALSE, FALSE, nullptr);
  339. if (m_hFence == nullptr) {
  340. AtlThrow(HRESULT_FROM_WIN32(GetLastError()));
  341. }
  342. }
  343. static void InitByteCode(D3D12_SHADER_BYTECODE *pBytecode, ID3D10Blob *pBlob) {
  344. if (pBlob == nullptr) {
  345. pBytecode->BytecodeLength = 0;
  346. pBytecode->pShaderBytecode = nullptr;
  347. }
  348. else {
  349. pBytecode->BytecodeLength = pBlob->GetBufferSize();
  350. pBytecode->pShaderBytecode = pBlob->GetBufferPointer();
  351. }
  352. }
  353. template <typename TKey, typename TValue>
  354. TValue map_get_or_null(const std::map<TKey, TValue> &amap, const TKey& key) {
  355. auto it = amap.find(key);
  356. if (it == amap.end()) return nullptr;
  357. return (*it).second;
  358. }
  359. void ShaderOpTest::CreatePipelineState() {
  360. CreateRootSignature();
  361. CreateShaders();
  362. if (m_pShaderOp->IsCompute()) {
  363. CComPtr<ID3D10Blob> pCS;
  364. pCS = m_Shaders[m_pShaderOp->CS];
  365. D3D12_COMPUTE_PIPELINE_STATE_DESC CDesc;
  366. ZeroMemory(&CDesc, sizeof(CDesc));
  367. CDesc.pRootSignature = m_pRootSignature.p;
  368. InitByteCode(&CDesc.CS, pCS);
  369. CHECK_HR(m_pDevice->CreateComputePipelineState(&CDesc, IID_PPV_ARGS(&m_pPSO)));
  370. }
  371. else {
  372. CComPtr<ID3D10Blob> pVS, pDS, pHS, pGS, pPS;
  373. pPS = map_get_or_null(m_Shaders, m_pShaderOp->PS);
  374. pVS = map_get_or_null(m_Shaders, m_pShaderOp->VS);
  375. pGS = map_get_or_null(m_Shaders, m_pShaderOp->GS);
  376. pHS = map_get_or_null(m_Shaders, m_pShaderOp->HS);
  377. pDS = map_get_or_null(m_Shaders, m_pShaderOp->DS);
  378. D3D12_GRAPHICS_PIPELINE_STATE_DESC GDesc;
  379. ZeroMemory(&GDesc, sizeof(GDesc));
  380. InitByteCode(&GDesc.VS, pVS);
  381. InitByteCode(&GDesc.HS, pHS);
  382. InitByteCode(&GDesc.DS, pDS);
  383. InitByteCode(&GDesc.GS, pGS);
  384. InitByteCode(&GDesc.PS, pPS);
  385. GDesc.InputLayout.NumElements = (UINT)m_pShaderOp->InputElements.size();
  386. GDesc.InputLayout.pInputElementDescs = m_pShaderOp->InputElements.data();
  387. GDesc.PrimitiveTopologyType = m_pShaderOp->PrimitiveTopologyType;
  388. GDesc.NumRenderTargets = (UINT)m_pShaderOp->RenderTargets.size();
  389. GDesc.SampleMask = m_pShaderOp->SampleMask;
  390. for (size_t i = 0; i < m_pShaderOp->RenderTargets.size(); ++i) {
  391. ShaderOpResource *R = m_pShaderOp->GetResourceByName(m_pShaderOp->RenderTargets[i]);
  392. GDesc.RTVFormats[i] = R->Desc.Format;
  393. }
  394. GDesc.SampleDesc.Count = 1; // TODO: read from file, set from shader operation; also apply to count
  395. GDesc.RasterizerState = CD3DX12_RASTERIZER_DESC(D3D12_DEFAULT); // TODO: read from file, set from op
  396. GDesc.BlendState = CD3DX12_BLEND_DESC(D3D12_DEFAULT); // TODO: read from file, set from op
  397. // TODO: pending values to set
  398. #if 0
  399. D3D12_STREAM_OUTPUT_DESC StreamOutput;
  400. D3D12_DEPTH_STENCIL_DESC DepthStencilState;
  401. D3D12_INDEX_BUFFER_STRIP_CUT_VALUE IBStripCutValue;
  402. DXGI_FORMAT DSVFormat;
  403. UINT NodeMask;
  404. D3D12_PIPELINE_STATE_FLAGS Flags;
  405. #endif
  406. GDesc.pRootSignature = m_pRootSignature.p;
  407. CHECK_HR(m_pDevice->CreateGraphicsPipelineState(&GDesc, IID_PPV_ARGS(&m_pPSO)));
  408. }
  409. }
  410. void ShaderOpTest::CreateResources() {
  411. CommandListRefs ResCommandList;
  412. ResCommandList.CreateForDevice(m_pDevice, true);
  413. ResCommandList.Allocator->SetName(L"ShaderOpTest Resource Creation Allocation");
  414. ResCommandList.Queue->SetName(L"ShaderOpTest Resource Creation Queue");
  415. ResCommandList.List->SetName(L"ShaderOpTest Resource Creation CommandList");
  416. ID3D12GraphicsCommandList *pList = ResCommandList.List.p;
  417. std::vector<CComPtr<ID3D12Resource> > intermediates;
  418. for (ShaderOpResource &R : m_pShaderOp->Resources) {
  419. if (m_ResourceData.count(R.Name) > 0) continue;
  420. // Initialize the upload resource early, to allow a by-name initializer
  421. // to set the desired width.
  422. bool initByName = R.Init && 0 == _stricmp("byname", R.Init);
  423. bool initZero = R.Init && 0 == _stricmp("zero", R.Init);
  424. bool initFromBytes = R.Init && 0 == _stricmp("frombytes", R.Init);
  425. bool hasInit = initByName || initZero || initFromBytes;
  426. bool isBuffer = R.Desc.Dimension == D3D12_RESOURCE_DIMENSION_BUFFER;
  427. std::vector<BYTE> values;
  428. if (hasInit) {
  429. if (isBuffer) {
  430. values.resize((size_t)R.Desc.Width);
  431. }
  432. else {
  433. // Probably needs more information.
  434. values.resize((size_t)(R.Desc.Width * R.Desc.Height *
  435. GetByteSizeForFormat(R.Desc.Format)));
  436. }
  437. if (initZero) {
  438. memset(values.data(), 0, values.size());
  439. }
  440. else if (initByName) {
  441. m_InitCallbackFn(R.Name, values, m_pShaderOp);
  442. if (isBuffer) {
  443. R.Desc.Width = values.size();
  444. }
  445. }
  446. else if (initFromBytes) {
  447. values = R.InitBytes;
  448. if (R.Desc.Width == 0) {
  449. if (isBuffer) {
  450. R.Desc.Width = values.size();
  451. }
  452. else if (R.Desc.Dimension == D3D12_RESOURCE_DIMENSION_TEXTURE1D) {
  453. R.Desc.Width = values.size() / GetByteSizeForFormat(R.Desc.Format);
  454. }
  455. }
  456. }
  457. }
  458. CComPtr<ID3D12Resource> pResource;
  459. CHECK_HR(m_pDevice->CreateCommittedResource(
  460. &R.HeapProperties, R.HeapFlags, &R.Desc, R.InitialResourceState,
  461. nullptr, IID_PPV_ARGS(&pResource)));
  462. ShaderOpResourceData &D = m_ResourceData[R.Name];
  463. D.ShaderOpRes = &R;
  464. D.Resource = pResource;
  465. D.ResourceState = R.InitialResourceState;
  466. SetObjectName(pResource, R.Name);
  467. if (hasInit) {
  468. CComPtr<ID3D12Resource> pIntermediate;
  469. CD3DX12_HEAP_PROPERTIES upload(D3D12_HEAP_TYPE_UPLOAD);
  470. D3D12_RESOURCE_DESC uploadDesc = R.Desc;
  471. // Calculate size required for intermediate buffer
  472. UINT64 totalBytes;
  473. m_pDevice->GetCopyableFootprints(&uploadDesc, 0, 1, 0, nullptr, nullptr, nullptr, &totalBytes);
  474. if (!isBuffer) {
  475. // Assuming a simple linear layout here.
  476. uploadDesc.Dimension = D3D12_RESOURCE_DIMENSION_BUFFER;
  477. uploadDesc.Width = totalBytes;
  478. uploadDesc.Height = 1;
  479. uploadDesc.MipLevels = 1;
  480. uploadDesc.Format = DXGI_FORMAT_UNKNOWN;
  481. uploadDesc.Layout = D3D12_TEXTURE_LAYOUT_ROW_MAJOR;
  482. }
  483. uploadDesc.Flags = D3D12_RESOURCE_FLAG_NONE;
  484. CHECK_HR(m_pDevice->CreateCommittedResource(
  485. &upload, D3D12_HEAP_FLAG_NONE, &uploadDesc,
  486. D3D12_RESOURCE_STATE_GENERIC_READ, nullptr,
  487. IID_PPV_ARGS(&pIntermediate)));
  488. intermediates.push_back(pIntermediate);
  489. char uploadObjectName[128];
  490. if (R.Name && SUCCEEDED(StringCchPrintfA(
  491. uploadObjectName, _countof(uploadObjectName),
  492. "Upload resource for %s", R.Name))) {
  493. SetObjectName(pIntermediate, uploadObjectName);
  494. }
  495. D3D12_SUBRESOURCE_DATA transferData;
  496. transferData.pData = values.data();
  497. transferData.RowPitch = values.size() / R.Desc.Height;
  498. transferData.SlicePitch = values.size();
  499. UpdateSubresources<1>(pList, pResource.p, pIntermediate.p, 0, 0, 1,
  500. &transferData);
  501. }
  502. if (R.ReadBack) {
  503. CComPtr<ID3D12Resource> pReadbackResource;
  504. CD3DX12_HEAP_PROPERTIES readback(D3D12_HEAP_TYPE_READBACK);
  505. UINT64 width = R.Desc.Dimension == D3D12_RESOURCE_DIMENSION_BUFFER
  506. ? R.Desc.Width
  507. : (R.Desc.Height * R.Desc.Width *
  508. GetByteSizeForFormat(R.Desc.Format));
  509. CD3DX12_RESOURCE_DESC readbackDesc(CD3DX12_RESOURCE_DESC::Buffer(width));
  510. readbackDesc.Flags = D3D12_RESOURCE_FLAG_NONE;
  511. CHECK_HR(m_pDevice->CreateCommittedResource(
  512. &readback, D3D12_HEAP_FLAG_NONE, &readbackDesc,
  513. D3D12_RESOURCE_STATE_COPY_DEST, nullptr,
  514. IID_PPV_ARGS(&pReadbackResource)));
  515. D.ReadBack = pReadbackResource;
  516. char readbackObjectName[128];
  517. if (R.Name && SUCCEEDED(StringCchPrintfA(
  518. readbackObjectName, _countof(readbackObjectName),
  519. "Readback resource for %s", R.Name))) {
  520. SetObjectName(pReadbackResource, readbackObjectName);
  521. }
  522. }
  523. if (R.TransitionTo != D.ResourceState) {
  524. RecordTransitionBarrier(pList, D.Resource, D.ResourceState,
  525. R.TransitionTo);
  526. D.ResourceState = R.TransitionTo;
  527. }
  528. }
  529. // Create a buffer to receive query results.
  530. {
  531. CComPtr<ID3D12Resource> pReadbackResource;
  532. CD3DX12_HEAP_PROPERTIES readback(D3D12_HEAP_TYPE_READBACK);
  533. CD3DX12_RESOURCE_DESC readbackDesc(CD3DX12_RESOURCE_DESC::Buffer(sizeof(D3D12_QUERY_DATA_PIPELINE_STATISTICS)));
  534. CHECK_HR(m_pDevice->CreateCommittedResource(
  535. &readback, D3D12_HEAP_FLAG_NONE, &readbackDesc,
  536. D3D12_RESOURCE_STATE_COPY_DEST, nullptr,
  537. IID_PPV_ARGS(&m_pQueryBuffer)));
  538. SetObjectName(m_pQueryBuffer, "Query Pipeline Readback Buffer");
  539. }
  540. CHECK_HR(pList->Close());
  541. ExecuteCommandList(ResCommandList.Queue, pList);
  542. WaitForSignal(ResCommandList.Queue, m_pFence, m_hFence, m_FenceValue++);
  543. }
  544. void ShaderOpTest::CreateRootSignature() {
  545. if (m_pShaderOp->RootSignature == nullptr) {
  546. AtlThrow(E_INVALIDARG);
  547. }
  548. CComPtr<ID3DBlob> pCode;
  549. CComPtr<ID3DBlob> pRootSignatureBlob;
  550. CComPtr<ID3DBlob> pError;
  551. std::string sQuoted;
  552. sQuoted.reserve(2 + strlen(m_pShaderOp->RootSignature) + 1);
  553. sQuoted.append("\"");
  554. sQuoted.append(m_pShaderOp->RootSignature);
  555. sQuoted.append("\"");
  556. char *ch = (char *)sQuoted.data();
  557. while (*ch) {
  558. if (*ch == '\r' || *ch == '\n') *ch = ' ';
  559. ++ch;
  560. }
  561. D3D_SHADER_MACRO M[2] = {
  562. { "RootSigVal", sQuoted.c_str() },
  563. { nullptr, nullptr }
  564. };
  565. HRESULT hr = D3DCompile(nullptr, 0, "RootSigShader", M, nullptr, sQuoted.c_str(),
  566. "rootsig_1_0", 0, 0, &pCode, &pError);
  567. if (FAILED(hr) && pError != nullptr) {
  568. ShaderOpLogFmt(L"Failed to compile root signature:\r\n%*S",
  569. (int)pError->GetBufferSize(),
  570. (LPCSTR)pError->GetBufferPointer());
  571. }
  572. CHECK_HR(hr);
  573. CHECK_HR(D3DGetBlobPart(pCode->GetBufferPointer(), pCode->GetBufferSize(),
  574. D3D_BLOB_ROOT_SIGNATURE, 0, &pRootSignatureBlob));
  575. CHECK_HR(m_pDevice->CreateRootSignature(
  576. 0, pRootSignatureBlob->GetBufferPointer(),
  577. pRootSignatureBlob->GetBufferSize(), IID_PPV_ARGS(&m_pRootSignature)));
  578. }
  579. static bool TargetUsesDxil(LPCSTR pText) {
  580. return (strlen(pText) > 3) && pText[3] >= '6'; // xx_6xx
  581. }
  582. static void splitWStringIntoVectors(LPWSTR str, wchar_t delim, std::vector<LPWSTR> &list) {
  583. if (str) {
  584. LPWSTR cur = str;
  585. list.push_back(cur);
  586. while (*cur != L'\0') {
  587. if (*cur == delim) {
  588. list.push_back(cur+1);
  589. *(cur) = L'\0';
  590. }
  591. cur++;
  592. }
  593. }
  594. }
  595. void ShaderOpTest::CreateShaders() {
  596. for (ShaderOpShader &S : m_pShaderOp->Shaders) {
  597. CComPtr<ID3DBlob> pCode;
  598. HRESULT hr = S_OK;
  599. LPCSTR pText = m_pShaderOp->GetShaderText(&S);
  600. if (S.Compiled) {
  601. int textLen = (int)strlen(pText);
  602. int decodedLen = Base64DecodeGetRequiredLength(textLen);
  603. // Length is an approximation, so we can't creat the final blob yet.
  604. std::vector<BYTE> decoded;
  605. decoded.resize(decodedLen);
  606. if (!Base64Decode(pText, textLen, decoded.data(), &decodedLen)) {
  607. ShaderOpLogFmt(L"Failed to decode compiled shader: %S\r\n", S.Name);
  608. CHECK_HR(E_FAIL);
  609. }
  610. // decodedLen should have the correct size now.
  611. CHECK_HR(D3DCreateBlob(decodedLen, &pCode));
  612. memcpy(pCode->GetBufferPointer(), decoded.data(), decodedLen);
  613. }
  614. else if (TargetUsesDxil(S.Target)) {
  615. CComPtr<IDxcCompiler> pCompiler;
  616. CComPtr<IDxcLibrary> pLibrary;
  617. CComPtr<IDxcBlobEncoding> pTextBlob;
  618. CComPtr<IDxcOperationResult> pResult;
  619. CA2W nameW(S.Name, CP_UTF8);
  620. CA2W entryPointW(S.EntryPoint, CP_UTF8);
  621. CA2W targetW(S.Target, CP_UTF8);
  622. CA2W argumentsW(S.Arguments, CP_UTF8);
  623. std::vector<LPWSTR> argumentsWList;
  624. splitWStringIntoVectors(argumentsW, L' ', argumentsWList);
  625. HRESULT resultCode;
  626. CHECK_HR(m_pDxcSupport->CreateInstance(CLSID_DxcLibrary, &pLibrary));
  627. CHECK_HR(pLibrary->CreateBlobWithEncodingFromPinned(
  628. pText, (UINT32)strlen(pText), CP_UTF8, &pTextBlob));
  629. CHECK_HR(m_pDxcSupport->CreateInstance(CLSID_DxcCompiler, &pCompiler));
  630. CHECK_HR(pCompiler->Compile(pTextBlob, nameW, entryPointW, targetW,
  631. (LPCWSTR *)argumentsWList.data(), (UINT32)argumentsWList.size(),
  632. nullptr, 0,
  633. nullptr, &pResult));
  634. CHECK_HR(pResult->GetStatus(&resultCode));
  635. if (FAILED(resultCode)) {
  636. CComPtr<IDxcBlobEncoding> errors;
  637. CHECK_HR(pResult->GetErrorBuffer(&errors));
  638. ShaderOpLogFmt(L"Failed to compile shader: %*S\r\n",
  639. (int)errors->GetBufferSize(),
  640. errors->GetBufferPointer());
  641. }
  642. CHECK_HR(resultCode);
  643. CHECK_HR(pResult->GetResult((IDxcBlob **)&pCode));
  644. #if 0 // use the following code to test compiled shader
  645. CComPtr<IDxcBlob> pCode;
  646. CHECK_HR(pResult->GetResult(&pCode));
  647. CComPtr<IDxcBlobEncoding> pBlob;
  648. CHECK_HR(pCompiler->Disassemble((IDxcBlob *)pCode, (IDxcBlobEncoding **)&pBlob));
  649. CComPtr<IDxcBlobEncoding> pUtf16Blob;
  650. pLibrary->GetBlobAsUtf16(pBlob, &pUtf16Blob);
  651. hlsl_test::LogCommentFmt(L"%*s", (int)pUtf16Blob->GetBufferSize() / 2, (LPCWSTR)pUtf16Blob->GetBufferPointer());
  652. #endif
  653. } else {
  654. CComPtr<ID3DBlob> pError;
  655. hr = D3DCompile(pText, strlen(pText), S.Name, nullptr, nullptr,
  656. S.EntryPoint, S.Target, 0, 0, &pCode, &pError);
  657. if (FAILED(hr) && pError != nullptr) {
  658. ShaderOpLogFmt(L"%*S\r\n", (int)pError->GetBufferSize(),
  659. ((LPCSTR)pError->GetBufferPointer()));
  660. }
  661. }
  662. CHECK_HR(hr);
  663. m_Shaders[S.Name] = pCode;
  664. }
  665. }
  666. void ShaderOpTest::GetPipelineStats(D3D12_QUERY_DATA_PIPELINE_STATISTICS *pStats) {
  667. MappedData M;
  668. M.reset(m_pQueryBuffer, sizeof(*pStats));
  669. memcpy(pStats, M.data(), sizeof(*pStats));
  670. }
  671. void ShaderOpTest::GetReadBackData(LPCSTR pResourceName, MappedData *pData) {
  672. pResourceName = m_pShaderOp->Strings.insert(pResourceName); // Unique
  673. ShaderOpResourceData &D = m_ResourceData.at(pResourceName);
  674. D3D12_RESOURCE_DESC Desc = D.ReadBack->GetDesc();
  675. UINT32 sizeInBytes = (UINT32)Desc.Width;
  676. pData->reset(D.ReadBack, sizeInBytes);
  677. }
  678. static void SetDescriptorHeaps(ID3D12GraphicsCommandList *pList,
  679. std::vector<ID3D12DescriptorHeap *> &heaps) {
  680. if (heaps.empty())
  681. return;
  682. std::vector<ID3D12DescriptorHeap *> localHeaps;
  683. for (auto &H : heaps) {
  684. if (H->GetDesc().Flags & D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE) {
  685. localHeaps.push_back(H);
  686. }
  687. }
  688. if (!localHeaps.empty())
  689. pList->SetDescriptorHeaps((UINT)localHeaps.size(), localHeaps.data());
  690. }
  691. void ShaderOpTest::RunCommandList() {
  692. ID3D12GraphicsCommandList *pList = m_CommandList.List.p;
  693. if (m_pShaderOp->IsCompute()) {
  694. pList->SetPipelineState(m_pPSO);
  695. pList->SetComputeRootSignature(m_pRootSignature);
  696. SetDescriptorHeaps(pList, m_DescriptorHeaps);
  697. SetRootValues(pList, m_pShaderOp->IsCompute());
  698. pList->Dispatch(m_pShaderOp->DispatchX, m_pShaderOp->DispatchY,
  699. m_pShaderOp->DispatchZ);
  700. } else {
  701. pList->SetPipelineState(m_pPSO);
  702. pList->SetGraphicsRootSignature(m_pRootSignature);
  703. SetDescriptorHeaps(pList, m_DescriptorHeaps);
  704. SetRootValues(pList, m_pShaderOp->IsCompute());
  705. if (!m_pShaderOp->RenderTargets.empty()) {
  706. // Use the first render target to set up the viewport and scissors.
  707. ShaderOpResource *R = m_pShaderOp->GetResourceByName(m_pShaderOp->RenderTargets[0]);
  708. D3D12_VIEWPORT viewport;
  709. D3D12_RECT scissorRect;
  710. memset(&viewport, 0, sizeof(viewport));
  711. viewport.Height = (FLOAT)R->Desc.Height;
  712. viewport.Width = (FLOAT)R->Desc.Width;
  713. viewport.MaxDepth = 1.0f;
  714. memset(&scissorRect, 0, sizeof(scissorRect));
  715. scissorRect.right = (LONG)viewport.Width;
  716. scissorRect.bottom = (LONG)viewport.Height;
  717. pList->RSSetViewports(1, &viewport);
  718. pList->RSSetScissorRects(1, &scissorRect);
  719. }
  720. // Indicate that the buffers will be used as render targets.
  721. D3D12_CPU_DESCRIPTOR_HANDLE rtvHandles[8];
  722. UINT rtvHandleCount = (UINT)m_pShaderOp->RenderTargets.size();
  723. for (size_t i = 0; i < rtvHandleCount; ++i) {
  724. auto &rt = m_pShaderOp->RenderTargets[i];
  725. ShaderOpDescriptorData &DData = m_DescriptorData[rt];
  726. rtvHandles[i] = DData.CPUHandle;
  727. RecordTransitionBarrier(pList, DData.ResData->Resource,
  728. DData.ResData->ResourceState,
  729. D3D12_RESOURCE_STATE_RENDER_TARGET);
  730. DData.ResData->ResourceState = D3D12_RESOURCE_STATE_RENDER_TARGET;
  731. }
  732. pList->OMSetRenderTargets(rtvHandleCount, rtvHandles, FALSE, nullptr);
  733. const float ClearColor[4] = { 0.0f, 0.2f, 0.4f, 1.0f };
  734. pList->ClearRenderTargetView(rtvHandles[0], ClearColor, 0, nullptr);
  735. // TODO: set all of this from m_pShaderOp.
  736. ShaderOpResourceData &VBufferData = this->m_ResourceData[m_pShaderOp->Strings.insert("VBuffer")];
  737. D3D_PRIMITIVE_TOPOLOGY topology = D3D_PRIMITIVE_TOPOLOGY_UNDEFINED;
  738. for (ShaderOpResource &resource : m_pShaderOp->Resources) {
  739. if (_strcmpi(resource.Name, "VBuffer") == 0) {
  740. topology = resource.PrimitiveTopology;
  741. break;
  742. }
  743. }
  744. pList->IASetPrimitiveTopology(topology);
  745. // Calculate the stride in bytes from the inputs, assuming linear & contiguous.
  746. UINT strideInBytes = 0;
  747. for (auto && IE : m_pShaderOp->InputElements) {
  748. strideInBytes += GetByteSizeForFormat(IE.Format);
  749. }
  750. D3D12_VERTEX_BUFFER_VIEW vertexBufferView;
  751. vertexBufferView.BufferLocation = VBufferData.Resource->GetGPUVirtualAddress();
  752. vertexBufferView.StrideInBytes = strideInBytes;
  753. vertexBufferView.SizeInBytes = (UINT)VBufferData.ShaderOpRes->Desc.Width;
  754. pList->IASetVertexBuffers(0, 1, &vertexBufferView);
  755. UINT vertexCount = vertexBufferView.SizeInBytes / vertexBufferView.StrideInBytes;
  756. UINT instanceCount = 1;
  757. UINT vertexCountPerInstance = vertexCount / instanceCount;
  758. pList->BeginQuery(m_pQueryHeap, D3D12_QUERY_TYPE_PIPELINE_STATISTICS, 0);
  759. pList->DrawInstanced(vertexCountPerInstance, instanceCount, 0, 0);
  760. pList->EndQuery(m_pQueryHeap, D3D12_QUERY_TYPE_PIPELINE_STATISTICS, 0);
  761. pList->ResolveQueryData(m_pQueryHeap, D3D12_QUERY_TYPE_PIPELINE_STATISTICS,
  762. 0, 1, m_pQueryBuffer, 0);
  763. }
  764. CHECK_HR(pList->Close());
  765. ExecuteCommandList(m_CommandList.Queue, pList);
  766. WaitForSignal(m_CommandList.Queue, m_pFence, m_hFence, m_FenceValue++);
  767. }
  768. void ShaderOpTest::RunShaderOp(ShaderOp *pShaderOp) {
  769. m_pShaderOp = pShaderOp;
  770. CreateDevice();
  771. CreateResources();
  772. CreateDescriptorHeaps();
  773. CreatePipelineState();
  774. CreateCommandList();
  775. RunCommandList();
  776. CopyBackResources();
  777. }
  778. void ShaderOpTest::RunShaderOp(std::shared_ptr<ShaderOp> ShaderOp) {
  779. m_OrigShaderOp = ShaderOp;
  780. RunShaderOp(m_OrigShaderOp.get());
  781. }
  782. void ShaderOpTest::SetRootValues(ID3D12GraphicsCommandList *pList,
  783. bool isCompute) {
  784. for (size_t i = 0; i < m_pShaderOp->RootValues.size(); ++i) {
  785. ShaderOpRootValue &V = m_pShaderOp->RootValues[i];
  786. UINT idx = V.Index == 0 ? (UINT)i : V.Index;
  787. if (V.ResName) {
  788. auto r_it = m_ResourceData.find(V.ResName);
  789. if (r_it == m_ResourceData.end()) {
  790. ShaderOpLogFmt(L"Root value #%u refers to missing resource %S", (unsigned)i, V.ResName);
  791. CHECK_HR(E_INVALIDARG);
  792. }
  793. // Issue a warning for trying to bind textures (GPU address will return null)
  794. ShaderOpResourceData &D = r_it->second;
  795. ID3D12Resource *pRes = D.Resource;
  796. if (isCompute) {
  797. switch (D.ShaderOpRes->TransitionTo) {
  798. case D3D12_RESOURCE_STATE_VERTEX_AND_CONSTANT_BUFFER:
  799. pList->SetComputeRootConstantBufferView(idx,
  800. pRes->GetGPUVirtualAddress());
  801. break;
  802. case D3D12_RESOURCE_STATE_UNORDERED_ACCESS:
  803. pList->SetComputeRootUnorderedAccessView(idx,
  804. pRes->GetGPUVirtualAddress());
  805. break;
  806. case D3D12_RESOURCE_STATE_NON_PIXEL_SHADER_RESOURCE:
  807. default:
  808. pList->SetComputeRootShaderResourceView(idx,
  809. pRes->GetGPUVirtualAddress());
  810. break;
  811. }
  812. }
  813. else {
  814. switch (D.ShaderOpRes->TransitionTo) {
  815. case D3D12_RESOURCE_STATE_VERTEX_AND_CONSTANT_BUFFER:
  816. pList->SetGraphicsRootConstantBufferView(idx,
  817. pRes->GetGPUVirtualAddress());
  818. break;
  819. case D3D12_RESOURCE_STATE_UNORDERED_ACCESS:
  820. pList->SetGraphicsRootUnorderedAccessView(idx,
  821. pRes->GetGPUVirtualAddress());
  822. break;
  823. case D3D12_RESOURCE_STATE_NON_PIXEL_SHADER_RESOURCE:
  824. default:
  825. pList->SetGraphicsRootShaderResourceView(idx,
  826. pRes->GetGPUVirtualAddress());
  827. break;
  828. }
  829. }
  830. }
  831. else if (V.HeapName) {
  832. D3D12_GPU_DESCRIPTOR_HANDLE heapBase(m_DescriptorHeapsByName[V.HeapName]->GetGPUDescriptorHandleForHeapStart());
  833. if (isCompute) {
  834. pList->SetComputeRootDescriptorTable(idx, heapBase);
  835. }
  836. else {
  837. pList->SetGraphicsRootDescriptorTable(idx, heapBase);
  838. }
  839. }
  840. }
  841. }
  842. void ShaderOpTest::SetDevice(ID3D12Device *pDevice) {
  843. m_pDevice = pDevice;
  844. }
  845. void ShaderOpTest::SetDxcSupport(dxc::DxcDllSupport *pDxcSupport) {
  846. m_pDxcSupport = pDxcSupport;
  847. }
  848. void ShaderOpTest::SetInitCallback(TInitCallbackFn InitCallbackFn) {
  849. m_InitCallbackFn = InitCallbackFn;
  850. }
  851. void ShaderOpTest::SetupRenderTarget(ShaderOp *pShaderOp, ID3D12Device *pDevice,
  852. ID3D12CommandQueue *pCommandQueue,
  853. ID3D12Resource *pRenderTarget) {
  854. SetDevice(pDevice);
  855. m_CommandList.Queue = pCommandQueue;
  856. // Simplification - add the render target name if missing, set it up 'by hand' if not.
  857. if (pShaderOp->RenderTargets.empty()) {
  858. pShaderOp->RenderTargets.push_back(pShaderOp->Strings.insert("RTarget"));
  859. ShaderOpResource R;
  860. ZeroMemory(&R, sizeof(R));
  861. R.Desc = pRenderTarget->GetDesc();
  862. R.Name = pShaderOp->Strings.insert("RTarget");
  863. R.HeapFlags = D3D12_HEAP_FLAG_NONE;
  864. R.Init = nullptr;
  865. R.InitialResourceState = D3D12_RESOURCE_STATE_PRESENT;
  866. R.ReadBack = FALSE;
  867. pShaderOp->Resources.push_back(R);
  868. ShaderOpResourceData &D = m_ResourceData[R.Name];
  869. D.ShaderOpRes = &pShaderOp->Resources.back();
  870. D.Resource = pRenderTarget;
  871. D.ResourceState = R.InitialResourceState;
  872. }
  873. // Create a render target heap to put this in.
  874. ShaderOpDescriptorHeap *pRtvHeap = pShaderOp->GetDescriptorHeapByName("RtvHeap");
  875. if (pRtvHeap == nullptr) {
  876. ShaderOpDescriptorHeap H;
  877. ZeroMemory(&H, sizeof(H));
  878. H.Name = pShaderOp->Strings.insert("RtvHeap");
  879. H.Desc.NumDescriptors = 1;
  880. H.Desc.Type = D3D12_DESCRIPTOR_HEAP_TYPE_RTV;
  881. pShaderOp->DescriptorHeaps.push_back(H);
  882. pRtvHeap = &pShaderOp->DescriptorHeaps.back();
  883. }
  884. if (pRtvHeap->Descriptors.empty()) {
  885. ShaderOpDescriptor D;
  886. ZeroMemory(&D, sizeof(D));
  887. D.Name = pShaderOp->Strings.insert("RTarget");
  888. D.ResName = D.Name;
  889. D.Kind = pShaderOp->Strings.insert("RTV");
  890. pRtvHeap->Descriptors.push_back(D);
  891. }
  892. }
  893. void ShaderOpTest::PresentRenderTarget(ShaderOp *pShaderOp,
  894. ID3D12CommandQueue *pCommandQueue,
  895. ID3D12Resource *pRenderTarget) {
  896. UNREFERENCED_PARAMETER(pShaderOp);
  897. CommandListRefs ResCommandList;
  898. ResCommandList.Queue = pCommandQueue;
  899. ResCommandList.CreateForDevice(m_pDevice, m_pShaderOp->IsCompute());
  900. ID3D12GraphicsCommandList *pList = ResCommandList.List;
  901. pList->SetName(L"ShaderOpTest Resource Present CommandList");
  902. RecordTransitionBarrier(pList, pRenderTarget,
  903. D3D12_RESOURCE_STATE_RENDER_TARGET,
  904. D3D12_RESOURCE_STATE_PRESENT);
  905. pList->Close();
  906. ExecuteCommandList(ResCommandList.Queue, pList);
  907. WaitForSignal(ResCommandList.Queue, m_pFence, m_hFence, m_FenceValue++);
  908. }
  909. ShaderOp *ShaderOpSet::GetShaderOp(LPCSTR pName) {
  910. for (auto &S : ShaderOps) {
  911. if (S->Name && 0 == _stricmp(pName, S->Name)) {
  912. return S.get();
  913. }
  914. }
  915. return nullptr;
  916. }
  917. ///////////////////////////////////////////////////////////////////////////////
  918. // ShaderOpTest library implementation for deserialization.
  919. #pragma region Parsing support
  920. // Use this class to initialize a ShaderOp object from an XML document.
  921. class ShaderOpParser {
  922. private:
  923. string_table *m_pStrings;
  924. bool ReadAtElementName(IXmlReader *pReader, LPCWSTR pName);
  925. HRESULT ReadAttrStr(IXmlReader *pReader, LPCWSTR pAttrName, LPCSTR *ppValue);
  926. HRESULT ReadAttrBOOL(IXmlReader *pReader, LPCWSTR pAttrName, BOOL *pValue, BOOL defaultValue = FALSE);
  927. HRESULT ReadAttrUINT64(IXmlReader *pReader, LPCWSTR pAttrName, UINT64 *pValue, UINT64 defaultValue = 0);
  928. HRESULT ReadAttrUINT16(IXmlReader *pReader, LPCWSTR pAttrName, UINT16 *pValue, UINT16 defaultValue = 0);
  929. HRESULT ReadAttrUINT(IXmlReader *pReader, LPCWSTR pAttrName, UINT *pValue, UINT defaultValue = 0);
  930. void ReadElementContentStr(IXmlReader *pReader, LPCSTR *ppValue);
  931. void ParseDescriptor(IXmlReader *pReader, ShaderOpDescriptor *pDesc);
  932. void ParseDescriptorHeap(IXmlReader *pReader, ShaderOpDescriptorHeap *pHeap);
  933. void ParseInputElement(IXmlReader *pReader, D3D12_INPUT_ELEMENT_DESC *pInputElement);
  934. void ParseInputElements(IXmlReader *pReader, std::vector<D3D12_INPUT_ELEMENT_DESC> *pInputElements);
  935. void ParseRenderTargets(IXmlReader *pReader, std::vector<LPCSTR> *pRenderTargets);
  936. void ParseRootValue(IXmlReader *pReader, ShaderOpRootValue *pRootValue);
  937. void ParseRootValues(IXmlReader *pReader, std::vector<ShaderOpRootValue> *pRootValues);
  938. void ParseResource(IXmlReader *pReader, ShaderOpResource *pResource);
  939. void ParseShader(IXmlReader *pReader, ShaderOpShader *pShader);
  940. public:
  941. void ParseShaderOpSet(IStream *pStream, ShaderOpSet *pShaderOpSet);
  942. void ParseShaderOpSet(IXmlReader *pReader, ShaderOpSet *pShaderOpSet);
  943. void ParseShaderOp(IXmlReader *pReader, ShaderOp *pShaderOp);
  944. };
  945. void ParseShaderOpSetFromStream(IStream *pStream, st::ShaderOpSet *pShaderOpSet) {
  946. ShaderOpParser parser;
  947. parser.ParseShaderOpSet(pStream, pShaderOpSet);
  948. }
  949. void ParseShaderOpSetFromXml(IXmlReader *pReader, st::ShaderOpSet *pShaderOpSet) {
  950. ShaderOpParser parser;
  951. parser.ParseShaderOpSet(pReader, pShaderOpSet);
  952. }
  953. enum class ParserEnumKind {
  954. INPUT_CLASSIFICATION,
  955. DXGI_FORMAT,
  956. HEAP_TYPE,
  957. CPU_PAGE_PROPERTY,
  958. MEMORY_POOL,
  959. RESOURCE_DIMENSION,
  960. TEXTURE_LAYOUT,
  961. RESOURCE_FLAG,
  962. HEAP_FLAG,
  963. RESOURCE_STATE,
  964. DESCRIPTOR_HEAP_TYPE,
  965. DESCRIPTOR_HEAP_FLAG,
  966. SRV_DIMENSION,
  967. UAV_DIMENSION,
  968. PRIMITIVE_TOPOLOGY,
  969. PRIMITIVE_TOPOLOGY_TYPE
  970. };
  971. struct ParserEnumValue {
  972. LPCWSTR Name;
  973. UINT Value;
  974. };
  975. struct ParserEnumTable {
  976. size_t ValueCount;
  977. const ParserEnumValue *Values;
  978. ParserEnumKind Kind;
  979. };
  980. static const ParserEnumValue INPUT_CLASSIFICATION_TABLE[] = {
  981. { L"INSTANCE", D3D12_INPUT_CLASSIFICATION_PER_INSTANCE_DATA },
  982. { L"VERTEX", D3D12_INPUT_CLASSIFICATION_PER_VERTEX_DATA }
  983. };
  984. static const ParserEnumValue DXGI_FORMAT_TABLE[] = {
  985. { L"UNKNOWN", DXGI_FORMAT_UNKNOWN },
  986. { L"R32G32B32A32_TYPELESS", DXGI_FORMAT_R32G32B32A32_TYPELESS },
  987. { L"R32G32B32A32_FLOAT", DXGI_FORMAT_R32G32B32A32_FLOAT },
  988. { L"R32G32B32A32_UINT", DXGI_FORMAT_R32G32B32A32_UINT },
  989. { L"R32G32B32A32_SINT", DXGI_FORMAT_R32G32B32A32_SINT },
  990. { L"R32G32B32_TYPELESS", DXGI_FORMAT_R32G32B32_TYPELESS },
  991. { L"R32G32B32_FLOAT", DXGI_FORMAT_R32G32B32_FLOAT },
  992. { L"R32G32B32_UINT", DXGI_FORMAT_R32G32B32_UINT },
  993. { L"R32G32B32_SINT", DXGI_FORMAT_R32G32B32_SINT },
  994. { L"R16G16B16A16_TYPELESS", DXGI_FORMAT_R16G16B16A16_TYPELESS },
  995. { L"R16G16B16A16_FLOAT", DXGI_FORMAT_R16G16B16A16_FLOAT },
  996. { L"R16G16B16A16_UNORM", DXGI_FORMAT_R16G16B16A16_UNORM },
  997. { L"R16G16B16A16_UINT", DXGI_FORMAT_R16G16B16A16_UINT },
  998. { L"R16G16B16A16_SNORM", DXGI_FORMAT_R16G16B16A16_SNORM },
  999. { L"R16G16B16A16_SINT", DXGI_FORMAT_R16G16B16A16_SINT },
  1000. { L"R32G32_TYPELESS", DXGI_FORMAT_R32G32_TYPELESS },
  1001. { L"R32G32_FLOAT", DXGI_FORMAT_R32G32_FLOAT },
  1002. { L"R32G32_UINT", DXGI_FORMAT_R32G32_UINT },
  1003. { L"R32G32_SINT", DXGI_FORMAT_R32G32_SINT },
  1004. { L"R32G8X24_TYPELESS", DXGI_FORMAT_R32G8X24_TYPELESS },
  1005. { L"D32_FLOAT_S8X24_UINT", DXGI_FORMAT_D32_FLOAT_S8X24_UINT },
  1006. { L"R32_FLOAT_X8X24_TYPELESS", DXGI_FORMAT_R32_FLOAT_X8X24_TYPELESS },
  1007. { L"X32_TYPELESS_G8X24_UINT", DXGI_FORMAT_X32_TYPELESS_G8X24_UINT },
  1008. { L"R10G10B10A2_TYPELESS", DXGI_FORMAT_R10G10B10A2_TYPELESS },
  1009. { L"R10G10B10A2_UNORM", DXGI_FORMAT_R10G10B10A2_UNORM },
  1010. { L"R10G10B10A2_UINT", DXGI_FORMAT_R10G10B10A2_UINT },
  1011. { L"R11G11B10_FLOAT", DXGI_FORMAT_R11G11B10_FLOAT },
  1012. { L"R8G8B8A8_TYPELESS", DXGI_FORMAT_R8G8B8A8_TYPELESS },
  1013. { L"R8G8B8A8_UNORM", DXGI_FORMAT_R8G8B8A8_UNORM },
  1014. { L"R8G8B8A8_UNORM_SRGB", DXGI_FORMAT_R8G8B8A8_UNORM_SRGB },
  1015. { L"R8G8B8A8_UINT", DXGI_FORMAT_R8G8B8A8_UINT },
  1016. { L"R8G8B8A8_SNORM", DXGI_FORMAT_R8G8B8A8_SNORM },
  1017. { L"R8G8B8A8_SINT", DXGI_FORMAT_R8G8B8A8_SINT },
  1018. { L"R16G16_TYPELESS", DXGI_FORMAT_R16G16_TYPELESS },
  1019. { L"R16G16_FLOAT", DXGI_FORMAT_R16G16_FLOAT },
  1020. { L"R16G16_UNORM", DXGI_FORMAT_R16G16_UNORM },
  1021. { L"R16G16_UINT", DXGI_FORMAT_R16G16_UINT },
  1022. { L"R16G16_SNORM", DXGI_FORMAT_R16G16_SNORM },
  1023. { L"R16G16_SINT", DXGI_FORMAT_R16G16_SINT },
  1024. { L"R32_TYPELESS", DXGI_FORMAT_R32_TYPELESS },
  1025. { L"D32_FLOAT", DXGI_FORMAT_D32_FLOAT },
  1026. { L"R32_FLOAT", DXGI_FORMAT_R32_FLOAT },
  1027. { L"R32_UINT", DXGI_FORMAT_R32_UINT },
  1028. { L"R32_SINT", DXGI_FORMAT_R32_SINT },
  1029. { L"R24G8_TYPELESS", DXGI_FORMAT_R24G8_TYPELESS },
  1030. { L"D24_UNORM_S8_UINT", DXGI_FORMAT_D24_UNORM_S8_UINT },
  1031. { L"R24_UNORM_X8_TYPELESS", DXGI_FORMAT_R24_UNORM_X8_TYPELESS },
  1032. { L"X24_TYPELESS_G8_UINT", DXGI_FORMAT_X24_TYPELESS_G8_UINT },
  1033. { L"R8G8_TYPELESS", DXGI_FORMAT_R8G8_TYPELESS },
  1034. { L"R8G8_UNORM", DXGI_FORMAT_R8G8_UNORM },
  1035. { L"R8G8_UINT", DXGI_FORMAT_R8G8_UINT },
  1036. { L"R8G8_SNORM", DXGI_FORMAT_R8G8_SNORM },
  1037. { L"R8G8_SINT", DXGI_FORMAT_R8G8_SINT },
  1038. { L"R16_TYPELESS", DXGI_FORMAT_R16_TYPELESS },
  1039. { L"R16_FLOAT", DXGI_FORMAT_R16_FLOAT },
  1040. { L"D16_UNORM", DXGI_FORMAT_D16_UNORM },
  1041. { L"R16_UNORM", DXGI_FORMAT_R16_UNORM },
  1042. { L"R16_UINT", DXGI_FORMAT_R16_UINT },
  1043. { L"R16_SNORM", DXGI_FORMAT_R16_SNORM },
  1044. { L"R16_SINT", DXGI_FORMAT_R16_SINT },
  1045. { L"R8_TYPELESS", DXGI_FORMAT_R8_TYPELESS },
  1046. { L"R8_UNORM", DXGI_FORMAT_R8_UNORM },
  1047. { L"R8_UINT", DXGI_FORMAT_R8_UINT },
  1048. { L"R8_SNORM", DXGI_FORMAT_R8_SNORM },
  1049. { L"R8_SINT", DXGI_FORMAT_R8_SINT },
  1050. { L"A8_UNORM", DXGI_FORMAT_A8_UNORM },
  1051. { L"R1_UNORM", DXGI_FORMAT_R1_UNORM },
  1052. { L"R9G9B9E5_SHAREDEXP", DXGI_FORMAT_R9G9B9E5_SHAREDEXP },
  1053. { L"R8G8_B8G8_UNORM", DXGI_FORMAT_R8G8_B8G8_UNORM },
  1054. { L"G8R8_G8B8_UNORM", DXGI_FORMAT_G8R8_G8B8_UNORM },
  1055. { L"BC1_TYPELESS", DXGI_FORMAT_BC1_TYPELESS },
  1056. { L"BC1_UNORM", DXGI_FORMAT_BC1_UNORM },
  1057. { L"BC1_UNORM_SRGB", DXGI_FORMAT_BC1_UNORM_SRGB },
  1058. { L"BC2_TYPELESS", DXGI_FORMAT_BC2_TYPELESS },
  1059. { L"BC2_UNORM", DXGI_FORMAT_BC2_UNORM },
  1060. { L"BC2_UNORM_SRGB", DXGI_FORMAT_BC2_UNORM_SRGB },
  1061. { L"BC3_TYPELESS", DXGI_FORMAT_BC3_TYPELESS },
  1062. { L"BC3_UNORM", DXGI_FORMAT_BC3_UNORM },
  1063. { L"BC3_UNORM_SRGB", DXGI_FORMAT_BC3_UNORM_SRGB },
  1064. { L"BC4_TYPELESS", DXGI_FORMAT_BC4_TYPELESS },
  1065. { L"BC4_UNORM", DXGI_FORMAT_BC4_UNORM },
  1066. { L"BC4_SNORM", DXGI_FORMAT_BC4_SNORM },
  1067. { L"BC5_TYPELESS", DXGI_FORMAT_BC5_TYPELESS },
  1068. { L"BC5_UNORM", DXGI_FORMAT_BC5_UNORM },
  1069. { L"BC5_SNORM", DXGI_FORMAT_BC5_SNORM },
  1070. { L"B5G6R5_UNORM", DXGI_FORMAT_B5G6R5_UNORM },
  1071. { L"B5G5R5A1_UNORM", DXGI_FORMAT_B5G5R5A1_UNORM },
  1072. { L"B8G8R8A8_UNORM", DXGI_FORMAT_B8G8R8A8_UNORM },
  1073. { L"B8G8R8X8_UNORM", DXGI_FORMAT_B8G8R8X8_UNORM },
  1074. { L"R10G10B10_XR_BIAS_A2_UNORM", DXGI_FORMAT_R10G10B10_XR_BIAS_A2_UNORM },
  1075. { L"B8G8R8A8_TYPELESS", DXGI_FORMAT_B8G8R8A8_TYPELESS },
  1076. { L"B8G8R8A8_UNORM_SRGB", DXGI_FORMAT_B8G8R8A8_UNORM_SRGB },
  1077. { L"B8G8R8X8_TYPELESS", DXGI_FORMAT_B8G8R8X8_TYPELESS },
  1078. { L"B8G8R8X8_UNORM_SRGB", DXGI_FORMAT_B8G8R8X8_UNORM_SRGB },
  1079. { L"BC6H_TYPELESS", DXGI_FORMAT_BC6H_TYPELESS },
  1080. { L"BC6H_UF16", DXGI_FORMAT_BC6H_UF16 },
  1081. { L"BC6H_SF16", DXGI_FORMAT_BC6H_SF16 },
  1082. { L"BC7_TYPELESS", DXGI_FORMAT_BC7_TYPELESS },
  1083. { L"BC7_UNORM", DXGI_FORMAT_BC7_UNORM },
  1084. { L"BC7_UNORM_SRGB", DXGI_FORMAT_BC7_UNORM_SRGB },
  1085. { L"AYUV", DXGI_FORMAT_AYUV },
  1086. { L"Y410", DXGI_FORMAT_Y410 },
  1087. { L"Y416", DXGI_FORMAT_Y416 },
  1088. { L"NV12", DXGI_FORMAT_NV12 },
  1089. { L"P010", DXGI_FORMAT_P010 },
  1090. { L"P016", DXGI_FORMAT_P016 },
  1091. { L"420_OPAQUE", DXGI_FORMAT_420_OPAQUE },
  1092. { L"YUY2", DXGI_FORMAT_YUY2 },
  1093. { L"Y210", DXGI_FORMAT_Y210 },
  1094. { L"Y216", DXGI_FORMAT_Y216 },
  1095. { L"NV11", DXGI_FORMAT_NV11 },
  1096. { L"AI44", DXGI_FORMAT_AI44 },
  1097. { L"IA44", DXGI_FORMAT_IA44 },
  1098. { L"P8", DXGI_FORMAT_P8 },
  1099. { L"A8P8", DXGI_FORMAT_A8P8 },
  1100. { L"B4G4R4A4_UNORM", DXGI_FORMAT_B4G4R4A4_UNORM },
  1101. { L"P208", DXGI_FORMAT_P208 },
  1102. { L"V208", DXGI_FORMAT_V208 },
  1103. { L"V408", DXGI_FORMAT_V408 }
  1104. };
  1105. static const ParserEnumValue HEAP_TYPE_TABLE[] = {
  1106. { L"DEFAULT", D3D12_HEAP_TYPE_DEFAULT },
  1107. { L"UPLOAD", D3D12_HEAP_TYPE_UPLOAD },
  1108. { L"READBACK", D3D12_HEAP_TYPE_READBACK },
  1109. { L"CUSTOM", D3D12_HEAP_TYPE_CUSTOM }
  1110. };
  1111. static const ParserEnumValue CPU_PAGE_PROPERTY_TABLE[] = {
  1112. { L"UNKNOWN", D3D12_CPU_PAGE_PROPERTY_UNKNOWN },
  1113. { L"NOT_AVAILABLE", D3D12_CPU_PAGE_PROPERTY_NOT_AVAILABLE },
  1114. { L"WRITE_COMBINE", D3D12_CPU_PAGE_PROPERTY_WRITE_COMBINE },
  1115. { L"WRITE_BACK", D3D12_CPU_PAGE_PROPERTY_WRITE_BACK }
  1116. };
  1117. static const ParserEnumValue MEMORY_POOL_TABLE[] = {
  1118. { L"UNKNOWN", D3D12_MEMORY_POOL_UNKNOWN },
  1119. { L"L0 ", D3D12_MEMORY_POOL_L0 },
  1120. { L"L1", D3D12_MEMORY_POOL_L1 }
  1121. };
  1122. static const ParserEnumValue RESOURCE_DIMENSION_TABLE[] = {
  1123. { L"UNKNOWN", D3D12_RESOURCE_DIMENSION_UNKNOWN },
  1124. { L"BUFFER", D3D12_RESOURCE_DIMENSION_BUFFER },
  1125. { L"TEXTURE1D", D3D12_RESOURCE_DIMENSION_TEXTURE1D },
  1126. { L"TEXTURE2D", D3D12_RESOURCE_DIMENSION_TEXTURE2D },
  1127. { L"TEXTURE3D", D3D12_RESOURCE_DIMENSION_TEXTURE3D }
  1128. };
  1129. static const ParserEnumValue TEXTURE_LAYOUT_TABLE[] = {
  1130. { L"UNKNOWN", D3D12_TEXTURE_LAYOUT_UNKNOWN },
  1131. { L"ROW_MAJOR", D3D12_TEXTURE_LAYOUT_ROW_MAJOR },
  1132. { L"UNDEFINED_SWIZZLE", D3D12_TEXTURE_LAYOUT_64KB_UNDEFINED_SWIZZLE },
  1133. { L"STANDARD_SWIZZLE", D3D12_TEXTURE_LAYOUT_64KB_STANDARD_SWIZZLE }
  1134. };
  1135. static const ParserEnumValue RESOURCE_FLAG_TABLE[] = {
  1136. { L"NONE", D3D12_RESOURCE_FLAG_NONE },
  1137. { L"ALLOW_RENDER_TARGET", D3D12_RESOURCE_FLAG_ALLOW_RENDER_TARGET },
  1138. { L"ALLOW_DEPTH_STENCIL", D3D12_RESOURCE_FLAG_ALLOW_DEPTH_STENCIL },
  1139. { L"ALLOW_UNORDERED_ACCESS", D3D12_RESOURCE_FLAG_ALLOW_UNORDERED_ACCESS },
  1140. { L"DENY_SHADER_RESOURCE", D3D12_RESOURCE_FLAG_DENY_SHADER_RESOURCE },
  1141. { L"ALLOW_CROSS_ADAPTER", D3D12_RESOURCE_FLAG_ALLOW_CROSS_ADAPTER },
  1142. { L"ALLOW_SIMULTANEOUS_ACCESS", D3D12_RESOURCE_FLAG_ALLOW_SIMULTANEOUS_ACCESS }
  1143. };
  1144. static const ParserEnumValue HEAP_FLAG_TABLE[] = {
  1145. { L"NONE", D3D12_HEAP_FLAG_NONE },
  1146. { L"SHARED", D3D12_HEAP_FLAG_SHARED },
  1147. { L"DENY_BUFFERS", D3D12_HEAP_FLAG_DENY_BUFFERS },
  1148. { L"ALLOW_DISPLAY", D3D12_HEAP_FLAG_ALLOW_DISPLAY },
  1149. { L"SHARED_CROSS_ADAPTER", D3D12_HEAP_FLAG_SHARED_CROSS_ADAPTER },
  1150. { L"DENY_RT_DS_TEXTURES", D3D12_HEAP_FLAG_DENY_RT_DS_TEXTURES },
  1151. { L"DENY_NON_RT_DS_TEXTURES", D3D12_HEAP_FLAG_DENY_NON_RT_DS_TEXTURES },
  1152. { L"ALLOW_ALL_BUFFERS_AND_TEXTURES",D3D12_HEAP_FLAG_ALLOW_ALL_BUFFERS_AND_TEXTURES },
  1153. { L"ALLOW_ONLY_BUFFERS", D3D12_HEAP_FLAG_ALLOW_ONLY_BUFFERS },
  1154. { L"ALLOW_ONLY_NON_RT_DS_TEXTURES", D3D12_HEAP_FLAG_ALLOW_ONLY_NON_RT_DS_TEXTURES },
  1155. { L"ALLOW_ONLY_RT_DS_TEXTURES", D3D12_HEAP_FLAG_ALLOW_ONLY_RT_DS_TEXTURES }
  1156. };
  1157. static const ParserEnumValue RESOURCE_STATE_TABLE[] = {
  1158. { L"COMMON", D3D12_RESOURCE_STATE_COMMON },
  1159. { L"VERTEX_AND_CONSTANT_BUFFER", D3D12_RESOURCE_STATE_VERTEX_AND_CONSTANT_BUFFER },
  1160. { L"INDEX_BUFFER", D3D12_RESOURCE_STATE_INDEX_BUFFER },
  1161. { L"RENDER_TARGET", D3D12_RESOURCE_STATE_RENDER_TARGET },
  1162. { L"UNORDERED_ACCESS", D3D12_RESOURCE_STATE_UNORDERED_ACCESS },
  1163. { L"DEPTH_WRITE", D3D12_RESOURCE_STATE_DEPTH_WRITE },
  1164. { L"DEPTH_READ", D3D12_RESOURCE_STATE_DEPTH_READ },
  1165. { L"NON_PIXEL_SHADER_RESOURCE", D3D12_RESOURCE_STATE_NON_PIXEL_SHADER_RESOURCE },
  1166. { L"PIXEL_SHADER_RESOURCE", D3D12_RESOURCE_STATE_PIXEL_SHADER_RESOURCE },
  1167. { L"STREAM_OUT", D3D12_RESOURCE_STATE_STREAM_OUT },
  1168. { L"INDIRECT_ARGUMENT", D3D12_RESOURCE_STATE_INDIRECT_ARGUMENT },
  1169. { L"COPY_DEST", D3D12_RESOURCE_STATE_COPY_DEST },
  1170. { L"COPY_SOURCE", D3D12_RESOURCE_STATE_COPY_SOURCE },
  1171. { L"RESOLVE_DEST", D3D12_RESOURCE_STATE_RESOLVE_DEST },
  1172. { L"RESOLVE_SOURCE", D3D12_RESOURCE_STATE_RESOLVE_SOURCE },
  1173. { L"GENERIC_READ", D3D12_RESOURCE_STATE_GENERIC_READ },
  1174. { L"PRESENT", D3D12_RESOURCE_STATE_PRESENT },
  1175. { L"PREDICATION", D3D12_RESOURCE_STATE_PREDICATION }
  1176. };
  1177. static const ParserEnumValue DESCRIPTOR_HEAP_TYPE_TABLE[] = {
  1178. { L"CBV_SRV_UAV", D3D12_DESCRIPTOR_HEAP_TYPE_CBV_SRV_UAV },
  1179. { L"SAMPLER", D3D12_DESCRIPTOR_HEAP_TYPE_SAMPLER },
  1180. { L"RTV", D3D12_DESCRIPTOR_HEAP_TYPE_RTV },
  1181. { L"DSV", D3D12_DESCRIPTOR_HEAP_TYPE_DSV }
  1182. };
  1183. static const ParserEnumValue DESCRIPTOR_HEAP_FLAG_TABLE[] = {
  1184. { L"NONE", D3D12_DESCRIPTOR_HEAP_FLAG_NONE },
  1185. { L"SHADER_VISIBLE", D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE }
  1186. };
  1187. static const ParserEnumValue SRV_DIMENSION_TABLE[] = {
  1188. { L"UNKNOWN", D3D12_SRV_DIMENSION_UNKNOWN },
  1189. { L"BUFFER", D3D12_SRV_DIMENSION_BUFFER },
  1190. { L"TEXTURE1D", D3D12_SRV_DIMENSION_TEXTURE1D },
  1191. { L"TEXTURE1DARRAY", D3D12_SRV_DIMENSION_TEXTURE1DARRAY },
  1192. { L"TEXTURE2D", D3D12_SRV_DIMENSION_TEXTURE2D },
  1193. { L"TEXTURE2DARRAY", D3D12_SRV_DIMENSION_TEXTURE2DARRAY },
  1194. { L"TEXTURE2DMS", D3D12_SRV_DIMENSION_TEXTURE2DMS },
  1195. { L"TEXTURE2DMSARRAY", D3D12_SRV_DIMENSION_TEXTURE2DMSARRAY },
  1196. { L"TEXTURE3D", D3D12_SRV_DIMENSION_TEXTURE3D },
  1197. { L"TEXTURECUBE", D3D12_SRV_DIMENSION_TEXTURECUBE },
  1198. { L"TEXTURECUBEARRAY", D3D12_SRV_DIMENSION_TEXTURECUBEARRAY }
  1199. };
  1200. static const ParserEnumValue UAV_DIMENSION_TABLE[] = {
  1201. { L"UNKNOWN", D3D12_UAV_DIMENSION_UNKNOWN },
  1202. { L"BUFFER", D3D12_UAV_DIMENSION_BUFFER },
  1203. { L"TEXTURE1D", D3D12_UAV_DIMENSION_TEXTURE1D },
  1204. { L"TEXTURE1DARRAY", D3D12_UAV_DIMENSION_TEXTURE1DARRAY },
  1205. { L"TEXTURE2D", D3D12_UAV_DIMENSION_TEXTURE2D },
  1206. { L"TEXTURE2DARRAY", D3D12_UAV_DIMENSION_TEXTURE2DARRAY },
  1207. { L"TEXTURE3D", D3D12_UAV_DIMENSION_TEXTURE3D }
  1208. };
  1209. static const ParserEnumValue PRIMITIVE_TOPOLOGY_TABLE[] = {
  1210. { L"UNDEFINED",D3D_PRIMITIVE_TOPOLOGY_UNDEFINED },
  1211. { L"POINTLIST",D3D_PRIMITIVE_TOPOLOGY_POINTLIST },
  1212. { L"LINELIST",D3D_PRIMITIVE_TOPOLOGY_LINELIST },
  1213. { L"LINESTRIP",D3D_PRIMITIVE_TOPOLOGY_LINESTRIP },
  1214. { L"TRIANGLELIST",D3D_PRIMITIVE_TOPOLOGY_TRIANGLELIST },
  1215. { L"TRIANGLESTRIP",D3D_PRIMITIVE_TOPOLOGY_TRIANGLESTRIP },
  1216. { L"LINELIST_ADJ",D3D_PRIMITIVE_TOPOLOGY_LINELIST_ADJ },
  1217. { L"LINESTRIP_ADJ",D3D_PRIMITIVE_TOPOLOGY_LINESTRIP_ADJ },
  1218. { L"TRIANGLELIST_ADJ",D3D_PRIMITIVE_TOPOLOGY_TRIANGLELIST_ADJ },
  1219. { L"TRIANGLESTRIP_ADJ",D3D_PRIMITIVE_TOPOLOGY_TRIANGLESTRIP_ADJ },
  1220. { L"1_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_1_CONTROL_POINT_PATCHLIST },
  1221. { L"2_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_2_CONTROL_POINT_PATCHLIST },
  1222. { L"3_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_3_CONTROL_POINT_PATCHLIST },
  1223. { L"4_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_4_CONTROL_POINT_PATCHLIST },
  1224. { L"5_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_5_CONTROL_POINT_PATCHLIST },
  1225. { L"6_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_6_CONTROL_POINT_PATCHLIST },
  1226. { L"7_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_7_CONTROL_POINT_PATCHLIST },
  1227. { L"8_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_8_CONTROL_POINT_PATCHLIST },
  1228. { L"9_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_9_CONTROL_POINT_PATCHLIST },
  1229. { L"10_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_10_CONTROL_POINT_PATCHLIST },
  1230. { L"11_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_11_CONTROL_POINT_PATCHLIST },
  1231. { L"12_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_12_CONTROL_POINT_PATCHLIST },
  1232. { L"13_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_13_CONTROL_POINT_PATCHLIST },
  1233. { L"14_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_14_CONTROL_POINT_PATCHLIST },
  1234. { L"15_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_15_CONTROL_POINT_PATCHLIST },
  1235. { L"16_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_16_CONTROL_POINT_PATCHLIST },
  1236. { L"17_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_17_CONTROL_POINT_PATCHLIST },
  1237. { L"18_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_18_CONTROL_POINT_PATCHLIST },
  1238. { L"19_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_19_CONTROL_POINT_PATCHLIST },
  1239. { L"20_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_20_CONTROL_POINT_PATCHLIST },
  1240. { L"21_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_21_CONTROL_POINT_PATCHLIST },
  1241. { L"22_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_22_CONTROL_POINT_PATCHLIST },
  1242. { L"23_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_23_CONTROL_POINT_PATCHLIST },
  1243. { L"24_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_24_CONTROL_POINT_PATCHLIST },
  1244. { L"25_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_25_CONTROL_POINT_PATCHLIST },
  1245. { L"26_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_26_CONTROL_POINT_PATCHLIST },
  1246. { L"27_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_27_CONTROL_POINT_PATCHLIST },
  1247. { L"28_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_28_CONTROL_POINT_PATCHLIST },
  1248. { L"29_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_29_CONTROL_POINT_PATCHLIST },
  1249. { L"30_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_30_CONTROL_POINT_PATCHLIST },
  1250. { L"31_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_31_CONTROL_POINT_PATCHLIST },
  1251. { L"32_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_32_CONTROL_POINT_PATCHLIST }
  1252. };
  1253. static const ParserEnumValue PRIMITIVE_TOPOLOGY_TYPE_TABLE[] = {
  1254. { L"UNDEFINED", D3D12_PRIMITIVE_TOPOLOGY_TYPE_UNDEFINED },
  1255. { L"POINT", D3D12_PRIMITIVE_TOPOLOGY_TYPE_POINT },
  1256. { L"LINE", D3D12_PRIMITIVE_TOPOLOGY_TYPE_LINE },
  1257. { L"TRIANGLE", D3D12_PRIMITIVE_TOPOLOGY_TYPE_TRIANGLE },
  1258. { L"PATCH", D3D12_PRIMITIVE_TOPOLOGY_TYPE_PATCH }
  1259. };
  1260. static const ParserEnumTable g_ParserEnumTables[] = {
  1261. { _countof(INPUT_CLASSIFICATION_TABLE), INPUT_CLASSIFICATION_TABLE, ParserEnumKind::INPUT_CLASSIFICATION },
  1262. { _countof(DXGI_FORMAT_TABLE), DXGI_FORMAT_TABLE, ParserEnumKind::DXGI_FORMAT },
  1263. { _countof(HEAP_TYPE_TABLE), HEAP_TYPE_TABLE, ParserEnumKind::HEAP_TYPE },
  1264. { _countof(CPU_PAGE_PROPERTY_TABLE), CPU_PAGE_PROPERTY_TABLE, ParserEnumKind::CPU_PAGE_PROPERTY },
  1265. { _countof(MEMORY_POOL_TABLE), MEMORY_POOL_TABLE, ParserEnumKind::MEMORY_POOL },
  1266. { _countof(RESOURCE_DIMENSION_TABLE), RESOURCE_DIMENSION_TABLE, ParserEnumKind::RESOURCE_DIMENSION },
  1267. { _countof(TEXTURE_LAYOUT_TABLE), TEXTURE_LAYOUT_TABLE, ParserEnumKind::TEXTURE_LAYOUT },
  1268. { _countof(RESOURCE_FLAG_TABLE), RESOURCE_FLAG_TABLE, ParserEnumKind::RESOURCE_FLAG },
  1269. { _countof(HEAP_FLAG_TABLE), HEAP_FLAG_TABLE, ParserEnumKind::HEAP_FLAG },
  1270. { _countof(RESOURCE_STATE_TABLE), RESOURCE_STATE_TABLE, ParserEnumKind::RESOURCE_STATE },
  1271. { _countof(DESCRIPTOR_HEAP_TYPE_TABLE), DESCRIPTOR_HEAP_TYPE_TABLE, ParserEnumKind::DESCRIPTOR_HEAP_TYPE },
  1272. { _countof(DESCRIPTOR_HEAP_FLAG_TABLE), DESCRIPTOR_HEAP_FLAG_TABLE, ParserEnumKind::DESCRIPTOR_HEAP_FLAG },
  1273. { _countof(SRV_DIMENSION_TABLE), SRV_DIMENSION_TABLE, ParserEnumKind::SRV_DIMENSION },
  1274. { _countof(UAV_DIMENSION_TABLE), UAV_DIMENSION_TABLE, ParserEnumKind::UAV_DIMENSION },
  1275. { _countof(PRIMITIVE_TOPOLOGY_TABLE), PRIMITIVE_TOPOLOGY_TABLE, ParserEnumKind::PRIMITIVE_TOPOLOGY },
  1276. { _countof(PRIMITIVE_TOPOLOGY_TYPE_TABLE), PRIMITIVE_TOPOLOGY_TYPE_TABLE, ParserEnumKind::PRIMITIVE_TOPOLOGY_TYPE },
  1277. };
  1278. static HRESULT GetEnumValue(LPCWSTR name, ParserEnumKind K, UINT *pValue) {
  1279. for (size_t i = 0; i < _countof(g_ParserEnumTables); ++i) {
  1280. const ParserEnumTable &T = g_ParserEnumTables[i];
  1281. if (T.Kind != K) {
  1282. continue;
  1283. }
  1284. for (size_t j = 0; j < T.ValueCount; ++j) {
  1285. if (_wcsicmp(name, T.Values[j].Name) == 0) {
  1286. *pValue = T.Values[j].Value;
  1287. return S_OK;
  1288. }
  1289. }
  1290. }
  1291. return E_INVALIDARG;
  1292. }
  1293. template <typename T>
  1294. static HRESULT GetEnumValueT(LPCWSTR name, ParserEnumKind K, T *pValue) {
  1295. UINT u;
  1296. HRESULT hr = GetEnumValue(name, K, &u);
  1297. *pValue = (T)u;
  1298. return hr;
  1299. }
  1300. template <typename T>
  1301. static HRESULT ReadAttrEnumT(IXmlReader *pReader, LPCWSTR pAttrName, ParserEnumKind K, T *pValue, T defaultValue, LPCWSTR pStripPrefix = nullptr) {
  1302. if (S_FALSE == CHECK_HR_RET(pReader->MoveToAttributeByName(pAttrName, nullptr))) {
  1303. *pValue = defaultValue;
  1304. return S_FALSE;
  1305. }
  1306. LPCWSTR pText;
  1307. CHECK_HR(pReader->GetValue(&pText, nullptr));
  1308. if (pStripPrefix && *pStripPrefix && _wcsnicmp(pStripPrefix, pText, wcslen(pStripPrefix)) == 0)
  1309. pText += wcslen(pStripPrefix);
  1310. CHECK_HR(GetEnumValueT(pText, K, pValue));
  1311. CHECK_HR(pReader->MoveToElement());
  1312. return S_OK;
  1313. }
  1314. static HRESULT ReadAttrINPUT_CLASSIFICATION(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_INPUT_CLASSIFICATION *pValue) {
  1315. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::INPUT_CLASSIFICATION, pValue, D3D12_INPUT_CLASSIFICATION_PER_VERTEX_DATA);
  1316. }
  1317. static HRESULT ReadAttrDESCRIPTOR_HEAP_TYPE(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_DESCRIPTOR_HEAP_TYPE *pValue) {
  1318. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::DESCRIPTOR_HEAP_TYPE, pValue, D3D12_DESCRIPTOR_HEAP_TYPE_CBV_SRV_UAV);
  1319. }
  1320. static HRESULT ReadAttrDESCRIPTOR_HEAP_FLAGS(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_DESCRIPTOR_HEAP_FLAGS *pValue) {
  1321. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::DESCRIPTOR_HEAP_FLAG, pValue, D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE);
  1322. }
  1323. static HRESULT ReadAttrDXGI_FORMAT(IXmlReader *pReader, LPCWSTR pAttrName, DXGI_FORMAT *pValue) {
  1324. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::DXGI_FORMAT, pValue, DXGI_FORMAT_UNKNOWN, L"DXGI_FORMAT_");
  1325. }
  1326. static HRESULT ReadAttrHEAP_TYPE(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_HEAP_TYPE *pValue) {
  1327. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::HEAP_TYPE, pValue, D3D12_HEAP_TYPE_DEFAULT);
  1328. }
  1329. static HRESULT ReadAttrCPU_PAGE_PROPERTY(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_CPU_PAGE_PROPERTY *pValue) {
  1330. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::CPU_PAGE_PROPERTY, pValue, D3D12_CPU_PAGE_PROPERTY_UNKNOWN);
  1331. }
  1332. static HRESULT ReadAttrMEMORY_POOL(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_MEMORY_POOL *pValue) {
  1333. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::MEMORY_POOL, pValue, D3D12_MEMORY_POOL_UNKNOWN);
  1334. }
  1335. static HRESULT ReadAttrRESOURCE_DIMENSION(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_RESOURCE_DIMENSION *pValue) {
  1336. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::RESOURCE_DIMENSION, pValue, D3D12_RESOURCE_DIMENSION_BUFFER);
  1337. }
  1338. static HRESULT ReadAttrTEXTURE_LAYOUT(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_TEXTURE_LAYOUT *pValue) {
  1339. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::TEXTURE_LAYOUT, pValue, D3D12_TEXTURE_LAYOUT_UNKNOWN);
  1340. }
  1341. static HRESULT ReadAttrRESOURCE_FLAGS(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_RESOURCE_FLAGS *pValue) {
  1342. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::RESOURCE_FLAG, pValue, D3D12_RESOURCE_FLAG_NONE);
  1343. }
  1344. static HRESULT ReadAttrHEAP_FLAGS(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_HEAP_FLAGS *pValue) {
  1345. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::HEAP_FLAG, pValue, D3D12_HEAP_FLAG_NONE);
  1346. }
  1347. static HRESULT ReadAttrRESOURCE_STATES(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_RESOURCE_STATES *pValue) {
  1348. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::RESOURCE_STATE, pValue, D3D12_RESOURCE_STATE_COMMON);
  1349. }
  1350. static HRESULT ReadAttrSRV_DIMENSION(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_SRV_DIMENSION *pValue) {
  1351. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::SRV_DIMENSION, pValue, D3D12_SRV_DIMENSION_BUFFER);
  1352. }
  1353. static HRESULT ReadAttrUAV_DIMENSION(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_UAV_DIMENSION *pValue) {
  1354. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::UAV_DIMENSION, pValue, D3D12_UAV_DIMENSION_BUFFER);
  1355. }
  1356. static HRESULT ReadAttrPRIMITIVE_TOPOLOGY(IXmlReader *pReader, LPCWSTR pAttrName, D3D_PRIMITIVE_TOPOLOGY *pValue) {
  1357. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::PRIMITIVE_TOPOLOGY, pValue, D3D_PRIMITIVE_TOPOLOGY_TRIANGLELIST);
  1358. }
  1359. static HRESULT ReadAttrPRIMITIVE_TOPOLOGY_TYPE(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_PRIMITIVE_TOPOLOGY_TYPE *pValue) {
  1360. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::PRIMITIVE_TOPOLOGY_TYPE, pValue, D3D12_PRIMITIVE_TOPOLOGY_TYPE_TRIANGLE);
  1361. }
  1362. HRESULT ShaderOpParser::ReadAttrStr(IXmlReader *pReader, LPCWSTR pAttrName, LPCSTR *ppValue) {
  1363. if (S_FALSE == CHECK_HR_RET(pReader->MoveToAttributeByName(pAttrName, nullptr))) {
  1364. *ppValue = nullptr;
  1365. return S_FALSE;
  1366. }
  1367. LPCWSTR pValue;
  1368. CHECK_HR(pReader->GetValue(&pValue, nullptr));
  1369. *ppValue = m_pStrings->insert(pValue);
  1370. CHECK_HR(pReader->MoveToElement());
  1371. return S_OK;
  1372. }
  1373. HRESULT ShaderOpParser::ReadAttrBOOL(IXmlReader *pReader, LPCWSTR pAttrName, BOOL *pValue, BOOL defaultValue) {
  1374. if (S_FALSE == CHECK_HR_RET(pReader->MoveToAttributeByName(pAttrName, nullptr))) {
  1375. *pValue = defaultValue;
  1376. return S_FALSE;
  1377. }
  1378. LPCWSTR pText;
  1379. CHECK_HR(pReader->GetValue(&pText, nullptr));
  1380. if (_wcsicmp(pText, L"true") == 0) {
  1381. *pValue = TRUE;
  1382. }
  1383. else {
  1384. *pValue = FALSE;
  1385. }
  1386. CHECK_HR(pReader->MoveToElement());
  1387. return S_OK;
  1388. }
  1389. HRESULT ShaderOpParser::ReadAttrUINT64(IXmlReader *pReader, LPCWSTR pAttrName, UINT64 *pValue, UINT64 defaultValue) {
  1390. if (S_FALSE == CHECK_HR_RET(pReader->MoveToAttributeByName(pAttrName, nullptr))) {
  1391. *pValue = defaultValue;
  1392. return S_FALSE;
  1393. }
  1394. LPCWSTR pText;
  1395. CHECK_HR(pReader->GetValue(&pText, nullptr));
  1396. long long ll = _wtol(pText);
  1397. if (errno == ERANGE) CHECK_HR(E_INVALIDARG);
  1398. *pValue = ll;
  1399. CHECK_HR(pReader->MoveToElement());
  1400. return S_OK;
  1401. }
  1402. HRESULT ShaderOpParser::ReadAttrUINT(IXmlReader *pReader, LPCWSTR pAttrName, UINT *pValue, UINT defaultValue) {
  1403. UINT64 u64;
  1404. HRESULT hrRead = CHECK_HR_RET(ReadAttrUINT64(pReader, pAttrName, &u64, defaultValue));
  1405. CHECK_HR(UInt64ToUInt(u64, pValue));
  1406. return hrRead;
  1407. }
  1408. HRESULT ShaderOpParser::ReadAttrUINT16(IXmlReader *pReader, LPCWSTR pAttrName, UINT16 *pValue, UINT16 defaultValue) {
  1409. UINT64 u64;
  1410. HRESULT hrRead = CHECK_HR_RET(ReadAttrUINT64(pReader, pAttrName, &u64, defaultValue));
  1411. CHECK_HR(UInt64ToUInt16(u64, pValue));
  1412. return hrRead;
  1413. }
  1414. void ShaderOpParser::ReadElementContentStr(IXmlReader *pReader, LPCSTR *ppValue) {
  1415. *ppValue = nullptr;
  1416. if (pReader->IsEmptyElement())
  1417. return;
  1418. UINT startDepth;
  1419. XmlNodeType nt;
  1420. CHECK_HR(pReader->GetDepth(&startDepth));
  1421. std::wstring value;
  1422. for (;;) {
  1423. UINT depth;
  1424. CHECK_HR(pReader->Read(&nt));
  1425. CHECK_HR(pReader->GetDepth(&depth));
  1426. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1427. break;
  1428. if (nt == XmlNodeType_CDATA || nt == XmlNodeType_Text || nt == XmlNodeType_Whitespace) {
  1429. LPCWSTR pText;
  1430. CHECK_HR(pReader->GetValue(&pText, nullptr));
  1431. value += pText;
  1432. }
  1433. }
  1434. *ppValue = m_pStrings->insert(value.c_str());
  1435. }
  1436. void ShaderOpParser::ParseDescriptor(IXmlReader *pReader, ShaderOpDescriptor *pDesc) {
  1437. if (!ReadAtElementName(pReader, L"Descriptor"))
  1438. return;
  1439. CHECK_HR(ReadAttrStr(pReader, L"Name", &pDesc->Name));
  1440. CHECK_HR(ReadAttrStr(pReader, L"ResName", &pDesc->ResName));
  1441. CHECK_HR(ReadAttrStr(pReader, L"CounterName", &pDesc->CounterName));
  1442. CHECK_HR(ReadAttrStr(pReader, L"Kind", &pDesc->Kind));
  1443. bool isSRV = pDesc->Kind && 0 == _stricmp(pDesc->Kind, "SRV");
  1444. pDesc->SrvDescPresent = false;
  1445. DXGI_FORMAT *pFormat;
  1446. if (isSRV) {
  1447. // D3D12_SHADER_RESOURCE_VIEW_DESC
  1448. pFormat = &pDesc->SrvDesc.Format;
  1449. }
  1450. else {
  1451. // D3D12_UNORDERED_ACCESS_VIEW_DESC - default for parsing
  1452. pFormat = &pDesc->UavDesc.Format;
  1453. }
  1454. HRESULT hrFormat = ReadAttrDXGI_FORMAT(pReader, L"Format", pFormat);
  1455. CHECK_HR(hrFormat);
  1456. if (isSRV) {
  1457. pDesc->SrvDesc.Shader4ComponentMapping = D3D12_DEFAULT_SHADER_4_COMPONENT_MAPPING;
  1458. pDesc->SrvDescPresent |= S_OK ==
  1459. CHECK_HR_RET(ReadAttrSRV_DIMENSION(pReader, L"Dimension", &pDesc->SrvDesc.ViewDimension));
  1460. switch (pDesc->SrvDesc.ViewDimension) {
  1461. case D3D12_SRV_DIMENSION_BUFFER:
  1462. pDesc->SrvDescPresent |= S_OK ==
  1463. CHECK_HR_RET(ReadAttrUINT64(pReader, L"FirstElement", &pDesc->SrvDesc.Buffer.FirstElement));
  1464. LPCSTR pFlags;
  1465. pDesc->SrvDescPresent |= S_OK ==
  1466. CHECK_HR_RET(ReadAttrStr(pReader, L"Flags", &pFlags));
  1467. if (pFlags && *pFlags && 0 == _stricmp(pFlags, "RAW")) {
  1468. pDesc->SrvDesc.Buffer.Flags = D3D12_BUFFER_SRV_FLAG_RAW;
  1469. }
  1470. else {
  1471. pDesc->SrvDesc.Buffer.Flags = D3D12_BUFFER_SRV_FLAG_NONE;
  1472. }
  1473. pDesc->SrvDescPresent |= S_OK ==
  1474. CHECK_HR_RET(ReadAttrUINT(pReader, L"NumElements", &pDesc->SrvDesc.Buffer.NumElements));
  1475. pDesc->SrvDescPresent |= S_OK ==
  1476. CHECK_HR_RET(ReadAttrUINT(pReader, L"StructureByteStride", &pDesc->SrvDesc.Buffer.StructureByteStride));
  1477. break;
  1478. default:
  1479. CHECK_HR(E_NOTIMPL);
  1480. }
  1481. }
  1482. else {
  1483. CHECK_HR(ReadAttrUAV_DIMENSION(pReader, L"Dimension", &pDesc->UavDesc.ViewDimension));
  1484. switch (pDesc->UavDesc.ViewDimension) {
  1485. case D3D12_UAV_DIMENSION_BUFFER:
  1486. CHECK_HR(ReadAttrUINT64(pReader, L"FirstElement", &pDesc->UavDesc.Buffer.FirstElement));
  1487. CHECK_HR(ReadAttrUINT(pReader, L"NumElements", &pDesc->UavDesc.Buffer.NumElements));
  1488. CHECK_HR(ReadAttrUINT(pReader, L"StructureByteStride", &pDesc->UavDesc.Buffer.StructureByteStride));
  1489. CHECK_HR(ReadAttrUINT64(pReader, L"CounterOffsetInBytes", &pDesc->UavDesc.Buffer.CounterOffsetInBytes));
  1490. LPCSTR pFlags;
  1491. CHECK_HR(ReadAttrStr(pReader, L"Flags", &pFlags));
  1492. if (pFlags && *pFlags && 0 == _stricmp(pFlags, "RAW")) {
  1493. pDesc->UavDesc.Buffer.Flags = D3D12_BUFFER_UAV_FLAG_RAW;
  1494. }
  1495. else {
  1496. pDesc->UavDesc.Buffer.Flags = D3D12_BUFFER_UAV_FLAG_NONE;
  1497. }
  1498. if (hrFormat == S_FALSE && pDesc->UavDesc.Buffer.Flags & D3D12_BUFFER_UAV_FLAG_RAW) {
  1499. pDesc->UavDesc.Format = DXGI_FORMAT_R32_TYPELESS;
  1500. }
  1501. break;
  1502. case D3D12_UAV_DIMENSION_TEXTURE1D:
  1503. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture1D.MipSlice));
  1504. break;
  1505. case D3D12_UAV_DIMENSION_TEXTURE1DARRAY:
  1506. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture1DArray.MipSlice));
  1507. CHECK_HR(ReadAttrUINT(pReader, L"FirstArraySlice", &pDesc->UavDesc.Texture1DArray.FirstArraySlice));
  1508. CHECK_HR(ReadAttrUINT(pReader, L"ArraySize", &pDesc->UavDesc.Texture1DArray.ArraySize));
  1509. break;
  1510. case D3D12_UAV_DIMENSION_TEXTURE2D:
  1511. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture2D.MipSlice));
  1512. CHECK_HR(ReadAttrUINT(pReader, L"PlaneSlice", &pDesc->UavDesc.Texture2D.PlaneSlice));
  1513. break;
  1514. case D3D12_UAV_DIMENSION_TEXTURE2DARRAY:
  1515. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture2DArray.MipSlice));
  1516. CHECK_HR(ReadAttrUINT(pReader, L"FirstArraySlice", &pDesc->UavDesc.Texture2DArray.FirstArraySlice));
  1517. CHECK_HR(ReadAttrUINT(pReader, L"ArraySize", &pDesc->UavDesc.Texture2DArray.ArraySize));
  1518. CHECK_HR(ReadAttrUINT(pReader, L"PlaneSlice", &pDesc->UavDesc.Texture2DArray.PlaneSlice));
  1519. break;
  1520. case D3D12_UAV_DIMENSION_TEXTURE3D:
  1521. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture3D.MipSlice));
  1522. CHECK_HR(ReadAttrUINT(pReader, L"FirstWSlice", &pDesc->UavDesc.Texture3D.FirstWSlice));
  1523. CHECK_HR(ReadAttrUINT(pReader, L"WSize", &pDesc->UavDesc.Texture3D.WSize));
  1524. break;
  1525. }
  1526. }
  1527. // If either is missing, set one from the other.
  1528. if (pDesc->Name && !pDesc->ResName) pDesc->ResName = pDesc->Name;
  1529. if (pDesc->ResName && !pDesc->Name) pDesc->Name = pDesc->ResName;
  1530. LPCSTR K = pDesc->Kind;
  1531. if (K == nullptr) {
  1532. ShaderOpLogFmt(L"Descriptor '%S' is missing Kind attribute.", pDesc->Name);
  1533. CHECK_HR(E_INVALIDARG);
  1534. } else if (0 != _stricmp(K, "UAV") && 0 != _stricmp(K, "SRV") &&
  1535. 0 != _stricmp(K, "CBV") && 0 != _stricmp(K, "RTV")) {
  1536. ShaderOpLogFmt(L"Descriptor '%S' references unknown kind '%S'",
  1537. pDesc->Name, K);
  1538. CHECK_HR(E_INVALIDARG);
  1539. }
  1540. }
  1541. void ShaderOpParser::ParseDescriptorHeap(IXmlReader *pReader, ShaderOpDescriptorHeap *pHeap) {
  1542. if (!ReadAtElementName(pReader, L"DescriptorHeap"))
  1543. return;
  1544. CHECK_HR(ReadAttrStr(pReader, L"Name", &pHeap->Name));
  1545. HRESULT hrFlags = ReadAttrDESCRIPTOR_HEAP_FLAGS(pReader, L"Flags", &pHeap->Desc.Flags);
  1546. CHECK_HR(hrFlags);
  1547. CHECK_HR(ReadAttrUINT(pReader, L"NodeMask", &pHeap->Desc.NodeMask));
  1548. CHECK_HR(ReadAttrUINT(pReader, L"NumDescriptors", &pHeap->Desc.NumDescriptors));
  1549. CHECK_HR(ReadAttrDESCRIPTOR_HEAP_TYPE(pReader, L"Type", &pHeap->Desc.Type));
  1550. if (pHeap->Desc.Type == D3D12_DESCRIPTOR_HEAP_TYPE_RTV && hrFlags == S_FALSE)
  1551. pHeap->Desc.Flags = D3D12_DESCRIPTOR_HEAP_FLAG_NONE;
  1552. if (pReader->IsEmptyElement())
  1553. return;
  1554. UINT startDepth;
  1555. XmlNodeType nt;
  1556. CHECK_HR(pReader->GetDepth(&startDepth));
  1557. std::wstring value;
  1558. for (;;) {
  1559. UINT depth;
  1560. CHECK_HR(pReader->Read(&nt));
  1561. CHECK_HR(pReader->GetDepth(&depth));
  1562. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1563. break;
  1564. if (nt == XmlNodeType_Element) {
  1565. LPCWSTR pLocalName;
  1566. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1567. if (0 == wcscmp(pLocalName, L"Descriptor")) {
  1568. ShaderOpDescriptor D;
  1569. ParseDescriptor(pReader, &D);
  1570. pHeap->Descriptors.push_back(D);
  1571. }
  1572. }
  1573. }
  1574. }
  1575. void ShaderOpParser::ParseInputElement(IXmlReader *pReader, D3D12_INPUT_ELEMENT_DESC *pInputElement) {
  1576. if (!ReadAtElementName(pReader, L"InputElement"))
  1577. return;
  1578. CHECK_HR(ReadAttrStr(pReader, L"SemanticName", &pInputElement->SemanticName));
  1579. CHECK_HR(ReadAttrUINT(pReader, L"SemanticIndex", &pInputElement->SemanticIndex));
  1580. CHECK_HR(ReadAttrDXGI_FORMAT(pReader, L"Format", &pInputElement->Format));
  1581. CHECK_HR(ReadAttrUINT(pReader, L"InputSlot", &pInputElement->InputSlot));
  1582. CHECK_HR(ReadAttrUINT(pReader, L"AlignedByteOffset", &pInputElement->AlignedByteOffset, D3D12_APPEND_ALIGNED_ELEMENT));
  1583. CHECK_HR(ReadAttrINPUT_CLASSIFICATION(pReader, L"InputSlotClass", &pInputElement->InputSlotClass));
  1584. CHECK_HR(ReadAttrUINT(pReader, L"InstanceDataStepRate", &pInputElement->InstanceDataStepRate));
  1585. }
  1586. void ShaderOpParser::ParseInputElements(IXmlReader *pReader, std::vector<D3D12_INPUT_ELEMENT_DESC> *pInputElements) {
  1587. if (!ReadAtElementName(pReader, L"InputElements"))
  1588. return;
  1589. if (pReader->IsEmptyElement()) return;
  1590. UINT startDepth;
  1591. XmlNodeType nt;
  1592. CHECK_HR(pReader->GetDepth(&startDepth));
  1593. for (;;) {
  1594. UINT depth;
  1595. CHECK_HR(pReader->Read(&nt));
  1596. CHECK_HR(pReader->GetDepth(&depth));
  1597. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1598. return;
  1599. if (nt == XmlNodeType_Element) {
  1600. LPCWSTR pLocalName;
  1601. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1602. if (0 == wcscmp(pLocalName, L"InputElement")) {
  1603. D3D12_INPUT_ELEMENT_DESC desc;
  1604. ParseInputElement(pReader, &desc);
  1605. pInputElements->push_back(desc);
  1606. }
  1607. }
  1608. }
  1609. }
  1610. void ShaderOpParser::ParseRenderTargets(IXmlReader *pReader, std::vector<LPCSTR> *pRenderTargets) {
  1611. if (!ReadAtElementName(pReader, L"RenderTargets"))
  1612. return;
  1613. if (pReader->IsEmptyElement()) return;
  1614. UINT startDepth;
  1615. XmlNodeType nt;
  1616. CHECK_HR(pReader->GetDepth(&startDepth));
  1617. for (;;) {
  1618. UINT depth;
  1619. CHECK_HR(pReader->Read(&nt));
  1620. CHECK_HR(pReader->GetDepth(&depth));
  1621. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1622. return;
  1623. if (nt == XmlNodeType_Element) {
  1624. LPCWSTR pLocalName;
  1625. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1626. if (0 == wcscmp(pLocalName, L"RenderTarget")) {
  1627. LPCSTR pName;
  1628. CHECK_HR(ReadAttrStr(pReader, L"Name", &pName));
  1629. pRenderTargets->push_back(pName);
  1630. }
  1631. }
  1632. }
  1633. }
  1634. void ShaderOpParser::ParseRootValue(IXmlReader *pReader, ShaderOpRootValue *pRootValue) {
  1635. if (!ReadAtElementName(pReader, L"RootValue"))
  1636. return;
  1637. CHECK_HR(ReadAttrStr(pReader, L"ResName", &pRootValue->ResName));
  1638. CHECK_HR(ReadAttrStr(pReader, L"HeapName", &pRootValue->HeapName));
  1639. CHECK_HR(ReadAttrUINT(pReader, L"Index", &pRootValue->Index));
  1640. }
  1641. void ShaderOpParser::ParseRootValues(IXmlReader *pReader, std::vector<ShaderOpRootValue> *pRootValues) {
  1642. if (!ReadAtElementName(pReader, L"RootValues"))
  1643. return;
  1644. if (pReader->IsEmptyElement()) return;
  1645. UINT startDepth;
  1646. XmlNodeType nt;
  1647. CHECK_HR(pReader->GetDepth(&startDepth));
  1648. for (;;) {
  1649. UINT depth;
  1650. CHECK_HR(pReader->Read(&nt));
  1651. CHECK_HR(pReader->GetDepth(&depth));
  1652. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1653. return;
  1654. if (nt == XmlNodeType_Element) {
  1655. LPCWSTR pLocalName;
  1656. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1657. if (0 == wcscmp(pLocalName, L"RootValue")) {
  1658. ShaderOpRootValue V;
  1659. ParseRootValue(pReader, &V);
  1660. pRootValues->push_back(V);
  1661. }
  1662. }
  1663. }
  1664. }
  1665. void ShaderOpParser::ParseShaderOpSet(IStream *pStream, ShaderOpSet *pShaderOpSet) {
  1666. CComPtr<IXmlReader> pReader;
  1667. CHECK_HR(CreateXmlReader(__uuidof(IXmlReader), (void **)&pReader, nullptr));
  1668. CHECK_HR(pReader->SetInput(pStream));
  1669. ParseShaderOpSet(pReader, pShaderOpSet);
  1670. }
  1671. void ShaderOpParser::ParseShaderOpSet(IXmlReader *pReader, ShaderOpSet *pShaderOpSet) {
  1672. if (!ReadAtElementName(pReader, L"ShaderOpSet"))
  1673. return;
  1674. UINT startDepth;
  1675. CHECK_HR(pReader->GetDepth(&startDepth));
  1676. XmlNodeType nt = XmlNodeType_Element;
  1677. for (;;) {
  1678. if (nt == XmlNodeType_Element) {
  1679. LPCWSTR pLocalName;
  1680. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1681. if (0 == wcscmp(pLocalName, L"ShaderOp")) {
  1682. pShaderOpSet->ShaderOps.emplace_back(std::make_unique<ShaderOp>());
  1683. ParseShaderOp(pReader, pShaderOpSet->ShaderOps.back().get());
  1684. }
  1685. }
  1686. else if (nt == XmlNodeType_EndElement) {
  1687. UINT depth;
  1688. CHECK_HR(pReader->GetDepth(&depth));
  1689. if (depth == startDepth + 1)
  1690. return;
  1691. }
  1692. CHECK_HR(pReader->Read(&nt));
  1693. }
  1694. }
  1695. void ShaderOpParser::ParseShaderOp(IXmlReader *pReader, ShaderOp *pShaderOp) {
  1696. m_pStrings = &pShaderOp->Strings;
  1697. // Look for a ShaderOp element.
  1698. if (!ReadAtElementName(pReader, L"ShaderOp"))
  1699. return;
  1700. CHECK_HR(ReadAttrStr(pReader, L"Name", &pShaderOp->Name));
  1701. CHECK_HR(ReadAttrStr(pReader, L"CS", &pShaderOp->CS));
  1702. CHECK_HR(ReadAttrStr(pReader, L"VS", &pShaderOp->VS));
  1703. CHECK_HR(ReadAttrStr(pReader, L"HS", &pShaderOp->HS));
  1704. CHECK_HR(ReadAttrStr(pReader, L"DS", &pShaderOp->DS));
  1705. CHECK_HR(ReadAttrStr(pReader, L"GS", &pShaderOp->GS));
  1706. CHECK_HR(ReadAttrStr(pReader, L"PS", &pShaderOp->PS));
  1707. CHECK_HR(ReadAttrUINT(pReader, L"DispatchX", &pShaderOp->DispatchX, 1));
  1708. CHECK_HR(ReadAttrUINT(pReader, L"DispatchY", &pShaderOp->DispatchY, 1));
  1709. CHECK_HR(ReadAttrUINT(pReader, L"DispatchZ", &pShaderOp->DispatchZ, 1));
  1710. CHECK_HR(ReadAttrPRIMITIVE_TOPOLOGY_TYPE(pReader, L"TopologyType", &pShaderOp->PrimitiveTopologyType));
  1711. UINT startDepth;
  1712. CHECK_HR(pReader->GetDepth(&startDepth));
  1713. XmlNodeType nt = XmlNodeType_Element;
  1714. for (;;) {
  1715. if (nt == XmlNodeType_Element) {
  1716. LPCWSTR pLocalName;
  1717. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1718. if (0 == wcscmp(pLocalName, L"InputElements")) {
  1719. ParseInputElements(pReader, &pShaderOp->InputElements);
  1720. }
  1721. else if (0 == wcscmp(pLocalName, L"Shader")) {
  1722. ShaderOpShader shader;
  1723. ParseShader(pReader, &shader);
  1724. pShaderOp->Shaders.push_back(shader);
  1725. }
  1726. else if (0 == wcscmp(pLocalName, L"RootSignature")) {
  1727. ReadElementContentStr(pReader, &pShaderOp->RootSignature);
  1728. }
  1729. else if (0 == wcscmp(pLocalName, L"RenderTargets")) {
  1730. ParseRenderTargets(pReader, &pShaderOp->RenderTargets);
  1731. }
  1732. else if (0 == wcscmp(pLocalName, L"Resource")) {
  1733. ShaderOpResource resource;
  1734. ParseResource(pReader, &resource);
  1735. pShaderOp->Resources.push_back(resource);
  1736. }
  1737. else if (0 == wcscmp(pLocalName, L"DescriptorHeap")) {
  1738. ShaderOpDescriptorHeap heap;
  1739. ParseDescriptorHeap(pReader, &heap);
  1740. pShaderOp->DescriptorHeaps.push_back(heap);
  1741. }
  1742. else if (0 == wcscmp(pLocalName, L"RootValues")) {
  1743. ParseRootValues(pReader, &pShaderOp->RootValues);
  1744. }
  1745. }
  1746. else if (nt == XmlNodeType_EndElement) {
  1747. UINT depth;
  1748. CHECK_HR(pReader->GetDepth(&depth));
  1749. if (depth == startDepth + 1)
  1750. return;
  1751. }
  1752. if (S_FALSE == CHECK_HR_RET(pReader->Read(&nt)))
  1753. return;
  1754. }
  1755. }
  1756. LPCWSTR SkipByteInitSeparators(LPCWSTR pText) {
  1757. while (*pText && (*pText == L' ' || *pText == L'\t' ||
  1758. *pText == L'\r' || *pText == L'\n' || *pText == L'{' ||
  1759. *pText == L'}' || *pText == L','))
  1760. ++pText;
  1761. return pText;
  1762. }
  1763. LPCWSTR FindByteInitSeparators(LPCWSTR pText) {
  1764. while (*pText &&
  1765. !(*pText == L' ' || *pText == L'\t' ||
  1766. *pText == L'\r' || *pText == L'\n' || *pText == L'{' ||
  1767. *pText == L'}' || *pText == L','))
  1768. ++pText;
  1769. return pText;
  1770. }
  1771. using namespace hlsl;
  1772. DXIL::ComponentType GetCompType(LPCWSTR pText, LPCWSTR pEnd) {
  1773. // if no prefix shown, use it as a default
  1774. if (pText == pEnd) return DXIL::ComponentType::F32;
  1775. // check if suffix starts with (half)
  1776. if (wcsncmp(pText, L"(half)", 6) == 0) {
  1777. return DXIL::ComponentType::F16;
  1778. }
  1779. switch (*(pEnd - 1)) {
  1780. case L'h':
  1781. case L'H':
  1782. return DXIL::ComponentType::F16;
  1783. case L'l':
  1784. case L'L':
  1785. return DXIL::ComponentType::F64;
  1786. case L'u':
  1787. case L'U':
  1788. return DXIL::ComponentType::U32;
  1789. case L'i':
  1790. case L'I':
  1791. return DXIL::ComponentType::I32;
  1792. case L'f':
  1793. case L'F':
  1794. default:
  1795. return DXIL::ComponentType::F32;
  1796. }
  1797. }
  1798. void ParseDataFromText(LPCWSTR pText, LPCWSTR pEnd, DXIL::ComponentType compType, std::vector<BYTE> &V) {
  1799. BYTE *pB;
  1800. if (compType == DXIL::ComponentType::F16 || compType == DXIL::ComponentType::F32) {
  1801. float fVal;
  1802. size_t wordSize = pEnd - pText;
  1803. if (wordSize >= 3 && 0 == _wcsnicmp(pEnd - 3, L"nan", 3)) {
  1804. fVal = NAN;
  1805. }
  1806. else if (wordSize >= 4 && 0 == _wcsnicmp(pEnd - 4, L"-inf", 4)) {
  1807. fVal = -(INFINITY);
  1808. }
  1809. else if ((wordSize >= 3 && 0 == _wcsnicmp(pEnd - 3, L"inf", 3)) ||
  1810. (wordSize >= 4 && 0 == _wcsnicmp(pEnd - 4, L"+inf", 4))) {
  1811. fVal = INFINITY;
  1812. }
  1813. else if (wordSize >= 7 && 0 == _wcsnicmp(pEnd - 7, L"-denorm", 7)) {
  1814. fVal = -(FLT_MIN / 2);
  1815. }
  1816. else if (wordSize >= 6 && 0 == _wcsnicmp(pEnd - 6, L"denorm", 6)) {
  1817. fVal = (FLT_MIN / 2);
  1818. }
  1819. else {
  1820. fVal = wcstof(pText, nullptr);
  1821. }
  1822. if (compType == DXIL::ComponentType::F16) {
  1823. uint16_t fp16Val = ConvertFloat32ToFloat16(fVal);
  1824. pB = (BYTE *)&fp16Val;
  1825. V.insert(V.end(), pB, pB + sizeof(uint16_t));
  1826. }
  1827. else {
  1828. pB = (BYTE *)&fVal;
  1829. V.insert(V.end(), pB, pB + sizeof(float));
  1830. }
  1831. }
  1832. else if (compType == DXIL::ComponentType::I32) {
  1833. int val = _wtoi(pText);
  1834. pB = (BYTE *)&val;
  1835. V.insert(V.end(), pB, pB + sizeof(int));
  1836. }
  1837. else {
  1838. DXASSERT_ARGS(false, "Unsupported stream component type : %u", compType);
  1839. }
  1840. }
  1841. void ShaderOpParser::ParseResource(IXmlReader *pReader, ShaderOpResource *pResource) {
  1842. if (!ReadAtElementName(pReader, L"Resource"))
  1843. return;
  1844. CHECK_HR(ReadAttrStr(pReader, L"Name", &pResource->Name));
  1845. CHECK_HR(ReadAttrStr(pReader, L"Init", &pResource->Init));
  1846. CHECK_HR(ReadAttrBOOL(pReader, L"ReadBack", &pResource->ReadBack));
  1847. CHECK_HR(ReadAttrHEAP_TYPE(pReader, L"HeapType", &pResource->HeapProperties.Type));
  1848. CHECK_HR(ReadAttrCPU_PAGE_PROPERTY(pReader, L"CPUPageProperty", &pResource->HeapProperties.CPUPageProperty));
  1849. CHECK_HR(ReadAttrMEMORY_POOL(pReader, L"MemoryPoolPreference", &pResource->HeapProperties.MemoryPoolPreference));
  1850. CHECK_HR(ReadAttrUINT(pReader, L"CreationNodeMask", &pResource->HeapProperties.CreationNodeMask));
  1851. CHECK_HR(ReadAttrUINT(pReader, L"VisibleNodeMask", &pResource->HeapProperties.VisibleNodeMask));
  1852. // D3D12_RESOURCE_DESC Desc;
  1853. CHECK_HR(ReadAttrRESOURCE_DIMENSION(pReader, L"Dimension", &pResource->Desc.Dimension));
  1854. CHECK_HR(ReadAttrUINT64(pReader, L"Alignment", &pResource->Desc.Alignment));
  1855. CHECK_HR(ReadAttrUINT64(pReader, L"Width", &pResource->Desc.Width));
  1856. CHECK_HR(ReadAttrUINT(pReader, L"Height", &pResource->Desc.Height));
  1857. CHECK_HR(ReadAttrUINT16(pReader, L"DepthOrArraySize", &pResource->Desc.DepthOrArraySize));
  1858. CHECK_HR(ReadAttrUINT16(pReader, L"MipLevels", &pResource->Desc.MipLevels));
  1859. CHECK_HR(ReadAttrDXGI_FORMAT(pReader, L"Format", &pResource->Desc.Format));
  1860. CHECK_HR(ReadAttrUINT(pReader, L"SampleCount", &pResource->Desc.SampleDesc.Count));
  1861. CHECK_HR(ReadAttrUINT(pReader, L"SampleQual", &pResource->Desc.SampleDesc.Quality));
  1862. CHECK_HR(ReadAttrTEXTURE_LAYOUT(pReader, L"Layout", &pResource->Desc.Layout));
  1863. CHECK_HR(ReadAttrRESOURCE_FLAGS(pReader, L"Flags", &pResource->Desc.Flags));
  1864. CHECK_HR(ReadAttrHEAP_FLAGS(pReader, L"HeapFlags", &pResource->HeapFlags));
  1865. CHECK_HR(ReadAttrRESOURCE_STATES(pReader, L"InitialResourceState", &pResource->InitialResourceState));
  1866. CHECK_HR(ReadAttrRESOURCE_STATES(pReader, L"TransitionTo", &pResource->TransitionTo));
  1867. CHECK_HR(ReadAttrPRIMITIVE_TOPOLOGY(pReader, L"Topology", &pResource->PrimitiveTopology));
  1868. // Set some fixed values.
  1869. if (pResource->Desc.Dimension == D3D12_RESOURCE_DIMENSION_BUFFER) {
  1870. pResource->Desc.Height = 1;
  1871. pResource->Desc.DepthOrArraySize = 1;
  1872. pResource->Desc.MipLevels = 1;
  1873. pResource->Desc.Format = DXGI_FORMAT_UNKNOWN;
  1874. pResource->Desc.SampleDesc.Count = 1;
  1875. pResource->Desc.SampleDesc.Quality = 0;
  1876. pResource->Desc.Layout = D3D12_TEXTURE_LAYOUT_ROW_MAJOR;
  1877. }
  1878. if (pResource->Desc.Dimension == D3D12_RESOURCE_DIMENSION_TEXTURE1D) {
  1879. if (pResource->Desc.Height == 0) pResource->Desc.Height = 1;
  1880. if (pResource->Desc.DepthOrArraySize == 0) pResource->Desc.DepthOrArraySize = 1;
  1881. if (pResource->Desc.SampleDesc.Count == 0) pResource->Desc.SampleDesc.Count = 1;
  1882. }
  1883. if (pResource->Desc.Dimension == D3D12_RESOURCE_DIMENSION_TEXTURE2D) {
  1884. if (pResource->Desc.DepthOrArraySize == 0) pResource->Desc.DepthOrArraySize = 1;
  1885. if (pResource->Desc.SampleDesc.Count == 0 ) pResource->Desc.SampleDesc.Count = 1;
  1886. }
  1887. // If the resource has text, that goes into the bytes initialization area.
  1888. if (pReader->IsEmptyElement())
  1889. return;
  1890. std::vector<BYTE> &V = pResource->InitBytes;
  1891. XmlNodeType nt;
  1892. CHECK_HR(pReader->GetNodeType(&nt));
  1893. for (;;) {
  1894. if (nt == XmlNodeType_EndElement) {
  1895. return;
  1896. }
  1897. if (nt == XmlNodeType_Text) {
  1898. // Handle the byte payload. '{', '}', ',', whitespace - these are all
  1899. // separators and are ignored in terms of structure. We simply read
  1900. // literals, figure out their type based on suffix, and write the bytes
  1901. // into the target array.
  1902. LPCWSTR pText;
  1903. pReader->GetValue(&pText, nullptr);
  1904. while (*pText) {
  1905. pText = SkipByteInitSeparators(pText);
  1906. if (!*pText) continue;
  1907. LPCWSTR pEnd = FindByteInitSeparators(pText);
  1908. // Consider looking for prefixes/suffixes to handle bases and types.
  1909. DXIL::ComponentType compType = GetCompType(pText, pEnd);
  1910. ParseDataFromText(pText, pEnd, compType, V);
  1911. pText = pEnd;
  1912. }
  1913. }
  1914. if (S_FALSE == CHECK_HR_RET(pReader->Read(&nt)))
  1915. return;
  1916. }
  1917. }
  1918. void ShaderOpParser::ParseShader(IXmlReader *pReader, ShaderOpShader *pShader) {
  1919. if (!ReadAtElementName(pReader, L"Shader"))
  1920. return;
  1921. CHECK_HR(ReadAttrStr(pReader, L"Name", &pShader->Name));
  1922. CHECK_HR(ReadAttrStr(pReader, L"EntryPoint", &pShader->EntryPoint));
  1923. CHECK_HR(ReadAttrStr(pReader, L"Target", &pShader->Target));
  1924. CHECK_HR(ReadAttrStr(pReader, L"Arguments", &pShader->Arguments));
  1925. CHECK_HR(ReadAttrBOOL(pReader, L"Compiled", &pShader->Compiled))
  1926. ReadElementContentStr(pReader, &pShader->Text);
  1927. bool hasText = pShader->Text && *pShader->Text;
  1928. if (hasText) {
  1929. LPCSTR pCheck;
  1930. CHECK_HR(ReadAttrStr(pReader, L"Text", &pCheck));
  1931. if (pCheck && *pCheck) {
  1932. ShaderOpLogFmt(L"Shader %S has text content and a Text attribute; it "
  1933. L"should only have one",
  1934. pShader->Name);
  1935. CHECK_HR(E_INVALIDARG);
  1936. }
  1937. }
  1938. else {
  1939. CHECK_HR(ReadAttrStr(pReader, L"Text", &pShader->Text));
  1940. }
  1941. if (pShader->EntryPoint == nullptr)
  1942. pShader->EntryPoint = m_pStrings->insert("main");
  1943. }
  1944. bool ShaderOpParser::ReadAtElementName(IXmlReader *pReader, LPCWSTR pName) {
  1945. XmlNodeType nt;
  1946. CHECK_HR(pReader->GetNodeType(&nt));
  1947. for (;;) {
  1948. if (nt == XmlNodeType_Element) {
  1949. LPCWSTR pLocalName;
  1950. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1951. if (0 == wcscmp(pLocalName, pName)) {
  1952. return true;
  1953. }
  1954. }
  1955. if (S_FALSE == CHECK_HR_RET(pReader->Read(&nt)))
  1956. return false;
  1957. }
  1958. }
  1959. #pragma endregion Parsing support
  1960. } // namespace st