shader_language.cpp 117 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134
  1. /*************************************************************************/
  2. /* shader_language.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2017 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. String ShaderLanguage::get_operator_text(Operator p_op) {
  43. static const char *op_names[OP_MAX] = { "==",
  44. "!=",
  45. "<",
  46. "<=",
  47. ">",
  48. ">=",
  49. "&&",
  50. "||",
  51. "!",
  52. "-",
  53. "+",
  54. "-",
  55. "*",
  56. "/",
  57. "%",
  58. "<<",
  59. ">>",
  60. "=",
  61. "+=",
  62. "-=",
  63. "*=",
  64. "/=",
  65. "%=",
  66. "<<=",
  67. ">>=",
  68. "&=",
  69. "|=",
  70. "^=",
  71. "&",
  72. "|",
  73. "^",
  74. "~",
  75. "++"
  76. "--",
  77. "()",
  78. "construct",
  79. "index" };
  80. return op_names[p_op];
  81. }
  82. const char *ShaderLanguage::token_names[TK_MAX] = {
  83. "EMPTY",
  84. "IDENTIFIER",
  85. "TRUE",
  86. "FALSE",
  87. "REAL_CONSTANT",
  88. "INT_CONSTANT",
  89. "TYPE_VOID",
  90. "TYPE_BOOL",
  91. "TYPE_BVEC2",
  92. "TYPE_BVEC3",
  93. "TYPE_BVEC4",
  94. "TYPE_INT",
  95. "TYPE_IVEC2",
  96. "TYPE_IVEC3",
  97. "TYPE_IVEC4",
  98. "TYPE_UINT",
  99. "TYPE_UVEC2",
  100. "TYPE_UVEC3",
  101. "TYPE_UVEC4",
  102. "TYPE_FLOAT",
  103. "TYPE_VEC2",
  104. "TYPE_VEC3",
  105. "TYPE_VEC4",
  106. "TYPE_MAT2",
  107. "TYPE_MAT3",
  108. "TYPE_MAT4",
  109. "TYPE_SAMPLER2D",
  110. "TYPE_ISAMPLER2D",
  111. "TYPE_USAMPLER2D",
  112. "TYPE_SAMPLERCUBE",
  113. "PRECISION_LOW",
  114. "PRECISION_MID",
  115. "PRECISION_HIGH",
  116. "OP_EQUAL",
  117. "OP_NOT_EQUAL",
  118. "OP_LESS",
  119. "OP_LESS_EQUAL",
  120. "OP_GREATER",
  121. "OP_GREATER_EQUAL",
  122. "OP_AND",
  123. "OP_OR",
  124. "OP_NOT",
  125. "OP_ADD",
  126. "OP_SUB",
  127. "OP_MUL",
  128. "OP_DIV",
  129. "OP_MOD",
  130. "OP_SHIFT_LEFT",
  131. "OP_SHIFT_RIGHT",
  132. "OP_ASSIGN",
  133. "OP_ASSIGN_ADD",
  134. "OP_ASSIGN_SUB",
  135. "OP_ASSIGN_MUL",
  136. "OP_ASSIGN_DIV",
  137. "OP_ASSIGN_MOD",
  138. "OP_ASSIGN_SHIFT_LEFT",
  139. "OP_ASSIGN_SHIFT_RIGHT",
  140. "OP_ASSIGN_BIT_AND",
  141. "OP_ASSIGN_BIT_OR",
  142. "OP_ASSIGN_BIT_XOR",
  143. "OP_BIT_AND",
  144. "OP_BIT_OR",
  145. "OP_BIT_XOR",
  146. "OP_BIT_INVERT",
  147. "OP_INCREMENT",
  148. "OP_DECREMENT",
  149. "CF_IF",
  150. "CF_ELSE",
  151. "CF_FOR",
  152. "CF_WHILE",
  153. "CF_DO",
  154. "CF_SWITCH",
  155. "CF_CASE",
  156. "CF_BREAK",
  157. "CF_CONTINUE",
  158. "CF_RETURN",
  159. "CF_DISCARD",
  160. "BRACKET_OPEN",
  161. "BRACKET_CLOSE",
  162. "CURLY_BRACKET_OPEN",
  163. "CURLY_BRACKET_CLOSE",
  164. "PARENTHESIS_OPEN",
  165. "PARENTHESIS_CLOSE",
  166. "QUESTION",
  167. "COMMA",
  168. "COLON",
  169. "SEMICOLON",
  170. "PERIOD",
  171. "UNIFORM",
  172. "VARYING",
  173. "IN",
  174. "OUT",
  175. "INOUT",
  176. "RENDER_MODE",
  177. "HINT_WHITE_TEXTURE",
  178. "HINT_BLACK_TEXTURE",
  179. "HINT_NORMAL_TEXTURE",
  180. "HINT_ANISO_TEXTURE",
  181. "HINT_ALBEDO_TEXTURE",
  182. "HINT_BLACK_ALBEDO_TEXTURE",
  183. "HINT_COLOR",
  184. "HINT_RANGE",
  185. "SHADER_TYPE",
  186. "CURSOR",
  187. "ERROR",
  188. "EOF",
  189. };
  190. String ShaderLanguage::get_token_text(Token p_token) {
  191. String name = token_names[p_token.type];
  192. if (p_token.type == TK_INT_CONSTANT || p_token.type == TK_REAL_CONSTANT) {
  193. name += "(" + rtos(p_token.constant) + ")";
  194. } else if (p_token.type == TK_IDENTIFIER) {
  195. name += "(" + String(p_token.text) + ")";
  196. } else if (p_token.type == TK_ERROR) {
  197. name += "(" + String(p_token.text) + ")";
  198. }
  199. return name;
  200. }
  201. ShaderLanguage::Token ShaderLanguage::_make_token(TokenType p_type, const StringName &p_text) {
  202. Token tk;
  203. tk.type = p_type;
  204. tk.text = p_text;
  205. tk.line = tk_line;
  206. if (tk.type == TK_ERROR) {
  207. _set_error(p_text);
  208. }
  209. return tk;
  210. }
  211. const ShaderLanguage::KeyWord ShaderLanguage::keyword_list[] = {
  212. { TK_TRUE, "true" },
  213. { TK_FALSE, "false" },
  214. { TK_TYPE_VOID, "void" },
  215. { TK_TYPE_BOOL, "bool" },
  216. { TK_TYPE_BVEC2, "bvec2" },
  217. { TK_TYPE_BVEC3, "bvec3" },
  218. { TK_TYPE_BVEC4, "bvec4" },
  219. { TK_TYPE_INT, "int" },
  220. { TK_TYPE_IVEC2, "ivec2" },
  221. { TK_TYPE_IVEC3, "ivec3" },
  222. { TK_TYPE_IVEC4, "ivec4" },
  223. { TK_TYPE_UINT, "uint" },
  224. { TK_TYPE_UVEC2, "uvec2" },
  225. { TK_TYPE_UVEC3, "uvec3" },
  226. { TK_TYPE_UVEC4, "uvec4" },
  227. { TK_TYPE_FLOAT, "float" },
  228. { TK_TYPE_VEC2, "vec2" },
  229. { TK_TYPE_VEC3, "vec3" },
  230. { TK_TYPE_VEC4, "vec4" },
  231. { TK_TYPE_MAT2, "mat2" },
  232. { TK_TYPE_MAT3, "mat3" },
  233. { TK_TYPE_MAT4, "mat4" },
  234. { TK_TYPE_SAMPLER2D, "sampler2D" },
  235. { TK_TYPE_ISAMPLER2D, "isampler2D" },
  236. { TK_TYPE_USAMPLER2D, "usampler2D" },
  237. { TK_TYPE_SAMPLERCUBE, "samplerCube" },
  238. { TK_PRECISION_LOW, "lowp" },
  239. { TK_PRECISION_MID, "mediump" },
  240. { TK_PRECISION_HIGH, "highp" },
  241. { TK_CF_IF, "if" },
  242. { TK_CF_ELSE, "else" },
  243. { TK_CF_FOR, "for" },
  244. { TK_CF_WHILE, "while" },
  245. { TK_CF_DO, "do" },
  246. { TK_CF_SWITCH, "switch" },
  247. { TK_CF_CASE, "case" },
  248. { TK_CF_BREAK, "break" },
  249. { TK_CF_CONTINUE, "continue" },
  250. { TK_CF_RETURN, "return" },
  251. { TK_UNIFORM, "uniform" },
  252. { TK_VARYING, "varying" },
  253. { TK_ARG_IN, "in" },
  254. { TK_ARG_OUT, "out" },
  255. { TK_ARG_INOUT, "inout" },
  256. { TK_RENDER_MODE, "render_mode" },
  257. { TK_HINT_WHITE_TEXTURE, "hint_white" },
  258. { TK_HINT_BLACK_TEXTURE, "hint_black" },
  259. { TK_HINT_NORMAL_TEXTURE, "hint_normal" },
  260. { TK_HINT_ANISO_TEXTURE, "hint_aniso" },
  261. { TK_HINT_ALBEDO_TEXTURE, "hint_albedo" },
  262. { TK_HINT_BLACK_ALBEDO_TEXTURE, "hint_black_albedo" },
  263. { TK_HINT_COLOR, "hint_color" },
  264. { TK_HINT_RANGE, "hint_range" },
  265. { TK_SHADER_TYPE, "shader_type" },
  266. { TK_ERROR, NULL }
  267. };
  268. ShaderLanguage::Token ShaderLanguage::_get_token() {
  269. #define GETCHAR(m_idx) (((char_idx + m_idx) < code.length()) ? code[char_idx + m_idx] : CharType(0))
  270. while (true) {
  271. char_idx++;
  272. switch (GETCHAR(-1)) {
  273. case 0:
  274. return _make_token(TK_EOF);
  275. case 0xFFFF:
  276. return _make_token(TK_CURSOR); //for completion
  277. case '\t':
  278. case '\r':
  279. case ' ':
  280. continue;
  281. case '\n':
  282. tk_line++;
  283. continue;
  284. case '/': {
  285. switch (GETCHAR(0)) {
  286. case '*': { // block comment
  287. char_idx++;
  288. while (true) {
  289. if (GETCHAR(0) == 0) {
  290. return _make_token(TK_EOF);
  291. }
  292. if (GETCHAR(0) == '*' && GETCHAR(1) == '/') {
  293. char_idx += 2;
  294. break;
  295. } else if (GETCHAR(0) == '\n') {
  296. tk_line++;
  297. }
  298. char_idx++;
  299. }
  300. } break;
  301. case '/': { // line comment skip
  302. while (true) {
  303. if (GETCHAR(0) == '\n') {
  304. char_idx++;
  305. break;
  306. }
  307. if (GETCHAR(0) == 0) {
  308. return _make_token(TK_EOF);
  309. }
  310. char_idx++;
  311. }
  312. } break;
  313. case '=': { // diveq
  314. char_idx++;
  315. return _make_token(TK_OP_ASSIGN_DIV);
  316. } break;
  317. default:
  318. return _make_token(TK_OP_DIV);
  319. }
  320. continue; //a comment, continue to next token
  321. } break;
  322. case '=': {
  323. if (GETCHAR(0) == '=') {
  324. char_idx++;
  325. return _make_token(TK_OP_EQUAL);
  326. }
  327. return _make_token(TK_OP_ASSIGN);
  328. } break;
  329. case '<': {
  330. if (GETCHAR(0) == '=') {
  331. char_idx++;
  332. return _make_token(TK_OP_LESS_EQUAL);
  333. } else if (GETCHAR(0) == '<') {
  334. char_idx++;
  335. if (GETCHAR(0) == '=') {
  336. char_idx++;
  337. return _make_token(TK_OP_ASSIGN_SHIFT_LEFT);
  338. }
  339. return _make_token(TK_OP_SHIFT_LEFT);
  340. }
  341. return _make_token(TK_OP_LESS);
  342. } break;
  343. case '>': {
  344. if (GETCHAR(0) == '=') {
  345. char_idx++;
  346. return _make_token(TK_OP_GREATER_EQUAL);
  347. } else if (GETCHAR(0) == '>') {
  348. char_idx++;
  349. if (GETCHAR(0) == '=') {
  350. char_idx++;
  351. return _make_token(TK_OP_ASSIGN_SHIFT_RIGHT);
  352. }
  353. return _make_token(TK_OP_SHIFT_RIGHT);
  354. }
  355. return _make_token(TK_OP_GREATER);
  356. } break;
  357. case '!': {
  358. if (GETCHAR(0) == '=') {
  359. char_idx++;
  360. return _make_token(TK_OP_NOT_EQUAL);
  361. }
  362. return _make_token(TK_OP_NOT);
  363. } break;
  364. //case '"' //string - no strings in shader
  365. //case '\'' //string - no strings in shader
  366. case '{':
  367. return _make_token(TK_CURLY_BRACKET_OPEN);
  368. case '}':
  369. return _make_token(TK_CURLY_BRACKET_CLOSE);
  370. case '[':
  371. return _make_token(TK_BRACKET_OPEN);
  372. case ']':
  373. return _make_token(TK_BRACKET_CLOSE);
  374. case '(':
  375. return _make_token(TK_PARENTHESIS_OPEN);
  376. case ')':
  377. return _make_token(TK_PARENTHESIS_CLOSE);
  378. case ',':
  379. return _make_token(TK_COMMA);
  380. case ';':
  381. return _make_token(TK_SEMICOLON);
  382. case '?':
  383. return _make_token(TK_QUESTION);
  384. case ':':
  385. return _make_token(TK_COLON);
  386. case '^':
  387. return _make_token(TK_OP_BIT_XOR);
  388. case '~':
  389. return _make_token(TK_OP_BIT_INVERT);
  390. case '&': {
  391. if (GETCHAR(0) == '=') {
  392. char_idx++;
  393. return _make_token(TK_OP_ASSIGN_BIT_AND);
  394. } else if (GETCHAR(0) == '&') {
  395. char_idx++;
  396. return _make_token(TK_OP_AND);
  397. }
  398. return _make_token(TK_OP_BIT_AND);
  399. } break;
  400. case '|': {
  401. if (GETCHAR(0) == '=') {
  402. char_idx++;
  403. return _make_token(TK_OP_ASSIGN_BIT_OR);
  404. } else if (GETCHAR(0) == '|') {
  405. char_idx++;
  406. return _make_token(TK_OP_OR);
  407. }
  408. return _make_token(TK_OP_BIT_OR);
  409. } break;
  410. case '*': {
  411. if (GETCHAR(0) == '=') {
  412. char_idx++;
  413. return _make_token(TK_OP_ASSIGN_MUL);
  414. }
  415. return _make_token(TK_OP_MUL);
  416. } break;
  417. case '+': {
  418. if (GETCHAR(0) == '=') {
  419. char_idx++;
  420. return _make_token(TK_OP_ASSIGN_ADD);
  421. } else if (GETCHAR(0) == '+') {
  422. char_idx++;
  423. return _make_token(TK_OP_INCREMENT);
  424. }
  425. return _make_token(TK_OP_ADD);
  426. } break;
  427. case '-': {
  428. if (GETCHAR(0) == '=') {
  429. char_idx++;
  430. return _make_token(TK_OP_ASSIGN_SUB);
  431. } else if (GETCHAR(0) == '-') {
  432. char_idx++;
  433. return _make_token(TK_OP_DECREMENT);
  434. }
  435. return _make_token(TK_OP_SUB);
  436. } break;
  437. case '%': {
  438. if (GETCHAR(0) == '=') {
  439. char_idx++;
  440. return _make_token(TK_OP_ASSIGN_MOD);
  441. }
  442. return _make_token(TK_OP_MOD);
  443. } break;
  444. default: {
  445. char_idx--; //go back one, since we have no idea what this is
  446. if (_is_number(GETCHAR(0)) || (GETCHAR(0) == '.' && _is_number(GETCHAR(1)))) {
  447. // parse number
  448. bool period_found = false;
  449. bool exponent_found = false;
  450. bool hexa_found = false;
  451. bool sign_found = false;
  452. bool minus_exponent_found = false;
  453. String str;
  454. int i = 0;
  455. while (true) {
  456. if (GETCHAR(i) == '.') {
  457. if (period_found || exponent_found)
  458. return _make_token(TK_ERROR, "Invalid numeric constant");
  459. period_found = true;
  460. } else if (GETCHAR(i) == 'x') {
  461. if (hexa_found || str.length() != 1 || str[0] != '0')
  462. return _make_token(TK_ERROR, "Invalid numeric constant");
  463. hexa_found = true;
  464. } else if (GETCHAR(i) == 'e') {
  465. if (hexa_found || exponent_found)
  466. return _make_token(TK_ERROR, "Invalid numeric constant");
  467. exponent_found = true;
  468. } else if (_is_number(GETCHAR(i))) {
  469. //all ok
  470. } else if (hexa_found && _is_hex(GETCHAR(i))) {
  471. } else if ((GETCHAR(i) == '-' || GETCHAR(i) == '+') && exponent_found) {
  472. if (sign_found)
  473. return _make_token(TK_ERROR, "Invalid numeric constant");
  474. sign_found = true;
  475. if (GETCHAR(i) == '-')
  476. minus_exponent_found = true;
  477. } else
  478. break;
  479. str += CharType(GETCHAR(i));
  480. i++;
  481. }
  482. if (!_is_number(str[str.length() - 1]))
  483. return _make_token(TK_ERROR, "Invalid numeric constant");
  484. char_idx += str.length();
  485. Token tk;
  486. if (period_found || minus_exponent_found)
  487. tk.type = TK_REAL_CONSTANT;
  488. else
  489. tk.type = TK_INT_CONSTANT;
  490. if (!str.is_valid_float()) {
  491. return _make_token(TK_ERROR, "Invalid numeric constant");
  492. }
  493. tk.constant = str.to_double();
  494. tk.line = tk_line;
  495. return tk;
  496. }
  497. if (GETCHAR(0) == '.') {
  498. //parse period
  499. char_idx++;
  500. return _make_token(TK_PERIOD);
  501. }
  502. if (_is_text_char(GETCHAR(0))) {
  503. // parse identifier
  504. String str;
  505. while (_is_text_char(GETCHAR(0))) {
  506. str += CharType(GETCHAR(0));
  507. char_idx++;
  508. }
  509. //see if keyword
  510. //should be converted to a static map
  511. int idx = 0;
  512. while (keyword_list[idx].text) {
  513. if (str == keyword_list[idx].text) {
  514. return _make_token(keyword_list[idx].token);
  515. }
  516. idx++;
  517. }
  518. return _make_token(TK_IDENTIFIER, str);
  519. }
  520. if (GETCHAR(0) > 32)
  521. return _make_token(TK_ERROR, "Tokenizer: Unknown character #" + itos(GETCHAR(0)) + ": '" + String::chr(GETCHAR(0)) + "'");
  522. else
  523. return _make_token(TK_ERROR, "Tokenizer: Unknown character #" + itos(GETCHAR(0)));
  524. } break;
  525. }
  526. }
  527. ERR_PRINT("BUG");
  528. return Token();
  529. }
  530. String ShaderLanguage::token_debug(const String &p_code) {
  531. clear();
  532. code = p_code;
  533. String output;
  534. Token tk = _get_token();
  535. while (tk.type != TK_EOF && tk.type != TK_ERROR) {
  536. output += itos(tk_line) + ": " + get_token_text(tk) + "\n";
  537. tk = _get_token();
  538. }
  539. return output;
  540. }
  541. bool ShaderLanguage::is_token_datatype(TokenType p_type) {
  542. return (
  543. p_type == TK_TYPE_VOID ||
  544. p_type == TK_TYPE_BOOL ||
  545. p_type == TK_TYPE_BVEC2 ||
  546. p_type == TK_TYPE_BVEC3 ||
  547. p_type == TK_TYPE_BVEC4 ||
  548. p_type == TK_TYPE_INT ||
  549. p_type == TK_TYPE_IVEC2 ||
  550. p_type == TK_TYPE_IVEC3 ||
  551. p_type == TK_TYPE_IVEC4 ||
  552. p_type == TK_TYPE_UINT ||
  553. p_type == TK_TYPE_UVEC2 ||
  554. p_type == TK_TYPE_UVEC3 ||
  555. p_type == TK_TYPE_UVEC4 ||
  556. p_type == TK_TYPE_FLOAT ||
  557. p_type == TK_TYPE_VEC2 ||
  558. p_type == TK_TYPE_VEC3 ||
  559. p_type == TK_TYPE_VEC4 ||
  560. p_type == TK_TYPE_MAT2 ||
  561. p_type == TK_TYPE_MAT3 ||
  562. p_type == TK_TYPE_MAT4 ||
  563. p_type == TK_TYPE_SAMPLER2D ||
  564. p_type == TK_TYPE_ISAMPLER2D ||
  565. p_type == TK_TYPE_USAMPLER2D ||
  566. p_type == TK_TYPE_SAMPLERCUBE);
  567. }
  568. ShaderLanguage::DataType ShaderLanguage::get_token_datatype(TokenType p_type) {
  569. return DataType(p_type - TK_TYPE_VOID);
  570. }
  571. bool ShaderLanguage::is_token_precision(TokenType p_type) {
  572. return (
  573. p_type == TK_PRECISION_LOW ||
  574. p_type == TK_PRECISION_MID ||
  575. p_type == TK_PRECISION_HIGH);
  576. }
  577. ShaderLanguage::DataPrecision ShaderLanguage::get_token_precision(TokenType p_type) {
  578. if (p_type == TK_PRECISION_LOW)
  579. return PRECISION_LOWP;
  580. else if (p_type == TK_PRECISION_HIGH)
  581. return PRECISION_HIGHP;
  582. else
  583. return PRECISION_MEDIUMP;
  584. }
  585. String ShaderLanguage::get_datatype_name(DataType p_type) {
  586. switch (p_type) {
  587. case TYPE_VOID: return "void";
  588. case TYPE_BOOL: return "bool";
  589. case TYPE_BVEC2: return "bvec2";
  590. case TYPE_BVEC3: return "bvec3";
  591. case TYPE_BVEC4: return "bvec4";
  592. case TYPE_INT: return "int";
  593. case TYPE_IVEC2: return "ivec2";
  594. case TYPE_IVEC3: return "ivec3";
  595. case TYPE_IVEC4: return "ivec4";
  596. case TYPE_UINT: return "uint";
  597. case TYPE_UVEC2: return "uvec2";
  598. case TYPE_UVEC3: return "uvec3";
  599. case TYPE_UVEC4: return "uvec4";
  600. case TYPE_FLOAT: return "float";
  601. case TYPE_VEC2: return "vec2";
  602. case TYPE_VEC3: return "vec3";
  603. case TYPE_VEC4: return "vec4";
  604. case TYPE_MAT2: return "mat2";
  605. case TYPE_MAT3: return "mat3";
  606. case TYPE_MAT4: return "mat4";
  607. case TYPE_SAMPLER2D: return "sampler2D";
  608. case TYPE_ISAMPLER2D: return "isampler2D";
  609. case TYPE_USAMPLER2D: return "usampler2D";
  610. case TYPE_SAMPLERCUBE: return "samplerCube";
  611. }
  612. return "";
  613. }
  614. bool ShaderLanguage::is_token_nonvoid_datatype(TokenType p_type) {
  615. return is_token_datatype(p_type) && p_type != TK_TYPE_VOID;
  616. }
  617. void ShaderLanguage::clear() {
  618. current_function = StringName();
  619. completion_type = COMPLETION_NONE;
  620. completion_block = NULL;
  621. completion_function = StringName();
  622. error_line = 0;
  623. tk_line = 1;
  624. char_idx = 0;
  625. error_set = false;
  626. error_str = "";
  627. while (nodes) {
  628. Node *n = nodes;
  629. nodes = nodes->next;
  630. memdelete(n);
  631. }
  632. }
  633. bool ShaderLanguage::_find_identifier(const BlockNode *p_block, const Map<StringName, DataType> &p_builtin_types, const StringName &p_identifier, DataType *r_data_type, IdentifierType *r_type) {
  634. if (p_builtin_types.has(p_identifier)) {
  635. if (r_data_type) {
  636. *r_data_type = p_builtin_types[p_identifier];
  637. }
  638. if (r_type) {
  639. *r_type = IDENTIFIER_BUILTIN_VAR;
  640. }
  641. return true;
  642. }
  643. FunctionNode *function = NULL;
  644. while (p_block) {
  645. if (p_block->variables.has(p_identifier)) {
  646. if (r_data_type) {
  647. *r_data_type = p_block->variables[p_identifier].type;
  648. }
  649. if (r_type) {
  650. *r_type = IDENTIFIER_LOCAL_VAR;
  651. }
  652. return true;
  653. }
  654. if (p_block->parent_function) {
  655. function = p_block->parent_function;
  656. break;
  657. } else {
  658. ERR_FAIL_COND_V(!p_block->parent_block, false);
  659. p_block = p_block->parent_block;
  660. }
  661. }
  662. if (function) {
  663. for (int i = 0; i < function->arguments.size(); i++) {
  664. if (function->arguments[i].name == p_identifier) {
  665. if (r_data_type) {
  666. *r_data_type = function->arguments[i].type;
  667. }
  668. if (r_type) {
  669. *r_type = IDENTIFIER_FUNCTION_ARGUMENT;
  670. }
  671. return true;
  672. }
  673. }
  674. }
  675. if (shader->varyings.has(p_identifier)) {
  676. if (r_data_type) {
  677. *r_data_type = shader->varyings[p_identifier].type;
  678. }
  679. if (r_type) {
  680. *r_type = IDENTIFIER_VARYING;
  681. }
  682. return true;
  683. }
  684. if (shader->uniforms.has(p_identifier)) {
  685. if (r_data_type) {
  686. *r_data_type = shader->uniforms[p_identifier].type;
  687. }
  688. if (r_type) {
  689. *r_type = IDENTIFIER_UNIFORM;
  690. }
  691. return true;
  692. }
  693. for (int i = 0; i < shader->functions.size(); i++) {
  694. if (!shader->functions[i].callable)
  695. continue;
  696. if (shader->functions[i].name == p_identifier) {
  697. if (r_data_type) {
  698. *r_data_type = shader->functions[i].function->return_type;
  699. }
  700. if (r_type) {
  701. *r_type = IDENTIFIER_FUNCTION;
  702. }
  703. }
  704. }
  705. return false;
  706. }
  707. bool ShaderLanguage::_validate_operator(OperatorNode *p_op, DataType *r_ret_type) {
  708. bool valid = false;
  709. DataType ret_type;
  710. switch (p_op->op) {
  711. case OP_EQUAL:
  712. case OP_NOT_EQUAL: {
  713. DataType na = p_op->arguments[0]->get_datatype();
  714. DataType nb = p_op->arguments[1]->get_datatype();
  715. valid = na == nb;
  716. ret_type = TYPE_BOOL;
  717. } break;
  718. case OP_LESS:
  719. case OP_LESS_EQUAL:
  720. case OP_GREATER:
  721. case OP_GREATER_EQUAL: {
  722. DataType na = p_op->arguments[0]->get_datatype();
  723. DataType nb = p_op->arguments[1]->get_datatype();
  724. valid = na == nb && (na == TYPE_UINT || na == TYPE_INT || na == TYPE_FLOAT);
  725. ret_type = TYPE_BOOL;
  726. } break;
  727. case OP_AND:
  728. case OP_OR: {
  729. DataType na = p_op->arguments[0]->get_datatype();
  730. DataType nb = p_op->arguments[1]->get_datatype();
  731. valid = na == nb && na == TYPE_BOOL;
  732. ret_type = TYPE_BOOL;
  733. } break;
  734. case OP_NOT: {
  735. DataType na = p_op->arguments[0]->get_datatype();
  736. valid = na == TYPE_BOOL;
  737. ret_type = TYPE_BOOL;
  738. } break;
  739. case OP_INCREMENT:
  740. case OP_DECREMENT:
  741. case OP_POST_INCREMENT:
  742. case OP_POST_DECREMENT:
  743. case OP_NEGATE: {
  744. DataType na = p_op->arguments[0]->get_datatype();
  745. valid = na > TYPE_BOOL && na < TYPE_MAT2;
  746. ret_type = na;
  747. } break;
  748. case OP_ADD:
  749. case OP_SUB:
  750. case OP_MUL:
  751. case OP_DIV: {
  752. DataType na = p_op->arguments[0]->get_datatype();
  753. DataType nb = p_op->arguments[1]->get_datatype();
  754. if (na > nb) {
  755. //make things easier;
  756. SWAP(na, nb);
  757. }
  758. if (na == nb) {
  759. valid = (na > TYPE_BOOL && na <= TYPE_MAT4);
  760. ret_type = na;
  761. } else if (na == TYPE_INT && nb == TYPE_IVEC2) {
  762. valid = true;
  763. ret_type = TYPE_IVEC2;
  764. } else if (na == TYPE_INT && nb == TYPE_IVEC3) {
  765. valid = true;
  766. ret_type = TYPE_IVEC3;
  767. } else if (na == TYPE_INT && nb == TYPE_IVEC4) {
  768. valid = true;
  769. ret_type = TYPE_IVEC4;
  770. } else if (na == TYPE_UINT && nb == TYPE_UVEC2) {
  771. valid = true;
  772. ret_type = TYPE_UVEC2;
  773. } else if (na == TYPE_UINT && nb == TYPE_UVEC3) {
  774. valid = true;
  775. ret_type = TYPE_UVEC3;
  776. } else if (na == TYPE_UINT && nb == TYPE_UVEC4) {
  777. valid = true;
  778. ret_type = TYPE_UVEC4;
  779. } else if (na == TYPE_FLOAT && nb == TYPE_VEC2) {
  780. valid = true;
  781. ret_type = TYPE_VEC2;
  782. } else if (na == TYPE_FLOAT && nb == TYPE_VEC3) {
  783. valid = true;
  784. ret_type = TYPE_VEC3;
  785. } else if (na == TYPE_FLOAT && nb == TYPE_VEC4) {
  786. valid = true;
  787. ret_type = TYPE_VEC4;
  788. } else if (p_op->op == OP_MUL && na == TYPE_FLOAT && nb == TYPE_MAT2) {
  789. valid = true;
  790. ret_type = TYPE_MAT2;
  791. } else if (p_op->op == OP_MUL && na == TYPE_FLOAT && nb == TYPE_MAT3) {
  792. valid = true;
  793. ret_type = TYPE_MAT3;
  794. } else if (p_op->op == OP_MUL && na == TYPE_FLOAT && nb == TYPE_MAT4) {
  795. valid = true;
  796. ret_type = TYPE_MAT4;
  797. } else if (p_op->op == OP_MUL && na == TYPE_VEC2 && nb == TYPE_MAT2) {
  798. valid = true;
  799. ret_type = TYPE_VEC2;
  800. } else if (p_op->op == OP_MUL && na == TYPE_VEC3 && nb == TYPE_MAT3) {
  801. valid = true;
  802. ret_type = TYPE_VEC3;
  803. } else if (p_op->op == OP_MUL && na == TYPE_VEC4 && nb == TYPE_MAT4) {
  804. valid = true;
  805. ret_type = TYPE_VEC4;
  806. }
  807. } break;
  808. case OP_ASSIGN_MOD:
  809. case OP_MOD: {
  810. /*
  811. * The operator modulus (%) operates on signed or unsigned integers or integer vectors. The operand
  812. * types must both be signed or both be unsigned. The operands cannot be vectors of differing size. If
  813. * one operand is a scalar and the other vector, then the scalar is applied component-wise to the vector,
  814. * resulting in the same type as the vector. If both are vectors of the same size, the result is computed
  815. * component-wise.
  816. */
  817. DataType na = p_op->arguments[0]->get_datatype();
  818. DataType nb = p_op->arguments[1]->get_datatype();
  819. if (na == TYPE_INT && nb == TYPE_INT) {
  820. valid = true;
  821. ret_type = TYPE_INT;
  822. } else if (na == TYPE_IVEC2 && nb == TYPE_INT) {
  823. valid = true;
  824. ret_type = TYPE_IVEC2;
  825. } else if (na == TYPE_IVEC3 && nb == TYPE_INT) {
  826. valid = true;
  827. ret_type = TYPE_IVEC3;
  828. } else if (na == TYPE_IVEC4 && nb == TYPE_INT) {
  829. valid = true;
  830. ret_type = TYPE_IVEC4;
  831. } else if (na == TYPE_IVEC2 && nb == TYPE_IVEC2) {
  832. valid = true;
  833. ret_type = TYPE_IVEC2;
  834. } else if (na == TYPE_IVEC3 && nb == TYPE_IVEC3) {
  835. valid = true;
  836. ret_type = TYPE_IVEC3;
  837. } else if (na == TYPE_IVEC4 && nb == TYPE_IVEC4) {
  838. valid = true;
  839. ret_type = TYPE_IVEC4;
  840. /////
  841. } else if (na == TYPE_UINT && nb == TYPE_UINT) {
  842. valid = true;
  843. ret_type = TYPE_UINT;
  844. } else if (na == TYPE_UVEC2 && nb == TYPE_UINT) {
  845. valid = true;
  846. ret_type = TYPE_UVEC2;
  847. } else if (na == TYPE_UVEC3 && nb == TYPE_UINT) {
  848. valid = true;
  849. ret_type = TYPE_UVEC3;
  850. } else if (na == TYPE_UVEC4 && nb == TYPE_UINT) {
  851. valid = true;
  852. ret_type = TYPE_UVEC4;
  853. } else if (na == TYPE_UVEC2 && nb == TYPE_UVEC2) {
  854. valid = true;
  855. ret_type = TYPE_UVEC2;
  856. } else if (na == TYPE_UVEC3 && nb == TYPE_UVEC3) {
  857. valid = true;
  858. ret_type = TYPE_UVEC3;
  859. } else if (na == TYPE_UVEC4 && nb == TYPE_UVEC4) {
  860. valid = true;
  861. ret_type = TYPE_UVEC4;
  862. }
  863. } break;
  864. case OP_ASSIGN_SHIFT_LEFT:
  865. case OP_ASSIGN_SHIFT_RIGHT:
  866. case OP_SHIFT_LEFT:
  867. case OP_SHIFT_RIGHT: {
  868. DataType na = p_op->arguments[0]->get_datatype();
  869. DataType nb = p_op->arguments[1]->get_datatype();
  870. if (na == TYPE_INT && nb == TYPE_INT) {
  871. valid = true;
  872. ret_type = TYPE_INT;
  873. } else if (na == TYPE_IVEC2 && nb == TYPE_INT) {
  874. valid = true;
  875. ret_type = TYPE_IVEC2;
  876. } else if (na == TYPE_IVEC3 && nb == TYPE_INT) {
  877. valid = true;
  878. ret_type = TYPE_IVEC3;
  879. } else if (na == TYPE_IVEC4 && nb == TYPE_INT) {
  880. valid = true;
  881. ret_type = TYPE_IVEC4;
  882. } else if (na == TYPE_IVEC2 && nb == TYPE_IVEC2) {
  883. valid = true;
  884. ret_type = TYPE_IVEC2;
  885. } else if (na == TYPE_IVEC3 && nb == TYPE_IVEC3) {
  886. valid = true;
  887. ret_type = TYPE_IVEC3;
  888. } else if (na == TYPE_IVEC4 && nb == TYPE_IVEC4) {
  889. valid = true;
  890. ret_type = TYPE_IVEC4;
  891. } else if (na == TYPE_UINT && nb == TYPE_UINT) {
  892. valid = true;
  893. ret_type = TYPE_UINT;
  894. } else if (na == TYPE_UVEC2 && nb == TYPE_UINT) {
  895. valid = true;
  896. ret_type = TYPE_UVEC2;
  897. } else if (na == TYPE_UVEC3 && nb == TYPE_UINT) {
  898. valid = true;
  899. ret_type = TYPE_UVEC3;
  900. } else if (na == TYPE_UVEC4 && nb == TYPE_UINT) {
  901. valid = true;
  902. ret_type = TYPE_UVEC4;
  903. } else if (na == TYPE_UVEC2 && nb == TYPE_UVEC2) {
  904. valid = true;
  905. ret_type = TYPE_UVEC2;
  906. } else if (na == TYPE_UVEC3 && nb == TYPE_UVEC3) {
  907. valid = true;
  908. ret_type = TYPE_UVEC3;
  909. } else if (na == TYPE_UVEC4 && nb == TYPE_UVEC4) {
  910. valid = true;
  911. ret_type = TYPE_UVEC4;
  912. }
  913. } break;
  914. case OP_ASSIGN: {
  915. DataType na = p_op->arguments[0]->get_datatype();
  916. DataType nb = p_op->arguments[1]->get_datatype();
  917. valid = na == nb;
  918. ret_type = na;
  919. } break;
  920. case OP_ASSIGN_ADD:
  921. case OP_ASSIGN_SUB:
  922. case OP_ASSIGN_MUL:
  923. case OP_ASSIGN_DIV: {
  924. DataType na = p_op->arguments[0]->get_datatype();
  925. DataType nb = p_op->arguments[1]->get_datatype();
  926. if (na == nb) {
  927. valid = (na > TYPE_BOOL && na < TYPE_MAT2) || (p_op->op == OP_ASSIGN_MUL && na >= TYPE_MAT2 && na <= TYPE_MAT4);
  928. ret_type = na;
  929. } else if (na == TYPE_IVEC2 && nb == TYPE_INT) {
  930. valid = true;
  931. ret_type = TYPE_IVEC2;
  932. } else if (na == TYPE_IVEC3 && nb == TYPE_INT) {
  933. valid = true;
  934. ret_type = TYPE_IVEC3;
  935. } else if (na == TYPE_IVEC4 && nb == TYPE_INT) {
  936. valid = true;
  937. ret_type = TYPE_IVEC4;
  938. } else if (na == TYPE_UVEC2 && nb == TYPE_UINT) {
  939. valid = true;
  940. ret_type = TYPE_UVEC2;
  941. } else if (na == TYPE_UVEC3 && nb == TYPE_UINT) {
  942. valid = true;
  943. ret_type = TYPE_UVEC3;
  944. } else if (na == TYPE_UVEC4 && nb == TYPE_UINT) {
  945. valid = true;
  946. ret_type = TYPE_UVEC4;
  947. } else if (na == TYPE_VEC2 && nb == TYPE_FLOAT) {
  948. valid = true;
  949. ret_type = TYPE_VEC2;
  950. } else if (na == TYPE_VEC3 && nb == TYPE_FLOAT) {
  951. valid = true;
  952. ret_type = TYPE_VEC3;
  953. } else if (na == TYPE_VEC4 && nb == TYPE_FLOAT) {
  954. valid = true;
  955. ret_type = TYPE_VEC4;
  956. } else if (p_op->op == OP_ASSIGN_MUL && na == TYPE_MAT2 && nb == TYPE_VEC2) {
  957. valid = true;
  958. ret_type = TYPE_MAT2;
  959. } else if (p_op->op == OP_ASSIGN_MUL && na == TYPE_MAT3 && nb == TYPE_VEC3) {
  960. valid = true;
  961. ret_type = TYPE_MAT3;
  962. } else if (p_op->op == OP_ASSIGN_MUL && na == TYPE_MAT4 && nb == TYPE_VEC4) {
  963. valid = true;
  964. ret_type = TYPE_MAT4;
  965. }
  966. } break;
  967. case OP_ASSIGN_BIT_AND:
  968. case OP_ASSIGN_BIT_OR:
  969. case OP_ASSIGN_BIT_XOR:
  970. case OP_BIT_AND:
  971. case OP_BIT_OR:
  972. case OP_BIT_XOR: {
  973. /*
  974. * The bitwise operators and (&), exclusive-or (^), and inclusive-or (|). The operands must be of type
  975. * signed or unsigned integers or integer vectors. The operands cannot be vectors of differing size. If
  976. * one operand is a scalar and the other a vector, the scalar is applied component-wise to the vector,
  977. * resulting in the same type as the vector. The fundamental types of the operands (signed or unsigned)
  978. * must match.
  979. */
  980. DataType na = p_op->arguments[0]->get_datatype();
  981. DataType nb = p_op->arguments[1]->get_datatype();
  982. if (na > nb && p_op->op >= OP_BIT_AND) {
  983. //can swap for non assign
  984. SWAP(na, nb);
  985. }
  986. if (na == TYPE_INT && nb == TYPE_INT) {
  987. valid = true;
  988. ret_type = TYPE_INT;
  989. } else if (na == TYPE_IVEC2 && nb == TYPE_INT) {
  990. valid = true;
  991. ret_type = TYPE_IVEC2;
  992. } else if (na == TYPE_IVEC3 && nb == TYPE_INT) {
  993. valid = true;
  994. ret_type = TYPE_IVEC3;
  995. } else if (na == TYPE_IVEC4 && nb == TYPE_INT) {
  996. valid = true;
  997. ret_type = TYPE_IVEC4;
  998. } else if (na == TYPE_IVEC2 && nb == TYPE_IVEC2) {
  999. valid = true;
  1000. ret_type = TYPE_IVEC2;
  1001. } else if (na == TYPE_IVEC3 && nb == TYPE_IVEC3) {
  1002. valid = true;
  1003. ret_type = TYPE_IVEC3;
  1004. } else if (na == TYPE_IVEC4 && nb == TYPE_IVEC4) {
  1005. valid = true;
  1006. ret_type = TYPE_IVEC4;
  1007. /////
  1008. } else if (na == TYPE_UINT && nb == TYPE_UINT) {
  1009. valid = true;
  1010. ret_type = TYPE_UINT;
  1011. } else if (na == TYPE_UVEC2 && nb == TYPE_UINT) {
  1012. valid = true;
  1013. ret_type = TYPE_UVEC2;
  1014. } else if (na == TYPE_UVEC3 && nb == TYPE_UINT) {
  1015. valid = true;
  1016. ret_type = TYPE_UVEC3;
  1017. } else if (na == TYPE_UVEC4 && nb == TYPE_UINT) {
  1018. valid = true;
  1019. ret_type = TYPE_UVEC4;
  1020. } else if (na == TYPE_UVEC2 && nb == TYPE_UVEC2) {
  1021. valid = true;
  1022. ret_type = TYPE_UVEC2;
  1023. } else if (na == TYPE_UVEC3 && nb == TYPE_UVEC3) {
  1024. valid = true;
  1025. ret_type = TYPE_UVEC3;
  1026. } else if (na == TYPE_UVEC4 && nb == TYPE_UVEC4) {
  1027. valid = true;
  1028. ret_type = TYPE_UVEC4;
  1029. }
  1030. } break;
  1031. case OP_BIT_INVERT: { //unaries
  1032. DataType na = p_op->arguments[0]->get_datatype();
  1033. valid = na >= TYPE_INT && na < TYPE_FLOAT;
  1034. ret_type = na;
  1035. } break;
  1036. case OP_SELECT_IF: {
  1037. DataType na = p_op->arguments[0]->get_datatype();
  1038. DataType nb = p_op->arguments[1]->get_datatype();
  1039. DataType nc = p_op->arguments[2]->get_datatype();
  1040. valid = na == TYPE_BOOL && (nb == nc);
  1041. ret_type = nb;
  1042. } break;
  1043. default: {
  1044. ERR_FAIL_V(false);
  1045. }
  1046. }
  1047. if (r_ret_type)
  1048. *r_ret_type = ret_type;
  1049. return valid;
  1050. }
  1051. const ShaderLanguage::BuiltinFuncDef ShaderLanguage::builtin_func_defs[] = {
  1052. //constructors
  1053. { "bool", TYPE_BOOL, { TYPE_BOOL, TYPE_VOID } },
  1054. { "bvec2", TYPE_BVEC2, { TYPE_BOOL, TYPE_VOID } },
  1055. { "bvec2", TYPE_BVEC2, { TYPE_BOOL, TYPE_BOOL, TYPE_VOID } },
  1056. { "bvec3", TYPE_BVEC3, { TYPE_BOOL, TYPE_VOID } },
  1057. { "bvec3", TYPE_BVEC3, { TYPE_BOOL, TYPE_BOOL, TYPE_BOOL, TYPE_VOID } },
  1058. { "bvec3", TYPE_BVEC3, { TYPE_BVEC2, TYPE_BOOL, TYPE_VOID } },
  1059. { "bvec3", TYPE_BVEC3, { TYPE_BOOL, TYPE_BVEC2, TYPE_VOID } },
  1060. { "bvec4", TYPE_BVEC4, { TYPE_BOOL, TYPE_VOID } },
  1061. { "bvec4", TYPE_BVEC4, { TYPE_BOOL, TYPE_BOOL, TYPE_BOOL, TYPE_BOOL, TYPE_VOID } },
  1062. { "bvec4", TYPE_BVEC4, { TYPE_BOOL, TYPE_BVEC2, TYPE_BOOL, TYPE_VOID } },
  1063. { "bvec4", TYPE_BVEC4, { TYPE_BVEC2, TYPE_BOOL, TYPE_BOOL, TYPE_VOID } },
  1064. { "bvec4", TYPE_BVEC4, { TYPE_BOOL, TYPE_BOOL, TYPE_BVEC2, TYPE_VOID } },
  1065. { "bvec4", TYPE_BVEC4, { TYPE_BOOL, TYPE_BVEC3, TYPE_VOID } },
  1066. { "bvec4", TYPE_BVEC4, { TYPE_BVEC3, TYPE_BOOL, TYPE_VOID } },
  1067. { "bvec4", TYPE_BVEC4, { TYPE_BVEC2, TYPE_BVEC2, TYPE_VOID } },
  1068. { "float", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1069. { "vec2", TYPE_VEC2, { TYPE_FLOAT, TYPE_VOID } },
  1070. { "vec2", TYPE_VEC2, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1071. { "vec3", TYPE_VEC3, { TYPE_FLOAT, TYPE_VOID } },
  1072. { "vec3", TYPE_VEC3, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1073. { "vec3", TYPE_VEC3, { TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1074. { "vec3", TYPE_VEC3, { TYPE_FLOAT, TYPE_VEC2, TYPE_VOID } },
  1075. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_VOID } },
  1076. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1077. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1078. { "vec4", TYPE_VEC4, { TYPE_VEC2, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1079. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VEC2, TYPE_VOID } },
  1080. { "vec4", TYPE_VEC4, { TYPE_FLOAT, TYPE_VEC3, TYPE_VOID } },
  1081. { "vec4", TYPE_VEC4, { TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1082. { "vec4", TYPE_VEC4, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1083. { "int", TYPE_INT, { TYPE_INT, TYPE_VOID } },
  1084. { "ivec2", TYPE_IVEC2, { TYPE_INT, TYPE_VOID } },
  1085. { "ivec2", TYPE_IVEC2, { TYPE_INT, TYPE_INT, TYPE_VOID } },
  1086. { "ivec3", TYPE_IVEC3, { TYPE_INT, TYPE_VOID } },
  1087. { "ivec3", TYPE_IVEC3, { TYPE_INT, TYPE_INT, TYPE_INT, TYPE_VOID } },
  1088. { "ivec3", TYPE_IVEC3, { TYPE_IVEC2, TYPE_INT, TYPE_VOID } },
  1089. { "ivec3", TYPE_IVEC3, { TYPE_INT, TYPE_IVEC2, TYPE_VOID } },
  1090. { "ivec4", TYPE_IVEC4, { TYPE_INT, TYPE_VOID } },
  1091. { "ivec4", TYPE_IVEC4, { TYPE_INT, TYPE_INT, TYPE_INT, TYPE_INT, TYPE_VOID } },
  1092. { "ivec4", TYPE_IVEC4, { TYPE_INT, TYPE_IVEC2, TYPE_INT, TYPE_VOID } },
  1093. { "ivec4", TYPE_IVEC4, { TYPE_IVEC2, TYPE_INT, TYPE_INT, TYPE_VOID } },
  1094. { "ivec4", TYPE_IVEC4, { TYPE_INT, TYPE_INT, TYPE_IVEC2, TYPE_VOID } },
  1095. { "ivec4", TYPE_IVEC4, { TYPE_INT, TYPE_IVEC3, TYPE_VOID } },
  1096. { "ivec4", TYPE_IVEC4, { TYPE_IVEC3, TYPE_INT, TYPE_VOID } },
  1097. { "ivec4", TYPE_IVEC4, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1098. { "uint", TYPE_UINT, { TYPE_UINT, TYPE_VOID } },
  1099. { "uvec2", TYPE_UVEC2, { TYPE_UINT, TYPE_VOID } },
  1100. { "uvec2", TYPE_UVEC2, { TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1101. { "uvec3", TYPE_UVEC3, { TYPE_UINT, TYPE_VOID } },
  1102. { "uvec3", TYPE_UVEC3, { TYPE_UINT, TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1103. { "uvec3", TYPE_UVEC3, { TYPE_UVEC2, TYPE_UINT, TYPE_VOID } },
  1104. { "uvec3", TYPE_UVEC3, { TYPE_UINT, TYPE_UVEC2, TYPE_VOID } },
  1105. { "uvec4", TYPE_UVEC4, { TYPE_UINT, TYPE_VOID } },
  1106. { "uvec4", TYPE_UVEC4, { TYPE_UINT, TYPE_UINT, TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1107. { "uvec4", TYPE_UVEC4, { TYPE_UINT, TYPE_UVEC2, TYPE_UINT, TYPE_VOID } },
  1108. { "uvec4", TYPE_UVEC4, { TYPE_UVEC2, TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1109. { "uvec4", TYPE_UVEC4, { TYPE_UINT, TYPE_UINT, TYPE_UVEC2, TYPE_VOID } },
  1110. { "uvec4", TYPE_UVEC4, { TYPE_UINT, TYPE_UVEC3, TYPE_VOID } },
  1111. { "uvec4", TYPE_UVEC4, { TYPE_UVEC3, TYPE_UINT, TYPE_VOID } },
  1112. { "uvec4", TYPE_UVEC4, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1113. { "mat2", TYPE_MAT2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1114. { "mat3", TYPE_MAT3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1115. { "mat4", TYPE_MAT4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1116. { "mat2", TYPE_MAT2, { TYPE_FLOAT, TYPE_VOID } },
  1117. { "mat3", TYPE_MAT3, { TYPE_FLOAT, TYPE_VOID } },
  1118. { "mat4", TYPE_MAT4, { TYPE_FLOAT, TYPE_VOID } },
  1119. //conversion scalars
  1120. { "int", TYPE_INT, { TYPE_BOOL, TYPE_VOID } },
  1121. { "int", TYPE_INT, { TYPE_INT, TYPE_VOID } },
  1122. { "int", TYPE_INT, { TYPE_UINT, TYPE_VOID } },
  1123. { "int", TYPE_INT, { TYPE_FLOAT, TYPE_VOID } },
  1124. { "float", TYPE_FLOAT, { TYPE_BOOL, TYPE_VOID } },
  1125. { "float", TYPE_FLOAT, { TYPE_INT, TYPE_VOID } },
  1126. { "float", TYPE_FLOAT, { TYPE_UINT, TYPE_VOID } },
  1127. { "float", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1128. { "uint", TYPE_UINT, { TYPE_BOOL, TYPE_VOID } },
  1129. { "uint", TYPE_UINT, { TYPE_INT, TYPE_VOID } },
  1130. { "uint", TYPE_UINT, { TYPE_UINT, TYPE_VOID } },
  1131. { "uint", TYPE_UINT, { TYPE_FLOAT, TYPE_VOID } },
  1132. { "bool", TYPE_BOOL, { TYPE_BOOL, TYPE_VOID } },
  1133. { "bool", TYPE_BOOL, { TYPE_INT, TYPE_VOID } },
  1134. { "bool", TYPE_BOOL, { TYPE_UINT, TYPE_VOID } },
  1135. { "bool", TYPE_BOOL, { TYPE_FLOAT, TYPE_VOID } },
  1136. //conversion vectors
  1137. { "ivec2", TYPE_IVEC2, { TYPE_BVEC2, TYPE_VOID } },
  1138. { "ivec2", TYPE_IVEC2, { TYPE_IVEC2, TYPE_VOID } },
  1139. { "ivec2", TYPE_IVEC2, { TYPE_UVEC2, TYPE_VOID } },
  1140. { "ivec2", TYPE_IVEC2, { TYPE_VEC2, TYPE_VOID } },
  1141. { "vec2", TYPE_VEC2, { TYPE_BVEC2, TYPE_VOID } },
  1142. { "vec2", TYPE_VEC2, { TYPE_IVEC2, TYPE_VOID } },
  1143. { "vec2", TYPE_VEC2, { TYPE_UVEC2, TYPE_VOID } },
  1144. { "vec2", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1145. { "uvec2", TYPE_UVEC2, { TYPE_BVEC2, TYPE_VOID } },
  1146. { "uvec2", TYPE_UVEC2, { TYPE_IVEC2, TYPE_VOID } },
  1147. { "uvec2", TYPE_UVEC2, { TYPE_UVEC2, TYPE_VOID } },
  1148. { "uvec2", TYPE_UVEC2, { TYPE_VEC2, TYPE_VOID } },
  1149. { "bvec2", TYPE_BVEC2, { TYPE_BVEC2, TYPE_VOID } },
  1150. { "bvec2", TYPE_BVEC2, { TYPE_IVEC2, TYPE_VOID } },
  1151. { "bvec2", TYPE_BVEC2, { TYPE_UVEC2, TYPE_VOID } },
  1152. { "bvec2", TYPE_BVEC2, { TYPE_VEC2, TYPE_VOID } },
  1153. { "ivec3", TYPE_IVEC3, { TYPE_BVEC3, TYPE_VOID } },
  1154. { "ivec3", TYPE_IVEC3, { TYPE_IVEC3, TYPE_VOID } },
  1155. { "ivec3", TYPE_IVEC3, { TYPE_UVEC3, TYPE_VOID } },
  1156. { "ivec3", TYPE_IVEC3, { TYPE_VEC3, TYPE_VOID } },
  1157. { "vec3", TYPE_VEC3, { TYPE_BVEC3, TYPE_VOID } },
  1158. { "vec3", TYPE_VEC3, { TYPE_IVEC3, TYPE_VOID } },
  1159. { "vec3", TYPE_VEC3, { TYPE_UVEC3, TYPE_VOID } },
  1160. { "vec3", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1161. { "uvec3", TYPE_UVEC3, { TYPE_BVEC3, TYPE_VOID } },
  1162. { "uvec3", TYPE_UVEC3, { TYPE_IVEC3, TYPE_VOID } },
  1163. { "uvec3", TYPE_UVEC3, { TYPE_UVEC3, TYPE_VOID } },
  1164. { "uvec3", TYPE_UVEC3, { TYPE_VEC3, TYPE_VOID } },
  1165. { "bvec3", TYPE_BVEC3, { TYPE_BVEC3, TYPE_VOID } },
  1166. { "bvec3", TYPE_BVEC3, { TYPE_IVEC3, TYPE_VOID } },
  1167. { "bvec3", TYPE_BVEC3, { TYPE_UVEC3, TYPE_VOID } },
  1168. { "bvec3", TYPE_BVEC3, { TYPE_VEC3, TYPE_VOID } },
  1169. { "ivec4", TYPE_IVEC4, { TYPE_BVEC4, TYPE_VOID } },
  1170. { "ivec4", TYPE_IVEC4, { TYPE_IVEC4, TYPE_VOID } },
  1171. { "ivec4", TYPE_IVEC4, { TYPE_UVEC4, TYPE_VOID } },
  1172. { "ivec4", TYPE_IVEC4, { TYPE_VEC4, TYPE_VOID } },
  1173. { "vec4", TYPE_VEC4, { TYPE_BVEC4, TYPE_VOID } },
  1174. { "vec4", TYPE_VEC4, { TYPE_IVEC4, TYPE_VOID } },
  1175. { "vec4", TYPE_VEC4, { TYPE_UVEC4, TYPE_VOID } },
  1176. { "vec4", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1177. { "uvec4", TYPE_UVEC4, { TYPE_BVEC4, TYPE_VOID } },
  1178. { "uvec4", TYPE_UVEC4, { TYPE_IVEC4, TYPE_VOID } },
  1179. { "uvec4", TYPE_UVEC4, { TYPE_UVEC4, TYPE_VOID } },
  1180. { "uvec4", TYPE_UVEC4, { TYPE_VEC4, TYPE_VOID } },
  1181. { "bvec4", TYPE_BVEC4, { TYPE_BVEC4, TYPE_VOID } },
  1182. { "bvec4", TYPE_BVEC4, { TYPE_IVEC4, TYPE_VOID } },
  1183. { "bvec4", TYPE_BVEC4, { TYPE_UVEC4, TYPE_VOID } },
  1184. { "bvec4", TYPE_BVEC4, { TYPE_VEC4, TYPE_VOID } },
  1185. //builtins - trigonometry
  1186. { "sin", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1187. { "cos", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1188. { "tan", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1189. { "asin", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1190. { "acos", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1191. { "atan", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1192. { "atan2", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1193. { "sinh", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1194. { "cosh", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1195. { "tanh", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1196. //builtins - exponential
  1197. { "pow", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1198. { "pow", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1199. { "pow", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1200. { "pow", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1201. { "exp", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1202. { "exp", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1203. { "exp", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1204. { "exp", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1205. { "log", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1206. { "log", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1207. { "log", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1208. { "log", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1209. { "sqrt", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1210. { "sqrt", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1211. { "sqrt", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1212. { "sqrt", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1213. //builtins - common
  1214. { "abs", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1215. { "abs", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1216. { "abs", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1217. { "abs", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1218. { "abs", TYPE_INT, { TYPE_INT, TYPE_VOID } },
  1219. { "abs", TYPE_IVEC2, { TYPE_IVEC2, TYPE_VOID } },
  1220. { "abs", TYPE_IVEC3, { TYPE_IVEC3, TYPE_VOID } },
  1221. { "abs", TYPE_IVEC4, { TYPE_IVEC4, TYPE_VOID } },
  1222. { "abs", TYPE_UINT, { TYPE_UINT, TYPE_VOID } },
  1223. { "abs", TYPE_UVEC2, { TYPE_UVEC2, TYPE_VOID } },
  1224. { "abs", TYPE_UVEC3, { TYPE_UVEC3, TYPE_VOID } },
  1225. { "abs", TYPE_UVEC4, { TYPE_UVEC4, TYPE_VOID } },
  1226. { "sign", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1227. { "sign", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1228. { "sign", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1229. { "sign", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1230. { "sign", TYPE_INT, { TYPE_INT, TYPE_VOID } },
  1231. { "sign", TYPE_IVEC2, { TYPE_IVEC2, TYPE_VOID } },
  1232. { "sign", TYPE_IVEC3, { TYPE_IVEC3, TYPE_VOID } },
  1233. { "sign", TYPE_IVEC4, { TYPE_IVEC4, TYPE_VOID } },
  1234. { "floor", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1235. { "floor", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1236. { "floor", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1237. { "floor", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1238. { "trunc", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1239. { "trunc", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1240. { "trunc", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1241. { "trunc", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1242. { "round", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1243. { "round", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1244. { "round", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1245. { "round", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1246. { "ceil", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1247. { "ceil", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1248. { "ceil", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1249. { "ceil", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1250. { "fract", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1251. { "fract", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1252. { "fract", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1253. { "fract", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1254. { "mod", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1255. { "mod", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1256. { "mod", TYPE_VEC2, { TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1257. { "mod", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1258. { "mod", TYPE_VEC3, { TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1259. { "mod", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1260. { "mod", TYPE_VEC4, { TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  1261. { "modf", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1262. { "modf", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1263. { "modf", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1264. { "modf", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1265. { "min", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1266. { "min", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1267. { "min", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1268. { "min", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1269. { "min", TYPE_INT, { TYPE_INT, TYPE_INT, TYPE_VOID } },
  1270. { "min", TYPE_IVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1271. { "min", TYPE_IVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1272. { "min", TYPE_IVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1273. { "min", TYPE_UINT, { TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1274. { "min", TYPE_UVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1275. { "min", TYPE_UVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1276. { "min", TYPE_UVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1277. { "max", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1278. { "max", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1279. { "max", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1280. { "max", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1281. { "max", TYPE_INT, { TYPE_INT, TYPE_INT, TYPE_VOID } },
  1282. { "max", TYPE_IVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1283. { "max", TYPE_IVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1284. { "max", TYPE_IVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1285. { "max", TYPE_UINT, { TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1286. { "max", TYPE_UVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1287. { "max", TYPE_UVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1288. { "max", TYPE_UVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1289. { "clamp", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1290. { "clamp", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1291. { "clamp", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1292. { "clamp", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1293. { "clamp", TYPE_VEC2, { TYPE_VEC2, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1294. { "clamp", TYPE_VEC3, { TYPE_VEC3, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1295. { "clamp", TYPE_VEC4, { TYPE_VEC4, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1296. { "clamp", TYPE_INT, { TYPE_INT, TYPE_INT, TYPE_INT, TYPE_VOID } },
  1297. { "clamp", TYPE_IVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1298. { "clamp", TYPE_IVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1299. { "clamp", TYPE_IVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1300. { "clamp", TYPE_IVEC2, { TYPE_IVEC2, TYPE_INT, TYPE_INT, TYPE_VOID } },
  1301. { "clamp", TYPE_IVEC3, { TYPE_IVEC3, TYPE_INT, TYPE_INT, TYPE_VOID } },
  1302. { "clamp", TYPE_IVEC4, { TYPE_IVEC4, TYPE_INT, TYPE_INT, TYPE_VOID } },
  1303. { "clamp", TYPE_UINT, { TYPE_UINT, TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1304. { "clamp", TYPE_UVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1305. { "clamp", TYPE_UVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1306. { "clamp", TYPE_UVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1307. { "clamp", TYPE_UVEC2, { TYPE_UVEC2, TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1308. { "clamp", TYPE_UVEC3, { TYPE_UVEC3, TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1309. { "clamp", TYPE_UVEC4, { TYPE_UVEC4, TYPE_UINT, TYPE_UINT, TYPE_VOID } },
  1310. { "mix", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1311. { "mix", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_BOOL, TYPE_VOID } },
  1312. { "mix", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1313. { "mix", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_BOOL, TYPE_VOID } },
  1314. { "mix", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_BVEC2, TYPE_VOID } },
  1315. { "mix", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1316. { "mix", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1317. { "mix", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_BOOL, TYPE_VOID } },
  1318. { "mix", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_BVEC3, TYPE_VOID } },
  1319. { "mix", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1320. { "mix", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  1321. { "mix", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_BOOL, TYPE_VOID } },
  1322. { "mix", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_BVEC3, TYPE_VOID } },
  1323. { "mix", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1324. { "step", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1325. { "step", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1326. { "step", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1327. { "step", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1328. { "step", TYPE_VEC2, { TYPE_FLOAT, TYPE_VEC2, TYPE_VOID } },
  1329. { "step", TYPE_VEC3, { TYPE_FLOAT, TYPE_VEC3, TYPE_VOID } },
  1330. { "step", TYPE_VEC4, { TYPE_FLOAT, TYPE_VEC4, TYPE_VOID } },
  1331. { "smoothstep", TYPE_FLOAT, { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT, TYPE_VOID } },
  1332. { "smoothstep", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1333. { "smoothstep", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1334. { "smoothstep", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1335. { "smoothstep", TYPE_VEC2, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VEC2, TYPE_VOID } },
  1336. { "smoothstep", TYPE_VEC3, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VEC3, TYPE_VOID } },
  1337. { "smoothstep", TYPE_VEC4, { TYPE_FLOAT, TYPE_FLOAT, TYPE_VEC4, TYPE_VOID } },
  1338. { "isnan", TYPE_BOOL, { TYPE_FLOAT, TYPE_VOID } },
  1339. { "isnan", TYPE_BOOL, { TYPE_VEC2, TYPE_VOID } },
  1340. { "isnan", TYPE_BOOL, { TYPE_VEC3, TYPE_VOID } },
  1341. { "isnan", TYPE_BOOL, { TYPE_VEC4, TYPE_VOID } },
  1342. { "isinf", TYPE_BOOL, { TYPE_FLOAT, TYPE_VOID } },
  1343. { "isinf", TYPE_BOOL, { TYPE_VEC2, TYPE_VOID } },
  1344. { "isinf", TYPE_BOOL, { TYPE_VEC3, TYPE_VOID } },
  1345. { "isinf", TYPE_BOOL, { TYPE_VEC4, TYPE_VOID } },
  1346. { "floatBitsToInt", TYPE_INT, { TYPE_FLOAT, TYPE_VOID } },
  1347. { "floatBitsToInt", TYPE_IVEC2, { TYPE_VEC2, TYPE_VOID } },
  1348. { "floatBitsToInt", TYPE_IVEC3, { TYPE_VEC3, TYPE_VOID } },
  1349. { "floatBitsToInt", TYPE_IVEC4, { TYPE_VEC4, TYPE_VOID } },
  1350. { "floatBitsToUInt", TYPE_UINT, { TYPE_FLOAT, TYPE_VOID } },
  1351. { "floatBitsToUInt", TYPE_UVEC2, { TYPE_VEC2, TYPE_VOID } },
  1352. { "floatBitsToUInt", TYPE_UVEC3, { TYPE_VEC3, TYPE_VOID } },
  1353. { "floatBitsToUInt", TYPE_UVEC4, { TYPE_VEC4, TYPE_VOID } },
  1354. { "intBitsToFloat", TYPE_FLOAT, { TYPE_INT, TYPE_VOID } },
  1355. { "intBitsToFloat", TYPE_VEC2, { TYPE_IVEC2, TYPE_VOID } },
  1356. { "intBitsToFloat", TYPE_VEC3, { TYPE_IVEC3, TYPE_VOID } },
  1357. { "intBitsToFloat", TYPE_VEC4, { TYPE_IVEC4, TYPE_VOID } },
  1358. { "uintBitsToFloat", TYPE_FLOAT, { TYPE_UINT, TYPE_VOID } },
  1359. { "uintBitsToFloat", TYPE_VEC2, { TYPE_UVEC2, TYPE_VOID } },
  1360. { "uintBitsToFloat", TYPE_VEC3, { TYPE_UVEC3, TYPE_VOID } },
  1361. { "uintBitsToFloat", TYPE_VEC4, { TYPE_UVEC4, TYPE_VOID } },
  1362. //builtins - geometric
  1363. { "length", TYPE_FLOAT, { TYPE_VEC2, TYPE_VOID } },
  1364. { "length", TYPE_FLOAT, { TYPE_VEC3, TYPE_VOID } },
  1365. { "length", TYPE_FLOAT, { TYPE_VEC4, TYPE_VOID } },
  1366. { "distance", TYPE_FLOAT, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1367. { "distance", TYPE_FLOAT, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1368. { "distance", TYPE_FLOAT, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1369. { "dot", TYPE_FLOAT, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1370. { "dot", TYPE_FLOAT, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1371. { "dot", TYPE_FLOAT, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1372. { "cross", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1373. { "normalize", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1374. { "normalize", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1375. { "normalize", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1376. { "reflect", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1377. { "refract", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1378. { "faceforward", TYPE_VEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1379. { "faceforward", TYPE_VEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1380. { "faceforward", TYPE_VEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1381. { "matrixCompMult", TYPE_MAT2, { TYPE_MAT2, TYPE_MAT2, TYPE_VOID } },
  1382. { "matrixCompMult", TYPE_MAT3, { TYPE_MAT3, TYPE_MAT3, TYPE_VOID } },
  1383. { "matrixCompMult", TYPE_MAT4, { TYPE_MAT4, TYPE_MAT4, TYPE_VOID } },
  1384. { "outerProduct", TYPE_MAT2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1385. { "outerProduct", TYPE_MAT3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1386. { "outerProduct", TYPE_MAT4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1387. { "transpose", TYPE_MAT2, { TYPE_MAT2, TYPE_VOID } },
  1388. { "transpose", TYPE_MAT3, { TYPE_MAT3, TYPE_VOID } },
  1389. { "transpose", TYPE_MAT4, { TYPE_MAT4, TYPE_VOID } },
  1390. { "determinant", TYPE_FLOAT, { TYPE_MAT2, TYPE_VOID } },
  1391. { "determinant", TYPE_FLOAT, { TYPE_MAT3, TYPE_VOID } },
  1392. { "determinant", TYPE_FLOAT, { TYPE_MAT4, TYPE_VOID } },
  1393. { "inverse", TYPE_MAT2, { TYPE_MAT2, TYPE_VOID } },
  1394. { "inverse", TYPE_MAT3, { TYPE_MAT3, TYPE_VOID } },
  1395. { "inverse", TYPE_MAT4, { TYPE_MAT4, TYPE_VOID } },
  1396. { "lessThan", TYPE_BVEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1397. { "lessThan", TYPE_BVEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1398. { "lessThan", TYPE_BVEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1399. { "lessThan", TYPE_BVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1400. { "lessThan", TYPE_BVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1401. { "lessThan", TYPE_BVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1402. { "lessThan", TYPE_BVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1403. { "lessThan", TYPE_BVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1404. { "lessThan", TYPE_BVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1405. { "greaterThan", TYPE_BVEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1406. { "greaterThan", TYPE_BVEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1407. { "greaterThan", TYPE_BVEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1408. { "greaterThan", TYPE_BVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1409. { "greaterThan", TYPE_BVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1410. { "greaterThan", TYPE_BVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1411. { "greaterThan", TYPE_BVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1412. { "greaterThan", TYPE_BVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1413. { "greaterThan", TYPE_BVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1414. { "lessThanEqual", TYPE_BVEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1415. { "lessThanEqual", TYPE_BVEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1416. { "lessThanEqual", TYPE_BVEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1417. { "lessThanEqual", TYPE_BVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1418. { "lessThanEqual", TYPE_BVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1419. { "lessThanEqual", TYPE_BVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1420. { "lessThanEqual", TYPE_BVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1421. { "lessThanEqual", TYPE_BVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1422. { "lessThanEqual", TYPE_BVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1423. { "greaterThanEqual", TYPE_BVEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1424. { "greaterThanEqual", TYPE_BVEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1425. { "greaterThanEqual", TYPE_BVEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1426. { "greaterThanEqual", TYPE_BVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1427. { "greaterThanEqual", TYPE_BVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1428. { "greaterThanEqual", TYPE_BVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1429. { "greaterThanEqual", TYPE_BVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1430. { "greaterThanEqual", TYPE_BVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1431. { "greaterThanEqual", TYPE_BVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1432. { "equal", TYPE_BVEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1433. { "equal", TYPE_BVEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1434. { "equal", TYPE_BVEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1435. { "equal", TYPE_BVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1436. { "equal", TYPE_BVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1437. { "equal", TYPE_BVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1438. { "equal", TYPE_BVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1439. { "equal", TYPE_BVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1440. { "equal", TYPE_BVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1441. { "equal", TYPE_BVEC2, { TYPE_BVEC2, TYPE_BVEC2, TYPE_VOID } },
  1442. { "equal", TYPE_BVEC3, { TYPE_BVEC3, TYPE_BVEC3, TYPE_VOID } },
  1443. { "equal", TYPE_BVEC4, { TYPE_BVEC4, TYPE_BVEC4, TYPE_VOID } },
  1444. { "notEqual", TYPE_BVEC2, { TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1445. { "notEqual", TYPE_BVEC3, { TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1446. { "notEqual", TYPE_BVEC4, { TYPE_VEC4, TYPE_VEC4, TYPE_VOID } },
  1447. { "notEqual", TYPE_BVEC2, { TYPE_IVEC2, TYPE_IVEC2, TYPE_VOID } },
  1448. { "notEqual", TYPE_BVEC3, { TYPE_IVEC3, TYPE_IVEC3, TYPE_VOID } },
  1449. { "notEqual", TYPE_BVEC4, { TYPE_IVEC4, TYPE_IVEC4, TYPE_VOID } },
  1450. { "notEqual", TYPE_BVEC2, { TYPE_UVEC2, TYPE_UVEC2, TYPE_VOID } },
  1451. { "notEqual", TYPE_BVEC3, { TYPE_UVEC3, TYPE_UVEC3, TYPE_VOID } },
  1452. { "notEqual", TYPE_BVEC4, { TYPE_UVEC4, TYPE_UVEC4, TYPE_VOID } },
  1453. { "notEqual", TYPE_BVEC2, { TYPE_BVEC2, TYPE_BVEC2, TYPE_VOID } },
  1454. { "notEqual", TYPE_BVEC3, { TYPE_BVEC3, TYPE_BVEC3, TYPE_VOID } },
  1455. { "notEqual", TYPE_BVEC4, { TYPE_BVEC4, TYPE_BVEC4, TYPE_VOID } },
  1456. { "any", TYPE_BOOL, { TYPE_BVEC2, TYPE_VOID } },
  1457. { "any", TYPE_BOOL, { TYPE_BVEC3, TYPE_VOID } },
  1458. { "any", TYPE_BOOL, { TYPE_BVEC4, TYPE_VOID } },
  1459. { "all", TYPE_BOOL, { TYPE_BVEC2, TYPE_VOID } },
  1460. { "all", TYPE_BOOL, { TYPE_BVEC3, TYPE_VOID } },
  1461. { "all", TYPE_BOOL, { TYPE_BVEC4, TYPE_VOID } },
  1462. { "not", TYPE_BOOL, { TYPE_BVEC2, TYPE_VOID } },
  1463. { "not", TYPE_BOOL, { TYPE_BVEC3, TYPE_VOID } },
  1464. { "not", TYPE_BOOL, { TYPE_BVEC4, TYPE_VOID } },
  1465. //builtins - texture
  1466. { "textureSize", TYPE_IVEC2, { TYPE_SAMPLER2D, TYPE_INT, TYPE_VOID } },
  1467. { "textureSize", TYPE_IVEC2, { TYPE_ISAMPLER2D, TYPE_INT, TYPE_VOID } },
  1468. { "textureSize", TYPE_IVEC2, { TYPE_USAMPLER2D, TYPE_INT, TYPE_VOID } },
  1469. { "textureSize", TYPE_IVEC2, { TYPE_SAMPLERCUBE, TYPE_INT, TYPE_VOID } },
  1470. { "texture", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC2, TYPE_VOID } },
  1471. { "texture", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1472. { "texture", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC2, TYPE_VOID } },
  1473. { "texture", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1474. { "texture", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC2, TYPE_VOID } },
  1475. { "texture", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1476. { "texture", TYPE_VEC4, { TYPE_SAMPLERCUBE, TYPE_VEC3, TYPE_VOID } },
  1477. { "texture", TYPE_VEC4, { TYPE_SAMPLERCUBE, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1478. { "textureProj", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC3, TYPE_VOID } },
  1479. { "textureProj", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC4, TYPE_VOID } },
  1480. { "textureProj", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1481. { "textureProj", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  1482. { "textureProj", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC3, TYPE_VOID } },
  1483. { "textureProj", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC4, TYPE_VOID } },
  1484. { "textureProj", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1485. { "textureProj", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  1486. { "textureProj", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC3, TYPE_VOID } },
  1487. { "textureProj", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC4, TYPE_VOID } },
  1488. { "textureProj", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1489. { "textureProj", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  1490. { "textureLod", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1491. { "textureLod", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1492. { "textureLod", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC2, TYPE_FLOAT, TYPE_VOID } },
  1493. { "textureLod", TYPE_VEC4, { TYPE_SAMPLERCUBE, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1494. { "texelFetch", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_IVEC2, TYPE_INT, TYPE_VOID } },
  1495. { "texelFetch", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_IVEC2, TYPE_INT, TYPE_VOID } },
  1496. { "texelFetch", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_IVEC2, TYPE_INT, TYPE_VOID } },
  1497. { "textureProjLod", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1498. { "textureProjLod", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  1499. { "textureProjLod", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1500. { "textureProjLod", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  1501. { "textureProjLod", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC3, TYPE_FLOAT, TYPE_VOID } },
  1502. { "textureProjLod", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC4, TYPE_FLOAT, TYPE_VOID } },
  1503. { "textureGrad", TYPE_VEC4, { TYPE_SAMPLER2D, TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1504. { "textureGrad", TYPE_IVEC4, { TYPE_ISAMPLER2D, TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1505. { "textureGrad", TYPE_UVEC4, { TYPE_USAMPLER2D, TYPE_VEC2, TYPE_VEC2, TYPE_VEC2, TYPE_VOID } },
  1506. { "textureGrad", TYPE_VEC4, { TYPE_SAMPLERCUBE, TYPE_VEC3, TYPE_VEC3, TYPE_VEC3, TYPE_VOID } },
  1507. { "textureScreen", TYPE_VEC4, { TYPE_VEC2, TYPE_VOID } },
  1508. { "dFdx", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1509. { "dFdx", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1510. { "dFdx", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1511. { "dFdx", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1512. { "dFdy", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1513. { "dFdy", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1514. { "dFdy", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1515. { "dFdy", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1516. { "fwidth", TYPE_FLOAT, { TYPE_FLOAT, TYPE_VOID } },
  1517. { "fwidth", TYPE_VEC2, { TYPE_VEC2, TYPE_VOID } },
  1518. { "fwidth", TYPE_VEC3, { TYPE_VEC3, TYPE_VOID } },
  1519. { "fwidth", TYPE_VEC4, { TYPE_VEC4, TYPE_VOID } },
  1520. { NULL, TYPE_VOID, { TYPE_VOID } }
  1521. };
  1522. bool ShaderLanguage::_validate_function_call(BlockNode *p_block, OperatorNode *p_func, DataType *r_ret_type) {
  1523. ERR_FAIL_COND_V(p_func->op != OP_CALL && p_func->op != OP_CONSTRUCT, NULL);
  1524. Vector<DataType> args;
  1525. ERR_FAIL_COND_V(p_func->arguments[0]->type != Node::TYPE_VARIABLE, NULL);
  1526. StringName name = static_cast<VariableNode *>(p_func->arguments[0])->name.operator String();
  1527. bool all_const = true;
  1528. for (int i = 1; i < p_func->arguments.size(); i++) {
  1529. if (p_func->arguments[i]->type != Node::TYPE_CONSTANT)
  1530. all_const = false;
  1531. args.push_back(p_func->arguments[i]->get_datatype());
  1532. }
  1533. int argcount = args.size();
  1534. bool failed_builtin = false;
  1535. if (argcount <= 4) {
  1536. // test builtins
  1537. int idx = 0;
  1538. while (builtin_func_defs[idx].name) {
  1539. if (name == builtin_func_defs[idx].name) {
  1540. failed_builtin = true;
  1541. bool fail = false;
  1542. for (int i = 0; i < argcount; i++) {
  1543. if (get_scalar_type(args[i]) == args[i] && p_func->arguments[i + 1]->type == Node::TYPE_CONSTANT && convert_constant(static_cast<ConstantNode *>(p_func->arguments[i + 1]), builtin_func_defs[idx].args[i])) {
  1544. //all good
  1545. } else if (args[i] != builtin_func_defs[idx].args[i]) {
  1546. fail = true;
  1547. break;
  1548. }
  1549. }
  1550. if (!fail && argcount < 4 && builtin_func_defs[idx].args[argcount] != TYPE_VOID)
  1551. fail = true; //make sure the number of arguments matches
  1552. if (!fail) {
  1553. if (r_ret_type)
  1554. *r_ret_type = builtin_func_defs[idx].rettype;
  1555. return true;
  1556. }
  1557. }
  1558. idx++;
  1559. }
  1560. }
  1561. if (failed_builtin) {
  1562. String err = "Invalid arguments for built-in function: " + String(name) + "(";
  1563. for (int i = 0; i < argcount; i++) {
  1564. if (i > 0)
  1565. err += ",";
  1566. if (p_func->arguments[i + 1]->type == Node::TYPE_CONSTANT && p_func->arguments[i + 1]->get_datatype() == TYPE_INT && static_cast<ConstantNode *>(p_func->arguments[i + 1])->values[0].sint < 0) {
  1567. err += "-";
  1568. }
  1569. err += get_datatype_name(args[i]);
  1570. }
  1571. err += ")";
  1572. _set_error(err);
  1573. return false;
  1574. }
  1575. #if 0
  1576. if (found_builtin) {
  1577. if (p_func->op==OP_CONSTRUCT && all_const) {
  1578. Vector<float> cdata;
  1579. for(int i=0;i<argcount;i++) {
  1580. Variant v = static_cast<ConstantNode*>(p_func->arguments[i+1])->value;
  1581. switch(v.get_type()) {
  1582. case Variant::REAL: cdata.push_back(v); break;
  1583. case Variant::INT: cdata.push_back(v); break;
  1584. case Variant::VECTOR2: { Vector2 v2=v; cdata.push_back(v2.x); cdata.push_back(v2.y); } break;
  1585. case Variant::VECTOR3: { Vector3 v3=v; cdata.push_back(v3.x); cdata.push_back(v3.y); cdata.push_back(v3.z);} break;
  1586. case Variant::PLANE: { Plane v4=v; cdata.push_back(v4.normal.x); cdata.push_back(v4.normal.y); cdata.push_back(v4.normal.z); cdata.push_back(v4.d); } break;
  1587. default: ERR_FAIL_V(NULL);
  1588. }
  1589. }
  1590. ConstantNode *cn = parser.create_node<ConstantNode>(p_func->parent);
  1591. Variant data;
  1592. switch(p_func->return_cache) {
  1593. case TYPE_FLOAT: data = cdata[0]; break;
  1594. case TYPE_VEC2:
  1595. if (cdata.size()==1)
  1596. data = Vector2(cdata[0],cdata[0]);
  1597. else
  1598. data = Vector2(cdata[0],cdata[1]);
  1599. break;
  1600. case TYPE_VEC3:
  1601. if (cdata.size()==1)
  1602. data = Vector3(cdata[0],cdata[0],cdata[0]);
  1603. else
  1604. data = Vector3(cdata[0],cdata[1],cdata[2]);
  1605. break;
  1606. case TYPE_VEC4:
  1607. if (cdata.size()==1)
  1608. data = Plane(cdata[0],cdata[0],cdata[0],cdata[0]);
  1609. else
  1610. data = Plane(cdata[0],cdata[1],cdata[2],cdata[3]);
  1611. break;
  1612. }
  1613. cn->datatype=p_func->return_cache;
  1614. cn->value=data;
  1615. return cn;
  1616. }
  1617. return p_func;
  1618. }
  1619. #endif
  1620. // try existing functions..
  1621. StringName exclude_function;
  1622. BlockNode *block = p_block;
  1623. while (block) {
  1624. if (block->parent_function) {
  1625. exclude_function = block->parent_function->name;
  1626. }
  1627. block = block->parent_block;
  1628. }
  1629. if (name == exclude_function) {
  1630. _set_error("Recursion is not allowed");
  1631. return false;
  1632. }
  1633. for (int i = 0; i < shader->functions.size(); i++) {
  1634. if (name != shader->functions[i].name)
  1635. continue;
  1636. if (!shader->functions[i].callable) {
  1637. _set_error("Function '" + String(name) + " can't be called from source code.");
  1638. return false;
  1639. }
  1640. FunctionNode *pfunc = shader->functions[i].function;
  1641. if (pfunc->arguments.size() != args.size())
  1642. continue;
  1643. bool fail = false;
  1644. for (int i = 0; i < args.size(); i++) {
  1645. if (get_scalar_type(args[i]) == args[i] && p_func->arguments[i + 1]->type == Node::TYPE_CONSTANT && convert_constant(static_cast<ConstantNode *>(p_func->arguments[i + 1]), pfunc->arguments[i].type)) {
  1646. //all good
  1647. } else if (args[i] != pfunc->arguments[i].type) {
  1648. fail = true;
  1649. break;
  1650. }
  1651. }
  1652. if (!fail) {
  1653. if (r_ret_type)
  1654. *r_ret_type = pfunc->return_type;
  1655. return true;
  1656. }
  1657. }
  1658. return false;
  1659. }
  1660. bool ShaderLanguage::_parse_function_arguments(BlockNode *p_block, const Map<StringName, DataType> &p_builtin_types, OperatorNode *p_func, int *r_complete_arg) {
  1661. TkPos pos = _get_tkpos();
  1662. Token tk = _get_token();
  1663. if (tk.type == TK_PARENTHESIS_CLOSE) {
  1664. return true;
  1665. }
  1666. _set_tkpos(pos);
  1667. while (true) {
  1668. if (r_complete_arg) {
  1669. pos = _get_tkpos();
  1670. tk = _get_token();
  1671. if (tk.type == TK_CURSOR) {
  1672. *r_complete_arg = p_func->arguments.size() - 1;
  1673. } else {
  1674. _set_tkpos(pos);
  1675. }
  1676. }
  1677. Node *arg = _parse_and_reduce_expression(p_block, p_builtin_types);
  1678. if (!arg) {
  1679. return false;
  1680. }
  1681. p_func->arguments.push_back(arg);
  1682. tk = _get_token();
  1683. if (tk.type == TK_PARENTHESIS_CLOSE) {
  1684. return true;
  1685. } else if (tk.type != TK_COMMA) {
  1686. // something is broken
  1687. _set_error("Expected ',' or ')' after argument");
  1688. return false;
  1689. }
  1690. }
  1691. return true;
  1692. }
  1693. bool ShaderLanguage::is_token_operator(TokenType p_type) {
  1694. return (p_type == TK_OP_EQUAL ||
  1695. p_type == TK_OP_NOT_EQUAL ||
  1696. p_type == TK_OP_LESS ||
  1697. p_type == TK_OP_LESS_EQUAL ||
  1698. p_type == TK_OP_GREATER ||
  1699. p_type == TK_OP_GREATER_EQUAL ||
  1700. p_type == TK_OP_AND ||
  1701. p_type == TK_OP_OR ||
  1702. p_type == TK_OP_NOT ||
  1703. p_type == TK_OP_ADD ||
  1704. p_type == TK_OP_SUB ||
  1705. p_type == TK_OP_MUL ||
  1706. p_type == TK_OP_DIV ||
  1707. p_type == TK_OP_MOD ||
  1708. p_type == TK_OP_SHIFT_LEFT ||
  1709. p_type == TK_OP_SHIFT_RIGHT ||
  1710. p_type == TK_OP_ASSIGN ||
  1711. p_type == TK_OP_ASSIGN_ADD ||
  1712. p_type == TK_OP_ASSIGN_SUB ||
  1713. p_type == TK_OP_ASSIGN_MUL ||
  1714. p_type == TK_OP_ASSIGN_DIV ||
  1715. p_type == TK_OP_ASSIGN_MOD ||
  1716. p_type == TK_OP_ASSIGN_SHIFT_LEFT ||
  1717. p_type == TK_OP_ASSIGN_SHIFT_RIGHT ||
  1718. p_type == TK_OP_ASSIGN_BIT_AND ||
  1719. p_type == TK_OP_ASSIGN_BIT_OR ||
  1720. p_type == TK_OP_ASSIGN_BIT_XOR ||
  1721. p_type == TK_OP_BIT_AND ||
  1722. p_type == TK_OP_BIT_OR ||
  1723. p_type == TK_OP_BIT_XOR ||
  1724. p_type == TK_OP_BIT_INVERT ||
  1725. p_type == TK_OP_INCREMENT ||
  1726. p_type == TK_OP_DECREMENT ||
  1727. p_type == TK_QUESTION ||
  1728. p_type == TK_COLON);
  1729. }
  1730. bool ShaderLanguage::convert_constant(ConstantNode *p_constant, DataType p_to_type, ConstantNode::Value *p_value) {
  1731. if (p_constant->datatype == p_to_type) {
  1732. if (p_value) {
  1733. for (int i = 0; i < p_constant->values.size(); i++) {
  1734. p_value[i] = p_constant->values[i];
  1735. }
  1736. }
  1737. return true;
  1738. } else if (p_constant->datatype == TYPE_INT && p_to_type == TYPE_FLOAT) {
  1739. if (p_value) {
  1740. p_value->real = p_constant->values[0].sint;
  1741. }
  1742. return true;
  1743. } else if (p_constant->datatype == TYPE_UINT && p_to_type == TYPE_FLOAT) {
  1744. if (p_value) {
  1745. p_value->real = p_constant->values[0].uint;
  1746. }
  1747. return true;
  1748. } else if (p_constant->datatype == TYPE_INT && p_to_type == TYPE_UINT) {
  1749. if (p_constant->values[0].sint < 0) {
  1750. return false;
  1751. }
  1752. if (p_value) {
  1753. p_value->uint = p_constant->values[0].sint;
  1754. }
  1755. return true;
  1756. } else if (p_constant->datatype == TYPE_UINT && p_to_type == TYPE_INT) {
  1757. if (p_constant->values[0].uint > 0x7FFFFFFF) {
  1758. return false;
  1759. }
  1760. if (p_value) {
  1761. p_value->sint = p_constant->values[0].uint;
  1762. }
  1763. return true;
  1764. } else
  1765. return false;
  1766. }
  1767. bool ShaderLanguage::is_scalar_type(DataType p_type) {
  1768. return p_type == TYPE_BOOL || p_type == TYPE_INT || p_type == TYPE_UINT || p_type == TYPE_FLOAT;
  1769. }
  1770. bool ShaderLanguage::is_sampler_type(DataType p_type) {
  1771. return p_type == TYPE_SAMPLER2D || p_type == TYPE_ISAMPLER2D || p_type == TYPE_USAMPLER2D || p_type == TYPE_SAMPLERCUBE;
  1772. }
  1773. void ShaderLanguage::get_keyword_list(List<String> *r_keywords) {
  1774. Set<String> kws;
  1775. int idx = 0;
  1776. while (keyword_list[idx].text) {
  1777. kws.insert(keyword_list[idx].text);
  1778. idx++;
  1779. }
  1780. idx = 0;
  1781. while (builtin_func_defs[idx].name) {
  1782. kws.insert(builtin_func_defs[idx].name);
  1783. idx++;
  1784. }
  1785. for (Set<String>::Element *E = kws.front(); E; E = E->next()) {
  1786. r_keywords->push_back(E->get());
  1787. }
  1788. }
  1789. void ShaderLanguage::get_builtin_funcs(List<String> *r_keywords) {
  1790. Set<String> kws;
  1791. int idx = 0;
  1792. while (builtin_func_defs[idx].name) {
  1793. kws.insert(builtin_func_defs[idx].name);
  1794. idx++;
  1795. }
  1796. for (Set<String>::Element *E = kws.front(); E; E = E->next()) {
  1797. r_keywords->push_back(E->get());
  1798. }
  1799. }
  1800. ShaderLanguage::DataType ShaderLanguage::get_scalar_type(DataType p_type) {
  1801. static const DataType scalar_types[] = {
  1802. TYPE_VOID,
  1803. TYPE_BOOL,
  1804. TYPE_BOOL,
  1805. TYPE_BOOL,
  1806. TYPE_BOOL,
  1807. TYPE_INT,
  1808. TYPE_INT,
  1809. TYPE_INT,
  1810. TYPE_INT,
  1811. TYPE_UINT,
  1812. TYPE_UINT,
  1813. TYPE_UINT,
  1814. TYPE_UINT,
  1815. TYPE_FLOAT,
  1816. TYPE_FLOAT,
  1817. TYPE_FLOAT,
  1818. TYPE_FLOAT,
  1819. TYPE_FLOAT,
  1820. TYPE_FLOAT,
  1821. TYPE_FLOAT,
  1822. TYPE_FLOAT,
  1823. TYPE_INT,
  1824. TYPE_UINT,
  1825. TYPE_FLOAT,
  1826. };
  1827. return scalar_types[p_type];
  1828. }
  1829. bool ShaderLanguage::_get_completable_identifier(BlockNode *p_block, CompletionType p_type, StringName &identifier) {
  1830. identifier = StringName();
  1831. TkPos pos;
  1832. Token tk = _get_token();
  1833. if (tk.type == TK_IDENTIFIER) {
  1834. identifier = tk.text;
  1835. pos = _get_tkpos();
  1836. tk = _get_token();
  1837. }
  1838. if (tk.type == TK_CURSOR) {
  1839. completion_type = p_type;
  1840. completion_line = tk_line;
  1841. completion_block = p_block;
  1842. pos = _get_tkpos();
  1843. tk = _get_token();
  1844. if (tk.type == TK_IDENTIFIER) {
  1845. identifier = identifier.operator String() + tk.text.operator String();
  1846. } else {
  1847. _set_tkpos(pos);
  1848. }
  1849. return true;
  1850. } else if (identifier != StringName()) {
  1851. _set_tkpos(pos);
  1852. }
  1853. return false;
  1854. }
  1855. ShaderLanguage::Node *ShaderLanguage::_parse_expression(BlockNode *p_block, const Map<StringName, DataType> &p_builtin_types) {
  1856. Vector<Expression> expression;
  1857. //Vector<TokenType> operators;
  1858. while (true) {
  1859. Node *expr = NULL;
  1860. TkPos prepos = _get_tkpos();
  1861. Token tk = _get_token();
  1862. TkPos pos = _get_tkpos();
  1863. if (tk.type == TK_PARENTHESIS_OPEN) {
  1864. //handle subexpression
  1865. expr = _parse_and_reduce_expression(p_block, p_builtin_types);
  1866. if (!expr)
  1867. return NULL;
  1868. tk = _get_token();
  1869. if (tk.type != TK_PARENTHESIS_CLOSE) {
  1870. _set_error("Expected ')' in expression");
  1871. return NULL;
  1872. }
  1873. } else if (tk.type == TK_REAL_CONSTANT) {
  1874. ConstantNode *constant = alloc_node<ConstantNode>();
  1875. ConstantNode::Value v;
  1876. v.real = tk.constant;
  1877. constant->values.push_back(v);
  1878. constant->datatype = TYPE_FLOAT;
  1879. expr = constant;
  1880. } else if (tk.type == TK_INT_CONSTANT) {
  1881. ConstantNode *constant = alloc_node<ConstantNode>();
  1882. ConstantNode::Value v;
  1883. v.sint = tk.constant;
  1884. constant->values.push_back(v);
  1885. constant->datatype = TYPE_INT;
  1886. expr = constant;
  1887. } else if (tk.type == TK_TRUE) {
  1888. //print_line("found true");
  1889. //handle true constant
  1890. ConstantNode *constant = alloc_node<ConstantNode>();
  1891. ConstantNode::Value v;
  1892. v.boolean = true;
  1893. constant->values.push_back(v);
  1894. constant->datatype = TYPE_BOOL;
  1895. expr = constant;
  1896. } else if (tk.type == TK_FALSE) {
  1897. //handle false constant
  1898. ConstantNode *constant = alloc_node<ConstantNode>();
  1899. ConstantNode::Value v;
  1900. v.boolean = false;
  1901. constant->values.push_back(v);
  1902. constant->datatype = TYPE_BOOL;
  1903. expr = constant;
  1904. } else if (tk.type == TK_TYPE_VOID) {
  1905. //make sure void is not used in expression
  1906. _set_error("Void value not allowed in Expression");
  1907. return NULL;
  1908. } else if (is_token_nonvoid_datatype(tk.type)) {
  1909. //basic type constructor
  1910. OperatorNode *func = alloc_node<OperatorNode>();
  1911. func->op = OP_CONSTRUCT;
  1912. if (is_token_precision(tk.type)) {
  1913. func->return_precision_cache = get_token_precision(tk.type);
  1914. tk = _get_token();
  1915. }
  1916. VariableNode *funcname = alloc_node<VariableNode>();
  1917. funcname->name = get_datatype_name(get_token_datatype(tk.type));
  1918. func->arguments.push_back(funcname);
  1919. tk = _get_token();
  1920. if (tk.type != TK_PARENTHESIS_OPEN) {
  1921. _set_error("Expected '(' after type name");
  1922. return NULL;
  1923. }
  1924. int carg = -1;
  1925. bool ok = _parse_function_arguments(p_block, p_builtin_types, func, &carg);
  1926. if (carg >= 0) {
  1927. completion_type = COMPLETION_CALL_ARGUMENTS;
  1928. completion_line = tk_line;
  1929. completion_block = p_block;
  1930. completion_function = funcname->name;
  1931. completion_argument = carg;
  1932. }
  1933. if (!ok)
  1934. return NULL;
  1935. if (!_validate_function_call(p_block, func, &func->return_cache)) {
  1936. _set_error("No matching constructor found for: '" + String(funcname->name) + "'");
  1937. return NULL;
  1938. }
  1939. //validate_Function_call()
  1940. expr = _reduce_expression(p_block, func);
  1941. } else if (tk.type == TK_IDENTIFIER) {
  1942. _set_tkpos(prepos);
  1943. StringName identifier;
  1944. _get_completable_identifier(p_block, COMPLETION_IDENTIFIER, identifier);
  1945. tk = _get_token();
  1946. if (tk.type == TK_PARENTHESIS_OPEN) {
  1947. //a function
  1948. StringName name = identifier;
  1949. OperatorNode *func = alloc_node<OperatorNode>();
  1950. func->op = OP_CALL;
  1951. VariableNode *funcname = alloc_node<VariableNode>();
  1952. funcname->name = name;
  1953. func->arguments.push_back(funcname);
  1954. int carg = -1;
  1955. bool ok = _parse_function_arguments(p_block, p_builtin_types, func, &carg);
  1956. //test if function was parsed first
  1957. for (int i = 0; i < shader->functions.size(); i++) {
  1958. if (shader->functions[i].name == name) {
  1959. //add to current function as dependency
  1960. for (int j = 0; j < shader->functions.size(); j++) {
  1961. if (shader->functions[j].name == current_function) {
  1962. shader->functions[j].uses_function.insert(name);
  1963. break;
  1964. }
  1965. }
  1966. break;
  1967. }
  1968. }
  1969. if (carg >= 0) {
  1970. completion_type = COMPLETION_CALL_ARGUMENTS;
  1971. completion_line = tk_line;
  1972. completion_block = p_block;
  1973. completion_function = funcname->name;
  1974. completion_argument = carg;
  1975. }
  1976. if (!ok)
  1977. return NULL;
  1978. if (!_validate_function_call(p_block, func, &func->return_cache)) {
  1979. _set_error("No matching function found for: '" + String(funcname->name) + "'");
  1980. return NULL;
  1981. }
  1982. expr = func;
  1983. } else {
  1984. //an identifier
  1985. _set_tkpos(pos);
  1986. DataType data_type;
  1987. IdentifierType ident_type;
  1988. if (!_find_identifier(p_block, p_builtin_types, identifier, &data_type, &ident_type)) {
  1989. _set_error("Unknown identifier in expression: " + String(identifier));
  1990. return NULL;
  1991. }
  1992. if (ident_type == IDENTIFIER_FUNCTION) {
  1993. _set_error("Can't use function as identifier: " + String(identifier));
  1994. return NULL;
  1995. }
  1996. VariableNode *varname = alloc_node<VariableNode>();
  1997. varname->name = identifier;
  1998. varname->datatype_cache = data_type;
  1999. expr = varname;
  2000. }
  2001. } else if (tk.type == TK_OP_ADD) {
  2002. continue; //this one does nothing
  2003. } else if (tk.type == TK_OP_SUB || tk.type == TK_OP_NOT || tk.type == TK_OP_BIT_INVERT || tk.type == TK_OP_INCREMENT || tk.type == TK_OP_DECREMENT) {
  2004. Expression e;
  2005. e.is_op = true;
  2006. switch (tk.type) {
  2007. case TK_OP_SUB: e.op = OP_NEGATE; break;
  2008. case TK_OP_NOT: e.op = OP_NOT; break;
  2009. case TK_OP_BIT_INVERT: e.op = OP_BIT_INVERT; break;
  2010. case TK_OP_INCREMENT: e.op = OP_INCREMENT; break;
  2011. case TK_OP_DECREMENT: e.op = OP_DECREMENT; break;
  2012. default: ERR_FAIL_V(NULL);
  2013. }
  2014. expression.push_back(e);
  2015. continue;
  2016. } else {
  2017. _set_error("Expected expression, found: " + get_token_text(tk));
  2018. return NULL;
  2019. //nothing
  2020. }
  2021. ERR_FAIL_COND_V(!expr, NULL);
  2022. /* OK now see what's NEXT to the operator.. */
  2023. /* OK now see what's NEXT to the operator.. */
  2024. /* OK now see what's NEXT to the operator.. */
  2025. while (true) {
  2026. TkPos pos = _get_tkpos();
  2027. tk = _get_token();
  2028. if (tk.type == TK_PERIOD) {
  2029. StringName identifier;
  2030. if (_get_completable_identifier(p_block, COMPLETION_INDEX, identifier)) {
  2031. completion_base = expr->get_datatype();
  2032. }
  2033. if (identifier == StringName()) {
  2034. _set_error("Expected identifier as member");
  2035. return NULL;
  2036. }
  2037. DataType dt = expr->get_datatype();
  2038. String ident = identifier;
  2039. bool ok = true;
  2040. DataType member_type;
  2041. switch (dt) {
  2042. case TYPE_BVEC2:
  2043. case TYPE_IVEC2:
  2044. case TYPE_UVEC2:
  2045. case TYPE_VEC2: {
  2046. int l = ident.length();
  2047. if (l == 1) {
  2048. member_type = DataType(dt - 1);
  2049. } else if (l == 2) {
  2050. member_type = dt;
  2051. } else {
  2052. ok = false;
  2053. break;
  2054. }
  2055. const CharType *c = ident.ptr();
  2056. for (int i = 0; i < l; i++) {
  2057. switch (c[i]) {
  2058. case 'r':
  2059. case 'g':
  2060. case 'x':
  2061. case 'y':
  2062. break;
  2063. default:
  2064. ok = false;
  2065. break;
  2066. }
  2067. }
  2068. } break;
  2069. case TYPE_BVEC3:
  2070. case TYPE_IVEC3:
  2071. case TYPE_UVEC3:
  2072. case TYPE_VEC3: {
  2073. int l = ident.length();
  2074. if (l == 1) {
  2075. member_type = DataType(dt - 2);
  2076. } else if (l == 2) {
  2077. member_type = DataType(dt - 1);
  2078. } else if (l == 3) {
  2079. member_type = dt;
  2080. } else {
  2081. ok = false;
  2082. break;
  2083. }
  2084. const CharType *c = ident.ptr();
  2085. for (int i = 0; i < l; i++) {
  2086. switch (c[i]) {
  2087. case 'r':
  2088. case 'g':
  2089. case 'b':
  2090. case 'x':
  2091. case 'y':
  2092. case 'z':
  2093. break;
  2094. default:
  2095. ok = false;
  2096. break;
  2097. }
  2098. }
  2099. } break;
  2100. case TYPE_BVEC4:
  2101. case TYPE_IVEC4:
  2102. case TYPE_UVEC4:
  2103. case TYPE_VEC4: {
  2104. int l = ident.length();
  2105. if (l == 1) {
  2106. member_type = DataType(dt - 3);
  2107. } else if (l == 2) {
  2108. member_type = DataType(dt - 2);
  2109. } else if (l == 3) {
  2110. member_type = DataType(dt - 1);
  2111. } else if (l == 4) {
  2112. member_type = dt;
  2113. } else {
  2114. ok = false;
  2115. break;
  2116. }
  2117. const CharType *c = ident.ptr();
  2118. for (int i = 0; i < l; i++) {
  2119. switch (c[i]) {
  2120. case 'r':
  2121. case 'g':
  2122. case 'b':
  2123. case 'a':
  2124. case 'x':
  2125. case 'y':
  2126. case 'z':
  2127. case 'w':
  2128. break;
  2129. default:
  2130. ok = false;
  2131. break;
  2132. }
  2133. }
  2134. } break;
  2135. default: {
  2136. ok = false;
  2137. }
  2138. }
  2139. if (!ok) {
  2140. _set_error("Invalid member for " + get_datatype_name(dt) + " expression: ." + ident);
  2141. return NULL;
  2142. }
  2143. MemberNode *mn = alloc_node<MemberNode>();
  2144. mn->basetype = dt;
  2145. mn->datatype = member_type;
  2146. mn->name = ident;
  2147. mn->owner = expr;
  2148. expr = mn;
  2149. //todo
  2150. //member (period) has priority over any operator
  2151. //creates a subindexing expression in place
  2152. /*} else if (tk.type==TK_BRACKET_OPEN) {
  2153. //todo
  2154. //subindexing has priority over any operator
  2155. //creates a subindexing expression in place
  2156. */
  2157. } else if (tk.type == TK_BRACKET_OPEN) {
  2158. Node *index = _parse_and_reduce_expression(p_block, p_builtin_types);
  2159. if (index->get_datatype() != TYPE_INT && index->get_datatype() != TYPE_UINT) {
  2160. _set_error("Only integer datatypes are allowed for indexing");
  2161. return NULL;
  2162. }
  2163. bool index_valid = false;
  2164. DataType member_type;
  2165. switch (expr->get_datatype()) {
  2166. case TYPE_BVEC2:
  2167. case TYPE_VEC2:
  2168. case TYPE_IVEC2:
  2169. case TYPE_UVEC2:
  2170. case TYPE_MAT2:
  2171. if (index->type == Node::TYPE_CONSTANT) {
  2172. uint32_t index_constant = static_cast<ConstantNode *>(index)->values[0].uint;
  2173. if (index_constant >= 2) {
  2174. _set_error("Index out of range (0-1)");
  2175. return NULL;
  2176. }
  2177. } else {
  2178. _set_error("Only integer constants are allowed as index at the moment");
  2179. return NULL;
  2180. }
  2181. index_valid = true;
  2182. switch (expr->get_datatype()) {
  2183. case TYPE_BVEC2: member_type = TYPE_BOOL; break;
  2184. case TYPE_VEC2: member_type = TYPE_FLOAT; break;
  2185. case TYPE_IVEC2: member_type = TYPE_INT; break;
  2186. case TYPE_UVEC2: member_type = TYPE_UINT; break;
  2187. case TYPE_MAT2: member_type = TYPE_VEC2; break;
  2188. }
  2189. break;
  2190. case TYPE_BVEC3:
  2191. case TYPE_VEC3:
  2192. case TYPE_IVEC3:
  2193. case TYPE_UVEC3:
  2194. case TYPE_MAT3:
  2195. if (index->type == Node::TYPE_CONSTANT) {
  2196. uint32_t index_constant = static_cast<ConstantNode *>(index)->values[0].uint;
  2197. if (index_constant >= 3) {
  2198. _set_error("Index out of range (0-2)");
  2199. return NULL;
  2200. }
  2201. } else {
  2202. _set_error("Only integer constants are allowed as index at the moment");
  2203. return NULL;
  2204. }
  2205. index_valid = true;
  2206. switch (expr->get_datatype()) {
  2207. case TYPE_BVEC3: member_type = TYPE_BOOL; break;
  2208. case TYPE_VEC3: member_type = TYPE_FLOAT; break;
  2209. case TYPE_IVEC3: member_type = TYPE_INT; break;
  2210. case TYPE_UVEC3: member_type = TYPE_UINT; break;
  2211. case TYPE_MAT3: member_type = TYPE_VEC3; break;
  2212. }
  2213. break;
  2214. case TYPE_BVEC4:
  2215. case TYPE_VEC4:
  2216. case TYPE_IVEC4:
  2217. case TYPE_UVEC4:
  2218. case TYPE_MAT4:
  2219. if (index->type == Node::TYPE_CONSTANT) {
  2220. uint32_t index_constant = static_cast<ConstantNode *>(index)->values[0].uint;
  2221. if (index_constant >= 4) {
  2222. _set_error("Index out of range (0-3)");
  2223. return NULL;
  2224. }
  2225. } else {
  2226. _set_error("Only integer constants are allowed as index at the moment");
  2227. return NULL;
  2228. }
  2229. index_valid = true;
  2230. switch (expr->get_datatype()) {
  2231. case TYPE_BVEC4: member_type = TYPE_BOOL; break;
  2232. case TYPE_VEC4: member_type = TYPE_FLOAT; break;
  2233. case TYPE_IVEC4: member_type = TYPE_INT; break;
  2234. case TYPE_UVEC4: member_type = TYPE_UINT; break;
  2235. case TYPE_MAT4: member_type = TYPE_VEC4; break;
  2236. }
  2237. break;
  2238. default: {
  2239. _set_error("Object of type '" + get_datatype_name(expr->get_datatype()) + "' can't be indexed");
  2240. return NULL;
  2241. }
  2242. }
  2243. if (!index_valid) {
  2244. _set_error("Invalid index");
  2245. return NULL;
  2246. }
  2247. OperatorNode *op = alloc_node<OperatorNode>();
  2248. op->op = OP_INDEX;
  2249. op->return_cache = member_type;
  2250. op->arguments.push_back(expr);
  2251. op->arguments.push_back(index);
  2252. expr = op;
  2253. tk = _get_token();
  2254. if (tk.type != TK_BRACKET_CLOSE) {
  2255. _set_error("Expected ']' after indexing expression");
  2256. return NULL;
  2257. }
  2258. } else if (tk.type == TK_OP_INCREMENT || tk.type == TK_OP_DECREMENT) {
  2259. OperatorNode *op = alloc_node<OperatorNode>();
  2260. op->op = tk.type == TK_OP_DECREMENT ? OP_POST_DECREMENT : OP_POST_INCREMENT;
  2261. op->arguments.push_back(expr);
  2262. if (!_validate_operator(op, &op->return_cache)) {
  2263. _set_error("Invalid base type for increment/decrement operator");
  2264. return NULL;
  2265. }
  2266. expr = op;
  2267. } else {
  2268. _set_tkpos(pos);
  2269. break;
  2270. }
  2271. }
  2272. Expression e;
  2273. e.is_op = false;
  2274. e.node = expr;
  2275. expression.push_back(e);
  2276. pos = _get_tkpos();
  2277. tk = _get_token();
  2278. if (is_token_operator(tk.type)) {
  2279. Expression o;
  2280. o.is_op = true;
  2281. switch (tk.type) {
  2282. case TK_OP_EQUAL: o.op = OP_EQUAL; break;
  2283. case TK_OP_NOT_EQUAL: o.op = OP_NOT_EQUAL; break;
  2284. case TK_OP_LESS: o.op = OP_LESS; break;
  2285. case TK_OP_LESS_EQUAL: o.op = OP_LESS_EQUAL; break;
  2286. case TK_OP_GREATER: o.op = OP_GREATER; break;
  2287. case TK_OP_GREATER_EQUAL: o.op = OP_GREATER_EQUAL; break;
  2288. case TK_OP_AND: o.op = OP_AND; break;
  2289. case TK_OP_OR: o.op = OP_OR; break;
  2290. case TK_OP_ADD: o.op = OP_ADD; break;
  2291. case TK_OP_SUB: o.op = OP_SUB; break;
  2292. case TK_OP_MUL: o.op = OP_MUL; break;
  2293. case TK_OP_DIV: o.op = OP_DIV; break;
  2294. case TK_OP_MOD: o.op = OP_MOD; break;
  2295. case TK_OP_SHIFT_LEFT: o.op = OP_SHIFT_LEFT; break;
  2296. case TK_OP_SHIFT_RIGHT: o.op = OP_SHIFT_RIGHT; break;
  2297. case TK_OP_ASSIGN: o.op = OP_ASSIGN; break;
  2298. case TK_OP_ASSIGN_ADD: o.op = OP_ASSIGN_ADD; break;
  2299. case TK_OP_ASSIGN_SUB: o.op = OP_ASSIGN_SUB; break;
  2300. case TK_OP_ASSIGN_MUL: o.op = OP_ASSIGN_MUL; break;
  2301. case TK_OP_ASSIGN_DIV: o.op = OP_ASSIGN_DIV; break;
  2302. case TK_OP_ASSIGN_MOD: o.op = OP_ASSIGN_MOD; break;
  2303. case TK_OP_ASSIGN_SHIFT_LEFT: o.op = OP_ASSIGN_SHIFT_LEFT; break;
  2304. case TK_OP_ASSIGN_SHIFT_RIGHT: o.op = OP_ASSIGN_SHIFT_RIGHT; break;
  2305. case TK_OP_ASSIGN_BIT_AND: o.op = OP_ASSIGN_BIT_AND; break;
  2306. case TK_OP_ASSIGN_BIT_OR: o.op = OP_ASSIGN_BIT_OR; break;
  2307. case TK_OP_ASSIGN_BIT_XOR: o.op = OP_ASSIGN_BIT_XOR; break;
  2308. case TK_OP_BIT_AND: o.op = OP_BIT_AND; break;
  2309. case TK_OP_BIT_OR: o.op = OP_BIT_OR; break;
  2310. case TK_OP_BIT_XOR: o.op = OP_BIT_XOR; break;
  2311. case TK_QUESTION: o.op = OP_SELECT_IF; break;
  2312. case TK_COLON: o.op = OP_SELECT_ELSE; break;
  2313. default: {
  2314. _set_error("Invalid token for operator: " + get_token_text(tk));
  2315. return NULL;
  2316. }
  2317. }
  2318. expression.push_back(o);
  2319. } else {
  2320. _set_tkpos(pos); //something else, so rollback and end
  2321. break;
  2322. }
  2323. }
  2324. /* Reduce the set set of expressions and place them in an operator tree, respecting precedence */
  2325. while (expression.size() > 1) {
  2326. int next_op = -1;
  2327. int min_priority = 0xFFFFF;
  2328. bool is_unary = false;
  2329. bool is_ternary = false;
  2330. for (int i = 0; i < expression.size(); i++) {
  2331. if (!expression[i].is_op) {
  2332. continue;
  2333. }
  2334. bool unary = false;
  2335. bool ternary = false;
  2336. int priority;
  2337. switch (expression[i].op) {
  2338. case OP_EQUAL: priority = 8; break;
  2339. case OP_NOT_EQUAL: priority = 8; break;
  2340. case OP_LESS: priority = 7; break;
  2341. case OP_LESS_EQUAL: priority = 7; break;
  2342. case OP_GREATER: priority = 7; break;
  2343. case OP_GREATER_EQUAL: priority = 7; break;
  2344. case OP_AND: priority = 12; break;
  2345. case OP_OR: priority = 14; break;
  2346. case OP_NOT:
  2347. priority = 3;
  2348. unary = true;
  2349. break;
  2350. case OP_NEGATE:
  2351. priority = 3;
  2352. unary = true;
  2353. break;
  2354. case OP_ADD: priority = 5; break;
  2355. case OP_SUB: priority = 5; break;
  2356. case OP_MUL: priority = 4; break;
  2357. case OP_DIV: priority = 4; break;
  2358. case OP_MOD: priority = 4; break;
  2359. case OP_SHIFT_LEFT: priority = 6; break;
  2360. case OP_SHIFT_RIGHT: priority = 6; break;
  2361. case OP_ASSIGN: priority = 16; break;
  2362. case OP_ASSIGN_ADD: priority = 16; break;
  2363. case OP_ASSIGN_SUB: priority = 16; break;
  2364. case OP_ASSIGN_MUL: priority = 16; break;
  2365. case OP_ASSIGN_DIV: priority = 16; break;
  2366. case OP_ASSIGN_MOD: priority = 16; break;
  2367. case OP_ASSIGN_SHIFT_LEFT: priority = 16; break;
  2368. case OP_ASSIGN_SHIFT_RIGHT: priority = 16; break;
  2369. case OP_ASSIGN_BIT_AND: priority = 16; break;
  2370. case OP_ASSIGN_BIT_OR: priority = 16; break;
  2371. case OP_ASSIGN_BIT_XOR: priority = 16; break;
  2372. case OP_BIT_AND: priority = 9; break;
  2373. case OP_BIT_OR: priority = 11; break;
  2374. case OP_BIT_XOR: priority = 10; break;
  2375. case OP_BIT_INVERT:
  2376. priority = 3;
  2377. unary = true;
  2378. break;
  2379. case OP_INCREMENT:
  2380. priority = 3;
  2381. unary = true;
  2382. break;
  2383. case OP_DECREMENT:
  2384. priority = 3;
  2385. unary = true;
  2386. break;
  2387. case OP_SELECT_IF:
  2388. priority = 15;
  2389. ternary = true;
  2390. break;
  2391. case OP_SELECT_ELSE:
  2392. priority = 15;
  2393. ternary = true;
  2394. break;
  2395. default:
  2396. ERR_FAIL_V(NULL); //unexpected operator
  2397. }
  2398. if (priority < min_priority) {
  2399. // < is used for left to right (default)
  2400. // <= is used for right to left
  2401. next_op = i;
  2402. min_priority = priority;
  2403. is_unary = unary;
  2404. is_ternary = ternary;
  2405. }
  2406. }
  2407. ERR_FAIL_COND_V(next_op == -1, NULL);
  2408. // OK! create operator..
  2409. // OK! create operator..
  2410. if (is_unary) {
  2411. int expr_pos = next_op;
  2412. while (expression[expr_pos].is_op) {
  2413. expr_pos++;
  2414. if (expr_pos == expression.size()) {
  2415. //can happen..
  2416. _set_error("Unexpected end of expression..");
  2417. return NULL;
  2418. }
  2419. }
  2420. //consecutively do unary opeators
  2421. for (int i = expr_pos - 1; i >= next_op; i--) {
  2422. OperatorNode *op = alloc_node<OperatorNode>();
  2423. op->op = expression[i].op;
  2424. op->arguments.push_back(expression[i + 1].node);
  2425. expression[i].is_op = false;
  2426. expression[i].node = op;
  2427. if (!_validate_operator(op, &op->return_cache)) {
  2428. String at;
  2429. for (int i = 0; i < op->arguments.size(); i++) {
  2430. if (i > 0)
  2431. at += " and ";
  2432. at += get_datatype_name(op->arguments[i]->get_datatype());
  2433. }
  2434. _set_error("Invalid arguments to unary operator '" + get_operator_text(op->op) + "' :" + at);
  2435. return NULL;
  2436. }
  2437. expression.remove(i + 1);
  2438. }
  2439. } else if (is_ternary) {
  2440. if (next_op < 1 || next_op >= (expression.size() - 1)) {
  2441. _set_error("Parser bug..");
  2442. ERR_FAIL_V(NULL);
  2443. }
  2444. if (next_op + 2 >= expression.size() || !expression[next_op + 2].is_op || expression[next_op + 2].op != OP_SELECT_ELSE) {
  2445. _set_error("Mising matching ':' for select operator");
  2446. return NULL;
  2447. }
  2448. OperatorNode *op = alloc_node<OperatorNode>();
  2449. op->op = expression[next_op].op;
  2450. op->arguments.push_back(expression[next_op - 1].node);
  2451. op->arguments.push_back(expression[next_op + 1].node);
  2452. op->arguments.push_back(expression[next_op + 3].node);
  2453. expression[next_op - 1].is_op = false;
  2454. expression[next_op - 1].node = op;
  2455. if (!_validate_operator(op, &op->return_cache)) {
  2456. String at;
  2457. for (int i = 0; i < op->arguments.size(); i++) {
  2458. if (i > 0)
  2459. at += " and ";
  2460. at += get_datatype_name(op->arguments[i]->get_datatype());
  2461. }
  2462. _set_error("Invalid argument to ternary ?: operator: " + at);
  2463. return NULL;
  2464. }
  2465. for (int i = 0; i < 4; i++) {
  2466. expression.remove(next_op);
  2467. }
  2468. } else {
  2469. if (next_op < 1 || next_op >= (expression.size() - 1)) {
  2470. _set_error("Parser bug..");
  2471. ERR_FAIL_V(NULL);
  2472. }
  2473. OperatorNode *op = alloc_node<OperatorNode>();
  2474. op->op = expression[next_op].op;
  2475. if (expression[next_op - 1].is_op) {
  2476. _set_error("Parser bug..");
  2477. ERR_FAIL_V(NULL);
  2478. }
  2479. if (expression[next_op + 1].is_op) {
  2480. // this is not invalid and can really appear
  2481. // but it becomes invalid anyway because no binary op
  2482. // can be followed by an unary op in a valid combination,
  2483. // due to how precedence works, unaries will always disappear first
  2484. _set_error("Parser bug..");
  2485. }
  2486. op->arguments.push_back(expression[next_op - 1].node); //expression goes as left
  2487. op->arguments.push_back(expression[next_op + 1].node); //next expression goes as right
  2488. expression[next_op - 1].node = op;
  2489. //replace all 3 nodes by this operator and make it an expression
  2490. if (!_validate_operator(op, &op->return_cache)) {
  2491. String at;
  2492. for (int i = 0; i < op->arguments.size(); i++) {
  2493. if (i > 0)
  2494. at += " and ";
  2495. at += get_datatype_name(op->arguments[i]->get_datatype());
  2496. }
  2497. _set_error("Invalid arguments to operator '" + get_operator_text(op->op) + "' :" + at);
  2498. return NULL;
  2499. }
  2500. expression.remove(next_op);
  2501. expression.remove(next_op);
  2502. }
  2503. }
  2504. return expression[0].node;
  2505. }
  2506. ShaderLanguage::Node *ShaderLanguage::_reduce_expression(BlockNode *p_block, ShaderLanguage::Node *p_node) {
  2507. if (p_node->type != Node::TYPE_OPERATOR)
  2508. return p_node;
  2509. //for now only reduce simple constructors
  2510. OperatorNode *op = static_cast<OperatorNode *>(p_node);
  2511. if (op->op == OP_CONSTRUCT) {
  2512. ERR_FAIL_COND_V(op->arguments[0]->type != Node::TYPE_VARIABLE, p_node);
  2513. VariableNode *vn = static_cast<VariableNode *>(op->arguments[0]);
  2514. //StringName name=vn->name;
  2515. DataType base = get_scalar_type(op->get_datatype());
  2516. Vector<ConstantNode::Value> values;
  2517. for (int i = 1; i < op->arguments.size(); i++) {
  2518. op->arguments[i] = _reduce_expression(p_block, op->arguments[i]);
  2519. if (op->arguments[i]->type == Node::TYPE_CONSTANT) {
  2520. ConstantNode *cn = static_cast<ConstantNode *>(op->arguments[i]);
  2521. if (get_scalar_type(cn->datatype) == base) {
  2522. for (int j = 0; j < cn->values.size(); j++) {
  2523. values.push_back(cn->values[j]);
  2524. }
  2525. } else if (get_scalar_type(cn->datatype) == cn->datatype) {
  2526. ConstantNode::Value v;
  2527. if (!convert_constant(cn, base, &v)) {
  2528. return p_node;
  2529. }
  2530. values.push_back(v);
  2531. } else {
  2532. return p_node;
  2533. }
  2534. } else {
  2535. return p_node;
  2536. }
  2537. }
  2538. ConstantNode *cn = alloc_node<ConstantNode>();
  2539. cn->datatype = op->get_datatype();
  2540. cn->values = values;
  2541. return cn;
  2542. } else if (op->op == OP_NEGATE) {
  2543. op->arguments[0] = _reduce_expression(p_block, op->arguments[0]);
  2544. if (op->arguments[0]->type == Node::TYPE_CONSTANT) {
  2545. ConstantNode *cn = static_cast<ConstantNode *>(op->arguments[0]);
  2546. DataType base = get_scalar_type(cn->datatype);
  2547. Vector<ConstantNode::Value> values;
  2548. for (int i = 0; i < cn->values.size(); i++) {
  2549. ConstantNode::Value nv;
  2550. switch (base) {
  2551. case TYPE_BOOL: {
  2552. nv.boolean = !cn->values[i].boolean;
  2553. } break;
  2554. case TYPE_INT: {
  2555. nv.sint = -cn->values[i].sint;
  2556. } break;
  2557. case TYPE_UINT: {
  2558. nv.uint = -cn->values[i].uint;
  2559. } break;
  2560. case TYPE_FLOAT: {
  2561. nv.real = -cn->values[i].real;
  2562. } break;
  2563. default: {}
  2564. }
  2565. values.push_back(nv);
  2566. }
  2567. cn->values = values;
  2568. return cn;
  2569. }
  2570. }
  2571. return p_node;
  2572. }
  2573. ShaderLanguage::Node *ShaderLanguage::_parse_and_reduce_expression(BlockNode *p_block, const Map<StringName, DataType> &p_builtin_types) {
  2574. ShaderLanguage::Node *expr = _parse_expression(p_block, p_builtin_types);
  2575. if (!expr) //errored
  2576. return NULL;
  2577. expr = _reduce_expression(p_block, expr);
  2578. return expr;
  2579. }
  2580. Error ShaderLanguage::_parse_block(BlockNode *p_block, const Map<StringName, DataType> &p_builtin_types, bool p_just_one, bool p_can_break, bool p_can_continue) {
  2581. while (true) {
  2582. TkPos pos = _get_tkpos();
  2583. Token tk = _get_token();
  2584. if (tk.type == TK_CURLY_BRACKET_CLOSE) { //end of block
  2585. if (p_just_one) {
  2586. _set_error("Unexpected '}'");
  2587. return ERR_PARSE_ERROR;
  2588. }
  2589. return OK;
  2590. } else if (is_token_precision(tk.type) || is_token_nonvoid_datatype(tk.type)) {
  2591. DataPrecision precision = PRECISION_DEFAULT;
  2592. if (is_token_precision(tk.type)) {
  2593. precision = get_token_precision(tk.type);
  2594. tk = _get_token();
  2595. if (!is_token_nonvoid_datatype(tk.type)) {
  2596. _set_error("Expected datatype after precission");
  2597. return ERR_PARSE_ERROR;
  2598. }
  2599. }
  2600. DataType type = get_token_datatype(tk.type);
  2601. tk = _get_token();
  2602. while (true) {
  2603. if (tk.type != TK_IDENTIFIER) {
  2604. _set_error("Expected identifier after type");
  2605. return ERR_PARSE_ERROR;
  2606. }
  2607. StringName name = tk.text;
  2608. if (_find_identifier(p_block, p_builtin_types, name)) {
  2609. _set_error("Redefinition of '" + String(name) + "'");
  2610. return ERR_PARSE_ERROR;
  2611. }
  2612. BlockNode::Variable var;
  2613. var.type = type;
  2614. var.precision = precision;
  2615. var.line = tk_line;
  2616. p_block->variables[name] = var;
  2617. tk = _get_token();
  2618. if (tk.type == TK_OP_ASSIGN) {
  2619. //variable creted with assignment! must parse an expression
  2620. Node *n = _parse_and_reduce_expression(p_block, p_builtin_types);
  2621. if (!n)
  2622. return ERR_PARSE_ERROR;
  2623. OperatorNode *assign = alloc_node<OperatorNode>();
  2624. VariableNode *vnode = alloc_node<VariableNode>();
  2625. vnode->name = name;
  2626. vnode->datatype_cache = type;
  2627. assign->arguments.push_back(vnode);
  2628. assign->arguments.push_back(n);
  2629. assign->op = OP_ASSIGN;
  2630. p_block->statements.push_back(assign);
  2631. tk = _get_token();
  2632. if (!_validate_operator(assign)) {
  2633. _set_error("Invalid assignment of '" + get_datatype_name(n->get_datatype()) + "' to '" + get_datatype_name(type) + "'");
  2634. return ERR_PARSE_ERROR;
  2635. }
  2636. }
  2637. if (tk.type == TK_COMMA) {
  2638. tk = _get_token();
  2639. //another variable
  2640. } else if (tk.type == TK_SEMICOLON) {
  2641. break;
  2642. } else {
  2643. _set_error("Expected ',' or ';' after variable");
  2644. return ERR_PARSE_ERROR;
  2645. }
  2646. }
  2647. } else if (tk.type == TK_CURLY_BRACKET_OPEN) {
  2648. //a sub block, just because..
  2649. BlockNode *block = alloc_node<BlockNode>();
  2650. block->parent_block = p_block;
  2651. _parse_block(block, p_builtin_types, false, p_can_break, p_can_continue);
  2652. p_block->statements.push_back(block);
  2653. } else if (tk.type == TK_CF_IF) {
  2654. //if () {}
  2655. tk = _get_token();
  2656. if (tk.type != TK_PARENTHESIS_OPEN) {
  2657. _set_error("Expected '(' after if");
  2658. return ERR_PARSE_ERROR;
  2659. }
  2660. ControlFlowNode *cf = alloc_node<ControlFlowNode>();
  2661. cf->flow_op = FLOW_OP_IF;
  2662. Node *n = _parse_and_reduce_expression(p_block, p_builtin_types);
  2663. if (!n)
  2664. return ERR_PARSE_ERROR;
  2665. tk = _get_token();
  2666. if (tk.type != TK_PARENTHESIS_CLOSE) {
  2667. _set_error("Expected ')' after expression");
  2668. return ERR_PARSE_ERROR;
  2669. }
  2670. BlockNode *block = alloc_node<BlockNode>();
  2671. block->parent_block = p_block;
  2672. cf->expressions.push_back(n);
  2673. cf->blocks.push_back(block);
  2674. p_block->statements.push_back(cf);
  2675. Error err = _parse_block(block, p_builtin_types, true, p_can_break, p_can_continue);
  2676. if (err)
  2677. return err;
  2678. pos = _get_tkpos();
  2679. tk = _get_token();
  2680. if (tk.type == TK_CF_ELSE) {
  2681. block = alloc_node<BlockNode>();
  2682. block->parent_block = p_block;
  2683. cf->blocks.push_back(block);
  2684. err = _parse_block(block, p_builtin_types, true, p_can_break, p_can_continue);
  2685. } else {
  2686. _set_tkpos(pos); //rollback
  2687. }
  2688. } else if (tk.type == TK_CF_WHILE) {
  2689. //if () {}
  2690. tk = _get_token();
  2691. if (tk.type != TK_PARENTHESIS_OPEN) {
  2692. _set_error("Expected '(' after if");
  2693. return ERR_PARSE_ERROR;
  2694. }
  2695. ControlFlowNode *cf = alloc_node<ControlFlowNode>();
  2696. cf->flow_op = FLOW_OP_WHILE;
  2697. Node *n = _parse_and_reduce_expression(p_block, p_builtin_types);
  2698. if (!n)
  2699. return ERR_PARSE_ERROR;
  2700. tk = _get_token();
  2701. if (tk.type != TK_PARENTHESIS_CLOSE) {
  2702. _set_error("Expected ')' after expression");
  2703. return ERR_PARSE_ERROR;
  2704. }
  2705. BlockNode *block = alloc_node<BlockNode>();
  2706. block->parent_block = p_block;
  2707. cf->expressions.push_back(n);
  2708. cf->blocks.push_back(block);
  2709. p_block->statements.push_back(cf);
  2710. Error err = _parse_block(block, p_builtin_types, true, p_can_break, p_can_continue);
  2711. if (err)
  2712. return err;
  2713. } else if (tk.type == TK_CF_RETURN) {
  2714. //check return type
  2715. BlockNode *b = p_block;
  2716. while (b && !b->parent_function) {
  2717. b = b->parent_block;
  2718. }
  2719. if (!b) {
  2720. _set_error("Bug");
  2721. return ERR_BUG;
  2722. }
  2723. ControlFlowNode *flow = alloc_node<ControlFlowNode>();
  2724. flow->flow_op = FLOW_OP_RETURN;
  2725. pos = _get_tkpos();
  2726. tk = _get_token();
  2727. if (tk.type == TK_SEMICOLON) {
  2728. //all is good
  2729. if (b->parent_function->return_type != TYPE_VOID) {
  2730. _set_error("Expected return with expression of type '" + get_datatype_name(b->parent_function->return_type) + "'");
  2731. return ERR_PARSE_ERROR;
  2732. }
  2733. } else {
  2734. _set_tkpos(pos); //rollback, wants expression
  2735. Node *expr = _parse_and_reduce_expression(p_block, p_builtin_types);
  2736. if (!expr)
  2737. return ERR_PARSE_ERROR;
  2738. if (b->parent_function->return_type != expr->get_datatype()) {
  2739. _set_error("Expected return expression of type '" + get_datatype_name(b->parent_function->return_type) + "'");
  2740. return ERR_PARSE_ERROR;
  2741. }
  2742. tk = _get_token();
  2743. if (tk.type != TK_SEMICOLON) {
  2744. _set_error("Expected ';' after return expression");
  2745. return ERR_PARSE_ERROR;
  2746. }
  2747. flow->expressions.push_back(expr);
  2748. }
  2749. p_block->statements.push_back(flow);
  2750. } else if (tk.type == TK_CF_DISCARD) {
  2751. //check return type
  2752. BlockNode *b = p_block;
  2753. while (b && !b->parent_function) {
  2754. b = b->parent_block;
  2755. }
  2756. if (!b) {
  2757. _set_error("Bug");
  2758. return ERR_BUG;
  2759. }
  2760. if (!b->parent_function->can_discard) {
  2761. _set_error("Use of 'discard' is not allowed here.");
  2762. return ERR_PARSE_ERROR;
  2763. }
  2764. ControlFlowNode *flow = alloc_node<ControlFlowNode>();
  2765. flow->flow_op = FLOW_OP_DISCARD;
  2766. pos = _get_tkpos();
  2767. tk = _get_token();
  2768. if (tk.type != TK_SEMICOLON) {
  2769. //all is good
  2770. _set_error("Expected ';' after discard");
  2771. }
  2772. p_block->statements.push_back(flow);
  2773. } else {
  2774. //nothng else, so expression
  2775. _set_tkpos(pos); //rollback
  2776. Node *expr = _parse_and_reduce_expression(p_block, p_builtin_types);
  2777. if (!expr)
  2778. return ERR_PARSE_ERROR;
  2779. p_block->statements.push_back(expr);
  2780. tk = _get_token();
  2781. if (tk.type != TK_SEMICOLON) {
  2782. _set_error("Expected ';' after statement");
  2783. return ERR_PARSE_ERROR;
  2784. }
  2785. }
  2786. if (p_just_one)
  2787. break;
  2788. }
  2789. return OK;
  2790. }
  2791. Error ShaderLanguage::_parse_shader(const Map<StringName, FunctionInfo> &p_functions, const Set<String> &p_render_modes, const Set<String> &p_shader_types) {
  2792. Token tk = _get_token();
  2793. if (tk.type != TK_SHADER_TYPE) {
  2794. _set_error("Expected 'shader_type' at the beginning of shader.");
  2795. return ERR_PARSE_ERROR;
  2796. }
  2797. tk = _get_token();
  2798. if (tk.type != TK_IDENTIFIER) {
  2799. _set_error("Expected identifier after 'shader_type', indicating type of shader.");
  2800. return ERR_PARSE_ERROR;
  2801. }
  2802. String shader_type_identifier;
  2803. shader_type_identifier = tk.text;
  2804. if (!p_shader_types.has(shader_type_identifier)) {
  2805. String valid;
  2806. for (Set<String>::Element *E = p_shader_types.front(); E; E = E->next()) {
  2807. if (valid != String()) {
  2808. valid += ", ";
  2809. }
  2810. valid += "'" + E->get() + "'";
  2811. }
  2812. _set_error("Invalid shader type, valid types are: " + valid);
  2813. return ERR_PARSE_ERROR;
  2814. }
  2815. tk = _get_token();
  2816. if (tk.type != TK_SEMICOLON) {
  2817. _set_error("Expected ';' after 'shader_type <type>'.");
  2818. }
  2819. tk = _get_token();
  2820. int texture_uniforms = 0;
  2821. int uniforms = 0;
  2822. while (tk.type != TK_EOF) {
  2823. switch (tk.type) {
  2824. case TK_RENDER_MODE: {
  2825. while (true) {
  2826. StringName mode;
  2827. _get_completable_identifier(NULL, COMPLETION_RENDER_MODE, mode);
  2828. if (mode == StringName()) {
  2829. _set_error("Expected identifier for render mode");
  2830. return ERR_PARSE_ERROR;
  2831. }
  2832. if (!p_render_modes.has(mode)) {
  2833. _set_error("Invalid render mode: '" + String(mode) + "'");
  2834. return ERR_PARSE_ERROR;
  2835. }
  2836. if (shader->render_modes.find(mode) != -1) {
  2837. _set_error("Duplicate render mode: '" + String(mode) + "'");
  2838. return ERR_PARSE_ERROR;
  2839. }
  2840. shader->render_modes.push_back(mode);
  2841. tk = _get_token();
  2842. if (tk.type == TK_COMMA) {
  2843. //all good, do nothing
  2844. } else if (tk.type == TK_SEMICOLON) {
  2845. break; //done
  2846. } else {
  2847. _set_error("Unexpected token: " + get_token_text(tk));
  2848. return ERR_PARSE_ERROR;
  2849. }
  2850. }
  2851. } break;
  2852. case TK_UNIFORM:
  2853. case TK_VARYING: {
  2854. bool uniform = tk.type == TK_UNIFORM;
  2855. DataPrecision precision = PRECISION_DEFAULT;
  2856. DataType type;
  2857. StringName name;
  2858. tk = _get_token();
  2859. if (is_token_precision(tk.type)) {
  2860. precision = get_token_precision(tk.type);
  2861. tk = _get_token();
  2862. }
  2863. if (!is_token_datatype(tk.type)) {
  2864. _set_error("Expected datatype. ");
  2865. return ERR_PARSE_ERROR;
  2866. }
  2867. type = get_token_datatype(tk.type);
  2868. if (type == TYPE_VOID) {
  2869. _set_error("void datatype not allowed here");
  2870. return ERR_PARSE_ERROR;
  2871. }
  2872. if (!uniform && type < TYPE_FLOAT && type > TYPE_VEC4) { // FIXME: always false! should it be || instead?
  2873. _set_error("Invalid type for varying, only float,vec2,vec3,vec4 allowed.");
  2874. return ERR_PARSE_ERROR;
  2875. }
  2876. tk = _get_token();
  2877. if (tk.type != TK_IDENTIFIER) {
  2878. _set_error("Expected identifier!");
  2879. return ERR_PARSE_ERROR;
  2880. }
  2881. name = tk.text;
  2882. if (_find_identifier(NULL, Map<StringName, DataType>(), name)) {
  2883. _set_error("Redefinition of '" + String(name) + "'");
  2884. return ERR_PARSE_ERROR;
  2885. }
  2886. if (uniform) {
  2887. ShaderNode::Uniform uniform;
  2888. if (is_sampler_type(type)) {
  2889. uniform.texture_order = texture_uniforms++;
  2890. uniform.order = -1;
  2891. } else {
  2892. uniform.texture_order = -1;
  2893. uniform.order = uniforms++;
  2894. }
  2895. uniform.type = type;
  2896. uniform.precission = precision;
  2897. //todo parse default value
  2898. tk = _get_token();
  2899. if (tk.type == TK_OP_ASSIGN) {
  2900. Node *expr = _parse_and_reduce_expression(NULL, Map<StringName, DataType>());
  2901. if (!expr)
  2902. return ERR_PARSE_ERROR;
  2903. if (expr->type != Node::TYPE_CONSTANT) {
  2904. _set_error("Expected constant expression after '='");
  2905. return ERR_PARSE_ERROR;
  2906. }
  2907. ConstantNode *cn = static_cast<ConstantNode *>(expr);
  2908. uniform.default_value.resize(cn->values.size());
  2909. if (!convert_constant(cn, uniform.type, uniform.default_value.ptr())) {
  2910. _set_error("Can't convert constant to " + get_datatype_name(uniform.type));
  2911. return ERR_PARSE_ERROR;
  2912. }
  2913. tk = _get_token();
  2914. }
  2915. if (tk.type == TK_COLON) {
  2916. //hint
  2917. tk = _get_token();
  2918. if (tk.type == TK_HINT_WHITE_TEXTURE) {
  2919. uniform.hint = ShaderNode::Uniform::HINT_WHITE;
  2920. } else if (tk.type == TK_HINT_BLACK_TEXTURE) {
  2921. uniform.hint = ShaderNode::Uniform::HINT_BLACK;
  2922. } else if (tk.type == TK_HINT_NORMAL_TEXTURE) {
  2923. uniform.hint = ShaderNode::Uniform::HINT_NORMAL;
  2924. } else if (tk.type == TK_HINT_ANISO_TEXTURE) {
  2925. uniform.hint = ShaderNode::Uniform::HINT_ANISO;
  2926. } else if (tk.type == TK_HINT_ALBEDO_TEXTURE) {
  2927. uniform.hint = ShaderNode::Uniform::HINT_ALBEDO;
  2928. } else if (tk.type == TK_HINT_BLACK_ALBEDO_TEXTURE) {
  2929. uniform.hint = ShaderNode::Uniform::HINT_BLACK_ALBEDO;
  2930. } else if (tk.type == TK_HINT_COLOR) {
  2931. if (type != TYPE_VEC4) {
  2932. _set_error("Color hint is for vec4 only");
  2933. return ERR_PARSE_ERROR;
  2934. }
  2935. uniform.hint = ShaderNode::Uniform::HINT_COLOR;
  2936. } else if (tk.type == TK_HINT_RANGE) {
  2937. uniform.hint = ShaderNode::Uniform::HINT_RANGE;
  2938. if (type != TYPE_FLOAT && type != TYPE_INT) {
  2939. _set_error("Range hint is for float and int only");
  2940. return ERR_PARSE_ERROR;
  2941. }
  2942. tk = _get_token();
  2943. if (tk.type != TK_PARENTHESIS_OPEN) {
  2944. _set_error("Expected '(' after hint_range");
  2945. return ERR_PARSE_ERROR;
  2946. }
  2947. tk = _get_token();
  2948. float sign = 1.0;
  2949. if (tk.type == TK_OP_SUB) {
  2950. sign = -1.0;
  2951. tk = _get_token();
  2952. }
  2953. if (tk.type != TK_REAL_CONSTANT && tk.type != TK_INT_CONSTANT) {
  2954. _set_error("Expected integer constant");
  2955. return ERR_PARSE_ERROR;
  2956. }
  2957. uniform.hint_range[0] = tk.constant;
  2958. uniform.hint_range[0] *= sign;
  2959. tk = _get_token();
  2960. if (tk.type != TK_COMMA) {
  2961. _set_error("Expected ',' after integer constant");
  2962. return ERR_PARSE_ERROR;
  2963. }
  2964. tk = _get_token();
  2965. sign = 1.0;
  2966. if (tk.type == TK_OP_SUB) {
  2967. sign = -1.0;
  2968. tk = _get_token();
  2969. }
  2970. if (tk.type != TK_REAL_CONSTANT && tk.type != TK_INT_CONSTANT) {
  2971. _set_error("Expected integer constant after ','");
  2972. return ERR_PARSE_ERROR;
  2973. }
  2974. uniform.hint_range[1] = tk.constant;
  2975. uniform.hint_range[1] *= sign;
  2976. tk = _get_token();
  2977. if (tk.type == TK_COMMA) {
  2978. tk = _get_token();
  2979. if (tk.type != TK_REAL_CONSTANT && tk.type != TK_INT_CONSTANT) {
  2980. _set_error("Expected integer constant after ','");
  2981. return ERR_PARSE_ERROR;
  2982. }
  2983. uniform.hint_range[2] = tk.constant;
  2984. tk = _get_token();
  2985. } else {
  2986. if (type == TYPE_INT) {
  2987. uniform.hint_range[2] = 1;
  2988. } else {
  2989. uniform.hint_range[2] = 0.001;
  2990. }
  2991. }
  2992. if (tk.type != TK_PARENTHESIS_CLOSE) {
  2993. _set_error("Expected ','");
  2994. return ERR_PARSE_ERROR;
  2995. }
  2996. } else {
  2997. _set_error("Expected valid type hint after ':'.");
  2998. }
  2999. if (uniform.hint != ShaderNode::Uniform::HINT_RANGE && uniform.hint != ShaderNode::Uniform::HINT_NONE && uniform.hint != ShaderNode::Uniform::HINT_COLOR && type <= TYPE_MAT4) {
  3000. _set_error("This hint is only for sampler types");
  3001. return ERR_PARSE_ERROR;
  3002. }
  3003. tk = _get_token();
  3004. }
  3005. shader->uniforms[name] = uniform;
  3006. if (tk.type != TK_SEMICOLON) {
  3007. _set_error("Expected ';'");
  3008. return ERR_PARSE_ERROR;
  3009. }
  3010. } else {
  3011. ShaderNode::Varying varying;
  3012. varying.type = type;
  3013. varying.precission = precision;
  3014. shader->varyings[name] = varying;
  3015. tk = _get_token();
  3016. if (tk.type != TK_SEMICOLON) {
  3017. _set_error("Expected ';'");
  3018. return ERR_PARSE_ERROR;
  3019. }
  3020. }
  3021. } break;
  3022. default: {
  3023. //function
  3024. DataPrecision precision = PRECISION_DEFAULT;
  3025. DataType type;
  3026. StringName name;
  3027. if (is_token_precision(tk.type)) {
  3028. precision = get_token_precision(tk.type);
  3029. tk = _get_token();
  3030. }
  3031. if (!is_token_datatype(tk.type)) {
  3032. _set_error("Expected function, uniform or varying ");
  3033. return ERR_PARSE_ERROR;
  3034. }
  3035. type = get_token_datatype(tk.type);
  3036. _get_completable_identifier(NULL, COMPLETION_MAIN_FUNCTION, name);
  3037. if (name == StringName()) {
  3038. _set_error("Expected function name after datatype");
  3039. return ERR_PARSE_ERROR;
  3040. }
  3041. if (_find_identifier(NULL, Map<StringName, DataType>(), name)) {
  3042. _set_error("Redefinition of '" + String(name) + "'");
  3043. return ERR_PARSE_ERROR;
  3044. }
  3045. tk = _get_token();
  3046. if (tk.type != TK_PARENTHESIS_OPEN) {
  3047. _set_error("Expected '(' after identifier");
  3048. return ERR_PARSE_ERROR;
  3049. }
  3050. Map<StringName, DataType> builtin_types;
  3051. if (p_functions.has(name)) {
  3052. builtin_types = p_functions[name].built_ins;
  3053. }
  3054. ShaderNode::Function function;
  3055. function.callable = !p_functions.has(name);
  3056. function.name = name;
  3057. FunctionNode *func_node = alloc_node<FunctionNode>();
  3058. function.function = func_node;
  3059. shader->functions.push_back(function);
  3060. func_node->name = name;
  3061. func_node->return_type = type;
  3062. func_node->return_precision = precision;
  3063. func_node->body = alloc_node<BlockNode>();
  3064. func_node->body->parent_function = func_node;
  3065. tk = _get_token();
  3066. while (true) {
  3067. if (tk.type == TK_PARENTHESIS_CLOSE) {
  3068. break;
  3069. }
  3070. ArgumentQualifier qualifier = ARGUMENT_QUALIFIER_IN;
  3071. if (tk.type == TK_ARG_IN) {
  3072. qualifier = ARGUMENT_QUALIFIER_IN;
  3073. tk = _get_token();
  3074. } else if (tk.type == TK_ARG_OUT) {
  3075. qualifier = ARGUMENT_QUALIFIER_OUT;
  3076. tk = _get_token();
  3077. } else if (tk.type == TK_ARG_INOUT) {
  3078. qualifier = ARGUMENT_QUALIFIER_INOUT;
  3079. tk = _get_token();
  3080. }
  3081. DataType ptype;
  3082. StringName pname;
  3083. DataPrecision pprecision = PRECISION_DEFAULT;
  3084. if (is_token_precision(tk.type)) {
  3085. pprecision = get_token_precision(tk.type);
  3086. tk = _get_token();
  3087. }
  3088. if (!is_token_datatype(tk.type)) {
  3089. _set_error("Expected a valid datatype for argument");
  3090. return ERR_PARSE_ERROR;
  3091. }
  3092. ptype = get_token_datatype(tk.type);
  3093. if (ptype == TYPE_VOID) {
  3094. _set_error("void not allowed in argument");
  3095. return ERR_PARSE_ERROR;
  3096. }
  3097. tk = _get_token();
  3098. if (tk.type != TK_IDENTIFIER) {
  3099. _set_error("Expected identifier for argument name");
  3100. return ERR_PARSE_ERROR;
  3101. }
  3102. pname = tk.text;
  3103. if (_find_identifier(func_node->body, builtin_types, pname)) {
  3104. _set_error("Redefinition of '" + String(pname) + "'");
  3105. return ERR_PARSE_ERROR;
  3106. }
  3107. FunctionNode::Argument arg;
  3108. arg.type = ptype;
  3109. arg.name = pname;
  3110. arg.precision = pprecision;
  3111. arg.qualifier = qualifier;
  3112. func_node->arguments.push_back(arg);
  3113. tk = _get_token();
  3114. if (tk.type == TK_COMMA) {
  3115. tk = _get_token();
  3116. //do none and go on
  3117. } else if (tk.type != TK_PARENTHESIS_CLOSE) {
  3118. _set_error("Expected ',' or ')' after identifier");
  3119. return ERR_PARSE_ERROR;
  3120. }
  3121. }
  3122. if (p_functions.has(name)) {
  3123. //if one of the core functions, make sure they are of the correct form
  3124. if (func_node->arguments.size() > 0) {
  3125. _set_error("Function '" + String(name) + "' expects no arguments.");
  3126. return ERR_PARSE_ERROR;
  3127. }
  3128. if (func_node->return_type != TYPE_VOID) {
  3129. _set_error("Function '" + String(name) + "' must be of void return type.");
  3130. return ERR_PARSE_ERROR;
  3131. }
  3132. }
  3133. //all good let's parse inside the function!
  3134. tk = _get_token();
  3135. if (tk.type != TK_CURLY_BRACKET_OPEN) {
  3136. _set_error("Expected '{' to begin function");
  3137. return ERR_PARSE_ERROR;
  3138. }
  3139. current_function = name;
  3140. Error err = _parse_block(func_node->body, builtin_types);
  3141. if (err)
  3142. return err;
  3143. current_function = StringName();
  3144. }
  3145. }
  3146. tk = _get_token();
  3147. }
  3148. return OK;
  3149. }
  3150. String ShaderLanguage::get_shader_type(const String &p_code) {
  3151. bool reading_type = false;
  3152. String cur_identifier;
  3153. for (int i = 0; i < p_code.length(); i++) {
  3154. if (p_code[i] == ';') {
  3155. break;
  3156. } else if (p_code[i] <= 32) {
  3157. if (cur_identifier != String()) {
  3158. if (!reading_type) {
  3159. if (cur_identifier != "shader_type") {
  3160. return String();
  3161. }
  3162. reading_type = true;
  3163. cur_identifier = String();
  3164. } else {
  3165. return cur_identifier;
  3166. }
  3167. }
  3168. } else {
  3169. cur_identifier += String::chr(p_code[i]);
  3170. }
  3171. }
  3172. if (reading_type)
  3173. return cur_identifier;
  3174. return String();
  3175. }
  3176. Error ShaderLanguage::compile(const String &p_code, const Map<StringName, FunctionInfo> &p_functions, const Set<String> &p_render_modes, const Set<String> &p_shader_types) {
  3177. clear();
  3178. code = p_code;
  3179. nodes = NULL;
  3180. shader = alloc_node<ShaderNode>();
  3181. Error err = _parse_shader(p_functions, p_render_modes, p_shader_types);
  3182. if (err != OK) {
  3183. return err;
  3184. }
  3185. return OK;
  3186. }
  3187. Error ShaderLanguage::complete(const String &p_code, const Map<StringName, FunctionInfo> &p_functions, const Set<String> &p_render_modes, const Set<String> &p_shader_types, List<String> *r_options, String &r_call_hint) {
  3188. clear();
  3189. code = p_code;
  3190. nodes = NULL;
  3191. shader = alloc_node<ShaderNode>();
  3192. Error err = _parse_shader(p_functions, p_render_modes, p_shader_types);
  3193. switch (completion_type) {
  3194. case COMPLETION_NONE: {
  3195. //do none
  3196. return ERR_PARSE_ERROR;
  3197. } break;
  3198. case COMPLETION_RENDER_MODE: {
  3199. for (const Set<String>::Element *E = p_render_modes.front(); E; E = E->next()) {
  3200. r_options->push_back(E->get());
  3201. }
  3202. return OK;
  3203. } break;
  3204. case COMPLETION_MAIN_FUNCTION: {
  3205. for (const Map<StringName, FunctionInfo>::Element *E = p_functions.front(); E; E = E->next()) {
  3206. r_options->push_back(E->key());
  3207. }
  3208. return OK;
  3209. } break;
  3210. case COMPLETION_IDENTIFIER:
  3211. case COMPLETION_FUNCTION_CALL: {
  3212. bool comp_ident = completion_type == COMPLETION_IDENTIFIER;
  3213. Set<String> matches;
  3214. StringName skip_function;
  3215. BlockNode *block = completion_block;
  3216. while (block) {
  3217. if (comp_ident) {
  3218. for (const Map<StringName, BlockNode::Variable>::Element *E = block->variables.front(); E; E = E->next()) {
  3219. if (E->get().line < completion_line) {
  3220. matches.insert(E->key());
  3221. }
  3222. }
  3223. }
  3224. if (block->parent_function) {
  3225. if (comp_ident) {
  3226. for (int i = 0; i < block->parent_function->arguments.size(); i++) {
  3227. matches.insert(block->parent_function->arguments[i].name);
  3228. }
  3229. }
  3230. skip_function = block->parent_function->name;
  3231. }
  3232. block = block->parent_block;
  3233. }
  3234. if (comp_ident && skip_function != StringName() && p_functions.has(skip_function)) {
  3235. for (Map<StringName, DataType>::Element *E = p_functions[skip_function].built_ins.front(); E; E = E->next()) {
  3236. matches.insert(E->key());
  3237. }
  3238. }
  3239. if (comp_ident) {
  3240. for (const Map<StringName, ShaderNode::Varying>::Element *E = shader->varyings.front(); E; E = E->next()) {
  3241. matches.insert(E->key());
  3242. }
  3243. for (const Map<StringName, ShaderNode::Uniform>::Element *E = shader->uniforms.front(); E; E = E->next()) {
  3244. matches.insert(E->key());
  3245. }
  3246. }
  3247. for (int i = 0; i < shader->functions.size(); i++) {
  3248. if (!shader->functions[i].callable || shader->functions[i].name == skip_function)
  3249. continue;
  3250. matches.insert(String(shader->functions[i].name) + "(");
  3251. }
  3252. int idx = 0;
  3253. while (builtin_func_defs[idx].name) {
  3254. matches.insert(String(builtin_func_defs[idx].name) + "(");
  3255. idx++;
  3256. }
  3257. for (Set<String>::Element *E = matches.front(); E; E = E->next()) {
  3258. r_options->push_back(E->get());
  3259. }
  3260. return OK;
  3261. } break;
  3262. case COMPLETION_CALL_ARGUMENTS: {
  3263. for (int i = 0; i < shader->functions.size(); i++) {
  3264. if (!shader->functions[i].callable)
  3265. continue;
  3266. if (shader->functions[i].name == completion_function) {
  3267. String calltip;
  3268. calltip += get_datatype_name(shader->functions[i].function->return_type);
  3269. calltip += " ";
  3270. calltip += shader->functions[i].name;
  3271. calltip += "(";
  3272. for (int j = 0; j < shader->functions[i].function->arguments.size(); j++) {
  3273. if (j > 0)
  3274. calltip += ", ";
  3275. else
  3276. calltip += " ";
  3277. if (j == completion_argument) {
  3278. calltip += CharType(0xFFFF);
  3279. }
  3280. calltip += get_datatype_name(shader->functions[i].function->arguments[j].type);
  3281. calltip += " ";
  3282. calltip += shader->functions[i].function->arguments[j].name;
  3283. if (j == completion_argument) {
  3284. calltip += CharType(0xFFFF);
  3285. }
  3286. }
  3287. if (shader->functions[i].function->arguments.size())
  3288. calltip += " ";
  3289. calltip += ")";
  3290. r_call_hint = calltip;
  3291. return OK;
  3292. }
  3293. }
  3294. int idx = 0;
  3295. String calltip;
  3296. while (builtin_func_defs[idx].name) {
  3297. if (completion_function == builtin_func_defs[idx].name) {
  3298. if (calltip.length())
  3299. calltip += "\n";
  3300. calltip += get_datatype_name(builtin_func_defs[idx].rettype);
  3301. calltip += " ";
  3302. calltip += builtin_func_defs[idx].name;
  3303. calltip += "(";
  3304. bool found_arg = false;
  3305. for (int i = 0; i < 4; i++) {
  3306. if (builtin_func_defs[idx].args[i] == TYPE_VOID)
  3307. break;
  3308. if (i > 0)
  3309. calltip += ", ";
  3310. else
  3311. calltip += " ";
  3312. if (i == completion_argument) {
  3313. calltip += CharType(0xFFFF);
  3314. }
  3315. calltip += get_datatype_name(builtin_func_defs[idx].args[i]);
  3316. if (i == completion_argument) {
  3317. calltip += CharType(0xFFFF);
  3318. }
  3319. found_arg = true;
  3320. }
  3321. if (found_arg)
  3322. calltip += " ";
  3323. calltip += ")";
  3324. }
  3325. idx++;
  3326. }
  3327. r_call_hint = calltip;
  3328. return OK;
  3329. } break;
  3330. case COMPLETION_INDEX: {
  3331. const char colv[4] = { 'r', 'g', 'b', 'a' };
  3332. const char coordv[4] = { 'x', 'y', 'z', 'w' };
  3333. int limit = 0;
  3334. switch (completion_base) {
  3335. case TYPE_BVEC2:
  3336. case TYPE_IVEC2:
  3337. case TYPE_UVEC2:
  3338. case TYPE_VEC2: {
  3339. limit = 2;
  3340. } break;
  3341. case TYPE_BVEC3:
  3342. case TYPE_IVEC3:
  3343. case TYPE_UVEC3:
  3344. case TYPE_VEC3: {
  3345. limit = 3;
  3346. } break;
  3347. case TYPE_BVEC4:
  3348. case TYPE_IVEC4:
  3349. case TYPE_UVEC4:
  3350. case TYPE_VEC4: {
  3351. limit = 4;
  3352. } break;
  3353. case TYPE_MAT2: limit = 2; break;
  3354. case TYPE_MAT3: limit = 3; break;
  3355. case TYPE_MAT4: limit = 4; break;
  3356. default: {}
  3357. }
  3358. for (int i = 0; i < limit; i++) {
  3359. r_options->push_back(String::chr(colv[i]));
  3360. r_options->push_back(String::chr(coordv[i]));
  3361. }
  3362. } break;
  3363. }
  3364. return ERR_PARSE_ERROR;
  3365. }
  3366. String ShaderLanguage::get_error_text() {
  3367. return error_str;
  3368. }
  3369. int ShaderLanguage::get_error_line() {
  3370. return error_line;
  3371. }
  3372. ShaderLanguage::ShaderNode *ShaderLanguage::get_shader() {
  3373. return shader;
  3374. }
  3375. ShaderLanguage::ShaderLanguage() {
  3376. nodes = NULL;
  3377. }
  3378. ShaderLanguage::~ShaderLanguage() {
  3379. clear();
  3380. }