BsMaterial.cpp 47 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466
  1. #include "BsMaterial.h"
  2. #include "BsException.h"
  3. #include "BsShader.h"
  4. #include "BsTechnique.h"
  5. #include "BsPass.h"
  6. #include "BsRenderAPI.h"
  7. #include "BsHardwareBufferManager.h"
  8. #include "BsGpuProgram.h"
  9. #include "BsGpuParamBlockBuffer.h"
  10. #include "BsGpuParamDesc.h"
  11. #include "BsMaterialRTTI.h"
  12. #include "BsMaterialManager.h"
  13. #include "BsDebug.h"
  14. #include "BsResources.h"
  15. #include "BsFrameAlloc.h"
  16. #include "BsMatrixNxM.h"
  17. #include "BsVectorNI.h"
  18. #include "BsMemorySerializer.h"
  19. #include "BsMaterialParams.h"
  20. namespace BansheeEngine
  21. {
  22. struct ShaderBlockDesc
  23. {
  24. String name;
  25. GpuParamBlockUsage usage;
  26. int size;
  27. bool create;
  28. };
  29. enum MaterialLoadFlags
  30. {
  31. Load_None = 0,
  32. Load_Shader = 1,
  33. Load_All = 2
  34. };
  35. const UINT32 PassParameters::NUM_PARAMS = 6;
  36. const UINT32 PassParametersCore::NUM_PARAMS = 6;
  37. SPtr<PassParametersCore> convertParamsToCore(const SPtr<PassParameters>& passParams)
  38. {
  39. SPtr<PassParametersCore> passParameters = bs_shared_ptr_new<PassParametersCore>();
  40. if (passParams->mVertParams != nullptr)
  41. passParameters->mVertParams = passParams->mVertParams->getCore();
  42. else
  43. passParameters->mVertParams = nullptr;
  44. if (passParams->mFragParams != nullptr)
  45. passParameters->mFragParams = passParams->mFragParams->getCore();
  46. else
  47. passParameters->mFragParams = nullptr;
  48. if (passParams->mGeomParams != nullptr)
  49. passParameters->mGeomParams = passParams->mGeomParams->getCore();
  50. else
  51. passParameters->mGeomParams = nullptr;
  52. if (passParams->mHullParams != nullptr)
  53. passParameters->mHullParams = passParams->mHullParams->getCore();
  54. else
  55. passParameters->mHullParams = nullptr;
  56. if (passParams->mDomainParams != nullptr)
  57. passParameters->mDomainParams = passParams->mDomainParams->getCore();
  58. else
  59. passParameters->mDomainParams = nullptr;
  60. if (passParams->mComputeParams != nullptr)
  61. passParameters->mComputeParams = passParams->mComputeParams->getCore();
  62. else
  63. passParameters->mComputeParams = nullptr;
  64. return passParameters;
  65. }
  66. bool areParamsEqual(const GpuParamDataDesc& paramA, const GpuParamDataDesc& paramB, bool ignoreBufferOffsets)
  67. {
  68. bool equal = paramA.arraySize == paramB.arraySize && paramA.elementSize == paramB.elementSize
  69. && paramA.type == paramB.type && paramA.arrayElementStride == paramB.arrayElementStride;
  70. if (!ignoreBufferOffsets)
  71. equal &= paramA.cpuMemOffset == paramB.cpuMemOffset && paramA.gpuMemOffset == paramB.gpuMemOffset;
  72. return equal;
  73. }
  74. Map<String, const GpuParamDataDesc*> determineValidDataParameters(const Vector<GpuParamDescPtr>& paramDescs)
  75. {
  76. Map<String, const GpuParamDataDesc*> foundDataParams;
  77. Map<String, bool> validParams;
  78. for (auto iter = paramDescs.begin(); iter != paramDescs.end(); ++iter)
  79. {
  80. const GpuParamDesc& curDesc = **iter;
  81. // Check regular data params
  82. for (auto iter2 = curDesc.params.begin(); iter2 != curDesc.params.end(); ++iter2)
  83. {
  84. bool isParameterValid = true;
  85. const GpuParamDataDesc& curParam = iter2->second;
  86. auto dataFindIter = validParams.find(iter2->first);
  87. if (dataFindIter == validParams.end())
  88. {
  89. validParams[iter2->first] = true;
  90. foundDataParams[iter2->first] = &curParam;
  91. }
  92. else
  93. {
  94. if (validParams[iter2->first])
  95. {
  96. auto dataFindIter2 = foundDataParams.find(iter2->first);
  97. const GpuParamDataDesc* otherParam = dataFindIter2->second;
  98. if (!areParamsEqual(curParam, *otherParam, true))
  99. {
  100. validParams[iter2->first] = false;
  101. foundDataParams.erase(dataFindIter2);
  102. }
  103. }
  104. }
  105. }
  106. }
  107. return foundDataParams;
  108. }
  109. Vector<const GpuParamObjectDesc*> determineValidObjectParameters(const Vector<GpuParamDescPtr>& paramDescs)
  110. {
  111. Vector<const GpuParamObjectDesc*> validParams;
  112. for (auto iter = paramDescs.begin(); iter != paramDescs.end(); ++iter)
  113. {
  114. const GpuParamDesc& curDesc = **iter;
  115. // Check sampler params
  116. for (auto iter2 = curDesc.samplers.begin(); iter2 != curDesc.samplers.end(); ++iter2)
  117. {
  118. validParams.push_back(&iter2->second);
  119. }
  120. // Check texture params
  121. for (auto iter2 = curDesc.textures.begin(); iter2 != curDesc.textures.end(); ++iter2)
  122. {
  123. validParams.push_back(&iter2->second);
  124. }
  125. // Check buffer params
  126. for (auto iter2 = curDesc.buffers.begin(); iter2 != curDesc.buffers.end(); ++iter2)
  127. {
  128. validParams.push_back(&iter2->second);
  129. }
  130. }
  131. return validParams;
  132. }
  133. Set<String> determineValidShareableParamBlocks(const Vector<GpuParamDescPtr>& paramDescs)
  134. {
  135. // Make sure param blocks with the same name actually are the same
  136. Map<String, std::pair<String, GpuParamDescPtr>> uniqueParamBlocks;
  137. Map<String, bool> validParamBlocks;
  138. for (auto iter = paramDescs.begin(); iter != paramDescs.end(); ++iter)
  139. {
  140. const GpuParamDesc& curDesc = **iter;
  141. for (auto blockIter = curDesc.paramBlocks.begin(); blockIter != curDesc.paramBlocks.end(); ++blockIter)
  142. {
  143. bool isBlockValid = true;
  144. const GpuParamBlockDesc& curBlock = blockIter->second;
  145. if (!curBlock.isShareable) // Non-shareable buffers are handled differently, they're allowed same names
  146. continue;
  147. auto iterFind = uniqueParamBlocks.find(blockIter->first);
  148. if (iterFind == uniqueParamBlocks.end())
  149. {
  150. uniqueParamBlocks[blockIter->first] = std::make_pair(blockIter->first, *iter);
  151. validParamBlocks[blockIter->first] = true;
  152. continue;
  153. }
  154. String otherBlockName = iterFind->second.first;
  155. GpuParamDescPtr otherDesc = iterFind->second.second;
  156. for (auto myParamIter = curDesc.params.begin(); myParamIter != curDesc.params.end(); ++myParamIter)
  157. {
  158. const GpuParamDataDesc& myParam = myParamIter->second;
  159. if (myParam.paramBlockSlot != curBlock.slot)
  160. continue; // Param is in another block, so we will check it when its time for that block
  161. auto otherParamFind = otherDesc->params.find(myParamIter->first);
  162. // Cannot find other param, blocks aren't equal
  163. if (otherParamFind == otherDesc->params.end())
  164. {
  165. isBlockValid = false;
  166. break;
  167. }
  168. const GpuParamDataDesc& otherParam = otherParamFind->second;
  169. if (!areParamsEqual(myParam, otherParam, false) || curBlock.name != otherBlockName)
  170. {
  171. isBlockValid = false;
  172. break;
  173. }
  174. }
  175. if (!isBlockValid)
  176. {
  177. if (validParamBlocks[blockIter->first])
  178. {
  179. LOGWRN("Found two param blocks with the same name but different contents: " + blockIter->first);
  180. validParamBlocks[blockIter->first] = false;
  181. }
  182. }
  183. }
  184. }
  185. Set<String> validParamBlocksReturn;
  186. for (auto iter = validParamBlocks.begin(); iter != validParamBlocks.end(); ++iter)
  187. {
  188. if (iter->second)
  189. validParamBlocksReturn.insert(iter->first);
  190. }
  191. return validParamBlocksReturn;
  192. }
  193. Map<String, String> determineParameterToBlockMapping(const Vector<GpuParamDescPtr>& paramDescs)
  194. {
  195. Map<String, String> paramToParamBlock;
  196. for (auto iter = paramDescs.begin(); iter != paramDescs.end(); ++iter)
  197. {
  198. const GpuParamDesc& curDesc = **iter;
  199. for (auto iter2 = curDesc.params.begin(); iter2 != curDesc.params.end(); ++iter2)
  200. {
  201. const GpuParamDataDesc& curParam = iter2->second;
  202. auto iterFind = paramToParamBlock.find(curParam.name);
  203. if (iterFind != paramToParamBlock.end())
  204. continue;
  205. for (auto iterBlock = curDesc.paramBlocks.begin(); iterBlock != curDesc.paramBlocks.end(); ++iterBlock)
  206. {
  207. if (iterBlock->second.slot == curParam.paramBlockSlot)
  208. {
  209. paramToParamBlock[curParam.name] = iterBlock->second.name;
  210. break;
  211. }
  212. }
  213. }
  214. }
  215. return paramToParamBlock;
  216. }
  217. Map<String, String> determineParamMappings(const Vector<GpuParamDescPtr>& paramDescs,
  218. const Map<String, SHADER_DATA_PARAM_DESC>& dataParams,
  219. const Map<String, SHADER_OBJECT_PARAM_DESC>& textureParams,
  220. const Map<String, SHADER_OBJECT_PARAM_DESC>& bufferParams,
  221. const Map<String, SHADER_OBJECT_PARAM_DESC>& samplerParams)
  222. {
  223. Map<String, String> validParams;
  224. Map<String, const GpuParamDataDesc*> validDataParameters = determineValidDataParameters(paramDescs);
  225. Vector<const GpuParamObjectDesc*> validObjectParameters = determineValidObjectParameters(paramDescs);
  226. Map<String, String> paramToParamBlockMap = determineParameterToBlockMapping(paramDescs);
  227. // Create data param mappings
  228. for (auto iter = dataParams.begin(); iter != dataParams.end(); ++iter)
  229. {
  230. auto findIter = validDataParameters.find(iter->second.gpuVariableName);
  231. // Not valid so we skip it
  232. if (findIter == validDataParameters.end())
  233. continue;
  234. if (findIter->second->type != iter->second.type && !(iter->second.type == GPDT_COLOR && findIter->second->type == GPDT_FLOAT4))
  235. {
  236. LOGWRN("Ignoring shader parameter \"" + iter->first + "\". Type doesn't match the one defined in the gpu program. "
  237. + "Shader defined type: " + toString(iter->second.type) + " - Gpu program defined type: " + toString(findIter->second->type));
  238. continue;
  239. }
  240. if (findIter->second->arraySize != iter->second.arraySize)
  241. {
  242. LOGWRN("Ignoring shader parameter \"" + iter->first + "\". Array size doesn't match the one defined in the gpu program."
  243. + "Shader defined array size: " + toString(iter->second.arraySize) + " - Gpu program defined array size: " + toString(findIter->second->arraySize));
  244. continue;
  245. }
  246. auto findBlockIter = paramToParamBlockMap.find(iter->second.gpuVariableName);
  247. if (findBlockIter == paramToParamBlockMap.end())
  248. BS_EXCEPT(InternalErrorException, "Parameter doesn't exist in param to param block map but exists in valid param map.");
  249. validParams[iter->first] = iter->second.gpuVariableName;
  250. }
  251. // Create object param mappings
  252. auto determineObjectMappings = [&](const Map<String, SHADER_OBJECT_PARAM_DESC>& params)
  253. {
  254. for (auto iter = params.begin(); iter != params.end(); ++iter)
  255. {
  256. const Vector<String>& gpuVariableNames = iter->second.gpuVariableNames;
  257. for (auto iter2 = gpuVariableNames.begin(); iter2 != gpuVariableNames.end(); ++iter2)
  258. {
  259. for (auto iter3 = validObjectParameters.begin(); iter3 != validObjectParameters.end(); ++iter3)
  260. {
  261. if ((*iter3)->name == (*iter2) && (*iter3)->type == iter->second.type)
  262. {
  263. validParams[iter->first] = *iter2;
  264. break;
  265. }
  266. }
  267. }
  268. }
  269. };
  270. determineObjectMappings(textureParams);
  271. determineObjectMappings(samplerParams);
  272. determineObjectMappings(bufferParams);
  273. return validParams;
  274. }
  275. Vector<ShaderBlockDesc> determineShaderBlockData(const Set<String>& paramBlocks, const Vector<GpuParamDescPtr>& paramDescs,
  276. const Map<String, SHADER_PARAM_BLOCK_DESC>& shaderDesc)
  277. {
  278. Vector<ShaderBlockDesc> output;
  279. for (auto iter = paramBlocks.begin(); iter != paramBlocks.end(); ++iter)
  280. {
  281. ShaderBlockDesc shaderBlockDesc;
  282. shaderBlockDesc.create = true;
  283. shaderBlockDesc.usage = GPBU_STATIC;
  284. shaderBlockDesc.size = 0;
  285. shaderBlockDesc.name = *iter;
  286. auto iterFind = shaderDesc.find(*iter);
  287. if (iterFind != shaderDesc.end())
  288. {
  289. shaderBlockDesc.create = !iterFind->second.shared && iterFind->second.rendererSemantic == StringID::NONE;
  290. shaderBlockDesc.usage = iterFind->second.usage;
  291. }
  292. for (auto iter2 = paramDescs.begin(); iter2 != paramDescs.end(); ++iter2)
  293. {
  294. auto findParamBlockDesc = (*iter2)->paramBlocks.find(*iter);
  295. if (findParamBlockDesc != (*iter2)->paramBlocks.end())
  296. {
  297. shaderBlockDesc.size = findParamBlockDesc->second.blockSize * sizeof(UINT32);
  298. break;
  299. }
  300. }
  301. output.push_back(shaderBlockDesc);
  302. }
  303. return output;
  304. }
  305. template<class T>
  306. bool isShaderValid(const T& shader) { return false; }
  307. template<>
  308. bool isShaderValid(const HShader& shader) { return shader.isLoaded(); }
  309. template<>
  310. bool isShaderValid(const SPtr<ShaderCore>& shader) { return shader != nullptr; }
  311. Vector<GpuParamDescPtr> MaterialBase::getAllParamDescs(const SPtr<Technique>& technique)
  312. {
  313. Vector<GpuParamDescPtr> allParamDescs;
  314. // Make sure all gpu programs are fully loaded
  315. for (UINT32 i = 0; i < technique->getNumPasses(); i++)
  316. {
  317. PassPtr curPass = technique->getPass(i);
  318. GpuProgramPtr vertProgram = curPass->getVertexProgram();
  319. if (vertProgram)
  320. {
  321. vertProgram->blockUntilCoreInitialized();
  322. allParamDescs.push_back(vertProgram->getParamDesc());
  323. }
  324. GpuProgramPtr fragProgram = curPass->getFragmentProgram();
  325. if (fragProgram)
  326. {
  327. fragProgram->blockUntilCoreInitialized();
  328. allParamDescs.push_back(fragProgram->getParamDesc());
  329. }
  330. GpuProgramPtr geomProgram = curPass->getGeometryProgram();
  331. if (geomProgram)
  332. {
  333. geomProgram->blockUntilCoreInitialized();
  334. allParamDescs.push_back(geomProgram->getParamDesc());
  335. }
  336. GpuProgramPtr hullProgram = curPass->getHullProgram();
  337. if (hullProgram)
  338. {
  339. hullProgram->blockUntilCoreInitialized();
  340. allParamDescs.push_back(hullProgram->getParamDesc());
  341. }
  342. GpuProgramPtr domainProgram = curPass->getDomainProgram();
  343. if (domainProgram)
  344. {
  345. domainProgram->blockUntilCoreInitialized();
  346. allParamDescs.push_back(domainProgram->getParamDesc());
  347. }
  348. GpuProgramPtr computeProgram = curPass->getComputeProgram();
  349. if (computeProgram)
  350. {
  351. computeProgram->blockUntilCoreInitialized();
  352. allParamDescs.push_back(computeProgram->getParamDesc());
  353. }
  354. }
  355. return allParamDescs;
  356. }
  357. Vector<GpuParamDescPtr> MaterialBase::getAllParamDescs(const SPtr<TechniqueCore>& technique)
  358. {
  359. Vector<GpuParamDescPtr> allParamDescs;
  360. // Make sure all gpu programs are fully loaded
  361. for (UINT32 i = 0; i < technique->getNumPasses(); i++)
  362. {
  363. SPtr<PassCore> curPass = technique->getPass(i);
  364. SPtr<GpuProgramCore> vertProgram = curPass->getVertexProgram();
  365. if (vertProgram)
  366. allParamDescs.push_back(vertProgram->getParamDesc());
  367. SPtr<GpuProgramCore> fragProgram = curPass->getFragmentProgram();
  368. if (fragProgram)
  369. allParamDescs.push_back(fragProgram->getParamDesc());
  370. SPtr<GpuProgramCore> geomProgram = curPass->getGeometryProgram();
  371. if (geomProgram)
  372. allParamDescs.push_back(geomProgram->getParamDesc());
  373. SPtr<GpuProgramCore> hullProgram = curPass->getHullProgram();
  374. if (hullProgram)
  375. allParamDescs.push_back(hullProgram->getParamDesc());
  376. SPtr<GpuProgramCore> domainProgram = curPass->getDomainProgram();
  377. if (domainProgram)
  378. allParamDescs.push_back(domainProgram->getParamDesc());
  379. SPtr<GpuProgramCore> computeProgram = curPass->getComputeProgram();
  380. if (computeProgram)
  381. allParamDescs.push_back(computeProgram->getParamDesc());
  382. }
  383. return allParamDescs;
  384. }
  385. template<bool Core>
  386. UINT32 TMaterial<Core>::getNumPasses() const
  387. {
  388. if (mShader == nullptr)
  389. return 0;
  390. return mShader->getBestTechnique()->getNumPasses();
  391. }
  392. template<bool Core>
  393. typename TMaterial<Core>::PassType TMaterial<Core>::getPass(UINT32 passIdx) const
  394. {
  395. if (passIdx < 0 || passIdx >= mShader->getBestTechnique()->getNumPasses())
  396. BS_EXCEPT(InvalidParametersException, "Invalid pass index.");
  397. return mShader->getBestTechnique()->getPass(passIdx);
  398. }
  399. template<bool Core>
  400. TMaterialParamStruct<Core> TMaterial<Core>::getParamStruct(const String& name) const
  401. {
  402. throwIfNotInitialized();
  403. auto iterFind = mValidParams.find(name);
  404. if (iterFind == mValidParams.end())
  405. {
  406. LOGWRN_VERBOSE("Material doesn't have a parameter named " + name);
  407. return TMaterialParamStruct<Core>();
  408. }
  409. const String& gpuVarName = iterFind->second;
  410. SPtr<Vector<TGpuParamStruct<Core>>> gpuParams = bs_shared_ptr_new<Vector<TGpuParamStruct<Core>>>();
  411. for (auto iter = mParametersPerPass.begin(); iter != mParametersPerPass.end(); ++iter)
  412. {
  413. SPtr<PassParamsType> params = *iter;
  414. for (UINT32 i = 0; i < PassParamsType::NUM_PARAMS; i++)
  415. {
  416. GpuParamsType& paramPtr = params->getParamByIdx(i);
  417. if (paramPtr)
  418. {
  419. if (paramPtr->hasParam(gpuVarName))
  420. {
  421. gpuParams->push_back(TGpuParamStruct<Core>());
  422. paramPtr->getStructParam(gpuVarName, gpuParams->back());
  423. }
  424. }
  425. }
  426. }
  427. return createStructParam(name, gpuParams);
  428. }
  429. template<bool Core>
  430. TMaterialParamTexture<Core> TMaterial<Core>::getParamTexture(const String& name) const
  431. {
  432. throwIfNotInitialized();
  433. auto iterFind = mValidParams.find(name);
  434. if (iterFind == mValidParams.end())
  435. {
  436. LOGWRN_VERBOSE("Material doesn't have a parameter named " + name);
  437. return TMaterialParamTexture<Core>();
  438. }
  439. const String& gpuVarName = iterFind->second;
  440. SPtr<Vector<TGpuParamTexture<Core>>> gpuParams = bs_shared_ptr_new<Vector<TGpuParamTexture<Core>>>();
  441. for (auto iter = mParametersPerPass.begin(); iter != mParametersPerPass.end(); ++iter)
  442. {
  443. SPtr<PassParamsType> params = *iter;
  444. for (UINT32 i = 0; i < PassParamsType::NUM_PARAMS; i++)
  445. {
  446. GpuParamsType& paramPtr = params->getParamByIdx(i);
  447. if (paramPtr)
  448. {
  449. if (paramPtr->hasTexture(gpuVarName))
  450. {
  451. gpuParams->push_back(TGpuParamTexture<Core>());
  452. paramPtr->getTextureParam(gpuVarName, gpuParams->back());
  453. }
  454. }
  455. }
  456. }
  457. return createTextureParam(name, gpuParams);
  458. }
  459. template<bool Core>
  460. TMaterialParamLoadStoreTexture<Core> TMaterial<Core>::getParamLoadStoreTexture(const String& name) const
  461. {
  462. throwIfNotInitialized();
  463. auto iterFind = mValidParams.find(name);
  464. if (iterFind == mValidParams.end())
  465. {
  466. LOGWRN_VERBOSE("Material doesn't have a parameter named " + name);
  467. return TMaterialParamLoadStoreTexture<Core>();
  468. }
  469. const String& gpuVarName = iterFind->second;
  470. SPtr<Vector<TGpuParamLoadStoreTexture<Core>>> gpuParams = bs_shared_ptr_new<Vector<TGpuParamLoadStoreTexture<Core>>>();
  471. for (auto iter = mParametersPerPass.begin(); iter != mParametersPerPass.end(); ++iter)
  472. {
  473. SPtr<PassParamsType> params = *iter;
  474. for (UINT32 i = 0; i < PassParamsType::NUM_PARAMS; i++)
  475. {
  476. GpuParamsType& paramPtr = params->getParamByIdx(i);
  477. if (paramPtr)
  478. {
  479. if (paramPtr->hasTexture(gpuVarName))
  480. {
  481. gpuParams->push_back(TGpuParamLoadStoreTexture<Core>());
  482. paramPtr->getLoadStoreTextureParam(gpuVarName, gpuParams->back());
  483. }
  484. }
  485. }
  486. }
  487. return createLoadStoreTextureParam(name, gpuParams);
  488. }
  489. template<bool Core>
  490. TMaterialParamSampState<Core> TMaterial<Core>::getParamSamplerState(const String& name) const
  491. {
  492. throwIfNotInitialized();
  493. auto iterFind = mValidParams.find(name);
  494. if (iterFind == mValidParams.end())
  495. {
  496. LOGWRN_VERBOSE("Material doesn't have a parameter named " + name);
  497. return TMaterialParamSampState<Core>();
  498. }
  499. const String& gpuVarName = iterFind->second;
  500. SPtr<Vector<TGpuParamSampState<Core>>> gpuParams = bs_shared_ptr_new<Vector<TGpuParamSampState<Core>>>();
  501. for (auto iter = mParametersPerPass.begin(); iter != mParametersPerPass.end(); ++iter)
  502. {
  503. SPtr<PassParamsType> params = *iter;
  504. for (UINT32 i = 0; i < PassParamsType::NUM_PARAMS; i++)
  505. {
  506. GpuParamsType& paramPtr = params->getParamByIdx(i);
  507. if (paramPtr)
  508. {
  509. if (paramPtr->hasSamplerState(gpuVarName))
  510. {
  511. gpuParams->push_back(TGpuParamSampState<Core>());
  512. paramPtr->getSamplerStateParam(gpuVarName, gpuParams->back());
  513. }
  514. }
  515. }
  516. }
  517. return createSamplerStateParam(name, gpuParams);
  518. }
  519. template<bool Core>
  520. void TMaterial<Core>::setParamBlockBuffer(const String& name, const ParamBlockPtrType& paramBlock)
  521. {
  522. auto iterFind = mValidShareableParamBlocks.find(name);
  523. if (iterFind == mValidShareableParamBlocks.end())
  524. {
  525. LOGWRN_VERBOSE("Material doesn't have a parameter block named " + name);
  526. return;
  527. }
  528. for (auto iter = mParametersPerPass.begin(); iter != mParametersPerPass.end(); ++iter)
  529. {
  530. SPtr<PassParamsType> params = *iter;
  531. for (UINT32 i = 0; i < PassParamsType::NUM_PARAMS; i++)
  532. {
  533. GpuParamsType& paramPtr = params->getParamByIdx(i);
  534. if (paramPtr)
  535. {
  536. if (paramPtr->hasParamBlock(name))
  537. paramPtr->setParamBlockBuffer(name, paramBlock);
  538. }
  539. }
  540. }
  541. }
  542. template<bool Core>
  543. void TMaterial<Core>::initBestTechnique()
  544. {
  545. mBestTechnique = nullptr;
  546. mParametersPerPass.clear();
  547. if (isShaderValid(mShader))
  548. {
  549. createCachedParams(mShader);
  550. mBestTechnique = mShader->getBestTechnique();
  551. if (mBestTechnique == nullptr)
  552. return;
  553. mValidShareableParamBlocks.clear();
  554. Vector<GpuParamDescPtr> allParamDescs = getAllParamDescs(mBestTechnique);
  555. mValidParams = determineParamMappings(allParamDescs, mShader->getDataParams(), mShader->getTextureParams(),
  556. mShader->getBufferParams(), mShader->getSamplerParams());
  557. // Fill out various helper structures
  558. Set<String> validShareableParamBlocks = determineValidShareableParamBlocks(allParamDescs);
  559. Vector<ShaderBlockDesc> paramBlockData = determineShaderBlockData(validShareableParamBlocks, allParamDescs, mShader->getParamBlocks());
  560. Map<String, ParamBlockPtrType> paramBlockBuffers;
  561. // Create param blocks
  562. for (auto& paramBlock : paramBlockData)
  563. {
  564. ParamBlockPtrType newParamBlockBuffer;
  565. if (paramBlock.create)
  566. {
  567. newParamBlockBuffer = ParamBlockType::create(paramBlock.size, paramBlock.usage);
  568. }
  569. paramBlockBuffers[paramBlock.name] = newParamBlockBuffer;
  570. mValidShareableParamBlocks.insert(paramBlock.name);
  571. }
  572. for (UINT32 i = 0; i < mBestTechnique->getNumPasses(); i++)
  573. {
  574. PassType curPass = mBestTechnique->getPass(i);
  575. SPtr<PassParamsType> params = SPtr<PassParamsType>(new PassParamsType());
  576. GpuProgramType vertProgram = curPass->getVertexProgram();
  577. if (vertProgram)
  578. params->mVertParams = vertProgram->createParameters();
  579. GpuProgramType fragProgram = curPass->getFragmentProgram();
  580. if (fragProgram)
  581. params->mFragParams = fragProgram->createParameters();
  582. GpuProgramType geomProgram = curPass->getGeometryProgram();
  583. if (geomProgram)
  584. params->mGeomParams = geomProgram->createParameters();
  585. GpuProgramType hullProgram = curPass->getHullProgram();
  586. if (hullProgram)
  587. params->mHullParams = hullProgram->createParameters();
  588. GpuProgramType domainProgram = curPass->getDomainProgram();
  589. if (domainProgram)
  590. params->mDomainParams = domainProgram->createParameters();
  591. GpuProgramType computeProgram = curPass->getComputeProgram();
  592. if (computeProgram)
  593. params->mComputeParams = computeProgram->createParameters();
  594. mParametersPerPass.push_back(params);
  595. }
  596. // Assign param block buffers
  597. for (auto iter = mParametersPerPass.begin(); iter != mParametersPerPass.end(); ++iter)
  598. {
  599. SPtr<PassParamsType> params = *iter;
  600. for (UINT32 i = 0; i < PassParamsType::NUM_PARAMS; i++)
  601. {
  602. GpuParamsType& paramPtr = params->getParamByIdx(i);
  603. if (paramPtr)
  604. {
  605. // Assign shareable buffers
  606. for (auto iterBlock = mValidShareableParamBlocks.begin(); iterBlock != mValidShareableParamBlocks.end(); ++iterBlock)
  607. {
  608. const String& paramBlockName = *iterBlock;
  609. if (paramPtr->hasParamBlock(paramBlockName))
  610. {
  611. ParamBlockPtrType blockBuffer = paramBlockBuffers[paramBlockName];
  612. paramPtr->setParamBlockBuffer(paramBlockName, blockBuffer);
  613. }
  614. }
  615. // Create non-shareable ones
  616. const GpuParamDesc& desc = paramPtr->getParamDesc();
  617. for (auto iterBlockDesc = desc.paramBlocks.begin(); iterBlockDesc != desc.paramBlocks.end(); ++iterBlockDesc)
  618. {
  619. if (!iterBlockDesc->second.isShareable)
  620. {
  621. ParamBlockPtrType newParamBlockBuffer = ParamBlockType::create(iterBlockDesc->second.blockSize * sizeof(UINT32));
  622. paramPtr->setParamBlockBuffer(iterBlockDesc->first, newParamBlockBuffer);
  623. }
  624. }
  625. }
  626. }
  627. }
  628. // Assign default parameters
  629. initDefaultParameters();
  630. }
  631. _markDependenciesDirty();
  632. }
  633. template <bool Core>
  634. template <typename T>
  635. void TMaterial<Core>::setParamValue(const String& name, UINT8* buffer, UINT32 numElements)
  636. {
  637. TMaterialDataParam<T, Core> param;
  638. getParam(name, param);
  639. T* ptr = (T*)buffer;
  640. for (UINT32 i = 0; i < numElements; i++)
  641. param.set(ptr[i], i);
  642. }
  643. template <bool Core>
  644. void TMaterial<Core>::initDefaultParameters()
  645. {
  646. const Map<String, SHADER_DATA_PARAM_DESC>& dataParams = mShader->getDataParams();
  647. for (auto& paramData : dataParams)
  648. {
  649. if (paramData.second.defaultValueIdx == (UINT32)-1)
  650. continue;
  651. auto iterFind = mValidParams.find(paramData.first);
  652. if (iterFind == mValidParams.end())
  653. continue;
  654. UINT8* buffer = (UINT8*)mShader->getDefaultValue(paramData.second.defaultValueIdx);
  655. if (buffer == nullptr)
  656. continue;
  657. switch (paramData.second.type)
  658. {
  659. case GPDT_FLOAT1:
  660. setParamValue<float>(iterFind->first, buffer, paramData.second.arraySize);
  661. break;
  662. case GPDT_FLOAT2:
  663. setParamValue<Vector2>(iterFind->first, buffer, paramData.second.arraySize);
  664. break;
  665. case GPDT_FLOAT3:
  666. setParamValue<Vector3>(iterFind->first, buffer, paramData.second.arraySize);
  667. break;
  668. case GPDT_FLOAT4:
  669. setParamValue<Vector4>(iterFind->first, buffer, paramData.second.arraySize);
  670. break;
  671. case GPDT_MATRIX_2X2:
  672. setParamValue<Matrix2>(iterFind->first, buffer, paramData.second.arraySize);
  673. break;
  674. case GPDT_MATRIX_2X3:
  675. setParamValue<Matrix2x3>(iterFind->first, buffer, paramData.second.arraySize);
  676. break;
  677. case GPDT_MATRIX_2X4:
  678. setParamValue<Matrix2x4>(iterFind->first, buffer, paramData.second.arraySize);
  679. break;
  680. case GPDT_MATRIX_3X2:
  681. setParamValue<Matrix3x2>(iterFind->first, buffer, paramData.second.arraySize);
  682. break;
  683. case GPDT_MATRIX_3X3:
  684. setParamValue<Matrix3>(iterFind->first, buffer, paramData.second.arraySize);
  685. break;
  686. case GPDT_MATRIX_3X4:
  687. setParamValue<Matrix3x4>(iterFind->first, buffer, paramData.second.arraySize);
  688. break;
  689. case GPDT_MATRIX_4X2:
  690. setParamValue<Matrix4x2>(iterFind->first, buffer, paramData.second.arraySize);
  691. break;
  692. case GPDT_MATRIX_4X3:
  693. setParamValue<Matrix4x3>(iterFind->first, buffer, paramData.second.arraySize);
  694. break;
  695. case GPDT_MATRIX_4X4:
  696. setParamValue<Matrix4>(iterFind->first, buffer, paramData.second.arraySize);
  697. break;
  698. case GPDT_INT1:
  699. setParamValue<int>(iterFind->first, buffer, paramData.second.arraySize);
  700. break;
  701. case GPDT_INT2:
  702. setParamValue<Vector2I>(iterFind->first, buffer, paramData.second.arraySize);
  703. break;
  704. case GPDT_INT3:
  705. setParamValue<Vector3I>(iterFind->first, buffer, paramData.second.arraySize);
  706. break;
  707. case GPDT_INT4:
  708. setParamValue<Vector4I>(iterFind->first, buffer, paramData.second.arraySize);
  709. break;
  710. case GPDT_BOOL:
  711. setParamValue<int>(iterFind->first, buffer, paramData.second.arraySize);
  712. break;
  713. case GPDT_COLOR:
  714. setParamValue<Color>(iterFind->first, buffer, paramData.second.arraySize);
  715. break;
  716. case GPDT_STRUCT:
  717. {
  718. TMaterialParamStruct<Core> param = getParamStruct(paramData.first);
  719. UINT32 elementSizeBytes = paramData.second.elementSize * sizeof(UINT32);
  720. UINT8* ptr = buffer;
  721. for (UINT32 i = 0; i < paramData.second.arraySize; i++)
  722. {
  723. param.set(ptr, elementSizeBytes, i);
  724. ptr += elementSizeBytes;
  725. }
  726. }
  727. break;
  728. }
  729. }
  730. const Map<String, SHADER_OBJECT_PARAM_DESC>& textureParams = mShader->getTextureParams();
  731. for (auto& param : textureParams)
  732. {
  733. if (param.second.defaultValueIdx == (UINT32)-1)
  734. continue;
  735. auto iterFind = mValidParams.find(param.first);
  736. if (iterFind == mValidParams.end())
  737. continue;
  738. TextureType defaultTex = mShader->getDefaultTexture(param.second.defaultValueIdx);
  739. getParamTexture(param.first).set(defaultTex);
  740. }
  741. const Map<String, SHADER_OBJECT_PARAM_DESC>& samplerParams = mShader->getSamplerParams();
  742. for (auto& param : samplerParams)
  743. {
  744. if (param.second.defaultValueIdx == (UINT32)-1)
  745. continue;
  746. auto iterFind = mValidParams.find(param.first);
  747. if (iterFind == mValidParams.end())
  748. continue;
  749. SamplerStateType defaultSampler = mShader->getDefaultSampler(param.second.defaultValueIdx);
  750. getParamSamplerState(param.first).set(defaultSampler);
  751. }
  752. }
  753. template <bool Core>
  754. template <typename T>
  755. void TMaterial<Core>::getParam(const String& name, TMaterialDataParam<T, Core>& output) const
  756. {
  757. throwIfNotInitialized();
  758. auto iterFind = mValidParams.find(name);
  759. if (iterFind == mValidParams.end())
  760. {
  761. LOGWRN_VERBOSE("Material doesn't have a parameter named " + name);
  762. return;
  763. }
  764. const String& gpuVarName = iterFind->second;
  765. SPtr<Vector<TGpuDataParam<T, Core>>> gpuParams = bs_shared_ptr_new<Vector<TGpuDataParam<T, Core>>>();
  766. for (auto iter = mParametersPerPass.begin(); iter != mParametersPerPass.end(); ++iter)
  767. {
  768. SPtr<PassParamsType> params = *iter;
  769. for (UINT32 i = 0; i < PassParamsType::NUM_PARAMS; i++)
  770. {
  771. GpuParamsType& paramPtr = params->getParamByIdx(i);
  772. if (paramPtr)
  773. {
  774. if (paramPtr->hasParam(gpuVarName))
  775. {
  776. gpuParams->push_back(TGpuDataParam<T, Core>());
  777. paramPtr->getParam<T>(gpuVarName, gpuParams->back());
  778. }
  779. }
  780. }
  781. }
  782. output = createDataParam(name, gpuParams);
  783. }
  784. template<bool Core>
  785. void TMaterial<Core>::throwIfNotInitialized() const
  786. {
  787. if (mShader == nullptr)
  788. {
  789. BS_EXCEPT(InternalErrorException, "Material does not have shader set.");
  790. }
  791. if (mBestTechnique == nullptr)
  792. {
  793. BS_EXCEPT(InternalErrorException, "Shader does not contain a supported technique.");
  794. }
  795. }
  796. template class TMaterial < false > ;
  797. template class TMaterial < true > ;
  798. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<float, false>&) const;
  799. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<int, false>&) const;
  800. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Color, false>&) const;
  801. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Vector2, false>&) const;
  802. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Vector3, false>&) const;
  803. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Vector4, false>&) const;
  804. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Vector2I, false>&) const;
  805. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Vector3I, false>&) const;
  806. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Vector4I, false>&) const;
  807. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix2, false>&) const;
  808. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix2x3, false>&) const;
  809. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix2x4, false>&) const;
  810. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix3, false>&) const;
  811. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix3x2, false>&) const;
  812. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix3x4, false>&) const;
  813. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix4, false>&) const;
  814. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix4x2, false>&) const;
  815. template BS_CORE_EXPORT void TMaterial<false>::getParam(const String&, TMaterialDataParam<Matrix4x3, false>&) const;
  816. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<float, true>&) const;
  817. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<int, true>&) const;
  818. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Color, true>&) const;
  819. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Vector2, true>&) const;
  820. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Vector3, true>&) const;
  821. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Vector4, true>&) const;
  822. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Vector2I, true>&) const;
  823. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Vector3I, true>&) const;
  824. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Vector4I, true>&) const;
  825. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix2, true>&) const;
  826. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix2x3, true>&) const;
  827. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix2x4, true>&) const;
  828. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix3, true>&) const;
  829. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix3x2, true>&) const;
  830. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix3x4, true>&) const;
  831. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix4, true>&) const;
  832. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix4x2, true>&) const;
  833. template BS_CORE_EXPORT void TMaterial<true>::getParam(const String&, TMaterialDataParam<Matrix4x3, true>&) const;
  834. MaterialCore::MaterialCore(const SPtr<ShaderCore>& shader)
  835. {
  836. setShader(shader);
  837. }
  838. MaterialCore::MaterialCore(const SPtr<ShaderCore>& shader, const SPtr<TechniqueCore>& bestTechnique,
  839. const Set<String>& validShareableParamBlocks, const Map<String, String>& validParams,
  840. const Vector<SPtr<PassParametersCore>>& passParams)
  841. {
  842. mShader = shader;
  843. mBestTechnique = bestTechnique;
  844. mValidShareableParamBlocks = validShareableParamBlocks;
  845. mValidParams = validParams;
  846. UINT32 numPassParams = (UINT32)passParams.size();
  847. mParametersPerPass.resize(numPassParams);
  848. for (UINT32 i = 0; i < numPassParams; i++)
  849. mParametersPerPass[i] = passParams[i];
  850. }
  851. void MaterialCore::setShader(const SPtr<ShaderCore>& shader)
  852. {
  853. mShader = shader;
  854. initBestTechnique();
  855. _markCoreDirty();
  856. }
  857. void MaterialCore::syncToCore(const CoreSyncData& data)
  858. {
  859. char* dataPtr = (char*)data.getBuffer();
  860. mValidShareableParamBlocks.clear();
  861. mValidParams.clear();
  862. mParametersPerPass.clear();
  863. UINT32 numPasses = 0;
  864. dataPtr = rttiReadElem(mValidShareableParamBlocks, dataPtr);
  865. dataPtr = rttiReadElem(mValidParams, dataPtr);
  866. dataPtr = rttiReadElem(numPasses, dataPtr);
  867. for (UINT32 i = 0; i < numPasses; i++)
  868. {
  869. SPtr<PassParametersCore>* passParameters = (SPtr<PassParametersCore>*)dataPtr;
  870. mParametersPerPass.push_back(*passParameters);
  871. passParameters->~SPtr<PassParametersCore>();
  872. dataPtr += sizeof(SPtr<PassParametersCore>);
  873. }
  874. SPtr<ShaderCore>* shader = (SPtr<ShaderCore>*)dataPtr;
  875. mShader = *shader;
  876. shader->~SPtr<ShaderCore>();
  877. dataPtr += sizeof(SPtr<ShaderCore>);
  878. SPtr<TechniqueCore>* technique = (SPtr<TechniqueCore>*)dataPtr;
  879. mBestTechnique = *technique;
  880. technique->~SPtr<TechniqueCore>();
  881. dataPtr += sizeof(SPtr<TechniqueCore>);
  882. }
  883. SPtr<MaterialCore> MaterialCore::create(const SPtr<ShaderCore>& shader)
  884. {
  885. MaterialCore* material = new (bs_alloc<MaterialCore>()) MaterialCore(shader);
  886. SPtr<MaterialCore> materialPtr = bs_shared_ptr<MaterialCore>(material);
  887. materialPtr->_setThisPtr(materialPtr);
  888. materialPtr->initialize();
  889. return materialPtr;
  890. }
  891. Material::Material()
  892. :mLoadFlags(Load_None)
  893. { }
  894. Material::Material(const HShader& shader)
  895. :mLoadFlags(Load_None)
  896. {
  897. mShader = shader;
  898. }
  899. void Material::initialize()
  900. {
  901. setShader(mShader); // Not calling directly in constructor because it calls virtual methods
  902. Resource::initialize();
  903. }
  904. void Material::setShader(const HShader& shader)
  905. {
  906. //if (mShader == shader)
  907. // return;
  908. mShader = shader;
  909. mBestTechnique = nullptr;
  910. mLoadFlags = Load_None;
  911. _markResourcesDirty();
  912. initializeIfLoaded();
  913. }
  914. void Material::_markCoreDirty()
  915. {
  916. markCoreDirty();
  917. }
  918. void Material::_markDependenciesDirty()
  919. {
  920. markDependenciesDirty();
  921. }
  922. void Material::_markResourcesDirty()
  923. {
  924. markListenerResourcesDirty();
  925. }
  926. SPtr<MaterialCore> Material::getCore() const
  927. {
  928. return std::static_pointer_cast<MaterialCore>(mCoreSpecific);
  929. }
  930. SPtr<CoreObjectCore> Material::createCore() const
  931. {
  932. MaterialCore* material = nullptr;
  933. SPtr<ShaderCore> shader;
  934. if (mShader.isLoaded())
  935. {
  936. shader = mShader->getCore();
  937. if (mBestTechnique != nullptr)
  938. {
  939. SPtr<TechniqueCore> technique = mBestTechnique->getCore();
  940. Vector<SPtr<PassParametersCore>> passParams;
  941. for (auto& passParam : mParametersPerPass)
  942. passParams.push_back(convertParamsToCore(passParam));
  943. material = new (bs_alloc<MaterialCore>()) MaterialCore(shader, technique, mValidShareableParamBlocks, mValidParams, passParams);
  944. }
  945. }
  946. if (material == nullptr)
  947. material = new (bs_alloc<MaterialCore>()) MaterialCore(shader);
  948. SPtr<MaterialCore> materialPtr = bs_shared_ptr<MaterialCore>(material);
  949. materialPtr->_setThisPtr(materialPtr);
  950. return materialPtr;
  951. }
  952. CoreSyncData Material::syncToCore(FrameAlloc* allocator)
  953. {
  954. UINT32 numPasses = (UINT32)mParametersPerPass.size();
  955. UINT32 size = sizeof(UINT32) + numPasses * sizeof(SPtr<PassParametersCore>)
  956. + sizeof(SPtr<ShaderCore>) + sizeof(SPtr<TechniqueCore>) + rttiGetElemSize(mValidShareableParamBlocks)
  957. + rttiGetElemSize(mValidParams);
  958. UINT8* buffer = allocator->alloc(size);
  959. char* dataPtr = (char*)buffer;
  960. dataPtr = rttiWriteElem(mValidShareableParamBlocks, dataPtr);
  961. dataPtr = rttiWriteElem(mValidParams, dataPtr);
  962. dataPtr = rttiWriteElem(numPasses, dataPtr);
  963. for (UINT32 i = 0; i < numPasses; i++)
  964. {
  965. SPtr<PassParametersCore>* passParameters = new (dataPtr) SPtr<PassParametersCore>();
  966. *passParameters = convertParamsToCore(mParametersPerPass[i]);
  967. dataPtr += sizeof(SPtr<PassParametersCore>);
  968. }
  969. SPtr<ShaderCore>* shader = new (dataPtr)SPtr<ShaderCore>();
  970. if (mShader.isLoaded(false))
  971. *shader = mShader->getCore();
  972. else
  973. *shader = nullptr;
  974. dataPtr += sizeof(SPtr<ShaderCore>);
  975. SPtr<TechniqueCore>* technique = new (dataPtr)SPtr<TechniqueCore>();
  976. if (mBestTechnique != nullptr)
  977. *technique = mBestTechnique->getCore();
  978. else
  979. *technique = nullptr;
  980. dataPtr += sizeof(SPtr<TechniqueCore>);
  981. return CoreSyncData(buffer, size);
  982. }
  983. void Material::getCoreDependencies(Vector<CoreObject*>& dependencies)
  984. {
  985. if (mShader.isLoaded())
  986. dependencies.push_back(mShader.get());
  987. for (auto& params : mParametersPerPass)
  988. {
  989. for (UINT32 i = 0; i < PassParameters::NUM_PARAMS; i++)
  990. {
  991. GpuParamsPtr gpuParams = params->getParamByIdx(i);
  992. if (gpuParams != nullptr)
  993. dependencies.push_back(gpuParams.get());
  994. }
  995. }
  996. }
  997. void Material::getListenerResources(Vector<HResource>& resources)
  998. {
  999. if (mShader != nullptr)
  1000. resources.push_back(mShader);
  1001. }
  1002. void Material::getResourceDependencies(FrameVector<HResource>& dependencies) const
  1003. {
  1004. if (mShader != nullptr)
  1005. dependencies.push_back(mShader);
  1006. if (mShader.isLoaded())
  1007. {
  1008. const Map<String, SHADER_OBJECT_PARAM_DESC>& textureParams = mShader->getTextureParams();
  1009. for (auto& param : textureParams)
  1010. {
  1011. if (param.second.defaultValueIdx != (UINT32)-1)
  1012. {
  1013. HTexture defaultTex = mShader->getDefaultTexture(param.second.defaultValueIdx);
  1014. if (defaultTex != nullptr)
  1015. dependencies.push_back(defaultTex);
  1016. }
  1017. }
  1018. }
  1019. }
  1020. void Material::initializeIfLoaded()
  1021. {
  1022. if (areDependenciesLoaded())
  1023. {
  1024. if (mLoadFlags != Load_All)
  1025. {
  1026. mLoadFlags = Load_All;
  1027. initBestTechnique();
  1028. markCoreDirty();
  1029. }
  1030. }
  1031. else
  1032. {
  1033. if (mShader.isLoaded() && mLoadFlags == Load_None)
  1034. {
  1035. mLoadFlags = Load_Shader;
  1036. markListenerResourcesDirty(); // Need to register resources dependent on shader now
  1037. }
  1038. }
  1039. }
  1040. void Material::createCachedParams(const HShader& shader)
  1041. {
  1042. mCachedParams = bs_shared_ptr_new<__MaterialParams>(shader);
  1043. }
  1044. void Material::notifyResourceLoaded(const HResource& resource)
  1045. {
  1046. initializeIfLoaded();
  1047. }
  1048. template<class T>
  1049. void copyParam(SPtr<GpuParams>& from, SPtr<GpuParams>& to, const String& name, UINT32 arraySize)
  1050. {
  1051. TGpuDataParam<T, false> newParam;
  1052. to->getParam(name, newParam);
  1053. TGpuDataParam<T, false> oldParam;
  1054. from->getParam(name, oldParam);
  1055. for (UINT32 i = 0; i < arraySize; i++)
  1056. newParam.set(oldParam.get(i), i);
  1057. }
  1058. void Material::notifyResourceChanged(const HResource& resource)
  1059. {
  1060. // Shader changed, so save parameters, rebuild material and restore parameters
  1061. Vector<SPtr<PassParameters>> oldPassParams = mParametersPerPass;
  1062. mLoadFlags = Load_None;
  1063. initializeIfLoaded();
  1064. for (UINT32 i = 0; i < (UINT32)oldPassParams.size(); i++)
  1065. {
  1066. if (i >= (UINT32)mParametersPerPass.size())
  1067. continue;
  1068. SPtr<PassParameters> oldParams = oldPassParams[i];
  1069. SPtr<PassParameters> newParams = mParametersPerPass[i];
  1070. for (UINT32 j = 0; j < PassParameters::NUM_PARAMS; j++)
  1071. {
  1072. SPtr<GpuParams>& oldGpuParams = oldParams->getParamByIdx(j);
  1073. SPtr<GpuParams>& newGpuParams = newParams->getParamByIdx(j);
  1074. if (oldGpuParams == nullptr || newGpuParams == nullptr)
  1075. continue;
  1076. const GpuParamDesc& oldDesc = oldGpuParams->getParamDesc();
  1077. const GpuParamDesc& newDesc = newGpuParams->getParamDesc();
  1078. for (auto& param : oldDesc.params)
  1079. {
  1080. auto iterFind = newDesc.params.find(param.first);
  1081. if (iterFind == newDesc.params.end())
  1082. continue;
  1083. if (param.second.type != iterFind->second.type)
  1084. continue;
  1085. UINT32 arraySize = std::min(param.second.arraySize, iterFind->second.arraySize);
  1086. switch (param.second.type)
  1087. {
  1088. case GPDT_FLOAT1:
  1089. copyParam<float>(oldGpuParams, newGpuParams, param.first, arraySize);
  1090. break;
  1091. case GPDT_FLOAT2:
  1092. copyParam<Vector2>(oldGpuParams, newGpuParams, param.first, arraySize);
  1093. break;
  1094. case GPDT_FLOAT3:
  1095. copyParam<Vector3>(oldGpuParams, newGpuParams, param.first, arraySize);
  1096. break;
  1097. case GPDT_FLOAT4:
  1098. copyParam<Vector4>(oldGpuParams, newGpuParams, param.first, arraySize);
  1099. break;
  1100. case GPDT_MATRIX_2X2:
  1101. copyParam<Matrix2>(oldGpuParams, newGpuParams, param.first, arraySize);
  1102. break;
  1103. case GPDT_MATRIX_2X3:
  1104. copyParam<Matrix2x3>(oldGpuParams, newGpuParams, param.first, arraySize);
  1105. break;
  1106. case GPDT_MATRIX_2X4:
  1107. copyParam<Matrix2x4>(oldGpuParams, newGpuParams, param.first, arraySize);
  1108. break;
  1109. case GPDT_MATRIX_3X2:
  1110. copyParam<Matrix3x2>(oldGpuParams, newGpuParams, param.first, arraySize);
  1111. break;
  1112. case GPDT_MATRIX_3X3:
  1113. copyParam<Matrix3>(oldGpuParams, newGpuParams, param.first, arraySize);
  1114. break;
  1115. case GPDT_MATRIX_3X4:
  1116. copyParam<Matrix3x4>(oldGpuParams, newGpuParams, param.first, arraySize);
  1117. break;
  1118. case GPDT_MATRIX_4X2:
  1119. copyParam<Matrix4x2>(oldGpuParams, newGpuParams, param.first, arraySize);
  1120. break;
  1121. case GPDT_MATRIX_4X3:
  1122. copyParam<Matrix4x3>(oldGpuParams, newGpuParams, param.first, arraySize);
  1123. break;
  1124. case GPDT_MATRIX_4X4:
  1125. copyParam<Matrix4>(oldGpuParams, newGpuParams, param.first, arraySize);
  1126. break;
  1127. case GPDT_INT1:
  1128. case GPDT_BOOL:
  1129. copyParam<int>(oldGpuParams, newGpuParams, param.first, arraySize);
  1130. break;
  1131. case GPDT_INT2:
  1132. copyParam<Vector2I>(oldGpuParams, newGpuParams, param.first, arraySize);
  1133. break;
  1134. case GPDT_INT3:
  1135. copyParam<Vector3I>(oldGpuParams, newGpuParams, param.first, arraySize);
  1136. break;
  1137. case GPDT_INT4:
  1138. copyParam<Vector4I>(oldGpuParams, newGpuParams, param.first, arraySize);
  1139. break;
  1140. case GPDT_COLOR:
  1141. copyParam<Color>(oldGpuParams, newGpuParams, param.first, arraySize);
  1142. break;
  1143. case GPDT_STRUCT:
  1144. {
  1145. GpuParamStruct newParam;
  1146. newGpuParams->getStructParam(param.first, newParam);
  1147. GpuParamStruct oldParam;
  1148. oldGpuParams->getStructParam(param.first, oldParam);
  1149. if (param.second.elementSize == iterFind->second.elementSize)
  1150. {
  1151. for (UINT32 k = 0; k < arraySize; k++)
  1152. {
  1153. UINT8* data = (UINT8*)bs_stack_alloc(param.second.elementSize);
  1154. oldParam.get(data, param.second.elementSize, k);
  1155. newParam.set(data, param.second.elementSize, k);
  1156. bs_stack_free(data);
  1157. }
  1158. }
  1159. }
  1160. break;
  1161. }
  1162. }
  1163. for (auto& param : oldDesc.textures)
  1164. {
  1165. auto iterFind = newDesc.textures.find(param.first);
  1166. if (iterFind == newDesc.textures.end())
  1167. continue;
  1168. UINT32 oldSlot = param.second.slot;
  1169. UINT32 newSlot = iterFind->second.slot;
  1170. bool isLoadStore = oldGpuParams->isLoadStoreTexture(oldSlot);
  1171. if (!isLoadStore)
  1172. newGpuParams->setTexture(newSlot, oldGpuParams->getTexture(oldSlot));
  1173. else
  1174. {
  1175. newGpuParams->setIsLoadStoreTexture(newSlot, true);
  1176. newGpuParams->setLoadStoreSurface(newSlot, oldGpuParams->getLoadStoreSurface(oldSlot));
  1177. }
  1178. }
  1179. for (auto& param : oldDesc.samplers)
  1180. {
  1181. auto iterFind = newDesc.samplers.find(param.first);
  1182. if (iterFind == newDesc.samplers.end())
  1183. continue;
  1184. newGpuParams->setSamplerState(iterFind->second.slot, oldGpuParams->getSamplerState(param.second.slot));
  1185. }
  1186. }
  1187. }
  1188. }
  1189. HMaterial Material::clone()
  1190. {
  1191. UINT32 bufferSize = 0;
  1192. MemorySerializer serializer;
  1193. UINT8* buffer = serializer.encode(this, bufferSize, (void*(*)(UINT32))&bs_alloc);
  1194. std::shared_ptr<Material> cloneObj = std::static_pointer_cast<Material>(serializer.decode(buffer, bufferSize));
  1195. bs_free(buffer);
  1196. return static_resource_cast<Material>(gResources()._createResourceHandle(cloneObj));
  1197. }
  1198. HMaterial Material::create()
  1199. {
  1200. MaterialPtr materialPtr = MaterialManager::instance().create();
  1201. return static_resource_cast<Material>(gResources()._createResourceHandle(materialPtr));
  1202. }
  1203. HMaterial Material::create(const HShader& shader)
  1204. {
  1205. MaterialPtr materialPtr = MaterialManager::instance().create(shader);
  1206. return static_resource_cast<Material>(gResources()._createResourceHandle(materialPtr));
  1207. }
  1208. RTTITypeBase* Material::getRTTIStatic()
  1209. {
  1210. return MaterialRTTI::instance();
  1211. }
  1212. RTTITypeBase* Material::getRTTI() const
  1213. {
  1214. return Material::getRTTIStatic();
  1215. }
  1216. }