shader_compiler_gles2.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833
  1. /*************************************************************************/
  2. /* shader_compiler_gles2.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* http://www.godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
  9. /* */
  10. /* Permission is hereby granted, free of charge, to any person obtaining */
  11. /* a copy of this software and associated documentation files (the */
  12. /* "Software"), to deal in the Software without restriction, including */
  13. /* without limitation the rights to use, copy, modify, merge, publish, */
  14. /* distribute, sublicense, and/or sell copies of the Software, and to */
  15. /* permit persons to whom the Software is furnished to do so, subject to */
  16. /* the following conditions: */
  17. /* */
  18. /* The above copyright notice and this permission notice shall be */
  19. /* included in all copies or substantial portions of the Software. */
  20. /* */
  21. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  22. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  23. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  24. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  25. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  26. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  27. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  28. /*************************************************************************/
  29. #include "shader_compiler_gles2.h"
  30. #include "print_string.h"
  31. #include "stdio.h"
  32. //#define DEBUG_SHADER_ENABLED
  33. typedef ShaderLanguage SL;
  34. struct CodeGLSL2 {
  35. String code;
  36. };
  37. static String _mktab(int p_level) {
  38. String tb;
  39. for(int i=0;i<p_level;i++) {
  40. tb+="\t";
  41. }
  42. return tb;
  43. }
  44. static String _typestr(SL::DataType p_type) {
  45. switch(p_type) {
  46. case SL::TYPE_VOID: return "void";
  47. case SL::TYPE_BOOL: return "bool";
  48. case SL::TYPE_FLOAT: return "float";
  49. case SL::TYPE_VEC2: return "vec2";
  50. case SL::TYPE_VEC3: return "vec3";
  51. case SL::TYPE_VEC4: return "vec4";
  52. case SL::TYPE_MAT3: return "mat3";
  53. case SL::TYPE_MAT4: return "mat4";
  54. case SL::TYPE_TEXTURE: return "sampler2D";
  55. case SL::TYPE_CUBEMAP: return "samplerCube";
  56. }
  57. return "";
  58. }
  59. static String _mknum(float p_num) {
  60. return String::num_real(p_num);
  61. }
  62. static String _opstr(SL::Operator p_op) {
  63. switch(p_op) {
  64. case SL::OP_ASSIGN: return "=";
  65. case SL::OP_ADD: return "+";
  66. case SL::OP_SUB: return "-";
  67. case SL::OP_MUL: return "*";
  68. case SL::OP_DIV: return "/";
  69. case SL::OP_ASSIGN_ADD: return "+=";
  70. case SL::OP_ASSIGN_SUB: return "-=";
  71. case SL::OP_ASSIGN_MUL: return "*=";
  72. case SL::OP_ASSIGN_DIV: return "/=";
  73. case SL::OP_NEG: return "-";
  74. case SL::OP_NOT: return "!";
  75. case SL::OP_CMP_EQ: return "==";
  76. case SL::OP_CMP_NEQ: return "!=";
  77. case SL::OP_CMP_LEQ: return "<=";
  78. case SL::OP_CMP_GEQ: return ">=";
  79. case SL::OP_CMP_LESS: return "<";
  80. case SL::OP_CMP_GREATER: return ">";
  81. case SL::OP_CMP_OR: return "||";
  82. case SL::OP_CMP_AND: return "&&";
  83. default: return "";
  84. }
  85. return "";
  86. }
  87. //#ifdef DEBUG_SHADER_ENABLED
  88. #if 1
  89. #define ENDL "\n"
  90. #else
  91. #define ENDL ""
  92. #endif
  93. String ShaderCompilerGLES2::dump_node_code(SL::Node *p_node,int p_level,bool p_assign_left) {
  94. String code;
  95. switch(p_node->type) {
  96. case SL::Node::TYPE_PROGRAM: {
  97. SL::ProgramNode *pnode=(SL::ProgramNode*)p_node;
  98. code+=dump_node_code(pnode->body,p_level);
  99. } break;
  100. case SL::Node::TYPE_FUNCTION: {
  101. } break;
  102. case SL::Node::TYPE_BLOCK: {
  103. SL::BlockNode *bnode=(SL::BlockNode*)p_node;
  104. //variables
  105. code+="{"ENDL;
  106. for(Map<StringName,SL::DataType>::Element *E=bnode->variables.front();E;E=E->next()) {
  107. code+=_mktab(p_level)+_typestr(E->value())+" "+replace_string(E->key())+";"ENDL;
  108. }
  109. for(int i=0;i<bnode->statements.size();i++) {
  110. code+=_mktab(p_level)+dump_node_code(bnode->statements[i],p_level)+";"ENDL;
  111. }
  112. code+="}"ENDL;
  113. } break;
  114. case SL::Node::TYPE_VARIABLE: {
  115. SL::VariableNode *vnode=(SL::VariableNode*)p_node;
  116. if (type==ShaderLanguage::SHADER_MATERIAL_VERTEX) {
  117. if (vnode->name==vname_vertex && p_assign_left) {
  118. vertex_code_writes_vertex=true;
  119. }
  120. if (vnode->name==vname_color_interp) {
  121. flags->use_color_interp=true;
  122. }
  123. if (vnode->name==vname_uv_interp) {
  124. flags->use_uv_interp=true;
  125. }
  126. if (vnode->name==vname_uv2_interp) {
  127. flags->use_uv2_interp=true;
  128. }
  129. if (vnode->name==vname_var1_interp) {
  130. flags->use_var1_interp=true;
  131. }
  132. if (vnode->name==vname_var2_interp) {
  133. flags->use_var2_interp=true;
  134. }
  135. if (vnode->name==vname_tangent_interp || vnode->name==vname_binormal_interp) {
  136. flags->use_tangent_interp=true;
  137. }
  138. }
  139. if (type==ShaderLanguage::SHADER_MATERIAL_FRAGMENT) {
  140. if (vnode->name==vname_discard) {
  141. uses_discard=true;
  142. }
  143. if (vnode->name==vname_normalmap) {
  144. uses_normalmap=true;
  145. }
  146. if (vnode->name==vname_screen_uv) {
  147. uses_screen_uv=true;
  148. }
  149. if (vnode->name==vname_diffuse_alpha && p_assign_left) {
  150. uses_alpha=true;
  151. }
  152. if (vnode->name==vname_color_interp) {
  153. flags->use_color_interp=true;
  154. }
  155. if (vnode->name==vname_uv_interp) {
  156. flags->use_uv_interp=true;
  157. }
  158. if (vnode->name==vname_uv2_interp) {
  159. flags->use_uv2_interp=true;
  160. }
  161. if (vnode->name==vname_var1_interp) {
  162. flags->use_var1_interp=true;
  163. }
  164. if (vnode->name==vname_var2_interp) {
  165. flags->use_var2_interp=true;
  166. }
  167. if (vnode->name==vname_tangent_interp || vnode->name==vname_binormal_interp) {
  168. flags->use_tangent_interp=true;
  169. }
  170. }
  171. if (type==ShaderLanguage::SHADER_MATERIAL_LIGHT) {
  172. if (vnode->name==vname_light) {
  173. uses_light=true;
  174. }
  175. }
  176. if (type==ShaderLanguage::SHADER_CANVAS_ITEM_VERTEX) {
  177. if (vnode->name==vname_var1_interp) {
  178. flags->use_var1_interp=true;
  179. }
  180. if (vnode->name==vname_var2_interp) {
  181. flags->use_var2_interp=true;
  182. }
  183. }
  184. if (type==ShaderLanguage::SHADER_CANVAS_ITEM_FRAGMENT) {
  185. if (vnode->name==vname_texpixel_size) {
  186. uses_texpixel_size=true;
  187. }
  188. if (vnode->name==vname_normal) {
  189. uses_normal=true;
  190. }
  191. if (vnode->name==vname_screen_uv) {
  192. uses_screen_uv=true;
  193. }
  194. if (vnode->name==vname_var1_interp) {
  195. flags->use_var1_interp=true;
  196. }
  197. if (vnode->name==vname_var2_interp) {
  198. flags->use_var2_interp=true;
  199. }
  200. }
  201. if (type==ShaderLanguage::SHADER_CANVAS_ITEM_LIGHT) {
  202. if (vnode->name==vname_light) {
  203. uses_light=true;
  204. }
  205. }
  206. if (vnode->name==vname_time) {
  207. uses_time=true;
  208. }
  209. code=replace_string(vnode->name);
  210. } break;
  211. case SL::Node::TYPE_CONSTANT: {
  212. SL::ConstantNode *cnode=(SL::ConstantNode*)p_node;
  213. switch(cnode->datatype) {
  214. case SL::TYPE_BOOL: code=cnode->value.operator bool()?"true":"false"; break;
  215. case SL::TYPE_FLOAT: code=_mknum(cnode->value); break; //force zeros, so GLSL doesn't confuse with integer.
  216. case SL::TYPE_VEC2: { Vector2 v = cnode->value; code="vec2("+_mknum(v.x)+", "+_mknum(v.y)+")"; } break;
  217. case SL::TYPE_VEC3: { Vector3 v = cnode->value; code="vec3("+_mknum(v.x)+", "+_mknum(v.y)+", "+_mknum(v.z)+")"; } break;
  218. case SL::TYPE_VEC4: { Plane v = cnode->value; code="vec4("+_mknum(v.normal.x)+", "+_mknum(v.normal.y)+", "+_mknum(v.normal.z)+", "+_mknum(v.d)+")"; } break;
  219. case SL::TYPE_MAT3: { Matrix3 x = cnode->value; code="mat3( vec3("+_mknum(x.get_axis(0).x)+", "+_mknum(x.get_axis(0).y)+", "+_mknum(x.get_axis(0).z)+"), vec3("+_mknum(x.get_axis(1).x)+", "+_mknum(x.get_axis(1).y)+", "+_mknum(x.get_axis(1).z)+"), vec3("+_mknum(x.get_axis(2).x)+", "+_mknum(x.get_axis(2).y)+", "+_mknum(x.get_axis(2).z)+"))"; } break;
  220. case SL::TYPE_MAT4: { Transform x = cnode->value; code="mat4( vec4("+_mknum(x.basis.get_axis(0).x)+", "+_mknum(x.basis.get_axis(0).y)+", "+_mknum(x.basis.get_axis(0).z)+",0.0), vec4("+_mknum(x.basis.get_axis(1).x)+", "+_mknum(x.basis.get_axis(1).y)+", "+_mknum(x.basis.get_axis(1).z)+",0.0), vec4("+_mknum(x.basis.get_axis(2).x)+", "+_mknum(x.basis.get_axis(2).y)+", "+_mknum(x.basis.get_axis(2).z)+",0.0), vec4("+_mknum(x.origin.x)+", "+_mknum(x.origin.y)+", "+_mknum(x.origin.z)+",1.0))"; } break;
  221. default: code="<error: "+Variant::get_type_name(cnode->value.get_type())+" ("+itos(cnode->datatype)+">";
  222. }
  223. } break;
  224. case SL::Node::TYPE_OPERATOR: {
  225. SL::OperatorNode *onode=(SL::OperatorNode*)p_node;
  226. switch(onode->op) {
  227. case SL::OP_ASSIGN_MUL: {
  228. if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC3 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT4) {
  229. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  230. String mul_r=dump_node_code(onode->arguments[1],p_level);
  231. code=mul_l+"=(vec4("+mul_l+",1.0)*("+mul_r+")).xyz";
  232. break;
  233. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT4 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC3) {
  234. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  235. String mul_r=dump_node_code(onode->arguments[1],p_level);
  236. code=mul_l+"=(("+mul_l+")*vec4("+mul_r+",1.0)).xyz";
  237. break;
  238. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC2 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT4) {
  239. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  240. String mul_r=dump_node_code(onode->arguments[1],p_level);
  241. code=mul_l+"=(vec4("+mul_l+",1.0,1.0)*("+mul_r+")).xy";
  242. break;
  243. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT4 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC2) {
  244. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  245. String mul_r=dump_node_code(onode->arguments[1],p_level);
  246. code=mul_l+"=(("+mul_l+")*vec4("+mul_r+",1.0,1.0)).xy";
  247. break;
  248. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC2 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT3) {
  249. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  250. String mul_r=dump_node_code(onode->arguments[1],p_level);
  251. code=mul_l+"=(("+mul_l+")*vec3("+mul_r+",1.0)).xy";
  252. break;
  253. }
  254. };
  255. case SL::OP_ASSIGN:
  256. case SL::OP_ASSIGN_ADD:
  257. case SL::OP_ASSIGN_SUB:
  258. case SL::OP_ASSIGN_DIV:
  259. code="("+dump_node_code(onode->arguments[0],p_level,true)+_opstr(onode->op)+dump_node_code(onode->arguments[1],p_level)+")";
  260. break;
  261. case SL::OP_MUL:
  262. if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT4 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC3) {
  263. code="("+dump_node_code(onode->arguments[0],p_level)+"*vec4("+dump_node_code(onode->arguments[1],p_level)+",1.0)).xyz";
  264. break;
  265. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC3 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT4) {
  266. code="(vec4("+dump_node_code(onode->arguments[0],p_level)+",1.0)*"+dump_node_code(onode->arguments[1],p_level)+").xyz";
  267. break;
  268. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT4 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC2) {
  269. code="("+dump_node_code(onode->arguments[0],p_level)+"*vec4("+dump_node_code(onode->arguments[1],p_level)+",1.0,1.0)).xyz";
  270. break;
  271. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC2 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT4) {
  272. code="(vec4("+dump_node_code(onode->arguments[0],p_level)+",1.0,1.0)*"+dump_node_code(onode->arguments[1],p_level)+").xyz";
  273. break;
  274. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT3 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC2) {
  275. code="("+dump_node_code(onode->arguments[0],p_level)+"*vec3("+dump_node_code(onode->arguments[1],p_level)+",1.0)).xy";
  276. break;
  277. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC2 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT3) {
  278. code="(vec3("+dump_node_code(onode->arguments[0],p_level)+",1.0)*"+dump_node_code(onode->arguments[1],p_level)+").xy";
  279. break;
  280. }
  281. case SL::OP_ADD:
  282. case SL::OP_SUB:
  283. case SL::OP_DIV:
  284. case SL::OP_CMP_EQ:
  285. case SL::OP_CMP_NEQ:
  286. case SL::OP_CMP_LEQ:
  287. case SL::OP_CMP_GEQ:
  288. case SL::OP_CMP_LESS:
  289. case SL::OP_CMP_GREATER:
  290. case SL::OP_CMP_OR:
  291. case SL::OP_CMP_AND:
  292. //handle binary
  293. code="("+dump_node_code(onode->arguments[0],p_level)+_opstr(onode->op)+dump_node_code(onode->arguments[1],p_level)+")";
  294. break;
  295. case SL::OP_NEG:
  296. case SL::OP_NOT:
  297. //handle unary
  298. code=_opstr(onode->op)+dump_node_code(onode->arguments[0],p_level);
  299. break;
  300. case SL::OP_CONSTRUCT:
  301. case SL::OP_CALL: {
  302. String callfunc=dump_node_code(onode->arguments[0],p_level);
  303. code=callfunc+"(";
  304. /*if (callfunc=="mat4") {
  305. //fix constructor for mat4
  306. for(int i=1;i<onode->arguments.size();i++) {
  307. if (i>1)
  308. code+=", ";
  309. //transform
  310. code+="vec4( "+dump_node_code(onode->arguments[i],p_level)+(i==4?",1.0)":",0.0)");
  311. }
  312. } else*/ if (callfunc=="tex") {
  313. code="texture2D( "+dump_node_code(onode->arguments[1],p_level)+","+dump_node_code(onode->arguments[2],p_level)+")";
  314. break;
  315. } else if (callfunc=="texcube") {
  316. code="(textureCube( "+dump_node_code(onode->arguments[1],p_level)+",("+dump_node_code(onode->arguments[2],p_level)+")).xyz";
  317. break;
  318. } else if (callfunc=="texscreen") {
  319. //create the call to sample the screen, and clamp it
  320. uses_texscreen=true;
  321. code="(texture2D( texscreen_tex, clamp(("+dump_node_code(onode->arguments[1],p_level)+").xy*texscreen_screen_mult,texscreen_screen_clamp.xy,texscreen_screen_clamp.zw))).rgb";
  322. //code="(texture2D( screen_texture, ("+dump_node_code(onode->arguments[1],p_level)+").xy).rgb";
  323. break;
  324. } else if (callfunc=="texpos") {
  325. //create the call to sample the screen, and clamp it
  326. uses_texpos=true;
  327. code="get_texpos("+dump_node_code(onode->arguments[1],p_level)+"";
  328. // code="get_texpos(gl_ProjectionMatrixInverse * texture2D( depth_texture, clamp(("+dump_node_code(onode->arguments[1],p_level)+").xy,vec2(0.0),vec2(1.0))*gl_LightSource[5].specular.zw+gl_LightSource[5].specular.xy)";
  329. //code="(texture2D( screen_texture, ("+dump_node_code(onode->arguments[1],p_level)+").xy).rgb";
  330. break;
  331. } else {
  332. for(int i=1;i<onode->arguments.size();i++) {
  333. if (i>1)
  334. code+=", ";
  335. //transform
  336. code+=dump_node_code(onode->arguments[i],p_level);
  337. }
  338. }
  339. code+=")";
  340. break;
  341. } break;
  342. default: {}
  343. }
  344. } break;
  345. case SL::Node::TYPE_CONTROL_FLOW: {
  346. SL::ControlFlowNode *cfnode=(SL::ControlFlowNode*)p_node;
  347. if (cfnode->flow_op==SL::FLOW_OP_IF) {
  348. code+="if ("+dump_node_code(cfnode->statements[0],p_level)+") {"ENDL;
  349. code+=dump_node_code(cfnode->statements[1],p_level+1);
  350. if (cfnode->statements.size()==3) {
  351. code+="} else {"ENDL;
  352. code+=dump_node_code(cfnode->statements[2],p_level+1);
  353. }
  354. code+="}"ENDL;
  355. } else if (cfnode->flow_op==SL::FLOW_OP_RETURN) {
  356. if (cfnode->statements.size()) {
  357. code="return "+dump_node_code(cfnode->statements[0],p_level);
  358. } else {
  359. code="return";
  360. }
  361. }
  362. } break;
  363. case SL::Node::TYPE_MEMBER: {
  364. SL::MemberNode *mnode=(SL::MemberNode*)p_node;
  365. String m;
  366. if (mnode->basetype==SL::TYPE_MAT4) {
  367. if (mnode->name=="x")
  368. m="[0]";
  369. else if (mnode->name=="y")
  370. m="[1]";
  371. else if (mnode->name=="z")
  372. m="[2]";
  373. else if (mnode->name=="w")
  374. m="[3]";
  375. } else if (mnode->basetype==SL::TYPE_MAT3) {
  376. if (mnode->name=="x")
  377. m="[0]";
  378. else if (mnode->name=="y")
  379. m="[1]";
  380. else if (mnode->name=="z")
  381. m="[2]";
  382. } else {
  383. m="."+mnode->name;
  384. }
  385. code=dump_node_code(mnode->owner,p_level)+m;
  386. } break;
  387. }
  388. return code;
  389. }
  390. Error ShaderCompilerGLES2::compile_node(SL::ProgramNode *p_program) {
  391. // feed the local replace table and global code
  392. global_code="";
  393. // uniforms first!
  394. int ubase=0;
  395. if (uniforms)
  396. ubase=uniforms->size();
  397. for(Map<StringName,SL::Uniform>::Element *E=p_program->uniforms.front();E;E=E->next()) {
  398. String uline="uniform "+_typestr(E->get().type)+" _"+E->key().operator String()+";"ENDL;
  399. global_code+=uline;
  400. if (uniforms) {
  401. //if (uniforms->has(E->key())) {
  402. // //repeated uniform, error
  403. // ERR_EXPLAIN("Uniform already exists from other shader: "+String(E->key()));
  404. // ERR_FAIL_COND_V(uniforms->has(E->key()),ERR_ALREADY_EXISTS);
  405. //
  406. // }
  407. SL::Uniform u = E->get();
  408. u.order+=ubase;
  409. uniforms->insert(E->key(),u);
  410. }
  411. }
  412. for(int i=0;i<p_program->functions.size();i++) {
  413. SL::FunctionNode *fnode=p_program->functions[i].function;
  414. StringName funcname=fnode->name;
  415. String newfuncname=replace_string(funcname);
  416. String header;
  417. header=_typestr(fnode->return_type)+" "+newfuncname+"(";
  418. for(int i=0;i<fnode->arguments.size();i++) {
  419. if (i>0)
  420. header+=", ";
  421. header+=_typestr(fnode->arguments[i].type)+" "+replace_string(fnode->arguments[i].name);
  422. }
  423. header+=") {"ENDL;
  424. String fcode=header;
  425. fcode+=dump_node_code(fnode->body,1);
  426. fcode+="}"ENDL;
  427. global_code+=fcode;
  428. }
  429. /* for(Map<StringName,SL::DataType>::Element *E=p_program->preexisting_variables.front();E;E=E->next()) {
  430. StringName varname=E->key();
  431. String newvarname=replace_string(varname);
  432. global_code+="uniform "+_typestr(E->get())+" "+newvarname+";"ENDL;
  433. }*/
  434. code=dump_node_code(p_program,0);
  435. #ifdef DEBUG_SHADER_ENABLED
  436. print_line("GLOBAL CODE:\n\n");
  437. print_line(global_code);
  438. global_code=global_code.replace("\n","");
  439. print_line("CODE:\n\n");
  440. print_line(code);
  441. code=code.replace("\n","");
  442. #endif
  443. return OK;
  444. }
  445. Error ShaderCompilerGLES2::create_glsl_120_code(void *p_str,SL::ProgramNode *p_program) {
  446. ShaderCompilerGLES2 *compiler=(ShaderCompilerGLES2*)p_str;
  447. return compiler->compile_node(p_program);
  448. }
  449. String ShaderCompilerGLES2::replace_string(const StringName& p_string) {
  450. Map<StringName,StringName>::Element *E=NULL;
  451. E=replace_table.find(p_string);
  452. if (E)
  453. return E->get();
  454. E=mode_replace_table[type].find(p_string);
  455. if (E)
  456. return E->get();
  457. return "_"+p_string.operator String();
  458. }
  459. Error ShaderCompilerGLES2::compile(const String& p_code, ShaderLanguage::ShaderType p_type, String& r_code_line, String& r_globals_line, Flags& r_flags, Map<StringName,ShaderLanguage::Uniform> *r_uniforms) {
  460. uses_texscreen=false;
  461. uses_texpos=false;
  462. uses_alpha=false;
  463. uses_discard=false;
  464. uses_screen_uv=false;
  465. uses_light=false;
  466. uses_time=false;
  467. uses_normalmap=false;
  468. uses_normal=false;
  469. uses_texpixel_size=false;
  470. vertex_code_writes_vertex=false;
  471. uniforms=r_uniforms;
  472. flags=&r_flags;
  473. r_flags.use_color_interp=false;
  474. r_flags.use_uv_interp=false;
  475. r_flags.use_uv2_interp=false;
  476. r_flags.use_tangent_interp=false;
  477. r_flags.use_var1_interp=false;
  478. r_flags.use_var2_interp=false;
  479. r_flags.uses_normalmap=false;
  480. r_flags.uses_normal=false;
  481. String error;
  482. int errline,errcol;
  483. type=p_type;
  484. Error err = SL::compile(p_code,p_type,create_glsl_120_code,this,&error,&errline,&errcol);
  485. if (err) {
  486. print_line("***Error precompiling shader: "+error);
  487. print_line("error "+itos(errline)+":"+itos(errcol));
  488. return err;
  489. }
  490. r_flags.uses_alpha=uses_alpha;
  491. r_flags.uses_texscreen=uses_texscreen;
  492. r_flags.uses_texpos=uses_texpos;
  493. r_flags.vertex_code_writes_vertex=vertex_code_writes_vertex;
  494. r_flags.uses_discard=uses_discard;
  495. r_flags.uses_screen_uv=uses_screen_uv;
  496. r_flags.uses_light=uses_light;
  497. r_flags.uses_time=uses_time;
  498. r_flags.uses_normalmap=uses_normalmap;
  499. r_flags.uses_normal=uses_normalmap;
  500. r_flags.uses_texpixel_size=uses_texpixel_size;
  501. r_code_line=code;
  502. r_globals_line=global_code;
  503. return OK;
  504. }
  505. ShaderCompilerGLES2::ShaderCompilerGLES2() {
  506. replace_table["bool"]= "bool";
  507. replace_table["float" ]= "float";
  508. replace_table["vec2" ]= "vec2";
  509. replace_table["vec3" ]= "vec3";
  510. replace_table["vec4" ]= "vec4";
  511. replace_table["mat3" ]= "mat3";
  512. replace_table["mat4" ]= "mat4";
  513. replace_table["texture" ]= "sampler2D";
  514. replace_table["cubemap" ]= "samplerCube";
  515. replace_table["sin"]= "sin";
  516. replace_table["cos" ]= "cos";
  517. replace_table["tan" ]= "tan";
  518. replace_table["asin" ]= "asin";
  519. replace_table["acos" ]= "acos";
  520. replace_table["atan" ]= "atan";
  521. replace_table["atan2"]= "atan";
  522. replace_table["sinh" ]= "sinh";
  523. replace_table["cosh" ]= "cosh";
  524. replace_table["tanh" ]= "tanh";
  525. replace_table["pow" ]= "pow";
  526. replace_table["exp" ]= "exp";
  527. replace_table["log" ]= "log";
  528. replace_table["sqrt"]= "sqrt";
  529. replace_table["abs" ]= "abs";
  530. replace_table["sign"]= "sign";
  531. replace_table["floor"]= "floor";
  532. replace_table["trunc"]= "trunc";
  533. #ifdef GLEW_ENABLED
  534. replace_table["round"]= "roundfix";
  535. #else
  536. replace_table["round"]= "round";
  537. #endif
  538. replace_table["ceil" ]= "ceil";
  539. replace_table["fract"]= "fract";
  540. replace_table["mod" ]= "mod";
  541. replace_table["min" ]= "min";
  542. replace_table["max"]= "max";
  543. replace_table["clamp"]= "clamp";
  544. replace_table["mix" ]= "mix";
  545. replace_table["step" ]= "step";
  546. replace_table["smoothstep" ]= "smoothstep";
  547. replace_table["length"]= "length";
  548. replace_table["distance"]= "distance";
  549. replace_table["dot" ]= "dot";
  550. replace_table["cross" ]="cross";
  551. replace_table["normalize"]= "normalize";
  552. replace_table["reflect"]= "reflect";
  553. replace_table["tex"]= "tex";
  554. replace_table["texa"]= "texa";
  555. replace_table["tex2"]= "tex2";
  556. replace_table["texcube"]= "textureCube";
  557. replace_table["texscreen"]= "texscreen";
  558. replace_table["texpos"]= "texpos";
  559. mode_replace_table[0]["SRC_VERTEX"]="vertex_in.xyz";
  560. mode_replace_table[0]["SRC_NORMAL"]="normal_in";
  561. mode_replace_table[0]["SRC_TANGENT"]="tangent_in";
  562. mode_replace_table[0]["SRC_BINORMALF"]="binormalf";
  563. mode_replace_table[0]["VERTEX"]="vertex_interp";
  564. mode_replace_table[0]["NORMAL"]="normal_interp";
  565. mode_replace_table[0]["TANGENT"]="tangent_interp";
  566. mode_replace_table[0]["BINORMAL"]="binormal_interp";
  567. mode_replace_table[0]["UV"]="uv_interp.xy";
  568. mode_replace_table[0]["UV2"]="uv_interp.zw";
  569. mode_replace_table[0]["COLOR"]="color_interp";
  570. //@TODO convert to glsl stuff
  571. mode_replace_table[0]["SPEC_EXP"]="vertex_specular_exp";
  572. mode_replace_table[0]["WORLD_MATRIX"]="world_transform";
  573. mode_replace_table[0]["INV_CAMERA_MATRIX"]="camera_inverse_transform";
  574. mode_replace_table[0]["PROJECTION_MATRIX"]="projection_transform";
  575. mode_replace_table[0]["MODELVIEW_MATRIX"]="modelview";
  576. mode_replace_table[0]["POINT_SIZE"]="gl_PointSize";
  577. mode_replace_table[0]["VAR1"]="var1_interp";
  578. mode_replace_table[0]["VAR2"]="var2_interp";
  579. // mode_replace_table[0]["SCREEN_POS"]="SCREEN_POS";
  580. // mode_replace_table[0]["SCREEN_SIZE"]="SCREEN_SIZE";
  581. mode_replace_table[0]["INSTANCE_ID"]="instance_id";
  582. mode_replace_table[0]["TIME"]="time";
  583. mode_replace_table[1]["VERTEX"]="vertex";
  584. //mode_replace_table[1]["POSITION"]="IN_POSITION";
  585. mode_replace_table[1]["NORMAL"]="normal";
  586. mode_replace_table[1]["TANGENT"]="tangent";
  587. mode_replace_table[1]["POSITION"]="gl_Position";
  588. mode_replace_table[1]["BINORMAL"]="binormal";
  589. mode_replace_table[1]["NORMALMAP"]="normalmap";
  590. mode_replace_table[1]["NORMALMAP_DEPTH"]="normaldepth";
  591. mode_replace_table[1]["VAR1"]="var1_interp";
  592. mode_replace_table[1]["VAR2"]="var2_interp";
  593. mode_replace_table[1]["UV"]="uv";
  594. mode_replace_table[1]["UV2"]="uv2";
  595. mode_replace_table[1]["SCREEN_UV"]="screen_uv";
  596. mode_replace_table[1]["VAR1"]="var1_interp";
  597. mode_replace_table[1]["VAR2"]="var2_interp";
  598. mode_replace_table[1]["COLOR"]="color";
  599. mode_replace_table[1]["DIFFUSE"]="diffuse.rgb";
  600. mode_replace_table[1]["DIFFUSE_ALPHA"]="diffuse";
  601. mode_replace_table[1]["SPECULAR"]="specular";
  602. mode_replace_table[1]["EMISSION"]="emission";
  603. mode_replace_table[1]["SHADE_PARAM"]="shade_param";
  604. mode_replace_table[1]["SPEC_EXP"]="specular_exp";
  605. mode_replace_table[1]["GLOW"]="glow";
  606. mode_replace_table[1]["DISCARD"]="discard_";
  607. mode_replace_table[1]["POINT_COORD"]="gl_PointCoord";
  608. mode_replace_table[1]["INV_CAMERA_MATRIX"]="camera_inverse_transform";
  609. //mode_replace_table[1]["SCREEN_POS"]="SCREEN_POS";
  610. //mode_replace_table[1]["SCREEN_TEXEL_SIZE"]="SCREEN_TEXEL_SIZE";
  611. mode_replace_table[1]["TIME"]="time";
  612. //////////////
  613. mode_replace_table[2]["NORMAL"]="normal";
  614. //mode_replace_table[2]["POSITION"]="IN_POSITION";
  615. mode_replace_table[2]["LIGHT_DIR"]="light_dir";
  616. mode_replace_table[2]["LIGHT_DIFFUSE"]="light_diffuse";
  617. mode_replace_table[2]["LIGHT_SPECULAR"]="light_specular";
  618. mode_replace_table[2]["EYE_VEC"]="eye_vec";
  619. mode_replace_table[2]["DIFFUSE"]="mdiffuse";
  620. mode_replace_table[2]["SPECULAR"]="specular";
  621. mode_replace_table[2]["SPECULAR_EXP"]="specular_exp";
  622. mode_replace_table[2]["SHADE_PARAM"]="shade_param";
  623. mode_replace_table[2]["LIGHT"]="light";
  624. mode_replace_table[2]["POINT_COORD"]="gl_PointCoord";
  625. mode_replace_table[2]["TIME"]="time";
  626. mode_replace_table[3]["SRC_VERTEX"]="src_vtx";
  627. mode_replace_table[3]["VERTEX"]="outvec.xy";
  628. mode_replace_table[3]["UV"]="uv_interp";
  629. mode_replace_table[3]["COLOR"]="color_interp";
  630. mode_replace_table[3]["VAR1"]="var1_interp";
  631. mode_replace_table[3]["VAR2"]="var2_interp";
  632. mode_replace_table[3]["POINT_SIZE"]="gl_PointSize";
  633. mode_replace_table[3]["WORLD_MATRIX"]="modelview_matrix";
  634. mode_replace_table[3]["PROJECTION_MATRIX"]="projection_matrix";
  635. mode_replace_table[3]["EXTRA_MATRIX"]="extra_matrix";
  636. mode_replace_table[3]["TIME"]="time";
  637. mode_replace_table[4]["POSITION"]="gl_Position";
  638. mode_replace_table[4]["NORMAL"]="normal";
  639. mode_replace_table[4]["UV"]="uv_interp";
  640. mode_replace_table[4]["SRC_COLOR"]="color_interp";
  641. mode_replace_table[4]["COLOR"]="color";
  642. mode_replace_table[4]["TEXTURE"]="texture";
  643. mode_replace_table[4]["TEXTURE_PIXEL_SIZE"]="texpixel_size";
  644. mode_replace_table[4]["VAR1"]="var1_interp";
  645. mode_replace_table[4]["VAR2"]="var2_interp";
  646. mode_replace_table[4]["SCREEN_UV"]="screen_uv";
  647. mode_replace_table[4]["POINT_COORD"]="gl_PointCoord";
  648. mode_replace_table[4]["TIME"]="time";
  649. mode_replace_table[5]["SRC_COLOR"]="color";
  650. mode_replace_table[5]["COLOR"]="color";
  651. mode_replace_table[5]["NORMAL"]="normal";
  652. mode_replace_table[5]["LIGHT_DIR"]="light_dir";
  653. mode_replace_table[5]["LIGHT_DISTANCE"]="light_distance";
  654. mode_replace_table[5]["LIGHT"]="light";
  655. mode_replace_table[5]["POINT_COORD"]="gl_PointCoord";
  656. mode_replace_table[5]["TIME"]="time";
  657. //mode_replace_table[2]["SCREEN_POS"]="SCREEN_POS";
  658. //mode_replace_table[2]["SCREEN_TEXEL_SIZE"]="SCREEN_TEXEL_SIZE";
  659. out_vertex_name="VERTEX";
  660. vname_discard="DISCARD";
  661. vname_screen_uv="SCREEN_UV";
  662. vname_diffuse_alpha="DIFFUSE_ALPHA";
  663. vname_color_interp="COLOR";
  664. vname_uv_interp="UV";
  665. vname_uv2_interp="UV2";
  666. vname_tangent_interp="TANGENT";
  667. vname_binormal_interp="BINORMAL";
  668. vname_var1_interp="VAR1";
  669. vname_var2_interp="VAR2";
  670. vname_vertex="VERTEX";
  671. vname_light="LIGHT";
  672. vname_time="TIME";
  673. vname_normalmap="NORMALMAP";
  674. vname_normal="NORMAL";
  675. vname_texpixel_size="TEXTURE_PIXEL_SIZE";
  676. }