shader_language.cpp 122 KB

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