2
0

processedShaderMaterial.cpp 49 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401
  1. //-----------------------------------------------------------------------------
  2. // Copyright (c) 2012 GarageGames, LLC
  3. //
  4. // Permission is hereby granted, free of charge, to any person obtaining a copy
  5. // of this software and associated documentation files (the "Software"), to
  6. // deal in the Software without restriction, including without limitation the
  7. // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
  8. // sell copies of the Software, and to permit persons to whom the Software is
  9. // furnished to do so, subject to the following conditions:
  10. //
  11. // The above copyright notice and this permission notice shall be included in
  12. // all copies or substantial portions of the Software.
  13. //
  14. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  19. // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  20. // IN THE SOFTWARE.
  21. //-----------------------------------------------------------------------------
  22. #include "platform/platform.h"
  23. #include "materials/processedShaderMaterial.h"
  24. #include "core/util/safeDelete.h"
  25. #include "gfx/sim/cubemapData.h"
  26. #include "gfx/gfxShader.h"
  27. #include "gfx/genericConstBuffer.h"
  28. #include "gfx/gfxPrimitiveBuffer.h"
  29. #include "scene/sceneRenderState.h"
  30. #include "shaderGen/shaderFeature.h"
  31. #include "shaderGen/shaderGenVars.h"
  32. #include "shaderGen/featureMgr.h"
  33. #include "shaderGen/shaderGen.h"
  34. #include "materials/sceneData.h"
  35. #include "materials/materialFeatureTypes.h"
  36. #include "materials/materialManager.h"
  37. #include "materials/shaderMaterialParameters.h"
  38. #include "materials/matTextureTarget.h"
  39. #include "gfx/util/screenspace.h"
  40. #include "math/util/matrixSet.h"
  41. // We need to include customMaterialDefinition for ShaderConstHandles::init
  42. #include "materials/customMaterialDefinition.h"
  43. ///
  44. /// ShaderConstHandles
  45. ///
  46. void ShaderConstHandles::init( GFXShader *shader, CustomMaterial* mat /*=NULL*/ )
  47. {
  48. mDiffuseColorSC = shader->getShaderConstHandle("$diffuseMaterialColor");
  49. mTexMatSC = shader->getShaderConstHandle(ShaderGenVars::texMat);
  50. mToneMapTexSC = shader->getShaderConstHandle(ShaderGenVars::toneMap);
  51. mSpecularColorSC = shader->getShaderConstHandle(ShaderGenVars::specularColor);
  52. mSpecularPowerSC = shader->getShaderConstHandle(ShaderGenVars::specularPower);
  53. mSpecularStrengthSC = shader->getShaderConstHandle(ShaderGenVars::specularStrength);
  54. mAccuScaleSC = shader->getShaderConstHandle("$accuScale");
  55. mAccuDirectionSC = shader->getShaderConstHandle("$accuDirection");
  56. mAccuStrengthSC = shader->getShaderConstHandle("$accuStrength");
  57. mAccuCoverageSC = shader->getShaderConstHandle("$accuCoverage");
  58. mAccuSpecularSC = shader->getShaderConstHandle("$accuSpecular");
  59. mParallaxInfoSC = shader->getShaderConstHandle("$parallaxInfo");
  60. mFogDataSC = shader->getShaderConstHandle(ShaderGenVars::fogData);
  61. mFogColorSC = shader->getShaderConstHandle(ShaderGenVars::fogColor);
  62. mDetailScaleSC = shader->getShaderConstHandle(ShaderGenVars::detailScale);
  63. mVisiblitySC = shader->getShaderConstHandle(ShaderGenVars::visibility);
  64. mColorMultiplySC = shader->getShaderConstHandle(ShaderGenVars::colorMultiply);
  65. mAlphaTestValueSC = shader->getShaderConstHandle(ShaderGenVars::alphaTestValue);
  66. mModelViewProjSC = shader->getShaderConstHandle(ShaderGenVars::modelview);
  67. mWorldViewOnlySC = shader->getShaderConstHandle(ShaderGenVars::worldViewOnly);
  68. mWorldToCameraSC = shader->getShaderConstHandle(ShaderGenVars::worldToCamera);
  69. mWorldToObjSC = shader->getShaderConstHandle(ShaderGenVars::worldToObj);
  70. mViewToObjSC = shader->getShaderConstHandle(ShaderGenVars::viewToObj);
  71. mCubeTransSC = shader->getShaderConstHandle(ShaderGenVars::cubeTrans);
  72. mObjTransSC = shader->getShaderConstHandle(ShaderGenVars::objTrans);
  73. mCubeEyePosSC = shader->getShaderConstHandle(ShaderGenVars::cubeEyePos);
  74. mEyePosSC = shader->getShaderConstHandle(ShaderGenVars::eyePos);
  75. mEyePosWorldSC = shader->getShaderConstHandle(ShaderGenVars::eyePosWorld);
  76. m_vEyeSC = shader->getShaderConstHandle(ShaderGenVars::vEye);
  77. mEyeMatSC = shader->getShaderConstHandle(ShaderGenVars::eyeMat);
  78. mOneOverFarplane = shader->getShaderConstHandle(ShaderGenVars::oneOverFarplane);
  79. mAccumTimeSC = shader->getShaderConstHandle(ShaderGenVars::accumTime);
  80. mMinnaertConstantSC = shader->getShaderConstHandle(ShaderGenVars::minnaertConstant);
  81. mSubSurfaceParamsSC = shader->getShaderConstHandle(ShaderGenVars::subSurfaceParams);
  82. mDiffuseAtlasParamsSC = shader->getShaderConstHandle(ShaderGenVars::diffuseAtlasParams);
  83. mDiffuseAtlasTileSC = shader->getShaderConstHandle(ShaderGenVars::diffuseAtlasTileParams);
  84. mBumpAtlasParamsSC = shader->getShaderConstHandle(ShaderGenVars::bumpAtlasParams);
  85. mBumpAtlasTileSC = shader->getShaderConstHandle(ShaderGenVars::bumpAtlasTileParams);
  86. mRTSizeSC = shader->getShaderConstHandle( "$targetSize" );
  87. mOneOverRTSizeSC = shader->getShaderConstHandle( "$oneOverTargetSize" );
  88. mDetailBumpStrength = shader->getShaderConstHandle( "$detailBumpStrength" );
  89. mViewProjSC = shader->getShaderConstHandle( "$viewProj" );
  90. // MFT_ImposterVert
  91. mImposterUVs = shader->getShaderConstHandle( "$imposterUVs" );
  92. mImposterLimits = shader->getShaderConstHandle( "$imposterLimits" );
  93. for (S32 i = 0; i < TEXTURE_STAGE_COUNT; ++i)
  94. mRTParamsSC[i] = shader->getShaderConstHandle( String::ToString( "$rtParams%d", i ) );
  95. // Clear any existing texture handles.
  96. dMemset( mTexHandlesSC, 0, sizeof( mTexHandlesSC ) );
  97. if(mat)
  98. {
  99. for (S32 i = 0; i < Material::MAX_TEX_PER_PASS; ++i)
  100. mTexHandlesSC[i] = shader->getShaderConstHandle(mat->mSamplerNames[i]);
  101. }
  102. }
  103. ///
  104. /// ShaderRenderPassData
  105. ///
  106. void ShaderRenderPassData::reset()
  107. {
  108. Parent::reset();
  109. shader = NULL;
  110. for ( U32 i=0; i < featureShaderHandles.size(); i++ )
  111. delete featureShaderHandles[i];
  112. featureShaderHandles.clear();
  113. }
  114. String ShaderRenderPassData::describeSelf() const
  115. {
  116. // First write the shader identification.
  117. String desc = String::ToString( "%s\n", shader->describeSelf().c_str() );
  118. // Let the parent get the rest.
  119. desc += Parent::describeSelf();
  120. return desc;
  121. }
  122. ///
  123. /// ProcessedShaderMaterial
  124. ///
  125. ProcessedShaderMaterial::ProcessedShaderMaterial()
  126. : mDefaultParameters( NULL ),
  127. mInstancingState( NULL )
  128. {
  129. VECTOR_SET_ASSOCIATION( mShaderConstDesc );
  130. VECTOR_SET_ASSOCIATION( mParameterHandles );
  131. }
  132. ProcessedShaderMaterial::ProcessedShaderMaterial(Material &mat)
  133. : mDefaultParameters( NULL ),
  134. mInstancingState( NULL )
  135. {
  136. VECTOR_SET_ASSOCIATION( mShaderConstDesc );
  137. VECTOR_SET_ASSOCIATION( mParameterHandles );
  138. mMaterial = &mat;
  139. }
  140. ProcessedShaderMaterial::~ProcessedShaderMaterial()
  141. {
  142. SAFE_DELETE(mInstancingState);
  143. SAFE_DELETE(mDefaultParameters);
  144. for (U32 i = 0; i < mParameterHandles.size(); i++)
  145. SAFE_DELETE(mParameterHandles[i]);
  146. }
  147. //
  148. // Material init
  149. //
  150. bool ProcessedShaderMaterial::init( const FeatureSet &features,
  151. const GFXVertexFormat *vertexFormat,
  152. const MatFeaturesDelegate &featuresDelegate )
  153. {
  154. // Load our textures
  155. _setStageData();
  156. // Determine how many stages we use
  157. mMaxStages = getNumStages();
  158. mVertexFormat = vertexFormat;
  159. mFeatures.clear();
  160. mStateHint.clear();
  161. SAFE_DELETE(mInstancingState);
  162. for( U32 i=0; i<mMaxStages; i++ )
  163. {
  164. MaterialFeatureData fd;
  165. // Determine the features of this stage
  166. _determineFeatures( i, fd, features );
  167. // Let the delegate poke at the features.
  168. if ( featuresDelegate )
  169. featuresDelegate( this, i, fd, features );
  170. // Create the passes for this stage
  171. if ( fd.features.isNotEmpty() )
  172. if( !_createPasses( fd, i, features ) )
  173. return false;
  174. }
  175. _initRenderPassDataStateBlocks();
  176. _initMaterialParameters();
  177. mDefaultParameters = allocMaterialParameters();
  178. setMaterialParameters( mDefaultParameters, 0 );
  179. mStateHint.init( this );
  180. // Enable instancing if we have it.
  181. if ( mFeatures.hasFeature( MFT_UseInstancing ) )
  182. {
  183. mInstancingState = new InstancingState();
  184. mInstancingState->setFormat( &_getRPD( 0 )->shader->mInstancingFormat, mVertexFormat );
  185. }
  186. // Check for a RenderTexTargetBin assignment
  187. // *IMPORTANT NOTE*
  188. // This is a temporary solution for getting diffuse mapping working with tex targets for standard materials
  189. // It should be removed once this is done properly, at that time the sAllowTextureTargetAssignment should also be removed
  190. // from Material (it is necessary for catching shadow maps/post effect this shouldn't be applied to)
  191. if (Material::sAllowTextureTargetAssignment)
  192. if (mMaterial && mMaterial->mDiffuseMapFilename[0].isNotEmpty() && mMaterial->mDiffuseMapFilename[0].substr( 0, 1 ).equal("#"))
  193. {
  194. String texTargetBufferName = mMaterial->mDiffuseMapFilename[0].substr(1, mMaterial->mDiffuseMapFilename[0].length() - 1);
  195. NamedTexTarget *texTarget = NamedTexTarget::find( texTargetBufferName );
  196. RenderPassData* rpd = getPass(0);
  197. if (rpd)
  198. {
  199. rpd->mTexSlot[0].texTarget = texTarget;
  200. rpd->mTexType[0] = Material::TexTarget;
  201. rpd->mSamplerNames[0] = "diffuseMap";
  202. }
  203. }
  204. return true;
  205. }
  206. U32 ProcessedShaderMaterial::getNumStages()
  207. {
  208. // Loops through all stages to determine how many
  209. // stages we actually use.
  210. //
  211. // The first stage is always active else we shouldn't be
  212. // creating the material to begin with.
  213. U32 numStages = 1;
  214. U32 i;
  215. for( i=1; i<Material::MAX_STAGES; i++ )
  216. {
  217. // Assume stage is inactive
  218. bool stageActive = false;
  219. // Cubemaps only on first stage
  220. if( i == 0 )
  221. {
  222. // If we have a cubemap the stage is active
  223. if( mMaterial->mCubemapData || mMaterial->mDynamicCubemap )
  224. {
  225. numStages++;
  226. continue;
  227. }
  228. }
  229. // If we have a texture for the a feature the
  230. // stage is active.
  231. if ( mStages[i].hasValidTex() )
  232. stageActive = true;
  233. // If this stage has specular lighting, it's active
  234. if ( mMaterial->mPixelSpecular[i] )
  235. stageActive = true;
  236. // If this stage has diffuse color, it's active
  237. if ( mMaterial->mDiffuse[i].alpha > 0 &&
  238. mMaterial->mDiffuse[i] != ColorF::WHITE )
  239. stageActive = true;
  240. // If we have a Material that is vertex lit
  241. // then it may not have a texture
  242. if( mMaterial->mVertLit[i] )
  243. stageActive = true;
  244. // Increment the number of active stages
  245. numStages += stageActive;
  246. }
  247. return numStages;
  248. }
  249. void ProcessedShaderMaterial::_determineFeatures( U32 stageNum,
  250. MaterialFeatureData &fd,
  251. const FeatureSet &features )
  252. {
  253. PROFILE_SCOPE( ProcessedShaderMaterial_DetermineFeatures );
  254. const F32 shaderVersion = GFX->getPixelShaderVersion();
  255. AssertFatal(shaderVersion > 0.0 , "Cannot create a shader material if we don't support shaders");
  256. bool lastStage = stageNum == (mMaxStages-1);
  257. // First we add all the features which the
  258. // material has defined.
  259. if ( mMaterial->isTranslucent() )
  260. {
  261. // Note: This is for decal blending into the prepass
  262. // for AL... it probably needs to be made clearer.
  263. if ( mMaterial->mTranslucentBlendOp == Material::LerpAlpha &&
  264. mMaterial->mTranslucentZWrite )
  265. fd.features.addFeature( MFT_IsTranslucentZWrite );
  266. else
  267. {
  268. fd.features.addFeature( MFT_IsTranslucent );
  269. fd.features.addFeature( MFT_ForwardShading );
  270. }
  271. }
  272. // TODO: This sort of sucks... BL should somehow force this
  273. // feature on from the outside and not this way.
  274. if ( dStrcmp( LIGHTMGR->getId(), "BLM" ) == 0 )
  275. fd.features.addFeature( MFT_ForwardShading );
  276. // Disabling the InterlacedPrePass feature for now. It is not ready for prime-time
  277. // and it should not be triggered off of the DoubleSided parameter. [2/5/2010 Pat]
  278. /*if ( mMaterial->isDoubleSided() )
  279. {
  280. fd.features.addFeature( MFT_InterlacedPrePass );
  281. }*/
  282. // Allow instancing if it was requested and the card supports
  283. // SM 3.0 or above.
  284. //
  285. // We also disable instancing for non-single pass materials
  286. // and glowing materials because its untested/unimplemented.
  287. //
  288. if ( features.hasFeature( MFT_UseInstancing ) &&
  289. mMaxStages == 1 &&
  290. !mMaterial->mGlow[0] &&
  291. !mMaterial->mDynamicCubemap &&
  292. shaderVersion >= 3.0f )
  293. fd.features.addFeature( MFT_UseInstancing );
  294. if ( mMaterial->mAlphaTest )
  295. fd.features.addFeature( MFT_AlphaTest );
  296. if ( mMaterial->mEmissive[stageNum] )
  297. fd.features.addFeature( MFT_IsEmissive );
  298. else
  299. fd.features.addFeature( MFT_RTLighting );
  300. if ( mMaterial->mAnimFlags[stageNum] )
  301. fd.features.addFeature( MFT_TexAnim );
  302. if ( mMaterial->mVertLit[stageNum] )
  303. fd.features.addFeature( MFT_VertLit );
  304. // cubemaps only available on stage 0 for now - bramage
  305. if ( stageNum < 1 &&
  306. ( ( mMaterial->mCubemapData && mMaterial->mCubemapData->mCubemap ) ||
  307. mMaterial->mDynamicCubemap ) )
  308. fd.features.addFeature( MFT_CubeMap );
  309. fd.features.addFeature( MFT_Visibility );
  310. if ( lastStage &&
  311. ( !gClientSceneGraph->usePostEffectFog() ||
  312. fd.features.hasFeature( MFT_IsTranslucent ) ||
  313. fd.features.hasFeature( MFT_ForwardShading )) )
  314. fd.features.addFeature( MFT_Fog );
  315. if ( mMaterial->mMinnaertConstant[stageNum] > 0.0f )
  316. fd.features.addFeature( MFT_MinnaertShading );
  317. if ( mMaterial->mSubSurface[stageNum] )
  318. fd.features.addFeature( MFT_SubSurface );
  319. if ( !mMaterial->mCellLayout[stageNum].isZero() )
  320. {
  321. fd.features.addFeature( MFT_DiffuseMapAtlas );
  322. if ( mMaterial->mNormalMapAtlas )
  323. fd.features.addFeature( MFT_NormalMapAtlas );
  324. }
  325. // Grab other features like normal maps, base texture, etc.
  326. FeatureSet mergeFeatures;
  327. mStages[stageNum].getFeatureSet( &mergeFeatures );
  328. fd.features.merge( mergeFeatures );
  329. if ( fd.features[ MFT_NormalMap ] )
  330. {
  331. if ( mStages[stageNum].getTex( MFT_NormalMap )->mFormat == GFXFormatDXT5 &&
  332. !mStages[stageNum].getTex( MFT_NormalMap )->mHasTransparency )
  333. fd.features.addFeature( MFT_IsDXTnm );
  334. }
  335. // Now for some more advanced features that we
  336. // cannot do on SM 2.0 and below.
  337. if ( shaderVersion > 2.0f )
  338. {
  339. // Only allow parallax if we have a normal map and
  340. // we're not using DXTnm compression.
  341. if ( mMaterial->mParallaxScale[stageNum] > 0.0f &&
  342. fd.features[ MFT_NormalMap ] &&
  343. !fd.features[ MFT_IsDXTnm ] )
  344. fd.features.addFeature( MFT_Parallax );
  345. // If not parallax then allow per-pixel specular if
  346. // we have real time lighting enabled.
  347. else if ( fd.features[MFT_RTLighting] &&
  348. mMaterial->mPixelSpecular[stageNum] )
  349. fd.features.addFeature( MFT_PixSpecular );
  350. }
  351. // Without realtime lighting and on lower end
  352. // shader models disable the specular map.
  353. if ( !fd.features[ MFT_RTLighting ] || shaderVersion == 2.0 )
  354. fd.features.removeFeature( MFT_SpecularMap );
  355. // If we have a specular map then make sure we
  356. // have per-pixel specular enabled.
  357. if( fd.features[ MFT_SpecularMap ] )
  358. {
  359. fd.features.addFeature( MFT_PixSpecular );
  360. // Check for an alpha channel on the specular map. If it has one (and it
  361. // has values less than 255) than the artist has put the gloss map into
  362. // the alpha channel.
  363. if( mStages[stageNum].getTex( MFT_SpecularMap )->mHasTransparency )
  364. fd.features.addFeature( MFT_GlossMap );
  365. }
  366. if ( mMaterial->mAccuEnabled[stageNum] )
  367. {
  368. fd.features.addFeature( MFT_AccuMap );
  369. mHasAccumulation = true;
  370. }
  371. // we need both diffuse and normal maps + sm3 to have an accu map
  372. if( fd.features[ MFT_AccuMap ] &&
  373. ( !fd.features[ MFT_DiffuseMap ] ||
  374. !fd.features[ MFT_NormalMap ] ||
  375. GFX->getPixelShaderVersion() < 3.0f ) ) {
  376. AssertWarn(false, "SAHARA: Using an Accu Map requires SM 3.0 and a normal map.");
  377. fd.features.removeFeature( MFT_AccuMap );
  378. mHasAccumulation = false;
  379. }
  380. // if we still have the AccuMap feature, we add all accu constant features
  381. if ( fd.features[ MFT_AccuMap ] ) {
  382. // add the dependencies of the accu map
  383. fd.features.addFeature( MFT_AccuScale );
  384. fd.features.addFeature( MFT_AccuDirection );
  385. fd.features.addFeature( MFT_AccuStrength );
  386. fd.features.addFeature( MFT_AccuCoverage );
  387. fd.features.addFeature( MFT_AccuSpecular );
  388. // now remove some features that are not compatible with this
  389. fd.features.removeFeature( MFT_UseInstancing );
  390. }
  391. // Without a base texture use the diffuse color
  392. // feature to ensure some sort of output.
  393. if (!fd.features[MFT_DiffuseMap])
  394. {
  395. fd.features.addFeature( MFT_DiffuseColor );
  396. // No texture coords... no overlay.
  397. fd.features.removeFeature( MFT_OverlayMap );
  398. }
  399. // If we have a diffuse map and the alpha on the diffuse isn't
  400. // zero and the color isn't pure white then multiply the color.
  401. else if ( mMaterial->mDiffuse[stageNum].alpha > 0.0f &&
  402. mMaterial->mDiffuse[stageNum] != ColorF::WHITE )
  403. fd.features.addFeature( MFT_DiffuseColor );
  404. // If lightmaps or tonemaps are enabled or we
  405. // don't have a second UV set then we cannot
  406. // use the overlay texture.
  407. if ( fd.features[MFT_LightMap] ||
  408. fd.features[MFT_ToneMap] ||
  409. mVertexFormat->getTexCoordCount() < 2 )
  410. fd.features.removeFeature( MFT_OverlayMap );
  411. // If tonemaps are enabled don't use lightmap
  412. if ( fd.features[MFT_ToneMap] || mVertexFormat->getTexCoordCount() < 2 )
  413. fd.features.removeFeature( MFT_LightMap );
  414. // Don't allow tonemaps if we don't have a second UV set
  415. if ( mVertexFormat->getTexCoordCount() < 2 )
  416. fd.features.removeFeature( MFT_ToneMap );
  417. // Always add the HDR output feature.
  418. //
  419. // It will be filtered out if it was disabled
  420. // for this material creation below.
  421. //
  422. // Also the shader code will evaluate to a nop
  423. // if HDR is not enabled in the scene.
  424. //
  425. fd.features.addFeature( MFT_HDROut );
  426. // If vertex color is enabled on the material's stage and
  427. // color is present in vertex format, add diffuse vertex
  428. // color feature.
  429. if ( mMaterial->mVertColor[ stageNum ] &&
  430. mVertexFormat->hasColor() )
  431. fd.features.addFeature( MFT_DiffuseVertColor );
  432. // Allow features to add themselves.
  433. for ( U32 i = 0; i < FEATUREMGR->getFeatureCount(); i++ )
  434. {
  435. const FeatureInfo &info = FEATUREMGR->getAt( i );
  436. info.feature->determineFeature( mMaterial,
  437. mVertexFormat,
  438. stageNum,
  439. *info.type,
  440. features,
  441. &fd );
  442. }
  443. // Now disable any features that were
  444. // not part of the input feature handle.
  445. fd.features.filter( features );
  446. }
  447. bool ProcessedShaderMaterial::_createPasses( MaterialFeatureData &stageFeatures, U32 stageNum, const FeatureSet &features )
  448. {
  449. // Creates passes for the given stage
  450. ShaderRenderPassData passData;
  451. U32 texIndex = 0;
  452. for( U32 i=0; i < FEATUREMGR->getFeatureCount(); i++ )
  453. {
  454. const FeatureInfo &info = FEATUREMGR->getAt( i );
  455. if ( !stageFeatures.features.hasFeature( *info.type ) )
  456. continue;
  457. U32 numTexReg = info.feature->getResources( stageFeatures ).numTexReg;
  458. // adds pass if blend op changes for feature
  459. _setPassBlendOp( info.feature, passData, texIndex, stageFeatures, stageNum, features );
  460. // Add pass if num tex reg is going to be too high
  461. if( passData.mNumTexReg + numTexReg > GFX->getNumSamplers() )
  462. {
  463. if( !_addPass( passData, texIndex, stageFeatures, stageNum, features ) )
  464. return false;
  465. _setPassBlendOp( info.feature, passData, texIndex, stageFeatures, stageNum, features );
  466. }
  467. passData.mNumTexReg += numTexReg;
  468. passData.mFeatureData.features.addFeature( *info.type );
  469. #if defined(TORQUE_DEBUG) && defined( TORQUE_OPENGL)
  470. U32 oldTexNumber = texIndex;
  471. #endif
  472. info.feature->setTexData( mStages[stageNum], stageFeatures, passData, texIndex );
  473. #if defined(TORQUE_DEBUG) && defined( TORQUE_OPENGL)
  474. if(oldTexNumber != texIndex)
  475. {
  476. for(int i = oldTexNumber; i < texIndex; i++)
  477. {
  478. AssertFatal(passData.mSamplerNames[ oldTexNumber ].isNotEmpty(), avar( "ERROR: ShaderGen feature %s don't set used sampler name", info.feature->getName().c_str()) );
  479. }
  480. }
  481. #endif
  482. // Add pass if tex units are maxed out
  483. if( texIndex > GFX->getNumSamplers() )
  484. {
  485. if( !_addPass( passData, texIndex, stageFeatures, stageNum, features ) )
  486. return false;
  487. _setPassBlendOp( info.feature, passData, texIndex, stageFeatures, stageNum, features );
  488. }
  489. }
  490. #if defined(TORQUE_DEBUG) && defined( TORQUE_OPENGL)
  491. for(int i = 0; i < texIndex; i++)
  492. {
  493. AssertFatal(passData.mSamplerNames[ i ].isNotEmpty(),"");
  494. }
  495. #endif
  496. const FeatureSet &passFeatures = passData.mFeatureData.codify();
  497. if ( passFeatures.isNotEmpty() )
  498. {
  499. mFeatures.merge( passFeatures );
  500. if( !_addPass( passData, texIndex, stageFeatures, stageNum, features ) )
  501. {
  502. mFeatures.clear();
  503. return false;
  504. }
  505. }
  506. return true;
  507. }
  508. void ProcessedShaderMaterial::_initMaterialParameters()
  509. {
  510. // Cleanup anything left first.
  511. SAFE_DELETE( mDefaultParameters );
  512. for ( U32 i = 0; i < mParameterHandles.size(); i++ )
  513. SAFE_DELETE( mParameterHandles[i] );
  514. // Gather the shaders as they all need to be
  515. // passed to the ShaderMaterialParameterHandles.
  516. Vector<GFXShader*> shaders;
  517. shaders.setSize( mPasses.size() );
  518. for ( U32 i = 0; i < mPasses.size(); i++ )
  519. shaders[i] = _getRPD(i)->shader;
  520. // Run through each shader and prepare its constants.
  521. for ( U32 i = 0; i < mPasses.size(); i++ )
  522. {
  523. const Vector<GFXShaderConstDesc>& desc = shaders[i]->getShaderConstDesc();
  524. Vector<GFXShaderConstDesc>::const_iterator p = desc.begin();
  525. for ( ; p != desc.end(); p++ )
  526. {
  527. // Add this to our list of shader constants
  528. GFXShaderConstDesc d(*p);
  529. mShaderConstDesc.push_back(d);
  530. ShaderMaterialParameterHandle* smph = new ShaderMaterialParameterHandle(d.name, shaders);
  531. mParameterHandles.push_back(smph);
  532. }
  533. }
  534. }
  535. bool ProcessedShaderMaterial::_addPass( ShaderRenderPassData &rpd,
  536. U32 &texIndex,
  537. MaterialFeatureData &fd,
  538. U32 stageNum,
  539. const FeatureSet &features )
  540. {
  541. // Set number of textures, stage, glow, etc.
  542. rpd.mNumTex = texIndex;
  543. rpd.mStageNum = stageNum;
  544. rpd.mGlow |= mMaterial->mGlow[stageNum];
  545. // Copy over features
  546. rpd.mFeatureData.materialFeatures = fd.features;
  547. Vector<String> samplers;
  548. samplers.setSize(Material::MAX_TEX_PER_PASS);
  549. for(int i = 0; i < Material::MAX_TEX_PER_PASS; ++i)
  550. {
  551. samplers[i] = (rpd.mSamplerNames[i].isEmpty() || rpd.mSamplerNames[i][0] == '$') ? rpd.mSamplerNames[i] : "$" + rpd.mSamplerNames[i];
  552. }
  553. // Generate shader
  554. GFXShader::setLogging( true, true );
  555. rpd.shader = SHADERGEN->getShader( rpd.mFeatureData, mVertexFormat, &mUserMacros, samplers );
  556. if( !rpd.shader )
  557. return false;
  558. rpd.shaderHandles.init( rpd.shader );
  559. // If a pass glows, we glow
  560. if( rpd.mGlow )
  561. mHasGlow = true;
  562. ShaderRenderPassData *newPass = new ShaderRenderPassData( rpd );
  563. mPasses.push_back( newPass );
  564. //initSamplerHandles
  565. ShaderConstHandles *handles = _getShaderConstHandles( mPasses.size()-1 );
  566. AssertFatal(handles,"");
  567. for(int i = 0; i < rpd.mNumTex; i++)
  568. {
  569. if(rpd.mSamplerNames[i].isEmpty())
  570. {
  571. handles->mTexHandlesSC[i] = newPass->shader->getShaderConstHandle( String::EmptyString );
  572. handles->mRTParamsSC[i] = newPass->shader->getShaderConstHandle( String::EmptyString );
  573. continue;
  574. }
  575. String samplerName = rpd.mSamplerNames[i];
  576. if( !samplerName.startsWith("$"))
  577. samplerName.insert(0, "$");
  578. GFXShaderConstHandle *handle = newPass->shader->getShaderConstHandle( samplerName );
  579. handles->mTexHandlesSC[i] = handle;
  580. handles->mRTParamsSC[i] = newPass->shader->getShaderConstHandle( String::ToString( "$rtParams%s", samplerName.c_str()+1 ) );
  581. AssertFatal( handle,"");
  582. }
  583. // Give each active feature a chance to create specialized shader consts.
  584. for( U32 i=0; i < FEATUREMGR->getFeatureCount(); i++ )
  585. {
  586. const FeatureInfo &info = FEATUREMGR->getAt( i );
  587. if ( !fd.features.hasFeature( *info.type ) )
  588. continue;
  589. ShaderFeatureConstHandles *fh = info.feature->createConstHandles( rpd.shader, mUserObject );
  590. if ( fh )
  591. newPass->featureShaderHandles.push_back( fh );
  592. }
  593. rpd.reset();
  594. texIndex = 0;
  595. return true;
  596. }
  597. void ProcessedShaderMaterial::_setPassBlendOp( ShaderFeature *sf,
  598. ShaderRenderPassData &passData,
  599. U32 &texIndex,
  600. MaterialFeatureData &stageFeatures,
  601. U32 stageNum,
  602. const FeatureSet &features )
  603. {
  604. if( sf->getBlendOp() == Material::None )
  605. {
  606. return;
  607. }
  608. // set up the current blend operation for multi-pass materials
  609. if( mPasses.size() > 0)
  610. {
  611. // If passData.numTexReg is 0, this is a brand new pass, so set the
  612. // blend operation to the first feature.
  613. if( passData.mNumTexReg == 0 )
  614. {
  615. passData.mBlendOp = sf->getBlendOp();
  616. }
  617. else
  618. {
  619. // numTegReg is more than zero, if this feature
  620. // doesn't have the same blend operation, then
  621. // we need to create yet another pass
  622. if( sf->getBlendOp() != passData.mBlendOp && mPasses[mPasses.size()-1]->mStageNum == stageNum)
  623. {
  624. _addPass( passData, texIndex, stageFeatures, stageNum, features );
  625. passData.mBlendOp = sf->getBlendOp();
  626. }
  627. }
  628. }
  629. }
  630. //
  631. // Runtime / rendering
  632. //
  633. bool ProcessedShaderMaterial::setupPass( SceneRenderState *state, const SceneData &sgData, U32 pass )
  634. {
  635. PROFILE_SCOPE( ProcessedShaderMaterial_SetupPass );
  636. // Make sure we have the pass
  637. if(pass >= mPasses.size())
  638. {
  639. // If we were rendering instanced data tell
  640. // the device to reset that vb stream.
  641. if ( mInstancingState )
  642. GFX->setVertexBuffer( NULL, 1 );
  643. return false;
  644. }
  645. _setRenderState( state, sgData, pass );
  646. // Set shaders
  647. ShaderRenderPassData* rpd = _getRPD(pass);
  648. if( rpd->shader )
  649. {
  650. GFX->setShader( rpd->shader );
  651. GFX->setShaderConstBuffer(_getShaderConstBuffer(pass));
  652. _setShaderConstants(state, sgData, pass);
  653. // If we're instancing then do the initial step to get
  654. // set the vb pointer to the const buffer.
  655. if ( mInstancingState )
  656. stepInstance();
  657. }
  658. else
  659. {
  660. GFX->setupGenericShaders();
  661. GFX->setShaderConstBuffer(NULL);
  662. }
  663. // Set our textures
  664. setTextureStages( state, sgData, pass );
  665. _setTextureTransforms(pass);
  666. return true;
  667. }
  668. void ProcessedShaderMaterial::setTextureStages( SceneRenderState *state, const SceneData &sgData, U32 pass )
  669. {
  670. PROFILE_SCOPE( ProcessedShaderMaterial_SetTextureStages );
  671. ShaderConstHandles *handles = _getShaderConstHandles(pass);
  672. AssertFatal(handles,"");
  673. // Set all of the textures we need to render the give pass.
  674. #ifdef TORQUE_DEBUG
  675. AssertFatal( pass<mPasses.size(), "Pass out of bounds" );
  676. #endif
  677. RenderPassData *rpd = mPasses[pass];
  678. GFXShaderConstBuffer* shaderConsts = _getShaderConstBuffer(pass);
  679. NamedTexTarget *texTarget;
  680. GFXTextureObject *texObject;
  681. for( U32 i=0; i<rpd->mNumTex; i++ )
  682. {
  683. U32 currTexFlag = rpd->mTexType[i];
  684. if (!LIGHTMGR || !LIGHTMGR->setTextureStage(sgData, currTexFlag, i, shaderConsts, handles))
  685. {
  686. switch( currTexFlag )
  687. {
  688. // If the flag is unset then assume its just
  689. // a regular texture to set... nothing special.
  690. case 0:
  691. default:
  692. GFX->setTexture(i, rpd->mTexSlot[i].texObject);
  693. break;
  694. case Material::NormalizeCube:
  695. GFX->setCubeTexture(i, Material::GetNormalizeCube());
  696. break;
  697. case Material::Lightmap:
  698. GFX->setTexture( i, sgData.lightmap );
  699. break;
  700. case Material::ToneMapTex:
  701. shaderConsts->setSafe(handles->mToneMapTexSC, (S32)i);
  702. GFX->setTexture(i, rpd->mTexSlot[i].texObject);
  703. break;
  704. case Material::Cube:
  705. GFX->setCubeTexture( i, rpd->mCubeMap );
  706. break;
  707. case Material::SGCube:
  708. GFX->setCubeTexture( i, sgData.cubemap );
  709. break;
  710. case Material::BackBuff:
  711. GFX->setTexture( i, sgData.backBuffTex );
  712. break;
  713. case Material::AccuMap:
  714. if ( sgData.accuTex )
  715. GFX->setTexture( i, sgData.accuTex );
  716. else
  717. GFX->setTexture( i, GFXTexHandle::ZERO );
  718. break;
  719. case Material::TexTarget:
  720. {
  721. texTarget = rpd->mTexSlot[i].texTarget;
  722. if ( !texTarget )
  723. {
  724. GFX->setTexture( i, NULL );
  725. break;
  726. }
  727. texObject = texTarget->getTexture();
  728. // If no texture is available then map the default 2x2
  729. // black texture to it. This at least will ensure that
  730. // we get consistant behavior across GPUs and platforms.
  731. if ( !texObject )
  732. texObject = GFXTexHandle::ZERO;
  733. if ( handles->mRTParamsSC[i]->isValid() && texObject )
  734. {
  735. const Point3I &targetSz = texObject->getSize();
  736. const RectI &targetVp = texTarget->getViewport();
  737. Point4F rtParams;
  738. ScreenSpace::RenderTargetParameters(targetSz, targetVp, rtParams);
  739. shaderConsts->set(handles->mRTParamsSC[i], rtParams);
  740. }
  741. GFX->setTexture( i, texObject );
  742. break;
  743. }
  744. }
  745. }
  746. }
  747. }
  748. void ProcessedShaderMaterial::_setTextureTransforms(const U32 pass)
  749. {
  750. PROFILE_SCOPE( ProcessedShaderMaterial_SetTextureTransforms );
  751. ShaderConstHandles* handles = _getShaderConstHandles(pass);
  752. if (handles->mTexMatSC->isValid())
  753. {
  754. MatrixF texMat( true );
  755. mMaterial->updateTimeBasedParams();
  756. F32 waveOffset = _getWaveOffset( pass ); // offset is between 0.0 and 1.0
  757. // handle scroll anim type
  758. if( mMaterial->mAnimFlags[pass] & Material::Scroll )
  759. {
  760. if( mMaterial->mAnimFlags[pass] & Material::Wave )
  761. {
  762. Point3F scrollOffset;
  763. scrollOffset.x = mMaterial->mScrollDir[pass].x * waveOffset;
  764. scrollOffset.y = mMaterial->mScrollDir[pass].y * waveOffset;
  765. scrollOffset.z = 1.0;
  766. texMat.setColumn( 3, scrollOffset );
  767. }
  768. else
  769. {
  770. Point3F offset( mMaterial->mScrollOffset[pass].x,
  771. mMaterial->mScrollOffset[pass].y,
  772. 1.0 );
  773. texMat.setColumn( 3, offset );
  774. }
  775. }
  776. // handle rotation
  777. if( mMaterial->mAnimFlags[pass] & Material::Rotate )
  778. {
  779. if( mMaterial->mAnimFlags[pass] & Material::Wave )
  780. {
  781. F32 rotPos = waveOffset * M_2PI;
  782. texMat.set( EulerF( 0.0, 0.0, rotPos ) );
  783. texMat.setColumn( 3, Point3F( 0.5, 0.5, 0.0 ) );
  784. MatrixF test( true );
  785. test.setColumn( 3, Point3F( mMaterial->mRotPivotOffset[pass].x,
  786. mMaterial->mRotPivotOffset[pass].y,
  787. 0.0 ) );
  788. texMat.mul( test );
  789. }
  790. else
  791. {
  792. texMat.set( EulerF( 0.0, 0.0, mMaterial->mRotPos[pass] ) );
  793. texMat.setColumn( 3, Point3F( 0.5, 0.5, 0.0 ) );
  794. MatrixF test( true );
  795. test.setColumn( 3, Point3F( mMaterial->mRotPivotOffset[pass].x,
  796. mMaterial->mRotPivotOffset[pass].y,
  797. 0.0 ) );
  798. texMat.mul( test );
  799. }
  800. }
  801. // Handle scale + wave offset
  802. if( mMaterial->mAnimFlags[pass] & Material::Scale &&
  803. mMaterial->mAnimFlags[pass] & Material::Wave )
  804. {
  805. F32 wOffset = fabs( waveOffset );
  806. texMat.setColumn( 3, Point3F( 0.5, 0.5, 0.0 ) );
  807. MatrixF temp( true );
  808. temp.setRow( 0, Point3F( wOffset, 0.0, 0.0 ) );
  809. temp.setRow( 1, Point3F( 0.0, wOffset, 0.0 ) );
  810. temp.setRow( 2, Point3F( 0.0, 0.0, wOffset ) );
  811. temp.setColumn( 3, Point3F( -wOffset * 0.5, -wOffset * 0.5, 0.0 ) );
  812. texMat.mul( temp );
  813. }
  814. // handle sequence
  815. if( mMaterial->mAnimFlags[pass] & Material::Sequence )
  816. {
  817. U32 frameNum = (U32)(MATMGR->getTotalTime() * mMaterial->mSeqFramePerSec[pass]);
  818. F32 offset = frameNum * mMaterial->mSeqSegSize[pass];
  819. if ( mMaterial->mAnimFlags[pass] & Material::Scale )
  820. texMat.scale( Point3F( mMaterial->mSeqSegSize[pass], 1.0f, 1.0f ) );
  821. Point3F texOffset = texMat.getPosition();
  822. texOffset.x += offset;
  823. texMat.setPosition( texOffset );
  824. }
  825. GFXShaderConstBuffer* shaderConsts = _getShaderConstBuffer(pass);
  826. shaderConsts->setSafe(handles->mTexMatSC, texMat);
  827. }
  828. }
  829. //--------------------------------------------------------------------------
  830. // Get wave offset for texture animations using a wave transform
  831. //--------------------------------------------------------------------------
  832. F32 ProcessedShaderMaterial::_getWaveOffset( U32 stage )
  833. {
  834. switch( mMaterial->mWaveType[stage] )
  835. {
  836. case Material::Sin:
  837. {
  838. return mMaterial->mWaveAmp[stage] * mSin( M_2PI * mMaterial->mWavePos[stage] );
  839. break;
  840. }
  841. case Material::Triangle:
  842. {
  843. F32 frac = mMaterial->mWavePos[stage] - mFloor( mMaterial->mWavePos[stage] );
  844. if( frac > 0.0 && frac <= 0.25 )
  845. {
  846. return mMaterial->mWaveAmp[stage] * frac * 4.0;
  847. }
  848. if( frac > 0.25 && frac <= 0.5 )
  849. {
  850. return mMaterial->mWaveAmp[stage] * ( 1.0 - ((frac-0.25)*4.0) );
  851. }
  852. if( frac > 0.5 && frac <= 0.75 )
  853. {
  854. return mMaterial->mWaveAmp[stage] * (frac-0.5) * -4.0;
  855. }
  856. if( frac > 0.75 && frac <= 1.0 )
  857. {
  858. return -mMaterial->mWaveAmp[stage] * ( 1.0 - ((frac-0.75)*4.0) );
  859. }
  860. break;
  861. }
  862. case Material::Square:
  863. {
  864. F32 frac = mMaterial->mWavePos[stage] - mFloor( mMaterial->mWavePos[stage] );
  865. if( frac > 0.0 && frac <= 0.5 )
  866. {
  867. return 0.0;
  868. }
  869. else
  870. {
  871. return mMaterial->mWaveAmp[stage];
  872. }
  873. break;
  874. }
  875. }
  876. return 0.0;
  877. }
  878. void ProcessedShaderMaterial::_setShaderConstants(SceneRenderState * state, const SceneData &sgData, U32 pass)
  879. {
  880. PROFILE_SCOPE( ProcessedShaderMaterial_SetShaderConstants );
  881. GFXShaderConstBuffer* shaderConsts = _getShaderConstBuffer(pass);
  882. ShaderConstHandles* handles = _getShaderConstHandles(pass);
  883. U32 stageNum = getStageFromPass(pass);
  884. // First we do all the constants which are not
  885. // controlled via the material... we have to
  886. // set these all the time as they could change.
  887. if ( handles->mFogDataSC->isValid() )
  888. {
  889. Point3F fogData;
  890. fogData.x = sgData.fogDensity;
  891. fogData.y = sgData.fogDensityOffset;
  892. fogData.z = sgData.fogHeightFalloff;
  893. shaderConsts->set( handles->mFogDataSC, fogData );
  894. }
  895. shaderConsts->setSafe(handles->mFogColorSC, sgData.fogColor);
  896. if( handles->mOneOverFarplane->isValid() )
  897. {
  898. const F32 &invfp = 1.0f / state->getFarPlane();
  899. Point4F oneOverFP(invfp, invfp, invfp, invfp);
  900. shaderConsts->set( handles->mOneOverFarplane, oneOverFP );
  901. }
  902. shaderConsts->setSafe( handles->mAccumTimeSC, MATMGR->getTotalTime() );
  903. // If the shader constants have not been lost then
  904. // they contain the content from a previous render pass.
  905. //
  906. // In this case we can skip updating the material constants
  907. // which do not change frame to frame.
  908. //
  909. // NOTE: This assumes we're not animating material parameters
  910. // in a way that doesn't cause a shader reload... this isn't
  911. // being done now, but it could change in the future.
  912. //
  913. if ( !shaderConsts->wasLost() )
  914. return;
  915. shaderConsts->setSafe(handles->mSpecularColorSC, mMaterial->mSpecular[stageNum]);
  916. shaderConsts->setSafe(handles->mSpecularPowerSC, mMaterial->mSpecularPower[stageNum]);
  917. shaderConsts->setSafe(handles->mSpecularStrengthSC, mMaterial->mSpecularStrength[stageNum]);
  918. shaderConsts->setSafe(handles->mParallaxInfoSC, mMaterial->mParallaxScale[stageNum]);
  919. shaderConsts->setSafe(handles->mMinnaertConstantSC, mMaterial->mMinnaertConstant[stageNum]);
  920. if ( handles->mSubSurfaceParamsSC->isValid() )
  921. {
  922. Point4F subSurfParams;
  923. dMemcpy( &subSurfParams, &mMaterial->mSubSurfaceColor[stageNum], sizeof(ColorF) );
  924. subSurfParams.w = mMaterial->mSubSurfaceRolloff[stageNum];
  925. shaderConsts->set(handles->mSubSurfaceParamsSC, subSurfParams);
  926. }
  927. if ( handles->mRTSizeSC->isValid() )
  928. {
  929. const Point2I &resolution = GFX->getActiveRenderTarget()->getSize();
  930. Point2F pixelShaderConstantData;
  931. pixelShaderConstantData.x = resolution.x;
  932. pixelShaderConstantData.y = resolution.y;
  933. shaderConsts->set( handles->mRTSizeSC, pixelShaderConstantData );
  934. }
  935. if ( handles->mOneOverRTSizeSC->isValid() )
  936. {
  937. const Point2I &resolution = GFX->getActiveRenderTarget()->getSize();
  938. Point2F oneOverTargetSize( 1.0f / (F32)resolution.x, 1.0f / (F32)resolution.y );
  939. shaderConsts->set( handles->mOneOverRTSizeSC, oneOverTargetSize );
  940. }
  941. // set detail scale
  942. shaderConsts->setSafe(handles->mDetailScaleSC, mMaterial->mDetailScale[stageNum]);
  943. shaderConsts->setSafe(handles->mDetailBumpStrength, mMaterial->mDetailNormalMapStrength[stageNum]);
  944. // MFT_ImposterVert
  945. if ( handles->mImposterUVs->isValid() )
  946. {
  947. U32 uvCount = getMin( mMaterial->mImposterUVs.size(), 64 ); // See imposter.hlsl
  948. AlignedArray<Point4F> imposterUVs( uvCount, sizeof( Point4F ), (U8*)mMaterial->mImposterUVs.address(), false );
  949. shaderConsts->set( handles->mImposterUVs, imposterUVs );
  950. }
  951. shaderConsts->setSafe( handles->mImposterLimits, mMaterial->mImposterLimits );
  952. // Diffuse
  953. shaderConsts->setSafe(handles->mDiffuseColorSC, mMaterial->mDiffuse[stageNum]);
  954. shaderConsts->setSafe( handles->mAlphaTestValueSC, mClampF( (F32)mMaterial->mAlphaRef / 255.0f, 0.0f, 1.0f ) );
  955. if(handles->mDiffuseAtlasParamsSC)
  956. {
  957. Point4F atlasParams(1.0f / mMaterial->mCellLayout[stageNum].x, // 1 / num_horizontal
  958. 1.0f / mMaterial->mCellLayout[stageNum].y, // 1 / num_vertical
  959. mMaterial->mCellSize[stageNum], // tile size in pixels
  960. getBinLog2(mMaterial->mCellSize[stageNum]) ); // pow of 2 of tile size in pixels 2^9 = 512, 2^10=1024 etc
  961. shaderConsts->setSafe(handles->mDiffuseAtlasParamsSC, atlasParams);
  962. }
  963. if(handles->mBumpAtlasParamsSC)
  964. {
  965. Point4F atlasParams(1.0f / mMaterial->mCellLayout[stageNum].x, // 1 / num_horizontal
  966. 1.0f / mMaterial->mCellLayout[stageNum].y, // 1 / num_vertical
  967. mMaterial->mCellSize[stageNum], // tile size in pixels
  968. getBinLog2(mMaterial->mCellSize[stageNum]) ); // pow of 2 of tile size in pixels 2^9 = 512, 2^10=1024 etc
  969. shaderConsts->setSafe(handles->mBumpAtlasParamsSC, atlasParams);
  970. }
  971. if(handles->mDiffuseAtlasTileSC)
  972. {
  973. // Sanity check the wrap flags
  974. //AssertWarn(mMaterial->mTextureAddressModeU == mMaterial->mTextureAddressModeV, "Addresing mode mismatch, texture atlasing will be confused");
  975. Point4F atlasTileParams( mMaterial->mCellIndex[stageNum].x, // Tile co-ordinate, ie: [0, 3]
  976. mMaterial->mCellIndex[stageNum].y,
  977. 0.0f, 0.0f ); // TODO: Wrap mode flags?
  978. shaderConsts->setSafe(handles->mDiffuseAtlasTileSC, atlasTileParams);
  979. }
  980. if(handles->mBumpAtlasTileSC)
  981. {
  982. // Sanity check the wrap flags
  983. //AssertWarn(mMaterial->mTextureAddressModeU == mMaterial->mTextureAddressModeV, "Addresing mode mismatch, texture atlasing will be confused");
  984. Point4F atlasTileParams( mMaterial->mCellIndex[stageNum].x, // Tile co-ordinate, ie: [0, 3]
  985. mMaterial->mCellIndex[stageNum].y,
  986. 0.0f, 0.0f ); // TODO: Wrap mode flags?
  987. shaderConsts->setSafe(handles->mBumpAtlasTileSC, atlasTileParams);
  988. }
  989. if( handles->mAccuScaleSC->isValid() )
  990. shaderConsts->set( handles->mAccuScaleSC, mMaterial->mAccuScale[stageNum] );
  991. if( handles->mAccuDirectionSC->isValid() )
  992. shaderConsts->set( handles->mAccuDirectionSC, mMaterial->mAccuDirection[stageNum] );
  993. if( handles->mAccuStrengthSC->isValid() )
  994. shaderConsts->set( handles->mAccuStrengthSC, mMaterial->mAccuStrength[stageNum] );
  995. if( handles->mAccuCoverageSC->isValid() )
  996. shaderConsts->set( handles->mAccuCoverageSC, mMaterial->mAccuCoverage[stageNum] );
  997. if( handles->mAccuSpecularSC->isValid() )
  998. shaderConsts->set( handles->mAccuSpecularSC, mMaterial->mAccuSpecular[stageNum] );
  999. }
  1000. bool ProcessedShaderMaterial::_hasCubemap(U32 pass)
  1001. {
  1002. // Only support cubemap on the first stage
  1003. if( mPasses[pass]->mStageNum > 0 )
  1004. return false;
  1005. if( mPasses[pass]->mCubeMap )
  1006. return true;
  1007. return false;
  1008. }
  1009. void ProcessedShaderMaterial::setTransforms(const MatrixSet &matrixSet, SceneRenderState *state, const U32 pass)
  1010. {
  1011. PROFILE_SCOPE( ProcessedShaderMaterial_setTransforms );
  1012. GFXShaderConstBuffer* shaderConsts = _getShaderConstBuffer(pass);
  1013. ShaderConstHandles* handles = _getShaderConstHandles(pass);
  1014. // The MatrixSet will lazily generate a matrix under the
  1015. // various 'get' methods, so inline the test for a valid
  1016. // shader constant handle to avoid that work when we can.
  1017. if ( handles->mModelViewProjSC->isValid() )
  1018. shaderConsts->set( handles->mModelViewProjSC, matrixSet.getWorldViewProjection() );
  1019. if ( handles->mObjTransSC->isValid() )
  1020. shaderConsts->set( handles->mObjTransSC, matrixSet.getObjectToWorld() );
  1021. if ( handles->mWorldToObjSC->isValid() )
  1022. shaderConsts->set( handles->mWorldToObjSC, matrixSet.getWorldToObject() );
  1023. if ( handles->mWorldToCameraSC->isValid() )
  1024. shaderConsts->set( handles->mWorldToCameraSC, matrixSet.getWorldToCamera() );
  1025. if ( handles->mWorldViewOnlySC->isValid() )
  1026. shaderConsts->set( handles->mWorldViewOnlySC, matrixSet.getObjectToCamera() );
  1027. if ( handles->mViewToObjSC->isValid() )
  1028. shaderConsts->set( handles->mViewToObjSC, matrixSet.getCameraToObject() );
  1029. if ( handles->mViewProjSC->isValid() )
  1030. shaderConsts->set( handles->mViewProjSC, matrixSet.getWorldToScreen() );
  1031. if ( handles->mCubeTransSC->isValid() &&
  1032. ( _hasCubemap(pass) || mMaterial->mDynamicCubemap ) )
  1033. {
  1034. // TODO: Could we not remove this constant? Use mObjTransSC and cast to float3x3 instead?
  1035. shaderConsts->set(handles->mCubeTransSC, matrixSet.getObjectToWorld(), GFXSCT_Float3x3);
  1036. }
  1037. if ( handles->m_vEyeSC->isValid() )
  1038. shaderConsts->set( handles->m_vEyeSC, state->getVectorEye() );
  1039. }
  1040. void ProcessedShaderMaterial::setSceneInfo(SceneRenderState * state, const SceneData& sgData, U32 pass)
  1041. {
  1042. PROFILE_SCOPE( ProcessedShaderMaterial_setSceneInfo );
  1043. GFXShaderConstBuffer* shaderConsts = _getShaderConstBuffer(pass);
  1044. ShaderConstHandles* handles = _getShaderConstHandles(pass);
  1045. // Set cubemap stuff here (it's convenient!)
  1046. const Point3F &eyePosWorld = state->getCameraPosition();
  1047. if ( handles->mCubeEyePosSC->isValid() )
  1048. {
  1049. if(_hasCubemap(pass) || mMaterial->mDynamicCubemap)
  1050. {
  1051. Point3F cubeEyePos = eyePosWorld - sgData.objTrans->getPosition();
  1052. shaderConsts->set(handles->mCubeEyePosSC, cubeEyePos);
  1053. }
  1054. }
  1055. shaderConsts->setSafe(handles->mVisiblitySC, sgData.visibility);
  1056. shaderConsts->setSafe(handles->mEyePosWorldSC, eyePosWorld);
  1057. if ( handles->mEyePosSC->isValid() )
  1058. {
  1059. MatrixF tempMat( *sgData.objTrans );
  1060. tempMat.inverse();
  1061. Point3F eyepos;
  1062. tempMat.mulP( eyePosWorld, &eyepos );
  1063. shaderConsts->set(handles->mEyePosSC, eyepos);
  1064. }
  1065. shaderConsts->setSafe(handles->mEyeMatSC, state->getCameraTransform());
  1066. ShaderRenderPassData *rpd = _getRPD( pass );
  1067. for ( U32 i=0; i < rpd->featureShaderHandles.size(); i++ )
  1068. rpd->featureShaderHandles[i]->setConsts( state, sgData, shaderConsts );
  1069. LIGHTMGR->setLightInfo( this, mMaterial, sgData, state, pass, shaderConsts );
  1070. }
  1071. void ProcessedShaderMaterial::setBuffers( GFXVertexBufferHandleBase *vertBuffer, GFXPrimitiveBufferHandle *primBuffer )
  1072. {
  1073. PROFILE_SCOPE(ProcessedShaderMaterial_setBuffers);
  1074. // If we're not instanced then just call the parent.
  1075. if ( !mInstancingState )
  1076. {
  1077. Parent::setBuffers( vertBuffer, primBuffer );
  1078. return;
  1079. }
  1080. PROFILE_SCOPE(ProcessedShaderMaterial_setBuffers_instancing);
  1081. const S32 instCount = mInstancingState->getCount();
  1082. AssertFatal( instCount > 0,
  1083. "ProcessedShaderMaterial::setBuffers - No instances rendered!" );
  1084. // Nothing special here.
  1085. GFX->setPrimitiveBuffer( *primBuffer );
  1086. // Set the first stream the the normal VB and set the
  1087. // correct frequency for the number of instances to render.
  1088. GFX->setVertexBuffer( *vertBuffer, 0, instCount );
  1089. // Get a volatile VB and fill it with the vertex data.
  1090. const GFXVertexFormat *instFormat = mInstancingState->getFormat();
  1091. GFXVertexBufferDataHandle instVB;
  1092. instVB.set( GFX, instFormat->getSizeInBytes(), instFormat, instCount, GFXBufferTypeVolatile );
  1093. U8 *dest = instVB.lock();
  1094. if(!dest) return;
  1095. dMemcpy( dest, mInstancingState->getBuffer(), instFormat->getSizeInBytes() * instCount );
  1096. instVB.unlock();
  1097. // Set the instance vb for streaming.
  1098. GFX->setVertexBuffer( instVB, 1, 1 );
  1099. // Finally set the vertex format which defines
  1100. // both of the streams.
  1101. GFX->setVertexFormat( mInstancingState->getDeclFormat() );
  1102. // Done... reset the count.
  1103. mInstancingState->resetStep();
  1104. }
  1105. bool ProcessedShaderMaterial::stepInstance()
  1106. {
  1107. PROFILE_SCOPE(ProcessedShaderMaterial_stepInstance);
  1108. AssertFatal( mInstancingState, "ProcessedShaderMaterial::stepInstance - This material isn't instanced!" );
  1109. return mInstancingState->step( &_getShaderConstBuffer( 0 )->mInstPtr );
  1110. }
  1111. MaterialParameters* ProcessedShaderMaterial::allocMaterialParameters()
  1112. {
  1113. ShaderMaterialParameters* smp = new ShaderMaterialParameters();
  1114. Vector<GFXShaderConstBufferRef> buffers( __FILE__, __LINE__ );
  1115. buffers.setSize(mPasses.size());
  1116. for (U32 i = 0; i < mPasses.size(); i++)
  1117. buffers[i] = _getRPD(i)->shader->allocConstBuffer();
  1118. // smp now owns these buffers.
  1119. smp->setBuffers(mShaderConstDesc, buffers);
  1120. return smp;
  1121. }
  1122. MaterialParameterHandle* ProcessedShaderMaterial::getMaterialParameterHandle(const String& name)
  1123. {
  1124. // Search our list
  1125. for (U32 i = 0; i < mParameterHandles.size(); i++)
  1126. {
  1127. if (mParameterHandles[i]->getName().equal(name))
  1128. return mParameterHandles[i];
  1129. }
  1130. // If we didn't find it, we have to add it to support shader reloading.
  1131. Vector<GFXShader*> shaders;
  1132. shaders.setSize(mPasses.size());
  1133. for (U32 i = 0; i < mPasses.size(); i++)
  1134. shaders[i] = _getRPD(i)->shader;
  1135. ShaderMaterialParameterHandle* smph = new ShaderMaterialParameterHandle( name, shaders );
  1136. mParameterHandles.push_back(smph);
  1137. return smph;
  1138. }
  1139. /// This is here to deal with the differences between ProcessedCustomMaterials and ProcessedShaderMaterials.
  1140. GFXShaderConstBuffer* ProcessedShaderMaterial::_getShaderConstBuffer( const U32 pass )
  1141. {
  1142. if (mCurrentParams && pass < mPasses.size())
  1143. {
  1144. return static_cast<ShaderMaterialParameters*>(mCurrentParams)->getBuffer(pass);
  1145. }
  1146. return NULL;
  1147. }
  1148. ShaderConstHandles* ProcessedShaderMaterial::_getShaderConstHandles(const U32 pass)
  1149. {
  1150. if (pass < mPasses.size())
  1151. {
  1152. return &_getRPD(pass)->shaderHandles;
  1153. }
  1154. return NULL;
  1155. }
  1156. void ProcessedShaderMaterial::dumpMaterialInfo()
  1157. {
  1158. for ( U32 i = 0; i < getNumPasses(); i++ )
  1159. {
  1160. const ShaderRenderPassData *passData = _getRPD( i );
  1161. if ( passData == NULL )
  1162. continue;
  1163. const GFXShader *shader = passData->shader;
  1164. if ( shader == NULL )
  1165. Con::printf( " [%i] [NULL shader]", i );
  1166. else
  1167. Con::printf( " [%i] %s", i, shader->describeSelf().c_str() );
  1168. }
  1169. }