2
0

shader_language.cpp 75 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673
  1. /*************************************************************************/
  2. /* shader_language.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2019 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2019 Godot Engine contributors (cf. AUTHORS.md) */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /*************************************************************************/
  30. #include "shader_language.h"
  31. #include "os/os.h"
  32. #include "print_string.h"
  33. static bool _is_text_char(CharType c) {
  34. return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || (c >= '0' && c <= '9') || c == '_';
  35. }
  36. static bool _is_number(CharType c) {
  37. return (c >= '0' && c <= '9');
  38. }
  39. static bool _is_hex(CharType c) {
  40. return (c >= '0' && c <= '9') || (c >= 'a' && c <= 'f') || (c >= 'A' && c <= 'F');
  41. }
  42. const char *ShaderLanguage::token_names[TK_MAX] = {
  43. "EMPTY",
  44. "INDENTIFIER",
  45. "TRUE",
  46. "FALSE",
  47. "REAL_CONSTANT",
  48. "TYPE_VOID",
  49. "TYPE_BOOL",
  50. "TYPE_FLOAT",
  51. "TYPE_VEC2",
  52. "TYPE_VEC3",
  53. "TYPE_VEC4",
  54. "TYPE_MAT2",
  55. "TYPE_MAT3",
  56. "TYPE_MAT4",
  57. "TYPE_TEXTURE",
  58. "TYPE_CUBEMAP",
  59. "TYPE_COLOR",
  60. "OP_EQUAL",
  61. "OP_NOT_EQUAL",
  62. "OP_LESS",
  63. "OP_LESS_EQUAL",
  64. "OP_GREATER",
  65. "OP_GREATER_EQUAL",
  66. "OP_AND",
  67. "OP_OR",
  68. "OP_NOT",
  69. "OP_ADD",
  70. "OP_SUB",
  71. "OP_MUL",
  72. "OP_DIV",
  73. "OP_NEG",
  74. "OP_ASSIGN",
  75. "OP_ASSIGN_ADD",
  76. "OP_ASSIGN_SUB",
  77. "OP_ASSIGN_MUL",
  78. "OP_ASSIGN_DIV",
  79. "CF_IF",
  80. "CF_ELSE",
  81. "CF_RETURN",
  82. "BRACKET_OPEN",
  83. "BRACKET_CLOSE",
  84. "CURLY_BRACKET_OPEN",
  85. "CURLY_BRACKET_CLOSE",
  86. "PARENTHESIS_OPEN",
  87. "PARENTHESIS_CLOSE",
  88. "COMMA",
  89. "SEMICOLON",
  90. "PERIOD",
  91. "UNIFORM",
  92. "ERROR",
  93. };
  94. ShaderLanguage::Token ShaderLanguage::read_token(const CharType *p_text, int p_len, int &r_line, int &r_chars) {
  95. #define GETCHAR(m_idx) ((m_idx < p_len) ? p_text[m_idx] : CharType(0))
  96. r_chars = 1; //by default everything eats one char
  97. switch (GETCHAR(0)) {
  98. case '\t':
  99. case '\r':
  100. case ' ':
  101. return Token();
  102. case '\n':
  103. r_line++;
  104. return Token();
  105. case '/': {
  106. switch (GETCHAR(1)) {
  107. case '*': { // block comment
  108. while (true) {
  109. if (GETCHAR(r_chars + 1) == 0) {
  110. r_chars += 1;
  111. break;
  112. }
  113. if (GETCHAR(r_chars + 1) == '*' && GETCHAR(r_chars + 2) == '/') {
  114. r_chars += 3;
  115. break;
  116. }
  117. if (GETCHAR(r_chars + 1) == '\n') {
  118. r_line++;
  119. }
  120. r_chars++;
  121. }
  122. return Token();
  123. } break;
  124. case '/': { // line comment skip
  125. while (GETCHAR(r_chars + 1) != '\n' && GETCHAR(r_chars + 1) != 0) {
  126. r_chars++;
  127. }
  128. r_chars++;
  129. //r_line++;
  130. return Token();
  131. } break;
  132. case '=': { // diveq
  133. r_chars = 2;
  134. return Token(TK_OP_ASSIGN_DIV);
  135. } break;
  136. default:
  137. return Token(TK_OP_DIV);
  138. }
  139. } break;
  140. case '=': {
  141. if (GETCHAR(1) == '=') {
  142. r_chars++;
  143. return Token(TK_OP_EQUAL);
  144. }
  145. return Token(TK_OP_ASSIGN);
  146. } break;
  147. case '<': {
  148. if (GETCHAR(1) == '=') {
  149. r_chars++;
  150. return Token(TK_OP_LESS_EQUAL);
  151. } /*else if (GETCHAR(1)=='<') {
  152. r_chars++;
  153. if (GETCHAR(2)=='=') {
  154. r_chars++;
  155. return Token(TK_OP_ASSIGN_SHIFT_LEFT);
  156. }
  157. return Token(TK_OP_SHIFT_LEFT);
  158. }*/
  159. return Token(TK_OP_LESS);
  160. } break;
  161. case '>': {
  162. if (GETCHAR(1) == '=') {
  163. r_chars++;
  164. return Token(TK_OP_GREATER_EQUAL);
  165. } /* else if (GETCHAR(1)=='<') {
  166. r_chars++;
  167. if (GETCHAR(2)=='=') {
  168. r_chars++;
  169. return Token(TK_OP_ASSIGN_SHIFT_RIGHT);
  170. }
  171. return Token(TK_OP_SHIFT_RIGHT);
  172. }*/
  173. return Token(TK_OP_GREATER);
  174. } break;
  175. case '!': {
  176. if (GETCHAR(1) == '=') {
  177. r_chars++;
  178. return Token(TK_OP_NOT_EQUAL);
  179. }
  180. return Token(TK_OP_NOT);
  181. } break;
  182. //case '"' //string - no strings in shader
  183. //case '\'' //string - no strings in shader
  184. case '{':
  185. return Token(TK_CURLY_BRACKET_OPEN);
  186. case '}':
  187. return Token(TK_CURLY_BRACKET_CLOSE);
  188. //case '[':
  189. // return Token(TK_BRACKET_OPEN);
  190. //case ']':
  191. // return Token(TK_BRACKET_CLOSE);
  192. case '(':
  193. return Token(TK_PARENTHESIS_OPEN);
  194. case ')':
  195. return Token(TK_PARENTHESIS_CLOSE);
  196. case ',':
  197. return Token(TK_COMMA);
  198. case ';':
  199. return Token(TK_SEMICOLON);
  200. //case '?':
  201. // return Token(TK_QUESTION_MARK);
  202. //case ':':
  203. // return Token(TK_COLON); //for methods maybe but now useless.
  204. //case '^':
  205. // return Token(TK_OP_BIT_XOR);
  206. //case '~':
  207. // return Token(TK_OP_BIT_INVERT);
  208. case '&': {
  209. if (GETCHAR(1) == '&') {
  210. r_chars++;
  211. return Token(TK_OP_AND);
  212. }
  213. return Token(TK_ERROR, "Unknown character");
  214. /*
  215. if (GETCHAR(1)=='=') {
  216. r_chars++;
  217. return Token(TK_OP_ASSIGN_BIT_AND);
  218. } else if (GETCHAR(1)=='&') {
  219. r_chars++;
  220. return Token(TK_OP_AND);
  221. }
  222. return TK_OP_BIT_AND;*/
  223. } break;
  224. case '|': {
  225. if (GETCHAR(1) == '|') {
  226. r_chars++;
  227. return Token(TK_OP_OR);
  228. }
  229. return Token(TK_ERROR, "Unknown character");
  230. /*
  231. if (GETCHAR(1)=='=') {
  232. r_chars++;
  233. return Token(TK_OP_ASSIGN_BIT_OR);
  234. } else if (GETCHAR(1)=='|') {
  235. r_chars++;
  236. return Token(TK_OP_OR);
  237. }
  238. return TK_OP_BIT_OR;
  239. */
  240. } break;
  241. case '*': {
  242. if (GETCHAR(1) == '=') {
  243. r_chars++;
  244. return Token(TK_OP_ASSIGN_MUL);
  245. }
  246. return TK_OP_MUL;
  247. } break;
  248. case '+': {
  249. if (GETCHAR(1) == '=') {
  250. r_chars++;
  251. return Token(TK_OP_ASSIGN_ADD);
  252. } /*else if (GETCHAR(1)=='+') {
  253. r_chars++;
  254. return Token(TK_OP_PLUS_PLUS);
  255. }*/
  256. return TK_OP_ADD;
  257. } break;
  258. case '-': {
  259. if (GETCHAR(1) == '=') {
  260. r_chars++;
  261. return Token(TK_OP_ASSIGN_SUB);
  262. } /* else if (GETCHAR(1)=='-') {
  263. r_chars++;
  264. return Token(TK_OP_MINUS_MINUS);
  265. }*/
  266. return TK_OP_SUB;
  267. } break;
  268. /*case '%': {
  269. if (GETCHAR(1)=='=') {
  270. r_chars++;
  271. return Token(TK_OP_ASSIGN_MOD);
  272. }
  273. return TK_OP_MOD;
  274. } break;*/
  275. default: {
  276. if (_is_number(GETCHAR(0)) || (GETCHAR(0) == '.' && _is_number(GETCHAR(1)))) {
  277. // parse number
  278. bool period_found = false;
  279. bool exponent_found = false;
  280. bool hexa_found = false;
  281. bool sign_found = false;
  282. String str;
  283. int i = 0;
  284. while (true) {
  285. if (GETCHAR(i) == '.') {
  286. if (period_found || exponent_found)
  287. return Token(TK_ERROR, "Invalid numeric constant");
  288. period_found = true;
  289. } else if (GETCHAR(i) == 'x') {
  290. if (hexa_found || str.length() != 1 || str[0] != '0')
  291. return Token(TK_ERROR, "Invalid numeric constant");
  292. hexa_found = true;
  293. } else if (GETCHAR(i) == 'e') {
  294. if (hexa_found || exponent_found)
  295. return Token(TK_ERROR, "Invalid numeric constant");
  296. exponent_found = true;
  297. } else if (_is_number(GETCHAR(i))) {
  298. //all ok
  299. } else if (hexa_found && _is_hex(GETCHAR(i))) {
  300. } else if ((GETCHAR(i) == '-' || GETCHAR(i) == '+') && exponent_found) {
  301. if (sign_found)
  302. return Token(TK_ERROR, "Invalid numeric constant");
  303. sign_found = true;
  304. } else
  305. break;
  306. str += CharType(GETCHAR(i));
  307. i++;
  308. }
  309. if (!_is_number(str[str.length() - 1]))
  310. return Token(TK_ERROR, "Invalid numeric constant");
  311. r_chars += str.length() - 1;
  312. return Token(TK_REAL_CONSTANT, str);
  313. /*
  314. if (period_found)
  315. return Token(TK_NUMBER_REAL,str);
  316. else
  317. return Token(TK_NUMBER_INTEGER,str);*/
  318. }
  319. if (GETCHAR(0) == '.') {
  320. //parse period
  321. return Token(TK_PERIOD);
  322. }
  323. if (_is_text_char(GETCHAR(0))) {
  324. // parse identifier
  325. String str;
  326. str += CharType(GETCHAR(0));
  327. while (_is_text_char(GETCHAR(r_chars))) {
  328. str += CharType(GETCHAR(r_chars));
  329. r_chars++;
  330. }
  331. //see if keyword
  332. struct _kws {
  333. TokenType token;
  334. const char *text;
  335. };
  336. static const _kws keyword_list[] = {
  337. { TK_TRUE, "true" },
  338. { TK_FALSE, "false" },
  339. { TK_TYPE_VOID, "void" },
  340. { TK_TYPE_BOOL, "bool" },
  341. /*{TK_TYPE_INT,"int"},
  342. {TK_TYPE_INT2,"int2"},
  343. {TK_TYPE_INT3,"int3"},
  344. {TK_TYPE_INT4,"int4"},*/
  345. { TK_TYPE_FLOAT, "float" },
  346. /*{TK_TYPE_FLOAT2,"float2"},
  347. {TK_TYPE_FLOAT3,"float3"},
  348. {TK_TYPE_FLOAT4,"float4"},*/
  349. { TK_TYPE_VEC2, "vec2" },
  350. { TK_TYPE_VEC3, "vec3" },
  351. { TK_TYPE_VEC4, "vec4" },
  352. { TK_TYPE_TEXTURE, "texture" },
  353. { TK_TYPE_CUBEMAP, "cubemap" },
  354. { TK_TYPE_COLOR, "color" },
  355. { TK_TYPE_MAT2, "mat2" },
  356. /*{TK_TYPE_MAT3,"mat3"},
  357. {TK_TYPE_MAT4,"mat3"},*/
  358. { TK_TYPE_MAT3, "mat3" },
  359. { TK_TYPE_MAT4, "mat4" },
  360. { TK_CF_IF, "if" },
  361. { TK_CF_ELSE, "else" },
  362. /*
  363. {TK_CF_FOR,"for"},
  364. {TK_CF_WHILE,"while"},
  365. {TK_CF_DO,"do"},
  366. {TK_CF_SWITCH,"switch"},
  367. {TK_CF_BREAK,"break"},
  368. {TK_CF_CONTINUE,"continue"},*/
  369. { TK_CF_RETURN, "return" },
  370. { TK_UNIFORM, "uniform" },
  371. { TK_ERROR, NULL }
  372. };
  373. int idx = 0;
  374. while (keyword_list[idx].text) {
  375. if (str == keyword_list[idx].text)
  376. return Token(keyword_list[idx].token);
  377. idx++;
  378. }
  379. return Token(TK_INDENTIFIER, str);
  380. }
  381. if (GETCHAR(0) > 32)
  382. return Token(TK_ERROR, "Tokenizer: Unknown character #" + itos(GETCHAR(0)) + ": '" + String::chr(GETCHAR(0)) + "'");
  383. else
  384. return Token(TK_ERROR, "Tokenizer: Unknown character #" + itos(GETCHAR(0)));
  385. } break;
  386. }
  387. ERR_PRINT("BUG");
  388. return Token();
  389. }
  390. Error ShaderLanguage::tokenize(const String &p_text, Vector<Token> *p_tokens, String *r_error, int *r_err_line, int *r_err_column) {
  391. int len = p_text.length();
  392. int pos = 0;
  393. int line = 0;
  394. int col = 0;
  395. while (pos < len) {
  396. int advance = 0;
  397. int prev_line = line;
  398. Token t = read_token(&p_text[pos], len - pos, line, advance);
  399. t.line = line;
  400. t.col = col;
  401. if (t.type == TK_ERROR) {
  402. if (r_error) {
  403. *r_error = t.text;
  404. *r_err_line = line;
  405. *r_err_column = col;
  406. return ERR_COMPILATION_FAILED;
  407. }
  408. }
  409. if (line == prev_line) {
  410. col += advance;
  411. } else {
  412. col = 0;
  413. //p_tokens->push_back(Token(TK_LINE,itos(line)))
  414. }
  415. if (t.type != TK_EMPTY)
  416. p_tokens->push_back(t);
  417. pos += advance;
  418. }
  419. return OK;
  420. }
  421. String ShaderLanguage::lex_debug(const String &p_code) {
  422. Vector<Token> tokens;
  423. String error;
  424. int errline, errcol;
  425. if (tokenize(p_code, &tokens, &error, &errline, &errcol) != OK)
  426. return error;
  427. String ret;
  428. for (int i = 0; i < tokens.size(); i++) {
  429. ret += String(token_names[tokens[i].type]) + ":" + itos(tokens[i].line) + ":" + itos(tokens[i].col) + ":" + tokens[i].text + "\n";
  430. }
  431. return ret;
  432. }
  433. bool ShaderLanguage::is_token_datatype(TokenType p_type) {
  434. return (p_type == TK_TYPE_VOID) ||
  435. (p_type == TK_TYPE_BOOL) ||
  436. (p_type == TK_TYPE_FLOAT) ||
  437. (p_type == TK_TYPE_VEC2) ||
  438. (p_type == TK_TYPE_VEC3) ||
  439. (p_type == TK_TYPE_VEC4) ||
  440. (p_type == TK_TYPE_COLOR) ||
  441. (p_type == TK_TYPE_MAT2) ||
  442. (p_type == TK_TYPE_MAT3) ||
  443. (p_type == TK_TYPE_MAT4) ||
  444. (p_type == TK_TYPE_CUBEMAP) ||
  445. (p_type == TK_TYPE_TEXTURE);
  446. }
  447. ShaderLanguage::DataType ShaderLanguage::get_token_datatype(TokenType p_type) {
  448. switch (p_type) {
  449. case TK_TYPE_VOID: return TYPE_VOID;
  450. case TK_TYPE_BOOL: return TYPE_BOOL;
  451. case TK_TYPE_FLOAT: return TYPE_FLOAT;
  452. case TK_TYPE_VEC2: return TYPE_VEC2;
  453. case TK_TYPE_VEC3: return TYPE_VEC3;
  454. case TK_TYPE_VEC4: return TYPE_VEC4;
  455. case TK_TYPE_COLOR: return TYPE_VEC4;
  456. case TK_TYPE_MAT2: return TYPE_MAT2;
  457. case TK_TYPE_MAT3: return TYPE_MAT3;
  458. case TK_TYPE_MAT4: return TYPE_MAT4;
  459. case TK_TYPE_TEXTURE: return TYPE_TEXTURE;
  460. case TK_TYPE_CUBEMAP: return TYPE_CUBEMAP;
  461. default: return TYPE_VOID;
  462. }
  463. return TYPE_VOID;
  464. }
  465. String ShaderLanguage::get_datatype_name(DataType p_type) {
  466. switch (p_type) {
  467. case TYPE_VOID: return "void";
  468. case TYPE_BOOL: return "bool";
  469. case TYPE_FLOAT: return "float";
  470. case TYPE_VEC2: return "vec2";
  471. case TYPE_VEC3: return "vec3";
  472. case TYPE_VEC4: return "vec4";
  473. case TYPE_MAT2: return "mat2";
  474. case TYPE_MAT3: return "mat3";
  475. case TYPE_MAT4: return "mat4";
  476. case TYPE_TEXTURE: return "texture";
  477. case TYPE_CUBEMAP: return "cubemap";
  478. default: return "";
  479. }
  480. return "";
  481. }
  482. bool ShaderLanguage::is_token_nonvoid_datatype(TokenType p_type) {
  483. return (p_type == TK_TYPE_BOOL) ||
  484. (p_type == TK_TYPE_FLOAT) ||
  485. (p_type == TK_TYPE_VEC2) ||
  486. (p_type == TK_TYPE_VEC3) ||
  487. (p_type == TK_TYPE_VEC4) ||
  488. (p_type == TK_TYPE_COLOR) ||
  489. (p_type == TK_TYPE_MAT2) ||
  490. (p_type == TK_TYPE_MAT3) ||
  491. (p_type == TK_TYPE_MAT4) ||
  492. (p_type == TK_TYPE_TEXTURE) ||
  493. (p_type == TK_TYPE_CUBEMAP);
  494. }
  495. bool ShaderLanguage::parser_is_at_function(Parser &parser) {
  496. return (is_token_datatype(parser.get_next_token_type(0)) && parser.get_next_token_type(1) == TK_INDENTIFIER && parser.get_next_token_type(2) == TK_PARENTHESIS_OPEN);
  497. }
  498. bool ShaderLanguage::test_existing_identifier(Node *p_node, const StringName p_identifier, bool p_func, bool p_var, bool p_builtin) {
  499. Node *node = p_node;
  500. while (node) {
  501. if (node->type == Node::TYPE_BLOCK) {
  502. BlockNode *block = (BlockNode *)node;
  503. if (block->variables.has(p_identifier))
  504. return true;
  505. } else if (node->type == Node::TYPE_PROGRAM) {
  506. ProgramNode *program = (ProgramNode *)node;
  507. for (int i = 0; i < program->functions.size(); i++) {
  508. if (program->functions[i].name == p_identifier) {
  509. return true;
  510. }
  511. }
  512. if (program->builtin_variables.has(p_identifier)) {
  513. return true;
  514. }
  515. if (program->uniforms.has(p_identifier)) {
  516. return true;
  517. }
  518. } else if (node->type == Node::TYPE_FUNCTION) {
  519. FunctionNode *func = (FunctionNode *)node;
  520. for (int i = 0; i < func->arguments.size(); i++)
  521. if (func->arguments[i].name == p_identifier)
  522. return true;
  523. }
  524. node = node->parent;
  525. }
  526. // try keywords
  527. int idx = 0;
  528. //todo optimize
  529. while (intrinsic_func_defs[idx].name) {
  530. if (p_identifier.operator String() == intrinsic_func_defs[idx].name)
  531. return true;
  532. idx++;
  533. }
  534. return false;
  535. }
  536. Error ShaderLanguage::parse_function(Parser &parser, BlockNode *p_block) {
  537. if (!p_block->parent || p_block->parent->type != Node::TYPE_PROGRAM) {
  538. parser.set_error("Misplaced function");
  539. return ERR_PARSE_ERROR;
  540. }
  541. ProgramNode *program = (ProgramNode *)p_block->parent;
  542. StringName name = parser.get_next_token(1).text;
  543. if (test_existing_identifier(p_block, name)) {
  544. parser.set_error("Duplicate Identifier (existing variable/builtin/function): " + name);
  545. return ERR_PARSE_ERROR;
  546. }
  547. FunctionNode *function = parser.create_node<FunctionNode>(program);
  548. function->body = parser.create_node<BlockNode>(function);
  549. function->name = name;
  550. function->return_type = get_token_datatype(parser.get_next_token_type(0));
  551. { //add to programnode
  552. ProgramNode::Function f;
  553. f.name = name;
  554. f.function = function;
  555. program->functions.push_back(f);
  556. }
  557. int ofs = 3;
  558. while (true) {
  559. //end of arguments
  560. if (parser.get_next_token_type(ofs) == TK_PARENTHESIS_CLOSE) {
  561. ofs++;
  562. break;
  563. }
  564. //next argument awaits
  565. if (parser.get_next_token_type(ofs) == TK_COMMA) {
  566. if (!is_token_nonvoid_datatype(parser.get_next_token_type(ofs + 1))) {
  567. parser.set_error("Expected Identifier or ')' following ','");
  568. return ERR_PARSE_ERROR;
  569. }
  570. ofs++;
  571. continue;
  572. }
  573. if (!is_token_nonvoid_datatype(parser.get_next_token_type(ofs + 0))) {
  574. parser.set_error("Invalid Argument Type");
  575. return ERR_PARSE_ERROR;
  576. }
  577. DataType identtype = get_token_datatype(parser.get_next_token_type(ofs + 0));
  578. if (parser.get_next_token_type(ofs + 1) != TK_INDENTIFIER) {
  579. parser.set_error("Expected Argument Identifier");
  580. return ERR_PARSE_ERROR;
  581. }
  582. StringName identname = parser.get_next_token(ofs + 1).text;
  583. if (test_existing_identifier(function, identname)) {
  584. parser.set_error("Duplicate Argument Identifier: " + identname);
  585. return ERR_DUPLICATE_SYMBOL;
  586. }
  587. FunctionNode::Argument arg;
  588. arg.name = identname;
  589. arg.type = identtype;
  590. //function->body->variables[arg.name]=arg.type;
  591. function->arguments.push_back(arg);
  592. ofs += 2;
  593. }
  594. parser.advance(ofs);
  595. // match {
  596. if (parser.get_next_token_type() != TK_CURLY_BRACKET_OPEN) {
  597. parser.set_error("Expected '{'");
  598. return ERR_PARSE_ERROR;
  599. }
  600. parser.advance();
  601. Error err = parse_block(parser, function->body);
  602. if (err)
  603. return err;
  604. // make sure that if the function has a return type, it does return something..
  605. if (function->return_type != TYPE_VOID) {
  606. bool found = false;
  607. for (int i = 0; i < function->body->statements.size(); i++) {
  608. if (function->body->statements[i]->type == Node::TYPE_CONTROL_FLOW) {
  609. ControlFlowNode *cf = (ControlFlowNode *)function->body->statements[i];
  610. if (cf->flow_op == FLOW_OP_RETURN) {
  611. // type of return was already checked when inserted
  612. // no need to check here
  613. found = true;
  614. }
  615. }
  616. }
  617. if (!found) {
  618. parser.set_error("Function must return a value (use the main block)");
  619. return ERR_PARSE_ERROR;
  620. }
  621. }
  622. return OK;
  623. }
  624. const ShaderLanguage::IntrinsicFuncDef ShaderLanguage::intrinsic_func_defs[] = {
  625. //constructors
  626. { "bool", TYPE_BOOL, { TYPE_BOOL, TYPE_VOID } },
  627. { "float", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  628. { "vec2", TYPE_VEC2, { TYPE_FLOAT, TYPE_VOID } },
  629. { "vec2", TYPE_VEC2, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  630. { "vec3", TYPE_VEC3, { TYPE_FLOAT, TYPE_VOID } },
  631. { "vec3", TYPE_VEC3, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  632. { "vec3", TYPE_VEC3, { TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  633. { "vec3", TYPE_VEC3, { TYPE_FLOAT, TYPE_VEC2, TYPE_VOID } },
  634. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_VOID } },
  635. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  636. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  637. { "vec4", TYPE_VEC4, { TYPE_VEC2, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  638. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VEC2, TYPE_VOID } },
  639. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_VEC3, TYPE_VOID } },
  640. { "vec4", TYPE_VEC4, { TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  641. { "vec4", TYPE_VEC4, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  642. { "mat2", TYPE_MAT2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  643. { "mat3", TYPE_MAT3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  644. { "mat4", TYPE_MAT4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  645. //intrinsics - trigonometry
  646. { "sin", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  647. { "cos", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  648. { "tan", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  649. { "asin", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  650. { "acos", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  651. { "atan", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  652. { "atan2", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  653. { "sinh", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  654. { "cosh", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  655. { "tanh", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  656. //intrinsics - exponential
  657. { "pow", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  658. { "pow", TYPE_VEC2, { TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  659. { "pow", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  660. { "pow", TYPE_VEC3, { TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  661. { "pow", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  662. { "pow", TYPE_VEC4, { TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  663. { "pow", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  664. { "exp", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  665. { "exp", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  666. { "exp", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  667. { "exp", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  668. { "log", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  669. { "log", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  670. { "log", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  671. { "log", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  672. { "sqrt", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  673. { "sqrt", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  674. { "sqrt", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  675. { "sqrt", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  676. //intrinsics - common
  677. { "abs", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  678. { "abs", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  679. { "abs", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  680. { "abs", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  681. { "sign", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  682. { "sign", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  683. { "sign", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  684. { "sign", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  685. { "floor", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  686. { "floor", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  687. { "floor", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  688. { "floor", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  689. { "trunc", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  690. { "trunc", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  691. { "trunc", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  692. { "trunc", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  693. { "round", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  694. { "round", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  695. { "round", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  696. { "round", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  697. { "ceil", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  698. { "ceil", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  699. { "ceil", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  700. { "ceil", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  701. { "fract", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  702. { "fract", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  703. { "fract", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  704. { "fract", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  705. { "mod", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  706. { "mod", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  707. { "mod", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  708. { "mod", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  709. { "min", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  710. { "min", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  711. { "min", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  712. { "min", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  713. { "max", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  714. { "max", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  715. { "max", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  716. { "max", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  717. { "clamp", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  718. { "clamp", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  719. { "clamp", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  720. { "clamp", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  721. { "clamp", TYPE_VEC2, { TYPE_VEC2, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  722. { "clamp", TYPE_VEC3, { TYPE_VEC3, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  723. { "clamp", TYPE_VEC4, { TYPE_VEC4, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  724. { "mix", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  725. { "mix", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  726. { "mix", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  727. { "mix", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  728. { "mix", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  729. { "mix", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  730. { "mix", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  731. { "step", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  732. { "step", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  733. { "step", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  734. { "step", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  735. { "step", TYPE_VEC2, { TYPE_FLOAT, TYPE_VEC2, TYPE_VOID } },
  736. { "step", TYPE_VEC3, { TYPE_FLOAT, TYPE_VEC3, TYPE_VOID } },
  737. { "step", TYPE_VEC4, { TYPE_FLOAT, TYPE_VEC4, TYPE_VOID } },
  738. { "smoothstep", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  739. { "smoothstep", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  740. { "smoothstep", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  741. { "smoothstep", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  742. { "smoothstep", TYPE_VEC2, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VEC2, TYPE_VOID } },
  743. { "smoothstep", TYPE_VEC3, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VEC3, TYPE_VOID } },
  744. { "smoothstep", TYPE_VEC4, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VEC4, TYPE_VOID } },
  745. //intrinsics - geometric
  746. { "length", TYPE_FLOAT, { TYPE_VEC2, TYPE_VOID } },
  747. { "length", TYPE_FLOAT, { TYPE_VEC3, TYPE_VOID } },
  748. { "length", TYPE_FLOAT, { TYPE_VEC4, TYPE_VOID } },
  749. { "distance", TYPE_FLOAT, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  750. { "distance", TYPE_FLOAT, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  751. { "distance", TYPE_FLOAT, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  752. { "dot", TYPE_FLOAT, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  753. { "dot", TYPE_FLOAT, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  754. { "dot", TYPE_FLOAT, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  755. { "cross", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  756. { "normalize", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  757. { "normalize", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  758. { "normalize", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  759. { "reflect", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  760. { "refract", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  761. //intrinsics - texture
  762. { "tex", TYPE_VEC4, { TYPE_TEXTURE, TYPE_VEC2, TYPE_VOID } },
  763. { "texcube", TYPE_VEC4, { TYPE_CUBEMAP, TYPE_VEC3, TYPE_VOID } },
  764. { "texscreen", TYPE_VEC3, { TYPE_VEC2, TYPE_VOID } },
  765. { "texpos", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  766. { NULL, TYPE_VOID, { TYPE_VOID } }
  767. };
  768. const ShaderLanguage::OperatorDef ShaderLanguage::operator_defs[] = {
  769. { OP_ASSIGN, TYPE_VOID, { TYPE_BOOL, TYPE_BOOL } },
  770. { OP_ASSIGN, TYPE_VOID, { TYPE_FLOAT, TYPE_FLOAT } },
  771. { OP_ASSIGN, TYPE_VOID, { TYPE_VEC2, TYPE_VEC2 } },
  772. { OP_ASSIGN, TYPE_VOID, { TYPE_VEC3, TYPE_VEC3 } },
  773. { OP_ASSIGN, TYPE_VOID, { TYPE_VEC4, TYPE_VEC4 } },
  774. { OP_ASSIGN, TYPE_VOID, { TYPE_MAT2, TYPE_MAT2 } },
  775. { OP_ASSIGN, TYPE_VOID, { TYPE_MAT3, TYPE_MAT3 } },
  776. { OP_ASSIGN, TYPE_VOID, { TYPE_MAT4, TYPE_MAT4 } },
  777. { OP_ADD, TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT } },
  778. { OP_ADD, TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2 } },
  779. { OP_ADD, TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3 } },
  780. { OP_ADD, TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4 } },
  781. { OP_SUB, TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT } },
  782. { OP_SUB, TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2 } },
  783. { OP_SUB, TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3 } },
  784. { OP_SUB, TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4 } },
  785. { OP_MUL, TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT } },
  786. { OP_MUL, TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2 } },
  787. { OP_MUL, TYPE_VEC2, { TYPE_VEC2, TYPE_FLOAT } },
  788. { OP_MUL, TYPE_VEC2, { TYPE_FLOAT, TYPE_VEC2 } },
  789. { OP_MUL, TYPE_VEC2, { TYPE_VEC2, TYPE_MAT3 } },
  790. { OP_MUL, TYPE_VEC2, { TYPE_MAT2, TYPE_VEC2 } },
  791. { OP_MUL, TYPE_VEC2, { TYPE_VEC2, TYPE_MAT2 } },
  792. { OP_MUL, TYPE_VEC2, { TYPE_MAT3, TYPE_VEC2 } },
  793. { OP_MUL, TYPE_VEC2, { TYPE_VEC2, TYPE_MAT4 } },
  794. { OP_MUL, TYPE_VEC2, { TYPE_MAT4, TYPE_VEC2 } },
  795. { OP_MUL, TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3 } },
  796. { OP_MUL, TYPE_VEC3, { TYPE_VEC3, TYPE_FLOAT } },
  797. { OP_MUL, TYPE_VEC3, { TYPE_FLOAT, TYPE_VEC3 } },
  798. { OP_MUL, TYPE_VEC3, { TYPE_MAT3, TYPE_VEC3 } },
  799. { OP_MUL, TYPE_VEC3, { TYPE_MAT4, TYPE_VEC3 } },
  800. { OP_MUL, TYPE_VEC3, { TYPE_VEC3, TYPE_MAT3 } },
  801. { OP_MUL, TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4 } },
  802. { OP_MUL, TYPE_VEC4, { TYPE_VEC4, TYPE_FLOAT } },
  803. { OP_MUL, TYPE_VEC4, { TYPE_FLOAT, TYPE_VEC4 } },
  804. { OP_MUL, TYPE_VEC4, { TYPE_MAT4, TYPE_VEC4 } },
  805. { OP_MUL, TYPE_VEC4, { TYPE_VEC4, TYPE_MAT4 } },
  806. { OP_MUL, TYPE_MAT2, { TYPE_MAT2, TYPE_MAT2 } },
  807. { OP_MUL, TYPE_MAT3, { TYPE_MAT3, TYPE_MAT3 } },
  808. { OP_MUL, TYPE_MAT4, { TYPE_MAT4, TYPE_MAT4 } },
  809. { OP_DIV, TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT } },
  810. { OP_DIV, TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2 } },
  811. { OP_DIV, TYPE_VEC2, { TYPE_VEC2, TYPE_FLOAT } },
  812. { OP_DIV, TYPE_VEC2, { TYPE_FLOAT, TYPE_VEC2 } },
  813. { OP_DIV, TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3 } },
  814. { OP_DIV, TYPE_VEC3, { TYPE_VEC3, TYPE_FLOAT } },
  815. { OP_DIV, TYPE_VEC3, { TYPE_FLOAT, TYPE_VEC3 } },
  816. { OP_DIV, TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4 } },
  817. { OP_DIV, TYPE_VEC4, { TYPE_VEC4, TYPE_FLOAT } },
  818. { OP_DIV, TYPE_VEC4, { TYPE_FLOAT, TYPE_VEC4 } },
  819. { OP_ASSIGN_ADD, TYPE_VOID, { TYPE_FLOAT, TYPE_FLOAT } },
  820. { OP_ASSIGN_ADD, TYPE_VOID, { TYPE_VEC2, TYPE_VEC2 } },
  821. { OP_ASSIGN_ADD, TYPE_VOID, { TYPE_VEC3, TYPE_VEC3 } },
  822. { OP_ASSIGN_ADD, TYPE_VOID, { TYPE_VEC4, TYPE_VEC4 } },
  823. { OP_ASSIGN_ADD, TYPE_VOID, { TYPE_VEC2, TYPE_FLOAT } },
  824. { OP_ASSIGN_ADD, TYPE_VOID, { TYPE_VEC3, TYPE_FLOAT } },
  825. { OP_ASSIGN_ADD, TYPE_VOID, { TYPE_VEC4, TYPE_FLOAT } },
  826. { OP_ASSIGN_SUB, TYPE_VOID, { TYPE_FLOAT, TYPE_FLOAT } },
  827. { OP_ASSIGN_SUB, TYPE_VOID, { TYPE_VEC2, TYPE_VEC2 } },
  828. { OP_ASSIGN_SUB, TYPE_VOID, { TYPE_VEC3, TYPE_VEC3 } },
  829. { OP_ASSIGN_SUB, TYPE_VOID, { TYPE_VEC4, TYPE_VEC4 } },
  830. { OP_ASSIGN_SUB, TYPE_VOID, { TYPE_VEC2, TYPE_FLOAT } },
  831. { OP_ASSIGN_SUB, TYPE_VOID, { TYPE_VEC3, TYPE_FLOAT } },
  832. { OP_ASSIGN_SUB, TYPE_VOID, { TYPE_VEC4, TYPE_FLOAT } },
  833. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_FLOAT, TYPE_FLOAT } },
  834. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC2, TYPE_VEC2 } },
  835. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC2, TYPE_FLOAT } },
  836. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC2, TYPE_MAT2 } },
  837. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_MAT2, TYPE_MAT2 } },
  838. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC3, TYPE_MAT3 } },
  839. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC3, TYPE_VEC3 } },
  840. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC3, TYPE_FLOAT } },
  841. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC3, TYPE_MAT4 } },
  842. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC4, TYPE_VEC4 } },
  843. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC4, TYPE_FLOAT } },
  844. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_VEC4, TYPE_MAT4 } },
  845. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_MAT3, TYPE_MAT3 } },
  846. { OP_ASSIGN_MUL, TYPE_VOID, { TYPE_MAT4, TYPE_MAT4 } },
  847. { OP_ASSIGN_DIV, TYPE_VOID, { TYPE_FLOAT, TYPE_FLOAT } },
  848. { OP_ASSIGN_DIV, TYPE_VOID, { TYPE_VEC2, TYPE_VEC2 } },
  849. { OP_ASSIGN_DIV, TYPE_VOID, { TYPE_VEC2, TYPE_FLOAT } },
  850. { OP_ASSIGN_DIV, TYPE_VOID, { TYPE_VEC3, TYPE_VEC3 } },
  851. { OP_ASSIGN_DIV, TYPE_VOID, { TYPE_VEC3, TYPE_FLOAT } },
  852. { OP_ASSIGN_DIV, TYPE_VOID, { TYPE_VEC4, TYPE_VEC4 } },
  853. { OP_ASSIGN_DIV, TYPE_VOID, { TYPE_VEC4, TYPE_FLOAT } },
  854. { OP_NEG, TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  855. { OP_NEG, TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  856. { OP_NEG, TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  857. { OP_NEG, TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  858. { OP_NOT, TYPE_BOOL, { TYPE_BOOL, TYPE_VOID } },
  859. { OP_CMP_EQ, TYPE_BOOL, { TYPE_BOOL, TYPE_BOOL } },
  860. { OP_CMP_EQ, TYPE_BOOL, { TYPE_FLOAT, TYPE_FLOAT } },
  861. { OP_CMP_EQ, TYPE_BOOL, { TYPE_VEC3, TYPE_VEC2 } },
  862. { OP_CMP_EQ, TYPE_BOOL, { TYPE_VEC3, TYPE_VEC3 } },
  863. { OP_CMP_EQ, TYPE_BOOL, { TYPE_VEC3, TYPE_VEC4 } },
  864. //{OP_CMP_EQ,TYPE_MAT3,{TYPE_MAT4,TYPE_MAT3}}, ??
  865. //{OP_CMP_EQ,TYPE_MAT4,{TYPE_MAT4,TYPE_MAT4}}, ??
  866. { OP_CMP_NEQ, TYPE_BOOL, { TYPE_BOOL, TYPE_BOOL } },
  867. { OP_CMP_NEQ, TYPE_BOOL, { TYPE_FLOAT, TYPE_FLOAT } },
  868. { OP_CMP_NEQ, TYPE_BOOL, { TYPE_VEC2, TYPE_VEC2 } },
  869. { OP_CMP_NEQ, TYPE_BOOL, { TYPE_VEC3, TYPE_VEC3 } },
  870. { OP_CMP_NEQ, TYPE_BOOL, { TYPE_VEC4, TYPE_VEC4 } },
  871. //{OP_CMP_NEQ,TYPE_MAT4,{TYPE_MAT4,TYPE_MAT4}}, //?
  872. { OP_CMP_LEQ, TYPE_BOOL, { TYPE_FLOAT, TYPE_FLOAT } },
  873. { OP_CMP_GEQ, TYPE_BOOL, { TYPE_FLOAT, TYPE_FLOAT } },
  874. { OP_CMP_LESS, TYPE_BOOL, { TYPE_FLOAT, TYPE_FLOAT } },
  875. { OP_CMP_GREATER, TYPE_BOOL, { TYPE_FLOAT, TYPE_FLOAT } },
  876. { OP_CMP_OR, TYPE_BOOL, { TYPE_BOOL, TYPE_BOOL } },
  877. { OP_CMP_AND, TYPE_BOOL, { TYPE_BOOL, TYPE_BOOL } },
  878. { OP_MAX, TYPE_VOID, { TYPE_VOID, TYPE_VOID } }
  879. };
  880. const ShaderLanguage::BuiltinsDef ShaderLanguage::vertex_builtins_defs[] = {
  881. { "SRC_VERTEX", TYPE_VEC3 },
  882. { "SRC_NORMAL", TYPE_VEC3 },
  883. { "SRC_TANGENT", TYPE_VEC3 },
  884. { "SRC_BINORMALF", TYPE_FLOAT },
  885. { "POSITION", TYPE_VEC4 },
  886. { "VERTEX", TYPE_VEC3 },
  887. { "NORMAL", TYPE_VEC3 },
  888. { "TANGENT", TYPE_VEC3 },
  889. { "BINORMAL", TYPE_VEC3 },
  890. { "UV", TYPE_VEC2 },
  891. { "UV2", TYPE_VEC2 },
  892. { "COLOR", TYPE_VEC4 },
  893. { "BONES", TYPE_VEC4 },
  894. { "WEIGHTS", TYPE_VEC4 },
  895. { "VAR1", TYPE_VEC4 },
  896. { "VAR2", TYPE_VEC4 },
  897. { "SPEC_EXP", TYPE_FLOAT },
  898. { "POINT_SIZE", TYPE_FLOAT },
  899. //builtins
  900. { "WORLD_MATRIX", TYPE_MAT4 },
  901. { "INV_CAMERA_MATRIX", TYPE_MAT4 },
  902. { "PROJECTION_MATRIX", TYPE_MAT4 },
  903. { "MODELVIEW_MATRIX", TYPE_MAT4 },
  904. { "INSTANCE_ID", TYPE_FLOAT },
  905. { "TIME", TYPE_FLOAT },
  906. { NULL, TYPE_VOID },
  907. };
  908. const ShaderLanguage::BuiltinsDef ShaderLanguage::fragment_builtins_defs[] = {
  909. { "VERTEX", TYPE_VEC3 },
  910. { "POSITION", TYPE_VEC4 },
  911. { "NORMAL", TYPE_VEC3 },
  912. { "TANGENT", TYPE_VEC3 },
  913. { "BINORMAL", TYPE_VEC3 },
  914. { "NORMALMAP", TYPE_VEC3 },
  915. { "NORMALMAP_DEPTH", TYPE_FLOAT },
  916. { "UV", TYPE_VEC2 },
  917. { "UV2", TYPE_VEC2 },
  918. { "COLOR", TYPE_VEC4 },
  919. { "NORMAL", TYPE_VEC3 },
  920. { "VAR1", TYPE_VEC4 },
  921. { "VAR2", TYPE_VEC4 },
  922. { "DIFFUSE", TYPE_VEC3 },
  923. { "DIFFUSE_ALPHA", TYPE_VEC4 },
  924. { "SPECULAR", TYPE_VEC3 },
  925. { "EMISSION", TYPE_VEC3 },
  926. { "SPEC_EXP", TYPE_FLOAT },
  927. { "GLOW", TYPE_FLOAT },
  928. { "SHADE_PARAM", TYPE_FLOAT },
  929. { "DISCARD", TYPE_BOOL },
  930. { "SCREEN_UV", TYPE_VEC2 },
  931. { "POINT_COORD", TYPE_VEC2 },
  932. { "INV_CAMERA_MATRIX", TYPE_MAT4 },
  933. // { "SCREEN_POS", TYPE_VEC2},
  934. // { "SCREEN_TEXEL_SIZE", TYPE_VEC2},
  935. { "TIME", TYPE_FLOAT },
  936. { NULL, TYPE_VOID }
  937. };
  938. const ShaderLanguage::BuiltinsDef ShaderLanguage::light_builtins_defs[] = {
  939. { "NORMAL", TYPE_VEC3 },
  940. { "LIGHT_DIR", TYPE_VEC3 },
  941. { "LIGHT_DIFFUSE", TYPE_VEC3 },
  942. { "LIGHT_SPECULAR", TYPE_VEC3 },
  943. { "EYE_VEC", TYPE_VEC3 },
  944. { "DIFFUSE", TYPE_VEC3 },
  945. { "SPECULAR", TYPE_VEC3 },
  946. { "SPECULAR_EXP", TYPE_FLOAT },
  947. { "SHADE_PARAM", TYPE_FLOAT },
  948. { "LIGHT", TYPE_VEC3 },
  949. { "SHADOW", TYPE_VEC3 },
  950. { "POINT_COORD", TYPE_VEC2 },
  951. // { "SCREEN_POS", TYPE_VEC2},
  952. // { "SCREEN_TEXEL_SIZE", TYPE_VEC2},
  953. { "TIME", TYPE_FLOAT },
  954. { NULL, TYPE_VOID }
  955. };
  956. const ShaderLanguage::BuiltinsDef ShaderLanguage::ci_vertex_builtins_defs[] = {
  957. { "SRC_VERTEX", TYPE_VEC2 },
  958. { "VERTEX", TYPE_VEC2 },
  959. { "WORLD_VERTEX", TYPE_VEC2 },
  960. { "UV", TYPE_VEC2 },
  961. { "COLOR", TYPE_VEC4 },
  962. { "VAR1", TYPE_VEC4 },
  963. { "VAR2", TYPE_VEC4 },
  964. { "POINT_SIZE", TYPE_FLOAT },
  965. //builtins
  966. { "WORLD_MATRIX", TYPE_MAT4 },
  967. { "PROJECTION_MATRIX", TYPE_MAT4 },
  968. { "EXTRA_MATRIX", TYPE_MAT4 },
  969. { "TIME", TYPE_FLOAT },
  970. { "AT_LIGHT_PASS", TYPE_BOOL },
  971. { NULL, TYPE_VOID },
  972. };
  973. const ShaderLanguage::BuiltinsDef ShaderLanguage::ci_fragment_builtins_defs[] = {
  974. { "SRC_COLOR", TYPE_VEC4 },
  975. { "POSITION", TYPE_VEC2 },
  976. { "NORMAL", TYPE_VEC3 },
  977. { "NORMALMAP", TYPE_VEC3 },
  978. { "NORMALMAP_DEPTH", TYPE_FLOAT },
  979. { "UV", TYPE_VEC2 },
  980. { "COLOR", TYPE_VEC4 },
  981. { "TEXTURE", TYPE_TEXTURE },
  982. { "TEXTURE_PIXEL_SIZE", TYPE_VEC2 },
  983. { "VAR1", TYPE_VEC4 },
  984. { "VAR2", TYPE_VEC4 },
  985. { "SCREEN_UV", TYPE_VEC2 },
  986. { "POINT_COORD", TYPE_VEC2 },
  987. // { "SCREEN_POS", TYPE_VEC2},
  988. // { "SCREEN_TEXEL_SIZE", TYPE_VEC2},
  989. { "TIME", TYPE_FLOAT },
  990. { "AT_LIGHT_PASS", TYPE_BOOL },
  991. { NULL, TYPE_VOID }
  992. };
  993. const ShaderLanguage::BuiltinsDef ShaderLanguage::ci_light_builtins_defs[] = {
  994. { "POSITION", TYPE_VEC2 },
  995. { "NORMAL", TYPE_VEC3 },
  996. { "UV", TYPE_VEC2 },
  997. { "COLOR", TYPE_VEC4 },
  998. { "TEXTURE", TYPE_TEXTURE },
  999. { "TEXTURE_PIXEL_SIZE", TYPE_VEC2 },
  1000. { "VAR1", TYPE_VEC4 },
  1001. { "VAR2", TYPE_VEC4 },
  1002. { "SCREEN_UV", TYPE_VEC2 },
  1003. { "LIGHT_VEC", TYPE_VEC2 },
  1004. { "LIGHT_HEIGHT", TYPE_FLOAT },
  1005. { "LIGHT_COLOR", TYPE_VEC4 },
  1006. { "LIGHT_UV", TYPE_VEC2 },
  1007. { "LIGHT_SHADOW", TYPE_VEC4 },
  1008. { "LIGHT", TYPE_VEC4 },
  1009. { "SHADOW", TYPE_VEC4 },
  1010. { "POINT_COORD", TYPE_VEC2 },
  1011. // { "SCREEN_POS", TYPE_VEC2},
  1012. // { "SCREEN_TEXEL_SIZE", TYPE_VEC2},
  1013. { "TIME", TYPE_FLOAT },
  1014. { NULL, TYPE_VOID }
  1015. };
  1016. const ShaderLanguage::BuiltinsDef ShaderLanguage::postprocess_fragment_builtins_defs[] = {
  1017. { "IN_COLOR", TYPE_VEC3 },
  1018. { "IN_POSITION", TYPE_VEC3 },
  1019. { "OUT_COLOR", TYPE_VEC3 },
  1020. { "SCREEN_POS", TYPE_VEC2 },
  1021. { "SCREEN_TEXEL_SIZE", TYPE_VEC2 },
  1022. { "TIME", TYPE_FLOAT },
  1023. { NULL, TYPE_VOID }
  1024. };
  1025. ShaderLanguage::DataType ShaderLanguage::compute_node_type(Node *p_node) {
  1026. switch (p_node->type) {
  1027. case Node::TYPE_PROGRAM: ERR_FAIL_V(TYPE_VOID);
  1028. case Node::TYPE_FUNCTION: return static_cast<FunctionNode *>(p_node)->return_type;
  1029. case Node::TYPE_BLOCK: ERR_FAIL_V(TYPE_VOID);
  1030. case Node::TYPE_VARIABLE: return static_cast<VariableNode *>(p_node)->datatype_cache;
  1031. case Node::TYPE_CONSTANT: return static_cast<ConstantNode *>(p_node)->datatype;
  1032. case Node::TYPE_OPERATOR: return static_cast<OperatorNode *>(p_node)->return_cache;
  1033. case Node::TYPE_CONTROL_FLOW: ERR_FAIL_V(TYPE_VOID);
  1034. case Node::TYPE_MEMBER: return static_cast<MemberNode *>(p_node)->datatype;
  1035. }
  1036. return TYPE_VOID;
  1037. }
  1038. ShaderLanguage::Node *ShaderLanguage::validate_function_call(Parser &parser, OperatorNode *p_func) {
  1039. ERR_FAIL_COND_V(p_func->op != OP_CALL && p_func->op != OP_CONSTRUCT, NULL);
  1040. Vector<DataType> args;
  1041. ERR_FAIL_COND_V(p_func->arguments[0]->type != Node::TYPE_VARIABLE, NULL);
  1042. String name = static_cast<VariableNode *>(p_func->arguments[0])->name.operator String();
  1043. bool all_const = true;
  1044. for (int i = 1; i < p_func->arguments.size(); i++) {
  1045. if (p_func->arguments[i]->type != Node::TYPE_CONSTANT)
  1046. all_const = false;
  1047. args.push_back(compute_node_type(p_func->arguments[i]));
  1048. }
  1049. int argcount = args.size();
  1050. bool found_intrinsic = false;
  1051. if (argcount <= 4) {
  1052. // test intrinsics
  1053. int idx = 0;
  1054. while (intrinsic_func_defs[idx].name) {
  1055. if (name == intrinsic_func_defs[idx].name) {
  1056. bool fail = false;
  1057. for (int i = 0; i < argcount; i++) {
  1058. if (args[i] != intrinsic_func_defs[idx].args[i]) {
  1059. fail = true;
  1060. break;
  1061. }
  1062. }
  1063. if (!fail && argcount < 4 && intrinsic_func_defs[idx].args[argcount] != TYPE_VOID)
  1064. fail = true; //make sure the number of arguments matches
  1065. if (!fail) {
  1066. p_func->return_cache = intrinsic_func_defs[idx].rettype;
  1067. found_intrinsic = true;
  1068. break;
  1069. }
  1070. }
  1071. idx++;
  1072. }
  1073. }
  1074. if (found_intrinsic) {
  1075. if (p_func->op == OP_CONSTRUCT && all_const) {
  1076. Vector<float> cdata;
  1077. for (int i = 0; i < argcount; i++) {
  1078. Variant v = static_cast<ConstantNode *>(p_func->arguments[i + 1])->value;
  1079. switch (v.get_type()) {
  1080. case Variant::REAL: cdata.push_back(v); break;
  1081. case Variant::VECTOR2: {
  1082. Vector2 v2 = v;
  1083. cdata.push_back(v2.x);
  1084. cdata.push_back(v2.y);
  1085. } break;
  1086. case Variant::VECTOR3: {
  1087. Vector3 v3 = v;
  1088. cdata.push_back(v3.x);
  1089. cdata.push_back(v3.y);
  1090. cdata.push_back(v3.z);
  1091. } break;
  1092. case Variant::PLANE: {
  1093. Plane v4 = v;
  1094. cdata.push_back(v4.normal.x);
  1095. cdata.push_back(v4.normal.y);
  1096. cdata.push_back(v4.normal.z);
  1097. cdata.push_back(v4.d);
  1098. } break;
  1099. default: ERR_FAIL_V(NULL);
  1100. }
  1101. }
  1102. ConstantNode *cn = parser.create_node<ConstantNode>(p_func->parent);
  1103. Variant data;
  1104. switch (p_func->return_cache) {
  1105. case TYPE_FLOAT: data = cdata[0]; break;
  1106. case TYPE_VEC2:
  1107. if (cdata.size() == 1)
  1108. data = Vector2(cdata[0], cdata[0]);
  1109. else
  1110. data = Vector2(cdata[0], cdata[1]);
  1111. break;
  1112. case TYPE_VEC3:
  1113. if (cdata.size() == 1)
  1114. data = Vector3(cdata[0], cdata[0], cdata[0]);
  1115. else
  1116. data = Vector3(cdata[0], cdata[1], cdata[2]);
  1117. break;
  1118. case TYPE_VEC4:
  1119. if (cdata.size() == 1)
  1120. data = Plane(cdata[0], cdata[0], cdata[0], cdata[0]);
  1121. else
  1122. data = Plane(cdata[0], cdata[1], cdata[2], cdata[3]);
  1123. break;
  1124. }
  1125. cn->datatype = p_func->return_cache;
  1126. cn->value = data;
  1127. return cn;
  1128. }
  1129. return p_func;
  1130. }
  1131. // try existing functions..
  1132. FunctionNode *exclude_function = NULL; //exclude current function (in case inside one)
  1133. Node *node = p_func;
  1134. while (node->parent) {
  1135. if (node->type == Node::TYPE_FUNCTION) {
  1136. exclude_function = (FunctionNode *)node;
  1137. }
  1138. node = node->parent;
  1139. }
  1140. ERR_FAIL_COND_V(node->type != Node::TYPE_PROGRAM, NULL);
  1141. ProgramNode *program = (ProgramNode *)node;
  1142. for (int i = 0; i < program->functions.size(); i++) {
  1143. if (program->functions[i].function == exclude_function)
  1144. continue;
  1145. FunctionNode *pfunc = program->functions[i].function;
  1146. if (pfunc->arguments.size() != args.size())
  1147. continue;
  1148. bool fail = false;
  1149. for (int i = 0; i < args.size(); i++) {
  1150. if (args[i] != pfunc->arguments[i].type) {
  1151. fail = true;
  1152. break;
  1153. }
  1154. }
  1155. if (!fail && name == program->functions[i].name) {
  1156. p_func->return_cache = pfunc->return_type;
  1157. return p_func;
  1158. }
  1159. }
  1160. return NULL;
  1161. }
  1162. ShaderLanguage::Node *ShaderLanguage::validate_operator(Parser &parser, OperatorNode *p_func) {
  1163. int argcount = p_func->arguments.size();
  1164. ERR_FAIL_COND_V(argcount > 2, NULL);
  1165. DataType argtype[2] = { TYPE_VOID, TYPE_VOID };
  1166. bool all_const = true;
  1167. for (int i = 0; i < argcount; i++) {
  1168. argtype[i] = compute_node_type(p_func->arguments[i]);
  1169. if (p_func->arguments[i]->type != Node::TYPE_CONSTANT)
  1170. all_const = false;
  1171. }
  1172. int idx = 0;
  1173. bool valid = false;
  1174. while (operator_defs[idx].op != OP_MAX) {
  1175. if (p_func->op == operator_defs[idx].op) {
  1176. if (operator_defs[idx].args[0] == argtype[0] && operator_defs[idx].args[1] == argtype[1]) {
  1177. p_func->return_cache = operator_defs[idx].rettype;
  1178. valid = true;
  1179. break;
  1180. }
  1181. }
  1182. idx++;
  1183. }
  1184. if (!valid)
  1185. return NULL;
  1186. #define _RCO2(m_op, m_vop) \
  1187. case m_op: { \
  1188. ConstantNode *cn = parser.create_node<ConstantNode>(p_func->parent); \
  1189. cn->datatype = p_func->return_cache; \
  1190. Variant::evaluate(m_vop, static_cast<ConstantNode *>(p_func->arguments[0])->value, static_cast<ConstantNode *>(p_func->arguments[1])->value, cn->value, valid); \
  1191. if (!valid) \
  1192. return NULL; \
  1193. return cn; \
  1194. } break;
  1195. #define _RCO1(m_op, m_vop) \
  1196. case m_op: { \
  1197. ConstantNode *cn = parser.create_node<ConstantNode>(p_func->parent); \
  1198. cn->datatype = p_func->return_cache; \
  1199. Variant::evaluate(m_vop, static_cast<ConstantNode *>(p_func->arguments[0])->value, Variant(), cn->value, valid); \
  1200. if (!valid) \
  1201. return NULL; \
  1202. return cn; \
  1203. } break;
  1204. if (all_const) {
  1205. //reduce constant operator
  1206. switch (p_func->op) {
  1207. _RCO2(OP_ADD, Variant::OP_ADD);
  1208. _RCO2(OP_SUB, Variant::OP_SUBSTRACT);
  1209. _RCO2(OP_MUL, Variant::OP_MULTIPLY);
  1210. _RCO2(OP_DIV, Variant::OP_DIVIDE);
  1211. _RCO1(OP_NEG, Variant::OP_NEGATE);
  1212. _RCO1(OP_NOT, Variant::OP_NOT);
  1213. _RCO2(OP_CMP_EQ, Variant::OP_EQUAL);
  1214. _RCO2(OP_CMP_NEQ, Variant::OP_NOT_EQUAL);
  1215. _RCO2(OP_CMP_LEQ, Variant::OP_LESS_EQUAL);
  1216. _RCO2(OP_CMP_GEQ, Variant::OP_GREATER_EQUAL);
  1217. _RCO2(OP_CMP_LESS, Variant::OP_LESS);
  1218. _RCO2(OP_CMP_GREATER, Variant::OP_GREATER);
  1219. _RCO2(OP_CMP_OR, Variant::OP_OR);
  1220. _RCO2(OP_CMP_AND, Variant::OP_AND);
  1221. default: {}
  1222. }
  1223. }
  1224. return p_func;
  1225. }
  1226. bool ShaderLanguage::is_token_operator(TokenType p_type) {
  1227. return (p_type == TK_OP_EQUAL) ||
  1228. (p_type == TK_OP_NOT_EQUAL) ||
  1229. (p_type == TK_OP_LESS) ||
  1230. (p_type == TK_OP_LESS_EQUAL) ||
  1231. (p_type == TK_OP_GREATER) ||
  1232. (p_type == TK_OP_GREATER_EQUAL) ||
  1233. (p_type == TK_OP_AND) ||
  1234. (p_type == TK_OP_OR) ||
  1235. (p_type == TK_OP_NOT) ||
  1236. (p_type == TK_OP_ADD) ||
  1237. (p_type == TK_OP_SUB) ||
  1238. (p_type == TK_OP_MUL) ||
  1239. (p_type == TK_OP_DIV) ||
  1240. (p_type == TK_OP_NEG) ||
  1241. (p_type == TK_OP_ASSIGN) ||
  1242. (p_type == TK_OP_ASSIGN_ADD) ||
  1243. (p_type == TK_OP_ASSIGN_SUB) ||
  1244. (p_type == TK_OP_ASSIGN_MUL) ||
  1245. (p_type == TK_OP_ASSIGN_DIV);
  1246. }
  1247. ShaderLanguage::Operator ShaderLanguage::get_token_operator(TokenType p_type) {
  1248. switch (p_type) {
  1249. case TK_OP_EQUAL: return OP_CMP_EQ;
  1250. case TK_OP_NOT_EQUAL: return OP_CMP_NEQ;
  1251. case TK_OP_LESS: return OP_CMP_LESS;
  1252. case TK_OP_LESS_EQUAL: return OP_CMP_LEQ;
  1253. case TK_OP_GREATER: return OP_CMP_GREATER;
  1254. case TK_OP_GREATER_EQUAL: return OP_CMP_GEQ;
  1255. case TK_OP_AND: return OP_CMP_AND;
  1256. case TK_OP_OR: return OP_CMP_OR;
  1257. case TK_OP_NOT: return OP_NOT;
  1258. case TK_OP_ADD: return OP_ADD;
  1259. case TK_OP_SUB: return OP_SUB;
  1260. case TK_OP_MUL: return OP_MUL;
  1261. case TK_OP_DIV: return OP_DIV;
  1262. case TK_OP_NEG: return OP_NEG;
  1263. case TK_OP_ASSIGN: return OP_ASSIGN;
  1264. case TK_OP_ASSIGN_ADD: return OP_ASSIGN_ADD;
  1265. case TK_OP_ASSIGN_SUB: return OP_ASSIGN_SUB;
  1266. case TK_OP_ASSIGN_MUL: return OP_ASSIGN_MUL;
  1267. case TK_OP_ASSIGN_DIV: return OP_ASSIGN_DIV;
  1268. default: ERR_FAIL_V(OP_MAX);
  1269. }
  1270. return OP_MAX;
  1271. }
  1272. Error ShaderLanguage::parse_expression(Parser &parser, Node *p_parent, Node **r_expr) {
  1273. Vector<Expression> expression;
  1274. //Vector<TokenType> operators;
  1275. while (true) {
  1276. Node *expr = NULL;
  1277. if (parser.get_next_token_type() == TK_PARENTHESIS_OPEN) {
  1278. //handle subexpression
  1279. parser.advance();
  1280. Error err = parse_expression(parser, p_parent, &expr);
  1281. if (err)
  1282. return err;
  1283. if (parser.get_next_token_type() != TK_PARENTHESIS_CLOSE) {
  1284. parser.set_error("Expected ')' in expression");
  1285. return ERR_PARSE_ERROR;
  1286. }
  1287. parser.advance();
  1288. } else if (parser.get_next_token_type() == TK_REAL_CONSTANT) {
  1289. ConstantNode *constant = parser.create_node<ConstantNode>(p_parent);
  1290. constant->value = parser.get_next_token().text.operator String().to_double();
  1291. constant->datatype = TYPE_FLOAT;
  1292. expr = constant;
  1293. parser.advance();
  1294. } else if (parser.get_next_token_type() == TK_TRUE) {
  1295. //print_line("found true");
  1296. //handle true constant
  1297. ConstantNode *constant = parser.create_node<ConstantNode>(p_parent);
  1298. constant->value = true;
  1299. constant->datatype = TYPE_BOOL;
  1300. expr = constant;
  1301. parser.advance();
  1302. } else if (parser.get_next_token_type() == TK_FALSE) {
  1303. //handle false constant
  1304. ConstantNode *constant = parser.create_node<ConstantNode>(p_parent);
  1305. constant->value = false;
  1306. constant->datatype = TYPE_BOOL;
  1307. expr = constant;
  1308. parser.advance();
  1309. } else if (parser.get_next_token_type() == TK_TYPE_VOID) {
  1310. //make sure void is not used in expression
  1311. parser.set_error("Void value not allowed in Expression");
  1312. return ERR_PARSE_ERROR;
  1313. } else if (parser.get_next_token_type(1) == TK_PARENTHESIS_OPEN && (is_token_nonvoid_datatype(parser.get_next_token_type()) || parser.get_next_token_type() == TK_INDENTIFIER)) {
  1314. //function or constructor
  1315. StringName name;
  1316. DataType constructor = TYPE_VOID;
  1317. if (is_token_nonvoid_datatype(parser.get_next_token_type())) {
  1318. constructor = get_token_datatype(parser.get_next_token_type());
  1319. switch (get_token_datatype(parser.get_next_token_type())) {
  1320. case TYPE_BOOL: name = "bool"; break;
  1321. case TYPE_FLOAT: name = "float"; break;
  1322. case TYPE_VEC2: name = "vec2"; break;
  1323. case TYPE_VEC3: name = "vec3"; break;
  1324. case TYPE_VEC4: name = "vec4"; break;
  1325. case TYPE_MAT2: name = "mat2"; break;
  1326. case TYPE_MAT3: name = "mat3"; break;
  1327. case TYPE_MAT4: name = "mat4"; break;
  1328. default: ERR_FAIL_V(ERR_BUG);
  1329. }
  1330. } else {
  1331. name = parser.get_next_token().text;
  1332. }
  1333. if (!test_existing_identifier(p_parent, name)) {
  1334. parser.set_error("Unknown identifier in expression: " + name);
  1335. return ERR_PARSE_ERROR;
  1336. }
  1337. parser.advance(2);
  1338. OperatorNode *func = parser.create_node<OperatorNode>(p_parent);
  1339. func->op = constructor != TYPE_VOID ? OP_CONSTRUCT : OP_CALL;
  1340. VariableNode *funcname = parser.create_node<VariableNode>(func);
  1341. funcname->name = name;
  1342. func->arguments.push_back(funcname);
  1343. //parse parameters
  1344. if (parser.get_next_token_type() == TK_PARENTHESIS_CLOSE) {
  1345. parser.advance();
  1346. } else {
  1347. while (true) {
  1348. Node *arg = NULL;
  1349. Error err = parse_expression(parser, func, &arg);
  1350. if (err)
  1351. return err;
  1352. func->arguments.push_back(arg);
  1353. if (parser.get_next_token_type() == TK_PARENTHESIS_CLOSE) {
  1354. parser.advance();
  1355. break;
  1356. } else if (parser.get_next_token_type() == TK_COMMA) {
  1357. if (parser.get_next_token_type(1) == TK_PARENTHESIS_CLOSE) {
  1358. parser.set_error("Expression expected");
  1359. return ERR_PARSE_ERROR;
  1360. }
  1361. parser.advance();
  1362. } else {
  1363. // something is broken
  1364. parser.set_error("Expected ',' or ')'");
  1365. return ERR_PARSE_ERROR;
  1366. }
  1367. }
  1368. }
  1369. expr = validate_function_call(parser, func);
  1370. if (!expr) {
  1371. parser.set_error("Invalid arguments to function/constructor: " + StringName(name));
  1372. return ERR_PARSE_ERROR;
  1373. }
  1374. } else if (parser.get_next_token_type() == TK_INDENTIFIER) {
  1375. //probably variable
  1376. Node *node = p_parent;
  1377. bool existing = false;
  1378. DataType datatype;
  1379. StringName identifier = parser.get_next_token().text;
  1380. while (node) {
  1381. if (node->type == Node::TYPE_BLOCK) {
  1382. BlockNode *block = (BlockNode *)node;
  1383. if (block->variables.has(identifier)) {
  1384. existing = true;
  1385. datatype = block->variables[identifier];
  1386. break;
  1387. }
  1388. }
  1389. if (node->type == Node::TYPE_FUNCTION) {
  1390. FunctionNode *function = (FunctionNode *)node;
  1391. for (int i = 0; i < function->arguments.size(); i++) {
  1392. if (function->arguments[i].name == identifier) {
  1393. existing = true;
  1394. datatype = function->arguments[i].type;
  1395. break;
  1396. }
  1397. }
  1398. if (existing)
  1399. break;
  1400. }
  1401. if (node->type == Node::TYPE_PROGRAM) {
  1402. ProgramNode *program = (ProgramNode *)node;
  1403. if (program->builtin_variables.has(identifier)) {
  1404. datatype = program->builtin_variables[identifier];
  1405. existing = true;
  1406. break;
  1407. }
  1408. if (program->uniforms.has(identifier)) {
  1409. datatype = program->uniforms[identifier].type;
  1410. existing = true;
  1411. break;
  1412. }
  1413. }
  1414. node = node->parent;
  1415. }
  1416. if (!existing) {
  1417. parser.set_error("Nonexistent identifier in expression: " + identifier);
  1418. return ERR_PARSE_ERROR;
  1419. }
  1420. VariableNode *varname = parser.create_node<VariableNode>(p_parent);
  1421. varname->name = identifier;
  1422. varname->datatype_cache = datatype;
  1423. parser.advance();
  1424. expr = varname;
  1425. } else if (parser.get_next_token_type() == TK_OP_SUB || parser.get_next_token_type() == TK_OP_NOT) {
  1426. //single prefix operators
  1427. TokenType token_type = parser.get_next_token_type();
  1428. parser.advance();
  1429. //Node *subexpr=NULL;
  1430. //Error err = parse_expression(parser,p_parent,&subexpr);
  1431. //if (err)
  1432. // return err;
  1433. //OperatorNode *op = parser.create_node<OperatorNode>(p_parent);
  1434. Expression e;
  1435. e.is_op = true;
  1436. switch (token_type) {
  1437. case TK_OP_SUB: e.op = TK_OP_NEG; break;
  1438. case TK_OP_NOT:
  1439. e.op = TK_OP_NOT;
  1440. break;
  1441. //case TK_OP_PLUS_PLUS: op->op=OP_PLUS_PLUS; break;
  1442. //case TK_OP_MINUS_MINUS: op->op=OP_MINUS_MINUS; break;
  1443. default: ERR_FAIL_V(ERR_BUG);
  1444. }
  1445. expression.push_back(e);
  1446. continue;
  1447. } else {
  1448. print_line("found bug?");
  1449. print_line("misplaced token: " + String(token_names[parser.get_next_token_type()]));
  1450. parser.set_error("Error parsing expression, misplaced: " + String(token_names[parser.get_next_token_type()]));
  1451. return ERR_PARSE_ERROR;
  1452. //nothing
  1453. }
  1454. ERR_FAIL_COND_V(!expr, ERR_BUG);
  1455. /* OK now see what's NEXT to the operator.. */
  1456. /* OK now see what's NEXT to the operator.. */
  1457. /* OK now see what's NEXT to the operator.. */
  1458. if (parser.get_next_token_type() == TK_PERIOD) {
  1459. if (parser.get_next_token_type(1) != TK_INDENTIFIER) {
  1460. parser.set_error("Expected identifier as member");
  1461. return ERR_PARSE_ERROR;
  1462. }
  1463. DataType dt = compute_node_type(expr);
  1464. String ident = parser.get_next_token(1).text;
  1465. bool ok = true;
  1466. DataType member_type;
  1467. switch (dt) {
  1468. case TYPE_VEC2: {
  1469. int l = ident.length();
  1470. if (l == 1) {
  1471. member_type = TYPE_FLOAT;
  1472. } else if (l == 2) {
  1473. member_type = TYPE_VEC2;
  1474. } else {
  1475. ok = false;
  1476. break;
  1477. }
  1478. const CharType *c = ident.ptr();
  1479. for (int i = 0; i < l; i++) {
  1480. switch (c[i]) {
  1481. case 'r':
  1482. case 'g':
  1483. case 'x':
  1484. case 'y':
  1485. break;
  1486. default:
  1487. ok = false;
  1488. break;
  1489. }
  1490. }
  1491. } break;
  1492. case TYPE_VEC3: {
  1493. int l = ident.length();
  1494. if (l == 1) {
  1495. member_type = TYPE_FLOAT;
  1496. } else if (l == 2) {
  1497. member_type = TYPE_VEC2;
  1498. } else if (l == 3) {
  1499. member_type = TYPE_VEC3;
  1500. } else {
  1501. ok = false;
  1502. break;
  1503. }
  1504. const CharType *c = ident.ptr();
  1505. for (int i = 0; i < l; i++) {
  1506. switch (c[i]) {
  1507. case 'r':
  1508. case 'g':
  1509. case 'b':
  1510. case 'x':
  1511. case 'y':
  1512. case 'z':
  1513. break;
  1514. default:
  1515. ok = false;
  1516. break;
  1517. }
  1518. }
  1519. } break;
  1520. case TYPE_VEC4: {
  1521. int l = ident.length();
  1522. if (l == 1) {
  1523. member_type = TYPE_FLOAT;
  1524. } else if (l == 2) {
  1525. member_type = TYPE_VEC2;
  1526. } else if (l == 3) {
  1527. member_type = TYPE_VEC3;
  1528. } else if (l == 4) {
  1529. member_type = TYPE_VEC4;
  1530. } else {
  1531. ok = false;
  1532. break;
  1533. }
  1534. const CharType *c = ident.ptr();
  1535. for (int i = 0; i < l; i++) {
  1536. switch (c[i]) {
  1537. case 'r':
  1538. case 'g':
  1539. case 'b':
  1540. case 'a':
  1541. case 'x':
  1542. case 'y':
  1543. case 'z':
  1544. case 'w':
  1545. break;
  1546. default:
  1547. ok = false;
  1548. break;
  1549. }
  1550. }
  1551. } break;
  1552. case TYPE_MAT2:
  1553. ok = (ident == "x" || ident == "y");
  1554. member_type = TYPE_VEC2;
  1555. break;
  1556. case TYPE_MAT3:
  1557. ok = (ident == "x" || ident == "y" || ident == "z");
  1558. member_type = TYPE_VEC3;
  1559. break;
  1560. case TYPE_MAT4:
  1561. ok = (ident == "x" || ident == "y" || ident == "z" || ident == "w");
  1562. member_type = TYPE_VEC4;
  1563. break;
  1564. default: {}
  1565. }
  1566. if (!ok) {
  1567. parser.set_error("Invalid member for expression: ." + ident);
  1568. return ERR_PARSE_ERROR;
  1569. }
  1570. MemberNode *mn = parser.create_node<MemberNode>(p_parent);
  1571. mn->basetype = dt;
  1572. mn->datatype = member_type;
  1573. mn->name = ident;
  1574. mn->owner = expr;
  1575. expr = mn;
  1576. parser.advance(2);
  1577. //todo
  1578. //member (period) has priority over any operator
  1579. //creates a subindexing expression in place
  1580. } else if (parser.get_next_token_type() == TK_BRACKET_OPEN) {
  1581. //todo
  1582. //subindexing has priority over any operator
  1583. //creates a subindexing expression in place
  1584. } /*else if (parser.get_next_token_type()==TK_OP_PLUS_PLUS || parser.get_next_token_type()==TK_OP_MINUS_MINUS) {
  1585. //todo
  1586. //inc/dec operators have priority over any operator
  1587. //creates a subindexing expression in place
  1588. //return OK; //wtfs
  1589. } */
  1590. Expression e;
  1591. e.is_op = false;
  1592. e.node = expr;
  1593. expression.push_back(e);
  1594. if (is_token_operator(parser.get_next_token_type())) {
  1595. Expression o;
  1596. o.is_op = true;
  1597. o.op = parser.get_next_token_type();
  1598. expression.push_back(o);
  1599. parser.advance();
  1600. } else {
  1601. break;
  1602. }
  1603. }
  1604. /* Reduce the set set of expressions and place them in an operator tree, respecting precedence */
  1605. while (expression.size() > 1) {
  1606. int next_op = -1;
  1607. int min_priority = 0xFFFFF;
  1608. bool is_unary = false;
  1609. for (int i = 0; i < expression.size(); i++) {
  1610. if (!expression[i].is_op) {
  1611. continue;
  1612. }
  1613. bool unary = false;
  1614. int priority;
  1615. switch (expression[i].op) {
  1616. case TK_OP_NOT:
  1617. priority = 0;
  1618. unary = true;
  1619. break;
  1620. case TK_OP_NEG:
  1621. priority = 0;
  1622. unary = true;
  1623. break;
  1624. case TK_OP_MUL: priority = 1; break;
  1625. case TK_OP_DIV: priority = 1; break;
  1626. case TK_OP_ADD: priority = 2; break;
  1627. case TK_OP_SUB:
  1628. priority = 2;
  1629. break;
  1630. // shift left/right =2
  1631. case TK_OP_LESS: priority = 4; break;
  1632. case TK_OP_LESS_EQUAL: priority = 4; break;
  1633. case TK_OP_GREATER: priority = 4; break;
  1634. case TK_OP_GREATER_EQUAL: priority = 4; break;
  1635. case TK_OP_EQUAL: priority = 5; break;
  1636. case TK_OP_NOT_EQUAL:
  1637. priority = 5;
  1638. break;
  1639. //bit and =5
  1640. //bit xor =6
  1641. //bit or=7
  1642. case TK_OP_AND: priority = 8; break;
  1643. case TK_OP_OR:
  1644. priority = 9;
  1645. break;
  1646. // ?: = 10
  1647. case TK_OP_ASSIGN_ADD: priority = 11; break;
  1648. case TK_OP_ASSIGN_SUB: priority = 11; break;
  1649. case TK_OP_ASSIGN_MUL: priority = 11; break;
  1650. case TK_OP_ASSIGN_DIV: priority = 11; break;
  1651. case TK_OP_ASSIGN: priority = 11; break;
  1652. default:
  1653. ERR_FAIL_V(ERR_BUG); //unexpected operator
  1654. }
  1655. if (priority < min_priority) {
  1656. // < is used for left to right (default)
  1657. // <= is used for right to left
  1658. next_op = i;
  1659. min_priority = priority;
  1660. is_unary = unary;
  1661. }
  1662. }
  1663. ERR_FAIL_COND_V(next_op == -1, ERR_BUG);
  1664. // OK! create operator..
  1665. // OK! create operator..
  1666. if (is_unary) {
  1667. int expr_pos = next_op;
  1668. while (expression[expr_pos].is_op) {
  1669. expr_pos++;
  1670. if (expr_pos == expression.size()) {
  1671. //can happen..
  1672. parser.set_error("Unexpected end of expression..");
  1673. return ERR_BUG;
  1674. }
  1675. }
  1676. //consecutively do unary opeators
  1677. for (int i = expr_pos - 1; i >= next_op; i--) {
  1678. OperatorNode *op = parser.create_node<OperatorNode>(p_parent);
  1679. op->op = get_token_operator(expression[i].op);
  1680. op->arguments.push_back(expression[i + 1].node);
  1681. expression[i].is_op = false;
  1682. expression[i].node = validate_operator(parser, op);
  1683. if (!expression[i].node) {
  1684. String at;
  1685. for (int i = 0; i < op->arguments.size(); i++) {
  1686. if (i > 0)
  1687. at += " and ";
  1688. at += get_datatype_name(compute_node_type(op->arguments[i]));
  1689. }
  1690. parser.set_error("Invalid argument to unary operator " + String(token_names[op->op]) + ": " + at);
  1691. return ERR_PARSE_ERROR;
  1692. }
  1693. expression.remove(i + 1);
  1694. }
  1695. } else {
  1696. if (next_op < 1 || next_op >= (expression.size() - 1)) {
  1697. parser.set_error("Parser bug..");
  1698. ERR_FAIL_V(ERR_BUG);
  1699. }
  1700. OperatorNode *op = parser.create_node<OperatorNode>(p_parent);
  1701. op->op = get_token_operator(expression[next_op].op);
  1702. if (expression[next_op - 1].is_op) {
  1703. parser.set_error("Parser bug..");
  1704. ERR_FAIL_V(ERR_BUG);
  1705. }
  1706. if (expression[next_op + 1].is_op) {
  1707. // this is not invalid and can really appear
  1708. // but it becomes invalid anyway because no binary op
  1709. // can be followed by an unary op in a valid combination,
  1710. // due to how precedence works, unaries will always dissapear first
  1711. parser.set_error("Parser bug..");
  1712. }
  1713. op->arguments.push_back(expression[next_op - 1].node); //expression goes as left
  1714. op->arguments.push_back(expression[next_op + 1].node); //next expression goes as right
  1715. //replace all 3 nodes by this operator and make it an expression
  1716. expression[next_op - 1].node = validate_operator(parser, op);
  1717. if (!expression[next_op - 1].node) {
  1718. String at;
  1719. for (int i = 0; i < op->arguments.size(); i++) {
  1720. if (i > 0)
  1721. at += " and ";
  1722. at += get_datatype_name(compute_node_type(op->arguments[i]));
  1723. }
  1724. static const char *op_names[OP_MAX] = { "=", "+", "-", "*", "/", "+=", "-=", "*=", "/=", "-", "!", "==", "!=", "<=", ">=", "<", ">", "||", "&&", "call", "()" };
  1725. parser.set_error("Invalid arguments to operator " + String(op_names[op->op]) + ": " + at);
  1726. return ERR_PARSE_ERROR;
  1727. }
  1728. expression.remove(next_op);
  1729. expression.remove(next_op);
  1730. }
  1731. #if 0
  1732. OperatorNode *op = parser.create_node<OperatorNode>(p_parent);
  1733. op->op=get_token_operator(operators[next_op]);
  1734. op->arguments.push_back(expressions[next_op]); //expression goes as left
  1735. op->arguments.push_back(expressions[next_op+1]); //next expression goes as right
  1736. expressions[next_op]=validate_operator(parser,op);
  1737. if (!expressions[next_op]) {
  1738. String at;
  1739. for(int i=0;i<op->arguments.size();i++) {
  1740. if (i>0)
  1741. at+=" and ";
  1742. at+=get_datatype_name(compute_node_type(op->arguments[i]));
  1743. }
  1744. parser.set_error("Invalid arguments to operator "+String(token_names[operators[next_op]])+": "+at);
  1745. return ERR_PARSE_ERROR;
  1746. }
  1747. expressions.remove(next_op+1);
  1748. operators.remove(next_op);
  1749. #endif
  1750. }
  1751. *r_expr = expression[0].node;
  1752. return OK;
  1753. /*
  1754. TokenType token_type=parser.get_next_token_type();
  1755. OperatorNode *op = parser.create_node<OperatorNode>(p_parent);
  1756. op->op=get_token_operator(parser.get_next_token_type());
  1757. op->arguments.push_back(*r_expr); //expression goes as left
  1758. parser.advance();
  1759. Node *right_expr=NULL;
  1760. Error err = parse_expression(parser,p_parent,&right_expr);
  1761. if (err)
  1762. return err;
  1763. op->arguments.push_back(right_expr);
  1764. if (!validate_operator(op)) {
  1765. parser.set_error("Invalid arguments to operator "+String(token_names[token_type]));
  1766. return ERR_PARSE_ERROR;
  1767. }
  1768. */
  1769. }
  1770. Error ShaderLanguage::parse_variable_declaration(Parser &parser, BlockNode *p_block) {
  1771. bool uniform = parser.get_next_token(-1).type == TK_UNIFORM;
  1772. DataType type = get_token_datatype(parser.get_next_token_type(0));
  1773. bool iscolor = parser.get_next_token_type(0) == TK_TYPE_COLOR;
  1774. if (type == TYPE_VOID) {
  1775. parser.set_error("Cannot Declare a 'void' Variable");
  1776. return ERR_PARSE_ERROR;
  1777. }
  1778. if (type == TYPE_TEXTURE && !uniform) {
  1779. parser.set_error("Cannot Declare a Non-Uniform Texture");
  1780. return ERR_PARSE_ERROR;
  1781. }
  1782. if (type == TYPE_CUBEMAP && !uniform) {
  1783. parser.set_error("Cannot Declare a Non-Uniform Cubemap");
  1784. return ERR_PARSE_ERROR;
  1785. }
  1786. parser.advance();
  1787. int found = 0;
  1788. while (true) {
  1789. if (found && parser.get_next_token_type() != TK_COMMA) {
  1790. break;
  1791. }
  1792. if (parser.get_next_token_type() != TK_INDENTIFIER) {
  1793. parser.set_error("Identifier Expected");
  1794. return ERR_PARSE_ERROR;
  1795. }
  1796. StringName name = parser.get_next_token().text;
  1797. if (test_existing_identifier(p_block, name)) {
  1798. parser.set_error("Duplicate Identifier (existing variable/function): " + name);
  1799. return ERR_PARSE_ERROR;
  1800. }
  1801. found = true;
  1802. parser.advance();
  1803. //see if declaration has an initializer
  1804. if (parser.get_next_token_type() == TK_OP_ASSIGN) {
  1805. parser.advance();
  1806. OperatorNode *op = parser.create_node<OperatorNode>(p_block);
  1807. VariableNode *var = parser.create_node<VariableNode>(op);
  1808. var->name = name;
  1809. var->datatype_cache = type;
  1810. var->uniform = uniform;
  1811. Node *expr;
  1812. Error err = parse_expression(parser, p_block, &expr);
  1813. if (err)
  1814. return err;
  1815. if (var->uniform) {
  1816. if (expr->type != Node::TYPE_CONSTANT) {
  1817. parser.set_error("Uniform can only be initialized to a constant.");
  1818. return ERR_PARSE_ERROR;
  1819. }
  1820. Uniform u;
  1821. u.order = parser.program->uniforms.size();
  1822. u.type = type;
  1823. u.default_value = static_cast<ConstantNode *>(expr)->value;
  1824. if (iscolor && u.default_value.get_type() == Variant::PLANE) {
  1825. Color c;
  1826. Plane p = u.default_value;
  1827. c = Color(p.normal.x, p.normal.y, p.normal.z, p.d);
  1828. u.default_value = c;
  1829. }
  1830. parser.program->uniforms[var->name] = u;
  1831. } else {
  1832. op->op = OP_ASSIGN;
  1833. op->arguments.push_back(var);
  1834. op->arguments.push_back(expr);
  1835. Node *n = validate_operator(parser, op);
  1836. if (!n) {
  1837. parser.set_error("Invalid initializer for variable: " + name);
  1838. return ERR_PARSE_ERROR;
  1839. }
  1840. p_block->statements.push_back(n);
  1841. }
  1842. } else {
  1843. //initialize it EMPTY
  1844. OperatorNode *op = parser.create_node<OperatorNode>(p_block);
  1845. VariableNode *var = parser.create_node<VariableNode>(op);
  1846. ConstantNode *con = parser.create_node<ConstantNode>(op);
  1847. var->name = name;
  1848. var->datatype_cache = type;
  1849. var->uniform = uniform;
  1850. con->datatype = type;
  1851. switch (type) {
  1852. case TYPE_BOOL: con->value = false; break;
  1853. case TYPE_FLOAT: con->value = 0.0; break;
  1854. case TYPE_VEC2: con->value = Vector2(); break;
  1855. case TYPE_VEC3: con->value = Vector3(); break;
  1856. case TYPE_VEC4: con->value = iscolor ? Variant(Color()) : Variant(Plane()); break;
  1857. case TYPE_MAT2: con->value = Matrix32(); break;
  1858. case TYPE_MAT3: con->value = Matrix3(); break;
  1859. case TYPE_MAT4: con->value = Transform(); break;
  1860. case TYPE_TEXTURE:
  1861. case TYPE_CUBEMAP: con->value = RID(); break;
  1862. default: {}
  1863. }
  1864. if (uniform) {
  1865. Uniform u;
  1866. u.type = type;
  1867. u.default_value = con->value;
  1868. u.order = parser.program->uniforms.size();
  1869. parser.program->uniforms[var->name] = u;
  1870. } else {
  1871. op->op = OP_ASSIGN;
  1872. op->arguments.push_back(var);
  1873. op->arguments.push_back(con);
  1874. p_block->statements.push_back(op);
  1875. }
  1876. }
  1877. if (!uniform)
  1878. p_block->variables[name] = type;
  1879. }
  1880. if (parser.get_next_token_type() != TK_SEMICOLON) {
  1881. parser.set_error("Expected ';'");
  1882. return ERR_PARSE_ERROR;
  1883. }
  1884. return OK;
  1885. }
  1886. Error ShaderLanguage::parse_flow_if(Parser &parser, Node *p_parent, Node **r_statement) {
  1887. ControlFlowNode *cf = parser.create_node<ControlFlowNode>(p_parent);
  1888. cf->flow_op = FLOW_OP_IF;
  1889. parser.advance();
  1890. if (parser.get_next_token_type() != TK_PARENTHESIS_OPEN) {
  1891. parser.set_error("Expected '(' after 'if'");
  1892. return ERR_PARSE_ERROR;
  1893. }
  1894. parser.advance();
  1895. Node *expression = NULL;
  1896. Error err = parse_expression(parser, cf, &expression);
  1897. if (err)
  1898. return err;
  1899. if (compute_node_type(expression) != TYPE_BOOL) {
  1900. parser.set_error("Expression for 'if' is not boolean");
  1901. return ERR_PARSE_ERROR;
  1902. }
  1903. cf->statements.push_back(expression);
  1904. if (parser.get_next_token_type() != TK_PARENTHESIS_CLOSE) {
  1905. parser.set_error("Expected ')' after expression");
  1906. return ERR_PARSE_ERROR;
  1907. }
  1908. parser.advance();
  1909. if (parser.get_next_token_type() != TK_CURLY_BRACKET_OPEN) {
  1910. parser.set_error("Expected statement block after 'if()'");
  1911. return ERR_PARSE_ERROR;
  1912. }
  1913. Node *substatement = NULL;
  1914. err = parse_statement(parser, cf, &substatement);
  1915. if (err)
  1916. return err;
  1917. cf->statements.push_back(substatement);
  1918. if (parser.get_next_token_type() == TK_CF_ELSE) {
  1919. parser.advance();
  1920. if (parser.get_next_token_type() != TK_CURLY_BRACKET_OPEN) {
  1921. parser.set_error("Expected statement block after 'else'");
  1922. return ERR_PARSE_ERROR;
  1923. }
  1924. substatement = NULL;
  1925. err = parse_statement(parser, cf, &substatement);
  1926. if (err)
  1927. return err;
  1928. cf->statements.push_back(substatement);
  1929. }
  1930. *r_statement = cf;
  1931. return OK;
  1932. }
  1933. Error ShaderLanguage::parse_flow_return(Parser &parser, Node *p_parent, Node **r_statement) {
  1934. FunctionNode *function = NULL;
  1935. Node *parent = p_parent;
  1936. while (parent) {
  1937. if (parent->type == Node::TYPE_FUNCTION) {
  1938. function = (FunctionNode *)parent;
  1939. break;
  1940. }
  1941. parent = parent->parent;
  1942. }
  1943. if (!function) {
  1944. parser.set_error("'return' must be inside a function");
  1945. return ERR_PARSE_ERROR;
  1946. }
  1947. ControlFlowNode *cf = parser.create_node<ControlFlowNode>(p_parent);
  1948. cf->flow_op = FLOW_OP_RETURN;
  1949. parser.advance();
  1950. if (function->return_type != TYPE_VOID) {
  1951. // should expect a return expression.
  1952. Node *expr = NULL;
  1953. Error err = parse_expression(parser, cf, &expr);
  1954. if (err)
  1955. return err;
  1956. if (compute_node_type(expr) != function->return_type) {
  1957. parser.set_error("Invalid type for 'return' expression");
  1958. return ERR_PARSE_ERROR;
  1959. }
  1960. cf->statements.push_back(expr);
  1961. }
  1962. *r_statement = cf;
  1963. if (parser.get_next_token_type() != TK_SEMICOLON) {
  1964. parser.set_error("Expected ';'");
  1965. return ERR_PARSE_ERROR;
  1966. }
  1967. return OK;
  1968. }
  1969. Error ShaderLanguage::parse_statement(Parser &parser, Node *p_parent, Node **r_statement) {
  1970. *r_statement = NULL;
  1971. TokenType token_type = parser.get_next_token_type();
  1972. if (token_type == TK_CURLY_BRACKET_OPEN) {
  1973. //sub-block
  1974. parser.advance();
  1975. BlockNode *block = parser.create_node<BlockNode>(p_parent);
  1976. *r_statement = block;
  1977. return parse_block(parser, block);
  1978. } else if (token_type == TK_SEMICOLON) {
  1979. // empty ;
  1980. parser.advance();
  1981. return OK;
  1982. } else if (token_type == TK_CF_IF) {
  1983. return parse_flow_if(parser, p_parent, r_statement);
  1984. } else if (token_type == TK_CF_RETURN) {
  1985. return parse_flow_return(parser, p_parent, r_statement);
  1986. } else {
  1987. Error err = parse_expression(parser, p_parent, r_statement);
  1988. if (err)
  1989. return err;
  1990. if (parser.get_next_token_type() != TK_SEMICOLON) {
  1991. parser.set_error("Expected ';'");
  1992. return ERR_PARSE_ERROR;
  1993. }
  1994. }
  1995. return OK;
  1996. }
  1997. Error ShaderLanguage::parse_block(Parser &parser, BlockNode *p_block) {
  1998. while (true) {
  1999. if (parser.is_at_end()) {
  2000. if (p_block->parent->type != Node::TYPE_PROGRAM) {
  2001. parser.set_error("Unexpected End of File");
  2002. return ERR_PARSE_ERROR;
  2003. }
  2004. return OK; //bye
  2005. }
  2006. TokenType token_type = parser.get_next_token_type();
  2007. if (token_type == TK_CURLY_BRACKET_CLOSE) {
  2008. if (p_block->parent->type == Node::TYPE_PROGRAM) {
  2009. parser.set_error("Unexpected '}'");
  2010. return ERR_PARSE_ERROR;
  2011. }
  2012. parser.advance();
  2013. return OK; // exit block
  2014. } else if (token_type == TK_UNIFORM) {
  2015. if (p_block != parser.program->body) {
  2016. parser.set_error("Uniform only allowed in main program body.");
  2017. return ERR_PARSE_ERROR;
  2018. }
  2019. parser.advance();
  2020. Error err = parse_variable_declaration(parser, p_block);
  2021. if (err)
  2022. return err;
  2023. } else if (is_token_datatype(token_type)) {
  2024. Error err = OK;
  2025. if (parser_is_at_function(parser))
  2026. err = parse_function(parser, p_block);
  2027. else {
  2028. err = parse_variable_declaration(parser, p_block);
  2029. }
  2030. if (err)
  2031. return err;
  2032. } else {
  2033. // must be a statement
  2034. Node *statement = NULL;
  2035. Error err = parse_statement(parser, p_block, &statement);
  2036. if (err)
  2037. return err;
  2038. if (statement) {
  2039. p_block->statements.push_back(statement);
  2040. }
  2041. }
  2042. }
  2043. return OK;
  2044. }
  2045. Error ShaderLanguage::parse(const Vector<Token> &p_tokens, ShaderType p_type, CompileFunc p_compile_func, void *p_userdata, String *r_error, int *r_err_line, int *r_err_column) {
  2046. Parser parser(p_tokens);
  2047. parser.program = parser.create_node<ProgramNode>(NULL);
  2048. parser.program->body = parser.create_node<BlockNode>(parser.program);
  2049. //add builtins
  2050. switch (p_type) {
  2051. case SHADER_MATERIAL_VERTEX: {
  2052. int idx = 0;
  2053. while (vertex_builtins_defs[idx].name) {
  2054. parser.program->builtin_variables[vertex_builtins_defs[idx].name] = vertex_builtins_defs[idx].type;
  2055. idx++;
  2056. }
  2057. } break;
  2058. case SHADER_MATERIAL_FRAGMENT: {
  2059. int idx = 0;
  2060. while (fragment_builtins_defs[idx].name) {
  2061. parser.program->builtin_variables[fragment_builtins_defs[idx].name] = fragment_builtins_defs[idx].type;
  2062. idx++;
  2063. }
  2064. } break;
  2065. case SHADER_MATERIAL_LIGHT: {
  2066. int idx = 0;
  2067. while (light_builtins_defs[idx].name) {
  2068. parser.program->builtin_variables[light_builtins_defs[idx].name] = light_builtins_defs[idx].type;
  2069. idx++;
  2070. }
  2071. } break;
  2072. case SHADER_CANVAS_ITEM_VERTEX: {
  2073. int idx = 0;
  2074. while (ci_vertex_builtins_defs[idx].name) {
  2075. parser.program->builtin_variables[ci_vertex_builtins_defs[idx].name] = ci_vertex_builtins_defs[idx].type;
  2076. idx++;
  2077. }
  2078. } break;
  2079. case SHADER_CANVAS_ITEM_FRAGMENT: {
  2080. int idx = 0;
  2081. while (ci_fragment_builtins_defs[idx].name) {
  2082. parser.program->builtin_variables[ci_fragment_builtins_defs[idx].name] = ci_fragment_builtins_defs[idx].type;
  2083. idx++;
  2084. }
  2085. } break;
  2086. case SHADER_CANVAS_ITEM_LIGHT: {
  2087. int idx = 0;
  2088. while (ci_light_builtins_defs[idx].name) {
  2089. parser.program->builtin_variables[ci_light_builtins_defs[idx].name] = ci_light_builtins_defs[idx].type;
  2090. idx++;
  2091. }
  2092. } break;
  2093. case SHADER_POST_PROCESS: {
  2094. int idx = 0;
  2095. while (postprocess_fragment_builtins_defs[idx].name) {
  2096. parser.program->builtin_variables[postprocess_fragment_builtins_defs[idx].name] = postprocess_fragment_builtins_defs[idx].type;
  2097. idx++;
  2098. }
  2099. } break;
  2100. }
  2101. Error err = parse_block(parser, parser.program->body);
  2102. if (err) {
  2103. parser.get_error(r_error, r_err_line, r_err_column);
  2104. return err;
  2105. }
  2106. if (p_compile_func) {
  2107. err = p_compile_func(p_userdata, parser.program);
  2108. }
  2109. //clean up nodes created
  2110. while (parser.nodegc.size()) {
  2111. memdelete(parser.nodegc.front()->get());
  2112. parser.nodegc.pop_front();
  2113. }
  2114. return err;
  2115. }
  2116. Error ShaderLanguage::compile(const String &p_code, ShaderType p_type, CompileFunc p_compile_func, void *p_userdata, String *r_error, int *r_err_line, int *r_err_column) {
  2117. *r_error = "";
  2118. *r_err_line = 0;
  2119. *r_err_column = 0;
  2120. Vector<Token> tokens;
  2121. Error err = tokenize(p_code, &tokens, r_error, r_err_line, r_err_column);
  2122. if (err != OK) {
  2123. print_line("tokenizer error!");
  2124. }
  2125. if (err != OK) {
  2126. return err;
  2127. }
  2128. err = parse(tokens, p_type, p_compile_func, p_userdata, r_error, r_err_line, r_err_column);
  2129. if (err != OK) {
  2130. return err;
  2131. }
  2132. return OK;
  2133. }
  2134. void ShaderLanguage::get_keyword_list(ShaderType p_type, List<String> *p_keywords) {
  2135. int idx = 0;
  2136. p_keywords->push_back("uniform");
  2137. p_keywords->push_back("texture");
  2138. p_keywords->push_back("cubemap");
  2139. p_keywords->push_back("color");
  2140. p_keywords->push_back("if");
  2141. p_keywords->push_back("else");
  2142. while (intrinsic_func_defs[idx].name) {
  2143. p_keywords->push_back(intrinsic_func_defs[idx].name);
  2144. idx++;
  2145. }
  2146. switch (p_type) {
  2147. case SHADER_MATERIAL_VERTEX: {
  2148. idx = 0;
  2149. while (vertex_builtins_defs[idx].name) {
  2150. p_keywords->push_back(vertex_builtins_defs[idx].name);
  2151. idx++;
  2152. }
  2153. } break;
  2154. case SHADER_MATERIAL_FRAGMENT: {
  2155. idx = 0;
  2156. while (fragment_builtins_defs[idx].name) {
  2157. p_keywords->push_back(fragment_builtins_defs[idx].name);
  2158. idx++;
  2159. }
  2160. } break;
  2161. case SHADER_MATERIAL_LIGHT: {
  2162. idx = 0;
  2163. while (light_builtins_defs[idx].name) {
  2164. p_keywords->push_back(light_builtins_defs[idx].name);
  2165. idx++;
  2166. }
  2167. } break;
  2168. case SHADER_CANVAS_ITEM_VERTEX: {
  2169. idx = 0;
  2170. while (ci_vertex_builtins_defs[idx].name) {
  2171. p_keywords->push_back(ci_vertex_builtins_defs[idx].name);
  2172. idx++;
  2173. }
  2174. } break;
  2175. case SHADER_CANVAS_ITEM_FRAGMENT: {
  2176. idx = 0;
  2177. while (ci_fragment_builtins_defs[idx].name) {
  2178. p_keywords->push_back(ci_fragment_builtins_defs[idx].name);
  2179. idx++;
  2180. }
  2181. } break;
  2182. case SHADER_CANVAS_ITEM_LIGHT: {
  2183. idx = 0;
  2184. while (ci_light_builtins_defs[idx].name) {
  2185. p_keywords->push_back(ci_light_builtins_defs[idx].name);
  2186. idx++;
  2187. }
  2188. } break;
  2189. case SHADER_POST_PROCESS: {
  2190. idx = 0;
  2191. while (postprocess_fragment_builtins_defs[idx].name) {
  2192. p_keywords->push_back(postprocess_fragment_builtins_defs[idx].name);
  2193. idx++;
  2194. }
  2195. } break;
  2196. }
  2197. }