shader_compiler_gles2.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728
  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. for(Map<StringName,SL::DataType>::Element *E=bnode->variables.front();E;E=E->next()) {
  106. code+=_mktab(p_level)+_typestr(E->value())+" "+replace_string(E->key())+";"ENDL;
  107. }
  108. for(int i=0;i<bnode->statements.size();i++) {
  109. code+=_mktab(p_level)+dump_node_code(bnode->statements[i],p_level)+";"ENDL;
  110. }
  111. } break;
  112. case SL::Node::TYPE_VARIABLE: {
  113. SL::VariableNode *vnode=(SL::VariableNode*)p_node;
  114. if (type==ShaderLanguage::SHADER_MATERIAL_VERTEX) {
  115. if (vnode->name==vname_vertex && p_assign_left) {
  116. vertex_code_writes_vertex=true;
  117. }
  118. if (vnode->name==vname_color_interp) {
  119. flags->use_color_interp=true;
  120. }
  121. if (vnode->name==vname_uv_interp) {
  122. flags->use_uv_interp=true;
  123. }
  124. if (vnode->name==vname_uv2_interp) {
  125. flags->use_uv2_interp=true;
  126. }
  127. if (vnode->name==vname_var1_interp) {
  128. flags->use_var1_interp=true;
  129. }
  130. if (vnode->name==vname_var2_interp) {
  131. flags->use_var2_interp=true;
  132. }
  133. if (vnode->name==vname_tangent_interp || vnode->name==vname_binormal_interp) {
  134. flags->use_tangent_interp=true;
  135. }
  136. }
  137. if (type==ShaderLanguage::SHADER_MATERIAL_FRAGMENT) {
  138. if (vnode->name==vname_discard) {
  139. uses_discard=true;
  140. }
  141. if (vnode->name==vname_screen_uv) {
  142. uses_screen_uv=true;
  143. }
  144. if (vnode->name==vname_diffuse_alpha && p_assign_left) {
  145. uses_alpha=true;
  146. }
  147. if (vnode->name==vname_color_interp) {
  148. flags->use_color_interp=true;
  149. }
  150. if (vnode->name==vname_uv_interp) {
  151. flags->use_uv_interp=true;
  152. }
  153. if (vnode->name==vname_uv2_interp) {
  154. flags->use_uv2_interp=true;
  155. }
  156. if (vnode->name==vname_var1_interp) {
  157. flags->use_var1_interp=true;
  158. }
  159. if (vnode->name==vname_var2_interp) {
  160. flags->use_var2_interp=true;
  161. }
  162. if (vnode->name==vname_tangent_interp || vnode->name==vname_binormal_interp) {
  163. flags->use_tangent_interp=true;
  164. }
  165. }
  166. if (type==ShaderLanguage::SHADER_MATERIAL_LIGHT) {
  167. if (vnode->name==vname_light) {
  168. uses_light=true;
  169. }
  170. }
  171. if (vnode->name==vname_time) {
  172. uses_time=true;
  173. }
  174. code=replace_string(vnode->name);
  175. } break;
  176. case SL::Node::TYPE_CONSTANT: {
  177. SL::ConstantNode *cnode=(SL::ConstantNode*)p_node;
  178. switch(cnode->datatype) {
  179. case SL::TYPE_BOOL: code=cnode->value.operator bool()?"true":"false"; break;
  180. case SL::TYPE_FLOAT: code=_mknum(cnode->value); break; //force zeros, so GLSL doesn't confuse with integer.
  181. case SL::TYPE_VEC2: { Vector2 v = cnode->value; code="vec2("+_mknum(v.x)+", "+_mknum(v.y)+")"; } break;
  182. case SL::TYPE_VEC3: { Vector3 v = cnode->value; code="vec3("+_mknum(v.x)+", "+_mknum(v.y)+", "+_mknum(v.z)+")"; } break;
  183. 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;
  184. 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;
  185. 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;
  186. default: code="<error: "+Variant::get_type_name(cnode->value.get_type())+" ("+itos(cnode->datatype)+">";
  187. }
  188. } break;
  189. case SL::Node::TYPE_OPERATOR: {
  190. SL::OperatorNode *onode=(SL::OperatorNode*)p_node;
  191. switch(onode->op) {
  192. case SL::OP_ASSIGN_MUL: {
  193. if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC3 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT4) {
  194. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  195. String mul_r=dump_node_code(onode->arguments[1],p_level);
  196. code=mul_l+"=(vec4("+mul_l+",1.0)*("+mul_r+")).xyz";
  197. break;
  198. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT4 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC3) {
  199. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  200. String mul_r=dump_node_code(onode->arguments[1],p_level);
  201. code=mul_l+"=(("+mul_l+")*vec4("+mul_r+",1.0)).xyz";
  202. break;
  203. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC2 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT4) {
  204. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  205. String mul_r=dump_node_code(onode->arguments[1],p_level);
  206. code=mul_l+"=(vec4("+mul_l+",1.0,1.0)*("+mul_r+")).xy";
  207. break;
  208. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT4 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC2) {
  209. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  210. String mul_r=dump_node_code(onode->arguments[1],p_level);
  211. code=mul_l+"=(("+mul_l+")*vec4("+mul_r+",1.0,1.0)).xy";
  212. break;
  213. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC2 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT3) {
  214. String mul_l=dump_node_code(onode->arguments[0],p_level,true);
  215. String mul_r=dump_node_code(onode->arguments[1],p_level);
  216. code=mul_l+"=(("+mul_l+")*vec3("+mul_r+",1.0)).xy";
  217. break;
  218. }
  219. };
  220. case SL::OP_ASSIGN:
  221. case SL::OP_ASSIGN_ADD:
  222. case SL::OP_ASSIGN_SUB:
  223. case SL::OP_ASSIGN_DIV:
  224. code="("+dump_node_code(onode->arguments[0],p_level,true)+_opstr(onode->op)+dump_node_code(onode->arguments[1],p_level)+")";
  225. break;
  226. case SL::OP_MUL:
  227. if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT4 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC3) {
  228. code="("+dump_node_code(onode->arguments[0],p_level)+"*vec4("+dump_node_code(onode->arguments[1],p_level)+",1.0)).xyz";
  229. break;
  230. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC3 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT4) {
  231. code="(vec4("+dump_node_code(onode->arguments[0],p_level)+",1.0)*"+dump_node_code(onode->arguments[1],p_level)+").xyz";
  232. break;
  233. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT4 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC2) {
  234. code="("+dump_node_code(onode->arguments[0],p_level)+"*vec4("+dump_node_code(onode->arguments[1],p_level)+",1.0,1.0)).xyz";
  235. break;
  236. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC2 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT4) {
  237. code="(vec4("+dump_node_code(onode->arguments[0],p_level)+",1.0,1.0)*"+dump_node_code(onode->arguments[1],p_level)+").xyz";
  238. break;
  239. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_MAT3 && onode->arguments[1]->get_datatype()==SL::TYPE_VEC2) {
  240. code="("+dump_node_code(onode->arguments[0],p_level)+"*vec3("+dump_node_code(onode->arguments[1],p_level)+",1.0)).xy";
  241. break;
  242. } else if (onode->arguments[0]->get_datatype()==SL::TYPE_VEC2 && onode->arguments[1]->get_datatype()==SL::TYPE_MAT3) {
  243. code="(vec3("+dump_node_code(onode->arguments[0],p_level)+",1.0)*"+dump_node_code(onode->arguments[1],p_level)+").xy";
  244. break;
  245. }
  246. case SL::OP_ADD:
  247. case SL::OP_SUB:
  248. case SL::OP_DIV:
  249. case SL::OP_CMP_EQ:
  250. case SL::OP_CMP_NEQ:
  251. case SL::OP_CMP_LEQ:
  252. case SL::OP_CMP_GEQ:
  253. case SL::OP_CMP_LESS:
  254. case SL::OP_CMP_GREATER:
  255. case SL::OP_CMP_OR:
  256. case SL::OP_CMP_AND:
  257. //handle binary
  258. code="("+dump_node_code(onode->arguments[0],p_level)+_opstr(onode->op)+dump_node_code(onode->arguments[1],p_level)+")";
  259. break;
  260. case SL::OP_NEG:
  261. case SL::OP_NOT:
  262. //handle unary
  263. code=_opstr(onode->op)+dump_node_code(onode->arguments[0],p_level);
  264. break;
  265. case SL::OP_CONSTRUCT:
  266. case SL::OP_CALL: {
  267. String callfunc=dump_node_code(onode->arguments[0],p_level);
  268. code=callfunc+"(";
  269. /*if (callfunc=="mat4") {
  270. //fix constructor for mat4
  271. for(int i=1;i<onode->arguments.size();i++) {
  272. if (i>1)
  273. code+=", ";
  274. //transform
  275. code+="vec4( "+dump_node_code(onode->arguments[i],p_level)+(i==4?",1.0)":",0.0)");
  276. }
  277. } else*/ if (callfunc=="tex") {
  278. code="texture2D( "+dump_node_code(onode->arguments[1],p_level)+","+dump_node_code(onode->arguments[2],p_level)+")";
  279. break;
  280. } else if (callfunc=="texcube") {
  281. code="(textureCube( "+dump_node_code(onode->arguments[1],p_level)+",("+dump_node_code(onode->arguments[2],p_level)+")).xyz";
  282. break;
  283. } else if (callfunc=="texscreen") {
  284. //create the call to sample the screen, and clamp it
  285. uses_texscreen=true;
  286. code="(texture2D( texscreen_tex, min(("+dump_node_code(onode->arguments[1],p_level)+").xy*texscreen_screen_mult,texscreen_screen_mult))).rgb";
  287. //code="(texture2D( screen_texture, ("+dump_node_code(onode->arguments[1],p_level)+").xy).rgb";
  288. break;
  289. } else if (callfunc=="texpos") {
  290. //create the call to sample the screen, and clamp it
  291. uses_texpos=true;
  292. code="get_texpos("+dump_node_code(onode->arguments[1],p_level)+"";
  293. // 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)";
  294. //code="(texture2D( screen_texture, ("+dump_node_code(onode->arguments[1],p_level)+").xy).rgb";
  295. break;
  296. } else {
  297. for(int i=1;i<onode->arguments.size();i++) {
  298. if (i>1)
  299. code+=", ";
  300. //transform
  301. code+=dump_node_code(onode->arguments[i],p_level);
  302. }
  303. }
  304. code+=")";
  305. break;
  306. } break;
  307. default: {}
  308. }
  309. } break;
  310. case SL::Node::TYPE_CONTROL_FLOW: {
  311. SL::ControlFlowNode *cfnode=(SL::ControlFlowNode*)p_node;
  312. if (cfnode->flow_op==SL::FLOW_OP_IF) {
  313. code+="if ("+dump_node_code(cfnode->statements[0],p_level)+") {"ENDL;
  314. code+=dump_node_code(cfnode->statements[1],p_level+1);
  315. if (cfnode->statements.size()==3) {
  316. code+="} else {"ENDL;
  317. code+=dump_node_code(cfnode->statements[2],p_level+1);
  318. }
  319. code+="}"ENDL;
  320. } else if (cfnode->flow_op==SL::FLOW_OP_RETURN) {
  321. if (cfnode->statements.size()) {
  322. code="return "+dump_node_code(cfnode->statements[0],p_level);
  323. } else {
  324. code="return";
  325. }
  326. }
  327. } break;
  328. case SL::Node::TYPE_MEMBER: {
  329. SL::MemberNode *mnode=(SL::MemberNode*)p_node;
  330. String m;
  331. if (mnode->basetype==SL::TYPE_MAT4) {
  332. if (mnode->name=="x")
  333. m="[0]";
  334. else if (mnode->name=="y")
  335. m="[1]";
  336. else if (mnode->name=="z")
  337. m="[2]";
  338. else if (mnode->name=="w")
  339. m="[3]";
  340. } else if (mnode->basetype==SL::TYPE_MAT3) {
  341. if (mnode->name=="x")
  342. m="[0]";
  343. else if (mnode->name=="y")
  344. m="[1]";
  345. else if (mnode->name=="z")
  346. m="[2]";
  347. } else {
  348. m="."+mnode->name;
  349. }
  350. code=dump_node_code(mnode->owner,p_level)+m;
  351. } break;
  352. }
  353. return code;
  354. }
  355. Error ShaderCompilerGLES2::compile_node(SL::ProgramNode *p_program) {
  356. // feed the local replace table and global code
  357. global_code="";
  358. // uniforms first!
  359. int ubase=0;
  360. if (uniforms)
  361. ubase=uniforms->size();
  362. for(Map<StringName,SL::Uniform>::Element *E=p_program->uniforms.front();E;E=E->next()) {
  363. String uline="uniform "+_typestr(E->get().type)+" _"+E->key().operator String()+";"ENDL;
  364. global_code+=uline;
  365. if (uniforms) {
  366. //if (uniforms->has(E->key())) {
  367. // //repeated uniform, error
  368. // ERR_EXPLAIN("Uniform already exists from other shader: "+String(E->key()));
  369. // ERR_FAIL_COND_V(uniforms->has(E->key()),ERR_ALREADY_EXISTS);
  370. //
  371. // }
  372. SL::Uniform u = E->get();
  373. u.order+=ubase;
  374. uniforms->insert(E->key(),u);
  375. }
  376. }
  377. for(int i=0;i<p_program->functions.size();i++) {
  378. SL::FunctionNode *fnode=p_program->functions[i].function;
  379. StringName funcname=fnode->name;
  380. String newfuncname=replace_string(funcname);
  381. String header;
  382. header=_typestr(fnode->return_type)+" "+newfuncname+"(";
  383. for(int i=0;i<fnode->arguments.size();i++) {
  384. if (i>0)
  385. header+=", ";
  386. header+=_typestr(fnode->arguments[i].type)+" "+replace_string(fnode->arguments[i].name);
  387. }
  388. header+=") {"ENDL;
  389. String fcode=header;
  390. fcode+=dump_node_code(fnode->body,1);
  391. fcode+="}"ENDL;
  392. global_code+=fcode;
  393. }
  394. /* for(Map<StringName,SL::DataType>::Element *E=p_program->preexisting_variables.front();E;E=E->next()) {
  395. StringName varname=E->key();
  396. String newvarname=replace_string(varname);
  397. global_code+="uniform "+_typestr(E->get())+" "+newvarname+";"ENDL;
  398. }*/
  399. code=dump_node_code(p_program,0);
  400. #ifdef DEBUG_SHADER_ENABLED
  401. print_line("GLOBAL CODE:\n\n");
  402. print_line(global_code);
  403. global_code=global_code.replace("\n","");
  404. print_line("CODE:\n\n");
  405. print_line(code);
  406. code=code.replace("\n","");
  407. #endif
  408. return OK;
  409. }
  410. Error ShaderCompilerGLES2::create_glsl_120_code(void *p_str,SL::ProgramNode *p_program) {
  411. ShaderCompilerGLES2 *compiler=(ShaderCompilerGLES2*)p_str;
  412. return compiler->compile_node(p_program);
  413. }
  414. String ShaderCompilerGLES2::replace_string(const StringName& p_string) {
  415. Map<StringName,StringName>::Element *E=NULL;
  416. E=replace_table.find(p_string);
  417. if (E)
  418. return E->get();
  419. E=mode_replace_table[type].find(p_string);
  420. if (E)
  421. return E->get();
  422. return "_"+p_string.operator String();
  423. }
  424. 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) {
  425. uses_texscreen=false;
  426. uses_texpos=false;
  427. uses_alpha=false;
  428. uses_discard=false;
  429. uses_screen_uv=false;
  430. uses_light=false;
  431. uses_time=false;
  432. vertex_code_writes_vertex=false;
  433. uniforms=r_uniforms;
  434. flags=&r_flags;
  435. r_flags.use_color_interp=false;
  436. r_flags.use_uv_interp=false;
  437. r_flags.use_uv2_interp=false;
  438. r_flags.use_tangent_interp=false;
  439. r_flags.use_var1_interp=false;
  440. r_flags.use_var2_interp=false;
  441. String error;
  442. int errline,errcol;
  443. type=p_type;
  444. Error err = SL::compile(p_code,p_type,create_glsl_120_code,this,&error,&errline,&errcol);
  445. if (err) {
  446. print_line("***Error precompiling shader: "+error);
  447. print_line("error "+itos(errline)+":"+itos(errcol));
  448. return err;
  449. }
  450. r_flags.uses_alpha=uses_alpha;
  451. r_flags.uses_texscreen=uses_texscreen;
  452. r_flags.uses_texpos=uses_texpos;
  453. r_flags.vertex_code_writes_vertex=vertex_code_writes_vertex;
  454. r_flags.uses_discard=uses_discard;
  455. r_flags.uses_screen_uv=uses_screen_uv;
  456. r_flags.uses_light=uses_light;
  457. r_flags.uses_time=uses_time;
  458. r_code_line=code;
  459. r_globals_line=global_code;
  460. return OK;
  461. }
  462. ShaderCompilerGLES2::ShaderCompilerGLES2() {
  463. replace_table["bool"]= "bool";
  464. replace_table["float" ]= "float";
  465. replace_table["vec2" ]= "vec2";
  466. replace_table["vec3" ]= "vec3";
  467. replace_table["vec4" ]= "vec4";
  468. replace_table["mat3" ]= "mat3";
  469. replace_table["mat4" ]= "mat4";
  470. replace_table["texture" ]= "sampler2D";
  471. replace_table["cubemap" ]= "samplerCube";
  472. replace_table["sin"]= "sin";
  473. replace_table["cos" ]= "cos";
  474. replace_table["tan" ]= "tan";
  475. replace_table["asin" ]= "asin";
  476. replace_table["acos" ]= "acos";
  477. replace_table["atan" ]= "atan";
  478. replace_table["atan2"]= "atan";
  479. replace_table["sinh" ]= "sinh";
  480. replace_table["cosh" ]= "cosh";
  481. replace_table["tanh" ]= "tanh";
  482. replace_table["pow" ]= "pow";
  483. replace_table["exp" ]= "exp";
  484. replace_table["log" ]= "log";
  485. replace_table["sqrt"]= "sqrt";
  486. replace_table["abs" ]= "abs";
  487. replace_table["sign"]= "sign";
  488. replace_table["floor"]= "floor";
  489. replace_table["trunc"]= "trunc";
  490. replace_table["round"]= "round";
  491. replace_table["ceil" ]= "ceil";
  492. replace_table["fract"]= "fract";
  493. replace_table["mod" ]= "mod";
  494. replace_table["min" ]= "min";
  495. replace_table["max"]= "max";
  496. replace_table["clamp"]= "clamp";
  497. replace_table["mix" ]= "mix";
  498. replace_table["step" ]= "step";
  499. replace_table["smoothstep" ]= "smoothstep";
  500. replace_table["length"]= "length";
  501. replace_table["distance"]= "distance";
  502. replace_table["dot" ]= "dot";
  503. replace_table["cross" ]="cross";
  504. replace_table["normalize"]= "normalize";
  505. replace_table["reflect"]= "reflect";
  506. replace_table["tex"]= "tex";
  507. replace_table["texa"]= "texa";
  508. replace_table["tex2"]= "tex2";
  509. replace_table["texcube"]= "textureCube";
  510. replace_table["texscreen"]= "texscreen";
  511. replace_table["texpos"]= "texpos";
  512. mode_replace_table[0]["SRC_VERTEX"]="vertex_in.xyz";
  513. mode_replace_table[0]["SRC_NORMAL"]="normal_in";
  514. mode_replace_table[0]["SRC_TANGENT"]="tangent_in";
  515. mode_replace_table[0]["SRC_BINORMALF"]="binormalf";
  516. mode_replace_table[0]["VERTEX"]="vertex_interp";
  517. mode_replace_table[0]["NORMAL"]="normal_interp";
  518. mode_replace_table[0]["TANGENT"]="tangent_interp";
  519. mode_replace_table[0]["BINORMAL"]="binormal_interp";
  520. mode_replace_table[0]["UV"]="uv_interp.xy";
  521. mode_replace_table[0]["UV2"]="uv_interp.zw";
  522. mode_replace_table[0]["COLOR"]="color_interp";
  523. //@TODO convert to glsl stuff
  524. mode_replace_table[0]["SPEC_EXP"]="vertex_specular_exp";
  525. mode_replace_table[0]["WORLD_MATRIX"]="world_transform";
  526. mode_replace_table[0]["INV_CAMERA_MATRIX"]="camera_inverse_transform";
  527. mode_replace_table[0]["PROJECTION_MATRIX"]="projection_transform";
  528. mode_replace_table[0]["MODELVIEW_MATRIX"]="modelview";
  529. mode_replace_table[0]["POINT_SIZE"]="gl_PointSize";
  530. mode_replace_table[0]["VAR1"]="var1_interp";
  531. mode_replace_table[0]["VAR2"]="var2_interp";
  532. // mode_replace_table[0]["SCREEN_POS"]="SCREEN_POS";
  533. // mode_replace_table[0]["SCREEN_SIZE"]="SCREEN_SIZE";
  534. mode_replace_table[0]["INSTANCE_ID"]="instance_id";
  535. mode_replace_table[0]["TIME"]="time";
  536. mode_replace_table[1]["VERTEX"]="vertex";
  537. //mode_replace_table[1]["POSITION"]="IN_POSITION";
  538. mode_replace_table[1]["NORMAL"]="normal";
  539. mode_replace_table[1]["TANGENT"]="tangent";
  540. mode_replace_table[1]["BINORMAL"]="binormal";
  541. mode_replace_table[1]["VAR1"]="var1_interp";
  542. mode_replace_table[1]["VAR2"]="var2_interp";
  543. mode_replace_table[1]["UV"]="uv";
  544. mode_replace_table[1]["UV2"]="uv2";
  545. mode_replace_table[1]["SCREEN_UV"]="screen_uv";
  546. mode_replace_table[1]["VAR1"]="var1_interp";
  547. mode_replace_table[1]["VAR2"]="var2_interp";
  548. mode_replace_table[1]["COLOR"]="color";
  549. mode_replace_table[1]["DIFFUSE"]="diffuse.rgb";
  550. mode_replace_table[1]["DIFFUSE_ALPHA"]="diffuse";
  551. mode_replace_table[1]["SPECULAR"]="specular";
  552. mode_replace_table[1]["EMISSION"]="emission";
  553. mode_replace_table[1]["SHADE_PARAM"]="shade_param";
  554. mode_replace_table[1]["SPEC_EXP"]="specular_exp";
  555. mode_replace_table[1]["GLOW"]="glow";
  556. mode_replace_table[1]["DISCARD"]="discard_";
  557. mode_replace_table[1]["POINT_COORD"]="gl_PointCoord";
  558. mode_replace_table[1]["INV_CAMERA_MATRIX"]="camera_inverse_transform";
  559. //mode_replace_table[1]["SCREEN_POS"]="SCREEN_POS";
  560. //mode_replace_table[1]["SCREEN_TEXEL_SIZE"]="SCREEN_TEXEL_SIZE";
  561. mode_replace_table[1]["TIME"]="time";
  562. //////////////
  563. mode_replace_table[2]["NORMAL"]="normal";
  564. //mode_replace_table[2]["POSITION"]="IN_POSITION";
  565. mode_replace_table[2]["LIGHT_DIR"]="light_dir";
  566. mode_replace_table[2]["LIGHT_DIFFUSE"]="light_diffuse";
  567. mode_replace_table[2]["LIGHT_SPECULAR"]="light_specular";
  568. mode_replace_table[2]["EYE_VEC"]="eye_vec";
  569. mode_replace_table[2]["DIFFUSE"]="mdiffuse";
  570. mode_replace_table[2]["SPECULAR"]="specular";
  571. mode_replace_table[2]["SPECULAR_EXP"]="specular_exp";
  572. mode_replace_table[2]["SHADE_PARAM"]="shade_param";
  573. mode_replace_table[2]["LIGHT"]="light";
  574. mode_replace_table[2]["POINT_COORD"]="gl_PointCoord";
  575. mode_replace_table[2]["TIME"]="time";
  576. //mode_replace_table[2]["SCREEN_POS"]="SCREEN_POS";
  577. //mode_replace_table[2]["SCREEN_TEXEL_SIZE"]="SCREEN_TEXEL_SIZE";
  578. out_vertex_name="VERTEX";
  579. vname_discard="DISCARD";
  580. vname_screen_uv="SCREEN_UV";
  581. vname_diffuse_alpha="DIFFUSE_ALPHA";
  582. vname_color_interp="COLOR";
  583. vname_uv_interp="UV";
  584. vname_uv2_interp="UV2";
  585. vname_tangent_interp="TANGENT";
  586. vname_binormal_interp="BINORMAL";
  587. vname_var1_interp="VAR1";
  588. vname_var2_interp="VAR2";
  589. vname_vertex="VERTEX";
  590. vname_light="LIGHT";
  591. vname_time="TIME";
  592. }