PolyGLSLShaderModule.cpp 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870
  1. /*
  2. Copyright (C) 2011 by Ivan Safrin
  3. Permission is hereby granted, free of charge, to any person obtaining a copy
  4. of this software and associated documentation files (the "Software"), to deal
  5. in the Software without restriction, including without limitation the rights
  6. to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  7. copies of the Software, and to permit persons to whom the Software is
  8. furnished to do so, subject to the following conditions:
  9. The above copyright notice and this permission notice shall be included in
  10. all copies or substantial portions of the Software.
  11. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  12. IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  13. FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  14. AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  15. LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  16. OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  17. THE SOFTWARE.
  18. */
  19. #ifdef _WINDOWS
  20. #include <windows.h>
  21. #endif
  22. #include "PolyGLHeaders.h"
  23. #include "PolyGLSLShaderModule.h"
  24. #include "PolyCoreServices.h"
  25. #include "PolyResourceManager.h"
  26. #include "PolyRenderer.h"
  27. #include "PolyGLSLProgram.h"
  28. #include "PolyGLSLShader.h"
  29. #include "PolyGLCubemap.h"
  30. #include "PolyMaterial.h"
  31. #include "PolyGLTexture.h"
  32. #include "tinyxml.h"
  33. using std::vector;
  34. using namespace Polycode;
  35. #if defined(_WINDOWS) && !defined(_MINGW)
  36. PFNGLUSEPROGRAMPROC glUseProgram;
  37. PFNGLUNIFORM1IPROC glUniform1i;
  38. PFNGLUNIFORM1FPROC glUniform1f;
  39. PFNGLUNIFORM2FPROC glUniform2f;
  40. PFNGLUNIFORM3FPROC glUniform3f;
  41. extern PFNGLACTIVETEXTUREPROC glActiveTexture;
  42. PFNGLCREATESHADERPROC glCreateShader;
  43. PFNGLSHADERSOURCEPROC glShaderSource;
  44. PFNGLCOMPILESHADERPROC glCompileShader;
  45. PFNGLCREATEPROGRAMPROC glCreateProgram;
  46. PFNGLATTACHSHADERPROC glAttachShader;
  47. PFNGLLINKPROGRAMPROC glLinkProgram;
  48. PFNGLDETACHSHADERPROC glDetachShader;
  49. PFNGLDELETESHADERPROC glDeleteShader;
  50. PFNGLDELETEPROGRAMPROC glDeleteProgram;
  51. PFNGLUNIFORMMATRIX4FVPROC glUniformMatrix4fv;
  52. PFNGLGETSHADERIVPROC glGetShaderiv;
  53. PFNGLGETSHADERINFOLOGPROC glGetShaderInfoLog;
  54. #ifndef _MINGW
  55. PFNGLGETUNIFORMLOCATIONARBPROC glGetUniformLocation;
  56. #endif
  57. #endif
  58. GLSLShaderModule::GLSLShaderModule() : PolycodeShaderModule() {
  59. #ifdef _WINDOWS
  60. glUseProgram = (PFNGLUSEPROGRAMPROC)wglGetProcAddress("glUseProgram");
  61. glUniform1i = (PFNGLUNIFORM1IPROC)wglGetProcAddress("glUniform1i");
  62. glUniform1f = (PFNGLUNIFORM1FPROC)wglGetProcAddress("glUniform1f");
  63. glUniform2f = (PFNGLUNIFORM2FPROC)wglGetProcAddress("glUniform2f");
  64. glUniform3f = (PFNGLUNIFORM3FPROC)wglGetProcAddress("glUniform3f");
  65. glCreateShader = (PFNGLCREATESHADERPROC)wglGetProcAddress("glCreateShader");
  66. glShaderSource = (PFNGLSHADERSOURCEPROC)wglGetProcAddress("glShaderSource");
  67. glCompileShader = (PFNGLCOMPILESHADERPROC)wglGetProcAddress("glCompileShader");
  68. glCreateProgram = (PFNGLCREATEPROGRAMPROC)wglGetProcAddress("glCreateProgram");
  69. glAttachShader = (PFNGLATTACHSHADERPROC)wglGetProcAddress("glAttachShader");
  70. glLinkProgram = (PFNGLLINKPROGRAMPROC)wglGetProcAddress("glLinkProgram");
  71. glDetachShader = (PFNGLDETACHSHADERPROC)wglGetProcAddress("glDetachShader");
  72. glDeleteShader = (PFNGLDELETESHADERPROC)wglGetProcAddress("glDeleteShader");
  73. glDeleteProgram = (PFNGLDELETEPROGRAMPROC)wglGetProcAddress("glDeleteProgram");
  74. glUniformMatrix4fv = (PFNGLUNIFORMMATRIX4FVPROC)wglGetProcAddress("glUniformMatrix4fv");
  75. glGetShaderiv = (PFNGLGETSHADERIVPROC)wglGetProcAddress("glGetShaderiv");
  76. glGetShaderInfoLog = (PFNGLGETSHADERINFOLOGPROC)wglGetProcAddress("glGetShaderInfoLog");
  77. #ifndef _MINGW
  78. glGetUniformLocation = (PFNGLGETUNIFORMLOCATIONARBPROC)wglGetProcAddress("glGetUniformLocation");
  79. #endif
  80. #endif
  81. }
  82. GLSLShaderModule::~GLSLShaderModule() {
  83. }
  84. bool GLSLShaderModule::acceptsExtension(const String& extension) {
  85. if(extension == "vert" || extension == "frag") {
  86. return true;
  87. } else {
  88. return false;
  89. }
  90. }
  91. String GLSLShaderModule::getShaderType() {
  92. return "glsl";
  93. }
  94. Shader *GLSLShaderModule::createShader(TiXmlNode *node) {
  95. TiXmlNode* pChild, *pChild2, *pChild3;
  96. GLSLProgram *vp = NULL;
  97. GLSLProgram *fp = NULL;
  98. GLSLShader *retShader = NULL;
  99. for (pChild = node->FirstChild(); pChild != 0; pChild = pChild->NextSibling()) {
  100. if(strcmp(pChild->Value(), "vp") == 0) {
  101. vp = (GLSLProgram*)CoreServices::getInstance()->getResourceManager()->getResource(Resource::RESOURCE_PROGRAM, String(pChild->ToElement()->Attribute("source")));
  102. if(vp) {
  103. for (pChild2 = pChild->FirstChild(); pChild2 != 0; pChild2 = pChild2->NextSibling()) {
  104. if(strcmp(pChild2->Value(), "params") == 0) {
  105. for (pChild3 = pChild2->FirstChild(); pChild3 != 0; pChild3 = pChild3->NextSibling()) {
  106. if(strcmp(pChild3->Value(), "param") == 0) {
  107. addParamToProgram(vp,pChild3);
  108. }
  109. }
  110. }
  111. }
  112. }
  113. }
  114. if(strcmp(pChild->Value(), "fp") == 0) {
  115. fp = (GLSLProgram*)CoreServices::getInstance()->getResourceManager()->getResource(Resource::RESOURCE_PROGRAM, String(pChild->ToElement()->Attribute("source")));
  116. if(fp) {
  117. for (pChild2 = pChild->FirstChild(); pChild2 != 0; pChild2 = pChild2->NextSibling()) {
  118. if(strcmp(pChild2->Value(), "params") == 0) {
  119. for (pChild3 = pChild2->FirstChild(); pChild3 != 0; pChild3 = pChild3->NextSibling()) {
  120. if(strcmp(pChild3->Value(), "param") == 0) {
  121. addParamToProgram(fp,pChild3);
  122. }
  123. }
  124. }
  125. }
  126. }
  127. }
  128. }
  129. if(vp != NULL && fp != NULL) {
  130. GLSLShader *cgShader = new GLSLShader(vp,fp);
  131. cgShader->setName(String(node->ToElement()->Attribute("name")));
  132. retShader = cgShader;
  133. shaders.push_back((Shader*)cgShader);
  134. }
  135. return retShader;
  136. }
  137. void GLSLShaderModule::clearShader() {
  138. glUseProgram(0);
  139. }
  140. void GLSLShaderModule::setGLSLAreaLightPositionParameter(Renderer *renderer, GLSLProgramParam &param, int lightIndex) {
  141. if(renderer->getNumAreaLights() > lightIndex) {
  142. vector<LightInfo> areaLights = renderer->getAreaLights();
  143. Vector3 lPos(areaLights[lightIndex].position.x,areaLights[lightIndex].position.y,areaLights[lightIndex].position.z);
  144. GLfloat LightPosition[] = {lPos.x, lPos.y, lPos.z, 1};
  145. glLightfv (GL_LIGHT0+lightIndex, GL_POSITION, LightPosition); //change the
  146. // glLightf(GL_LIGHT0+lightIndex, GL_CONSTANT_ATTENUATION, areaLights[lightIndex].distance);
  147. // glLightf(GL_LIGHT0+lightIndex, GL_LINEAR_ATTENUATION, areaLights[lightIndex].intensity);
  148. // glLightf(GL_LIGHT0+lightIndex, GL_QUADRATIC_ATTENUATION, areaLights[lightIndex].intensity);
  149. } else {
  150. }
  151. }
  152. void GLSLShaderModule::setGLSLSpotLightPositionParameter(Renderer *renderer, GLSLProgramParam &param, int lightIndex) {
  153. if(renderer->getNumSpotLights() > lightIndex) {
  154. vector<LightInfo> spotLights = renderer->getSpotLights();
  155. Vector3 lPos(spotLights[lightIndex].position.x,spotLights[lightIndex].position.y,spotLights[lightIndex].position.z);
  156. lPos = renderer->getCameraMatrix().inverse() * lPos;
  157. // cgGLSetParameter4f(param.cgParam, lPos.x,lPos.y,lPos.z, spotLights[lightIndex].distance);
  158. } else {
  159. // cgGLSetParameter4f(param.cgParam, 0,0,0,0);
  160. }
  161. }
  162. void GLSLShaderModule::setGLSLSpotLightDirectionParameter(Renderer *renderer, GLSLProgramParam &param, int lightIndex) {
  163. if(renderer->getNumSpotLights() > lightIndex) {
  164. vector<LightInfo> spotLights = renderer->getSpotLights();
  165. Vector3 lPos(spotLights[lightIndex].dir.x,spotLights[lightIndex].dir.y,spotLights[lightIndex].dir.z);
  166. lPos = renderer->getCameraMatrix().inverse().rotateVector(lPos);
  167. // cgGLSetParameter3f(param.cgParam, lPos.x,lPos.y,lPos.z);
  168. } else {
  169. // cgGLSetParameter3f(param.cgParam, 0.0f,0.0f,0.0f);
  170. }
  171. }
  172. void GLSLShaderModule::setGLSLAreaLightColorParameter(Renderer *renderer, GLSLProgramParam &param, int lightIndex) {
  173. if(renderer->getNumAreaLights() > lightIndex) {
  174. vector<LightInfo> areaLights = renderer->getAreaLights();
  175. GLfloat DiffuseLight[] = {areaLights[lightIndex].color.x, areaLights[lightIndex].color.y, areaLights[lightIndex].color.z};
  176. glLightfv (GL_LIGHT0+lightIndex, GL_DIFFUSE, DiffuseLight);
  177. // cgGLSetParameter4f(param.cgParam, areaLights[lightIndex].color.x,areaLights[lightIndex].color.y,areaLights[lightIndex].color.z, areaLights[lightIndex].intensity);
  178. } else {
  179. // cgGLSetParameter4f(param.cgParam, 0,0,0,0);
  180. }
  181. }
  182. void GLSLShaderModule::setGLSLSpotLightColorParameter(Renderer *renderer, GLSLProgramParam &param, int lightIndex) {
  183. if(renderer->getNumSpotLights() > lightIndex) {
  184. vector<LightInfo> spotLights = renderer->getSpotLights();
  185. // cgGLSetParameter4f(param.cgParam, spotLights[lightIndex].color.x,spotLights[lightIndex].color.y,spotLights[lightIndex].color.z, spotLights[lightIndex].intensity);
  186. } else {
  187. // cgGLSetParameter4f(param.cgParam, 0,0,0,0);
  188. }
  189. }
  190. void GLSLShaderModule::setGLSLSpotLightTextureMatrixParameter(Renderer *renderer, GLSLProgramParam &param, int lightIndex) {
  191. if(renderer->getNumLights() > lightIndex) {
  192. vector<LightInfo> spotLights = renderer->getSpotLights();
  193. glMatrixMode(GL_MODELVIEW);
  194. glPushMatrix();
  195. glLoadMatrixd(spotLights[lightIndex].textureMatrix.ml);
  196. // cgGLSetStateMatrixParameter(param.cgParam, GLSL_GL_MODELVIEW_MATRIX,GLSL_GL_MATRIX_IDENTITY);
  197. glPopMatrix();
  198. }
  199. }
  200. void GLSLShaderModule::updateGLSLParam(Renderer *renderer, GLSLShader *glslShader, GLSLProgramParam &param, ShaderBinding *materialOptions, ShaderBinding *localOptions) {
  201. if(param.isAuto) {
  202. switch(param.autoID) {
  203. case GLSLProgramParam::POLY_MODELVIEWPROJ_MATRIX:
  204. // cgGLSetStateMatrixParameter(param.cgParam, GLSL_GL_MODELVIEW_PROJECTION_MATRIX,GLSL_GL_MATRIX_IDENTITY);
  205. break;
  206. case GLSLProgramParam::POLY_SPOT_LIGHT_TEXTUREMATRIX_0:
  207. setGLSLSpotLightTextureMatrixParameter(renderer, param, 0);
  208. break;
  209. case GLSLProgramParam::POLY_SPOT_LIGHT_TEXTUREMATRIX_1:
  210. setGLSLSpotLightTextureMatrixParameter(renderer, param, 1);
  211. break;
  212. case GLSLProgramParam::POLY_SPOT_LIGHT_TEXTUREMATRIX_2:
  213. setGLSLSpotLightTextureMatrixParameter(renderer, param, 2);
  214. break;
  215. case GLSLProgramParam::POLY_SPOT_LIGHT_TEXTUREMATRIX_3:
  216. setGLSLSpotLightTextureMatrixParameter(renderer, param, 3);
  217. break;
  218. case GLSLProgramParam::POLY_AMBIENTCOLOR:
  219. // cgGLSetParameter3f(param.cgParam, renderer->ambientColor.r,renderer->ambientColor.g,renderer->ambientColor.b);
  220. break;
  221. case GLSLProgramParam::POLY_CLEARCOLOR:
  222. // cgGLSetParameter3f(param.cgParam, renderer->clearColor.r,renderer->clearColor.g,renderer->clearColor.b);
  223. break;
  224. case GLSLProgramParam::POLY_SPOT_LIGHT_DIRECTION_0:
  225. setGLSLSpotLightDirectionParameter(renderer, param, 0);
  226. break;
  227. case GLSLProgramParam::POLY_SPOT_LIGHT_DIRECTION_1:
  228. setGLSLSpotLightDirectionParameter(renderer, param, 1);
  229. break;
  230. case GLSLProgramParam::POLY_SPOT_LIGHT_DIRECTION_2:
  231. setGLSLSpotLightDirectionParameter(renderer, param, 2);
  232. break;
  233. case GLSLProgramParam::POLY_SPOT_LIGHT_DIRECTION_3:
  234. setGLSLSpotLightDirectionParameter(renderer, param, 3);
  235. break;
  236. case GLSLProgramParam::POLY_AREA_LIGHT_POSITION_0:
  237. setGLSLAreaLightPositionParameter(renderer, param, 0);
  238. break;
  239. case GLSLProgramParam::POLY_AREA_LIGHT_POSITION_1:
  240. setGLSLAreaLightPositionParameter(renderer, param, 1);
  241. break;
  242. case GLSLProgramParam::POLY_AREA_LIGHT_POSITION_2:
  243. setGLSLAreaLightPositionParameter(renderer, param, 2);
  244. break;
  245. case GLSLProgramParam::POLY_AREA_LIGHT_POSITION_3:
  246. setGLSLAreaLightPositionParameter(renderer, param, 3);
  247. break;
  248. case GLSLProgramParam::POLY_AREA_LIGHT_POSITION_4:
  249. setGLSLAreaLightPositionParameter(renderer, param, 4);
  250. break;
  251. case GLSLProgramParam::POLY_AREA_LIGHT_POSITION_5:
  252. setGLSLAreaLightPositionParameter(renderer, param, 5);
  253. break;
  254. case GLSLProgramParam::POLY_AREA_LIGHT_POSITION_6:
  255. setGLSLAreaLightPositionParameter(renderer, param, 6);
  256. break;
  257. case GLSLProgramParam::POLY_AREA_LIGHT_POSITION_7:
  258. setGLSLAreaLightPositionParameter(renderer, param, 7);
  259. break;
  260. case GLSLProgramParam::POLY_SPOT_LIGHT_POSITION_0:
  261. setGLSLSpotLightPositionParameter(renderer, param, 0);
  262. break;
  263. case GLSLProgramParam::POLY_SPOT_LIGHT_POSITION_1:
  264. setGLSLSpotLightPositionParameter(renderer, param, 1);
  265. break;
  266. case GLSLProgramParam::POLY_SPOT_LIGHT_POSITION_2:
  267. setGLSLSpotLightPositionParameter(renderer, param, 2);
  268. break;
  269. case GLSLProgramParam::POLY_SPOT_LIGHT_POSITION_3:
  270. setGLSLSpotLightPositionParameter(renderer, param, 3);
  271. break;
  272. case GLSLProgramParam::POLY_AREA_LIGHT_COLOR_0:
  273. setGLSLAreaLightColorParameter(renderer, param, 0);
  274. break;
  275. case GLSLProgramParam::POLY_AREA_LIGHT_COLOR_1:
  276. setGLSLAreaLightColorParameter(renderer, param, 1);
  277. break;
  278. case GLSLProgramParam::POLY_AREA_LIGHT_COLOR_2:
  279. setGLSLAreaLightColorParameter(renderer, param, 2);
  280. break;
  281. case GLSLProgramParam::POLY_AREA_LIGHT_COLOR_3:
  282. setGLSLAreaLightColorParameter(renderer, param, 3);
  283. break;
  284. case GLSLProgramParam::POLY_AREA_LIGHT_COLOR_4:
  285. setGLSLAreaLightColorParameter(renderer, param, 4);
  286. break;
  287. case GLSLProgramParam::POLY_AREA_LIGHT_COLOR_5:
  288. setGLSLAreaLightColorParameter(renderer, param, 5);
  289. break;
  290. case GLSLProgramParam::POLY_AREA_LIGHT_COLOR_6:
  291. setGLSLAreaLightColorParameter(renderer, param, 6);
  292. break;
  293. case GLSLProgramParam::POLY_AREA_LIGHT_COLOR_7:
  294. setGLSLAreaLightColorParameter(renderer, param, 7);
  295. break;
  296. case GLSLProgramParam::POLY_SPOT_LIGHT_COLOR_0:
  297. setGLSLSpotLightColorParameter(renderer, param, 0);
  298. break;
  299. case GLSLProgramParam::POLY_SPOT_LIGHT_COLOR_1:
  300. setGLSLSpotLightColorParameter(renderer, param, 1);
  301. break;
  302. case GLSLProgramParam::POLY_SPOT_LIGHT_COLOR_2:
  303. setGLSLSpotLightColorParameter(renderer, param, 2);
  304. break;
  305. case GLSLProgramParam::POLY_SPOT_LIGHT_COLOR_3:
  306. setGLSLSpotLightColorParameter(renderer, param, 3);
  307. break;
  308. case GLSLProgramParam::POLY_MODELVIEW_MATRIX:
  309. // cgGLSetStateMatrixParameter(param.cgParam, GLSL_GL_MODELVIEW_MATRIX,GLSL_GL_MATRIX_IDENTITY); }
  310. break;
  311. case GLSLProgramParam::POLY_MODELVIEW_INVERSE_MATRIX:
  312. // cgGLSetStateMatrixParameter(param.cgParam, GLSL_GL_MODELVIEW_MATRIX,GLSL_GL_MATRIX_INVERSE_TRANSPOSE);
  313. break;
  314. case GLSLProgramParam::POLY_EXPOSURE_LEVEL:
  315. // cgGLSetParameter1f(param.cgParam, renderer->exposureLevel);
  316. break;
  317. }
  318. } else {
  319. void *paramData = param.defaultData;
  320. LocalShaderParam *localParam = materialOptions->getLocalParamByName(param.name);
  321. if(localParam)
  322. paramData = localParam->data;
  323. localParam = localOptions->getLocalParamByName(param.name);
  324. if(localParam)
  325. paramData = localParam->data;
  326. Number *fval;
  327. switch(param.paramType) {
  328. case GLSLProgramParam::PARAM_Number:
  329. {
  330. fval = (Number*)paramData;
  331. int paramLocation = glGetUniformLocation(glslShader->shader_id, param.name.c_str());
  332. glUniform1f(paramLocation, *fval);
  333. break;
  334. }
  335. case GLSLProgramParam::PARAM_Number2:
  336. {
  337. Vector2 *fval2 = (Vector2*)paramData;
  338. int paramLocation = glGetUniformLocation(glslShader->shader_id, param.name.c_str());
  339. glUniform2f(paramLocation, fval2->x, fval2->y); break;
  340. }
  341. case GLSLProgramParam::PARAM_Number3:
  342. {
  343. Vector3 *fval3 = (Vector3*)paramData;
  344. int paramLocation = glGetUniformLocation(glslShader->shader_id, param.name.c_str());
  345. glUniform3f(paramLocation, fval3->x,fval3->y,fval3->z);
  346. break;
  347. }
  348. }
  349. }
  350. }
  351. bool GLSLShaderModule::applyShaderMaterial(Renderer *renderer, Material *material, ShaderBinding *localOptions, unsigned int shaderIndex) {
  352. GLSLShader *glslShader = (GLSLShader*)material->getShader(shaderIndex);
  353. glPushMatrix();
  354. glLoadIdentity();
  355. int numRendererAreaLights = renderer->getNumAreaLights();
  356. int numRendererSpotLights = renderer->getNumSpotLights();
  357. int numTotalLights = glslShader->numAreaLights + glslShader->numSpotLights;
  358. if(numTotalLights > 0) {
  359. renderer->sortLights();
  360. }
  361. for(int i=0 ; i < numTotalLights; i++) {
  362. GLfloat resetData[] = {0.0, 0.0, 0.0, 0.0};
  363. glLightfv (GL_LIGHT0+i, GL_DIFFUSE, resetData);
  364. glLightfv (GL_LIGHT0+i, GL_SPECULAR, resetData);
  365. glLightfv (GL_LIGHT0+i, GL_AMBIENT, resetData);
  366. glLightfv (GL_LIGHT0+i, GL_POSITION, resetData);
  367. glLightf (GL_LIGHT0+i, GL_SPOT_CUTOFF, 180);
  368. glLightf (GL_LIGHT0+i, GL_CONSTANT_ATTENUATION,1.0);
  369. glLightf (GL_LIGHT0+i, GL_LINEAR_ATTENUATION,0.0);
  370. glLightf (GL_LIGHT0+i, GL_QUADRATIC_ATTENUATION, 0.0);
  371. }
  372. int lightIndex = 0;
  373. vector<LightInfo> areaLights = renderer->getAreaLights();
  374. // printf("Applying {\n");
  375. // for(int z=0;z < areaLights.size(); z++) {
  376. // LightInfo light = areaLights[z];
  377. // printf("Light: %f %f %f\n", light.position.x, light.position.y, light.position.z);
  378. // }
  379. // printf("}\n");
  380. GLfloat ambientVal[] = {1, 1, 1, 1.0};
  381. for(int i=0; i < glslShader->numAreaLights; i++) {
  382. LightInfo light;
  383. if(i < numRendererAreaLights) {
  384. light = areaLights[i];
  385. light.position = renderer->getCameraMatrix().inverse() * light.position;
  386. ambientVal[0] = renderer->ambientColor.r;
  387. ambientVal[1] = renderer->ambientColor.g;
  388. ambientVal[2] = renderer->ambientColor.b;
  389. ambientVal[3] = 1;
  390. GLfloat data4[] = {light.color.x * light.intensity, light.color.y * light.intensity, light.color.z * light.intensity, 1.0};
  391. glLightfv (GL_LIGHT0+lightIndex, GL_DIFFUSE, data4);
  392. data4[0] = light.specularColor.r* light.intensity;
  393. data4[1] = light.specularColor.g* light.intensity;
  394. data4[2] = light.specularColor.b* light.intensity;
  395. data4[3] = light.specularColor.a* light.intensity;
  396. glLightfv (GL_LIGHT0+lightIndex, GL_SPECULAR, data4);
  397. data4[3] = 1.0;
  398. glLightfv (GL_LIGHT0+lightIndex, GL_AMBIENT, ambientVal);
  399. glLightf (GL_LIGHT0+lightIndex, GL_SPOT_CUTOFF, 180);
  400. data4[0] = light.position.x;
  401. data4[1] = light.position.y;
  402. data4[2] = light.position.z;
  403. glLightfv (GL_LIGHT0+lightIndex, GL_POSITION, data4);
  404. glLightf (GL_LIGHT0+lightIndex, GL_CONSTANT_ATTENUATION, light.constantAttenuation);
  405. glLightf (GL_LIGHT0+lightIndex, GL_LINEAR_ATTENUATION, light.linearAttenuation);
  406. glLightf (GL_LIGHT0+lightIndex, GL_QUADRATIC_ATTENUATION, light.quadraticAttenuation);
  407. }
  408. lightIndex++;
  409. }
  410. vector<LightInfo> spotLights = renderer->getSpotLights();
  411. // vector<Texture*> shadowMapTextures = renderer->getShadowMapTextures();
  412. char texName[32];
  413. char matName[32];
  414. int shadowMapTextureIndex = 0;
  415. glUseProgram(glslShader->shader_id);
  416. int textureIndex = 0;
  417. for(int i=0; i < glslShader->numSpotLights; i++) {
  418. LightInfo light;
  419. Vector3 pos;
  420. Vector3 dir;
  421. if(i < numRendererSpotLights) {
  422. light = spotLights[i];
  423. pos = light.position;
  424. dir = light.dir;
  425. pos = renderer->getCameraMatrix().inverse() * pos;
  426. dir = renderer->getCameraMatrix().inverse().rotateVector(dir);
  427. ambientVal[0] = renderer->ambientColor.r;
  428. ambientVal[1] = renderer->ambientColor.g;
  429. ambientVal[2] = renderer->ambientColor.b;
  430. ambientVal[3] = 1;
  431. GLfloat data4[] = {light.color.x * light.intensity, light.color.y * light.intensity, light.color.z * light.intensity, 1.0};
  432. glLightfv (GL_LIGHT0+lightIndex, GL_DIFFUSE, data4);
  433. data4[0] = light.specularColor.r* light.intensity;
  434. data4[1] = light.specularColor.g* light.intensity;
  435. data4[2] = light.specularColor.b* light.intensity;
  436. data4[3] = light.specularColor.a* light.intensity;
  437. glLightfv (GL_LIGHT0+lightIndex, GL_SPECULAR, data4);
  438. data4[3] = 1.0;
  439. glLightfv (GL_LIGHT0+lightIndex, GL_AMBIENT, ambientVal);
  440. glLightf (GL_LIGHT0+lightIndex, GL_SPOT_CUTOFF, light.spotlightCutoff);
  441. glLightf (GL_LIGHT0+lightIndex, GL_SPOT_EXPONENT, light.spotlightExponent);
  442. data4[0] = dir.x;
  443. data4[1] = dir.y;
  444. data4[2] = dir.z;
  445. glLightfv (GL_LIGHT0+lightIndex, GL_SPOT_DIRECTION, data4);
  446. data4[0] = pos.x;
  447. data4[1] = pos.y;
  448. data4[2] = pos.z;
  449. glLightfv (GL_LIGHT0+lightIndex, GL_POSITION, data4);
  450. glLightf (GL_LIGHT0+lightIndex, GL_CONSTANT_ATTENUATION, light.constantAttenuation);
  451. glLightf (GL_LIGHT0+lightIndex, GL_LINEAR_ATTENUATION, light.linearAttenuation);
  452. glLightf (GL_LIGHT0+lightIndex, GL_QUADRATIC_ATTENUATION, light.quadraticAttenuation);
  453. if(light.shadowsEnabled) {
  454. if(shadowMapTextureIndex < 4) {
  455. switch(shadowMapTextureIndex) {
  456. case 0:
  457. strcpy(texName, "shadowMap0");
  458. strcpy(matName, "shadowMatrix0");
  459. break;
  460. case 1:
  461. strcpy(texName, "shadowMap1");
  462. strcpy(matName, "shadowMatrix1");
  463. break;
  464. case 2:
  465. strcpy(texName, "shadowMap2");
  466. strcpy(matName, "shadowMatrix2");
  467. break;
  468. case 3:
  469. strcpy(texName, "shadowMap3");
  470. strcpy(matName, "shadowMatrix3");
  471. break;
  472. }
  473. int texture_location = glGetUniformLocation(glslShader->shader_id, texName);
  474. glUniform1i(texture_location, textureIndex);
  475. glActiveTexture(GL_TEXTURE0 + textureIndex);
  476. glBindTexture(GL_TEXTURE_2D, ((OpenGLTexture*)light.shadowMapTexture)->getTextureID());
  477. textureIndex++;
  478. // glMatrixMode(GL_MODELVIEW);
  479. // glPushMatrix();
  480. // glLoadMatrixd(light.textureMatrix.ml);
  481. int mloc = glGetUniformLocation(glslShader->shader_id, matName);
  482. light.textureMatrix = light.textureMatrix;
  483. GLfloat mat[16];
  484. for(int z=0; z < 16; z++) {
  485. mat[z] = light.textureMatrix.ml[z];
  486. }
  487. glUniformMatrix4fv(mloc, 1, false, mat);
  488. // glPopMatrix();
  489. }
  490. shadowMapTextureIndex++;
  491. }
  492. else {
  493. light.shadowsEnabled = false;
  494. }
  495. }
  496. lightIndex++;
  497. }
  498. glPopMatrix();
  499. glEnable(GL_TEXTURE_2D);
  500. Matrix4 modelMatrix = renderer->getCurrentModelMatrix();
  501. int mloc = glGetUniformLocation(glslShader->shader_id, "modelMatrix");
  502. GLfloat mat[16];
  503. for(int z=0; z < 16; z++) {
  504. mat[z] = modelMatrix.ml[z];
  505. }
  506. glUniformMatrix4fv(mloc, 1, false, mat);
  507. GLSLShaderBinding *cgBinding = (GLSLShaderBinding*)material->getShaderBinding(shaderIndex);
  508. for(int i=0; i < glslShader->vp->params.size(); i++) {
  509. GLSLProgramParam param = glslShader->vp->params[i];
  510. updateGLSLParam(renderer, glslShader, param, material->getShaderBinding(shaderIndex), localOptions);
  511. }
  512. for(int i=0; i < glslShader->fp->params.size(); i++) {
  513. GLSLProgramParam param = glslShader->fp->params[i];
  514. updateGLSLParam(renderer, glslShader, param, material->getShaderBinding(shaderIndex), localOptions);
  515. }
  516. for(int i=0; i < cgBinding->textures.size(); i++) {
  517. int texture_location = glGetUniformLocation(glslShader->shader_id, cgBinding->textures[i].name.c_str());
  518. glUniform1i(texture_location, textureIndex);
  519. glActiveTexture(GL_TEXTURE0 + textureIndex);
  520. glBindTexture(GL_TEXTURE_2D, ((OpenGLTexture*)cgBinding->textures[i].texture)->getTextureID());
  521. textureIndex++;
  522. }
  523. for(int i=0; i < cgBinding->cubemaps.size(); i++) {
  524. int texture_location = glGetUniformLocation(glslShader->shader_id, cgBinding->cubemaps[i].name.c_str());
  525. glUniform1i(texture_location, textureIndex);
  526. glActiveTexture(GL_TEXTURE0 + textureIndex);
  527. glBindTexture(GL_TEXTURE_CUBE_MAP, ((OpenGLCubemap*)cgBinding->cubemaps[i].cubemap)->getTextureID());
  528. textureIndex++;
  529. }
  530. cgBinding = (GLSLShaderBinding*)localOptions;
  531. for(int i=0; i < cgBinding->textures.size(); i++) {
  532. int texture_location = glGetUniformLocation(glslShader->shader_id, cgBinding->textures[i].name.c_str());
  533. glUniform1i(texture_location, textureIndex);
  534. glActiveTexture(GL_TEXTURE0 + textureIndex);
  535. glBindTexture(GL_TEXTURE_2D, ((OpenGLTexture*)cgBinding->textures[i].texture)->getTextureID());
  536. textureIndex++;
  537. }
  538. // Logger::log("applying %s (%s %s)\n", material->getShader()->getName().c_str(), cgShader->vp->getResourceName().c_str(), cgShader->fp->getResourceName().c_str());
  539. /*
  540. vector<Texture*> shadowMapTextures = renderer->getShadowMapTextures();
  541. char texName[32];
  542. for(int i=0; i< 4; i++) {
  543. if(i < shadowMapTextures.size()) {
  544. switch(i) {
  545. case 0:
  546. strcpy(texName, "shadowMap0");
  547. break;
  548. case 1:
  549. strcpy(texName, "shadowMap1");
  550. break;
  551. case 2:
  552. strcpy(texName, "shadowMap2");
  553. break;
  554. case 3:
  555. strcpy(texName, "shadowMap3");
  556. break;
  557. }
  558. int texture_location = glGetUniformLocation(glslShader->shader_id, texName);
  559. glUniform1i(texture_location, textureIndex);
  560. glActiveTexture(GL_TEXTURE0 + textureIndex);
  561. glBindTexture(GL_TEXTURE_2D, ((OpenGLTexture*)shadowMapTextures[i])->getTextureID());
  562. textureIndex++;
  563. }
  564. }
  565. */
  566. /*
  567. cgBinding = (GLSLShaderBinding*)localOptions;
  568. for(int i=0; i < cgBinding->textures.size(); i++) {
  569. cgGLSetTextureParameter(cgBinding->textures[i].vpParam, ((OpenGLTexture*)cgBinding->textures[i].texture)->getTextureID());
  570. cgGLEnableTextureParameter(cgBinding->textures[i].vpParam);
  571. }
  572. vector<Texture*> shadowMapTextures = renderer->getShadowMapTextures();
  573. char texName[32];
  574. for(int i=0; i< 4; i++) {
  575. if(i < shadowMapTextures.size()) {
  576. switch(i) {
  577. case 0:
  578. strcpy(texName, "shadowMap0");
  579. break;
  580. case 1:
  581. strcpy(texName, "shadowMap1");
  582. break;
  583. case 2:
  584. strcpy(texName, "shadowMap2");
  585. break;
  586. case 3:
  587. strcpy(texName, "shadowMap3");
  588. break;
  589. }
  590. cgGLSetTextureParameter(cgGetNamedParameter(cgShader->fp->program, texName), ((OpenGLTexture*)shadowMapTextures[i])->getTextureID());
  591. cgGLEnableTextureParameter(cgGetNamedParameter(cgShader->fp->program, texName));
  592. }
  593. }
  594. */
  595. return true;
  596. }
  597. void GLSLShaderModule::addParamToProgram(GLSLProgram *program,TiXmlNode *node) {
  598. bool isAuto = false;
  599. int autoID = 0;
  600. int paramType = GLSLProgramParam::PARAM_UNKNOWN;
  601. void *defaultData = NULL;
  602. if(strcmp(node->ToElement()->Attribute("type"), "auto") == 0) {
  603. isAuto = true;
  604. String pid = node->ToElement()->Attribute("id");
  605. if(pid == "POLY_MODELVIEWPROJ_MATRIX")
  606. autoID = GLSLProgramParam::POLY_MODELVIEWPROJ_MATRIX;
  607. else if(pid == "POLY_AREA_LIGHT_POSITION_0")
  608. autoID = GLSLProgramParam::POLY_AREA_LIGHT_POSITION_0;
  609. else if(pid == "POLY_AREA_LIGHT_POSITION_1")
  610. autoID = GLSLProgramParam::POLY_AREA_LIGHT_POSITION_1;
  611. else if(pid == "POLY_AREA_LIGHT_POSITION_2")
  612. autoID = GLSLProgramParam::POLY_AREA_LIGHT_POSITION_2;
  613. else if(pid == "POLY_AREA_LIGHT_POSITION_3")
  614. autoID = GLSLProgramParam::POLY_AREA_LIGHT_POSITION_3;
  615. else if(pid == "POLY_AREA_LIGHT_POSITION_4")
  616. autoID = GLSLProgramParam::POLY_AREA_LIGHT_POSITION_4;
  617. else if(pid == "POLY_AREA_LIGHT_POSITION_5")
  618. autoID = GLSLProgramParam::POLY_AREA_LIGHT_POSITION_5;
  619. else if(pid == "POLY_AREA_LIGHT_POSITION_6")
  620. autoID = GLSLProgramParam::POLY_AREA_LIGHT_POSITION_6;
  621. else if(pid == "POLY_AREA_LIGHT_POSITION_7")
  622. autoID = GLSLProgramParam::POLY_AREA_LIGHT_POSITION_7;
  623. else if(pid == "POLY_SPOT_LIGHT_POSITION_0")
  624. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_POSITION_0;
  625. else if(pid == "POLY_SPOT_LIGHT_POSITION_1")
  626. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_POSITION_1;
  627. else if(pid == "POLY_SPOT_LIGHT_POSITION_2")
  628. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_POSITION_2;
  629. else if(pid == "POLY_SPOT_LIGHT_POSITION_3")
  630. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_POSITION_3;
  631. else if(pid == "POLY_AREA_LIGHT_COLOR_0")
  632. autoID = GLSLProgramParam::POLY_AREA_LIGHT_COLOR_0;
  633. else if(pid == "POLY_AREA_LIGHT_COLOR_1")
  634. autoID = GLSLProgramParam::POLY_AREA_LIGHT_COLOR_1;
  635. else if(pid == "POLY_AREA_LIGHT_COLOR_2")
  636. autoID = GLSLProgramParam::POLY_AREA_LIGHT_COLOR_2;
  637. else if(pid == "POLY_AREA_LIGHT_COLOR_3")
  638. autoID = GLSLProgramParam::POLY_AREA_LIGHT_COLOR_3;
  639. else if(pid == "POLY_AREA_LIGHT_COLOR_4")
  640. autoID = GLSLProgramParam::POLY_AREA_LIGHT_COLOR_4;
  641. else if(pid == "POLY_AREA_LIGHT_COLOR_5")
  642. autoID = GLSLProgramParam::POLY_AREA_LIGHT_COLOR_5;
  643. else if(pid == "POLY_AREA_LIGHT_COLOR_6")
  644. autoID = GLSLProgramParam::POLY_AREA_LIGHT_COLOR_6;
  645. else if(pid == "POLY_AREA_LIGHT_COLOR_7")
  646. autoID = GLSLProgramParam::POLY_AREA_LIGHT_COLOR_7;
  647. else if(pid == "POLY_SPOT_LIGHT_COLOR_0")
  648. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_COLOR_0;
  649. else if(pid == "POLY_SPOT_LIGHT_COLOR_1")
  650. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_COLOR_1;
  651. else if(pid == "POLY_SPOT_LIGHT_COLOR_2")
  652. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_COLOR_2;
  653. else if(pid == "POLY_SPOT_LIGHT_COLOR_3")
  654. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_COLOR_3;
  655. else if(pid == "POLY_SPOT_LIGHT_DIRECTION_0")
  656. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_DIRECTION_0;
  657. else if(pid == "POLY_SPOT_LIGHT_DIRECTION_1")
  658. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_DIRECTION_1;
  659. else if(pid == "POLY_SPOT_LIGHT_DIRECTION_2")
  660. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_DIRECTION_2;
  661. else if(pid == "POLY_SPOT_LIGHT_DIRECTION_3")
  662. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_DIRECTION_3;
  663. else if(pid == "POLY_SPOT_LIGHT_TEXTUREMATRIX_0")
  664. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_TEXTUREMATRIX_0;
  665. else if(pid == "POLY_SPOT_LIGHT_TEXTUREMATRIX_1")
  666. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_TEXTUREMATRIX_1;
  667. else if(pid == "POLY_SPOT_LIGHT_TEXTUREMATRIX_2")
  668. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_TEXTUREMATRIX_2;
  669. else if(pid == "POLY_SPOT_LIGHT_TEXTUREMATRIX_3")
  670. autoID = GLSLProgramParam::POLY_SPOT_LIGHT_TEXTUREMATRIX_3;
  671. else if(pid == "POLY_MODELVIEW_MATRIX")
  672. autoID = GLSLProgramParam::POLY_MODELVIEW_MATRIX;
  673. else if(pid == "POLY_MODELVIEW_INVERSE_MATRIX")
  674. autoID = GLSLProgramParam::POLY_MODELVIEW_INVERSE_MATRIX;
  675. else if(pid == "POLY_EXPOSURE_LEVEL")
  676. autoID = GLSLProgramParam::POLY_EXPOSURE_LEVEL;
  677. else if(pid == "POLY_CLEARCOLOR")
  678. autoID = GLSLProgramParam::POLY_CLEARCOLOR;
  679. else if(pid == "POLY_AMBIENTCOLOR")
  680. autoID = GLSLProgramParam::POLY_AMBIENTCOLOR;
  681. else
  682. isAuto = false;
  683. } else {
  684. defaultData = GLSLProgramParam::createParamData(&paramType, node->ToElement()->Attribute("type"), node->ToElement()->Attribute("default"));
  685. }
  686. program->addParam(node->ToElement()->Attribute("name"), isAuto, autoID, paramType, defaultData);
  687. }
  688. void GLSLShaderModule::reloadPrograms() {
  689. for(int i=0; i < programs.size(); i++) {
  690. GLSLProgram *program = programs[i];
  691. recreateGLSLProgram(program, program->getResourcePath(), program->type);
  692. }
  693. }
  694. void GLSLShaderModule::recreateGLSLProgram(GLSLProgram *prog, const String& fileName, int type) {
  695. OSFILE *file = OSBasics::open(fileName, "r");
  696. OSBasics::seek(file, 0, SEEK_END);
  697. long progsize = OSBasics::tell(file);
  698. OSBasics::seek(file, 0, SEEK_SET);
  699. char *buffer = (char*)malloc(progsize+1);
  700. memset(buffer, 0, progsize+1);
  701. OSBasics::read(buffer, progsize, 1, file);
  702. OSBasics::close(file);
  703. if(type == GLSLProgram::TYPE_VERT) {
  704. prog->program = glCreateShader(GL_VERTEX_SHADER);
  705. } else {
  706. prog->program = glCreateShader(GL_FRAGMENT_SHADER);
  707. }
  708. glShaderSource(prog->program, 1, (const GLchar**)&buffer, 0);
  709. glCompileShader(prog->program);
  710. GLint compiled = true;
  711. glGetShaderiv(prog->program, GL_COMPILE_STATUS, &compiled);
  712. if(!compiled) {
  713. GLint length;
  714. GLchar* log;
  715. glGetShaderiv(prog->program, GL_INFO_LOG_LENGTH, &length);
  716. log = (GLchar*)malloc(length);
  717. glGetShaderInfoLog(prog->program, length, &length, log);
  718. printf("GLSL ERROR: %s\n", log);
  719. free(log);
  720. }
  721. free(buffer);
  722. }
  723. GLSLProgram *GLSLShaderModule::createGLSLProgram(const String& fileName, int type) {
  724. GLSLProgram *prog = new GLSLProgram(type);
  725. recreateGLSLProgram(prog, fileName, type);
  726. programs.push_back(prog);
  727. return prog;
  728. }
  729. Resource* GLSLShaderModule::createProgramFromFile(const String& extension, const String& fullPath) {
  730. if(extension == "vert") {
  731. Logger::log("Adding GLSL vertex program %s\n", fullPath.c_str());
  732. return createGLSLProgram(fullPath, GLSLProgram::TYPE_VERT);
  733. }
  734. if(extension == "frag") {
  735. Logger::log("Adding GLSL fragment program %s\n", fullPath.c_str());
  736. return createGLSLProgram(fullPath, GLSLProgram::TYPE_FRAG);
  737. }
  738. return NULL;
  739. }