visual_shader_editor_plugin.cpp 178 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375
  1. /*************************************************************************/
  2. /* visual_shader_editor_plugin.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2020 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2020 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 "visual_shader_editor_plugin.h"
  31. #include "core/input/input.h"
  32. #include "core/io/resource_loader.h"
  33. #include "core/math/math_defs.h"
  34. #include "core/os/keyboard.h"
  35. #include "core/project_settings.h"
  36. #include "core/version.h"
  37. #include "editor/editor_log.h"
  38. #include "editor/editor_properties.h"
  39. #include "editor/editor_scale.h"
  40. #include "scene/animation/animation_player.h"
  41. #include "scene/gui/menu_button.h"
  42. #include "scene/gui/panel.h"
  43. #include "scene/main/window.h"
  44. #include "scene/resources/visual_shader_nodes.h"
  45. #include "servers/display_server.h"
  46. #include "servers/rendering/shader_types.h"
  47. Control *VisualShaderNodePlugin::create_editor(const Ref<Resource> &p_parent_resource, const Ref<VisualShaderNode> &p_node) {
  48. if (get_script_instance()) {
  49. return get_script_instance()->call("create_editor", p_parent_resource, p_node);
  50. }
  51. return nullptr;
  52. }
  53. void VisualShaderNodePlugin::_bind_methods() {
  54. BIND_VMETHOD(MethodInfo(Variant::OBJECT, "create_editor", PropertyInfo(Variant::OBJECT, "parent_resource", PROPERTY_HINT_RESOURCE_TYPE, "Resource"), PropertyInfo(Variant::OBJECT, "for_node", PROPERTY_HINT_RESOURCE_TYPE, "VisualShaderNode")));
  55. }
  56. ///////////////////
  57. void VisualShaderEditor::edit(VisualShader *p_visual_shader) {
  58. bool changed = false;
  59. if (p_visual_shader) {
  60. if (visual_shader.is_null()) {
  61. changed = true;
  62. } else {
  63. if (visual_shader.ptr() != p_visual_shader) {
  64. changed = true;
  65. }
  66. }
  67. visual_shader = Ref<VisualShader>(p_visual_shader);
  68. if (!visual_shader->is_connected("changed", callable_mp(this, &VisualShaderEditor::_update_preview))) {
  69. visual_shader->connect("changed", callable_mp(this, &VisualShaderEditor::_update_preview));
  70. }
  71. #ifndef DISABLE_DEPRECATED
  72. String version = VERSION_BRANCH;
  73. if (visual_shader->get_version() != version) {
  74. visual_shader->update_version(version);
  75. }
  76. #endif
  77. visual_shader->set_graph_offset(graph->get_scroll_ofs() / EDSCALE);
  78. } else {
  79. if (visual_shader.is_valid()) {
  80. if (visual_shader->is_connected("changed", callable_mp(this, &VisualShaderEditor::_update_preview))) {
  81. visual_shader->disconnect("changed", callable_mp(this, &VisualShaderEditor::_update_preview));
  82. }
  83. }
  84. visual_shader.unref();
  85. }
  86. if (visual_shader.is_null()) {
  87. hide();
  88. } else {
  89. if (changed) { // to avoid tree collapse
  90. _clear_buffer();
  91. _update_options_menu();
  92. _update_preview();
  93. }
  94. _update_graph();
  95. }
  96. }
  97. void VisualShaderEditor::add_plugin(const Ref<VisualShaderNodePlugin> &p_plugin) {
  98. if (plugins.find(p_plugin) != -1) {
  99. return;
  100. }
  101. plugins.push_back(p_plugin);
  102. }
  103. void VisualShaderEditor::remove_plugin(const Ref<VisualShaderNodePlugin> &p_plugin) {
  104. plugins.erase(p_plugin);
  105. }
  106. void VisualShaderEditor::clear_custom_types() {
  107. for (int i = 0; i < add_options.size(); i++) {
  108. if (add_options[i].is_custom) {
  109. add_options.remove(i);
  110. i--;
  111. }
  112. }
  113. }
  114. void VisualShaderEditor::add_custom_type(const String &p_name, const Ref<Script> &p_script, const String &p_description, int p_return_icon_type, const String &p_category, bool p_highend) {
  115. ERR_FAIL_COND(!p_name.is_valid_identifier());
  116. ERR_FAIL_COND(!p_script.is_valid());
  117. for (int i = 0; i < add_options.size(); i++) {
  118. if (add_options[i].is_custom) {
  119. if (add_options[i].script == p_script) {
  120. return;
  121. }
  122. }
  123. }
  124. AddOption ao;
  125. ao.name = p_name;
  126. ao.script = p_script;
  127. ao.return_type = p_return_icon_type;
  128. ao.description = p_description;
  129. ao.category = p_category;
  130. ao.highend = p_highend;
  131. ao.is_custom = true;
  132. bool begin = false;
  133. String root = p_category.split("/")[0];
  134. for (int i = 0; i < add_options.size(); i++) {
  135. if (add_options[i].is_custom) {
  136. if (add_options[i].category == root) {
  137. if (!begin) {
  138. begin = true;
  139. }
  140. } else {
  141. if (begin) {
  142. add_options.insert(i, ao);
  143. return;
  144. }
  145. }
  146. }
  147. }
  148. add_options.push_back(ao);
  149. }
  150. bool VisualShaderEditor::_is_available(int p_mode) {
  151. int current_mode = edit_type->get_selected();
  152. if (p_mode != -1) {
  153. switch (current_mode) {
  154. case VisualShader::TYPE_VERTEX:
  155. current_mode = 1;
  156. break;
  157. case VisualShader::TYPE_FRAGMENT:
  158. current_mode = 2;
  159. break;
  160. case VisualShader::TYPE_LIGHT:
  161. current_mode = 4;
  162. break;
  163. default:
  164. break;
  165. }
  166. int temp_mode = 0;
  167. if (p_mode & VisualShader::TYPE_FRAGMENT) {
  168. temp_mode |= 2;
  169. }
  170. if (p_mode & VisualShader::TYPE_LIGHT) {
  171. temp_mode |= 4;
  172. }
  173. if (temp_mode == 0) {
  174. temp_mode |= 1;
  175. }
  176. p_mode = temp_mode;
  177. }
  178. return (p_mode == -1 || (p_mode & current_mode) != 0);
  179. }
  180. void VisualShaderEditor::update_custom_nodes() {
  181. if (members_dialog->is_visible()) {
  182. return;
  183. }
  184. clear_custom_types();
  185. List<StringName> class_list;
  186. ScriptServer::get_global_class_list(&class_list);
  187. Dictionary added;
  188. for (int i = 0; i < class_list.size(); i++) {
  189. if (ScriptServer::get_global_class_native_base(class_list[i]) == "VisualShaderNodeCustom") {
  190. String script_path = ScriptServer::get_global_class_path(class_list[i]);
  191. Ref<Resource> res = ResourceLoader::load(script_path);
  192. ERR_FAIL_COND(res.is_null());
  193. ERR_FAIL_COND(!res->is_class("Script"));
  194. Ref<Script> script = Ref<Script>(res);
  195. Ref<VisualShaderNodeCustom> ref;
  196. ref.instance();
  197. ref->set_script(script);
  198. String name;
  199. if (ref->has_method("_get_name")) {
  200. name = (String)ref->call("_get_name");
  201. } else {
  202. name = "Unnamed";
  203. }
  204. String description = "";
  205. if (ref->has_method("_get_description")) {
  206. description = (String)ref->call("_get_description");
  207. }
  208. int return_icon_type = -1;
  209. if (ref->has_method("_get_return_icon_type")) {
  210. return_icon_type = (int)ref->call("_get_return_icon_type");
  211. }
  212. String category = "";
  213. if (ref->has_method("_get_category")) {
  214. category = (String)ref->call("_get_category");
  215. }
  216. String subcategory = "";
  217. if (ref->has_method("_get_subcategory")) {
  218. subcategory = (String)ref->call("_get_subcategory");
  219. }
  220. bool highend = false;
  221. if (ref->has_method("_is_highend")) {
  222. highend = (bool)ref->call("_is_highend");
  223. }
  224. Dictionary dict;
  225. dict["name"] = name;
  226. dict["script"] = script;
  227. dict["description"] = description;
  228. dict["return_icon_type"] = return_icon_type;
  229. category = category.rstrip("/");
  230. category = category.lstrip("/");
  231. category = "Addons/" + category;
  232. if (subcategory != "") {
  233. category += "/" + subcategory;
  234. }
  235. dict["category"] = category;
  236. dict["highend"] = highend;
  237. String key;
  238. key = category + "/" + name;
  239. added[key] = dict;
  240. }
  241. }
  242. Array keys = added.keys();
  243. keys.sort();
  244. for (int i = 0; i < keys.size(); i++) {
  245. const Variant &key = keys.get(i);
  246. const Dictionary &value = (Dictionary)added[key];
  247. add_custom_type(value["name"], value["script"], value["description"], value["return_icon_type"], value["category"], value["highend"]);
  248. }
  249. _update_options_menu();
  250. }
  251. String VisualShaderEditor::_get_description(int p_idx) {
  252. return add_options[p_idx].description;
  253. }
  254. void VisualShaderEditor::_update_options_menu() {
  255. node_desc->set_text("");
  256. members_dialog->get_ok()->set_disabled(true);
  257. members->clear();
  258. TreeItem *root = members->create_item();
  259. String filter = node_filter->get_text().strip_edges();
  260. bool use_filter = !filter.empty();
  261. bool is_first_item = true;
  262. Color unsupported_color = get_theme_color("error_color", "Editor");
  263. Color supported_color = get_theme_color("warning_color", "Editor");
  264. static bool low_driver = ProjectSettings::get_singleton()->get("rendering/quality/driver/driver_name") == "GLES2";
  265. Map<String, TreeItem *> folders;
  266. int current_func = -1;
  267. if (!visual_shader.is_null()) {
  268. current_func = visual_shader->get_mode();
  269. }
  270. Vector<AddOption> custom_options;
  271. Vector<AddOption> embedded_options;
  272. for (int i = 0; i < add_options.size(); i++) {
  273. if (!use_filter || add_options[i].name.findn(filter) != -1) {
  274. if ((add_options[i].func != current_func && add_options[i].func != -1) || !_is_available(add_options[i].mode)) {
  275. continue;
  276. }
  277. const_cast<AddOption &>(add_options[i]).temp_idx = i; // save valid id
  278. if (add_options[i].is_custom) {
  279. custom_options.push_back(add_options[i]);
  280. } else {
  281. embedded_options.push_back(add_options[i]);
  282. }
  283. }
  284. }
  285. Vector<AddOption> options;
  286. SortArray<AddOption, _OptionComparator> sorter;
  287. sorter.sort(custom_options.ptrw(), custom_options.size());
  288. options.append_array(custom_options);
  289. options.append_array(embedded_options);
  290. for (int i = 0; i < options.size(); i++) {
  291. String path = options[i].category;
  292. Vector<String> subfolders = path.split("/");
  293. TreeItem *category = nullptr;
  294. if (!folders.has(path)) {
  295. category = root;
  296. String path_temp = "";
  297. for (int j = 0; j < subfolders.size(); j++) {
  298. path_temp += subfolders[j];
  299. if (!folders.has(path_temp)) {
  300. category = members->create_item(category);
  301. category->set_selectable(0, false);
  302. category->set_collapsed(!use_filter);
  303. category->set_text(0, subfolders[j]);
  304. folders.insert(path_temp, category);
  305. } else {
  306. category = folders[path_temp];
  307. }
  308. }
  309. } else {
  310. category = folders[path];
  311. }
  312. TreeItem *item = members->create_item(category);
  313. if (options[i].highend && low_driver) {
  314. item->set_custom_color(0, unsupported_color);
  315. } else if (options[i].highend) {
  316. item->set_custom_color(0, supported_color);
  317. }
  318. item->set_text(0, options[i].name);
  319. if (is_first_item && use_filter) {
  320. item->select(0);
  321. node_desc->set_text(options[i].description);
  322. is_first_item = false;
  323. }
  324. switch (options[i].return_type) {
  325. case VisualShaderNode::PORT_TYPE_SCALAR:
  326. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("float", "EditorIcons"));
  327. break;
  328. case VisualShaderNode::PORT_TYPE_SCALAR_INT:
  329. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("int", "EditorIcons"));
  330. break;
  331. case VisualShaderNode::PORT_TYPE_VECTOR:
  332. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Vector3", "EditorIcons"));
  333. break;
  334. case VisualShaderNode::PORT_TYPE_BOOLEAN:
  335. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("bool", "EditorIcons"));
  336. break;
  337. case VisualShaderNode::PORT_TYPE_TRANSFORM:
  338. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Transform", "EditorIcons"));
  339. break;
  340. case VisualShaderNode::PORT_TYPE_SAMPLER:
  341. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("ImageTexture", "EditorIcons"));
  342. break;
  343. default:
  344. break;
  345. }
  346. item->set_meta("id", options[i].temp_idx);
  347. }
  348. }
  349. Size2 VisualShaderEditor::get_minimum_size() const {
  350. return Size2(10, 200);
  351. }
  352. void VisualShaderEditor::_draw_color_over_button(Object *obj, Color p_color) {
  353. Button *button = Object::cast_to<Button>(obj);
  354. if (!button) {
  355. return;
  356. }
  357. Ref<StyleBox> normal = get_theme_stylebox("normal", "Button");
  358. button->draw_rect(Rect2(normal->get_offset(), button->get_size() - normal->get_minimum_size()), p_color);
  359. }
  360. static Ref<StyleBoxEmpty> make_empty_stylebox(float p_margin_left = -1, float p_margin_top = -1, float p_margin_right = -1, float p_margin_bottom = -1) {
  361. Ref<StyleBoxEmpty> style(memnew(StyleBoxEmpty));
  362. style->set_default_margin(MARGIN_LEFT, p_margin_left * EDSCALE);
  363. style->set_default_margin(MARGIN_RIGHT, p_margin_right * EDSCALE);
  364. style->set_default_margin(MARGIN_BOTTOM, p_margin_bottom * EDSCALE);
  365. style->set_default_margin(MARGIN_TOP, p_margin_top * EDSCALE);
  366. return style;
  367. }
  368. void VisualShaderEditor::_update_created_node(GraphNode *node) {
  369. if (EditorSettings::get_singleton()->get("interface/theme/use_graph_node_headers")) {
  370. Ref<StyleBoxFlat> sb = node->get_theme_stylebox("frame", "GraphNode");
  371. Color c = sb->get_border_color();
  372. Color ic;
  373. Color mono_color;
  374. if (((c.r + c.g + c.b) / 3) < 0.7) {
  375. mono_color = Color(1.0, 1.0, 1.0);
  376. ic = Color(0.0, 0.0, 0.0, 0.7);
  377. } else {
  378. mono_color = Color(0.0, 0.0, 0.0);
  379. ic = Color(1.0, 1.0, 1.0, 0.7);
  380. }
  381. mono_color.a = 0.85;
  382. c = mono_color;
  383. node->add_theme_color_override("title_color", c);
  384. c.a = 0.7;
  385. node->add_theme_color_override("close_color", c);
  386. node->add_theme_color_override("resizer_color", ic);
  387. }
  388. }
  389. void VisualShaderEditor::_update_graph() {
  390. if (updating) {
  391. return;
  392. }
  393. if (visual_shader.is_null()) {
  394. return;
  395. }
  396. graph->set_scroll_ofs(visual_shader->get_graph_offset() * EDSCALE);
  397. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  398. graph->clear_connections();
  399. //erase all nodes
  400. for (int i = 0; i < graph->get_child_count(); i++) {
  401. if (Object::cast_to<GraphNode>(graph->get_child(i))) {
  402. Node *node = graph->get_child(i);
  403. graph->remove_child(node);
  404. memdelete(node);
  405. i--;
  406. }
  407. }
  408. static const Color type_color[6] = {
  409. Color(0.38, 0.85, 0.96), // scalar (float)
  410. Color(0.49, 0.78, 0.94), // scalar (int)
  411. Color(0.84, 0.49, 0.93), // vector
  412. Color(0.55, 0.65, 0.94), // boolean
  413. Color(0.96, 0.66, 0.43), // transform
  414. Color(1.0, 1.0, 0.0), // sampler
  415. };
  416. List<VisualShader::Connection> connections;
  417. visual_shader->get_node_connections(type, &connections);
  418. Ref<StyleBoxEmpty> label_style = make_empty_stylebox(2, 1, 2, 1);
  419. Vector<int> nodes = visual_shader->get_node_list(type);
  420. Control *offset;
  421. for (int n_i = 0; n_i < nodes.size(); n_i++) {
  422. Vector2 position = visual_shader->get_node_position(type, nodes[n_i]);
  423. Ref<VisualShaderNode> vsnode = visual_shader->get_node(type, nodes[n_i]);
  424. Ref<VisualShaderNodeGroupBase> group_node = Object::cast_to<VisualShaderNodeGroupBase>(vsnode.ptr());
  425. bool is_group = !group_node.is_null();
  426. Size2 size = Size2(0, 0);
  427. Ref<VisualShaderNodeExpression> expression_node = Object::cast_to<VisualShaderNodeExpression>(group_node.ptr());
  428. bool is_expression = !expression_node.is_null();
  429. String expression = "";
  430. GraphNode *node = memnew(GraphNode);
  431. if (is_group) {
  432. size = group_node->get_size();
  433. node->set_resizable(true);
  434. node->connect("resize_request", callable_mp(this, &VisualShaderEditor::_node_resized), varray((int)type, nodes[n_i]));
  435. }
  436. if (is_expression) {
  437. expression = expression_node->get_expression();
  438. }
  439. node->set_offset(position);
  440. node->set_title(vsnode->get_caption());
  441. node->set_name(itos(nodes[n_i]));
  442. if (nodes[n_i] >= 2) {
  443. node->set_show_close_button(true);
  444. node->connect("close_request", callable_mp(this, &VisualShaderEditor::_delete_request), varray(nodes[n_i]), CONNECT_DEFERRED);
  445. }
  446. node->connect("dragged", callable_mp(this, &VisualShaderEditor::_node_dragged), varray(nodes[n_i]));
  447. Control *custom_editor = nullptr;
  448. int port_offset = 0;
  449. if (is_group) {
  450. port_offset += 2;
  451. }
  452. Ref<VisualShaderNodeUniform> uniform = vsnode;
  453. Ref<VisualShaderNodeFloatUniform> float_uniform = vsnode;
  454. Ref<VisualShaderNodeIntUniform> int_uniform = vsnode;
  455. Ref<VisualShaderNodeVec3Uniform> vec3_uniform = vsnode;
  456. Ref<VisualShaderNodeColorUniform> color_uniform = vsnode;
  457. Ref<VisualShaderNodeBooleanUniform> bool_uniform = vsnode;
  458. Ref<VisualShaderNodeTransformUniform> transform_uniform = vsnode;
  459. if (uniform.is_valid()) {
  460. graph->add_child(node);
  461. _update_created_node(node);
  462. LineEdit *uniform_name = memnew(LineEdit);
  463. uniform_name->set_text(uniform->get_uniform_name());
  464. node->add_child(uniform_name);
  465. uniform_name->connect("text_entered", callable_mp(this, &VisualShaderEditor::_line_edit_changed), varray(uniform_name, nodes[n_i]));
  466. uniform_name->connect("focus_exited", callable_mp(this, &VisualShaderEditor::_line_edit_focus_out), varray(uniform_name, nodes[n_i]));
  467. String error = vsnode->get_warning(visual_shader->get_mode(), type);
  468. if (error != String()) {
  469. offset = memnew(Control);
  470. offset->set_custom_minimum_size(Size2(0, 4 * EDSCALE));
  471. node->add_child(offset);
  472. Label *error_label = memnew(Label);
  473. error_label->add_theme_color_override("font_color", get_theme_color("error_color", "Editor"));
  474. error_label->set_text(error);
  475. node->add_child(error_label);
  476. }
  477. if (vsnode->get_input_port_count() == 0 && vsnode->get_output_port_count() == 1 && vsnode->get_output_port_name(0) == "") {
  478. //shortcut
  479. VisualShaderNode::PortType port_right = vsnode->get_output_port_type(0);
  480. node->set_slot(0, false, VisualShaderNode::PORT_TYPE_SCALAR, Color(), true, port_right, type_color[port_right]);
  481. if (!float_uniform.is_valid() && !int_uniform.is_valid() && !vec3_uniform.is_valid() && !color_uniform.is_valid() && !bool_uniform.is_valid() && !transform_uniform.is_valid()) {
  482. continue;
  483. }
  484. }
  485. port_offset++;
  486. }
  487. for (int i = 0; i < plugins.size(); i++) {
  488. custom_editor = plugins.write[i]->create_editor(visual_shader, vsnode);
  489. if (custom_editor) {
  490. break;
  491. }
  492. }
  493. if (custom_editor && !float_uniform.is_valid() && !int_uniform.is_valid() && !vec3_uniform.is_valid() && !bool_uniform.is_valid() && !transform_uniform.is_valid() && vsnode->get_output_port_count() > 0 && vsnode->get_output_port_name(0) == "" && (vsnode->get_input_port_count() == 0 || vsnode->get_input_port_name(0) == "")) {
  494. //will be embedded in first port
  495. } else if (custom_editor) {
  496. port_offset++;
  497. node->add_child(custom_editor);
  498. if (color_uniform.is_valid()) {
  499. custom_editor->call_deferred("_show_prop_names", true);
  500. }
  501. if (float_uniform.is_valid() || int_uniform.is_valid() || vec3_uniform.is_valid() || bool_uniform.is_valid() || transform_uniform.is_valid()) {
  502. custom_editor->call_deferred("_show_prop_names", true);
  503. continue;
  504. }
  505. custom_editor = nullptr;
  506. }
  507. if (is_group) {
  508. offset = memnew(Control);
  509. offset->set_custom_minimum_size(Size2(0, 6 * EDSCALE));
  510. node->add_child(offset);
  511. if (group_node->is_editable()) {
  512. HBoxContainer *hb2 = memnew(HBoxContainer);
  513. Button *add_input_btn = memnew(Button);
  514. add_input_btn->set_text(TTR("Add Input"));
  515. add_input_btn->connect("pressed", callable_mp(this, &VisualShaderEditor::_add_input_port), varray(nodes[n_i], group_node->get_free_input_port_id(), VisualShaderNode::PORT_TYPE_VECTOR, "input" + itos(group_node->get_free_input_port_id())), CONNECT_DEFERRED);
  516. hb2->add_child(add_input_btn);
  517. hb2->add_spacer();
  518. Button *add_output_btn = memnew(Button);
  519. add_output_btn->set_text(TTR("Add Output"));
  520. add_output_btn->connect("pressed", callable_mp(this, &VisualShaderEditor::_add_output_port), varray(nodes[n_i], group_node->get_free_output_port_id(), VisualShaderNode::PORT_TYPE_VECTOR, "output" + itos(group_node->get_free_output_port_id())), CONNECT_DEFERRED);
  521. hb2->add_child(add_output_btn);
  522. node->add_child(hb2);
  523. }
  524. }
  525. for (int i = 0; i < MAX(vsnode->get_input_port_count(), vsnode->get_output_port_count()); i++) {
  526. if (vsnode->is_port_separator(i)) {
  527. node->add_child(memnew(HSeparator));
  528. port_offset++;
  529. }
  530. bool valid_left = i < vsnode->get_input_port_count();
  531. VisualShaderNode::PortType port_left = VisualShaderNode::PORT_TYPE_SCALAR;
  532. bool port_left_used = false;
  533. String name_left;
  534. if (valid_left) {
  535. name_left = vsnode->get_input_port_name(i);
  536. port_left = vsnode->get_input_port_type(i);
  537. for (List<VisualShader::Connection>::Element *E = connections.front(); E; E = E->next()) {
  538. if (E->get().to_node == nodes[n_i] && E->get().to_port == i) {
  539. port_left_used = true;
  540. }
  541. }
  542. }
  543. bool valid_right = i < vsnode->get_output_port_count();
  544. VisualShaderNode::PortType port_right = VisualShaderNode::PORT_TYPE_SCALAR;
  545. String name_right;
  546. if (valid_right) {
  547. name_right = vsnode->get_output_port_name(i);
  548. port_right = vsnode->get_output_port_type(i);
  549. }
  550. HBoxContainer *hb = memnew(HBoxContainer);
  551. hb->add_theme_constant_override("separation", 7 * EDSCALE);
  552. Variant default_value;
  553. if (valid_left && !port_left_used) {
  554. default_value = vsnode->get_input_port_default_value(i);
  555. }
  556. if (default_value.get_type() != Variant::NIL) { // only a label
  557. Button *button = memnew(Button);
  558. hb->add_child(button);
  559. button->connect("pressed", callable_mp(this, &VisualShaderEditor::_edit_port_default_input), varray(button, nodes[n_i], i));
  560. switch (default_value.get_type()) {
  561. case Variant::COLOR: {
  562. button->set_custom_minimum_size(Size2(30, 0) * EDSCALE);
  563. button->connect("draw", callable_mp(this, &VisualShaderEditor::_draw_color_over_button), varray(button, default_value));
  564. } break;
  565. case Variant::BOOL: {
  566. button->set_text(((bool)default_value) ? "true" : "false");
  567. } break;
  568. case Variant::INT:
  569. case Variant::FLOAT: {
  570. button->set_text(String::num(default_value, 4));
  571. } break;
  572. case Variant::VECTOR3: {
  573. Vector3 v = default_value;
  574. button->set_text(String::num(v.x, 3) + "," + String::num(v.y, 3) + "," + String::num(v.z, 3));
  575. } break;
  576. default: {
  577. }
  578. }
  579. }
  580. if (i == 0 && custom_editor) {
  581. hb->add_child(custom_editor);
  582. custom_editor->set_h_size_flags(SIZE_EXPAND_FILL);
  583. } else {
  584. if (valid_left) {
  585. if (is_group) {
  586. OptionButton *type_box = memnew(OptionButton);
  587. hb->add_child(type_box);
  588. type_box->add_item(TTR("Float"));
  589. type_box->add_item(TTR("Int"));
  590. type_box->add_item(TTR("Vector"));
  591. type_box->add_item(TTR("Boolean"));
  592. type_box->add_item(TTR("Transform"));
  593. type_box->add_item(TTR("Sampler"));
  594. type_box->select(group_node->get_input_port_type(i));
  595. type_box->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  596. type_box->connect("item_selected", callable_mp(this, &VisualShaderEditor::_change_input_port_type), varray(nodes[n_i], i), CONNECT_DEFERRED);
  597. LineEdit *name_box = memnew(LineEdit);
  598. hb->add_child(name_box);
  599. name_box->set_custom_minimum_size(Size2(65 * EDSCALE, 0));
  600. name_box->set_h_size_flags(SIZE_EXPAND_FILL);
  601. name_box->set_text(name_left);
  602. name_box->connect("text_entered", callable_mp(this, &VisualShaderEditor::_change_input_port_name), varray(name_box, nodes[n_i], i));
  603. name_box->connect("focus_exited", callable_mp(this, &VisualShaderEditor::_port_name_focus_out), varray(name_box, nodes[n_i], i, false));
  604. Button *remove_btn = memnew(Button);
  605. remove_btn->set_icon(EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Remove", "EditorIcons"));
  606. remove_btn->set_tooltip(TTR("Remove") + " " + name_left);
  607. remove_btn->connect("pressed", callable_mp(this, &VisualShaderEditor::_remove_input_port), varray(nodes[n_i], i), CONNECT_DEFERRED);
  608. hb->add_child(remove_btn);
  609. } else {
  610. Label *label = memnew(Label);
  611. label->set_text(name_left);
  612. label->add_theme_style_override("normal", label_style); //more compact
  613. hb->add_child(label);
  614. if (vsnode->get_input_port_default_hint(i) != "" && !port_left_used) {
  615. Label *hint_label = memnew(Label);
  616. hint_label->set_text("[" + vsnode->get_input_port_default_hint(i) + "]");
  617. hint_label->add_theme_color_override("font_color", get_theme_color("font_color_readonly", "TextEdit"));
  618. hint_label->add_theme_style_override("normal", label_style);
  619. hb->add_child(hint_label);
  620. }
  621. }
  622. }
  623. if (!is_group) {
  624. hb->add_spacer();
  625. }
  626. if (valid_right) {
  627. if (is_group) {
  628. Button *remove_btn = memnew(Button);
  629. remove_btn->set_icon(EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Remove", "EditorIcons"));
  630. remove_btn->set_tooltip(TTR("Remove") + " " + name_left);
  631. remove_btn->connect("pressed", callable_mp(this, &VisualShaderEditor::_remove_output_port), varray(nodes[n_i], i), CONNECT_DEFERRED);
  632. hb->add_child(remove_btn);
  633. LineEdit *name_box = memnew(LineEdit);
  634. hb->add_child(name_box);
  635. name_box->set_custom_minimum_size(Size2(65 * EDSCALE, 0));
  636. name_box->set_h_size_flags(SIZE_EXPAND_FILL);
  637. name_box->set_text(name_right);
  638. name_box->connect("text_entered", callable_mp(this, &VisualShaderEditor::_change_output_port_name), varray(name_box, nodes[n_i], i));
  639. name_box->connect("focus_exited", callable_mp(this, &VisualShaderEditor::_port_name_focus_out), varray(name_box, nodes[n_i], i, true));
  640. OptionButton *type_box = memnew(OptionButton);
  641. hb->add_child(type_box);
  642. type_box->add_item(TTR("Float"));
  643. type_box->add_item(TTR("Int"));
  644. type_box->add_item(TTR("Vector"));
  645. type_box->add_item(TTR("Boolean"));
  646. type_box->add_item(TTR("Transform"));
  647. type_box->select(group_node->get_output_port_type(i));
  648. type_box->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  649. type_box->connect("item_selected", callable_mp(this, &VisualShaderEditor::_change_output_port_type), varray(nodes[n_i], i), CONNECT_DEFERRED);
  650. } else {
  651. Label *label = memnew(Label);
  652. label->set_text(name_right);
  653. label->add_theme_style_override("normal", label_style); //more compact
  654. hb->add_child(label);
  655. }
  656. }
  657. }
  658. if (valid_right && edit_type->get_selected() == VisualShader::TYPE_FRAGMENT && port_right != VisualShaderNode::PORT_TYPE_TRANSFORM && port_right != VisualShaderNode::PORT_TYPE_SAMPLER) {
  659. TextureButton *preview = memnew(TextureButton);
  660. preview->set_toggle_mode(true);
  661. preview->set_normal_texture(get_theme_icon("GuiVisibilityHidden", "EditorIcons"));
  662. preview->set_pressed_texture(get_theme_icon("GuiVisibilityVisible", "EditorIcons"));
  663. preview->set_v_size_flags(SIZE_SHRINK_CENTER);
  664. if (vsnode->get_output_port_for_preview() == i) {
  665. preview->set_pressed(true);
  666. }
  667. preview->connect("pressed", callable_mp(this, &VisualShaderEditor::_preview_select_port), varray(nodes[n_i], i), CONNECT_DEFERRED);
  668. hb->add_child(preview);
  669. }
  670. if (is_group) {
  671. offset = memnew(Control);
  672. offset->set_custom_minimum_size(Size2(0, 5 * EDSCALE));
  673. node->add_child(offset);
  674. port_offset++;
  675. }
  676. node->add_child(hb);
  677. node->set_slot(i + port_offset, valid_left, port_left, type_color[port_left], valid_right, port_right, type_color[port_right]);
  678. }
  679. if (vsnode->get_output_port_for_preview() >= 0) {
  680. int port_type = vsnode->get_output_port_type(vsnode->get_output_port_for_preview());
  681. if (port_type != VisualShaderNode::PORT_TYPE_TRANSFORM && port_type != VisualShaderNode::PORT_TYPE_SAMPLER) {
  682. offset = memnew(Control);
  683. offset->set_custom_minimum_size(Size2(0, 5 * EDSCALE));
  684. node->add_child(offset);
  685. VisualShaderNodePortPreview *port_preview = memnew(VisualShaderNodePortPreview);
  686. port_preview->setup(visual_shader, type, nodes[n_i], vsnode->get_output_port_for_preview());
  687. port_preview->set_h_size_flags(SIZE_SHRINK_CENTER);
  688. node->add_child(port_preview);
  689. }
  690. }
  691. offset = memnew(Control);
  692. offset->set_custom_minimum_size(Size2(0, 4 * EDSCALE));
  693. node->add_child(offset);
  694. String error = vsnode->get_warning(visual_shader->get_mode(), type);
  695. if (error != String()) {
  696. Label *error_label = memnew(Label);
  697. error_label->add_theme_color_override("font_color", get_theme_color("error_color", "Editor"));
  698. error_label->set_text(error);
  699. node->add_child(error_label);
  700. }
  701. if (is_expression) {
  702. TextEdit *expression_box = memnew(TextEdit);
  703. Ref<CodeHighlighter> expression_syntax_highlighter;
  704. expression_syntax_highlighter.instance();
  705. expression_node->set_control(expression_box, 0);
  706. node->add_child(expression_box);
  707. Color background_color = EDITOR_GET("text_editor/highlighting/background_color");
  708. Color text_color = EDITOR_GET("text_editor/highlighting/text_color");
  709. Color keyword_color = EDITOR_GET("text_editor/highlighting/keyword_color");
  710. Color comment_color = EDITOR_GET("text_editor/highlighting/comment_color");
  711. Color symbol_color = EDITOR_GET("text_editor/highlighting/symbol_color");
  712. expression_box->set_syntax_highlighter(expression_syntax_highlighter);
  713. expression_box->add_theme_color_override("background_color", background_color);
  714. for (List<String>::Element *E = keyword_list.front(); E; E = E->next()) {
  715. expression_syntax_highlighter->add_keyword_color(E->get(), keyword_color);
  716. }
  717. expression_box->add_theme_font_override("font", get_theme_font("expression", "EditorFonts"));
  718. expression_box->add_theme_color_override("font_color", text_color);
  719. expression_syntax_highlighter->set_symbol_color(symbol_color);
  720. expression_syntax_highlighter->add_color_region("/*", "*/", comment_color, false);
  721. expression_syntax_highlighter->add_color_region("//", "", comment_color, false);
  722. expression_box->set_text(expression);
  723. expression_box->set_context_menu_enabled(false);
  724. expression_box->set_show_line_numbers(true);
  725. expression_box->connect("focus_exited", callable_mp(this, &VisualShaderEditor::_expression_focus_out), varray(expression_box, nodes[n_i]));
  726. }
  727. if (!uniform.is_valid()) {
  728. graph->add_child(node);
  729. _update_created_node(node);
  730. if (is_group) {
  731. call_deferred("_set_node_size", (int)type, nodes[n_i], size);
  732. }
  733. }
  734. }
  735. for (List<VisualShader::Connection>::Element *E = connections.front(); E; E = E->next()) {
  736. int from = E->get().from_node;
  737. int from_idx = E->get().from_port;
  738. int to = E->get().to_node;
  739. int to_idx = E->get().to_port;
  740. graph->connect_node(itos(from), from_idx, itos(to), to_idx);
  741. }
  742. }
  743. void VisualShaderEditor::_add_input_port(int p_node, int p_port, int p_port_type, const String &p_name) {
  744. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  745. Ref<VisualShaderNodeExpression> node = visual_shader->get_node(type, p_node);
  746. if (node.is_null()) {
  747. return;
  748. }
  749. undo_redo->create_action(TTR("Add input port"));
  750. undo_redo->add_do_method(node.ptr(), "add_input_port", p_port, p_port_type, p_name);
  751. undo_redo->add_undo_method(node.ptr(), "remove_input_port", p_port);
  752. undo_redo->add_do_method(this, "_update_graph");
  753. undo_redo->add_undo_method(this, "_update_graph");
  754. undo_redo->add_do_method(this, "_rebuild");
  755. undo_redo->add_undo_method(this, "_rebuild");
  756. undo_redo->commit_action();
  757. }
  758. void VisualShaderEditor::_add_output_port(int p_node, int p_port, int p_port_type, const String &p_name) {
  759. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  760. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  761. if (node.is_null()) {
  762. return;
  763. }
  764. undo_redo->create_action(TTR("Add output port"));
  765. undo_redo->add_do_method(node.ptr(), "add_output_port", p_port, p_port_type, p_name);
  766. undo_redo->add_undo_method(node.ptr(), "remove_output_port", p_port);
  767. undo_redo->add_do_method(this, "_update_graph");
  768. undo_redo->add_undo_method(this, "_update_graph");
  769. undo_redo->add_do_method(this, "_rebuild");
  770. undo_redo->add_undo_method(this, "_rebuild");
  771. undo_redo->commit_action();
  772. }
  773. void VisualShaderEditor::_change_input_port_type(int p_type, int p_node, int p_port) {
  774. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  775. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  776. if (node.is_null()) {
  777. return;
  778. }
  779. undo_redo->create_action(TTR("Change input port type"));
  780. undo_redo->add_do_method(node.ptr(), "set_input_port_type", p_port, p_type);
  781. undo_redo->add_undo_method(node.ptr(), "set_input_port_type", p_port, node->get_input_port_type(p_port));
  782. undo_redo->add_do_method(this, "_update_graph");
  783. undo_redo->add_undo_method(this, "_update_graph");
  784. undo_redo->add_do_method(this, "_rebuild");
  785. undo_redo->add_undo_method(this, "_rebuild");
  786. undo_redo->commit_action();
  787. }
  788. void VisualShaderEditor::_change_output_port_type(int p_type, int p_node, int p_port) {
  789. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  790. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  791. if (node.is_null()) {
  792. return;
  793. }
  794. undo_redo->create_action(TTR("Change output port type"));
  795. undo_redo->add_do_method(node.ptr(), "set_output_port_type", p_port, p_type);
  796. undo_redo->add_undo_method(node.ptr(), "set_output_port_type", p_port, node->get_output_port_type(p_port));
  797. undo_redo->add_do_method(this, "_update_graph");
  798. undo_redo->add_undo_method(this, "_update_graph");
  799. undo_redo->add_do_method(this, "_rebuild");
  800. undo_redo->add_undo_method(this, "_rebuild");
  801. undo_redo->commit_action();
  802. }
  803. void VisualShaderEditor::_change_input_port_name(const String &p_text, Object *line_edit, int p_node_id, int p_port_id) {
  804. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  805. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node_id);
  806. ERR_FAIL_COND(!node.is_valid());
  807. undo_redo->create_action(TTR("Change input port name"));
  808. undo_redo->add_do_method(node.ptr(), "set_input_port_name", p_port_id, p_text);
  809. undo_redo->add_undo_method(node.ptr(), "set_input_port_name", p_port_id, node->get_input_port_name(p_port_id));
  810. undo_redo->add_do_method(this, "_rebuild");
  811. undo_redo->add_undo_method(this, "_rebuild");
  812. undo_redo->commit_action();
  813. }
  814. void VisualShaderEditor::_change_output_port_name(const String &p_text, Object *line_edit, int p_node_id, int p_port_id) {
  815. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  816. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node_id);
  817. ERR_FAIL_COND(!node.is_valid());
  818. undo_redo->create_action(TTR("Change output port name"));
  819. undo_redo->add_do_method(node.ptr(), "set_output_port_name", p_port_id, p_text);
  820. undo_redo->add_undo_method(node.ptr(), "set_output_port_name", p_port_id, node->get_output_port_name(p_port_id));
  821. undo_redo->add_do_method(this, "_rebuild");
  822. undo_redo->add_undo_method(this, "_rebuild");
  823. undo_redo->commit_action();
  824. }
  825. void VisualShaderEditor::_remove_input_port(int p_node, int p_port) {
  826. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  827. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  828. if (node.is_null()) {
  829. return;
  830. }
  831. undo_redo->create_action(TTR("Remove input port"));
  832. List<VisualShader::Connection> conns;
  833. visual_shader->get_node_connections(type, &conns);
  834. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  835. int from_node = E->get().from_node;
  836. int from_port = E->get().from_port;
  837. int to_node = E->get().to_node;
  838. int to_port = E->get().to_port;
  839. if (to_node == p_node) {
  840. if (to_port == p_port) {
  841. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  842. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  843. } else if (to_port > p_port) {
  844. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  845. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  846. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port - 1);
  847. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port - 1);
  848. }
  849. }
  850. }
  851. undo_redo->add_do_method(node.ptr(), "remove_input_port", p_port);
  852. undo_redo->add_undo_method(node.ptr(), "add_input_port", p_port, (int)node->get_input_port_type(p_port), node->get_input_port_name(p_port));
  853. undo_redo->add_do_method(this, "_update_graph");
  854. undo_redo->add_undo_method(this, "_update_graph");
  855. undo_redo->add_do_method(this, "_rebuild");
  856. undo_redo->add_undo_method(this, "_rebuild");
  857. undo_redo->commit_action();
  858. }
  859. void VisualShaderEditor::_remove_output_port(int p_node, int p_port) {
  860. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  861. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  862. if (node.is_null()) {
  863. return;
  864. }
  865. undo_redo->create_action(TTR("Remove output port"));
  866. List<VisualShader::Connection> conns;
  867. visual_shader->get_node_connections(type, &conns);
  868. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  869. int from_node = E->get().from_node;
  870. int from_port = E->get().from_port;
  871. int to_node = E->get().to_node;
  872. int to_port = E->get().to_port;
  873. if (from_node == p_node) {
  874. if (from_port == p_port) {
  875. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  876. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  877. } else if (from_port > p_port) {
  878. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  879. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  880. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port - 1, to_node, to_port);
  881. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port - 1, to_node, to_port);
  882. }
  883. }
  884. }
  885. undo_redo->add_do_method(node.ptr(), "remove_output_port", p_port);
  886. undo_redo->add_undo_method(node.ptr(), "add_output_port", p_port, (int)node->get_output_port_type(p_port), node->get_output_port_name(p_port));
  887. undo_redo->add_do_method(this, "_update_graph");
  888. undo_redo->add_undo_method(this, "_update_graph");
  889. undo_redo->add_do_method(this, "_rebuild");
  890. undo_redo->add_undo_method(this, "_rebuild");
  891. undo_redo->commit_action();
  892. }
  893. void VisualShaderEditor::_expression_focus_out(Object *text_edit, int p_node) {
  894. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  895. Ref<VisualShaderNodeExpression> node = visual_shader->get_node(type, p_node);
  896. if (node.is_null()) {
  897. return;
  898. }
  899. TextEdit *expression_box = Object::cast_to<TextEdit>(text_edit);
  900. if (node->get_expression() == expression_box->get_text()) {
  901. return;
  902. }
  903. undo_redo->create_action(TTR("Set expression"));
  904. undo_redo->add_do_method(node.ptr(), "set_expression", expression_box->get_text());
  905. undo_redo->add_undo_method(node.ptr(), "set_expression", node->get_expression());
  906. undo_redo->add_do_method(this, "_rebuild");
  907. undo_redo->add_undo_method(this, "_rebuild");
  908. undo_redo->commit_action();
  909. }
  910. void VisualShaderEditor::_rebuild() {
  911. if (visual_shader != nullptr) {
  912. EditorNode::get_singleton()->get_log()->clear();
  913. visual_shader->rebuild();
  914. }
  915. }
  916. void VisualShaderEditor::_set_node_size(int p_type, int p_node, const Vector2 &p_size) {
  917. VisualShader::Type type = VisualShader::Type(p_type);
  918. Ref<VisualShaderNode> node = visual_shader->get_node(type, p_node);
  919. if (node.is_null()) {
  920. return;
  921. }
  922. Ref<VisualShaderNodeGroupBase> group_node = Object::cast_to<VisualShaderNodeGroupBase>(node.ptr());
  923. if (group_node.is_null()) {
  924. return;
  925. }
  926. Vector2 size = p_size;
  927. group_node->set_size(size);
  928. GraphNode *gn = nullptr;
  929. if (edit_type->get_selected() == p_type) { // check - otherwise the error will be emitted
  930. Node *node2 = graph->get_node(itos(p_node));
  931. gn = Object::cast_to<GraphNode>(node2);
  932. if (!gn) {
  933. return;
  934. }
  935. gn->set_custom_minimum_size(size);
  936. gn->set_size(Size2(1, 1));
  937. }
  938. Ref<VisualShaderNodeExpression> expression_node = Object::cast_to<VisualShaderNodeExpression>(node.ptr());
  939. if (!expression_node.is_null()) {
  940. Control *text_box = expression_node->get_control(0);
  941. Size2 box_size = size;
  942. if (gn != nullptr) {
  943. if (box_size.x < 150 * EDSCALE || box_size.y < 0) {
  944. box_size.x = gn->get_size().x;
  945. }
  946. }
  947. box_size.x -= text_box->get_margin(MARGIN_LEFT);
  948. box_size.x -= 28 * EDSCALE;
  949. box_size.y -= text_box->get_margin(MARGIN_TOP);
  950. box_size.y -= 28 * EDSCALE;
  951. text_box->set_custom_minimum_size(Size2(box_size.x, box_size.y));
  952. text_box->set_size(Size2(1, 1));
  953. }
  954. }
  955. void VisualShaderEditor::_node_resized(const Vector2 &p_new_size, int p_type, int p_node) {
  956. VisualShader::Type type = VisualShader::Type(p_type);
  957. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  958. if (node.is_null()) {
  959. return;
  960. }
  961. undo_redo->create_action(TTR("Resize VisualShader node"), UndoRedo::MERGE_ENDS);
  962. undo_redo->add_do_method(this, "_set_node_size", p_type, p_node, p_new_size);
  963. undo_redo->add_undo_method(this, "_set_node_size", p_type, p_node, node->get_size());
  964. undo_redo->commit_action();
  965. }
  966. void VisualShaderEditor::_preview_select_port(int p_node, int p_port) {
  967. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  968. Ref<VisualShaderNode> node = visual_shader->get_node(type, p_node);
  969. if (node.is_null()) {
  970. return;
  971. }
  972. if (node->get_output_port_for_preview() == p_port) {
  973. p_port = -1; //toggle it
  974. }
  975. undo_redo->create_action(TTR("Set Uniform Name"));
  976. undo_redo->add_do_method(node.ptr(), "set_output_port_for_preview", p_port);
  977. undo_redo->add_undo_method(node.ptr(), "set_output_port_for_preview", node->get_output_port_for_preview());
  978. undo_redo->add_do_method(this, "_update_graph");
  979. undo_redo->add_undo_method(this, "_update_graph");
  980. undo_redo->commit_action();
  981. }
  982. void VisualShaderEditor::_line_edit_changed(const String &p_text, Object *line_edit, int p_node_id) {
  983. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  984. Ref<VisualShaderNodeUniform> node = visual_shader->get_node(type, p_node_id);
  985. ERR_FAIL_COND(!node.is_valid());
  986. String validated_name = visual_shader->validate_uniform_name(p_text, node);
  987. updating = true;
  988. undo_redo->create_action(TTR("Set Uniform Name"));
  989. undo_redo->add_do_method(node.ptr(), "set_uniform_name", validated_name);
  990. undo_redo->add_undo_method(node.ptr(), "set_uniform_name", node->get_uniform_name());
  991. undo_redo->add_do_method(this, "_update_graph");
  992. undo_redo->add_undo_method(this, "_update_graph");
  993. undo_redo->commit_action();
  994. updating = false;
  995. Object::cast_to<LineEdit>(line_edit)->set_text(validated_name);
  996. }
  997. void VisualShaderEditor::_line_edit_focus_out(Object *line_edit, int p_node_id) {
  998. String text = Object::cast_to<LineEdit>(line_edit)->get_text();
  999. _line_edit_changed(text, line_edit, p_node_id);
  1000. }
  1001. void VisualShaderEditor::_port_name_focus_out(Object *line_edit, int p_node_id, int p_port_id, bool p_output) {
  1002. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1003. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node_id);
  1004. ERR_FAIL_COND(!node.is_valid());
  1005. String text = Object::cast_to<LineEdit>(line_edit)->get_text();
  1006. if (!p_output) {
  1007. if (node->get_input_port_name(p_port_id) == text) {
  1008. return;
  1009. }
  1010. } else {
  1011. if (node->get_output_port_name(p_port_id) == text) {
  1012. return;
  1013. }
  1014. }
  1015. List<String> input_names;
  1016. List<String> output_names;
  1017. for (int i = 0; i < node->get_input_port_count(); i++) {
  1018. if (!p_output && i == p_port_id) {
  1019. continue;
  1020. }
  1021. input_names.push_back(node->get_input_port_name(i));
  1022. }
  1023. for (int i = 0; i < node->get_output_port_count(); i++) {
  1024. if (p_output && i == p_port_id) {
  1025. continue;
  1026. }
  1027. output_names.push_back(node->get_output_port_name(i));
  1028. }
  1029. String validated_name = visual_shader->validate_port_name(text, input_names, output_names);
  1030. if (validated_name == "") {
  1031. if (!p_output) {
  1032. Object::cast_to<LineEdit>(line_edit)->set_text(node->get_input_port_name(p_port_id));
  1033. } else {
  1034. Object::cast_to<LineEdit>(line_edit)->set_text(node->get_output_port_name(p_port_id));
  1035. }
  1036. return;
  1037. }
  1038. if (!p_output) {
  1039. _change_input_port_name(validated_name, line_edit, p_node_id, p_port_id);
  1040. } else {
  1041. _change_output_port_name(validated_name, line_edit, p_node_id, p_port_id);
  1042. }
  1043. }
  1044. void VisualShaderEditor::_port_edited() {
  1045. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1046. Variant value = property_editor->get_variant();
  1047. Ref<VisualShaderNode> vsn = visual_shader->get_node(type, editing_node);
  1048. ERR_FAIL_COND(!vsn.is_valid());
  1049. undo_redo->create_action(TTR("Set Input Default Port"));
  1050. undo_redo->add_do_method(vsn.ptr(), "set_input_port_default_value", editing_port, value);
  1051. undo_redo->add_undo_method(vsn.ptr(), "set_input_port_default_value", editing_port, vsn->get_input_port_default_value(editing_port));
  1052. undo_redo->add_do_method(this, "_update_graph");
  1053. undo_redo->add_undo_method(this, "_update_graph");
  1054. undo_redo->commit_action();
  1055. property_editor->hide();
  1056. }
  1057. void VisualShaderEditor::_edit_port_default_input(Object *p_button, int p_node, int p_port) {
  1058. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1059. Ref<VisualShaderNode> vsn = visual_shader->get_node(type, p_node);
  1060. Button *button = Object::cast_to<Button>(p_button);
  1061. ERR_FAIL_COND(!button);
  1062. Variant value = vsn->get_input_port_default_value(p_port);
  1063. property_editor->set_position(button->get_screen_position() + Vector2(0, button->get_size().height));
  1064. property_editor->edit(nullptr, "", value.get_type(), value, 0, "");
  1065. property_editor->popup();
  1066. editing_node = p_node;
  1067. editing_port = p_port;
  1068. }
  1069. void VisualShaderEditor::_add_custom_node(const String &p_path) {
  1070. int idx = -1;
  1071. for (int i = custom_node_option_idx; i < add_options.size(); i++) {
  1072. if (add_options[i].script.is_valid()) {
  1073. if (add_options[i].script->get_path() == p_path) {
  1074. idx = i;
  1075. break;
  1076. }
  1077. }
  1078. }
  1079. if (idx != -1) {
  1080. _add_node(idx);
  1081. }
  1082. }
  1083. void VisualShaderEditor::_add_texture_node(const String &p_path) {
  1084. VisualShaderNodeTexture *texture = (VisualShaderNodeTexture *)_add_node(texture_node_option_idx, -1);
  1085. texture->set_texture(ResourceLoader::load(p_path));
  1086. }
  1087. VisualShaderNode *VisualShaderEditor::_add_node(int p_idx, int p_op_idx) {
  1088. ERR_FAIL_INDEX_V(p_idx, add_options.size(), nullptr);
  1089. Ref<VisualShaderNode> vsnode;
  1090. bool is_custom = add_options[p_idx].is_custom;
  1091. if (!is_custom && add_options[p_idx].type != String()) {
  1092. VisualShaderNode *vsn = Object::cast_to<VisualShaderNode>(ClassDB::instance(add_options[p_idx].type));
  1093. ERR_FAIL_COND_V(!vsn, nullptr);
  1094. VisualShaderNodeFloatConstant *constant = Object::cast_to<VisualShaderNodeFloatConstant>(vsn);
  1095. if (constant) {
  1096. if ((int)add_options[p_idx].value != -1) {
  1097. constant->set_constant(add_options[p_idx].value);
  1098. }
  1099. }
  1100. if (p_op_idx != -1) {
  1101. VisualShaderNodeInput *input = Object::cast_to<VisualShaderNodeInput>(vsn);
  1102. if (input) {
  1103. input->set_input_name(add_options[p_idx].sub_func_str);
  1104. }
  1105. VisualShaderNodeIs *is = Object::cast_to<VisualShaderNodeIs>(vsn);
  1106. if (is) {
  1107. is->set_function((VisualShaderNodeIs::Function)p_op_idx);
  1108. }
  1109. VisualShaderNodeCompare *cmp = Object::cast_to<VisualShaderNodeCompare>(vsn);
  1110. if (cmp) {
  1111. cmp->set_function((VisualShaderNodeCompare::Function)p_op_idx);
  1112. }
  1113. VisualShaderNodeColorOp *colorOp = Object::cast_to<VisualShaderNodeColorOp>(vsn);
  1114. if (colorOp) {
  1115. colorOp->set_operator((VisualShaderNodeColorOp::Operator)p_op_idx);
  1116. }
  1117. VisualShaderNodeColorFunc *colorFunc = Object::cast_to<VisualShaderNodeColorFunc>(vsn);
  1118. if (colorFunc) {
  1119. colorFunc->set_function((VisualShaderNodeColorFunc::Function)p_op_idx);
  1120. }
  1121. VisualShaderNodeFloatOp *floatOp = Object::cast_to<VisualShaderNodeFloatOp>(vsn);
  1122. if (floatOp) {
  1123. floatOp->set_operator((VisualShaderNodeFloatOp::Operator)p_op_idx);
  1124. }
  1125. VisualShaderNodeIntOp *intOp = Object::cast_to<VisualShaderNodeIntOp>(vsn);
  1126. if (intOp) {
  1127. intOp->set_operator((VisualShaderNodeIntOp::Operator)p_op_idx);
  1128. }
  1129. VisualShaderNodeFloatFunc *floatFunc = Object::cast_to<VisualShaderNodeFloatFunc>(vsn);
  1130. if (floatFunc) {
  1131. floatFunc->set_function((VisualShaderNodeFloatFunc::Function)p_op_idx);
  1132. }
  1133. VisualShaderNodeIntFunc *intFunc = Object::cast_to<VisualShaderNodeIntFunc>(vsn);
  1134. if (intFunc) {
  1135. intFunc->set_function((VisualShaderNodeIntFunc::Function)p_op_idx);
  1136. }
  1137. VisualShaderNodeVectorOp *vecOp = Object::cast_to<VisualShaderNodeVectorOp>(vsn);
  1138. if (vecOp) {
  1139. vecOp->set_operator((VisualShaderNodeVectorOp::Operator)p_op_idx);
  1140. }
  1141. VisualShaderNodeVectorFunc *vecFunc = Object::cast_to<VisualShaderNodeVectorFunc>(vsn);
  1142. if (vecFunc) {
  1143. vecFunc->set_function((VisualShaderNodeVectorFunc::Function)p_op_idx);
  1144. }
  1145. VisualShaderNodeTransformFunc *matFunc = Object::cast_to<VisualShaderNodeTransformFunc>(vsn);
  1146. if (matFunc) {
  1147. matFunc->set_function((VisualShaderNodeTransformFunc::Function)p_op_idx);
  1148. }
  1149. VisualShaderNodeScalarDerivativeFunc *sderFunc = Object::cast_to<VisualShaderNodeScalarDerivativeFunc>(vsn);
  1150. if (sderFunc) {
  1151. sderFunc->set_function((VisualShaderNodeScalarDerivativeFunc::Function)p_op_idx);
  1152. }
  1153. VisualShaderNodeVectorDerivativeFunc *vderFunc = Object::cast_to<VisualShaderNodeVectorDerivativeFunc>(vsn);
  1154. if (vderFunc) {
  1155. vderFunc->set_function((VisualShaderNodeVectorDerivativeFunc::Function)p_op_idx);
  1156. }
  1157. VisualShaderNodeMultiplyAdd *fmaFunc = Object::cast_to<VisualShaderNodeMultiplyAdd>(vsn);
  1158. if (fmaFunc) {
  1159. fmaFunc->set_type((VisualShaderNodeMultiplyAdd::Type)p_op_idx);
  1160. }
  1161. }
  1162. vsnode = Ref<VisualShaderNode>(vsn);
  1163. } else {
  1164. ERR_FAIL_COND_V(add_options[p_idx].script.is_null(), nullptr);
  1165. String base_type = add_options[p_idx].script->get_instance_base_type();
  1166. VisualShaderNode *vsn = Object::cast_to<VisualShaderNode>(ClassDB::instance(base_type));
  1167. ERR_FAIL_COND_V(!vsn, nullptr);
  1168. vsnode = Ref<VisualShaderNode>(vsn);
  1169. vsnode->set_script(add_options[p_idx].script);
  1170. }
  1171. Point2 position = graph->get_scroll_ofs();
  1172. if (saved_node_pos_dirty) {
  1173. position += saved_node_pos;
  1174. } else {
  1175. position += graph->get_size() * 0.5;
  1176. position /= EDSCALE;
  1177. }
  1178. saved_node_pos_dirty = false;
  1179. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1180. int id_to_use = visual_shader->get_valid_node_id(type);
  1181. undo_redo->create_action(TTR("Add Node to Visual Shader"));
  1182. undo_redo->add_do_method(visual_shader.ptr(), "add_node", type, vsnode, position, id_to_use);
  1183. undo_redo->add_undo_method(visual_shader.ptr(), "remove_node", type, id_to_use);
  1184. VisualShaderNodeExpression *expr = Object::cast_to<VisualShaderNodeExpression>(vsnode.ptr());
  1185. if (expr) {
  1186. undo_redo->add_do_method(expr, "set_size", Size2(250 * EDSCALE, 150 * EDSCALE));
  1187. }
  1188. if (to_node != -1 && to_slot != -1) {
  1189. if (vsnode->get_output_port_count() > 0) {
  1190. int _from_node = id_to_use;
  1191. int _from_slot = 0;
  1192. if (visual_shader->is_port_types_compatible(vsnode->get_output_port_type(_from_slot), visual_shader->get_node(type, to_node)->get_input_port_type(to_slot))) {
  1193. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, _from_node, _from_slot, to_node, to_slot);
  1194. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, _from_node, _from_slot, to_node, to_slot);
  1195. }
  1196. }
  1197. } else if (from_node != -1 && from_slot != -1) {
  1198. if (vsnode->get_input_port_count() > 0) {
  1199. int _to_node = id_to_use;
  1200. int _to_slot = 0;
  1201. if (visual_shader->is_port_types_compatible(visual_shader->get_node(type, from_node)->get_output_port_type(from_slot), vsnode->get_input_port_type(_to_slot))) {
  1202. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, from_node, from_slot, _to_node, _to_slot);
  1203. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_slot, _to_node, _to_slot);
  1204. }
  1205. }
  1206. }
  1207. undo_redo->add_do_method(this, "_update_graph");
  1208. undo_redo->add_undo_method(this, "_update_graph");
  1209. undo_redo->commit_action();
  1210. return vsnode.ptr();
  1211. }
  1212. void VisualShaderEditor::_node_dragged(const Vector2 &p_from, const Vector2 &p_to, int p_node) {
  1213. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1214. updating = true;
  1215. undo_redo->create_action(TTR("Node Moved"));
  1216. undo_redo->add_do_method(visual_shader.ptr(), "set_node_position", type, p_node, p_to);
  1217. undo_redo->add_undo_method(visual_shader.ptr(), "set_node_position", type, p_node, p_from);
  1218. undo_redo->add_do_method(this, "_update_graph");
  1219. undo_redo->add_undo_method(this, "_update_graph");
  1220. undo_redo->commit_action();
  1221. updating = false;
  1222. }
  1223. void VisualShaderEditor::_connection_request(const String &p_from, int p_from_index, const String &p_to, int p_to_index) {
  1224. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1225. int from = p_from.to_int();
  1226. int to = p_to.to_int();
  1227. if (!visual_shader->can_connect_nodes(type, from, p_from_index, to, p_to_index)) {
  1228. return;
  1229. }
  1230. undo_redo->create_action(TTR("Nodes Connected"));
  1231. List<VisualShader::Connection> conns;
  1232. visual_shader->get_node_connections(type, &conns);
  1233. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  1234. if (E->get().to_node == to && E->get().to_port == p_to_index) {
  1235. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  1236. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  1237. }
  1238. }
  1239. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, from, p_from_index, to, p_to_index);
  1240. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from, p_from_index, to, p_to_index);
  1241. undo_redo->add_do_method(this, "_update_graph");
  1242. undo_redo->add_undo_method(this, "_update_graph");
  1243. undo_redo->commit_action();
  1244. }
  1245. void VisualShaderEditor::_disconnection_request(const String &p_from, int p_from_index, const String &p_to, int p_to_index) {
  1246. graph->disconnect_node(p_from, p_from_index, p_to, p_to_index);
  1247. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1248. int from = p_from.to_int();
  1249. int to = p_to.to_int();
  1250. //updating = true; seems graph edit can handle this, no need to protect
  1251. undo_redo->create_action(TTR("Nodes Disconnected"));
  1252. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from, p_from_index, to, p_to_index);
  1253. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes", type, from, p_from_index, to, p_to_index);
  1254. undo_redo->add_do_method(this, "_update_graph");
  1255. undo_redo->add_undo_method(this, "_update_graph");
  1256. undo_redo->commit_action();
  1257. //updating = false;
  1258. }
  1259. void VisualShaderEditor::_connection_to_empty(const String &p_from, int p_from_slot, const Vector2 &p_release_position) {
  1260. from_node = p_from.to_int();
  1261. from_slot = p_from_slot;
  1262. _show_members_dialog(true);
  1263. }
  1264. void VisualShaderEditor::_connection_from_empty(const String &p_to, int p_to_slot, const Vector2 &p_release_position) {
  1265. to_node = p_to.to_int();
  1266. to_slot = p_to_slot;
  1267. _show_members_dialog(true);
  1268. }
  1269. void VisualShaderEditor::_delete_request(int which) {
  1270. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1271. Ref<VisualShaderNode> node = Ref<VisualShaderNode>(visual_shader->get_node(type, which));
  1272. undo_redo->create_action(TTR("Delete Node"));
  1273. undo_redo->add_do_method(visual_shader.ptr(), "remove_node", type, which);
  1274. undo_redo->add_undo_method(visual_shader.ptr(), "add_node", type, node, visual_shader->get_node_position(type, which), which);
  1275. undo_redo->add_do_method(this, "_clear_buffer");
  1276. undo_redo->add_undo_method(this, "_clear_buffer");
  1277. // restore size, inputs and outputs if node is group
  1278. VisualShaderNodeGroupBase *group = Object::cast_to<VisualShaderNodeGroupBase>(node.ptr());
  1279. if (group) {
  1280. undo_redo->add_undo_method(group, "set_size", group->get_size());
  1281. undo_redo->add_undo_method(group, "set_inputs", group->get_inputs());
  1282. undo_redo->add_undo_method(group, "set_outputs", group->get_outputs());
  1283. }
  1284. // restore expression text if node is expression
  1285. VisualShaderNodeExpression *expression = Object::cast_to<VisualShaderNodeExpression>(node.ptr());
  1286. if (expression) {
  1287. undo_redo->add_undo_method(expression, "set_expression", expression->get_expression());
  1288. }
  1289. List<VisualShader::Connection> conns;
  1290. visual_shader->get_node_connections(type, &conns);
  1291. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  1292. if (E->get().from_node == which || E->get().to_node == which) {
  1293. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  1294. }
  1295. }
  1296. undo_redo->add_do_method(this, "_update_graph");
  1297. undo_redo->add_undo_method(this, "_update_graph");
  1298. undo_redo->commit_action();
  1299. }
  1300. void VisualShaderEditor::_node_selected(Object *p_node) {
  1301. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1302. GraphNode *gn = Object::cast_to<GraphNode>(p_node);
  1303. ERR_FAIL_COND(!gn);
  1304. int id = String(gn->get_name()).to_int();
  1305. Ref<VisualShaderNode> vsnode = visual_shader->get_node(type, id);
  1306. ERR_FAIL_COND(!vsnode.is_valid());
  1307. //do not rely on this, makes editor more complex
  1308. //EditorNode::get_singleton()->push_item(vsnode.ptr(), "", true);
  1309. }
  1310. void VisualShaderEditor::_graph_gui_input(const Ref<InputEvent> &p_event) {
  1311. Ref<InputEventMouseButton> mb = p_event;
  1312. if (mb.is_valid() && mb->is_pressed() && mb->get_button_index() == BUTTON_RIGHT) {
  1313. List<int> to_change;
  1314. for (int i = 0; i < graph->get_child_count(); i++) {
  1315. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  1316. if (gn) {
  1317. if (gn->is_selected() && gn->is_close_button_visible()) {
  1318. to_change.push_back(gn->get_name().operator String().to_int());
  1319. }
  1320. }
  1321. }
  1322. if (to_change.empty() && copy_nodes_buffer.empty()) {
  1323. _show_members_dialog(true);
  1324. } else {
  1325. popup_menu->set_item_disabled(NodeMenuOptions::COPY, to_change.empty());
  1326. popup_menu->set_item_disabled(NodeMenuOptions::PASTE, copy_nodes_buffer.empty());
  1327. popup_menu->set_item_disabled(NodeMenuOptions::DELETE, to_change.empty());
  1328. popup_menu->set_item_disabled(NodeMenuOptions::DUPLICATE, to_change.empty());
  1329. menu_point = graph->get_local_mouse_position();
  1330. Point2 gpos = Input::get_singleton()->get_mouse_position();
  1331. popup_menu->set_position(gpos);
  1332. popup_menu->popup();
  1333. }
  1334. }
  1335. }
  1336. void VisualShaderEditor::_show_members_dialog(bool at_mouse_pos) {
  1337. if (at_mouse_pos) {
  1338. saved_node_pos_dirty = true;
  1339. saved_node_pos = graph->get_local_mouse_position();
  1340. Point2 gpos = Input::get_singleton()->get_mouse_position();
  1341. members_dialog->popup();
  1342. members_dialog->set_position(gpos);
  1343. } else {
  1344. members_dialog->popup();
  1345. saved_node_pos_dirty = false;
  1346. members_dialog->set_position(graph->get_global_position() + Point2(5 * EDSCALE, 65 * EDSCALE));
  1347. }
  1348. // keep dialog within window bounds
  1349. Size2 window_size = DisplayServer::get_singleton()->window_get_size();
  1350. Rect2 dialog_rect = Rect2(members_dialog->get_position(), members_dialog->get_size());
  1351. if (dialog_rect.position.y + dialog_rect.size.y > window_size.y) {
  1352. int difference = dialog_rect.position.y + dialog_rect.size.y - window_size.y;
  1353. members_dialog->set_position(members_dialog->get_position() - Point2(0, difference));
  1354. }
  1355. if (dialog_rect.position.x + dialog_rect.size.x > window_size.x) {
  1356. int difference = dialog_rect.position.x + dialog_rect.size.x - window_size.x;
  1357. members_dialog->set_position(members_dialog->get_position() - Point2(difference, 0));
  1358. }
  1359. node_filter->call_deferred("grab_focus"); // still not visible
  1360. node_filter->select_all();
  1361. }
  1362. void VisualShaderEditor::_sbox_input(const Ref<InputEvent> &p_ie) {
  1363. Ref<InputEventKey> ie = p_ie;
  1364. if (ie.is_valid() && (ie->get_keycode() == KEY_UP ||
  1365. ie->get_keycode() == KEY_DOWN ||
  1366. ie->get_keycode() == KEY_ENTER ||
  1367. ie->get_keycode() == KEY_KP_ENTER)) {
  1368. members->call("_gui_input", ie);
  1369. node_filter->accept_event();
  1370. }
  1371. }
  1372. void VisualShaderEditor::_notification(int p_what) {
  1373. if (p_what == NOTIFICATION_ENTER_TREE) {
  1374. node_filter->set_clear_button_enabled(true);
  1375. // collapse tree by default
  1376. TreeItem *category = members->get_root()->get_children();
  1377. while (category) {
  1378. category->set_collapsed(true);
  1379. TreeItem *sub_category = category->get_children();
  1380. while (sub_category) {
  1381. sub_category->set_collapsed(true);
  1382. sub_category = sub_category->get_next();
  1383. }
  1384. category = category->get_next();
  1385. }
  1386. }
  1387. if (p_what == NOTIFICATION_DRAG_BEGIN) {
  1388. Dictionary dd = get_viewport()->gui_get_drag_data();
  1389. if (members->is_visible_in_tree() && dd.has("id")) {
  1390. members->set_drop_mode_flags(Tree::DROP_MODE_ON_ITEM);
  1391. }
  1392. } else if (p_what == NOTIFICATION_DRAG_END) {
  1393. members->set_drop_mode_flags(0);
  1394. }
  1395. if (p_what == NOTIFICATION_ENTER_TREE || p_what == NOTIFICATION_THEME_CHANGED) {
  1396. highend_label->set_modulate(get_theme_color("vulkan_color", "Editor"));
  1397. error_panel->add_theme_style_override("panel", get_theme_stylebox("bg", "Tree"));
  1398. error_label->add_theme_color_override("font_color", get_theme_color("error_color", "Editor"));
  1399. node_filter->set_right_icon(Control::get_theme_icon("Search", "EditorIcons"));
  1400. preview_shader->set_icon(Control::get_theme_icon("Shader", "EditorIcons"));
  1401. {
  1402. Color background_color = EDITOR_GET("text_editor/highlighting/background_color");
  1403. Color text_color = EDITOR_GET("text_editor/highlighting/text_color");
  1404. Color keyword_color = EDITOR_GET("text_editor/highlighting/keyword_color");
  1405. Color comment_color = EDITOR_GET("text_editor/highlighting/comment_color");
  1406. Color symbol_color = EDITOR_GET("text_editor/highlighting/symbol_color");
  1407. preview_text->add_theme_color_override("background_color", background_color);
  1408. for (List<String>::Element *E = keyword_list.front(); E; E = E->next()) {
  1409. syntax_highlighter->add_keyword_color(E->get(), keyword_color);
  1410. }
  1411. preview_text->add_theme_font_override("font", get_theme_font("expression", "EditorFonts"));
  1412. preview_text->add_theme_color_override("font_color", text_color);
  1413. syntax_highlighter->set_symbol_color(symbol_color);
  1414. syntax_highlighter->add_color_region("/*", "*/", comment_color, false);
  1415. syntax_highlighter->add_color_region("//", "", comment_color, false);
  1416. error_text->add_theme_font_override("font", get_theme_font("status_source", "EditorFonts"));
  1417. error_text->add_theme_color_override("font_color", get_theme_color("error_color", "Editor"));
  1418. }
  1419. tools->set_icon(EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Tools", "EditorIcons"));
  1420. if (p_what == NOTIFICATION_THEME_CHANGED && is_visible_in_tree()) {
  1421. _update_graph();
  1422. }
  1423. }
  1424. }
  1425. void VisualShaderEditor::_scroll_changed(const Vector2 &p_scroll) {
  1426. if (updating) {
  1427. return;
  1428. }
  1429. updating = true;
  1430. visual_shader->set_graph_offset(p_scroll / EDSCALE);
  1431. updating = false;
  1432. }
  1433. void VisualShaderEditor::_node_changed(int p_id) {
  1434. if (updating) {
  1435. return;
  1436. }
  1437. if (is_visible_in_tree()) {
  1438. _update_graph();
  1439. }
  1440. }
  1441. void VisualShaderEditor::_dup_update_excluded(int p_type, Set<int> &r_excluded) {
  1442. r_excluded.clear();
  1443. VisualShader::Type type = (VisualShader::Type)p_type;
  1444. for (int i = 0; i < graph->get_child_count(); i++) {
  1445. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  1446. if (gn) {
  1447. int id = String(gn->get_name()).to_int();
  1448. Ref<VisualShaderNode> node = visual_shader->get_node(type, id);
  1449. Ref<VisualShaderNodeOutput> output = node;
  1450. if (output.is_valid()) {
  1451. r_excluded.insert(id);
  1452. continue;
  1453. }
  1454. r_excluded.insert(id);
  1455. }
  1456. }
  1457. }
  1458. void VisualShaderEditor::_dup_copy_nodes(int p_type, List<int> &r_nodes, Set<int> &r_excluded) {
  1459. VisualShader::Type type = (VisualShader::Type)p_type;
  1460. selection_center.x = 0.0f;
  1461. selection_center.y = 0.0f;
  1462. for (int i = 0; i < graph->get_child_count(); i++) {
  1463. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  1464. if (gn) {
  1465. int id = String(gn->get_name()).to_int();
  1466. Ref<VisualShaderNode> node = visual_shader->get_node(type, id);
  1467. Ref<VisualShaderNodeOutput> output = node;
  1468. if (output.is_valid()) { // can't duplicate output
  1469. r_excluded.insert(id);
  1470. continue;
  1471. }
  1472. if (node.is_valid() && gn->is_selected()) {
  1473. Vector2 pos = visual_shader->get_node_position(type, id);
  1474. selection_center += pos;
  1475. r_nodes.push_back(id);
  1476. }
  1477. r_excluded.insert(id);
  1478. }
  1479. }
  1480. selection_center /= (float)r_nodes.size();
  1481. }
  1482. void VisualShaderEditor::_dup_paste_nodes(int p_type, int p_pasted_type, List<int> &r_nodes, Set<int> &r_excluded, const Vector2 &p_offset, bool p_select) {
  1483. VisualShader::Type type = (VisualShader::Type)p_type;
  1484. VisualShader::Type pasted_type = (VisualShader::Type)p_pasted_type;
  1485. int base_id = visual_shader->get_valid_node_id(type);
  1486. int id_from = base_id;
  1487. Map<int, int> connection_remap;
  1488. Set<int> unsupported_set;
  1489. for (List<int>::Element *E = r_nodes.front(); E; E = E->next()) {
  1490. connection_remap[E->get()] = id_from;
  1491. Ref<VisualShaderNode> node = visual_shader->get_node(pasted_type, E->get());
  1492. bool unsupported = false;
  1493. for (int i = 0; i < add_options.size(); i++) {
  1494. if (add_options[i].type == node->get_class_name()) {
  1495. if (!_is_available(add_options[i].mode)) {
  1496. unsupported = true;
  1497. }
  1498. break;
  1499. }
  1500. }
  1501. if (unsupported) {
  1502. unsupported_set.insert(E->get());
  1503. continue;
  1504. }
  1505. Ref<VisualShaderNode> dupli = node->duplicate();
  1506. undo_redo->add_do_method(visual_shader.ptr(), "add_node", type, dupli, visual_shader->get_node_position(pasted_type, E->get()) + p_offset, id_from);
  1507. undo_redo->add_undo_method(visual_shader.ptr(), "remove_node", type, id_from);
  1508. // duplicate size, inputs and outputs if node is group
  1509. Ref<VisualShaderNodeGroupBase> group = Object::cast_to<VisualShaderNodeGroupBase>(node.ptr());
  1510. if (!group.is_null()) {
  1511. undo_redo->add_do_method(dupli.ptr(), "set_size", group->get_size());
  1512. undo_redo->add_do_method(dupli.ptr(), "set_inputs", group->get_inputs());
  1513. undo_redo->add_do_method(dupli.ptr(), "set_outputs", group->get_outputs());
  1514. }
  1515. // duplicate expression text if node is expression
  1516. Ref<VisualShaderNodeExpression> expression = Object::cast_to<VisualShaderNodeExpression>(node.ptr());
  1517. if (!expression.is_null()) {
  1518. undo_redo->add_do_method(dupli.ptr(), "set_expression", expression->get_expression());
  1519. }
  1520. id_from++;
  1521. }
  1522. List<VisualShader::Connection> conns;
  1523. visual_shader->get_node_connections(pasted_type, &conns);
  1524. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  1525. if (unsupported_set.has(E->get().from_node) || unsupported_set.has(E->get().to_node)) {
  1526. continue;
  1527. }
  1528. if (connection_remap.has(E->get().from_node) && connection_remap.has(E->get().to_node)) {
  1529. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes_forced", type, connection_remap[E->get().from_node], E->get().from_port, connection_remap[E->get().to_node], E->get().to_port);
  1530. }
  1531. }
  1532. undo_redo->add_do_method(this, "_update_graph");
  1533. undo_redo->add_undo_method(this, "_update_graph");
  1534. undo_redo->commit_action();
  1535. if (p_select) {
  1536. // reselect duplicated nodes by excluding the other ones
  1537. for (int i = 0; i < graph->get_child_count(); i++) {
  1538. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  1539. if (gn) {
  1540. int id = String(gn->get_name()).to_int();
  1541. if (!r_excluded.has(id)) {
  1542. gn->set_selected(true);
  1543. } else {
  1544. gn->set_selected(false);
  1545. }
  1546. }
  1547. }
  1548. }
  1549. }
  1550. void VisualShaderEditor::_clear_buffer() {
  1551. copy_nodes_buffer.clear();
  1552. copy_nodes_excluded_buffer.clear();
  1553. }
  1554. void VisualShaderEditor::_duplicate_nodes() {
  1555. int type = edit_type->get_selected();
  1556. List<int> nodes;
  1557. Set<int> excluded;
  1558. _dup_copy_nodes(type, nodes, excluded);
  1559. if (nodes.empty()) {
  1560. return;
  1561. }
  1562. undo_redo->create_action(TTR("Duplicate Nodes"));
  1563. _dup_paste_nodes(type, type, nodes, excluded, Vector2(10, 10) * EDSCALE, true);
  1564. }
  1565. void VisualShaderEditor::_copy_nodes() {
  1566. copy_type = edit_type->get_selected();
  1567. _clear_buffer();
  1568. _dup_copy_nodes(copy_type, copy_nodes_buffer, copy_nodes_excluded_buffer);
  1569. }
  1570. void VisualShaderEditor::_paste_nodes(bool p_use_custom_position, const Vector2 &p_custom_position) {
  1571. if (copy_nodes_buffer.empty()) {
  1572. return;
  1573. }
  1574. int type = edit_type->get_selected();
  1575. undo_redo->create_action(TTR("Paste Nodes"));
  1576. float scale = graph->get_zoom();
  1577. Vector2 mpos;
  1578. if (p_use_custom_position) {
  1579. mpos = p_custom_position;
  1580. } else {
  1581. mpos = graph->get_local_mouse_position();
  1582. }
  1583. _dup_paste_nodes(type, copy_type, copy_nodes_buffer, copy_nodes_excluded_buffer, (graph->get_scroll_ofs() / scale + mpos / scale - selection_center), false);
  1584. _dup_update_excluded(type, copy_nodes_excluded_buffer); // to prevent selection of previous copies at new paste
  1585. }
  1586. void VisualShaderEditor::_delete_nodes() {
  1587. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1588. List<int> to_erase;
  1589. for (int i = 0; i < graph->get_child_count(); i++) {
  1590. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  1591. if (gn) {
  1592. if (gn->is_selected() && gn->is_close_button_visible()) {
  1593. to_erase.push_back(gn->get_name().operator String().to_int());
  1594. }
  1595. }
  1596. }
  1597. if (to_erase.empty()) {
  1598. return;
  1599. }
  1600. undo_redo->create_action(TTR("Delete Nodes"));
  1601. for (List<int>::Element *F = to_erase.front(); F; F = F->next()) {
  1602. Ref<VisualShaderNode> node = visual_shader->get_node(type, F->get());
  1603. undo_redo->add_do_method(visual_shader.ptr(), "remove_node", type, F->get());
  1604. undo_redo->add_undo_method(visual_shader.ptr(), "add_node", type, node, visual_shader->get_node_position(type, F->get()), F->get());
  1605. undo_redo->add_do_method(this, "_clear_buffer");
  1606. undo_redo->add_undo_method(this, "_clear_buffer");
  1607. // restore size, inputs and outputs if node is group
  1608. VisualShaderNodeGroupBase *group = Object::cast_to<VisualShaderNodeGroupBase>(node.ptr());
  1609. if (group) {
  1610. undo_redo->add_undo_method(group, "set_size", group->get_size());
  1611. undo_redo->add_undo_method(group, "set_inputs", group->get_inputs());
  1612. undo_redo->add_undo_method(group, "set_outputs", group->get_outputs());
  1613. }
  1614. // restore expression text if node is expression
  1615. VisualShaderNodeExpression *expression = Object::cast_to<VisualShaderNodeExpression>(node.ptr());
  1616. if (expression) {
  1617. undo_redo->add_undo_method(expression, "set_expression", expression->get_expression());
  1618. }
  1619. }
  1620. List<VisualShader::Connection> conns;
  1621. visual_shader->get_node_connections(type, &conns);
  1622. List<VisualShader::Connection> used_conns;
  1623. for (List<int>::Element *F = to_erase.front(); F; F = F->next()) {
  1624. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  1625. if (E->get().from_node == F->get() || E->get().to_node == F->get()) {
  1626. bool cancel = false;
  1627. for (List<VisualShader::Connection>::Element *R = used_conns.front(); R; R = R->next()) {
  1628. if (R->get().from_node == E->get().from_node && R->get().from_port == E->get().from_port && R->get().to_node == E->get().to_node && R->get().to_port == E->get().to_port) {
  1629. cancel = true; // to avoid ERR_ALREADY_EXISTS warning
  1630. break;
  1631. }
  1632. }
  1633. if (!cancel) {
  1634. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  1635. used_conns.push_back(E->get());
  1636. }
  1637. }
  1638. }
  1639. }
  1640. undo_redo->add_do_method(this, "_update_graph");
  1641. undo_redo->add_undo_method(this, "_update_graph");
  1642. undo_redo->commit_action();
  1643. }
  1644. void VisualShaderEditor::_mode_selected(int p_id) {
  1645. _update_options_menu();
  1646. _update_graph();
  1647. }
  1648. void VisualShaderEditor::_input_select_item(Ref<VisualShaderNodeInput> input, String name) {
  1649. String prev_name = input->get_input_name();
  1650. if (name == prev_name) {
  1651. return;
  1652. }
  1653. bool type_changed = input->get_input_type_by_name(name) != input->get_input_type_by_name(prev_name);
  1654. UndoRedo *undo_redo = EditorNode::get_singleton()->get_undo_redo();
  1655. undo_redo->create_action(TTR("Visual Shader Input Type Changed"));
  1656. undo_redo->add_do_method(input.ptr(), "set_input_name", name);
  1657. undo_redo->add_undo_method(input.ptr(), "set_input_name", prev_name);
  1658. if (type_changed) {
  1659. //restore connections if type changed
  1660. VisualShader::Type type = VisualShader::Type(edit_type->get_selected());
  1661. int id = visual_shader->find_node_id(type, input);
  1662. List<VisualShader::Connection> conns;
  1663. visual_shader->get_node_connections(type, &conns);
  1664. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  1665. if (E->get().from_node == id) {
  1666. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  1667. }
  1668. }
  1669. }
  1670. undo_redo->add_do_method(VisualShaderEditor::get_singleton(), "_update_graph");
  1671. undo_redo->add_undo_method(VisualShaderEditor::get_singleton(), "_update_graph");
  1672. undo_redo->commit_action();
  1673. }
  1674. void VisualShaderEditor::_member_filter_changed(const String &p_text) {
  1675. _update_options_menu();
  1676. }
  1677. void VisualShaderEditor::_member_selected() {
  1678. TreeItem *item = members->get_selected();
  1679. if (item != nullptr && item->has_meta("id")) {
  1680. members_dialog->get_ok()->set_disabled(false);
  1681. highend_label->set_visible(add_options[item->get_meta("id")].highend);
  1682. node_desc->set_text(_get_description(item->get_meta("id")));
  1683. } else {
  1684. highend_label->set_visible(false);
  1685. members_dialog->get_ok()->set_disabled(true);
  1686. node_desc->set_text("");
  1687. }
  1688. }
  1689. void VisualShaderEditor::_member_unselected() {
  1690. }
  1691. void VisualShaderEditor::_member_create() {
  1692. TreeItem *item = members->get_selected();
  1693. if (item != nullptr && item->has_meta("id")) {
  1694. int idx = members->get_selected()->get_meta("id");
  1695. _add_node(idx, add_options[idx].sub_func);
  1696. members_dialog->hide();
  1697. }
  1698. }
  1699. void VisualShaderEditor::_member_cancel() {
  1700. to_node = -1;
  1701. to_slot = -1;
  1702. from_node = -1;
  1703. from_slot = -1;
  1704. }
  1705. void VisualShaderEditor::_tools_menu_option(int p_idx) {
  1706. TreeItem *category = members->get_root()->get_children();
  1707. switch (p_idx) {
  1708. case EXPAND_ALL:
  1709. while (category) {
  1710. category->set_collapsed(false);
  1711. TreeItem *sub_category = category->get_children();
  1712. while (sub_category) {
  1713. sub_category->set_collapsed(false);
  1714. sub_category = sub_category->get_next();
  1715. }
  1716. category = category->get_next();
  1717. }
  1718. break;
  1719. case COLLAPSE_ALL:
  1720. while (category) {
  1721. category->set_collapsed(true);
  1722. TreeItem *sub_category = category->get_children();
  1723. while (sub_category) {
  1724. sub_category->set_collapsed(true);
  1725. sub_category = sub_category->get_next();
  1726. }
  1727. category = category->get_next();
  1728. }
  1729. break;
  1730. default:
  1731. break;
  1732. }
  1733. }
  1734. void VisualShaderEditor::_node_menu_id_pressed(int p_idx) {
  1735. switch (p_idx) {
  1736. case NodeMenuOptions::ADD:
  1737. _show_members_dialog(true);
  1738. break;
  1739. case NodeMenuOptions::COPY:
  1740. _copy_nodes();
  1741. break;
  1742. case NodeMenuOptions::PASTE:
  1743. _paste_nodes(true, menu_point);
  1744. break;
  1745. case NodeMenuOptions::DELETE:
  1746. _delete_nodes();
  1747. break;
  1748. case NodeMenuOptions::DUPLICATE:
  1749. _duplicate_nodes();
  1750. break;
  1751. }
  1752. }
  1753. Variant VisualShaderEditor::get_drag_data_fw(const Point2 &p_point, Control *p_from) {
  1754. if (p_from == members) {
  1755. TreeItem *it = members->get_item_at_position(p_point);
  1756. if (!it) {
  1757. return Variant();
  1758. }
  1759. if (!it->has_meta("id")) {
  1760. return Variant();
  1761. }
  1762. int id = it->get_meta("id");
  1763. AddOption op = add_options[id];
  1764. Dictionary d;
  1765. d["id"] = id;
  1766. if (op.sub_func == -1) {
  1767. d["sub_func"] = op.sub_func_str;
  1768. } else {
  1769. d["sub_func"] = op.sub_func;
  1770. }
  1771. Label *label = memnew(Label);
  1772. label->set_text(it->get_text(0));
  1773. set_drag_preview(label);
  1774. return d;
  1775. }
  1776. return Variant();
  1777. }
  1778. bool VisualShaderEditor::can_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) const {
  1779. if (p_from == graph) {
  1780. Dictionary d = p_data;
  1781. if (d.has("id")) {
  1782. return true;
  1783. }
  1784. if (d.has("files")) {
  1785. return true;
  1786. }
  1787. }
  1788. return false;
  1789. }
  1790. void VisualShaderEditor::drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) {
  1791. if (p_from == graph) {
  1792. Dictionary d = p_data;
  1793. if (d.has("id")) {
  1794. int idx = d["id"];
  1795. saved_node_pos = p_point;
  1796. saved_node_pos_dirty = true;
  1797. _add_node(idx, add_options[idx].sub_func);
  1798. } else if (d.has("files")) {
  1799. if (d["files"].get_type() == Variant::PACKED_STRING_ARRAY) {
  1800. int j = 0;
  1801. PackedStringArray arr = d["files"];
  1802. for (int i = 0; i < arr.size(); i++) {
  1803. String type = ResourceLoader::get_resource_type(arr[i]);
  1804. if (type == "GDScript") {
  1805. Ref<Script> script = ResourceLoader::load(arr[i]);
  1806. if (script->get_instance_base_type() == "VisualShaderNodeCustom") {
  1807. saved_node_pos = p_point + Vector2(0, j * 210 * EDSCALE);
  1808. saved_node_pos_dirty = true;
  1809. _add_custom_node(arr[i]);
  1810. j++;
  1811. }
  1812. } else if (ClassDB::get_parent_class(type) == "Texture2D") {
  1813. saved_node_pos = p_point + Vector2(0, j * 210 * EDSCALE);
  1814. saved_node_pos_dirty = true;
  1815. _add_texture_node(arr[i]);
  1816. j++;
  1817. }
  1818. }
  1819. }
  1820. }
  1821. }
  1822. }
  1823. void VisualShaderEditor::_show_preview_text() {
  1824. preview_showed = !preview_showed;
  1825. preview_vbox->set_visible(preview_showed);
  1826. if (preview_showed) {
  1827. if (pending_update_preview) {
  1828. _update_preview();
  1829. pending_update_preview = false;
  1830. }
  1831. }
  1832. }
  1833. static ShaderLanguage::DataType _get_global_variable_type(const StringName &p_variable) {
  1834. RS::GlobalVariableType gvt = RS::get_singleton()->global_variable_get_type(p_variable);
  1835. return RS::global_variable_type_get_shader_datatype(gvt);
  1836. }
  1837. void VisualShaderEditor::_update_preview() {
  1838. if (!preview_showed) {
  1839. pending_update_preview = true;
  1840. return;
  1841. }
  1842. String code = visual_shader->get_code();
  1843. preview_text->set_text(code);
  1844. ShaderLanguage sl;
  1845. Error err = sl.compile(code, ShaderTypes::get_singleton()->get_functions(RenderingServer::ShaderMode(visual_shader->get_mode())), ShaderTypes::get_singleton()->get_modes(RenderingServer::ShaderMode(visual_shader->get_mode())), ShaderTypes::get_singleton()->get_types(), _get_global_variable_type);
  1846. for (int i = 0; i < preview_text->get_line_count(); i++) {
  1847. preview_text->set_line_as_marked(i, false);
  1848. }
  1849. if (err != OK) {
  1850. preview_text->set_line_as_marked(sl.get_error_line() - 1, true);
  1851. error_text->set_visible(true);
  1852. String text = "error(" + itos(sl.get_error_line()) + "): " + sl.get_error_text();
  1853. error_text->set_text(text);
  1854. shader_error = true;
  1855. } else {
  1856. error_text->set_visible(false);
  1857. shader_error = false;
  1858. }
  1859. }
  1860. void VisualShaderEditor::_bind_methods() {
  1861. ClassDB::bind_method("_rebuild", &VisualShaderEditor::_rebuild);
  1862. ClassDB::bind_method("_update_graph", &VisualShaderEditor::_update_graph);
  1863. ClassDB::bind_method("_update_options_menu", &VisualShaderEditor::_update_options_menu);
  1864. ClassDB::bind_method("_add_node", &VisualShaderEditor::_add_node);
  1865. ClassDB::bind_method("_node_changed", &VisualShaderEditor::_node_changed);
  1866. ClassDB::bind_method("_input_select_item", &VisualShaderEditor::_input_select_item);
  1867. ClassDB::bind_method("_set_node_size", &VisualShaderEditor::_set_node_size);
  1868. ClassDB::bind_method("_clear_buffer", &VisualShaderEditor::_clear_buffer);
  1869. ClassDB::bind_method(D_METHOD("get_drag_data_fw"), &VisualShaderEditor::get_drag_data_fw);
  1870. ClassDB::bind_method(D_METHOD("can_drop_data_fw"), &VisualShaderEditor::can_drop_data_fw);
  1871. ClassDB::bind_method(D_METHOD("drop_data_fw"), &VisualShaderEditor::drop_data_fw);
  1872. ClassDB::bind_method("_is_available", &VisualShaderEditor::_is_available);
  1873. }
  1874. VisualShaderEditor *VisualShaderEditor::singleton = nullptr;
  1875. VisualShaderEditor::VisualShaderEditor() {
  1876. singleton = this;
  1877. updating = false;
  1878. saved_node_pos_dirty = false;
  1879. saved_node_pos = Point2(0, 0);
  1880. ShaderLanguage::get_keyword_list(&keyword_list);
  1881. preview_showed = false;
  1882. pending_update_preview = false;
  1883. shader_error = false;
  1884. to_node = -1;
  1885. to_slot = -1;
  1886. from_node = -1;
  1887. from_slot = -1;
  1888. main_box = memnew(HSplitContainer);
  1889. main_box->set_v_size_flags(SIZE_EXPAND_FILL);
  1890. main_box->set_h_size_flags(SIZE_EXPAND_FILL);
  1891. add_child(main_box);
  1892. graph = memnew(GraphEdit);
  1893. graph->get_zoom_hbox()->set_h_size_flags(SIZE_EXPAND_FILL);
  1894. graph->set_v_size_flags(SIZE_EXPAND_FILL);
  1895. graph->set_h_size_flags(SIZE_EXPAND_FILL);
  1896. main_box->add_child(graph);
  1897. graph->set_drag_forwarding(this);
  1898. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_SCALAR);
  1899. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_SCALAR_INT);
  1900. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_BOOLEAN);
  1901. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_VECTOR);
  1902. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_TRANSFORM);
  1903. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_SAMPLER);
  1904. //graph->add_valid_left_disconnect_type(0);
  1905. graph->set_v_size_flags(SIZE_EXPAND_FILL);
  1906. graph->connect("connection_request", callable_mp(this, &VisualShaderEditor::_connection_request), varray(), CONNECT_DEFERRED);
  1907. graph->connect("disconnection_request", callable_mp(this, &VisualShaderEditor::_disconnection_request), varray(), CONNECT_DEFERRED);
  1908. graph->connect("node_selected", callable_mp(this, &VisualShaderEditor::_node_selected));
  1909. graph->connect("scroll_offset_changed", callable_mp(this, &VisualShaderEditor::_scroll_changed));
  1910. graph->connect("duplicate_nodes_request", callable_mp(this, &VisualShaderEditor::_duplicate_nodes));
  1911. graph->connect("copy_nodes_request", callable_mp(this, &VisualShaderEditor::_copy_nodes));
  1912. graph->connect("paste_nodes_request", callable_mp(this, &VisualShaderEditor::_paste_nodes));
  1913. graph->connect("delete_nodes_request", callable_mp(this, &VisualShaderEditor::_delete_nodes));
  1914. graph->connect("gui_input", callable_mp(this, &VisualShaderEditor::_graph_gui_input));
  1915. graph->connect("connection_to_empty", callable_mp(this, &VisualShaderEditor::_connection_to_empty));
  1916. graph->connect("connection_from_empty", callable_mp(this, &VisualShaderEditor::_connection_from_empty));
  1917. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR);
  1918. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR_INT);
  1919. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_VECTOR);
  1920. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_BOOLEAN);
  1921. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShaderNode::PORT_TYPE_SCALAR_INT);
  1922. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShaderNode::PORT_TYPE_SCALAR);
  1923. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShaderNode::PORT_TYPE_VECTOR);
  1924. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShaderNode::PORT_TYPE_BOOLEAN);
  1925. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_SCALAR);
  1926. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_SCALAR_INT);
  1927. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR);
  1928. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_BOOLEAN);
  1929. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_SCALAR);
  1930. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_SCALAR_INT);
  1931. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_VECTOR);
  1932. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_BOOLEAN);
  1933. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_TRANSFORM, VisualShaderNode::PORT_TYPE_TRANSFORM);
  1934. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SAMPLER, VisualShaderNode::PORT_TYPE_SAMPLER);
  1935. VSeparator *vs = memnew(VSeparator);
  1936. graph->get_zoom_hbox()->add_child(vs);
  1937. graph->get_zoom_hbox()->move_child(vs, 0);
  1938. edit_type = memnew(OptionButton);
  1939. edit_type->add_item(TTR("Vertex"));
  1940. edit_type->add_item(TTR("Fragment"));
  1941. edit_type->add_item(TTR("Light"));
  1942. edit_type->select(1);
  1943. edit_type->connect("item_selected", callable_mp(this, &VisualShaderEditor::_mode_selected));
  1944. graph->get_zoom_hbox()->add_child(edit_type);
  1945. graph->get_zoom_hbox()->move_child(edit_type, 0);
  1946. add_node = memnew(Button);
  1947. add_node->set_flat(true);
  1948. graph->get_zoom_hbox()->add_child(add_node);
  1949. add_node->set_text(TTR("Add Node..."));
  1950. graph->get_zoom_hbox()->move_child(add_node, 0);
  1951. add_node->connect("pressed", callable_mp(this, &VisualShaderEditor::_show_members_dialog), varray(false));
  1952. preview_shader = memnew(Button);
  1953. preview_shader->set_flat(true);
  1954. preview_shader->set_toggle_mode(true);
  1955. preview_shader->set_tooltip(TTR("Show resulted shader code."));
  1956. graph->get_zoom_hbox()->add_child(preview_shader);
  1957. preview_shader->connect("pressed", callable_mp(this, &VisualShaderEditor::_show_preview_text));
  1958. ///////////////////////////////////////
  1959. // PREVIEW PANEL
  1960. ///////////////////////////////////////
  1961. preview_vbox = memnew(VBoxContainer);
  1962. preview_vbox->set_visible(preview_showed);
  1963. main_box->add_child(preview_vbox);
  1964. preview_text = memnew(TextEdit);
  1965. syntax_highlighter.instance();
  1966. preview_vbox->add_child(preview_text);
  1967. preview_text->set_h_size_flags(SIZE_EXPAND_FILL);
  1968. preview_text->set_v_size_flags(SIZE_EXPAND_FILL);
  1969. preview_text->set_custom_minimum_size(Size2(400 * EDSCALE, 0));
  1970. preview_text->set_syntax_highlighter(syntax_highlighter);
  1971. preview_text->set_show_line_numbers(true);
  1972. preview_text->set_readonly(true);
  1973. error_text = memnew(Label);
  1974. preview_vbox->add_child(error_text);
  1975. error_text->set_visible(false);
  1976. ///////////////////////////////////////
  1977. // POPUP MENU
  1978. ///////////////////////////////////////
  1979. popup_menu = memnew(PopupMenu);
  1980. add_child(popup_menu);
  1981. popup_menu->add_item("Add Node", NodeMenuOptions::ADD);
  1982. popup_menu->add_separator();
  1983. popup_menu->add_item("Copy", NodeMenuOptions::COPY);
  1984. popup_menu->add_item("Paste", NodeMenuOptions::PASTE);
  1985. popup_menu->add_item("Delete", NodeMenuOptions::DELETE);
  1986. popup_menu->add_item("Duplicate", NodeMenuOptions::DUPLICATE);
  1987. popup_menu->connect("id_pressed", callable_mp(this, &VisualShaderEditor::_node_menu_id_pressed));
  1988. ///////////////////////////////////////
  1989. // SHADER NODES TREE
  1990. ///////////////////////////////////////
  1991. VBoxContainer *members_vb = memnew(VBoxContainer);
  1992. members_vb->set_v_size_flags(SIZE_EXPAND_FILL);
  1993. HBoxContainer *filter_hb = memnew(HBoxContainer);
  1994. members_vb->add_child(filter_hb);
  1995. node_filter = memnew(LineEdit);
  1996. filter_hb->add_child(node_filter);
  1997. node_filter->connect("text_changed", callable_mp(this, &VisualShaderEditor::_member_filter_changed));
  1998. node_filter->connect("gui_input", callable_mp(this, &VisualShaderEditor::_sbox_input));
  1999. node_filter->set_h_size_flags(SIZE_EXPAND_FILL);
  2000. node_filter->set_placeholder(TTR("Search"));
  2001. tools = memnew(MenuButton);
  2002. filter_hb->add_child(tools);
  2003. tools->set_tooltip(TTR("Options"));
  2004. tools->get_popup()->connect("id_pressed", callable_mp(this, &VisualShaderEditor::_tools_menu_option));
  2005. tools->get_popup()->add_item(TTR("Expand All"), EXPAND_ALL);
  2006. tools->get_popup()->add_item(TTR("Collapse All"), COLLAPSE_ALL);
  2007. members = memnew(Tree);
  2008. members_vb->add_child(members);
  2009. members->set_drag_forwarding(this);
  2010. members->set_h_size_flags(SIZE_EXPAND_FILL);
  2011. members->set_v_size_flags(SIZE_EXPAND_FILL);
  2012. members->set_hide_root(true);
  2013. members->set_allow_reselect(true);
  2014. members->set_hide_folding(false);
  2015. members->set_custom_minimum_size(Size2(180 * EDSCALE, 200 * EDSCALE));
  2016. members->connect("item_activated", callable_mp(this, &VisualShaderEditor::_member_create));
  2017. members->connect("item_selected", callable_mp(this, &VisualShaderEditor::_member_selected));
  2018. members->connect("nothing_selected", callable_mp(this, &VisualShaderEditor::_member_unselected));
  2019. HBoxContainer *desc_hbox = memnew(HBoxContainer);
  2020. members_vb->add_child(desc_hbox);
  2021. Label *desc_label = memnew(Label);
  2022. desc_hbox->add_child(desc_label);
  2023. desc_label->set_text(TTR("Description:"));
  2024. desc_hbox->add_spacer();
  2025. highend_label = memnew(Label);
  2026. desc_hbox->add_child(highend_label);
  2027. highend_label->set_visible(false);
  2028. highend_label->set_text("Vulkan");
  2029. highend_label->set_mouse_filter(Control::MOUSE_FILTER_STOP);
  2030. highend_label->set_tooltip(TTR("High-end node"));
  2031. node_desc = memnew(RichTextLabel);
  2032. members_vb->add_child(node_desc);
  2033. node_desc->set_h_size_flags(SIZE_EXPAND_FILL);
  2034. node_desc->set_v_size_flags(SIZE_FILL);
  2035. node_desc->set_custom_minimum_size(Size2(0, 70 * EDSCALE));
  2036. members_dialog = memnew(ConfirmationDialog);
  2037. members_dialog->set_title(TTR("Create Shader Node"));
  2038. members_dialog->set_exclusive(false);
  2039. members_dialog->add_child(members_vb);
  2040. members_dialog->get_ok()->set_text(TTR("Create"));
  2041. members_dialog->get_ok()->connect("pressed", callable_mp(this, &VisualShaderEditor::_member_create));
  2042. members_dialog->get_ok()->set_disabled(true);
  2043. members_dialog->connect("cancelled", callable_mp(this, &VisualShaderEditor::_member_cancel));
  2044. add_child(members_dialog);
  2045. alert = memnew(AcceptDialog);
  2046. alert->get_label()->set_autowrap(true);
  2047. alert->get_label()->set_align(Label::ALIGN_CENTER);
  2048. alert->get_label()->set_valign(Label::VALIGN_CENTER);
  2049. alert->get_label()->set_custom_minimum_size(Size2(400, 60) * EDSCALE);
  2050. add_child(alert);
  2051. ///////////////////////////////////////
  2052. // SHADER NODES TREE OPTIONS
  2053. ///////////////////////////////////////
  2054. // COLOR
  2055. add_options.push_back(AddOption("ColorFunc", "Color", "Common", "VisualShaderNodeColorFunc", TTR("Color function."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2056. add_options.push_back(AddOption("ColorOp", "Color", "Common", "VisualShaderNodeColorOp", TTR("Color operator."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2057. add_options.push_back(AddOption("Grayscale", "Color", "Functions", "VisualShaderNodeColorFunc", TTR("Grayscale function."), VisualShaderNodeColorFunc::FUNC_GRAYSCALE, VisualShaderNode::PORT_TYPE_VECTOR));
  2058. add_options.push_back(AddOption("HSV2RGB", "Color", "Functions", "VisualShaderNodeVectorFunc", TTR("Converts HSV vector to RGB equivalent."), VisualShaderNodeVectorFunc::FUNC_HSV2RGB, VisualShaderNode::PORT_TYPE_VECTOR));
  2059. add_options.push_back(AddOption("RGB2HSV", "Color", "Functions", "VisualShaderNodeVectorFunc", TTR("Converts RGB vector to HSV equivalent."), VisualShaderNodeVectorFunc::FUNC_RGB2HSV, VisualShaderNode::PORT_TYPE_VECTOR));
  2060. add_options.push_back(AddOption("Sepia", "Color", "Functions", "VisualShaderNodeColorFunc", TTR("Sepia function."), VisualShaderNodeColorFunc::FUNC_SEPIA, VisualShaderNode::PORT_TYPE_VECTOR));
  2061. add_options.push_back(AddOption("Burn", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Burn operator."), VisualShaderNodeColorOp::OP_BURN, VisualShaderNode::PORT_TYPE_VECTOR));
  2062. add_options.push_back(AddOption("Darken", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Darken operator."), VisualShaderNodeColorOp::OP_DARKEN, VisualShaderNode::PORT_TYPE_VECTOR));
  2063. add_options.push_back(AddOption("Difference", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Difference operator."), VisualShaderNodeColorOp::OP_DIFFERENCE, VisualShaderNode::PORT_TYPE_VECTOR));
  2064. add_options.push_back(AddOption("Dodge", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Dodge operator."), VisualShaderNodeColorOp::OP_DODGE, VisualShaderNode::PORT_TYPE_VECTOR));
  2065. add_options.push_back(AddOption("HardLight", "Color", "Operators", "VisualShaderNodeColorOp", TTR("HardLight operator."), VisualShaderNodeColorOp::OP_HARD_LIGHT, VisualShaderNode::PORT_TYPE_VECTOR));
  2066. add_options.push_back(AddOption("Lighten", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Lighten operator."), VisualShaderNodeColorOp::OP_LIGHTEN, VisualShaderNode::PORT_TYPE_VECTOR));
  2067. add_options.push_back(AddOption("Overlay", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Overlay operator."), VisualShaderNodeColorOp::OP_OVERLAY, VisualShaderNode::PORT_TYPE_VECTOR));
  2068. add_options.push_back(AddOption("Screen", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Screen operator."), VisualShaderNodeColorOp::OP_SCREEN, VisualShaderNode::PORT_TYPE_VECTOR));
  2069. add_options.push_back(AddOption("SoftLight", "Color", "Operators", "VisualShaderNodeColorOp", TTR("SoftLight operator."), VisualShaderNodeColorOp::OP_SOFT_LIGHT, VisualShaderNode::PORT_TYPE_VECTOR));
  2070. add_options.push_back(AddOption("ColorConstant", "Color", "Variables", "VisualShaderNodeColorConstant", TTR("Color constant."), -1, -1));
  2071. add_options.push_back(AddOption("ColorUniform", "Color", "Variables", "VisualShaderNodeColorUniform", TTR("Color uniform."), -1, -1));
  2072. // CONDITIONAL
  2073. const String &compare_func_desc = TTR("Returns the boolean result of the %s comparison between two parameters.");
  2074. add_options.push_back(AddOption("Equal", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Equal (==)")), VisualShaderNodeCompare::FUNC_EQUAL, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2075. add_options.push_back(AddOption("GreaterThan", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Greater Than (>)")), VisualShaderNodeCompare::FUNC_GREATER_THAN, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2076. add_options.push_back(AddOption("GreaterThanEqual", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Greater Than or Equal (>=)")), VisualShaderNodeCompare::FUNC_GREATER_THAN_EQUAL, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2077. add_options.push_back(AddOption("If", "Conditional", "Functions", "VisualShaderNodeIf", TTR("Returns an associated vector if the provided scalars are equal, greater or less."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2078. add_options.push_back(AddOption("IsInf", "Conditional", "Functions", "VisualShaderNodeIs", TTR("Returns the boolean result of the comparison between INF and a scalar parameter."), VisualShaderNodeIs::FUNC_IS_INF, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2079. add_options.push_back(AddOption("IsNaN", "Conditional", "Functions", "VisualShaderNodeIs", TTR("Returns the boolean result of the comparison between NaN and a scalar parameter."), VisualShaderNodeIs::FUNC_IS_NAN, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2080. add_options.push_back(AddOption("LessThan", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Less Than (<)")), VisualShaderNodeCompare::FUNC_LESS_THAN, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2081. add_options.push_back(AddOption("LessThanEqual", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Less Than or Equal (<=)")), VisualShaderNodeCompare::FUNC_LESS_THAN_EQUAL, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2082. add_options.push_back(AddOption("NotEqual", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Not Equal (!=)")), VisualShaderNodeCompare::FUNC_NOT_EQUAL, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2083. add_options.push_back(AddOption("Switch", "Conditional", "Functions", "VisualShaderNodeSwitch", TTR("Returns an associated vector if the provided boolean value is true or false."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2084. add_options.push_back(AddOption("SwitchS", "Conditional", "Functions", "VisualShaderNodeScalarSwitch", TTR("Returns an associated scalar if the provided boolean value is true or false."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2085. add_options.push_back(AddOption("Compare", "Conditional", "Common", "VisualShaderNodeCompare", TTR("Returns the boolean result of the comparison between two parameters."), -1, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2086. add_options.push_back(AddOption("Is", "Conditional", "Common", "VisualShaderNodeIs", TTR("Returns the boolean result of the comparison between INF (or NaN) and a scalar parameter."), -1, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2087. add_options.push_back(AddOption("BooleanConstant", "Conditional", "Variables", "VisualShaderNodeBooleanConstant", TTR("Boolean constant."), -1, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2088. add_options.push_back(AddOption("BooleanUniform", "Conditional", "Variables", "VisualShaderNodeBooleanUniform", TTR("Boolean uniform."), -1, VisualShaderNode::PORT_TYPE_BOOLEAN));
  2089. // INPUT
  2090. // SPATIAL-FOR-ALL
  2091. const String input_param_shader_modes = TTR("'%s' input parameter for all shader modes.");
  2092. add_options.push_back(AddOption("Camera", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "camera"), "camera", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_SPATIAL));
  2093. add_options.push_back(AddOption("InvCamera", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "inv_camera"), "inv_camera", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_SPATIAL));
  2094. add_options.push_back(AddOption("InvProjection", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "inv_projection"), "inv_projection", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_SPATIAL));
  2095. add_options.push_back(AddOption("Normal", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "normal"), "normal", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_SPATIAL));
  2096. add_options.push_back(AddOption("OutputIsSRGB", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "output_is_srgb"), "output_is_srgb", VisualShaderNode::PORT_TYPE_BOOLEAN, -1, Shader::MODE_SPATIAL));
  2097. add_options.push_back(AddOption("Projection", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "camera"), "projection", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_SPATIAL));
  2098. add_options.push_back(AddOption("Time", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "time"), "time", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_SPATIAL));
  2099. add_options.push_back(AddOption("ViewportSize", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "viewport_size"), "viewport_size", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_SPATIAL));
  2100. add_options.push_back(AddOption("World", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "world"), "world", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_SPATIAL));
  2101. // CANVASITEM-FOR-ALL
  2102. add_options.push_back(AddOption("Alpha", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "alpha"), "alpha", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_CANVAS_ITEM));
  2103. add_options.push_back(AddOption("Color", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "color"), "color", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_CANVAS_ITEM));
  2104. add_options.push_back(AddOption("TexturePixelSize", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "texture_pixel_size"), "texture_pixel_size", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_CANVAS_ITEM));
  2105. add_options.push_back(AddOption("Time", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "time"), "time", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_CANVAS_ITEM));
  2106. add_options.push_back(AddOption("UV", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "uv"), "uv", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_CANVAS_ITEM));
  2107. /////////////////
  2108. add_options.push_back(AddOption("Input", "Input", "Common", "VisualShaderNodeInput", TTR("Input parameter.")));
  2109. // SPATIAL INPUTS
  2110. const String input_param_for_vertex_and_fragment_shader_modes = TTR("'%s' input parameter for vertex and fragment shader modes.");
  2111. const String input_param_for_fragment_and_light_shader_modes = TTR("'%s' input parameter for fragment and light shader modes.");
  2112. const String input_param_for_fragment_shader_mode = TTR("'%s' input parameter for fragment shader mode.");
  2113. const String input_param_for_light_shader_mode = TTR("'%s' input parameter for light shader mode.");
  2114. const String input_param_for_vertex_shader_mode = TTR("'%s' input parameter for vertex shader mode.");
  2115. const String input_param_for_vertex_and_fragment_shader_mode = TTR("'%s' input parameter for vertex and fragment shader mode.");
  2116. add_options.push_back(AddOption("Alpha", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "alpha"), "alpha", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2117. add_options.push_back(AddOption("Binormal", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "binormal"), "binormal", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2118. add_options.push_back(AddOption("Color", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "color"), "color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2119. add_options.push_back(AddOption("DepthTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "depth_texture"), "depth_texture", VisualShaderNode::PORT_TYPE_SAMPLER, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2120. add_options.push_back(AddOption("FragCoord", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "fragcoord"), "fragcoord", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2121. add_options.push_back(AddOption("FrontFacing", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "front_facing"), "front_facing", VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2122. add_options.push_back(AddOption("PointCoord", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "point_coord"), "point_coord", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2123. add_options.push_back(AddOption("ScreenTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "screen_texture"), "screen_texture", VisualShaderNode::PORT_TYPE_SAMPLER, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2124. add_options.push_back(AddOption("ScreenUV", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "screen_uv"), "screen_uv", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2125. add_options.push_back(AddOption("Side", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "side"), "side", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2126. add_options.push_back(AddOption("Tangent", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "tangent"), "tangent", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2127. add_options.push_back(AddOption("UV", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "uv"), "uv", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2128. add_options.push_back(AddOption("UV2", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "uv2"), "uv2", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2129. add_options.push_back(AddOption("Vertex", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "vertex"), "vertex", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2130. add_options.push_back(AddOption("View", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "view"), "view", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SPATIAL));
  2131. add_options.push_back(AddOption("Albedo", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "albedo"), "albedo", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2132. add_options.push_back(AddOption("Attenuation", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "attenuation"), "attenuation", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2133. add_options.push_back(AddOption("Diffuse", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "diffuse"), "diffuse", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2134. add_options.push_back(AddOption("FragCoord", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "fragcoord"), "fragcoord", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2135. add_options.push_back(AddOption("Light", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light"), "light", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2136. add_options.push_back(AddOption("LightColor", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_color"), "light_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2137. add_options.push_back(AddOption("Roughness", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "roughness"), "roughness", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2138. add_options.push_back(AddOption("Specular", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "specular"), "specular", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2139. add_options.push_back(AddOption("Transmission", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "transmission"), "transmission", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2140. add_options.push_back(AddOption("View", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "view"), "view", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2141. add_options.push_back(AddOption("Alpha", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "alpha"), "alpha", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2142. add_options.push_back(AddOption("Binormal", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "binormal"), "binormal", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2143. add_options.push_back(AddOption("Color", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "color"), "color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2144. add_options.push_back(AddOption("ModelView", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "modelview"), "modelview", VisualShaderNode::PORT_TYPE_TRANSFORM, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2145. add_options.push_back(AddOption("PointSize", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "point_size"), "point_size", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2146. add_options.push_back(AddOption("Tangent", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_mode, "tangent"), "tangent", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2147. add_options.push_back(AddOption("UV", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "uv"), "uv", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2148. add_options.push_back(AddOption("UV2", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "uv2"), "uv2", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2149. add_options.push_back(AddOption("Vertex", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "vertex"), "vertex", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_SPATIAL));
  2150. // CANVASITEM INPUTS
  2151. add_options.push_back(AddOption("FragCoord", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "fragcoord"), "fragcoord", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  2152. add_options.push_back(AddOption("LightPass", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "light_pass"), "light_pass", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  2153. add_options.push_back(AddOption("NormalTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "normal_texture"), "normal_texture", VisualShaderNode::PORT_TYPE_SAMPLER, VisualShader::TYPE_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  2154. add_options.push_back(AddOption("PointCoord", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "point_coord"), "point_coord", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  2155. add_options.push_back(AddOption("ScreenPixelSize", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "screen_pixel_size"), "screen_pixel_size", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  2156. add_options.push_back(AddOption("ScreenTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "screen_texture"), "screen_texture", VisualShaderNode::PORT_TYPE_SAMPLER, VisualShader::TYPE_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  2157. add_options.push_back(AddOption("ScreenUV", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "screen_uv"), "screen_uv", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  2158. add_options.push_back(AddOption("Texture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "texture"), "texture", VisualShaderNode::PORT_TYPE_SAMPLER, VisualShader::TYPE_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  2159. add_options.push_back(AddOption("FragCoord", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "fragcoord"), "fragcoord", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2160. add_options.push_back(AddOption("LightAlpha", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_alpha"), "light_alpha", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2161. add_options.push_back(AddOption("LightColor", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_color"), "light_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2162. add_options.push_back(AddOption("LightHeight", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_height"), "light_height", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2163. add_options.push_back(AddOption("LightUV", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_uv"), "light_uv", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2164. add_options.push_back(AddOption("LightVector", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_vec"), "light_vec", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2165. add_options.push_back(AddOption("Normal", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "normal"), "normal", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2166. add_options.push_back(AddOption("PointCoord", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "point_coord"), "point_coord", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2167. add_options.push_back(AddOption("ScreenUV", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "screen_uv"), "screen_uv", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2168. add_options.push_back(AddOption("ShadowAlpha", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "shadow_alpha"), "shadow_alpha", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2169. add_options.push_back(AddOption("ShadowColor", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "shadow_color"), "shadow_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2170. add_options.push_back(AddOption("ShadowVec", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "shadow_vec"), "shadow_vec", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2171. add_options.push_back(AddOption("Texture", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "texture"), "texture", VisualShaderNode::PORT_TYPE_SAMPLER, VisualShader::TYPE_LIGHT, Shader::MODE_CANVAS_ITEM));
  2172. add_options.push_back(AddOption("Extra", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "extra"), "extra", VisualShaderNode::PORT_TYPE_TRANSFORM, VisualShader::TYPE_VERTEX, Shader::MODE_CANVAS_ITEM));
  2173. add_options.push_back(AddOption("LightPass", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "light_pass"), "light_pass", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_CANVAS_ITEM));
  2174. add_options.push_back(AddOption("PointSize", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "point_size"), "point_size", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_CANVAS_ITEM));
  2175. add_options.push_back(AddOption("Projection", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "projection"), "projection", VisualShaderNode::PORT_TYPE_TRANSFORM, VisualShader::TYPE_VERTEX, Shader::MODE_CANVAS_ITEM));
  2176. add_options.push_back(AddOption("Vertex", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "vertex"), "vertex", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_CANVAS_ITEM));
  2177. add_options.push_back(AddOption("World", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "world"), "world", VisualShaderNode::PORT_TYPE_TRANSFORM, VisualShader::TYPE_VERTEX, Shader::MODE_CANVAS_ITEM));
  2178. // PARTICLES INPUTS
  2179. add_options.push_back(AddOption("Active", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "active"), "active", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2180. add_options.push_back(AddOption("Alpha", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "alpha"), "alpha", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2181. add_options.push_back(AddOption("Color", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "color"), "color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2182. add_options.push_back(AddOption("Custom", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "custom"), "custom", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2183. add_options.push_back(AddOption("CustomAlpha", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "custom_alpha"), "custom_alpha", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2184. add_options.push_back(AddOption("Delta", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "delta"), "delta", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2185. add_options.push_back(AddOption("EmissionTransform", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "emission_transform"), "emission_transform", VisualShaderNode::PORT_TYPE_TRANSFORM, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2186. add_options.push_back(AddOption("Index", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "index"), "index", VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2187. add_options.push_back(AddOption("LifeTime", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "lifetime"), "lifetime", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2188. add_options.push_back(AddOption("Restart", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "restart"), "restart", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2189. add_options.push_back(AddOption("Time", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "time"), "time", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2190. add_options.push_back(AddOption("Transform", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "transform"), "transform", VisualShaderNode::PORT_TYPE_TRANSFORM, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2191. add_options.push_back(AddOption("Velocity", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "velocity"), "velocity", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_VERTEX, Shader::MODE_PARTICLES));
  2192. // SKY INPUTS
  2193. add_options.push_back(AddOption("AtCubeMapPass", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "at_cubemap_pass"), "at_cubemap_pass", VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2194. add_options.push_back(AddOption("AtHalfResPass", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "at_half_res_pass"), "at_half_res_pass", VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2195. add_options.push_back(AddOption("AtQuarterResPass", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "at_quarter_res_pass"), "at_quarter_res_pass", VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2196. add_options.push_back(AddOption("EyeDir", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "eyedir"), "eyedir", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2197. add_options.push_back(AddOption("HalfResColor", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "half_res_color"), "half_res_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2198. add_options.push_back(AddOption("HalfResAlpha", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "half_res_alpha"), "half_res_alpha", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2199. add_options.push_back(AddOption("Light0Color", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light0_color"), "light0_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2200. add_options.push_back(AddOption("Light0Direction", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light0_direction"), "light0_direction", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2201. add_options.push_back(AddOption("Light0Enabled", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light0_enabled"), "light0_enabled", VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2202. add_options.push_back(AddOption("Light0Energy", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light0_energy"), "light0_energy", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2203. add_options.push_back(AddOption("Light1Color", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light1_color"), "light1_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2204. add_options.push_back(AddOption("Light1Direction", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light1_direction"), "light1_direction", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2205. add_options.push_back(AddOption("Light1Enabled", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light1_enabled"), "light1_enabled", VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2206. add_options.push_back(AddOption("Light1Energy", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light1_energy"), "light1_energy", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2207. add_options.push_back(AddOption("Light2Color", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light2_color"), "light2_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2208. add_options.push_back(AddOption("Light2Direction", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light2_direction"), "light2_direction", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2209. add_options.push_back(AddOption("Light2Enabled", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light2_enabled"), "light2_enabled", VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2210. add_options.push_back(AddOption("Light2Energy", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light2_energy"), "light2_energy", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2211. add_options.push_back(AddOption("Light3Color", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light3_color"), "light3_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2212. add_options.push_back(AddOption("Light3Direction", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light3_direction"), "light3_direction", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2213. add_options.push_back(AddOption("Light3Enabled", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light3_enabled"), "light3_enabled", VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2214. add_options.push_back(AddOption("Light3Energy", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "light3_energy"), "light3_energy", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2215. add_options.push_back(AddOption("Position", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "position"), "position", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2216. add_options.push_back(AddOption("QuarterResColor", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "quarter_res_color"), "quarter_res_color", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2217. add_options.push_back(AddOption("QuarterResAlpha", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "quarter_res_alpha"), "quarter_res_alpha", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2218. add_options.push_back(AddOption("Radiance", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "radiance"), "radiance", VisualShaderNode::PORT_TYPE_SAMPLER, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2219. add_options.push_back(AddOption("ScreenUV", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "screen_uv"), "screen_uv", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2220. add_options.push_back(AddOption("SkyCoords", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "sky_coords"), "sky_coords", VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2221. add_options.push_back(AddOption("Time", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "time"), "time", VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT, Shader::MODE_SKY));
  2222. // SCALAR
  2223. add_options.push_back(AddOption("FloatFunc", "Scalar", "Common", "VisualShaderNodeFloatFunc", TTR("Float function."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2224. add_options.push_back(AddOption("IntFunc", "Scalar", "Common", "VisualShaderNodeIntFunc", TTR("Integer function."), -1, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2225. add_options.push_back(AddOption("FloatOp", "Scalar", "Common", "VisualShaderNodeFloatOp", TTR("Float operator."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2226. add_options.push_back(AddOption("IntOp", "Scalar", "Common", "VisualShaderNodeIntOp", TTR("Integer operator."), -1, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2227. //CONSTANTS
  2228. add_options.push_back(AddOption("E", "Scalar", "Constants", "VisualShaderNodeFloatConstant", TTR("E constant (2.718282). Represents the base of the natural logarithm."), -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, Math_E));
  2229. add_options.push_back(AddOption("Epsilon", "Scalar", "Constants", "VisualShaderNodeFloatConstant", TTR("Epsilon constant (0.00001). Smallest possible scalar number."), -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, CMP_EPSILON));
  2230. add_options.push_back(AddOption("Phi", "Scalar", "Constants", "VisualShaderNodeFloatConstant", TTR("Phi constant (1.618034). Golden ratio."), -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, 1.618034f));
  2231. add_options.push_back(AddOption("Pi/4", "Scalar", "Constants", "VisualShaderNodeFloatConstant", TTR("Pi/4 constant (0.785398) or 45 degrees."), -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, Math_PI / 4));
  2232. add_options.push_back(AddOption("Pi/2", "Scalar", "Constants", "VisualShaderNodeFloatConstant", TTR("Pi/2 constant (1.570796) or 90 degrees."), -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, Math_PI / 2));
  2233. add_options.push_back(AddOption("Pi", "Scalar", "Constants", "VisualShaderNodeFloatConstant", TTR("Pi constant (3.141593) or 180 degrees."), -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, Math_PI));
  2234. add_options.push_back(AddOption("Tau", "Scalar", "Constants", "VisualShaderNodeFloatConstant", TTR("Tau constant (6.283185) or 360 degrees."), -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, Math_TAU));
  2235. add_options.push_back(AddOption("Sqrt2", "Scalar", "Constants", "VisualShaderNodeFloatConstant", TTR("Sqrt2 constant (1.414214). Square root of 2."), -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, Math_SQRT2));
  2236. // FUNCTIONS
  2237. add_options.push_back(AddOption("Abs", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the absolute value of the parameter."), VisualShaderNodeFloatFunc::FUNC_ABS, VisualShaderNode::PORT_TYPE_SCALAR));
  2238. add_options.push_back(AddOption("Abs", "Scalar", "Functions", "VisualShaderNodeIntFunc", TTR("Returns the absolute value of the parameter."), VisualShaderNodeIntFunc::FUNC_ABS, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2239. add_options.push_back(AddOption("ACos", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the arc-cosine of the parameter."), VisualShaderNodeFloatFunc::FUNC_ACOS, VisualShaderNode::PORT_TYPE_SCALAR));
  2240. add_options.push_back(AddOption("ACosH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the inverse hyperbolic cosine of the parameter."), VisualShaderNodeFloatFunc::FUNC_ACOSH, VisualShaderNode::PORT_TYPE_SCALAR));
  2241. add_options.push_back(AddOption("ASin", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the arc-sine of the parameter."), VisualShaderNodeFloatFunc::FUNC_ASIN, VisualShaderNode::PORT_TYPE_SCALAR));
  2242. add_options.push_back(AddOption("ASinH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the inverse hyperbolic sine of the parameter."), VisualShaderNodeFloatFunc::FUNC_ASINH, VisualShaderNode::PORT_TYPE_SCALAR));
  2243. add_options.push_back(AddOption("ATan", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the arc-tangent of the parameter."), VisualShaderNodeFloatFunc::FUNC_ATAN, VisualShaderNode::PORT_TYPE_SCALAR));
  2244. add_options.push_back(AddOption("ATan2", "Scalar", "Functions", "VisualShaderNodeFloatOp", TTR("Returns the arc-tangent of the parameters."), VisualShaderNodeFloatOp::OP_ATAN2, VisualShaderNode::PORT_TYPE_SCALAR));
  2245. add_options.push_back(AddOption("ATanH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the inverse hyperbolic tangent of the parameter."), VisualShaderNodeFloatFunc::FUNC_ATANH, VisualShaderNode::PORT_TYPE_SCALAR));
  2246. add_options.push_back(AddOption("Ceil", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Finds the nearest integer that is greater than or equal to the parameter."), VisualShaderNodeFloatFunc::FUNC_CEIL, VisualShaderNode::PORT_TYPE_SCALAR));
  2247. add_options.push_back(AddOption("Clamp", "Scalar", "Functions", "VisualShaderNodeScalarClamp", TTR("Constrains a value to lie between two further values."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2248. add_options.push_back(AddOption("Clamp", "Scalar", "Functions", "VisualShaderNodeIntFunc", TTR("Constrains a value to lie between two further values."), VisualShaderNodeIntFunc::FUNC_CLAMP, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2249. add_options.push_back(AddOption("Cos", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the cosine of the parameter."), VisualShaderNodeFloatFunc::FUNC_COS, VisualShaderNode::PORT_TYPE_SCALAR));
  2250. add_options.push_back(AddOption("CosH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the hyperbolic cosine of the parameter."), VisualShaderNodeFloatFunc::FUNC_COSH, VisualShaderNode::PORT_TYPE_SCALAR));
  2251. add_options.push_back(AddOption("Degrees", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Converts a quantity in radians to degrees."), VisualShaderNodeFloatFunc::FUNC_DEGREES, VisualShaderNode::PORT_TYPE_SCALAR));
  2252. add_options.push_back(AddOption("Exp", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Base-e Exponential."), VisualShaderNodeFloatFunc::FUNC_EXP, VisualShaderNode::PORT_TYPE_SCALAR));
  2253. add_options.push_back(AddOption("Exp2", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Base-2 Exponential."), VisualShaderNodeFloatFunc::FUNC_EXP2, VisualShaderNode::PORT_TYPE_SCALAR));
  2254. add_options.push_back(AddOption("Floor", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Finds the nearest integer less than or equal to the parameter."), VisualShaderNodeFloatFunc::FUNC_FLOOR, VisualShaderNode::PORT_TYPE_SCALAR));
  2255. add_options.push_back(AddOption("Fract", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Computes the fractional part of the argument."), VisualShaderNodeFloatFunc::FUNC_FRAC, VisualShaderNode::PORT_TYPE_SCALAR));
  2256. add_options.push_back(AddOption("InverseSqrt", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the inverse of the square root of the parameter."), VisualShaderNodeFloatFunc::FUNC_INVERSE_SQRT, VisualShaderNode::PORT_TYPE_SCALAR));
  2257. add_options.push_back(AddOption("Log", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Natural logarithm."), VisualShaderNodeFloatFunc::FUNC_LOG, VisualShaderNode::PORT_TYPE_SCALAR));
  2258. add_options.push_back(AddOption("Log2", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Base-2 logarithm."), VisualShaderNodeFloatFunc::FUNC_LOG2, VisualShaderNode::PORT_TYPE_SCALAR));
  2259. add_options.push_back(AddOption("Max", "Scalar", "Functions", "VisualShaderNodeFloatOp", TTR("Returns the greater of two values."), VisualShaderNodeFloatOp::OP_MAX, VisualShaderNode::PORT_TYPE_SCALAR));
  2260. add_options.push_back(AddOption("Min", "Scalar", "Functions", "VisualShaderNodeFloatOp", TTR("Returns the lesser of two values."), VisualShaderNodeFloatOp::OP_MIN, VisualShaderNode::PORT_TYPE_SCALAR));
  2261. add_options.push_back(AddOption("Mix", "Scalar", "Functions", "VisualShaderNodeScalarInterp", TTR("Linear interpolation between two scalars."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2262. add_options.push_back(AddOption("MultiplyAdd", "Scalar", "Functions", "VisualShaderNodeMultiplyAdd", TTR("Performs a fused multiply-add operation (a * b + c) on scalars."), VisualShaderNodeMultiplyAdd::TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR));
  2263. add_options.push_back(AddOption("Negate", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the opposite value of the parameter."), VisualShaderNodeFloatFunc::FUNC_NEGATE, VisualShaderNode::PORT_TYPE_SCALAR));
  2264. add_options.push_back(AddOption("Negate", "Scalar", "Functions", "VisualShaderNodeIntFunc", TTR("Returns the opposite value of the parameter."), VisualShaderNodeIntFunc::FUNC_NEGATE, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2265. add_options.push_back(AddOption("OneMinus", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("1.0 - scalar"), VisualShaderNodeFloatFunc::FUNC_ONEMINUS, VisualShaderNode::PORT_TYPE_SCALAR));
  2266. add_options.push_back(AddOption("Pow", "Scalar", "Functions", "VisualShaderNodeFloatOp", TTR("Returns the value of the first parameter raised to the power of the second."), VisualShaderNodeFloatOp::OP_POW, VisualShaderNode::PORT_TYPE_SCALAR));
  2267. add_options.push_back(AddOption("Radians", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Converts a quantity in degrees to radians."), VisualShaderNodeFloatFunc::FUNC_RADIANS, VisualShaderNode::PORT_TYPE_SCALAR));
  2268. add_options.push_back(AddOption("Reciprocal", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("1.0 / scalar"), VisualShaderNodeFloatFunc::FUNC_RECIPROCAL, VisualShaderNode::PORT_TYPE_SCALAR));
  2269. add_options.push_back(AddOption("Round", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Finds the nearest integer to the parameter."), VisualShaderNodeFloatFunc::FUNC_ROUND, VisualShaderNode::PORT_TYPE_SCALAR));
  2270. add_options.push_back(AddOption("RoundEven", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Finds the nearest even integer to the parameter."), VisualShaderNodeFloatFunc::FUNC_ROUNDEVEN, VisualShaderNode::PORT_TYPE_SCALAR));
  2271. add_options.push_back(AddOption("Saturate", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Clamps the value between 0.0 and 1.0."), VisualShaderNodeFloatFunc::FUNC_SATURATE, VisualShaderNode::PORT_TYPE_SCALAR));
  2272. add_options.push_back(AddOption("Sign", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Extracts the sign of the parameter."), VisualShaderNodeFloatFunc::FUNC_SIGN, VisualShaderNode::PORT_TYPE_SCALAR));
  2273. add_options.push_back(AddOption("Sign", "Scalar", "Functions", "VisualShaderNodeIntFunc", TTR("Extracts the sign of the parameter."), VisualShaderNodeIntFunc::FUNC_SIGN, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2274. add_options.push_back(AddOption("Sin", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the sine of the parameter."), VisualShaderNodeFloatFunc::FUNC_SIN, VisualShaderNode::PORT_TYPE_SCALAR));
  2275. add_options.push_back(AddOption("SinH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the hyperbolic sine of the parameter."), VisualShaderNodeFloatFunc::FUNC_SINH, VisualShaderNode::PORT_TYPE_SCALAR));
  2276. add_options.push_back(AddOption("Sqrt", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the square root of the parameter."), VisualShaderNodeFloatFunc::FUNC_SQRT, VisualShaderNode::PORT_TYPE_SCALAR));
  2277. add_options.push_back(AddOption("SmoothStep", "Scalar", "Functions", "VisualShaderNodeScalarSmoothStep", TTR("SmoothStep function( scalar(edge0), scalar(edge1), scalar(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge0' and 1.0 if x is larger than 'edge1'. Otherwise the return value is interpolated between 0.0 and 1.0 using Hermite polynomials."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2278. add_options.push_back(AddOption("Step", "Scalar", "Functions", "VisualShaderNodeFloatOp", TTR("Step function( scalar(edge), scalar(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge' and otherwise 1.0."), VisualShaderNodeFloatOp::OP_STEP, VisualShaderNode::PORT_TYPE_SCALAR));
  2279. add_options.push_back(AddOption("Tan", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the tangent of the parameter."), VisualShaderNodeFloatFunc::FUNC_TAN, VisualShaderNode::PORT_TYPE_SCALAR));
  2280. add_options.push_back(AddOption("TanH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the hyperbolic tangent of the parameter."), VisualShaderNodeFloatFunc::FUNC_TANH, VisualShaderNode::PORT_TYPE_SCALAR));
  2281. add_options.push_back(AddOption("Trunc", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Finds the truncated value of the parameter."), VisualShaderNodeFloatFunc::FUNC_TRUNC, VisualShaderNode::PORT_TYPE_SCALAR));
  2282. add_options.push_back(AddOption("Add", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Sums two floating-point scalars."), VisualShaderNodeFloatOp::OP_ADD, VisualShaderNode::PORT_TYPE_SCALAR));
  2283. add_options.push_back(AddOption("Add", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Sums two integer scalars."), VisualShaderNodeIntOp::OP_ADD, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2284. add_options.push_back(AddOption("Divide", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Divides two floating-point scalars."), VisualShaderNodeFloatOp::OP_DIV, VisualShaderNode::PORT_TYPE_SCALAR));
  2285. add_options.push_back(AddOption("Divide", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Divides two integer scalars."), VisualShaderNodeIntOp::OP_DIV, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2286. add_options.push_back(AddOption("Multiply", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Multiplies two floating-point scalars."), VisualShaderNodeFloatOp::OP_MUL, VisualShaderNode::PORT_TYPE_SCALAR));
  2287. add_options.push_back(AddOption("Multiply", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Multiplies two integer scalars."), VisualShaderNodeIntOp::OP_MUL, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2288. add_options.push_back(AddOption("Remainder", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Returns the remainder of the two floating-point scalars."), VisualShaderNodeFloatOp::OP_MOD, VisualShaderNode::PORT_TYPE_SCALAR));
  2289. add_options.push_back(AddOption("Remainder", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Returns the remainder of the two integer scalars."), VisualShaderNodeIntOp::OP_MOD, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2290. add_options.push_back(AddOption("Subtract", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Subtracts two floating-point scalars."), VisualShaderNodeFloatOp::OP_SUB, VisualShaderNode::PORT_TYPE_SCALAR));
  2291. add_options.push_back(AddOption("Subtract", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Subtracts two integer scalars."), VisualShaderNodeIntOp::OP_SUB, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2292. add_options.push_back(AddOption("FloatConstant", "Scalar", "Variables", "VisualShaderNodeFloatConstant", TTR("Scalar floating-point constant."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2293. add_options.push_back(AddOption("IntConstant", "Scalar", "Variables", "VisualShaderNodeIntConstant", TTR("Scalar integer constant."), -1, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2294. add_options.push_back(AddOption("FloatUniform", "Scalar", "Variables", "VisualShaderNodeFloatUniform", TTR("Scalar floating-point uniform."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2295. add_options.push_back(AddOption("IntUniform", "Scalar", "Variables", "VisualShaderNodeIntUniform", TTR("Scalar integer uniform."), -1, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  2296. // TEXTURES
  2297. add_options.push_back(AddOption("CubeMap", "Textures", "Functions", "VisualShaderNodeCubemap", TTR("Perform the cubic texture lookup."), -1, -1));
  2298. texture_node_option_idx = add_options.size();
  2299. add_options.push_back(AddOption("Texture2D", "Textures", "Functions", "VisualShaderNodeTexture", TTR("Perform the texture lookup."), -1, -1));
  2300. add_options.push_back(AddOption("CubeMapUniform", "Textures", "Variables", "VisualShaderNodeCubemapUniform", TTR("Cubic texture uniform lookup."), -1, -1));
  2301. add_options.push_back(AddOption("Texture2DArray", "Textures", "Functions", "VisualShaderNodeTexture2DArray", TTR("Perform the 2D-array texture lookup."), -1, -1, -1, -1, -1));
  2302. add_options.push_back(AddOption("TextureUniform", "Textures", "Variables", "VisualShaderNodeTextureUniform", TTR("2D texture uniform lookup."), -1, -1));
  2303. add_options.push_back(AddOption("TextureUniformTriplanar", "Textures", "Variables", "VisualShaderNodeTextureUniformTriplanar", TTR("2D texture uniform lookup with triplanar."), -1, -1, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, Shader::MODE_SPATIAL));
  2304. add_options.push_back(AddOption("Texture2DArrayUniform", "Textures", "Variables", "VisualShaderNodeTexture2DArrayUniform", TTR("2D array of textures uniform lookup."), -1, -1, -1, -1, -1));
  2305. // TRANSFORM
  2306. add_options.push_back(AddOption("TransformFunc", "Transform", "Common", "VisualShaderNodeTransformFunc", TTR("Transform function."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  2307. add_options.push_back(AddOption("OuterProduct", "Transform", "Composition", "VisualShaderNodeOuterProduct", TTR("Calculate the outer product of a pair of vectors.\n\nOuterProduct treats the first parameter 'c' as a column vector (matrix with one column) and the second parameter 'r' as a row vector (matrix with one row) and does a linear algebraic matrix multiply 'c * r', yielding a matrix whose number of rows is the number of components in 'c' and whose number of columns is the number of components in 'r'."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  2308. add_options.push_back(AddOption("TransformCompose", "Transform", "Composition", "VisualShaderNodeTransformCompose", TTR("Composes transform from four vectors."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  2309. add_options.push_back(AddOption("TransformDecompose", "Transform", "Composition", "VisualShaderNodeTransformDecompose", TTR("Decomposes transform to four vectors.")));
  2310. add_options.push_back(AddOption("Determinant", "Transform", "Functions", "VisualShaderNodeDeterminant", TTR("Calculates the determinant of a transform."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2311. add_options.push_back(AddOption("Inverse", "Transform", "Functions", "VisualShaderNodeTransformFunc", TTR("Calculates the inverse of a transform."), VisualShaderNodeTransformFunc::FUNC_INVERSE, VisualShaderNode::PORT_TYPE_TRANSFORM));
  2312. add_options.push_back(AddOption("Transpose", "Transform", "Functions", "VisualShaderNodeTransformFunc", TTR("Calculates the transpose of a transform."), VisualShaderNodeTransformFunc::FUNC_TRANSPOSE, VisualShaderNode::PORT_TYPE_TRANSFORM));
  2313. add_options.push_back(AddOption("TransformMult", "Transform", "Operators", "VisualShaderNodeTransformMult", TTR("Multiplies transform by transform."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  2314. add_options.push_back(AddOption("TransformVectorMult", "Transform", "Operators", "VisualShaderNodeTransformVecMult", TTR("Multiplies vector by transform."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2315. add_options.push_back(AddOption("TransformConstant", "Transform", "Variables", "VisualShaderNodeTransformConstant", TTR("Transform constant."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  2316. add_options.push_back(AddOption("TransformUniform", "Transform", "Variables", "VisualShaderNodeTransformUniform", TTR("Transform uniform."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  2317. // VECTOR
  2318. add_options.push_back(AddOption("VectorFunc", "Vector", "Common", "VisualShaderNodeVectorFunc", TTR("Vector function."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2319. add_options.push_back(AddOption("VectorOp", "Vector", "Common", "VisualShaderNodeVectorOp", TTR("Vector operator."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2320. add_options.push_back(AddOption("VectorCompose", "Vector", "Composition", "VisualShaderNodeVectorCompose", TTR("Composes vector from three scalars."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2321. add_options.push_back(AddOption("VectorDecompose", "Vector", "Composition", "VisualShaderNodeVectorDecompose", TTR("Decomposes vector to three scalars.")));
  2322. add_options.push_back(AddOption("Abs", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the absolute value of the parameter."), VisualShaderNodeVectorFunc::FUNC_ABS, VisualShaderNode::PORT_TYPE_VECTOR));
  2323. add_options.push_back(AddOption("ACos", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the arc-cosine of the parameter."), VisualShaderNodeVectorFunc::FUNC_ACOS, VisualShaderNode::PORT_TYPE_VECTOR));
  2324. add_options.push_back(AddOption("ACosH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the inverse hyperbolic cosine of the parameter."), VisualShaderNodeVectorFunc::FUNC_ACOSH, VisualShaderNode::PORT_TYPE_VECTOR));
  2325. add_options.push_back(AddOption("ASin", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the arc-sine of the parameter."), VisualShaderNodeVectorFunc::FUNC_ASIN, VisualShaderNode::PORT_TYPE_VECTOR));
  2326. add_options.push_back(AddOption("ASinH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the inverse hyperbolic sine of the parameter."), VisualShaderNodeVectorFunc::FUNC_ASINH, VisualShaderNode::PORT_TYPE_VECTOR));
  2327. add_options.push_back(AddOption("ATan", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the arc-tangent of the parameter."), VisualShaderNodeVectorFunc::FUNC_ATAN, VisualShaderNode::PORT_TYPE_VECTOR));
  2328. add_options.push_back(AddOption("ATan2", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Returns the arc-tangent of the parameters."), VisualShaderNodeVectorOp::OP_ATAN2, VisualShaderNode::PORT_TYPE_VECTOR));
  2329. add_options.push_back(AddOption("ATanH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the inverse hyperbolic tangent of the parameter."), VisualShaderNodeVectorFunc::FUNC_ATANH, VisualShaderNode::PORT_TYPE_VECTOR));
  2330. add_options.push_back(AddOption("Ceil", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Finds the nearest integer that is greater than or equal to the parameter."), VisualShaderNodeVectorFunc::FUNC_CEIL, VisualShaderNode::PORT_TYPE_VECTOR));
  2331. add_options.push_back(AddOption("Clamp", "Vector", "Functions", "VisualShaderNodeVectorClamp", TTR("Constrains a value to lie between two further values."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2332. add_options.push_back(AddOption("Cos", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the cosine of the parameter."), VisualShaderNodeVectorFunc::FUNC_COS, VisualShaderNode::PORT_TYPE_VECTOR));
  2333. add_options.push_back(AddOption("CosH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the hyperbolic cosine of the parameter."), VisualShaderNodeVectorFunc::FUNC_COSH, VisualShaderNode::PORT_TYPE_VECTOR));
  2334. add_options.push_back(AddOption("Cross", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Calculates the cross product of two vectors."), VisualShaderNodeVectorOp::OP_CROSS, VisualShaderNode::PORT_TYPE_VECTOR));
  2335. add_options.push_back(AddOption("Degrees", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Converts a quantity in radians to degrees."), VisualShaderNodeVectorFunc::FUNC_DEGREES, VisualShaderNode::PORT_TYPE_VECTOR));
  2336. add_options.push_back(AddOption("Distance", "Vector", "Functions", "VisualShaderNodeVectorDistance", TTR("Returns the distance between two points."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2337. add_options.push_back(AddOption("Dot", "Vector", "Functions", "VisualShaderNodeDotProduct", TTR("Calculates the dot product of two vectors."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2338. add_options.push_back(AddOption("Exp", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Base-e Exponential."), VisualShaderNodeVectorFunc::FUNC_EXP, VisualShaderNode::PORT_TYPE_VECTOR));
  2339. add_options.push_back(AddOption("Exp2", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Base-2 Exponential."), VisualShaderNodeVectorFunc::FUNC_EXP2, VisualShaderNode::PORT_TYPE_VECTOR));
  2340. add_options.push_back(AddOption("FaceForward", "Vector", "Functions", "VisualShaderNodeFaceForward", TTR("Returns the vector that points in the same direction as a reference vector. The function has three vector parameters : N, the vector to orient, I, the incident vector, and Nref, the reference vector. If the dot product of I and Nref is smaller than zero the return value is N. Otherwise -N is returned."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2341. add_options.push_back(AddOption("Floor", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Finds the nearest integer less than or equal to the parameter."), VisualShaderNodeVectorFunc::FUNC_FLOOR, VisualShaderNode::PORT_TYPE_VECTOR));
  2342. add_options.push_back(AddOption("Fract", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Computes the fractional part of the argument."), VisualShaderNodeVectorFunc::FUNC_FRAC, VisualShaderNode::PORT_TYPE_VECTOR));
  2343. add_options.push_back(AddOption("InverseSqrt", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the inverse of the square root of the parameter."), VisualShaderNodeVectorFunc::FUNC_INVERSE_SQRT, VisualShaderNode::PORT_TYPE_VECTOR));
  2344. add_options.push_back(AddOption("Length", "Vector", "Functions", "VisualShaderNodeVectorLen", TTR("Calculates the length of a vector."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2345. add_options.push_back(AddOption("Log", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Natural logarithm."), VisualShaderNodeVectorFunc::FUNC_LOG, VisualShaderNode::PORT_TYPE_VECTOR));
  2346. add_options.push_back(AddOption("Log2", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Base-2 logarithm."), VisualShaderNodeVectorFunc::FUNC_LOG2, VisualShaderNode::PORT_TYPE_VECTOR));
  2347. add_options.push_back(AddOption("Max", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Returns the greater of two values."), VisualShaderNodeVectorOp::OP_MAX, VisualShaderNode::PORT_TYPE_VECTOR));
  2348. add_options.push_back(AddOption("Min", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Returns the lesser of two values."), VisualShaderNodeVectorOp::OP_MIN, VisualShaderNode::PORT_TYPE_VECTOR));
  2349. add_options.push_back(AddOption("Mix", "Vector", "Functions", "VisualShaderNodeVectorInterp", TTR("Linear interpolation between two vectors."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2350. add_options.push_back(AddOption("MixS", "Vector", "Functions", "VisualShaderNodeVectorScalarMix", TTR("Linear interpolation between two vectors using scalar."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2351. add_options.push_back(AddOption("MultiplyAdd", "Vector", "Functions", "VisualShaderNodeMultiplyAdd", TTR("Performs a fused multiply-add operation (a * b + c) on vectors."), VisualShaderNodeMultiplyAdd::TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR));
  2352. add_options.push_back(AddOption("Negate", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the opposite value of the parameter."), VisualShaderNodeVectorFunc::FUNC_NEGATE, VisualShaderNode::PORT_TYPE_VECTOR));
  2353. add_options.push_back(AddOption("Normalize", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Calculates the normalize product of vector."), VisualShaderNodeVectorFunc::FUNC_NORMALIZE, VisualShaderNode::PORT_TYPE_VECTOR));
  2354. add_options.push_back(AddOption("OneMinus", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("1.0 - vector"), VisualShaderNodeVectorFunc::FUNC_ONEMINUS, VisualShaderNode::PORT_TYPE_VECTOR));
  2355. add_options.push_back(AddOption("Pow", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Returns the value of the first parameter raised to the power of the second."), VisualShaderNodeVectorOp::OP_POW, VisualShaderNode::PORT_TYPE_VECTOR));
  2356. add_options.push_back(AddOption("Radians", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Converts a quantity in degrees to radians."), VisualShaderNodeVectorFunc::FUNC_RADIANS, VisualShaderNode::PORT_TYPE_VECTOR));
  2357. add_options.push_back(AddOption("Reciprocal", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("1.0 / vector"), VisualShaderNodeVectorFunc::FUNC_RECIPROCAL, VisualShaderNode::PORT_TYPE_VECTOR));
  2358. add_options.push_back(AddOption("Reflect", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Returns the vector that points in the direction of reflection ( a : incident vector, b : normal vector )."), VisualShaderNodeVectorOp::OP_REFLECT, VisualShaderNode::PORT_TYPE_VECTOR));
  2359. add_options.push_back(AddOption("Refract", "Vector", "Functions", "VisualShaderNodeVectorRefract", TTR("Returns the vector that points in the direction of refraction."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2360. add_options.push_back(AddOption("Round", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Finds the nearest integer to the parameter."), VisualShaderNodeVectorFunc::FUNC_ROUND, VisualShaderNode::PORT_TYPE_VECTOR));
  2361. add_options.push_back(AddOption("RoundEven", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Finds the nearest even integer to the parameter."), VisualShaderNodeVectorFunc::FUNC_ROUNDEVEN, VisualShaderNode::PORT_TYPE_VECTOR));
  2362. add_options.push_back(AddOption("Saturate", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Clamps the value between 0.0 and 1.0."), VisualShaderNodeVectorFunc::FUNC_SATURATE, VisualShaderNode::PORT_TYPE_VECTOR));
  2363. add_options.push_back(AddOption("Sign", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Extracts the sign of the parameter."), VisualShaderNodeVectorFunc::FUNC_SIGN, VisualShaderNode::PORT_TYPE_VECTOR));
  2364. add_options.push_back(AddOption("Sin", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the sine of the parameter."), VisualShaderNodeVectorFunc::FUNC_SIN, VisualShaderNode::PORT_TYPE_VECTOR));
  2365. add_options.push_back(AddOption("SinH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the hyperbolic sine of the parameter."), VisualShaderNodeVectorFunc::FUNC_SINH, VisualShaderNode::PORT_TYPE_VECTOR));
  2366. add_options.push_back(AddOption("Sqrt", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the square root of the parameter."), VisualShaderNodeVectorFunc::FUNC_SQRT, VisualShaderNode::PORT_TYPE_VECTOR));
  2367. add_options.push_back(AddOption("SmoothStep", "Vector", "Functions", "VisualShaderNodeVectorSmoothStep", TTR("SmoothStep function( vector(edge0), vector(edge1), vector(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge0' and 1.0 if 'x' is larger than 'edge1'. Otherwise the return value is interpolated between 0.0 and 1.0 using Hermite polynomials."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2368. add_options.push_back(AddOption("SmoothStepS", "Vector", "Functions", "VisualShaderNodeVectorScalarSmoothStep", TTR("SmoothStep function( scalar(edge0), scalar(edge1), vector(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge0' and 1.0 if 'x' is larger than 'edge1'. Otherwise the return value is interpolated between 0.0 and 1.0 using Hermite polynomials."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2369. add_options.push_back(AddOption("Step", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Step function( vector(edge), vector(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge' and otherwise 1.0."), VisualShaderNodeVectorOp::OP_STEP, VisualShaderNode::PORT_TYPE_VECTOR));
  2370. add_options.push_back(AddOption("StepS", "Vector", "Functions", "VisualShaderNodeVectorScalarStep", TTR("Step function( scalar(edge), vector(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge' and otherwise 1.0."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2371. add_options.push_back(AddOption("Tan", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the tangent of the parameter."), VisualShaderNodeVectorFunc::FUNC_TAN, VisualShaderNode::PORT_TYPE_VECTOR));
  2372. add_options.push_back(AddOption("TanH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the hyperbolic tangent of the parameter."), VisualShaderNodeVectorFunc::FUNC_TANH, VisualShaderNode::PORT_TYPE_VECTOR));
  2373. add_options.push_back(AddOption("Trunc", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Finds the truncated value of the parameter."), VisualShaderNodeVectorFunc::FUNC_TRUNC, VisualShaderNode::PORT_TYPE_VECTOR));
  2374. add_options.push_back(AddOption("Add", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Adds vector to vector."), VisualShaderNodeVectorOp::OP_ADD, VisualShaderNode::PORT_TYPE_VECTOR));
  2375. add_options.push_back(AddOption("Divide", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Divides vector by vector."), VisualShaderNodeVectorOp::OP_DIV, VisualShaderNode::PORT_TYPE_VECTOR));
  2376. add_options.push_back(AddOption("Multiply", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Multiplies vector by vector."), VisualShaderNodeVectorOp::OP_MUL, VisualShaderNode::PORT_TYPE_VECTOR));
  2377. add_options.push_back(AddOption("Remainder", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Returns the remainder of the two vectors."), VisualShaderNodeVectorOp::OP_MOD, VisualShaderNode::PORT_TYPE_VECTOR));
  2378. add_options.push_back(AddOption("Subtract", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Subtracts vector from vector."), VisualShaderNodeVectorOp::OP_SUB, VisualShaderNode::PORT_TYPE_VECTOR));
  2379. add_options.push_back(AddOption("VectorConstant", "Vector", "Variables", "VisualShaderNodeVec3Constant", TTR("Vector constant."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2380. add_options.push_back(AddOption("VectorUniform", "Vector", "Variables", "VisualShaderNodeVec3Uniform", TTR("Vector uniform."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  2381. // SPECIAL
  2382. add_options.push_back(AddOption("Expression", "Special", "", "VisualShaderNodeExpression", TTR("Custom Godot Shader Language expression, with custom amount of input and output ports. This is a direct injection of code into the vertex/fragment/light function, do not use it to write the function declarations inside.")));
  2383. add_options.push_back(AddOption("Fresnel", "Special", "", "VisualShaderNodeFresnel", TTR("Returns falloff based on the dot product of surface normal and view direction of camera (pass associated inputs to it)."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  2384. add_options.push_back(AddOption("GlobalExpression", "Special", "", "VisualShaderNodeGlobalExpression", TTR("Custom Godot Shader Language expression, which is placed on top of the resulted shader. You can place various function definitions inside and call it later in the Expressions. You can also declare varyings, uniforms and constants.")));
  2385. add_options.push_back(AddOption("ScalarDerivativeFunc", "Special", "Common", "VisualShaderNodeScalarDerivativeFunc", TTR("(Fragment/Light mode only) Scalar derivative function."), -1, VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, -1, -1, true));
  2386. add_options.push_back(AddOption("VectorDerivativeFunc", "Special", "Common", "VisualShaderNodeVectorDerivativeFunc", TTR("(Fragment/Light mode only) Vector derivative function."), -1, VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, -1, -1, true));
  2387. add_options.push_back(AddOption("DdX", "Special", "Derivative", "VisualShaderNodeVectorDerivativeFunc", TTR("(Fragment/Light mode only) (Vector) Derivative in 'x' using local differencing."), VisualShaderNodeVectorDerivativeFunc::FUNC_X, VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, -1, -1, true));
  2388. add_options.push_back(AddOption("DdXS", "Special", "Derivative", "VisualShaderNodeScalarDerivativeFunc", TTR("(Fragment/Light mode only) (Scalar) Derivative in 'x' using local differencing."), VisualShaderNodeScalarDerivativeFunc::FUNC_X, VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, -1, -1, true));
  2389. add_options.push_back(AddOption("DdY", "Special", "Derivative", "VisualShaderNodeVectorDerivativeFunc", TTR("(Fragment/Light mode only) (Vector) Derivative in 'y' using local differencing."), VisualShaderNodeVectorDerivativeFunc::FUNC_Y, VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, -1, -1, true));
  2390. add_options.push_back(AddOption("DdYS", "Special", "Derivative", "VisualShaderNodeScalarDerivativeFunc", TTR("(Fragment/Light mode only) (Scalar) Derivative in 'y' using local differencing."), VisualShaderNodeScalarDerivativeFunc::FUNC_Y, VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, -1, -1, true));
  2391. add_options.push_back(AddOption("Sum", "Special", "Derivative", "VisualShaderNodeVectorDerivativeFunc", TTR("(Fragment/Light mode only) (Vector) Sum of absolute derivative in 'x' and 'y'."), VisualShaderNodeVectorDerivativeFunc::FUNC_SUM, VisualShaderNode::PORT_TYPE_VECTOR, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, -1, -1, true));
  2392. add_options.push_back(AddOption("SumS", "Special", "Derivative", "VisualShaderNodeScalarDerivativeFunc", TTR("(Fragment/Light mode only) (Scalar) Sum of absolute derivative in 'x' and 'y'."), VisualShaderNodeScalarDerivativeFunc::FUNC_SUM, VisualShaderNode::PORT_TYPE_SCALAR, VisualShader::TYPE_FRAGMENT | VisualShader::TYPE_LIGHT, -1, -1, true));
  2393. custom_node_option_idx = add_options.size();
  2394. /////////////////////////////////////////////////////////////////////
  2395. _update_options_menu();
  2396. error_panel = memnew(PanelContainer);
  2397. add_child(error_panel);
  2398. error_label = memnew(Label);
  2399. error_panel->add_child(error_label);
  2400. error_label->set_text("eh");
  2401. error_panel->hide();
  2402. undo_redo = EditorNode::get_singleton()->get_undo_redo();
  2403. Ref<VisualShaderNodePluginDefault> default_plugin;
  2404. default_plugin.instance();
  2405. add_plugin(default_plugin);
  2406. property_editor = memnew(CustomPropertyEditor);
  2407. add_child(property_editor);
  2408. property_editor->connect("variant_changed", callable_mp(this, &VisualShaderEditor::_port_edited));
  2409. }
  2410. void VisualShaderEditorPlugin::edit(Object *p_object) {
  2411. visual_shader_editor->edit(Object::cast_to<VisualShader>(p_object));
  2412. }
  2413. bool VisualShaderEditorPlugin::handles(Object *p_object) const {
  2414. return p_object->is_class("VisualShader");
  2415. }
  2416. void VisualShaderEditorPlugin::make_visible(bool p_visible) {
  2417. if (p_visible) {
  2418. //editor->hide_animation_player_editors();
  2419. //editor->animation_panel_make_visible(true);
  2420. button->show();
  2421. editor->make_bottom_panel_item_visible(visual_shader_editor);
  2422. visual_shader_editor->update_custom_nodes();
  2423. visual_shader_editor->set_process_input(true);
  2424. //visual_shader_editor->set_process(true);
  2425. } else {
  2426. if (visual_shader_editor->is_visible_in_tree()) {
  2427. editor->hide_bottom_panel();
  2428. }
  2429. button->hide();
  2430. visual_shader_editor->set_process_input(false);
  2431. //visual_shader_editor->set_process(false);
  2432. }
  2433. }
  2434. VisualShaderEditorPlugin::VisualShaderEditorPlugin(EditorNode *p_node) {
  2435. editor = p_node;
  2436. visual_shader_editor = memnew(VisualShaderEditor);
  2437. visual_shader_editor->set_custom_minimum_size(Size2(0, 300) * EDSCALE);
  2438. button = editor->add_bottom_panel_item(TTR("VisualShader"), visual_shader_editor);
  2439. button->hide();
  2440. }
  2441. VisualShaderEditorPlugin::~VisualShaderEditorPlugin() {
  2442. }
  2443. ////////////////
  2444. class VisualShaderNodePluginInputEditor : public OptionButton {
  2445. GDCLASS(VisualShaderNodePluginInputEditor, OptionButton);
  2446. Ref<VisualShaderNodeInput> input;
  2447. public:
  2448. void _notification(int p_what) {
  2449. if (p_what == NOTIFICATION_READY) {
  2450. connect("item_selected", callable_mp(this, &VisualShaderNodePluginInputEditor::_item_selected));
  2451. }
  2452. }
  2453. void _item_selected(int p_item) {
  2454. VisualShaderEditor::get_singleton()->call_deferred("_input_select_item", input, get_item_text(p_item));
  2455. }
  2456. void setup(const Ref<VisualShaderNodeInput> &p_input) {
  2457. input = p_input;
  2458. Ref<Texture2D> type_icon[6] = {
  2459. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("float", "EditorIcons"),
  2460. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("int", "EditorIcons"),
  2461. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Vector3", "EditorIcons"),
  2462. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("bool", "EditorIcons"),
  2463. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Transform", "EditorIcons"),
  2464. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("ImageTexture", "EditorIcons"),
  2465. };
  2466. add_item("[None]");
  2467. int to_select = -1;
  2468. for (int i = 0; i < input->get_input_index_count(); i++) {
  2469. if (input->get_input_name() == input->get_input_index_name(i)) {
  2470. to_select = i + 1;
  2471. }
  2472. add_icon_item(type_icon[input->get_input_index_type(i)], input->get_input_index_name(i));
  2473. }
  2474. if (to_select >= 0) {
  2475. select(to_select);
  2476. }
  2477. }
  2478. };
  2479. class VisualShaderNodePluginDefaultEditor : public VBoxContainer {
  2480. GDCLASS(VisualShaderNodePluginDefaultEditor, VBoxContainer);
  2481. Ref<Resource> parent_resource;
  2482. public:
  2483. void _property_changed(const String &p_property, const Variant &p_value, const String &p_field = "", bool p_changing = false) {
  2484. if (p_changing) {
  2485. return;
  2486. }
  2487. UndoRedo *undo_redo = EditorNode::get_singleton()->get_undo_redo();
  2488. updating = true;
  2489. undo_redo->create_action(TTR("Edit Visual Property") + ": " + p_property, UndoRedo::MERGE_ENDS);
  2490. undo_redo->add_do_property(node.ptr(), p_property, p_value);
  2491. undo_redo->add_undo_property(node.ptr(), p_property, node->get(p_property));
  2492. if (p_value.get_type() == Variant::OBJECT) {
  2493. RES prev_res = node->get(p_property);
  2494. RES curr_res = p_value;
  2495. if (curr_res.is_null()) {
  2496. undo_redo->add_do_method(this, "_open_inspector", (RES)parent_resource.ptr());
  2497. } else {
  2498. undo_redo->add_do_method(this, "_open_inspector", (RES)curr_res.ptr());
  2499. }
  2500. if (!prev_res.is_null()) {
  2501. undo_redo->add_undo_method(this, "_open_inspector", (RES)prev_res.ptr());
  2502. } else {
  2503. undo_redo->add_undo_method(this, "_open_inspector", (RES)parent_resource.ptr());
  2504. }
  2505. undo_redo->add_do_method(this, "_refresh_request");
  2506. undo_redo->add_undo_method(this, "_refresh_request");
  2507. }
  2508. undo_redo->commit_action();
  2509. updating = false;
  2510. }
  2511. void _node_changed() {
  2512. if (updating) {
  2513. return;
  2514. }
  2515. for (int i = 0; i < properties.size(); i++) {
  2516. properties[i]->update_property();
  2517. }
  2518. }
  2519. void _refresh_request() {
  2520. VisualShaderEditor::get_singleton()->call_deferred("_update_graph");
  2521. }
  2522. void _resource_selected(const String &p_path, RES p_resource) {
  2523. _open_inspector(p_resource);
  2524. }
  2525. void _open_inspector(RES p_resource) {
  2526. EditorNode::get_singleton()->get_inspector()->edit(p_resource.ptr());
  2527. }
  2528. bool updating;
  2529. Ref<VisualShaderNode> node;
  2530. Vector<EditorProperty *> properties;
  2531. Vector<Label *> prop_names;
  2532. void _show_prop_names(bool p_show) {
  2533. for (int i = 0; i < prop_names.size(); i++) {
  2534. prop_names[i]->set_visible(p_show);
  2535. }
  2536. }
  2537. void setup(Ref<Resource> p_parent_resource, Vector<EditorProperty *> p_properties, const Vector<StringName> &p_names, Ref<VisualShaderNode> p_node) {
  2538. parent_resource = p_parent_resource;
  2539. updating = false;
  2540. node = p_node;
  2541. properties = p_properties;
  2542. for (int i = 0; i < p_properties.size(); i++) {
  2543. HBoxContainer *hbox = memnew(HBoxContainer);
  2544. hbox->set_h_size_flags(SIZE_EXPAND_FILL);
  2545. add_child(hbox);
  2546. Label *prop_name = memnew(Label);
  2547. String prop_name_str = p_names[i];
  2548. prop_name_str = prop_name_str.capitalize() + ":";
  2549. prop_name->set_text(prop_name_str);
  2550. prop_name->set_visible(false);
  2551. hbox->add_child(prop_name);
  2552. prop_names.push_back(prop_name);
  2553. p_properties[i]->set_h_size_flags(SIZE_EXPAND_FILL);
  2554. hbox->add_child(p_properties[i]);
  2555. bool res_prop = Object::cast_to<EditorPropertyResource>(p_properties[i]);
  2556. if (res_prop) {
  2557. p_properties[i]->connect("resource_selected", callable_mp(this, &VisualShaderNodePluginDefaultEditor::_resource_selected));
  2558. }
  2559. properties[i]->connect("property_changed", callable_mp(this, &VisualShaderNodePluginDefaultEditor::_property_changed));
  2560. properties[i]->set_object_and_property(node.ptr(), p_names[i]);
  2561. properties[i]->update_property();
  2562. properties[i]->set_name_split_ratio(0);
  2563. }
  2564. node->connect("changed", callable_mp(this, &VisualShaderNodePluginDefaultEditor::_node_changed));
  2565. node->connect("editor_refresh_request", callable_mp(this, &VisualShaderNodePluginDefaultEditor::_refresh_request), varray(), CONNECT_DEFERRED);
  2566. }
  2567. static void _bind_methods() {
  2568. ClassDB::bind_method("_refresh_request", &VisualShaderNodePluginDefaultEditor::_refresh_request); // Used by UndoRedo.
  2569. ClassDB::bind_method("_open_inspector", &VisualShaderNodePluginDefaultEditor::_open_inspector); // Used by UndoRedo.
  2570. ClassDB::bind_method("_show_prop_names", &VisualShaderNodePluginDefaultEditor::_show_prop_names); // Used with call_deferred.
  2571. }
  2572. };
  2573. Control *VisualShaderNodePluginDefault::create_editor(const Ref<Resource> &p_parent_resource, const Ref<VisualShaderNode> &p_node) {
  2574. if (p_node->is_class("VisualShaderNodeInput")) {
  2575. //create input
  2576. VisualShaderNodePluginInputEditor *input_editor = memnew(VisualShaderNodePluginInputEditor);
  2577. input_editor->setup(p_node);
  2578. return input_editor;
  2579. }
  2580. Vector<StringName> properties = p_node->get_editable_properties();
  2581. if (properties.size() == 0) {
  2582. return nullptr;
  2583. }
  2584. List<PropertyInfo> props;
  2585. p_node->get_property_list(&props);
  2586. Vector<PropertyInfo> pinfo;
  2587. for (List<PropertyInfo>::Element *E = props.front(); E; E = E->next()) {
  2588. for (int i = 0; i < properties.size(); i++) {
  2589. if (E->get().name == String(properties[i])) {
  2590. pinfo.push_back(E->get());
  2591. }
  2592. }
  2593. }
  2594. if (pinfo.size() == 0) {
  2595. return nullptr;
  2596. }
  2597. properties.clear();
  2598. Ref<VisualShaderNode> node = p_node;
  2599. Vector<EditorProperty *> editors;
  2600. for (int i = 0; i < pinfo.size(); i++) {
  2601. EditorProperty *prop = EditorInspector::instantiate_property_editor(node.ptr(), pinfo[i].type, pinfo[i].name, pinfo[i].hint, pinfo[i].hint_string, pinfo[i].usage);
  2602. if (!prop) {
  2603. return nullptr;
  2604. }
  2605. if (Object::cast_to<EditorPropertyResource>(prop)) {
  2606. Object::cast_to<EditorPropertyResource>(prop)->set_use_sub_inspector(false);
  2607. prop->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  2608. } else if (Object::cast_to<EditorPropertyTransform>(prop) || Object::cast_to<EditorPropertyVector3>(prop)) {
  2609. prop->set_custom_minimum_size(Size2(250 * EDSCALE, 0));
  2610. } else if (Object::cast_to<EditorPropertyFloat>(prop)) {
  2611. prop->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  2612. } else if (Object::cast_to<EditorPropertyEnum>(prop)) {
  2613. prop->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  2614. Object::cast_to<EditorPropertyEnum>(prop)->set_option_button_clip(false);
  2615. }
  2616. editors.push_back(prop);
  2617. properties.push_back(pinfo[i].name);
  2618. }
  2619. VisualShaderNodePluginDefaultEditor *editor = memnew(VisualShaderNodePluginDefaultEditor);
  2620. editor->setup(p_parent_resource, editors, properties, p_node);
  2621. return editor;
  2622. }
  2623. void EditorPropertyShaderMode::_option_selected(int p_which) {
  2624. //will not use this, instead will do all the logic setting manually
  2625. //emit_signal("property_changed", get_edited_property(), p_which);
  2626. Ref<VisualShader> visual_shader(Object::cast_to<VisualShader>(get_edited_object()));
  2627. if (visual_shader->get_mode() == p_which) {
  2628. return;
  2629. }
  2630. UndoRedo *undo_redo = EditorNode::get_singleton()->get_undo_redo();
  2631. undo_redo->create_action(TTR("Visual Shader Mode Changed"));
  2632. //do is easy
  2633. undo_redo->add_do_method(visual_shader.ptr(), "set_mode", p_which);
  2634. undo_redo->add_undo_method(visual_shader.ptr(), "set_mode", visual_shader->get_mode());
  2635. //now undo is hell
  2636. //1. restore connections to output
  2637. for (int i = 0; i < VisualShader::TYPE_MAX; i++) {
  2638. VisualShader::Type type = VisualShader::Type(i);
  2639. List<VisualShader::Connection> conns;
  2640. visual_shader->get_node_connections(type, &conns);
  2641. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  2642. if (E->get().to_node == VisualShader::NODE_ID_OUTPUT) {
  2643. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2644. }
  2645. }
  2646. }
  2647. //2. restore input indices
  2648. for (int i = 0; i < VisualShader::TYPE_MAX; i++) {
  2649. VisualShader::Type type = VisualShader::Type(i);
  2650. Vector<int> nodes = visual_shader->get_node_list(type);
  2651. for (int j = 0; j < nodes.size(); j++) {
  2652. Ref<VisualShaderNodeInput> input = visual_shader->get_node(type, nodes[j]);
  2653. if (!input.is_valid()) {
  2654. continue;
  2655. }
  2656. undo_redo->add_undo_method(input.ptr(), "set_input_name", input->get_input_name());
  2657. }
  2658. }
  2659. //3. restore enums and flags
  2660. List<PropertyInfo> props;
  2661. visual_shader->get_property_list(&props);
  2662. for (List<PropertyInfo>::Element *E = props.front(); E; E = E->next()) {
  2663. if (E->get().name.begins_with("flags/") || E->get().name.begins_with("modes/")) {
  2664. undo_redo->add_undo_property(visual_shader.ptr(), E->get().name, visual_shader->get(E->get().name));
  2665. }
  2666. }
  2667. undo_redo->add_do_method(VisualShaderEditor::get_singleton(), "_update_options_menu");
  2668. undo_redo->add_undo_method(VisualShaderEditor::get_singleton(), "_update_options_menu");
  2669. //update graph
  2670. undo_redo->add_do_method(VisualShaderEditor::get_singleton(), "_update_graph");
  2671. undo_redo->add_undo_method(VisualShaderEditor::get_singleton(), "_update_graph");
  2672. undo_redo->commit_action();
  2673. }
  2674. void EditorPropertyShaderMode::update_property() {
  2675. int which = get_edited_object()->get(get_edited_property());
  2676. options->select(which);
  2677. }
  2678. void EditorPropertyShaderMode::setup(const Vector<String> &p_options) {
  2679. for (int i = 0; i < p_options.size(); i++) {
  2680. options->add_item(p_options[i], i);
  2681. }
  2682. }
  2683. void EditorPropertyShaderMode::set_option_button_clip(bool p_enable) {
  2684. options->set_clip_text(p_enable);
  2685. }
  2686. void EditorPropertyShaderMode::_bind_methods() {
  2687. }
  2688. EditorPropertyShaderMode::EditorPropertyShaderMode() {
  2689. options = memnew(OptionButton);
  2690. options->set_clip_text(true);
  2691. add_child(options);
  2692. add_focusable(options);
  2693. options->connect("item_selected", callable_mp(this, &EditorPropertyShaderMode::_option_selected));
  2694. }
  2695. bool EditorInspectorShaderModePlugin::can_handle(Object *p_object) {
  2696. return true; //can handle everything
  2697. }
  2698. void EditorInspectorShaderModePlugin::parse_begin(Object *p_object) {
  2699. //do none
  2700. }
  2701. bool EditorInspectorShaderModePlugin::parse_property(Object *p_object, Variant::Type p_type, const String &p_path, PropertyHint p_hint, const String &p_hint_text, int p_usage, bool p_wide) {
  2702. if (p_path == "mode" && p_object->is_class("VisualShader") && p_type == Variant::INT) {
  2703. EditorPropertyShaderMode *editor = memnew(EditorPropertyShaderMode);
  2704. Vector<String> options = p_hint_text.split(",");
  2705. editor->setup(options);
  2706. add_property_editor(p_path, editor);
  2707. return true;
  2708. }
  2709. return false; //can be overridden, although it will most likely be last anyway
  2710. }
  2711. void EditorInspectorShaderModePlugin::parse_end() {
  2712. //do none
  2713. }
  2714. //////////////////////////////////
  2715. void VisualShaderNodePortPreview::_shader_changed() {
  2716. if (shader.is_null()) {
  2717. return;
  2718. }
  2719. Vector<VisualShader::DefaultTextureParam> default_textures;
  2720. String shader_code = shader->generate_preview_shader(type, node, port, default_textures);
  2721. Ref<Shader> preview_shader;
  2722. preview_shader.instance();
  2723. preview_shader->set_code(shader_code);
  2724. for (int i = 0; i < default_textures.size(); i++) {
  2725. preview_shader->set_default_texture_param(default_textures[i].name, default_textures[i].param);
  2726. }
  2727. Ref<ShaderMaterial> material;
  2728. material.instance();
  2729. material->set_shader(preview_shader);
  2730. //find if a material is also being edited and copy parameters to this one
  2731. for (int i = EditorNode::get_singleton()->get_editor_history()->get_path_size() - 1; i >= 0; i--) {
  2732. Object *object = ObjectDB::get_instance(EditorNode::get_singleton()->get_editor_history()->get_path_object(i));
  2733. if (!object) {
  2734. continue;
  2735. }
  2736. ShaderMaterial *src_mat = Object::cast_to<ShaderMaterial>(object);
  2737. if (src_mat && src_mat->get_shader().is_valid()) {
  2738. List<PropertyInfo> params;
  2739. src_mat->get_shader()->get_param_list(&params);
  2740. for (List<PropertyInfo>::Element *E = params.front(); E; E = E->next()) {
  2741. material->set(E->get().name, src_mat->get(E->get().name));
  2742. }
  2743. }
  2744. }
  2745. set_material(material);
  2746. }
  2747. void VisualShaderNodePortPreview::setup(const Ref<VisualShader> &p_shader, VisualShader::Type p_type, int p_node, int p_port) {
  2748. shader = p_shader;
  2749. shader->connect("changed", callable_mp(this, &VisualShaderNodePortPreview::_shader_changed));
  2750. type = p_type;
  2751. port = p_port;
  2752. node = p_node;
  2753. update();
  2754. _shader_changed();
  2755. }
  2756. Size2 VisualShaderNodePortPreview::get_minimum_size() const {
  2757. return Size2(100, 100) * EDSCALE;
  2758. }
  2759. void VisualShaderNodePortPreview::_notification(int p_what) {
  2760. if (p_what == NOTIFICATION_DRAW) {
  2761. Vector<Vector2> points;
  2762. Vector<Vector2> uvs;
  2763. Vector<Color> colors;
  2764. points.push_back(Vector2());
  2765. uvs.push_back(Vector2(0, 0));
  2766. colors.push_back(Color(1, 1, 1, 1));
  2767. points.push_back(Vector2(get_size().width, 0));
  2768. uvs.push_back(Vector2(1, 0));
  2769. colors.push_back(Color(1, 1, 1, 1));
  2770. points.push_back(get_size());
  2771. uvs.push_back(Vector2(1, 1));
  2772. colors.push_back(Color(1, 1, 1, 1));
  2773. points.push_back(Vector2(0, get_size().height));
  2774. uvs.push_back(Vector2(0, 1));
  2775. colors.push_back(Color(1, 1, 1, 1));
  2776. draw_primitive(points, colors, uvs);
  2777. }
  2778. }
  2779. void VisualShaderNodePortPreview::_bind_methods() {
  2780. }
  2781. VisualShaderNodePortPreview::VisualShaderNodePortPreview() {
  2782. }
  2783. //////////////////////////////////
  2784. String VisualShaderConversionPlugin::converts_to() const {
  2785. return "Shader";
  2786. }
  2787. bool VisualShaderConversionPlugin::handles(const Ref<Resource> &p_resource) const {
  2788. Ref<VisualShader> vshader = p_resource;
  2789. return vshader.is_valid();
  2790. }
  2791. Ref<Resource> VisualShaderConversionPlugin::convert(const Ref<Resource> &p_resource) const {
  2792. Ref<VisualShader> vshader = p_resource;
  2793. ERR_FAIL_COND_V(!vshader.is_valid(), Ref<Resource>());
  2794. Ref<Shader> shader;
  2795. shader.instance();
  2796. String code = vshader->get_code();
  2797. shader->set_code(code);
  2798. return shader;
  2799. }