ShaderOpTest.cpp 86 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142
  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 <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/HLSL/DxilConstants.h" // ComponentType
  24. #include "WexTestClass.h" // TAEF
  25. #include "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(), 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. dxc::WriteUtf8ToConsoleSizeT((char *)pBlob->GetBufferPointer(), pBlob->GetBufferSize());
  650. #endif
  651. } else {
  652. CComPtr<ID3DBlob> pError;
  653. hr = D3DCompile(pText, strlen(pText), S.Name, nullptr, nullptr,
  654. S.EntryPoint, S.Target, 0, 0, &pCode, &pError);
  655. if (FAILED(hr) && pError != nullptr) {
  656. ShaderOpLogFmt(L"%*S\r\n", (int)pError->GetBufferSize(),
  657. ((LPCSTR)pError->GetBufferPointer()));
  658. }
  659. }
  660. CHECK_HR(hr);
  661. m_Shaders[S.Name] = pCode;
  662. }
  663. }
  664. void ShaderOpTest::GetPipelineStats(D3D12_QUERY_DATA_PIPELINE_STATISTICS *pStats) {
  665. MappedData M;
  666. M.reset(m_pQueryBuffer, sizeof(*pStats));
  667. memcpy(pStats, M.data(), sizeof(*pStats));
  668. }
  669. void ShaderOpTest::GetReadBackData(LPCSTR pResourceName, MappedData *pData) {
  670. pResourceName = m_pShaderOp->Strings.insert(pResourceName); // Unique
  671. ShaderOpResourceData &D = m_ResourceData.at(pResourceName);
  672. D3D12_RESOURCE_DESC Desc = D.ReadBack->GetDesc();
  673. UINT32 sizeInBytes = (UINT32)Desc.Width;
  674. pData->reset(D.ReadBack, sizeInBytes);
  675. }
  676. static void SetDescriptorHeaps(ID3D12GraphicsCommandList *pList,
  677. std::vector<ID3D12DescriptorHeap *> &heaps) {
  678. if (heaps.empty())
  679. return;
  680. std::vector<ID3D12DescriptorHeap *> localHeaps;
  681. for (auto &H : heaps) {
  682. if (H->GetDesc().Flags & D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE) {
  683. localHeaps.push_back(H);
  684. }
  685. }
  686. if (!localHeaps.empty())
  687. pList->SetDescriptorHeaps((UINT)localHeaps.size(), localHeaps.data());
  688. }
  689. void ShaderOpTest::RunCommandList() {
  690. ID3D12GraphicsCommandList *pList = m_CommandList.List.p;
  691. if (m_pShaderOp->IsCompute()) {
  692. pList->SetPipelineState(m_pPSO);
  693. pList->SetComputeRootSignature(m_pRootSignature);
  694. SetDescriptorHeaps(pList, m_DescriptorHeaps);
  695. SetRootValues(pList, m_pShaderOp->IsCompute());
  696. pList->Dispatch(m_pShaderOp->DispatchX, m_pShaderOp->DispatchY,
  697. m_pShaderOp->DispatchZ);
  698. } else {
  699. pList->SetPipelineState(m_pPSO);
  700. pList->SetGraphicsRootSignature(m_pRootSignature);
  701. SetDescriptorHeaps(pList, m_DescriptorHeaps);
  702. SetRootValues(pList, m_pShaderOp->IsCompute());
  703. if (!m_pShaderOp->RenderTargets.empty()) {
  704. // Use the first render target to set up the viewport and scissors.
  705. ShaderOpResource *R = m_pShaderOp->GetResourceByName(m_pShaderOp->RenderTargets[0]);
  706. D3D12_VIEWPORT viewport;
  707. D3D12_RECT scissorRect;
  708. memset(&viewport, 0, sizeof(viewport));
  709. viewport.Height = (FLOAT)R->Desc.Height;
  710. viewport.Width = (FLOAT)R->Desc.Width;
  711. viewport.MaxDepth = 1.0f;
  712. memset(&scissorRect, 0, sizeof(scissorRect));
  713. scissorRect.right = (LONG)viewport.Width;
  714. scissorRect.bottom = (LONG)viewport.Height;
  715. pList->RSSetViewports(1, &viewport);
  716. pList->RSSetScissorRects(1, &scissorRect);
  717. }
  718. // Indicate that the buffers will be used as render targets.
  719. D3D12_CPU_DESCRIPTOR_HANDLE rtvHandles[8];
  720. UINT rtvHandleCount = (UINT)m_pShaderOp->RenderTargets.size();
  721. for (size_t i = 0; i < rtvHandleCount; ++i) {
  722. auto &rt = m_pShaderOp->RenderTargets[i];
  723. ShaderOpDescriptorData &DData = m_DescriptorData[rt];
  724. rtvHandles[i] = DData.CPUHandle;
  725. RecordTransitionBarrier(pList, DData.ResData->Resource,
  726. DData.ResData->ResourceState,
  727. D3D12_RESOURCE_STATE_RENDER_TARGET);
  728. DData.ResData->ResourceState = D3D12_RESOURCE_STATE_RENDER_TARGET;
  729. }
  730. pList->OMSetRenderTargets(rtvHandleCount, rtvHandles, FALSE, nullptr);
  731. const float ClearColor[4] = { 0.0f, 0.2f, 0.4f, 1.0f };
  732. pList->ClearRenderTargetView(rtvHandles[0], ClearColor, 0, nullptr);
  733. // TODO: set all of this from m_pShaderOp.
  734. ShaderOpResourceData &VBufferData = this->m_ResourceData[m_pShaderOp->Strings.insert("VBuffer")];
  735. D3D_PRIMITIVE_TOPOLOGY topology = D3D_PRIMITIVE_TOPOLOGY_UNDEFINED;
  736. for (ShaderOpResource &resource : m_pShaderOp->Resources) {
  737. if (_strcmpi(resource.Name, "VBuffer") == 0) {
  738. topology = resource.PrimitiveTopology;
  739. break;
  740. }
  741. }
  742. pList->IASetPrimitiveTopology(topology);
  743. // Calculate the stride in bytes from the inputs, assuming linear & contiguous.
  744. UINT strideInBytes = 0;
  745. for (auto && IE : m_pShaderOp->InputElements) {
  746. strideInBytes += GetByteSizeForFormat(IE.Format);
  747. }
  748. D3D12_VERTEX_BUFFER_VIEW vertexBufferView;
  749. vertexBufferView.BufferLocation = VBufferData.Resource->GetGPUVirtualAddress();
  750. vertexBufferView.StrideInBytes = strideInBytes;
  751. vertexBufferView.SizeInBytes = (UINT)VBufferData.ShaderOpRes->Desc.Width;
  752. pList->IASetVertexBuffers(0, 1, &vertexBufferView);
  753. UINT vertexCount = vertexBufferView.SizeInBytes / vertexBufferView.StrideInBytes;
  754. UINT instanceCount = 1;
  755. UINT vertexCountPerInstance = vertexCount / instanceCount;
  756. pList->BeginQuery(m_pQueryHeap, D3D12_QUERY_TYPE_PIPELINE_STATISTICS, 0);
  757. pList->DrawInstanced(vertexCountPerInstance, instanceCount, 0, 0);
  758. pList->EndQuery(m_pQueryHeap, D3D12_QUERY_TYPE_PIPELINE_STATISTICS, 0);
  759. pList->ResolveQueryData(m_pQueryHeap, D3D12_QUERY_TYPE_PIPELINE_STATISTICS,
  760. 0, 1, m_pQueryBuffer, 0);
  761. }
  762. CHECK_HR(pList->Close());
  763. ExecuteCommandList(m_CommandList.Queue, pList);
  764. WaitForSignal(m_CommandList.Queue, m_pFence, m_hFence, m_FenceValue++);
  765. }
  766. void ShaderOpTest::RunShaderOp(ShaderOp *pShaderOp) {
  767. m_pShaderOp = pShaderOp;
  768. CreateDevice();
  769. CreateResources();
  770. CreateDescriptorHeaps();
  771. CreatePipelineState();
  772. CreateCommandList();
  773. RunCommandList();
  774. CopyBackResources();
  775. }
  776. void ShaderOpTest::RunShaderOp(std::shared_ptr<ShaderOp> ShaderOp) {
  777. m_OrigShaderOp = ShaderOp;
  778. RunShaderOp(m_OrigShaderOp.get());
  779. }
  780. void ShaderOpTest::SetRootValues(ID3D12GraphicsCommandList *pList,
  781. bool isCompute) {
  782. for (size_t i = 0; i < m_pShaderOp->RootValues.size(); ++i) {
  783. ShaderOpRootValue &V = m_pShaderOp->RootValues[i];
  784. UINT idx = V.Index == 0 ? (UINT)i : V.Index;
  785. if (V.ResName) {
  786. auto r_it = m_ResourceData.find(V.ResName);
  787. if (r_it == m_ResourceData.end()) {
  788. ShaderOpLogFmt(L"Root value #%u refers to missing resource %S", (unsigned)i, V.ResName);
  789. CHECK_HR(E_INVALIDARG);
  790. }
  791. // Issue a warning for trying to bind textures (GPU address will return null)
  792. ShaderOpResourceData &D = r_it->second;
  793. ID3D12Resource *pRes = D.Resource;
  794. if (isCompute) {
  795. switch (D.ShaderOpRes->TransitionTo) {
  796. case D3D12_RESOURCE_STATE_VERTEX_AND_CONSTANT_BUFFER:
  797. pList->SetComputeRootConstantBufferView(idx,
  798. pRes->GetGPUVirtualAddress());
  799. break;
  800. case D3D12_RESOURCE_STATE_UNORDERED_ACCESS:
  801. pList->SetComputeRootUnorderedAccessView(idx,
  802. pRes->GetGPUVirtualAddress());
  803. break;
  804. case D3D12_RESOURCE_STATE_NON_PIXEL_SHADER_RESOURCE:
  805. default:
  806. pList->SetComputeRootShaderResourceView(idx,
  807. pRes->GetGPUVirtualAddress());
  808. break;
  809. }
  810. }
  811. else {
  812. switch (D.ShaderOpRes->TransitionTo) {
  813. case D3D12_RESOURCE_STATE_VERTEX_AND_CONSTANT_BUFFER:
  814. pList->SetGraphicsRootConstantBufferView(idx,
  815. pRes->GetGPUVirtualAddress());
  816. break;
  817. case D3D12_RESOURCE_STATE_UNORDERED_ACCESS:
  818. pList->SetGraphicsRootUnorderedAccessView(idx,
  819. pRes->GetGPUVirtualAddress());
  820. break;
  821. case D3D12_RESOURCE_STATE_NON_PIXEL_SHADER_RESOURCE:
  822. default:
  823. pList->SetGraphicsRootShaderResourceView(idx,
  824. pRes->GetGPUVirtualAddress());
  825. break;
  826. }
  827. }
  828. }
  829. else if (V.HeapName) {
  830. D3D12_GPU_DESCRIPTOR_HANDLE heapBase(m_DescriptorHeapsByName[V.HeapName]->GetGPUDescriptorHandleForHeapStart());
  831. if (isCompute) {
  832. pList->SetComputeRootDescriptorTable(idx, heapBase);
  833. }
  834. else {
  835. pList->SetGraphicsRootDescriptorTable(idx, heapBase);
  836. }
  837. }
  838. }
  839. }
  840. void ShaderOpTest::SetDevice(ID3D12Device *pDevice) {
  841. m_pDevice = pDevice;
  842. }
  843. void ShaderOpTest::SetDxcSupport(dxc::DxcDllSupport *pDxcSupport) {
  844. m_pDxcSupport = pDxcSupport;
  845. }
  846. void ShaderOpTest::SetInitCallback(TInitCallbackFn InitCallbackFn) {
  847. m_InitCallbackFn = InitCallbackFn;
  848. }
  849. void ShaderOpTest::SetupRenderTarget(ShaderOp *pShaderOp, ID3D12Device *pDevice,
  850. ID3D12CommandQueue *pCommandQueue,
  851. ID3D12Resource *pRenderTarget) {
  852. SetDevice(pDevice);
  853. m_CommandList.Queue = pCommandQueue;
  854. // Simplification - add the render target name if missing, set it up 'by hand' if not.
  855. if (pShaderOp->RenderTargets.empty()) {
  856. pShaderOp->RenderTargets.push_back(pShaderOp->Strings.insert("RTarget"));
  857. ShaderOpResource R;
  858. ZeroMemory(&R, sizeof(R));
  859. R.Desc = pRenderTarget->GetDesc();
  860. R.Name = pShaderOp->Strings.insert("RTarget");
  861. R.HeapFlags = D3D12_HEAP_FLAG_NONE;
  862. R.Init = nullptr;
  863. R.InitialResourceState = D3D12_RESOURCE_STATE_PRESENT;
  864. R.ReadBack = FALSE;
  865. pShaderOp->Resources.push_back(R);
  866. ShaderOpResourceData &D = m_ResourceData[R.Name];
  867. D.ShaderOpRes = &pShaderOp->Resources.back();
  868. D.Resource = pRenderTarget;
  869. D.ResourceState = R.InitialResourceState;
  870. }
  871. // Create a render target heap to put this in.
  872. ShaderOpDescriptorHeap *pRtvHeap = pShaderOp->GetDescriptorHeapByName("RtvHeap");
  873. if (pRtvHeap == nullptr) {
  874. ShaderOpDescriptorHeap H;
  875. ZeroMemory(&H, sizeof(H));
  876. H.Name = pShaderOp->Strings.insert("RtvHeap");
  877. H.Desc.NumDescriptors = 1;
  878. H.Desc.Type = D3D12_DESCRIPTOR_HEAP_TYPE_RTV;
  879. pShaderOp->DescriptorHeaps.push_back(H);
  880. pRtvHeap = &pShaderOp->DescriptorHeaps.back();
  881. }
  882. if (pRtvHeap->Descriptors.empty()) {
  883. ShaderOpDescriptor D;
  884. ZeroMemory(&D, sizeof(D));
  885. D.Name = pShaderOp->Strings.insert("RTarget");
  886. D.ResName = D.Name;
  887. D.Kind = pShaderOp->Strings.insert("RTV");
  888. pRtvHeap->Descriptors.push_back(D);
  889. }
  890. }
  891. void ShaderOpTest::PresentRenderTarget(ShaderOp *pShaderOp,
  892. ID3D12CommandQueue *pCommandQueue,
  893. ID3D12Resource *pRenderTarget) {
  894. UNREFERENCED_PARAMETER(pShaderOp);
  895. CommandListRefs ResCommandList;
  896. ResCommandList.Queue = pCommandQueue;
  897. ResCommandList.CreateForDevice(m_pDevice, m_pShaderOp->IsCompute());
  898. ID3D12GraphicsCommandList *pList = ResCommandList.List;
  899. pList->SetName(L"ShaderOpTest Resource Present CommandList");
  900. RecordTransitionBarrier(pList, pRenderTarget,
  901. D3D12_RESOURCE_STATE_RENDER_TARGET,
  902. D3D12_RESOURCE_STATE_PRESENT);
  903. pList->Close();
  904. ExecuteCommandList(ResCommandList.Queue, pList);
  905. WaitForSignal(ResCommandList.Queue, m_pFence, m_hFence, m_FenceValue++);
  906. }
  907. ShaderOp *ShaderOpSet::GetShaderOp(LPCSTR pName) {
  908. for (auto &S : ShaderOps) {
  909. if (S->Name && 0 == _stricmp(pName, S->Name)) {
  910. return S.get();
  911. }
  912. }
  913. return nullptr;
  914. }
  915. ///////////////////////////////////////////////////////////////////////////////
  916. // ShaderOpTest library implementation for deserialization.
  917. #pragma region Parsing support
  918. // Use this class to initialize a ShaderOp object from an XML document.
  919. class ShaderOpParser {
  920. private:
  921. string_table *m_pStrings;
  922. bool ReadAtElementName(IXmlReader *pReader, LPCWSTR pName);
  923. HRESULT ReadAttrStr(IXmlReader *pReader, LPCWSTR pAttrName, LPCSTR *ppValue);
  924. HRESULT ReadAttrBOOL(IXmlReader *pReader, LPCWSTR pAttrName, BOOL *pValue, BOOL defaultValue = FALSE);
  925. HRESULT ReadAttrUINT64(IXmlReader *pReader, LPCWSTR pAttrName, UINT64 *pValue, UINT64 defaultValue = 0);
  926. HRESULT ReadAttrUINT16(IXmlReader *pReader, LPCWSTR pAttrName, UINT16 *pValue, UINT16 defaultValue = 0);
  927. HRESULT ReadAttrUINT(IXmlReader *pReader, LPCWSTR pAttrName, UINT *pValue, UINT defaultValue = 0);
  928. void ReadElementContentStr(IXmlReader *pReader, LPCSTR *ppValue);
  929. void ParseDescriptor(IXmlReader *pReader, ShaderOpDescriptor *pDesc);
  930. void ParseDescriptorHeap(IXmlReader *pReader, ShaderOpDescriptorHeap *pHeap);
  931. void ParseInputElement(IXmlReader *pReader, D3D12_INPUT_ELEMENT_DESC *pInputElement);
  932. void ParseInputElements(IXmlReader *pReader, std::vector<D3D12_INPUT_ELEMENT_DESC> *pInputElements);
  933. void ParseRenderTargets(IXmlReader *pReader, std::vector<LPCSTR> *pRenderTargets);
  934. void ParseRootValue(IXmlReader *pReader, ShaderOpRootValue *pRootValue);
  935. void ParseRootValues(IXmlReader *pReader, std::vector<ShaderOpRootValue> *pRootValues);
  936. void ParseResource(IXmlReader *pReader, ShaderOpResource *pResource);
  937. void ParseShader(IXmlReader *pReader, ShaderOpShader *pShader);
  938. public:
  939. void ParseShaderOpSet(IStream *pStream, ShaderOpSet *pShaderOpSet);
  940. void ParseShaderOpSet(IXmlReader *pReader, ShaderOpSet *pShaderOpSet);
  941. void ParseShaderOp(IXmlReader *pReader, ShaderOp *pShaderOp);
  942. };
  943. void ParseShaderOpSetFromStream(IStream *pStream, st::ShaderOpSet *pShaderOpSet) {
  944. ShaderOpParser parser;
  945. parser.ParseShaderOpSet(pStream, pShaderOpSet);
  946. }
  947. void ParseShaderOpSetFromXml(IXmlReader *pReader, st::ShaderOpSet *pShaderOpSet) {
  948. ShaderOpParser parser;
  949. parser.ParseShaderOpSet(pReader, pShaderOpSet);
  950. }
  951. enum class ParserEnumKind {
  952. INPUT_CLASSIFICATION,
  953. DXGI_FORMAT,
  954. HEAP_TYPE,
  955. CPU_PAGE_PROPERTY,
  956. MEMORY_POOL,
  957. RESOURCE_DIMENSION,
  958. TEXTURE_LAYOUT,
  959. RESOURCE_FLAG,
  960. HEAP_FLAG,
  961. RESOURCE_STATE,
  962. DESCRIPTOR_HEAP_TYPE,
  963. DESCRIPTOR_HEAP_FLAG,
  964. SRV_DIMENSION,
  965. UAV_DIMENSION,
  966. PRIMITIVE_TOPOLOGY,
  967. PRIMITIVE_TOPOLOGY_TYPE
  968. };
  969. struct ParserEnumValue {
  970. LPCWSTR Name;
  971. UINT Value;
  972. };
  973. struct ParserEnumTable {
  974. size_t ValueCount;
  975. const ParserEnumValue *Values;
  976. ParserEnumKind Kind;
  977. };
  978. static const ParserEnumValue INPUT_CLASSIFICATION_TABLE[] = {
  979. { L"INSTANCE", D3D12_INPUT_CLASSIFICATION_PER_INSTANCE_DATA },
  980. { L"VERTEX", D3D12_INPUT_CLASSIFICATION_PER_VERTEX_DATA }
  981. };
  982. static const ParserEnumValue DXGI_FORMAT_TABLE[] = {
  983. { L"UNKNOWN", DXGI_FORMAT_UNKNOWN },
  984. { L"R32G32B32A32_TYPELESS", DXGI_FORMAT_R32G32B32A32_TYPELESS },
  985. { L"R32G32B32A32_FLOAT", DXGI_FORMAT_R32G32B32A32_FLOAT },
  986. { L"R32G32B32A32_UINT", DXGI_FORMAT_R32G32B32A32_UINT },
  987. { L"R32G32B32A32_SINT", DXGI_FORMAT_R32G32B32A32_SINT },
  988. { L"R32G32B32_TYPELESS", DXGI_FORMAT_R32G32B32_TYPELESS },
  989. { L"R32G32B32_FLOAT", DXGI_FORMAT_R32G32B32_FLOAT },
  990. { L"R32G32B32_UINT", DXGI_FORMAT_R32G32B32_UINT },
  991. { L"R32G32B32_SINT", DXGI_FORMAT_R32G32B32_SINT },
  992. { L"R16G16B16A16_TYPELESS", DXGI_FORMAT_R16G16B16A16_TYPELESS },
  993. { L"R16G16B16A16_FLOAT", DXGI_FORMAT_R16G16B16A16_FLOAT },
  994. { L"R16G16B16A16_UNORM", DXGI_FORMAT_R16G16B16A16_UNORM },
  995. { L"R16G16B16A16_UINT", DXGI_FORMAT_R16G16B16A16_UINT },
  996. { L"R16G16B16A16_SNORM", DXGI_FORMAT_R16G16B16A16_SNORM },
  997. { L"R16G16B16A16_SINT", DXGI_FORMAT_R16G16B16A16_SINT },
  998. { L"R32G32_TYPELESS", DXGI_FORMAT_R32G32_TYPELESS },
  999. { L"R32G32_FLOAT", DXGI_FORMAT_R32G32_FLOAT },
  1000. { L"R32G32_UINT", DXGI_FORMAT_R32G32_UINT },
  1001. { L"R32G32_SINT", DXGI_FORMAT_R32G32_SINT },
  1002. { L"R32G8X24_TYPELESS", DXGI_FORMAT_R32G8X24_TYPELESS },
  1003. { L"D32_FLOAT_S8X24_UINT", DXGI_FORMAT_D32_FLOAT_S8X24_UINT },
  1004. { L"R32_FLOAT_X8X24_TYPELESS", DXGI_FORMAT_R32_FLOAT_X8X24_TYPELESS },
  1005. { L"X32_TYPELESS_G8X24_UINT", DXGI_FORMAT_X32_TYPELESS_G8X24_UINT },
  1006. { L"R10G10B10A2_TYPELESS", DXGI_FORMAT_R10G10B10A2_TYPELESS },
  1007. { L"R10G10B10A2_UNORM", DXGI_FORMAT_R10G10B10A2_UNORM },
  1008. { L"R10G10B10A2_UINT", DXGI_FORMAT_R10G10B10A2_UINT },
  1009. { L"R11G11B10_FLOAT", DXGI_FORMAT_R11G11B10_FLOAT },
  1010. { L"R8G8B8A8_TYPELESS", DXGI_FORMAT_R8G8B8A8_TYPELESS },
  1011. { L"R8G8B8A8_UNORM", DXGI_FORMAT_R8G8B8A8_UNORM },
  1012. { L"R8G8B8A8_UNORM_SRGB", DXGI_FORMAT_R8G8B8A8_UNORM_SRGB },
  1013. { L"R8G8B8A8_UINT", DXGI_FORMAT_R8G8B8A8_UINT },
  1014. { L"R8G8B8A8_SNORM", DXGI_FORMAT_R8G8B8A8_SNORM },
  1015. { L"R8G8B8A8_SINT", DXGI_FORMAT_R8G8B8A8_SINT },
  1016. { L"R16G16_TYPELESS", DXGI_FORMAT_R16G16_TYPELESS },
  1017. { L"R16G16_FLOAT", DXGI_FORMAT_R16G16_FLOAT },
  1018. { L"R16G16_UNORM", DXGI_FORMAT_R16G16_UNORM },
  1019. { L"R16G16_UINT", DXGI_FORMAT_R16G16_UINT },
  1020. { L"R16G16_SNORM", DXGI_FORMAT_R16G16_SNORM },
  1021. { L"R16G16_SINT", DXGI_FORMAT_R16G16_SINT },
  1022. { L"R32_TYPELESS", DXGI_FORMAT_R32_TYPELESS },
  1023. { L"D32_FLOAT", DXGI_FORMAT_D32_FLOAT },
  1024. { L"R32_FLOAT", DXGI_FORMAT_R32_FLOAT },
  1025. { L"R32_UINT", DXGI_FORMAT_R32_UINT },
  1026. { L"R32_SINT", DXGI_FORMAT_R32_SINT },
  1027. { L"R24G8_TYPELESS", DXGI_FORMAT_R24G8_TYPELESS },
  1028. { L"D24_UNORM_S8_UINT", DXGI_FORMAT_D24_UNORM_S8_UINT },
  1029. { L"R24_UNORM_X8_TYPELESS", DXGI_FORMAT_R24_UNORM_X8_TYPELESS },
  1030. { L"X24_TYPELESS_G8_UINT", DXGI_FORMAT_X24_TYPELESS_G8_UINT },
  1031. { L"R8G8_TYPELESS", DXGI_FORMAT_R8G8_TYPELESS },
  1032. { L"R8G8_UNORM", DXGI_FORMAT_R8G8_UNORM },
  1033. { L"R8G8_UINT", DXGI_FORMAT_R8G8_UINT },
  1034. { L"R8G8_SNORM", DXGI_FORMAT_R8G8_SNORM },
  1035. { L"R8G8_SINT", DXGI_FORMAT_R8G8_SINT },
  1036. { L"R16_TYPELESS", DXGI_FORMAT_R16_TYPELESS },
  1037. { L"R16_FLOAT", DXGI_FORMAT_R16_FLOAT },
  1038. { L"D16_UNORM", DXGI_FORMAT_D16_UNORM },
  1039. { L"R16_UNORM", DXGI_FORMAT_R16_UNORM },
  1040. { L"R16_UINT", DXGI_FORMAT_R16_UINT },
  1041. { L"R16_SNORM", DXGI_FORMAT_R16_SNORM },
  1042. { L"R16_SINT", DXGI_FORMAT_R16_SINT },
  1043. { L"R8_TYPELESS", DXGI_FORMAT_R8_TYPELESS },
  1044. { L"R8_UNORM", DXGI_FORMAT_R8_UNORM },
  1045. { L"R8_UINT", DXGI_FORMAT_R8_UINT },
  1046. { L"R8_SNORM", DXGI_FORMAT_R8_SNORM },
  1047. { L"R8_SINT", DXGI_FORMAT_R8_SINT },
  1048. { L"A8_UNORM", DXGI_FORMAT_A8_UNORM },
  1049. { L"R1_UNORM", DXGI_FORMAT_R1_UNORM },
  1050. { L"R9G9B9E5_SHAREDEXP", DXGI_FORMAT_R9G9B9E5_SHAREDEXP },
  1051. { L"R8G8_B8G8_UNORM", DXGI_FORMAT_R8G8_B8G8_UNORM },
  1052. { L"G8R8_G8B8_UNORM", DXGI_FORMAT_G8R8_G8B8_UNORM },
  1053. { L"BC1_TYPELESS", DXGI_FORMAT_BC1_TYPELESS },
  1054. { L"BC1_UNORM", DXGI_FORMAT_BC1_UNORM },
  1055. { L"BC1_UNORM_SRGB", DXGI_FORMAT_BC1_UNORM_SRGB },
  1056. { L"BC2_TYPELESS", DXGI_FORMAT_BC2_TYPELESS },
  1057. { L"BC2_UNORM", DXGI_FORMAT_BC2_UNORM },
  1058. { L"BC2_UNORM_SRGB", DXGI_FORMAT_BC2_UNORM_SRGB },
  1059. { L"BC3_TYPELESS", DXGI_FORMAT_BC3_TYPELESS },
  1060. { L"BC3_UNORM", DXGI_FORMAT_BC3_UNORM },
  1061. { L"BC3_UNORM_SRGB", DXGI_FORMAT_BC3_UNORM_SRGB },
  1062. { L"BC4_TYPELESS", DXGI_FORMAT_BC4_TYPELESS },
  1063. { L"BC4_UNORM", DXGI_FORMAT_BC4_UNORM },
  1064. { L"BC4_SNORM", DXGI_FORMAT_BC4_SNORM },
  1065. { L"BC5_TYPELESS", DXGI_FORMAT_BC5_TYPELESS },
  1066. { L"BC5_UNORM", DXGI_FORMAT_BC5_UNORM },
  1067. { L"BC5_SNORM", DXGI_FORMAT_BC5_SNORM },
  1068. { L"B5G6R5_UNORM", DXGI_FORMAT_B5G6R5_UNORM },
  1069. { L"B5G5R5A1_UNORM", DXGI_FORMAT_B5G5R5A1_UNORM },
  1070. { L"B8G8R8A8_UNORM", DXGI_FORMAT_B8G8R8A8_UNORM },
  1071. { L"B8G8R8X8_UNORM", DXGI_FORMAT_B8G8R8X8_UNORM },
  1072. { L"R10G10B10_XR_BIAS_A2_UNORM", DXGI_FORMAT_R10G10B10_XR_BIAS_A2_UNORM },
  1073. { L"B8G8R8A8_TYPELESS", DXGI_FORMAT_B8G8R8A8_TYPELESS },
  1074. { L"B8G8R8A8_UNORM_SRGB", DXGI_FORMAT_B8G8R8A8_UNORM_SRGB },
  1075. { L"B8G8R8X8_TYPELESS", DXGI_FORMAT_B8G8R8X8_TYPELESS },
  1076. { L"B8G8R8X8_UNORM_SRGB", DXGI_FORMAT_B8G8R8X8_UNORM_SRGB },
  1077. { L"BC6H_TYPELESS", DXGI_FORMAT_BC6H_TYPELESS },
  1078. { L"BC6H_UF16", DXGI_FORMAT_BC6H_UF16 },
  1079. { L"BC6H_SF16", DXGI_FORMAT_BC6H_SF16 },
  1080. { L"BC7_TYPELESS", DXGI_FORMAT_BC7_TYPELESS },
  1081. { L"BC7_UNORM", DXGI_FORMAT_BC7_UNORM },
  1082. { L"BC7_UNORM_SRGB", DXGI_FORMAT_BC7_UNORM_SRGB },
  1083. { L"AYUV", DXGI_FORMAT_AYUV },
  1084. { L"Y410", DXGI_FORMAT_Y410 },
  1085. { L"Y416", DXGI_FORMAT_Y416 },
  1086. { L"NV12", DXGI_FORMAT_NV12 },
  1087. { L"P010", DXGI_FORMAT_P010 },
  1088. { L"P016", DXGI_FORMAT_P016 },
  1089. { L"420_OPAQUE", DXGI_FORMAT_420_OPAQUE },
  1090. { L"YUY2", DXGI_FORMAT_YUY2 },
  1091. { L"Y210", DXGI_FORMAT_Y210 },
  1092. { L"Y216", DXGI_FORMAT_Y216 },
  1093. { L"NV11", DXGI_FORMAT_NV11 },
  1094. { L"AI44", DXGI_FORMAT_AI44 },
  1095. { L"IA44", DXGI_FORMAT_IA44 },
  1096. { L"P8", DXGI_FORMAT_P8 },
  1097. { L"A8P8", DXGI_FORMAT_A8P8 },
  1098. { L"B4G4R4A4_UNORM", DXGI_FORMAT_B4G4R4A4_UNORM },
  1099. { L"P208", DXGI_FORMAT_P208 },
  1100. { L"V208", DXGI_FORMAT_V208 },
  1101. { L"V408", DXGI_FORMAT_V408 }
  1102. };
  1103. static const ParserEnumValue HEAP_TYPE_TABLE[] = {
  1104. { L"DEFAULT", D3D12_HEAP_TYPE_DEFAULT },
  1105. { L"UPLOAD", D3D12_HEAP_TYPE_UPLOAD },
  1106. { L"READBACK", D3D12_HEAP_TYPE_READBACK },
  1107. { L"CUSTOM", D3D12_HEAP_TYPE_CUSTOM }
  1108. };
  1109. static const ParserEnumValue CPU_PAGE_PROPERTY_TABLE[] = {
  1110. { L"UNKNOWN", D3D12_CPU_PAGE_PROPERTY_UNKNOWN },
  1111. { L"NOT_AVAILABLE", D3D12_CPU_PAGE_PROPERTY_NOT_AVAILABLE },
  1112. { L"WRITE_COMBINE", D3D12_CPU_PAGE_PROPERTY_WRITE_COMBINE },
  1113. { L"WRITE_BACK", D3D12_CPU_PAGE_PROPERTY_WRITE_BACK }
  1114. };
  1115. static const ParserEnumValue MEMORY_POOL_TABLE[] = {
  1116. { L"UNKNOWN", D3D12_MEMORY_POOL_UNKNOWN },
  1117. { L"L0 ", D3D12_MEMORY_POOL_L0 },
  1118. { L"L1", D3D12_MEMORY_POOL_L1 }
  1119. };
  1120. static const ParserEnumValue RESOURCE_DIMENSION_TABLE[] = {
  1121. { L"UNKNOWN", D3D12_RESOURCE_DIMENSION_UNKNOWN },
  1122. { L"BUFFER", D3D12_RESOURCE_DIMENSION_BUFFER },
  1123. { L"TEXTURE1D", D3D12_RESOURCE_DIMENSION_TEXTURE1D },
  1124. { L"TEXTURE2D", D3D12_RESOURCE_DIMENSION_TEXTURE2D },
  1125. { L"TEXTURE3D", D3D12_RESOURCE_DIMENSION_TEXTURE3D }
  1126. };
  1127. static const ParserEnumValue TEXTURE_LAYOUT_TABLE[] = {
  1128. { L"UNKNOWN", D3D12_TEXTURE_LAYOUT_UNKNOWN },
  1129. { L"ROW_MAJOR", D3D12_TEXTURE_LAYOUT_ROW_MAJOR },
  1130. { L"UNDEFINED_SWIZZLE", D3D12_TEXTURE_LAYOUT_64KB_UNDEFINED_SWIZZLE },
  1131. { L"STANDARD_SWIZZLE", D3D12_TEXTURE_LAYOUT_64KB_STANDARD_SWIZZLE }
  1132. };
  1133. static const ParserEnumValue RESOURCE_FLAG_TABLE[] = {
  1134. { L"NONE", D3D12_RESOURCE_FLAG_NONE },
  1135. { L"ALLOW_RENDER_TARGET", D3D12_RESOURCE_FLAG_ALLOW_RENDER_TARGET },
  1136. { L"ALLOW_DEPTH_STENCIL", D3D12_RESOURCE_FLAG_ALLOW_DEPTH_STENCIL },
  1137. { L"ALLOW_UNORDERED_ACCESS", D3D12_RESOURCE_FLAG_ALLOW_UNORDERED_ACCESS },
  1138. { L"DENY_SHADER_RESOURCE", D3D12_RESOURCE_FLAG_DENY_SHADER_RESOURCE },
  1139. { L"ALLOW_CROSS_ADAPTER", D3D12_RESOURCE_FLAG_ALLOW_CROSS_ADAPTER },
  1140. { L"ALLOW_SIMULTANEOUS_ACCESS", D3D12_RESOURCE_FLAG_ALLOW_SIMULTANEOUS_ACCESS }
  1141. };
  1142. static const ParserEnumValue HEAP_FLAG_TABLE[] = {
  1143. { L"NONE", D3D12_HEAP_FLAG_NONE },
  1144. { L"SHARED", D3D12_HEAP_FLAG_SHARED },
  1145. { L"DENY_BUFFERS", D3D12_HEAP_FLAG_DENY_BUFFERS },
  1146. { L"ALLOW_DISPLAY", D3D12_HEAP_FLAG_ALLOW_DISPLAY },
  1147. { L"SHARED_CROSS_ADAPTER", D3D12_HEAP_FLAG_SHARED_CROSS_ADAPTER },
  1148. { L"DENY_RT_DS_TEXTURES", D3D12_HEAP_FLAG_DENY_RT_DS_TEXTURES },
  1149. { L"DENY_NON_RT_DS_TEXTURES", D3D12_HEAP_FLAG_DENY_NON_RT_DS_TEXTURES },
  1150. { L"ALLOW_ALL_BUFFERS_AND_TEXTURES",D3D12_HEAP_FLAG_ALLOW_ALL_BUFFERS_AND_TEXTURES },
  1151. { L"ALLOW_ONLY_BUFFERS", D3D12_HEAP_FLAG_ALLOW_ONLY_BUFFERS },
  1152. { L"ALLOW_ONLY_NON_RT_DS_TEXTURES", D3D12_HEAP_FLAG_ALLOW_ONLY_NON_RT_DS_TEXTURES },
  1153. { L"ALLOW_ONLY_RT_DS_TEXTURES", D3D12_HEAP_FLAG_ALLOW_ONLY_RT_DS_TEXTURES }
  1154. };
  1155. static const ParserEnumValue RESOURCE_STATE_TABLE[] = {
  1156. { L"COMMON", D3D12_RESOURCE_STATE_COMMON },
  1157. { L"VERTEX_AND_CONSTANT_BUFFER", D3D12_RESOURCE_STATE_VERTEX_AND_CONSTANT_BUFFER },
  1158. { L"INDEX_BUFFER", D3D12_RESOURCE_STATE_INDEX_BUFFER },
  1159. { L"RENDER_TARGET", D3D12_RESOURCE_STATE_RENDER_TARGET },
  1160. { L"UNORDERED_ACCESS", D3D12_RESOURCE_STATE_UNORDERED_ACCESS },
  1161. { L"DEPTH_WRITE", D3D12_RESOURCE_STATE_DEPTH_WRITE },
  1162. { L"DEPTH_READ", D3D12_RESOURCE_STATE_DEPTH_READ },
  1163. { L"NON_PIXEL_SHADER_RESOURCE", D3D12_RESOURCE_STATE_NON_PIXEL_SHADER_RESOURCE },
  1164. { L"PIXEL_SHADER_RESOURCE", D3D12_RESOURCE_STATE_PIXEL_SHADER_RESOURCE },
  1165. { L"STREAM_OUT", D3D12_RESOURCE_STATE_STREAM_OUT },
  1166. { L"INDIRECT_ARGUMENT", D3D12_RESOURCE_STATE_INDIRECT_ARGUMENT },
  1167. { L"COPY_DEST", D3D12_RESOURCE_STATE_COPY_DEST },
  1168. { L"COPY_SOURCE", D3D12_RESOURCE_STATE_COPY_SOURCE },
  1169. { L"RESOLVE_DEST", D3D12_RESOURCE_STATE_RESOLVE_DEST },
  1170. { L"RESOLVE_SOURCE", D3D12_RESOURCE_STATE_RESOLVE_SOURCE },
  1171. { L"GENERIC_READ", D3D12_RESOURCE_STATE_GENERIC_READ },
  1172. { L"PRESENT", D3D12_RESOURCE_STATE_PRESENT },
  1173. { L"PREDICATION", D3D12_RESOURCE_STATE_PREDICATION }
  1174. };
  1175. static const ParserEnumValue DESCRIPTOR_HEAP_TYPE_TABLE[] = {
  1176. { L"CBV_SRV_UAV", D3D12_DESCRIPTOR_HEAP_TYPE_CBV_SRV_UAV },
  1177. { L"SAMPLER", D3D12_DESCRIPTOR_HEAP_TYPE_SAMPLER },
  1178. { L"RTV", D3D12_DESCRIPTOR_HEAP_TYPE_RTV },
  1179. { L"DSV", D3D12_DESCRIPTOR_HEAP_TYPE_DSV }
  1180. };
  1181. static const ParserEnumValue DESCRIPTOR_HEAP_FLAG_TABLE[] = {
  1182. { L"NONE", D3D12_DESCRIPTOR_HEAP_FLAG_NONE },
  1183. { L"SHADER_VISIBLE", D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE }
  1184. };
  1185. static const ParserEnumValue SRV_DIMENSION_TABLE[] = {
  1186. { L"UNKNOWN", D3D12_SRV_DIMENSION_UNKNOWN },
  1187. { L"BUFFER", D3D12_SRV_DIMENSION_BUFFER },
  1188. { L"TEXTURE1D", D3D12_SRV_DIMENSION_TEXTURE1D },
  1189. { L"TEXTURE1DARRAY", D3D12_SRV_DIMENSION_TEXTURE1DARRAY },
  1190. { L"TEXTURE2D", D3D12_SRV_DIMENSION_TEXTURE2D },
  1191. { L"TEXTURE2DARRAY", D3D12_SRV_DIMENSION_TEXTURE2DARRAY },
  1192. { L"TEXTURE2DMS", D3D12_SRV_DIMENSION_TEXTURE2DMS },
  1193. { L"TEXTURE2DMSARRAY", D3D12_SRV_DIMENSION_TEXTURE2DMSARRAY },
  1194. { L"TEXTURE3D", D3D12_SRV_DIMENSION_TEXTURE3D },
  1195. { L"TEXTURECUBE", D3D12_SRV_DIMENSION_TEXTURECUBE },
  1196. { L"TEXTURECUBEARRAY", D3D12_SRV_DIMENSION_TEXTURECUBEARRAY }
  1197. };
  1198. static const ParserEnumValue UAV_DIMENSION_TABLE[] = {
  1199. { L"UNKNOWN", D3D12_UAV_DIMENSION_UNKNOWN },
  1200. { L"BUFFER", D3D12_UAV_DIMENSION_BUFFER },
  1201. { L"TEXTURE1D", D3D12_UAV_DIMENSION_TEXTURE1D },
  1202. { L"TEXTURE1DARRAY", D3D12_UAV_DIMENSION_TEXTURE1DARRAY },
  1203. { L"TEXTURE2D", D3D12_UAV_DIMENSION_TEXTURE2D },
  1204. { L"TEXTURE2DARRAY", D3D12_UAV_DIMENSION_TEXTURE2DARRAY },
  1205. { L"TEXTURE3D", D3D12_UAV_DIMENSION_TEXTURE3D }
  1206. };
  1207. static const ParserEnumValue PRIMITIVE_TOPOLOGY_TABLE[] = {
  1208. { L"UNDEFINED",D3D_PRIMITIVE_TOPOLOGY_UNDEFINED },
  1209. { L"POINTLIST",D3D_PRIMITIVE_TOPOLOGY_POINTLIST },
  1210. { L"LINELIST",D3D_PRIMITIVE_TOPOLOGY_LINELIST },
  1211. { L"LINESTRIP",D3D_PRIMITIVE_TOPOLOGY_LINESTRIP },
  1212. { L"TRIANGLELIST",D3D_PRIMITIVE_TOPOLOGY_TRIANGLELIST },
  1213. { L"TRIANGLESTRIP",D3D_PRIMITIVE_TOPOLOGY_TRIANGLESTRIP },
  1214. { L"LINELIST_ADJ",D3D_PRIMITIVE_TOPOLOGY_LINELIST_ADJ },
  1215. { L"LINESTRIP_ADJ",D3D_PRIMITIVE_TOPOLOGY_LINESTRIP_ADJ },
  1216. { L"TRIANGLELIST_ADJ",D3D_PRIMITIVE_TOPOLOGY_TRIANGLELIST_ADJ },
  1217. { L"TRIANGLESTRIP_ADJ",D3D_PRIMITIVE_TOPOLOGY_TRIANGLESTRIP_ADJ },
  1218. { L"1_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_1_CONTROL_POINT_PATCHLIST },
  1219. { L"2_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_2_CONTROL_POINT_PATCHLIST },
  1220. { L"3_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_3_CONTROL_POINT_PATCHLIST },
  1221. { L"4_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_4_CONTROL_POINT_PATCHLIST },
  1222. { L"5_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_5_CONTROL_POINT_PATCHLIST },
  1223. { L"6_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_6_CONTROL_POINT_PATCHLIST },
  1224. { L"7_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_7_CONTROL_POINT_PATCHLIST },
  1225. { L"8_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_8_CONTROL_POINT_PATCHLIST },
  1226. { L"9_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_9_CONTROL_POINT_PATCHLIST },
  1227. { L"10_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_10_CONTROL_POINT_PATCHLIST },
  1228. { L"11_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_11_CONTROL_POINT_PATCHLIST },
  1229. { L"12_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_12_CONTROL_POINT_PATCHLIST },
  1230. { L"13_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_13_CONTROL_POINT_PATCHLIST },
  1231. { L"14_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_14_CONTROL_POINT_PATCHLIST },
  1232. { L"15_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_15_CONTROL_POINT_PATCHLIST },
  1233. { L"16_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_16_CONTROL_POINT_PATCHLIST },
  1234. { L"17_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_17_CONTROL_POINT_PATCHLIST },
  1235. { L"18_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_18_CONTROL_POINT_PATCHLIST },
  1236. { L"19_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_19_CONTROL_POINT_PATCHLIST },
  1237. { L"20_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_20_CONTROL_POINT_PATCHLIST },
  1238. { L"21_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_21_CONTROL_POINT_PATCHLIST },
  1239. { L"22_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_22_CONTROL_POINT_PATCHLIST },
  1240. { L"23_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_23_CONTROL_POINT_PATCHLIST },
  1241. { L"24_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_24_CONTROL_POINT_PATCHLIST },
  1242. { L"25_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_25_CONTROL_POINT_PATCHLIST },
  1243. { L"26_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_26_CONTROL_POINT_PATCHLIST },
  1244. { L"27_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_27_CONTROL_POINT_PATCHLIST },
  1245. { L"28_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_28_CONTROL_POINT_PATCHLIST },
  1246. { L"29_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_29_CONTROL_POINT_PATCHLIST },
  1247. { L"30_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_30_CONTROL_POINT_PATCHLIST },
  1248. { L"31_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_31_CONTROL_POINT_PATCHLIST },
  1249. { L"32_CONTROL_POINT_PATCHLIST",D3D_PRIMITIVE_TOPOLOGY_32_CONTROL_POINT_PATCHLIST }
  1250. };
  1251. static const ParserEnumValue PRIMITIVE_TOPOLOGY_TYPE_TABLE[] = {
  1252. { L"UNDEFINED", D3D12_PRIMITIVE_TOPOLOGY_TYPE_UNDEFINED },
  1253. { L"POINT", D3D12_PRIMITIVE_TOPOLOGY_TYPE_POINT },
  1254. { L"LINE", D3D12_PRIMITIVE_TOPOLOGY_TYPE_LINE },
  1255. { L"TRIANGLE", D3D12_PRIMITIVE_TOPOLOGY_TYPE_TRIANGLE },
  1256. { L"PATCH", D3D12_PRIMITIVE_TOPOLOGY_TYPE_PATCH }
  1257. };
  1258. static const ParserEnumTable g_ParserEnumTables[] = {
  1259. { _countof(INPUT_CLASSIFICATION_TABLE), INPUT_CLASSIFICATION_TABLE, ParserEnumKind::INPUT_CLASSIFICATION },
  1260. { _countof(DXGI_FORMAT_TABLE), DXGI_FORMAT_TABLE, ParserEnumKind::DXGI_FORMAT },
  1261. { _countof(HEAP_TYPE_TABLE), HEAP_TYPE_TABLE, ParserEnumKind::HEAP_TYPE },
  1262. { _countof(CPU_PAGE_PROPERTY_TABLE), CPU_PAGE_PROPERTY_TABLE, ParserEnumKind::CPU_PAGE_PROPERTY },
  1263. { _countof(MEMORY_POOL_TABLE), MEMORY_POOL_TABLE, ParserEnumKind::MEMORY_POOL },
  1264. { _countof(RESOURCE_DIMENSION_TABLE), RESOURCE_DIMENSION_TABLE, ParserEnumKind::RESOURCE_DIMENSION },
  1265. { _countof(TEXTURE_LAYOUT_TABLE), TEXTURE_LAYOUT_TABLE, ParserEnumKind::TEXTURE_LAYOUT },
  1266. { _countof(RESOURCE_FLAG_TABLE), RESOURCE_FLAG_TABLE, ParserEnumKind::RESOURCE_FLAG },
  1267. { _countof(HEAP_FLAG_TABLE), HEAP_FLAG_TABLE, ParserEnumKind::HEAP_FLAG },
  1268. { _countof(RESOURCE_STATE_TABLE), RESOURCE_STATE_TABLE, ParserEnumKind::RESOURCE_STATE },
  1269. { _countof(DESCRIPTOR_HEAP_TYPE_TABLE), DESCRIPTOR_HEAP_TYPE_TABLE, ParserEnumKind::DESCRIPTOR_HEAP_TYPE },
  1270. { _countof(DESCRIPTOR_HEAP_FLAG_TABLE), DESCRIPTOR_HEAP_FLAG_TABLE, ParserEnumKind::DESCRIPTOR_HEAP_FLAG },
  1271. { _countof(SRV_DIMENSION_TABLE), SRV_DIMENSION_TABLE, ParserEnumKind::SRV_DIMENSION },
  1272. { _countof(UAV_DIMENSION_TABLE), UAV_DIMENSION_TABLE, ParserEnumKind::UAV_DIMENSION },
  1273. { _countof(PRIMITIVE_TOPOLOGY_TABLE), PRIMITIVE_TOPOLOGY_TABLE, ParserEnumKind::PRIMITIVE_TOPOLOGY },
  1274. { _countof(PRIMITIVE_TOPOLOGY_TYPE_TABLE), PRIMITIVE_TOPOLOGY_TYPE_TABLE, ParserEnumKind::PRIMITIVE_TOPOLOGY_TYPE },
  1275. };
  1276. static HRESULT GetEnumValue(LPCWSTR name, ParserEnumKind K, UINT *pValue) {
  1277. for (size_t i = 0; i < _countof(g_ParserEnumTables); ++i) {
  1278. const ParserEnumTable &T = g_ParserEnumTables[i];
  1279. if (T.Kind != K) {
  1280. continue;
  1281. }
  1282. for (size_t j = 0; j < T.ValueCount; ++j) {
  1283. if (_wcsicmp(name, T.Values[j].Name) == 0) {
  1284. *pValue = T.Values[j].Value;
  1285. return S_OK;
  1286. }
  1287. }
  1288. }
  1289. return E_INVALIDARG;
  1290. }
  1291. template <typename T>
  1292. static HRESULT GetEnumValueT(LPCWSTR name, ParserEnumKind K, T *pValue) {
  1293. UINT u;
  1294. HRESULT hr = GetEnumValue(name, K, &u);
  1295. *pValue = (T)u;
  1296. return hr;
  1297. }
  1298. template <typename T>
  1299. static HRESULT ReadAttrEnumT(IXmlReader *pReader, LPCWSTR pAttrName, ParserEnumKind K, T *pValue, T defaultValue, LPCWSTR pStripPrefix = nullptr) {
  1300. if (S_FALSE == CHECK_HR_RET(pReader->MoveToAttributeByName(pAttrName, nullptr))) {
  1301. *pValue = defaultValue;
  1302. return S_FALSE;
  1303. }
  1304. LPCWSTR pText;
  1305. CHECK_HR(pReader->GetValue(&pText, nullptr));
  1306. if (pStripPrefix && *pStripPrefix && _wcsnicmp(pAttrName, pText, wcslen(pStripPrefix)) == 0)
  1307. pText += wcslen(pStripPrefix);
  1308. CHECK_HR(GetEnumValueT(pText, K, pValue));
  1309. CHECK_HR(pReader->MoveToElement());
  1310. return S_OK;
  1311. }
  1312. static HRESULT ReadAttrINPUT_CLASSIFICATION(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_INPUT_CLASSIFICATION *pValue) {
  1313. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::INPUT_CLASSIFICATION, pValue, D3D12_INPUT_CLASSIFICATION_PER_VERTEX_DATA);
  1314. }
  1315. static HRESULT ReadAttrDESCRIPTOR_HEAP_TYPE(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_DESCRIPTOR_HEAP_TYPE *pValue) {
  1316. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::DESCRIPTOR_HEAP_TYPE, pValue, D3D12_DESCRIPTOR_HEAP_TYPE_CBV_SRV_UAV);
  1317. }
  1318. static HRESULT ReadAttrDESCRIPTOR_HEAP_FLAGS(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_DESCRIPTOR_HEAP_FLAGS *pValue) {
  1319. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::DESCRIPTOR_HEAP_FLAG, pValue, D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE);
  1320. }
  1321. static HRESULT ReadAttrDXGI_FORMAT(IXmlReader *pReader, LPCWSTR pAttrName, DXGI_FORMAT *pValue) {
  1322. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::DXGI_FORMAT, pValue, DXGI_FORMAT_UNKNOWN, L"DXGI_FORMAT_");
  1323. }
  1324. static HRESULT ReadAttrHEAP_TYPE(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_HEAP_TYPE *pValue) {
  1325. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::HEAP_TYPE, pValue, D3D12_HEAP_TYPE_DEFAULT);
  1326. }
  1327. static HRESULT ReadAttrCPU_PAGE_PROPERTY(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_CPU_PAGE_PROPERTY *pValue) {
  1328. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::CPU_PAGE_PROPERTY, pValue, D3D12_CPU_PAGE_PROPERTY_UNKNOWN);
  1329. }
  1330. static HRESULT ReadAttrMEMORY_POOL(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_MEMORY_POOL *pValue) {
  1331. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::MEMORY_POOL, pValue, D3D12_MEMORY_POOL_UNKNOWN);
  1332. }
  1333. static HRESULT ReadAttrRESOURCE_DIMENSION(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_RESOURCE_DIMENSION *pValue) {
  1334. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::RESOURCE_DIMENSION, pValue, D3D12_RESOURCE_DIMENSION_BUFFER);
  1335. }
  1336. static HRESULT ReadAttrTEXTURE_LAYOUT(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_TEXTURE_LAYOUT *pValue) {
  1337. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::TEXTURE_LAYOUT, pValue, D3D12_TEXTURE_LAYOUT_UNKNOWN);
  1338. }
  1339. static HRESULT ReadAttrRESOURCE_FLAGS(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_RESOURCE_FLAGS *pValue) {
  1340. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::RESOURCE_FLAG, pValue, D3D12_RESOURCE_FLAG_NONE);
  1341. }
  1342. static HRESULT ReadAttrHEAP_FLAGS(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_HEAP_FLAGS *pValue) {
  1343. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::HEAP_FLAG, pValue, D3D12_HEAP_FLAG_NONE);
  1344. }
  1345. static HRESULT ReadAttrRESOURCE_STATES(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_RESOURCE_STATES *pValue) {
  1346. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::RESOURCE_STATE, pValue, D3D12_RESOURCE_STATE_COMMON);
  1347. }
  1348. static HRESULT ReadAttrSRV_DIMENSION(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_SRV_DIMENSION *pValue) {
  1349. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::SRV_DIMENSION, pValue, D3D12_SRV_DIMENSION_BUFFER);
  1350. }
  1351. static HRESULT ReadAttrUAV_DIMENSION(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_UAV_DIMENSION *pValue) {
  1352. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::UAV_DIMENSION, pValue, D3D12_UAV_DIMENSION_BUFFER);
  1353. }
  1354. static HRESULT ReadAttrPRIMITIVE_TOPOLOGY(IXmlReader *pReader, LPCWSTR pAttrName, D3D_PRIMITIVE_TOPOLOGY *pValue) {
  1355. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::PRIMITIVE_TOPOLOGY, pValue, D3D_PRIMITIVE_TOPOLOGY_TRIANGLELIST);
  1356. }
  1357. static HRESULT ReadAttrPRIMITIVE_TOPOLOGY_TYPE(IXmlReader *pReader, LPCWSTR pAttrName, D3D12_PRIMITIVE_TOPOLOGY_TYPE *pValue) {
  1358. return ReadAttrEnumT(pReader, pAttrName, ParserEnumKind::PRIMITIVE_TOPOLOGY_TYPE, pValue, D3D12_PRIMITIVE_TOPOLOGY_TYPE_TRIANGLE);
  1359. }
  1360. HRESULT ShaderOpParser::ReadAttrStr(IXmlReader *pReader, LPCWSTR pAttrName, LPCSTR *ppValue) {
  1361. if (S_FALSE == CHECK_HR_RET(pReader->MoveToAttributeByName(pAttrName, nullptr))) {
  1362. *ppValue = nullptr;
  1363. return S_FALSE;
  1364. }
  1365. LPCWSTR pValue;
  1366. CHECK_HR(pReader->GetValue(&pValue, nullptr));
  1367. *ppValue = m_pStrings->insert(pValue);
  1368. CHECK_HR(pReader->MoveToElement());
  1369. return S_OK;
  1370. }
  1371. HRESULT ShaderOpParser::ReadAttrBOOL(IXmlReader *pReader, LPCWSTR pAttrName, BOOL *pValue, BOOL defaultValue) {
  1372. if (S_FALSE == CHECK_HR_RET(pReader->MoveToAttributeByName(pAttrName, nullptr))) {
  1373. *pValue = defaultValue;
  1374. return S_FALSE;
  1375. }
  1376. LPCWSTR pText;
  1377. CHECK_HR(pReader->GetValue(&pText, nullptr));
  1378. if (_wcsicmp(pText, L"true") == 0) {
  1379. *pValue = TRUE;
  1380. }
  1381. else {
  1382. *pValue = FALSE;
  1383. }
  1384. CHECK_HR(pReader->MoveToElement());
  1385. return S_OK;
  1386. }
  1387. HRESULT ShaderOpParser::ReadAttrUINT64(IXmlReader *pReader, LPCWSTR pAttrName, UINT64 *pValue, UINT64 defaultValue) {
  1388. if (S_FALSE == CHECK_HR_RET(pReader->MoveToAttributeByName(pAttrName, nullptr))) {
  1389. *pValue = defaultValue;
  1390. return S_FALSE;
  1391. }
  1392. LPCWSTR pText;
  1393. CHECK_HR(pReader->GetValue(&pText, nullptr));
  1394. long long ll = _wtol(pText);
  1395. if (errno == ERANGE) CHECK_HR(E_INVALIDARG);
  1396. *pValue = ll;
  1397. CHECK_HR(pReader->MoveToElement());
  1398. return S_OK;
  1399. }
  1400. HRESULT ShaderOpParser::ReadAttrUINT(IXmlReader *pReader, LPCWSTR pAttrName, UINT *pValue, UINT defaultValue) {
  1401. UINT64 u64;
  1402. HRESULT hrRead = CHECK_HR_RET(ReadAttrUINT64(pReader, pAttrName, &u64, defaultValue));
  1403. CHECK_HR(UInt64ToUInt(u64, pValue));
  1404. return hrRead;
  1405. }
  1406. HRESULT ShaderOpParser::ReadAttrUINT16(IXmlReader *pReader, LPCWSTR pAttrName, UINT16 *pValue, UINT16 defaultValue) {
  1407. UINT64 u64;
  1408. HRESULT hrRead = CHECK_HR_RET(ReadAttrUINT64(pReader, pAttrName, &u64, defaultValue));
  1409. CHECK_HR(UInt64ToUInt16(u64, pValue));
  1410. return hrRead;
  1411. }
  1412. void ShaderOpParser::ReadElementContentStr(IXmlReader *pReader, LPCSTR *ppValue) {
  1413. *ppValue = nullptr;
  1414. if (pReader->IsEmptyElement())
  1415. return;
  1416. UINT startDepth;
  1417. XmlNodeType nt;
  1418. CHECK_HR(pReader->GetDepth(&startDepth));
  1419. std::wstring value;
  1420. for (;;) {
  1421. UINT depth;
  1422. CHECK_HR(pReader->Read(&nt));
  1423. CHECK_HR(pReader->GetDepth(&depth));
  1424. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1425. break;
  1426. if (nt == XmlNodeType_CDATA || nt == XmlNodeType_Text || nt == XmlNodeType_Whitespace) {
  1427. LPCWSTR pText;
  1428. CHECK_HR(pReader->GetValue(&pText, nullptr));
  1429. value += pText;
  1430. }
  1431. }
  1432. *ppValue = m_pStrings->insert(value.c_str());
  1433. }
  1434. void ShaderOpParser::ParseDescriptor(IXmlReader *pReader, ShaderOpDescriptor *pDesc) {
  1435. if (!ReadAtElementName(pReader, L"Descriptor"))
  1436. return;
  1437. CHECK_HR(ReadAttrStr(pReader, L"Name", &pDesc->Name));
  1438. CHECK_HR(ReadAttrStr(pReader, L"ResName", &pDesc->ResName));
  1439. CHECK_HR(ReadAttrStr(pReader, L"CounterName", &pDesc->CounterName));
  1440. CHECK_HR(ReadAttrStr(pReader, L"Kind", &pDesc->Kind));
  1441. bool isSRV = pDesc->Kind && 0 == _stricmp(pDesc->Kind, "SRV");
  1442. pDesc->SrvDescPresent = false;
  1443. DXGI_FORMAT *pFormat;
  1444. if (isSRV) {
  1445. // D3D12_SHADER_RESOURCE_VIEW_DESC
  1446. pFormat = &pDesc->SrvDesc.Format;
  1447. }
  1448. else {
  1449. // D3D12_UNORDERED_ACCESS_VIEW_DESC - default for parsing
  1450. pFormat = &pDesc->UavDesc.Format;
  1451. }
  1452. HRESULT hrFormat = ReadAttrDXGI_FORMAT(pReader, L"Format", pFormat);
  1453. CHECK_HR(hrFormat);
  1454. if (isSRV) {
  1455. pDesc->SrvDesc.Shader4ComponentMapping = D3D12_DEFAULT_SHADER_4_COMPONENT_MAPPING;
  1456. pDesc->SrvDescPresent |= S_OK ==
  1457. CHECK_HR_RET(ReadAttrSRV_DIMENSION(pReader, L"Dimension", &pDesc->SrvDesc.ViewDimension));
  1458. switch (pDesc->SrvDesc.ViewDimension) {
  1459. case D3D12_SRV_DIMENSION_BUFFER:
  1460. pDesc->SrvDescPresent |= S_OK ==
  1461. CHECK_HR_RET(ReadAttrUINT64(pReader, L"FirstElement", &pDesc->SrvDesc.Buffer.FirstElement));
  1462. LPCSTR pFlags;
  1463. pDesc->SrvDescPresent |= S_OK ==
  1464. CHECK_HR_RET(ReadAttrStr(pReader, L"Flags", &pFlags));
  1465. if (pFlags && *pFlags && 0 == _stricmp(pFlags, "RAW")) {
  1466. pDesc->SrvDesc.Buffer.Flags = D3D12_BUFFER_SRV_FLAG_RAW;
  1467. }
  1468. else {
  1469. pDesc->SrvDesc.Buffer.Flags = D3D12_BUFFER_SRV_FLAG_NONE;
  1470. }
  1471. pDesc->SrvDescPresent |= S_OK ==
  1472. CHECK_HR_RET(ReadAttrUINT(pReader, L"NumElements", &pDesc->SrvDesc.Buffer.NumElements));
  1473. pDesc->SrvDescPresent |= S_OK ==
  1474. CHECK_HR_RET(ReadAttrUINT(pReader, L"StructureByteStride", &pDesc->SrvDesc.Buffer.StructureByteStride));
  1475. break;
  1476. default:
  1477. CHECK_HR(E_NOTIMPL);
  1478. }
  1479. }
  1480. else {
  1481. CHECK_HR(ReadAttrUAV_DIMENSION(pReader, L"Dimension", &pDesc->UavDesc.ViewDimension));
  1482. switch (pDesc->UavDesc.ViewDimension) {
  1483. case D3D12_UAV_DIMENSION_BUFFER:
  1484. CHECK_HR(ReadAttrUINT64(pReader, L"FirstElement", &pDesc->UavDesc.Buffer.FirstElement));
  1485. CHECK_HR(ReadAttrUINT(pReader, L"NumElements", &pDesc->UavDesc.Buffer.NumElements));
  1486. CHECK_HR(ReadAttrUINT(pReader, L"StructureByteStride", &pDesc->UavDesc.Buffer.StructureByteStride));
  1487. CHECK_HR(ReadAttrUINT64(pReader, L"CounterOffsetInBytes", &pDesc->UavDesc.Buffer.CounterOffsetInBytes));
  1488. LPCSTR pFlags;
  1489. CHECK_HR(ReadAttrStr(pReader, L"Flags", &pFlags));
  1490. if (pFlags && *pFlags && 0 == _stricmp(pFlags, "RAW")) {
  1491. pDesc->UavDesc.Buffer.Flags = D3D12_BUFFER_UAV_FLAG_RAW;
  1492. }
  1493. else {
  1494. pDesc->UavDesc.Buffer.Flags = D3D12_BUFFER_UAV_FLAG_NONE;
  1495. }
  1496. if (hrFormat == S_FALSE && pDesc->UavDesc.Buffer.Flags & D3D12_BUFFER_UAV_FLAG_RAW) {
  1497. pDesc->UavDesc.Format = DXGI_FORMAT_R32_TYPELESS;
  1498. }
  1499. break;
  1500. case D3D12_UAV_DIMENSION_TEXTURE1D:
  1501. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture1D.MipSlice));
  1502. break;
  1503. case D3D12_UAV_DIMENSION_TEXTURE1DARRAY:
  1504. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture1DArray.MipSlice));
  1505. CHECK_HR(ReadAttrUINT(pReader, L"FirstArraySlice", &pDesc->UavDesc.Texture1DArray.FirstArraySlice));
  1506. CHECK_HR(ReadAttrUINT(pReader, L"ArraySize", &pDesc->UavDesc.Texture1DArray.ArraySize));
  1507. break;
  1508. case D3D12_UAV_DIMENSION_TEXTURE2D:
  1509. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture2D.MipSlice));
  1510. CHECK_HR(ReadAttrUINT(pReader, L"PlaneSlice", &pDesc->UavDesc.Texture2D.PlaneSlice));
  1511. break;
  1512. case D3D12_UAV_DIMENSION_TEXTURE2DARRAY:
  1513. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture2DArray.MipSlice));
  1514. CHECK_HR(ReadAttrUINT(pReader, L"FirstArraySlice", &pDesc->UavDesc.Texture2DArray.FirstArraySlice));
  1515. CHECK_HR(ReadAttrUINT(pReader, L"ArraySize", &pDesc->UavDesc.Texture2DArray.ArraySize));
  1516. CHECK_HR(ReadAttrUINT(pReader, L"PlaneSlice", &pDesc->UavDesc.Texture2DArray.PlaneSlice));
  1517. break;
  1518. case D3D12_UAV_DIMENSION_TEXTURE3D:
  1519. CHECK_HR(ReadAttrUINT(pReader, L"MipSlice", &pDesc->UavDesc.Texture3D.MipSlice));
  1520. CHECK_HR(ReadAttrUINT(pReader, L"FirstWSlice", &pDesc->UavDesc.Texture3D.FirstWSlice));
  1521. CHECK_HR(ReadAttrUINT(pReader, L"WSize", &pDesc->UavDesc.Texture3D.WSize));
  1522. break;
  1523. }
  1524. }
  1525. // If either is missing, set one from the other.
  1526. if (pDesc->Name && !pDesc->ResName) pDesc->ResName = pDesc->Name;
  1527. if (pDesc->ResName && !pDesc->Name) pDesc->Name = pDesc->ResName;
  1528. LPCSTR K = pDesc->Kind;
  1529. if (K == nullptr) {
  1530. ShaderOpLogFmt(L"Descriptor '%S' is missing Kind attribute.", pDesc->Name);
  1531. CHECK_HR(E_INVALIDARG);
  1532. } else if (0 != _stricmp(K, "UAV") && 0 != _stricmp(K, "SRV") &&
  1533. 0 != _stricmp(K, "CBV") && 0 != _stricmp(K, "RTV")) {
  1534. ShaderOpLogFmt(L"Descriptor '%S' references unknown kind '%S'",
  1535. pDesc->Name, K);
  1536. CHECK_HR(E_INVALIDARG);
  1537. }
  1538. }
  1539. void ShaderOpParser::ParseDescriptorHeap(IXmlReader *pReader, ShaderOpDescriptorHeap *pHeap) {
  1540. if (!ReadAtElementName(pReader, L"DescriptorHeap"))
  1541. return;
  1542. CHECK_HR(ReadAttrStr(pReader, L"Name", &pHeap->Name));
  1543. HRESULT hrFlags = ReadAttrDESCRIPTOR_HEAP_FLAGS(pReader, L"Flags", &pHeap->Desc.Flags);
  1544. CHECK_HR(hrFlags);
  1545. CHECK_HR(ReadAttrUINT(pReader, L"NodeMask", &pHeap->Desc.NodeMask));
  1546. CHECK_HR(ReadAttrUINT(pReader, L"NumDescriptors", &pHeap->Desc.NumDescriptors));
  1547. CHECK_HR(ReadAttrDESCRIPTOR_HEAP_TYPE(pReader, L"Type", &pHeap->Desc.Type));
  1548. if (pHeap->Desc.Type == D3D12_DESCRIPTOR_HEAP_TYPE_RTV && hrFlags == S_FALSE)
  1549. pHeap->Desc.Flags = D3D12_DESCRIPTOR_HEAP_FLAG_NONE;
  1550. if (pReader->IsEmptyElement())
  1551. return;
  1552. UINT startDepth;
  1553. XmlNodeType nt;
  1554. CHECK_HR(pReader->GetDepth(&startDepth));
  1555. std::wstring value;
  1556. for (;;) {
  1557. UINT depth;
  1558. CHECK_HR(pReader->Read(&nt));
  1559. CHECK_HR(pReader->GetDepth(&depth));
  1560. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1561. break;
  1562. if (nt == XmlNodeType_Element) {
  1563. LPCWSTR pLocalName;
  1564. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1565. if (0 == wcscmp(pLocalName, L"Descriptor")) {
  1566. ShaderOpDescriptor D;
  1567. ParseDescriptor(pReader, &D);
  1568. pHeap->Descriptors.push_back(D);
  1569. }
  1570. }
  1571. }
  1572. }
  1573. void ShaderOpParser::ParseInputElement(IXmlReader *pReader, D3D12_INPUT_ELEMENT_DESC *pInputElement) {
  1574. if (!ReadAtElementName(pReader, L"InputElement"))
  1575. return;
  1576. CHECK_HR(ReadAttrStr(pReader, L"SemanticName", &pInputElement->SemanticName));
  1577. CHECK_HR(ReadAttrUINT(pReader, L"SemanticIndex", &pInputElement->SemanticIndex));
  1578. CHECK_HR(ReadAttrDXGI_FORMAT(pReader, L"Format", &pInputElement->Format));
  1579. CHECK_HR(ReadAttrUINT(pReader, L"InputSlot", &pInputElement->InputSlot));
  1580. CHECK_HR(ReadAttrUINT(pReader, L"AlignedByteOffset", &pInputElement->AlignedByteOffset, D3D12_APPEND_ALIGNED_ELEMENT));
  1581. CHECK_HR(ReadAttrINPUT_CLASSIFICATION(pReader, L"InputSlotClass", &pInputElement->InputSlotClass));
  1582. CHECK_HR(ReadAttrUINT(pReader, L"InstanceDataStepRate", &pInputElement->InstanceDataStepRate));
  1583. }
  1584. void ShaderOpParser::ParseInputElements(IXmlReader *pReader, std::vector<D3D12_INPUT_ELEMENT_DESC> *pInputElements) {
  1585. if (!ReadAtElementName(pReader, L"InputElements"))
  1586. return;
  1587. if (pReader->IsEmptyElement()) return;
  1588. UINT startDepth;
  1589. XmlNodeType nt;
  1590. CHECK_HR(pReader->GetDepth(&startDepth));
  1591. for (;;) {
  1592. UINT depth;
  1593. CHECK_HR(pReader->Read(&nt));
  1594. CHECK_HR(pReader->GetDepth(&depth));
  1595. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1596. return;
  1597. if (nt == XmlNodeType_Element) {
  1598. LPCWSTR pLocalName;
  1599. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1600. if (0 == wcscmp(pLocalName, L"InputElement")) {
  1601. D3D12_INPUT_ELEMENT_DESC desc;
  1602. ParseInputElement(pReader, &desc);
  1603. pInputElements->push_back(desc);
  1604. }
  1605. }
  1606. }
  1607. }
  1608. void ShaderOpParser::ParseRenderTargets(IXmlReader *pReader, std::vector<LPCSTR> *pRenderTargets) {
  1609. if (!ReadAtElementName(pReader, L"RenderTargets"))
  1610. return;
  1611. if (pReader->IsEmptyElement()) return;
  1612. UINT startDepth;
  1613. XmlNodeType nt;
  1614. CHECK_HR(pReader->GetDepth(&startDepth));
  1615. for (;;) {
  1616. UINT depth;
  1617. CHECK_HR(pReader->Read(&nt));
  1618. CHECK_HR(pReader->GetDepth(&depth));
  1619. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1620. return;
  1621. if (nt == XmlNodeType_Element) {
  1622. LPCWSTR pLocalName;
  1623. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1624. if (0 == wcscmp(pLocalName, L"RenderTarget")) {
  1625. LPCSTR pName;
  1626. CHECK_HR(ReadAttrStr(pReader, L"Name", &pName));
  1627. pRenderTargets->push_back(pName);
  1628. }
  1629. }
  1630. }
  1631. }
  1632. void ShaderOpParser::ParseRootValue(IXmlReader *pReader, ShaderOpRootValue *pRootValue) {
  1633. if (!ReadAtElementName(pReader, L"RootValue"))
  1634. return;
  1635. CHECK_HR(ReadAttrStr(pReader, L"ResName", &pRootValue->ResName));
  1636. CHECK_HR(ReadAttrStr(pReader, L"HeapName", &pRootValue->HeapName));
  1637. CHECK_HR(ReadAttrUINT(pReader, L"Index", &pRootValue->Index));
  1638. }
  1639. void ShaderOpParser::ParseRootValues(IXmlReader *pReader, std::vector<ShaderOpRootValue> *pRootValues) {
  1640. if (!ReadAtElementName(pReader, L"RootValues"))
  1641. return;
  1642. if (pReader->IsEmptyElement()) return;
  1643. UINT startDepth;
  1644. XmlNodeType nt;
  1645. CHECK_HR(pReader->GetDepth(&startDepth));
  1646. for (;;) {
  1647. UINT depth;
  1648. CHECK_HR(pReader->Read(&nt));
  1649. CHECK_HR(pReader->GetDepth(&depth));
  1650. if (nt == XmlNodeType_EndElement && depth == startDepth + 1)
  1651. return;
  1652. if (nt == XmlNodeType_Element) {
  1653. LPCWSTR pLocalName;
  1654. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1655. if (0 == wcscmp(pLocalName, L"RootValue")) {
  1656. ShaderOpRootValue V;
  1657. ParseRootValue(pReader, &V);
  1658. pRootValues->push_back(V);
  1659. }
  1660. }
  1661. }
  1662. }
  1663. void ShaderOpParser::ParseShaderOpSet(IStream *pStream, ShaderOpSet *pShaderOpSet) {
  1664. CComPtr<IXmlReader> pReader;
  1665. CHECK_HR(CreateXmlReader(__uuidof(IXmlReader), (void **)&pReader, nullptr));
  1666. CHECK_HR(pReader->SetInput(pStream));
  1667. ParseShaderOpSet(pReader, pShaderOpSet);
  1668. }
  1669. void ShaderOpParser::ParseShaderOpSet(IXmlReader *pReader, ShaderOpSet *pShaderOpSet) {
  1670. if (!ReadAtElementName(pReader, L"ShaderOpSet"))
  1671. return;
  1672. UINT startDepth;
  1673. CHECK_HR(pReader->GetDepth(&startDepth));
  1674. XmlNodeType nt = XmlNodeType_Element;
  1675. for (;;) {
  1676. if (nt == XmlNodeType_Element) {
  1677. LPCWSTR pLocalName;
  1678. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1679. if (0 == wcscmp(pLocalName, L"ShaderOp")) {
  1680. pShaderOpSet->ShaderOps.emplace_back(std::make_unique<ShaderOp>());
  1681. ParseShaderOp(pReader, pShaderOpSet->ShaderOps.back().get());
  1682. }
  1683. }
  1684. else if (nt == XmlNodeType_EndElement) {
  1685. UINT depth;
  1686. CHECK_HR(pReader->GetDepth(&depth));
  1687. if (depth == startDepth + 1)
  1688. return;
  1689. }
  1690. CHECK_HR(pReader->Read(&nt));
  1691. }
  1692. }
  1693. void ShaderOpParser::ParseShaderOp(IXmlReader *pReader, ShaderOp *pShaderOp) {
  1694. m_pStrings = &pShaderOp->Strings;
  1695. // Look for a ShaderOp element.
  1696. if (!ReadAtElementName(pReader, L"ShaderOp"))
  1697. return;
  1698. CHECK_HR(ReadAttrStr(pReader, L"Name", &pShaderOp->Name));
  1699. CHECK_HR(ReadAttrStr(pReader, L"CS", &pShaderOp->CS));
  1700. CHECK_HR(ReadAttrStr(pReader, L"VS", &pShaderOp->VS));
  1701. CHECK_HR(ReadAttrStr(pReader, L"HS", &pShaderOp->HS));
  1702. CHECK_HR(ReadAttrStr(pReader, L"DS", &pShaderOp->DS));
  1703. CHECK_HR(ReadAttrStr(pReader, L"GS", &pShaderOp->GS));
  1704. CHECK_HR(ReadAttrStr(pReader, L"PS", &pShaderOp->PS));
  1705. CHECK_HR(ReadAttrUINT(pReader, L"DispatchX", &pShaderOp->DispatchX, 1));
  1706. CHECK_HR(ReadAttrUINT(pReader, L"DispatchY", &pShaderOp->DispatchY, 1));
  1707. CHECK_HR(ReadAttrUINT(pReader, L"DispatchZ", &pShaderOp->DispatchZ, 1));
  1708. CHECK_HR(ReadAttrPRIMITIVE_TOPOLOGY_TYPE(pReader, L"TopologyType", &pShaderOp->PrimitiveTopologyType));
  1709. UINT startDepth;
  1710. CHECK_HR(pReader->GetDepth(&startDepth));
  1711. XmlNodeType nt = XmlNodeType_Element;
  1712. for (;;) {
  1713. if (nt == XmlNodeType_Element) {
  1714. LPCWSTR pLocalName;
  1715. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1716. if (0 == wcscmp(pLocalName, L"InputElements")) {
  1717. ParseInputElements(pReader, &pShaderOp->InputElements);
  1718. }
  1719. else if (0 == wcscmp(pLocalName, L"Shader")) {
  1720. ShaderOpShader shader;
  1721. ParseShader(pReader, &shader);
  1722. pShaderOp->Shaders.push_back(shader);
  1723. }
  1724. else if (0 == wcscmp(pLocalName, L"RootSignature")) {
  1725. ReadElementContentStr(pReader, &pShaderOp->RootSignature);
  1726. }
  1727. else if (0 == wcscmp(pLocalName, L"RenderTargets")) {
  1728. ParseRenderTargets(pReader, &pShaderOp->RenderTargets);
  1729. }
  1730. else if (0 == wcscmp(pLocalName, L"Resource")) {
  1731. ShaderOpResource resource;
  1732. ParseResource(pReader, &resource);
  1733. pShaderOp->Resources.push_back(resource);
  1734. }
  1735. else if (0 == wcscmp(pLocalName, L"DescriptorHeap")) {
  1736. ShaderOpDescriptorHeap heap;
  1737. ParseDescriptorHeap(pReader, &heap);
  1738. pShaderOp->DescriptorHeaps.push_back(heap);
  1739. }
  1740. else if (0 == wcscmp(pLocalName, L"RootValues")) {
  1741. ParseRootValues(pReader, &pShaderOp->RootValues);
  1742. }
  1743. }
  1744. else if (nt == XmlNodeType_EndElement) {
  1745. UINT depth;
  1746. CHECK_HR(pReader->GetDepth(&depth));
  1747. if (depth == startDepth + 1)
  1748. return;
  1749. }
  1750. if (S_FALSE == CHECK_HR_RET(pReader->Read(&nt)))
  1751. return;
  1752. }
  1753. }
  1754. LPCWSTR SkipByteInitSeparators(LPCWSTR pText) {
  1755. while (*pText && (*pText == L' ' || *pText == L'\t' ||
  1756. *pText == L'\r' || *pText == L'\n' || *pText == L'{' ||
  1757. *pText == L'}' || *pText == L','))
  1758. ++pText;
  1759. return pText;
  1760. }
  1761. LPCWSTR FindByteInitSeparators(LPCWSTR pText) {
  1762. while (*pText &&
  1763. !(*pText == L' ' || *pText == L'\t' ||
  1764. *pText == L'\r' || *pText == L'\n' || *pText == L'{' ||
  1765. *pText == L'}' || *pText == L','))
  1766. ++pText;
  1767. return pText;
  1768. }
  1769. using namespace hlsl;
  1770. DXIL::ComponentType GetCompType(LPCWSTR pText, LPCWSTR pEnd) {
  1771. // if no prefix shown, use it as a default
  1772. if (pText == pEnd) return DXIL::ComponentType::F32;
  1773. // check if suffix starts with (half)
  1774. if (wcsncmp(pText, L"(half)", 6) == 0) {
  1775. return DXIL::ComponentType::F16;
  1776. }
  1777. switch (*(pEnd - 1)) {
  1778. case L'h':
  1779. case L'H':
  1780. return DXIL::ComponentType::F16;
  1781. case L'l':
  1782. case L'L':
  1783. return DXIL::ComponentType::F64;
  1784. case L'u':
  1785. case L'U':
  1786. return DXIL::ComponentType::U32;
  1787. case L'i':
  1788. case L'I':
  1789. return DXIL::ComponentType::I32;
  1790. case L'f':
  1791. case L'F':
  1792. default:
  1793. return DXIL::ComponentType::F32;
  1794. }
  1795. }
  1796. void ParseDataFromText(LPCWSTR pText, LPCWSTR pEnd, DXIL::ComponentType compType, std::vector<BYTE> &V) {
  1797. BYTE *pB;
  1798. if (compType == DXIL::ComponentType::F16 || compType == DXIL::ComponentType::F32) {
  1799. float fVal;
  1800. size_t wordSize = pEnd - pText;
  1801. if (wordSize >= 3 && 0 == _wcsnicmp(pEnd - 3, L"nan", 3)) {
  1802. fVal = NAN;
  1803. }
  1804. else if (wordSize >= 4 && 0 == _wcsnicmp(pEnd - 4, L"-inf", 4)) {
  1805. fVal = -(INFINITY);
  1806. }
  1807. else if ((wordSize >= 3 && 0 == _wcsnicmp(pEnd - 3, L"inf", 3)) ||
  1808. (wordSize >= 4 && 0 == _wcsnicmp(pEnd - 4, L"+inf", 4))) {
  1809. fVal = INFINITY;
  1810. }
  1811. else if (wordSize >= 7 && 0 == _wcsnicmp(pEnd - 7, L"-denorm", 7)) {
  1812. fVal = -(FLT_MIN / 2);
  1813. }
  1814. else if (wordSize >= 6 && 0 == _wcsnicmp(pEnd - 6, L"denorm", 6)) {
  1815. fVal = (FLT_MIN / 2);
  1816. }
  1817. else {
  1818. fVal = wcstof(pText, nullptr);
  1819. }
  1820. if (compType == DXIL::ComponentType::F16) {
  1821. uint16_t fp16Val = ConvertFloat32ToFloat16(fVal);
  1822. pB = (BYTE *)&fp16Val;
  1823. V.insert(V.end(), pB, pB + sizeof(uint16_t));
  1824. }
  1825. else {
  1826. pB = (BYTE *)&fVal;
  1827. V.insert(V.end(), pB, pB + sizeof(float));
  1828. }
  1829. }
  1830. else if (compType == DXIL::ComponentType::I32) {
  1831. int val = _wtoi(pText);
  1832. pB = (BYTE *)&val;
  1833. V.insert(V.end(), pB, pB + sizeof(int));
  1834. }
  1835. else {
  1836. DXASSERT_ARGS(false, "Unsupported stream component type : %u", compType);
  1837. }
  1838. }
  1839. void ShaderOpParser::ParseResource(IXmlReader *pReader, ShaderOpResource *pResource) {
  1840. if (!ReadAtElementName(pReader, L"Resource"))
  1841. return;
  1842. CHECK_HR(ReadAttrStr(pReader, L"Name", &pResource->Name));
  1843. CHECK_HR(ReadAttrStr(pReader, L"Init", &pResource->Init));
  1844. CHECK_HR(ReadAttrBOOL(pReader, L"ReadBack", &pResource->ReadBack));
  1845. CHECK_HR(ReadAttrHEAP_TYPE(pReader, L"HeapType", &pResource->HeapProperties.Type));
  1846. CHECK_HR(ReadAttrCPU_PAGE_PROPERTY(pReader, L"CPUPageProperty", &pResource->HeapProperties.CPUPageProperty));
  1847. CHECK_HR(ReadAttrMEMORY_POOL(pReader, L"MemoryPoolPreference", &pResource->HeapProperties.MemoryPoolPreference));
  1848. CHECK_HR(ReadAttrUINT(pReader, L"CreationNodeMask", &pResource->HeapProperties.CreationNodeMask));
  1849. CHECK_HR(ReadAttrUINT(pReader, L"VisibleNodeMask", &pResource->HeapProperties.VisibleNodeMask));
  1850. // D3D12_RESOURCE_DESC Desc;
  1851. CHECK_HR(ReadAttrRESOURCE_DIMENSION(pReader, L"Dimension", &pResource->Desc.Dimension));
  1852. CHECK_HR(ReadAttrUINT64(pReader, L"Alignment", &pResource->Desc.Alignment));
  1853. CHECK_HR(ReadAttrUINT64(pReader, L"Width", &pResource->Desc.Width));
  1854. CHECK_HR(ReadAttrUINT(pReader, L"Height", &pResource->Desc.Height));
  1855. CHECK_HR(ReadAttrUINT16(pReader, L"DepthOrArraySize", &pResource->Desc.DepthOrArraySize));
  1856. CHECK_HR(ReadAttrUINT16(pReader, L"MipLevels", &pResource->Desc.MipLevels));
  1857. CHECK_HR(ReadAttrDXGI_FORMAT(pReader, L"Format", &pResource->Desc.Format));
  1858. CHECK_HR(ReadAttrUINT(pReader, L"SampleCount", &pResource->Desc.SampleDesc.Count));
  1859. CHECK_HR(ReadAttrUINT(pReader, L"SampleQual", &pResource->Desc.SampleDesc.Quality));
  1860. CHECK_HR(ReadAttrTEXTURE_LAYOUT(pReader, L"Layout", &pResource->Desc.Layout));
  1861. CHECK_HR(ReadAttrRESOURCE_FLAGS(pReader, L"Flags", &pResource->Desc.Flags));
  1862. CHECK_HR(ReadAttrHEAP_FLAGS(pReader, L"HeapFlags", &pResource->HeapFlags));
  1863. CHECK_HR(ReadAttrRESOURCE_STATES(pReader, L"InitialResourceState", &pResource->InitialResourceState));
  1864. CHECK_HR(ReadAttrRESOURCE_STATES(pReader, L"TransitionTo", &pResource->TransitionTo));
  1865. CHECK_HR(ReadAttrPRIMITIVE_TOPOLOGY(pReader, L"Topology", &pResource->PrimitiveTopology));
  1866. // Set some fixed values.
  1867. if (pResource->Desc.Dimension == D3D12_RESOURCE_DIMENSION_BUFFER) {
  1868. pResource->Desc.Height = 1;
  1869. pResource->Desc.DepthOrArraySize = 1;
  1870. pResource->Desc.MipLevels = 1;
  1871. pResource->Desc.Format = DXGI_FORMAT_UNKNOWN;
  1872. pResource->Desc.SampleDesc.Count = 1;
  1873. pResource->Desc.SampleDesc.Quality = 0;
  1874. pResource->Desc.Layout = D3D12_TEXTURE_LAYOUT_ROW_MAJOR;
  1875. }
  1876. if (pResource->Desc.Dimension == D3D12_RESOURCE_DIMENSION_TEXTURE1D) {
  1877. if (pResource->Desc.Height == 0) pResource->Desc.Height = 1;
  1878. if (pResource->Desc.DepthOrArraySize == 0) pResource->Desc.DepthOrArraySize = 1;
  1879. if (pResource->Desc.SampleDesc.Count == 0) pResource->Desc.SampleDesc.Count = 1;
  1880. }
  1881. if (pResource->Desc.Dimension == D3D12_RESOURCE_DIMENSION_TEXTURE2D) {
  1882. if (pResource->Desc.DepthOrArraySize == 0) pResource->Desc.DepthOrArraySize = 1;
  1883. if (pResource->Desc.SampleDesc.Count == 0 ) pResource->Desc.SampleDesc.Count = 1;
  1884. }
  1885. // If the resource has text, that goes into the bytes initialization area.
  1886. if (pReader->IsEmptyElement())
  1887. return;
  1888. std::vector<BYTE> &V = pResource->InitBytes;
  1889. XmlNodeType nt;
  1890. CHECK_HR(pReader->GetNodeType(&nt));
  1891. for (;;) {
  1892. if (nt == XmlNodeType_EndElement) {
  1893. return;
  1894. }
  1895. if (nt == XmlNodeType_Text) {
  1896. // Handle the byte payload. '{', '}', ',', whitespace - these are all
  1897. // separators and are ignored in terms of structure. We simply read
  1898. // literals, figure out their type based on suffix, and write the bytes
  1899. // into the target array.
  1900. LPCWSTR pText;
  1901. pReader->GetValue(&pText, nullptr);
  1902. while (*pText) {
  1903. pText = SkipByteInitSeparators(pText);
  1904. if (!*pText) continue;
  1905. LPCWSTR pEnd = FindByteInitSeparators(pText);
  1906. // Consider looking for prefixes/suffixes to handle bases and types.
  1907. DXIL::ComponentType compType = GetCompType(pText, pEnd);
  1908. ParseDataFromText(pText, pEnd, compType, V);
  1909. pText = pEnd;
  1910. }
  1911. }
  1912. if (S_FALSE == CHECK_HR_RET(pReader->Read(&nt)))
  1913. return;
  1914. }
  1915. }
  1916. void ShaderOpParser::ParseShader(IXmlReader *pReader, ShaderOpShader *pShader) {
  1917. if (!ReadAtElementName(pReader, L"Shader"))
  1918. return;
  1919. CHECK_HR(ReadAttrStr(pReader, L"Name", &pShader->Name));
  1920. CHECK_HR(ReadAttrStr(pReader, L"EntryPoint", &pShader->EntryPoint));
  1921. CHECK_HR(ReadAttrStr(pReader, L"Target", &pShader->Target));
  1922. CHECK_HR(ReadAttrStr(pReader, L"Arguments", &pShader->Arguments));
  1923. CHECK_HR(ReadAttrBOOL(pReader, L"Compiled", &pShader->Compiled))
  1924. ReadElementContentStr(pReader, &pShader->Text);
  1925. bool hasText = pShader->Text && *pShader->Text;
  1926. if (hasText) {
  1927. LPCSTR pCheck;
  1928. CHECK_HR(ReadAttrStr(pReader, L"Text", &pCheck));
  1929. if (pCheck && *pCheck) {
  1930. ShaderOpLogFmt(L"Shader %S has text content and a Text attribute; it "
  1931. L"should only have one",
  1932. pShader->Name);
  1933. CHECK_HR(E_INVALIDARG);
  1934. }
  1935. }
  1936. else {
  1937. CHECK_HR(ReadAttrStr(pReader, L"Text", &pShader->Text));
  1938. }
  1939. if (pShader->EntryPoint == nullptr)
  1940. pShader->EntryPoint = m_pStrings->insert("main");
  1941. }
  1942. bool ShaderOpParser::ReadAtElementName(IXmlReader *pReader, LPCWSTR pName) {
  1943. XmlNodeType nt;
  1944. CHECK_HR(pReader->GetNodeType(&nt));
  1945. for (;;) {
  1946. if (nt == XmlNodeType_Element) {
  1947. LPCWSTR pLocalName;
  1948. CHECK_HR(pReader->GetLocalName(&pLocalName, nullptr));
  1949. if (0 == wcscmp(pLocalName, pName)) {
  1950. return true;
  1951. }
  1952. }
  1953. if (S_FALSE == CHECK_HR_RET(pReader->Read(&nt)))
  1954. return false;
  1955. }
  1956. }
  1957. #pragma endregion Parsing support
  1958. } // namespace st