visual_shader_editor_plugin.cpp 249 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993
  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-2021 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2021 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/config/project_settings.h"
  32. #include "core/input/input.h"
  33. #include "core/io/resource_loader.h"
  34. #include "core/math/math_defs.h"
  35. #include "core/os/keyboard.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 "scene/resources/visual_shader_particle_nodes.h"
  46. #include "scene/resources/visual_shader_sdf_nodes.h"
  47. #include "servers/display_server.h"
  48. #include "servers/rendering/shader_types.h"
  49. struct FloatConstantDef {
  50. String name;
  51. float value = 0;
  52. String desc;
  53. };
  54. static FloatConstantDef float_constant_defs[] = {
  55. { "E", Math_E, TTR("E constant (2.718282). Represents the base of the natural logarithm.") },
  56. { "Epsilon", CMP_EPSILON, TTR("Epsilon constant (0.00001). Smallest possible scalar number.") },
  57. { "Phi", 1.618034f, TTR("Phi constant (1.618034). Golden ratio.") },
  58. { "Pi/4", Math_PI / 4, TTR("Pi/4 constant (0.785398) or 45 degrees.") },
  59. { "Pi/2", Math_PI / 2, TTR("Pi/2 constant (1.570796) or 90 degrees.") },
  60. { "Pi", Math_PI, TTR("Pi constant (3.141593) or 180 degrees.") },
  61. { "Tau", Math_TAU, TTR("Tau constant (6.283185) or 360 degrees.") },
  62. { "Sqrt2", Math_SQRT2, TTR("Sqrt2 constant (1.414214). Square root of 2.") }
  63. };
  64. const int MAX_FLOAT_CONST_DEFS = sizeof(float_constant_defs) / sizeof(FloatConstantDef);
  65. ///////////////////
  66. Control *VisualShaderNodePlugin::create_editor(const Ref<Resource> &p_parent_resource, const Ref<VisualShaderNode> &p_node) {
  67. if (get_script_instance()) {
  68. return get_script_instance()->call("_create_editor", p_parent_resource, p_node);
  69. }
  70. return nullptr;
  71. }
  72. void VisualShaderNodePlugin::_bind_methods() {
  73. 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")));
  74. }
  75. ///////////////////
  76. 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) {
  77. Ref<StyleBoxEmpty> style(memnew(StyleBoxEmpty));
  78. style->set_default_margin(SIDE_LEFT, p_margin_left * EDSCALE);
  79. style->set_default_margin(SIDE_RIGHT, p_margin_right * EDSCALE);
  80. style->set_default_margin(SIDE_BOTTOM, p_margin_bottom * EDSCALE);
  81. style->set_default_margin(SIDE_TOP, p_margin_top * EDSCALE);
  82. return style;
  83. }
  84. ///////////////////
  85. VisualShaderGraphPlugin::VisualShaderGraphPlugin() {
  86. }
  87. void VisualShaderGraphPlugin::_bind_methods() {
  88. ClassDB::bind_method("add_node", &VisualShaderGraphPlugin::add_node);
  89. ClassDB::bind_method("remove_node", &VisualShaderGraphPlugin::remove_node);
  90. ClassDB::bind_method("connect_nodes", &VisualShaderGraphPlugin::connect_nodes);
  91. ClassDB::bind_method("disconnect_nodes", &VisualShaderGraphPlugin::disconnect_nodes);
  92. ClassDB::bind_method("set_node_position", &VisualShaderGraphPlugin::set_node_position);
  93. ClassDB::bind_method("set_node_size", &VisualShaderGraphPlugin::set_node_size);
  94. ClassDB::bind_method("update_node", &VisualShaderGraphPlugin::update_node);
  95. ClassDB::bind_method("update_node_deferred", &VisualShaderGraphPlugin::update_node_deferred);
  96. ClassDB::bind_method("set_input_port_default_value", &VisualShaderGraphPlugin::set_input_port_default_value);
  97. ClassDB::bind_method("set_uniform_name", &VisualShaderGraphPlugin::set_uniform_name);
  98. ClassDB::bind_method("set_expression", &VisualShaderGraphPlugin::set_expression);
  99. ClassDB::bind_method("update_curve", &VisualShaderGraphPlugin::update_curve);
  100. ClassDB::bind_method("update_constant", &VisualShaderGraphPlugin::update_constant);
  101. }
  102. void VisualShaderGraphPlugin::register_shader(VisualShader *p_shader) {
  103. visual_shader = Ref<VisualShader>(p_shader);
  104. }
  105. void VisualShaderGraphPlugin::set_connections(List<VisualShader::Connection> &p_connections) {
  106. connections = p_connections;
  107. }
  108. void VisualShaderGraphPlugin::show_port_preview(VisualShader::Type p_type, int p_node_id, int p_port_id) {
  109. if (visual_shader->get_shader_type() == p_type && links.has(p_node_id) && links[p_node_id].output_ports.has(p_port_id)) {
  110. for (Map<int, Port>::Element *E = links[p_node_id].output_ports.front(); E; E = E->next()) {
  111. if (E->value().preview_button != nullptr) {
  112. E->value().preview_button->set_pressed(false);
  113. }
  114. }
  115. if (links[p_node_id].preview_visible && !is_dirty() && links[p_node_id].preview_box != nullptr) {
  116. links[p_node_id].graph_node->remove_child(links[p_node_id].preview_box);
  117. memdelete(links[p_node_id].preview_box);
  118. links[p_node_id].graph_node->set_size(Vector2(-1, -1));
  119. links[p_node_id].preview_visible = false;
  120. }
  121. if (p_port_id != -1 && links[p_node_id].output_ports[p_port_id].preview_button != nullptr) {
  122. if (is_dirty()) {
  123. links[p_node_id].preview_pos = links[p_node_id].graph_node->get_child_count();
  124. }
  125. VBoxContainer *vbox = memnew(VBoxContainer);
  126. links[p_node_id].graph_node->add_child(vbox);
  127. if (links[p_node_id].preview_pos != -1) {
  128. links[p_node_id].graph_node->move_child(vbox, links[p_node_id].preview_pos);
  129. }
  130. Control *offset = memnew(Control);
  131. offset->set_custom_minimum_size(Size2(0, 5 * EDSCALE));
  132. vbox->add_child(offset);
  133. VisualShaderNodePortPreview *port_preview = memnew(VisualShaderNodePortPreview);
  134. port_preview->setup(visual_shader, visual_shader->get_shader_type(), p_node_id, p_port_id);
  135. port_preview->set_h_size_flags(Control::SIZE_SHRINK_CENTER);
  136. vbox->add_child(port_preview);
  137. links[p_node_id].preview_visible = true;
  138. links[p_node_id].preview_box = vbox;
  139. links[p_node_id].output_ports[p_port_id].preview_button->set_pressed(true);
  140. }
  141. }
  142. }
  143. void VisualShaderGraphPlugin::update_node_deferred(VisualShader::Type p_type, int p_node_id) {
  144. call_deferred("update_node", p_type, p_node_id);
  145. }
  146. void VisualShaderGraphPlugin::update_node(VisualShader::Type p_type, int p_node_id) {
  147. if (p_type != visual_shader->get_shader_type() || !links.has(p_node_id)) {
  148. return;
  149. }
  150. remove_node(p_type, p_node_id);
  151. add_node(p_type, p_node_id);
  152. }
  153. void VisualShaderGraphPlugin::set_input_port_default_value(VisualShader::Type p_type, int p_node_id, int p_port_id, Variant p_value) {
  154. if (p_type != visual_shader->get_shader_type() || !links.has(p_node_id)) {
  155. return;
  156. }
  157. Button *button = links[p_node_id].input_ports[p_port_id].default_input_button;
  158. switch (p_value.get_type()) {
  159. case Variant::COLOR: {
  160. button->set_custom_minimum_size(Size2(30, 0) * EDSCALE);
  161. if (!button->is_connected("draw", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_draw_color_over_button))) {
  162. button->connect("draw", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_draw_color_over_button), varray(button, p_value));
  163. }
  164. } break;
  165. case Variant::BOOL: {
  166. button->set_text(((bool)p_value) ? "true" : "false");
  167. } break;
  168. case Variant::INT:
  169. case Variant::FLOAT: {
  170. button->set_text(String::num(p_value, 4));
  171. } break;
  172. case Variant::VECTOR3: {
  173. Vector3 v = p_value;
  174. button->set_text(String::num(v.x, 3) + "," + String::num(v.y, 3) + "," + String::num(v.z, 3));
  175. } break;
  176. default: {
  177. }
  178. }
  179. }
  180. void VisualShaderGraphPlugin::set_uniform_name(VisualShader::Type p_type, int p_node_id, const String &p_name) {
  181. if (visual_shader->get_shader_type() == p_type && links.has(p_node_id) && links[p_node_id].uniform_name != nullptr) {
  182. links[p_node_id].uniform_name->set_text(p_name);
  183. }
  184. }
  185. void VisualShaderGraphPlugin::update_curve(int p_node_id) {
  186. if (links.has(p_node_id) && links[p_node_id].curve_editor) {
  187. if (((VisualShaderNodeCurveTexture *)links[p_node_id].visual_node)->get_texture().is_valid()) {
  188. links[p_node_id].curve_editor->set_curve(((VisualShaderNodeCurveTexture *)links[p_node_id].visual_node)->get_texture()->get_curve());
  189. }
  190. }
  191. }
  192. int VisualShaderGraphPlugin::get_constant_index(float p_constant) const {
  193. for (int i = 0; i < MAX_FLOAT_CONST_DEFS; i++) {
  194. if (Math::is_equal_approx(p_constant, float_constant_defs[i].value)) {
  195. return i + 1;
  196. }
  197. }
  198. return 0;
  199. }
  200. void VisualShaderGraphPlugin::update_constant(VisualShader::Type p_type, int p_node_id) {
  201. if (p_type != visual_shader->get_shader_type() || !links.has(p_node_id) || !links[p_node_id].const_op) {
  202. return;
  203. }
  204. VisualShaderNodeFloatConstant *float_const = Object::cast_to<VisualShaderNodeFloatConstant>(links[p_node_id].visual_node);
  205. if (!float_const) {
  206. return;
  207. }
  208. links[p_node_id].const_op->select(get_constant_index(float_const->get_constant()));
  209. links[p_node_id].graph_node->set_size(Size2(-1, -1));
  210. }
  211. void VisualShaderGraphPlugin::set_expression(VisualShader::Type p_type, int p_node_id, const String &p_expression) {
  212. if (p_type != visual_shader->get_shader_type() || !links.has(p_node_id) || !links[p_node_id].expression_edit) {
  213. return;
  214. }
  215. links[p_node_id].expression_edit->set_text(p_expression);
  216. }
  217. void VisualShaderGraphPlugin::update_node_size(int p_node_id) {
  218. if (!links.has(p_node_id)) {
  219. return;
  220. }
  221. links[p_node_id].graph_node->set_size(Size2(-1, -1));
  222. }
  223. void VisualShaderGraphPlugin::register_default_input_button(int p_node_id, int p_port_id, Button *p_button) {
  224. links[p_node_id].input_ports.insert(p_port_id, { p_button });
  225. }
  226. void VisualShaderGraphPlugin::register_constant_option_btn(int p_node_id, OptionButton *p_button) {
  227. links[p_node_id].const_op = p_button;
  228. }
  229. void VisualShaderGraphPlugin::register_expression_edit(int p_node_id, CodeEdit *p_expression_edit) {
  230. links[p_node_id].expression_edit = p_expression_edit;
  231. }
  232. void VisualShaderGraphPlugin::register_curve_editor(int p_node_id, CurveEditor *p_curve_editor) {
  233. links[p_node_id].curve_editor = p_curve_editor;
  234. }
  235. void VisualShaderGraphPlugin::update_uniform_refs() {
  236. for (Map<int, Link>::Element *E = links.front(); E; E = E->next()) {
  237. VisualShaderNodeUniformRef *ref = Object::cast_to<VisualShaderNodeUniformRef>(E->get().visual_node);
  238. if (ref) {
  239. remove_node(E->get().type, E->key());
  240. add_node(E->get().type, E->key());
  241. }
  242. }
  243. }
  244. VisualShader::Type VisualShaderGraphPlugin::get_shader_type() const {
  245. return visual_shader->get_shader_type();
  246. }
  247. void VisualShaderGraphPlugin::set_node_position(VisualShader::Type p_type, int p_id, const Vector2 &p_position) {
  248. if (visual_shader->get_shader_type() == p_type && links.has(p_id)) {
  249. links[p_id].graph_node->set_position_offset(p_position);
  250. }
  251. }
  252. void VisualShaderGraphPlugin::set_node_size(VisualShader::Type p_type, int p_id, const Vector2 &p_size) {
  253. if (visual_shader->get_shader_type() == p_type && links.has(p_id)) {
  254. links[p_id].graph_node->set_size(p_size);
  255. }
  256. }
  257. bool VisualShaderGraphPlugin::is_preview_visible(int p_id) const {
  258. return links[p_id].preview_visible;
  259. }
  260. void VisualShaderGraphPlugin::clear_links() {
  261. links.clear();
  262. }
  263. bool VisualShaderGraphPlugin::is_dirty() const {
  264. return dirty;
  265. }
  266. void VisualShaderGraphPlugin::make_dirty(bool p_enabled) {
  267. dirty = p_enabled;
  268. }
  269. void VisualShaderGraphPlugin::register_link(VisualShader::Type p_type, int p_id, VisualShaderNode *p_visual_node, GraphNode *p_graph_node) {
  270. links.insert(p_id, { p_type, p_visual_node, p_graph_node, p_visual_node->get_output_port_for_preview() != -1, -1, Map<int, InputPort>(), Map<int, Port>(), nullptr, nullptr, nullptr, nullptr, nullptr });
  271. }
  272. void VisualShaderGraphPlugin::register_output_port(int p_node_id, int p_port, TextureButton *p_button) {
  273. links[p_node_id].output_ports.insert(p_port, { p_button });
  274. }
  275. void VisualShaderGraphPlugin::register_uniform_name(int p_node_id, LineEdit *p_uniform_name) {
  276. links[p_node_id].uniform_name = p_uniform_name;
  277. }
  278. void VisualShaderGraphPlugin::update_theme() {
  279. vector_expanded_color[0] = VisualShaderEditor::get_singleton()->get_theme_color("axis_x_color", "Editor"); // red
  280. vector_expanded_color[1] = VisualShaderEditor::get_singleton()->get_theme_color("axis_y_color", "Editor"); // green
  281. vector_expanded_color[2] = VisualShaderEditor::get_singleton()->get_theme_color("axis_z_color", "Editor"); // blue
  282. }
  283. void VisualShaderGraphPlugin::add_node(VisualShader::Type p_type, int p_id) {
  284. if (p_type != visual_shader->get_shader_type()) {
  285. return;
  286. }
  287. Control *offset;
  288. static Ref<StyleBoxEmpty> label_style = make_empty_stylebox(2, 1, 2, 1);
  289. static const Color type_color[6] = {
  290. Color(0.38, 0.85, 0.96), // scalar (float)
  291. Color(0.49, 0.78, 0.94), // scalar (int)
  292. Color(0.84, 0.49, 0.93), // vector
  293. Color(0.55, 0.65, 0.94), // boolean
  294. Color(0.96, 0.66, 0.43), // transform
  295. Color(1.0, 1.0, 0.0), // sampler
  296. };
  297. static const String vector_expanded_name[3] = {
  298. "red",
  299. "green",
  300. "blue"
  301. };
  302. Ref<VisualShaderNode> vsnode = visual_shader->get_node(p_type, p_id);
  303. Ref<VisualShaderNodeResizableBase> resizable_node = Object::cast_to<VisualShaderNodeResizableBase>(vsnode.ptr());
  304. bool is_resizable = !resizable_node.is_null();
  305. Size2 size = Size2(0, 0);
  306. Ref<VisualShaderNodeGroupBase> group_node = Object::cast_to<VisualShaderNodeGroupBase>(vsnode.ptr());
  307. bool is_group = !group_node.is_null();
  308. bool is_comment = false;
  309. Ref<VisualShaderNodeExpression> expression_node = Object::cast_to<VisualShaderNodeExpression>(group_node.ptr());
  310. bool is_expression = !expression_node.is_null();
  311. String expression = "";
  312. VisualShaderNodeCustom *custom_node = Object::cast_to<VisualShaderNodeCustom>(vsnode.ptr());
  313. if (custom_node) {
  314. custom_node->_set_initialized(true);
  315. }
  316. GraphNode *node = memnew(GraphNode);
  317. register_link(p_type, p_id, vsnode.ptr(), node);
  318. if (is_resizable) {
  319. size = resizable_node->get_size();
  320. node->set_resizable(true);
  321. node->connect("resize_request", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_node_resized), varray((int)p_type, p_id));
  322. }
  323. if (is_expression) {
  324. expression = expression_node->get_expression();
  325. }
  326. node->set_position_offset(visual_shader->get_node_position(p_type, p_id));
  327. node->set_title(vsnode->get_caption());
  328. node->set_name(itos(p_id));
  329. if (p_id >= 2) {
  330. node->set_show_close_button(true);
  331. node->connect("close_request", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_delete_node_request), varray(p_type, p_id), CONNECT_DEFERRED);
  332. }
  333. node->connect("dragged", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_node_dragged), varray(p_id));
  334. Control *custom_editor = nullptr;
  335. int port_offset = 0;
  336. if (is_group) {
  337. port_offset += 2;
  338. }
  339. if (is_resizable) {
  340. Ref<VisualShaderNodeComment> comment_node = Object::cast_to<VisualShaderNodeComment>(vsnode.ptr());
  341. if (comment_node.is_valid()) {
  342. is_comment = true;
  343. node->set_comment(true);
  344. Label *comment_label = memnew(Label);
  345. node->add_child(comment_label);
  346. comment_label->set_h_size_flags(Control::SIZE_EXPAND_FILL);
  347. comment_label->set_v_size_flags(Control::SIZE_EXPAND_FILL);
  348. comment_label->set_text(comment_node->get_description());
  349. }
  350. }
  351. Ref<VisualShaderNodeParticleEmit> emit = vsnode;
  352. if (emit.is_valid()) {
  353. node->set_custom_minimum_size(Size2(200 * EDSCALE, 0));
  354. }
  355. Ref<VisualShaderNodeUniform> uniform = vsnode;
  356. if (uniform.is_valid()) {
  357. VisualShaderEditor::get_singleton()->graph->add_child(node);
  358. VisualShaderEditor::get_singleton()->_update_created_node(node);
  359. LineEdit *uniform_name = memnew(LineEdit);
  360. register_uniform_name(p_id, uniform_name);
  361. uniform_name->set_text(uniform->get_uniform_name());
  362. node->add_child(uniform_name);
  363. uniform_name->connect("text_submitted", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_uniform_line_edit_changed), varray(p_id));
  364. uniform_name->connect("focus_exited", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_uniform_line_edit_focus_out), varray(uniform_name, p_id));
  365. if (vsnode->get_input_port_count() == 0 && vsnode->get_output_port_count() == 1 && vsnode->get_output_port_name(0) == "") {
  366. //shortcut
  367. VisualShaderNode::PortType port_right = vsnode->get_output_port_type(0);
  368. node->set_slot(0, false, VisualShaderNode::PORT_TYPE_SCALAR, Color(), true, port_right, type_color[port_right]);
  369. if (!vsnode->is_use_prop_slots()) {
  370. return;
  371. }
  372. }
  373. port_offset++;
  374. }
  375. for (int i = 0; i < VisualShaderEditor::get_singleton()->plugins.size(); i++) {
  376. vsnode->set_meta("id", p_id);
  377. vsnode->set_meta("shader_type", (int)p_type);
  378. custom_editor = VisualShaderEditor::get_singleton()->plugins.write[i]->create_editor(visual_shader, vsnode);
  379. vsnode->remove_meta("id");
  380. vsnode->remove_meta("shader_type");
  381. if (custom_editor) {
  382. if (vsnode->is_show_prop_names()) {
  383. custom_editor->call_deferred("_show_prop_names", true);
  384. }
  385. break;
  386. }
  387. }
  388. Ref<VisualShaderNodeCurveTexture> curve = vsnode;
  389. if (curve.is_valid()) {
  390. if (curve->get_texture().is_valid() && !curve->get_texture()->is_connected("changed", callable_mp(VisualShaderEditor::get_singleton()->get_graph_plugin(), &VisualShaderGraphPlugin::update_curve))) {
  391. curve->get_texture()->connect("changed", callable_mp(VisualShaderEditor::get_singleton()->get_graph_plugin(), &VisualShaderGraphPlugin::update_curve), varray(p_id));
  392. }
  393. HBoxContainer *hbox = memnew(HBoxContainer);
  394. custom_editor->set_h_size_flags(Control::SIZE_EXPAND_FILL);
  395. hbox->add_child(custom_editor);
  396. custom_editor = hbox;
  397. }
  398. Ref<VisualShaderNodeFloatConstant> float_const = vsnode;
  399. if (float_const.is_valid()) {
  400. HBoxContainer *hbox = memnew(HBoxContainer);
  401. hbox->add_child(custom_editor);
  402. OptionButton *btn = memnew(OptionButton);
  403. hbox->add_child(btn);
  404. register_constant_option_btn(p_id, btn);
  405. btn->add_item("");
  406. for (int i = 0; i < MAX_FLOAT_CONST_DEFS; i++) {
  407. btn->add_item(float_constant_defs[i].name);
  408. }
  409. btn->select(get_constant_index(float_const->get_constant()));
  410. btn->connect("item_selected", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_float_constant_selected), varray(p_id));
  411. custom_editor = hbox;
  412. }
  413. if (custom_editor && !vsnode->is_use_prop_slots() && vsnode->get_output_port_count() > 0 && vsnode->get_output_port_name(0) == "" && (vsnode->get_input_port_count() == 0 || vsnode->get_input_port_name(0) == "")) {
  414. //will be embedded in first port
  415. } else if (custom_editor) {
  416. port_offset++;
  417. node->add_child(custom_editor);
  418. if (curve.is_valid()) {
  419. VisualShaderEditor::get_singleton()->graph->add_child(node);
  420. VisualShaderEditor::get_singleton()->_update_created_node(node);
  421. CurveEditor *curve_editor = memnew(CurveEditor);
  422. node->add_child(curve_editor);
  423. register_curve_editor(p_id, curve_editor);
  424. curve_editor->set_custom_minimum_size(Size2(300, 0));
  425. curve_editor->set_h_size_flags(Control::SIZE_EXPAND_FILL);
  426. if (curve->get_texture().is_valid()) {
  427. curve_editor->set_curve(curve->get_texture()->get_curve());
  428. }
  429. TextureButton *preview = memnew(TextureButton);
  430. preview->set_toggle_mode(true);
  431. preview->set_normal_texture(VisualShaderEditor::get_singleton()->get_theme_icon("GuiVisibilityHidden", "EditorIcons"));
  432. preview->set_pressed_texture(VisualShaderEditor::get_singleton()->get_theme_icon("GuiVisibilityVisible", "EditorIcons"));
  433. preview->set_v_size_flags(Control::SIZE_SHRINK_CENTER);
  434. register_output_port(p_id, 0, preview);
  435. preview->connect("pressed", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_preview_select_port), varray(p_id, 0), CONNECT_DEFERRED);
  436. custom_editor->add_child(preview);
  437. VisualShaderNode::PortType port_left = vsnode->get_input_port_type(0);
  438. VisualShaderNode::PortType port_right = vsnode->get_output_port_type(0);
  439. node->set_slot(0, true, port_left, type_color[port_left], true, port_right, type_color[port_right]);
  440. VisualShaderEditor::get_singleton()->call_deferred("_set_node_size", (int)p_type, p_id, size);
  441. }
  442. if (vsnode->is_use_prop_slots()) {
  443. return;
  444. }
  445. custom_editor = nullptr;
  446. }
  447. if (is_group) {
  448. offset = memnew(Control);
  449. offset->set_custom_minimum_size(Size2(0, 6 * EDSCALE));
  450. node->add_child(offset);
  451. if (group_node->is_editable()) {
  452. HBoxContainer *hb2 = memnew(HBoxContainer);
  453. String input_port_name = "input" + itos(group_node->get_free_input_port_id());
  454. String output_port_name = "output" + itos(group_node->get_free_output_port_id());
  455. for (int i = 0; i < MAX(vsnode->get_input_port_count(), vsnode->get_output_port_count()); i++) {
  456. if (i < vsnode->get_input_port_count()) {
  457. if (input_port_name == vsnode->get_input_port_name(i)) {
  458. input_port_name = "_" + input_port_name;
  459. }
  460. }
  461. if (i < vsnode->get_output_port_count()) {
  462. if (output_port_name == vsnode->get_output_port_name(i)) {
  463. output_port_name = "_" + output_port_name;
  464. }
  465. }
  466. }
  467. Button *add_input_btn = memnew(Button);
  468. add_input_btn->set_text(TTR("Add Input"));
  469. add_input_btn->connect("pressed", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_add_input_port), varray(p_id, group_node->get_free_input_port_id(), VisualShaderNode::PORT_TYPE_VECTOR, input_port_name), CONNECT_DEFERRED);
  470. hb2->add_child(add_input_btn);
  471. hb2->add_spacer();
  472. Button *add_output_btn = memnew(Button);
  473. add_output_btn->set_text(TTR("Add Output"));
  474. add_output_btn->connect("pressed", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_add_output_port), varray(p_id, group_node->get_free_output_port_id(), VisualShaderNode::PORT_TYPE_VECTOR, output_port_name), CONNECT_DEFERRED);
  475. hb2->add_child(add_output_btn);
  476. node->add_child(hb2);
  477. }
  478. }
  479. int output_port_count = 0;
  480. for (int i = 0; i < vsnode->get_output_port_count(); i++) {
  481. if (vsnode->_is_output_port_expanded(i)) {
  482. if (vsnode->get_output_port_type(i) == VisualShaderNode::PORT_TYPE_VECTOR) {
  483. output_port_count += 3;
  484. }
  485. }
  486. output_port_count++;
  487. }
  488. int max_ports = MAX(vsnode->get_input_port_count(), output_port_count);
  489. VisualShaderNode::PortType expanded_type = VisualShaderNode::PORT_TYPE_SCALAR;
  490. int expanded_port_counter = 0;
  491. for (int i = 0, j = 0; i < max_ports; i++, j++) {
  492. if (expanded_type == VisualShaderNode::PORT_TYPE_VECTOR && expanded_port_counter >= 3) {
  493. expanded_type = VisualShaderNode::PORT_TYPE_SCALAR;
  494. expanded_port_counter = 0;
  495. i -= 3;
  496. }
  497. if (vsnode->is_port_separator(i)) {
  498. node->add_child(memnew(HSeparator));
  499. port_offset++;
  500. }
  501. bool valid_left = j < vsnode->get_input_port_count();
  502. VisualShaderNode::PortType port_left = VisualShaderNode::PORT_TYPE_SCALAR;
  503. bool port_left_used = false;
  504. String name_left;
  505. if (valid_left) {
  506. name_left = vsnode->get_input_port_name(i);
  507. port_left = vsnode->get_input_port_type(i);
  508. for (List<VisualShader::Connection>::Element *E = connections.front(); E; E = E->next()) {
  509. if (E->get().to_node == p_id && E->get().to_port == j) {
  510. port_left_used = true;
  511. }
  512. }
  513. }
  514. bool valid_right = true;
  515. VisualShaderNode::PortType port_right = VisualShaderNode::PORT_TYPE_SCALAR;
  516. String name_right;
  517. if (expanded_type == VisualShaderNode::PORT_TYPE_SCALAR) {
  518. valid_right = i < vsnode->get_output_port_count();
  519. if (valid_right) {
  520. name_right = vsnode->get_output_port_name(i);
  521. port_right = vsnode->get_output_port_type(i);
  522. }
  523. } else {
  524. name_right = vector_expanded_name[expanded_port_counter++];
  525. }
  526. HBoxContainer *hb = memnew(HBoxContainer);
  527. hb->add_theme_constant_override("separation", 7 * EDSCALE);
  528. Variant default_value;
  529. if (valid_left && !port_left_used) {
  530. default_value = vsnode->get_input_port_default_value(i);
  531. }
  532. Button *button = memnew(Button);
  533. hb->add_child(button);
  534. register_default_input_button(p_id, i, button);
  535. button->connect("pressed", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_edit_port_default_input), varray(button, p_id, i));
  536. if (default_value.get_type() != Variant::NIL) { // only a label
  537. set_input_port_default_value(p_type, p_id, i, default_value);
  538. } else {
  539. button->hide();
  540. }
  541. if (i == 0 && custom_editor) {
  542. hb->add_child(custom_editor);
  543. custom_editor->set_h_size_flags(Control::SIZE_EXPAND_FILL);
  544. } else {
  545. if (valid_left) {
  546. if (is_group) {
  547. OptionButton *type_box = memnew(OptionButton);
  548. hb->add_child(type_box);
  549. type_box->add_item(TTR("Float"));
  550. type_box->add_item(TTR("Int"));
  551. type_box->add_item(TTR("Vector"));
  552. type_box->add_item(TTR("Boolean"));
  553. type_box->add_item(TTR("Transform"));
  554. type_box->add_item(TTR("Sampler"));
  555. type_box->select(group_node->get_input_port_type(i));
  556. type_box->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  557. type_box->connect("item_selected", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_change_input_port_type), varray(p_id, i), CONNECT_DEFERRED);
  558. LineEdit *name_box = memnew(LineEdit);
  559. hb->add_child(name_box);
  560. name_box->set_custom_minimum_size(Size2(65 * EDSCALE, 0));
  561. name_box->set_h_size_flags(Control::SIZE_EXPAND_FILL);
  562. name_box->set_text(name_left);
  563. name_box->connect("text_submitted", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_change_input_port_name), varray(name_box, p_id, i), CONNECT_DEFERRED);
  564. name_box->connect("focus_exited", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_port_name_focus_out), varray(name_box, p_id, i, false), CONNECT_DEFERRED);
  565. Button *remove_btn = memnew(Button);
  566. remove_btn->set_icon(EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Remove", "EditorIcons"));
  567. remove_btn->set_tooltip(TTR("Remove") + " " + name_left);
  568. remove_btn->connect("pressed", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_remove_input_port), varray(p_id, i), CONNECT_DEFERRED);
  569. hb->add_child(remove_btn);
  570. } else {
  571. Label *label = memnew(Label);
  572. label->set_text(name_left);
  573. label->add_theme_style_override("normal", label_style); //more compact
  574. hb->add_child(label);
  575. if (vsnode->get_input_port_default_hint(i) != "" && !port_left_used) {
  576. Label *hint_label = memnew(Label);
  577. hint_label->set_text("[" + vsnode->get_input_port_default_hint(i) + "]");
  578. hint_label->add_theme_color_override("font_color", VisualShaderEditor::get_singleton()->get_theme_color("font_readonly_color", "TextEdit"));
  579. hint_label->add_theme_style_override("normal", label_style);
  580. hb->add_child(hint_label);
  581. }
  582. }
  583. }
  584. if (!is_group) {
  585. hb->add_spacer();
  586. }
  587. if (valid_right) {
  588. if (is_group) {
  589. Button *remove_btn = memnew(Button);
  590. remove_btn->set_icon(EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Remove", "EditorIcons"));
  591. remove_btn->set_tooltip(TTR("Remove") + " " + name_left);
  592. remove_btn->connect("pressed", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_remove_output_port), varray(p_id, i), CONNECT_DEFERRED);
  593. hb->add_child(remove_btn);
  594. LineEdit *name_box = memnew(LineEdit);
  595. hb->add_child(name_box);
  596. name_box->set_custom_minimum_size(Size2(65 * EDSCALE, 0));
  597. name_box->set_h_size_flags(Control::SIZE_EXPAND_FILL);
  598. name_box->set_text(name_right);
  599. name_box->connect("text_submitted", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_change_output_port_name), varray(name_box, p_id, i), CONNECT_DEFERRED);
  600. name_box->connect("focus_exited", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_port_name_focus_out), varray(name_box, p_id, i, true), CONNECT_DEFERRED);
  601. OptionButton *type_box = memnew(OptionButton);
  602. hb->add_child(type_box);
  603. type_box->add_item(TTR("Float"));
  604. type_box->add_item(TTR("Int"));
  605. type_box->add_item(TTR("Vector"));
  606. type_box->add_item(TTR("Boolean"));
  607. type_box->add_item(TTR("Transform"));
  608. type_box->select(group_node->get_output_port_type(i));
  609. type_box->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  610. type_box->connect("item_selected", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_change_output_port_type), varray(p_id, i), CONNECT_DEFERRED);
  611. } else {
  612. Label *label = memnew(Label);
  613. label->set_text(name_right);
  614. label->add_theme_style_override("normal", label_style); //more compact
  615. hb->add_child(label);
  616. }
  617. }
  618. }
  619. if (valid_right) {
  620. if (vsnode->is_output_port_expandable(i)) {
  621. TextureButton *expand = memnew(TextureButton);
  622. expand->set_toggle_mode(true);
  623. expand->set_normal_texture(VisualShaderEditor::get_singleton()->get_theme_icon("GuiTreeArrowDown", "EditorIcons"));
  624. expand->set_pressed_texture(VisualShaderEditor::get_singleton()->get_theme_icon("GuiTreeArrowRight", "EditorIcons"));
  625. expand->set_v_size_flags(Control::SIZE_SHRINK_CENTER);
  626. expand->set_pressed(vsnode->_is_output_port_expanded(i));
  627. expand->connect("pressed", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_expand_output_port), varray(p_id, i, !vsnode->_is_output_port_expanded(i)), CONNECT_DEFERRED);
  628. hb->add_child(expand);
  629. }
  630. if (visual_shader->get_shader_type() == VisualShader::TYPE_FRAGMENT && port_right != VisualShaderNode::PORT_TYPE_TRANSFORM && port_right != VisualShaderNode::PORT_TYPE_SAMPLER) {
  631. TextureButton *preview = memnew(TextureButton);
  632. preview->set_toggle_mode(true);
  633. preview->set_normal_texture(VisualShaderEditor::get_singleton()->get_theme_icon("GuiVisibilityHidden", "EditorIcons"));
  634. preview->set_pressed_texture(VisualShaderEditor::get_singleton()->get_theme_icon("GuiVisibilityVisible", "EditorIcons"));
  635. preview->set_v_size_flags(Control::SIZE_SHRINK_CENTER);
  636. register_output_port(p_id, j, preview);
  637. preview->connect("pressed", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_preview_select_port), varray(p_id, j), CONNECT_DEFERRED);
  638. hb->add_child(preview);
  639. }
  640. }
  641. if (is_group) {
  642. offset = memnew(Control);
  643. offset->set_custom_minimum_size(Size2(0, 5 * EDSCALE));
  644. node->add_child(offset);
  645. port_offset++;
  646. }
  647. node->add_child(hb);
  648. if (expanded_type != VisualShaderNode::PORT_TYPE_SCALAR) {
  649. continue;
  650. }
  651. node->set_slot(i + port_offset, valid_left, port_left, type_color[port_left], valid_right, port_right, type_color[port_right]);
  652. if (vsnode->_is_output_port_expanded(i)) {
  653. if (vsnode->get_output_port_type(i) == VisualShaderNode::PORT_TYPE_VECTOR) {
  654. port_offset++;
  655. valid_left = (i + 1) < vsnode->get_input_port_count();
  656. port_left = VisualShaderNode::PORT_TYPE_SCALAR;
  657. if (valid_left) {
  658. port_left = vsnode->get_input_port_type(i + 1);
  659. }
  660. node->set_slot(i + port_offset, valid_left, port_left, type_color[port_left], true, VisualShaderNode::PORT_TYPE_SCALAR, vector_expanded_color[0]);
  661. port_offset++;
  662. valid_left = (i + 2) < vsnode->get_input_port_count();
  663. port_left = VisualShaderNode::PORT_TYPE_SCALAR;
  664. if (valid_left) {
  665. port_left = vsnode->get_input_port_type(i + 2);
  666. }
  667. node->set_slot(i + port_offset, valid_left, port_left, type_color[port_left], true, VisualShaderNode::PORT_TYPE_SCALAR, vector_expanded_color[1]);
  668. port_offset++;
  669. valid_left = (i + 3) < vsnode->get_input_port_count();
  670. port_left = VisualShaderNode::PORT_TYPE_SCALAR;
  671. if (valid_left) {
  672. port_left = vsnode->get_input_port_type(i + 3);
  673. }
  674. node->set_slot(i + port_offset, valid_left, port_left, type_color[port_left], true, VisualShaderNode::PORT_TYPE_SCALAR, vector_expanded_color[2]);
  675. expanded_type = VisualShaderNode::PORT_TYPE_VECTOR;
  676. }
  677. }
  678. }
  679. if (vsnode->get_output_port_for_preview() >= 0) {
  680. show_port_preview(p_type, p_id, vsnode->get_output_port_for_preview());
  681. }
  682. offset = memnew(Control);
  683. offset->set_custom_minimum_size(Size2(0, 4 * EDSCALE));
  684. node->add_child(offset);
  685. String error = vsnode->get_warning(visual_shader->get_mode(), p_type);
  686. if (error != String()) {
  687. Label *error_label = memnew(Label);
  688. error_label->add_theme_color_override("font_color", VisualShaderEditor::get_singleton()->get_theme_color("error_color", "Editor"));
  689. error_label->set_text(error);
  690. node->add_child(error_label);
  691. }
  692. if (is_expression) {
  693. CodeEdit *expression_box = memnew(CodeEdit);
  694. Ref<CodeHighlighter> expression_syntax_highlighter;
  695. expression_syntax_highlighter.instantiate();
  696. expression_node->set_ctrl_pressed(expression_box, 0);
  697. node->add_child(expression_box);
  698. register_expression_edit(p_id, expression_box);
  699. Color background_color = EDITOR_GET("text_editor/highlighting/background_color");
  700. Color text_color = EDITOR_GET("text_editor/highlighting/text_color");
  701. Color keyword_color = EDITOR_GET("text_editor/highlighting/keyword_color");
  702. Color control_flow_keyword_color = EDITOR_GET("text_editor/highlighting/control_flow_keyword_color");
  703. Color comment_color = EDITOR_GET("text_editor/highlighting/comment_color");
  704. Color symbol_color = EDITOR_GET("text_editor/highlighting/symbol_color");
  705. Color function_color = EDITOR_GET("text_editor/highlighting/function_color");
  706. Color number_color = EDITOR_GET("text_editor/highlighting/number_color");
  707. Color members_color = EDITOR_GET("text_editor/highlighting/member_variable_color");
  708. expression_box->set_syntax_highlighter(expression_syntax_highlighter);
  709. expression_box->add_theme_color_override("background_color", background_color);
  710. for (List<String>::Element *E = VisualShaderEditor::get_singleton()->keyword_list.front(); E; E = E->next()) {
  711. if (ShaderLanguage::is_control_flow_keyword(E->get())) {
  712. expression_syntax_highlighter->add_keyword_color(E->get(), control_flow_keyword_color);
  713. } else {
  714. expression_syntax_highlighter->add_keyword_color(E->get(), keyword_color);
  715. }
  716. }
  717. expression_box->add_theme_font_override("font", VisualShaderEditor::get_singleton()->get_theme_font("expression", "EditorFonts"));
  718. expression_box->add_theme_font_size_override("font_size", VisualShaderEditor::get_singleton()->get_theme_font_size("expression_size", "EditorFonts"));
  719. expression_box->add_theme_color_override("font_color", text_color);
  720. expression_syntax_highlighter->set_number_color(number_color);
  721. expression_syntax_highlighter->set_symbol_color(symbol_color);
  722. expression_syntax_highlighter->set_function_color(function_color);
  723. expression_syntax_highlighter->set_member_variable_color(members_color);
  724. expression_syntax_highlighter->add_color_region("/*", "*/", comment_color, false);
  725. expression_syntax_highlighter->add_color_region("//", "", comment_color, true);
  726. expression_box->clear_comment_delimiters();
  727. expression_box->add_comment_delimiter("/*", "*/", false);
  728. expression_box->add_comment_delimiter("//", "", true);
  729. expression_box->set_text(expression);
  730. expression_box->set_context_menu_enabled(false);
  731. expression_box->set_draw_line_numbers(true);
  732. expression_box->connect("focus_exited", callable_mp(VisualShaderEditor::get_singleton(), &VisualShaderEditor::_expression_focus_out), varray(expression_box, p_id));
  733. }
  734. if (!uniform.is_valid()) {
  735. VisualShaderEditor::get_singleton()->graph->add_child(node);
  736. if (is_comment) {
  737. VisualShaderEditor::get_singleton()->graph->move_child(node, 0); // to prevents a bug where comment node overlaps its content
  738. }
  739. VisualShaderEditor::get_singleton()->_update_created_node(node);
  740. if (is_resizable) {
  741. VisualShaderEditor::get_singleton()->call_deferred("_set_node_size", (int)p_type, p_id, size);
  742. }
  743. }
  744. }
  745. void VisualShaderGraphPlugin::remove_node(VisualShader::Type p_type, int p_id) {
  746. if (visual_shader->get_shader_type() == p_type && links.has(p_id)) {
  747. links[p_id].graph_node->get_parent()->remove_child(links[p_id].graph_node);
  748. memdelete(links[p_id].graph_node);
  749. links.erase(p_id);
  750. }
  751. }
  752. void VisualShaderGraphPlugin::connect_nodes(VisualShader::Type p_type, int p_from_node, int p_from_port, int p_to_node, int p_to_port) {
  753. if (visual_shader->get_shader_type() == p_type) {
  754. VisualShaderEditor::get_singleton()->graph->connect_node(itos(p_from_node), p_from_port, itos(p_to_node), p_to_port);
  755. if (links[p_to_node].input_ports.has(p_to_port) && links[p_to_node].input_ports[p_to_port].default_input_button != nullptr) {
  756. links[p_to_node].input_ports[p_to_port].default_input_button->hide();
  757. }
  758. }
  759. }
  760. void VisualShaderGraphPlugin::disconnect_nodes(VisualShader::Type p_type, int p_from_node, int p_from_port, int p_to_node, int p_to_port) {
  761. if (visual_shader->get_shader_type() == p_type) {
  762. VisualShaderEditor::get_singleton()->graph->disconnect_node(itos(p_from_node), p_from_port, itos(p_to_node), p_to_port);
  763. if (links[p_to_node].input_ports.has(p_to_port) && links[p_to_node].input_ports[p_to_port].default_input_button != nullptr && links[p_to_node].visual_node->get_input_port_default_value(p_to_port).get_type() != Variant::NIL) {
  764. links[p_to_node].input_ports[p_to_port].default_input_button->show();
  765. set_input_port_default_value(p_type, p_to_node, p_to_port, links[p_to_node].visual_node->get_input_port_default_value(p_to_port));
  766. }
  767. }
  768. }
  769. VisualShaderGraphPlugin::~VisualShaderGraphPlugin() {
  770. }
  771. /////////////////
  772. void VisualShaderEditor::edit(VisualShader *p_visual_shader) {
  773. bool changed = false;
  774. if (p_visual_shader) {
  775. if (visual_shader.is_null()) {
  776. changed = true;
  777. } else {
  778. if (visual_shader.ptr() != p_visual_shader) {
  779. changed = true;
  780. }
  781. }
  782. visual_shader = Ref<VisualShader>(p_visual_shader);
  783. graph_plugin->register_shader(visual_shader.ptr());
  784. if (!visual_shader->is_connected("changed", callable_mp(this, &VisualShaderEditor::_update_preview))) {
  785. visual_shader->connect("changed", callable_mp(this, &VisualShaderEditor::_update_preview));
  786. }
  787. #ifndef DISABLE_DEPRECATED
  788. String version = VERSION_BRANCH;
  789. if (visual_shader->get_version() != version) {
  790. visual_shader->update_version(version);
  791. }
  792. #endif
  793. visual_shader->set_graph_offset(graph->get_scroll_ofs() / EDSCALE);
  794. _set_mode(visual_shader->get_mode());
  795. } else {
  796. if (visual_shader.is_valid()) {
  797. if (visual_shader->is_connected("changed", callable_mp(this, &VisualShaderEditor::_update_preview))) {
  798. visual_shader->disconnect("changed", callable_mp(this, &VisualShaderEditor::_update_preview));
  799. }
  800. }
  801. visual_shader.unref();
  802. }
  803. if (visual_shader.is_null()) {
  804. hide();
  805. } else {
  806. if (changed) { // to avoid tree collapse
  807. _clear_buffer();
  808. _update_options_menu();
  809. _update_preview();
  810. _update_graph();
  811. }
  812. }
  813. }
  814. void VisualShaderEditor::add_plugin(const Ref<VisualShaderNodePlugin> &p_plugin) {
  815. if (plugins.find(p_plugin) != -1) {
  816. return;
  817. }
  818. plugins.push_back(p_plugin);
  819. }
  820. void VisualShaderEditor::remove_plugin(const Ref<VisualShaderNodePlugin> &p_plugin) {
  821. plugins.erase(p_plugin);
  822. }
  823. void VisualShaderEditor::clear_custom_types() {
  824. for (int i = 0; i < add_options.size(); i++) {
  825. if (add_options[i].is_custom) {
  826. add_options.remove(i);
  827. i--;
  828. }
  829. }
  830. }
  831. 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) {
  832. ERR_FAIL_COND(!p_name.is_valid_identifier());
  833. ERR_FAIL_COND(!p_script.is_valid());
  834. for (int i = 0; i < add_options.size(); i++) {
  835. if (add_options[i].is_custom) {
  836. if (add_options[i].script == p_script) {
  837. return;
  838. }
  839. }
  840. }
  841. AddOption ao;
  842. ao.name = p_name;
  843. ao.script = p_script;
  844. ao.return_type = p_return_icon_type;
  845. ao.description = p_description;
  846. ao.category = p_category;
  847. ao.highend = p_highend;
  848. ao.is_custom = true;
  849. bool begin = false;
  850. String root = p_category.split("/")[0];
  851. for (int i = 0; i < add_options.size(); i++) {
  852. if (add_options[i].is_custom) {
  853. if (add_options[i].category == root) {
  854. if (!begin) {
  855. begin = true;
  856. }
  857. } else {
  858. if (begin) {
  859. add_options.insert(i, ao);
  860. return;
  861. }
  862. }
  863. }
  864. }
  865. add_options.push_back(ao);
  866. }
  867. bool VisualShaderEditor::_is_available(int p_mode) {
  868. int current_mode = edit_type->get_selected();
  869. if (p_mode != -1) {
  870. switch (current_mode) {
  871. case 0: // Vertex / Emit
  872. current_mode = 1;
  873. break;
  874. case 1: // Fragment / Process
  875. current_mode = 2;
  876. break;
  877. case 2: // Light / Collide
  878. current_mode = 4;
  879. break;
  880. default:
  881. break;
  882. }
  883. }
  884. return (p_mode == -1 || (p_mode & current_mode) != 0);
  885. }
  886. void VisualShaderEditor::update_custom_nodes() {
  887. if (members_dialog->is_visible()) {
  888. return;
  889. }
  890. clear_custom_types();
  891. List<StringName> class_list;
  892. ScriptServer::get_global_class_list(&class_list);
  893. Dictionary added;
  894. for (int i = 0; i < class_list.size(); i++) {
  895. if (ScriptServer::get_global_class_native_base(class_list[i]) == "VisualShaderNodeCustom") {
  896. String script_path = ScriptServer::get_global_class_path(class_list[i]);
  897. Ref<Resource> res = ResourceLoader::load(script_path);
  898. ERR_FAIL_COND(res.is_null());
  899. ERR_FAIL_COND(!res->is_class("Script"));
  900. Ref<Script> script = Ref<Script>(res);
  901. Ref<VisualShaderNodeCustom> ref;
  902. ref.instantiate();
  903. ref->set_script(script);
  904. String name;
  905. if (ref->has_method("_get_name")) {
  906. name = (String)ref->call("_get_name");
  907. } else {
  908. name = "Unnamed";
  909. }
  910. String description = "";
  911. if (ref->has_method("_get_description")) {
  912. description = (String)ref->call("_get_description");
  913. }
  914. int return_icon_type = -1;
  915. if (ref->has_method("_get_return_icon_type")) {
  916. return_icon_type = (int)ref->call("_get_return_icon_type");
  917. }
  918. String category = "";
  919. if (ref->has_method("_get_category")) {
  920. category = (String)ref->call("_get_category");
  921. }
  922. String subcategory = "";
  923. if (ref->has_method("_get_subcategory")) {
  924. subcategory = (String)ref->call("_get_subcategory");
  925. }
  926. bool highend = false;
  927. if (ref->has_method("_is_highend")) {
  928. highend = (bool)ref->call("_is_highend");
  929. }
  930. Dictionary dict;
  931. dict["name"] = name;
  932. dict["script"] = script;
  933. dict["description"] = description;
  934. dict["return_icon_type"] = return_icon_type;
  935. category = category.rstrip("/");
  936. category = category.lstrip("/");
  937. category = "Addons/" + category;
  938. if (subcategory != "") {
  939. category += "/" + subcategory;
  940. }
  941. dict["category"] = category;
  942. dict["highend"] = highend;
  943. String key;
  944. key = category + "/" + name;
  945. added[key] = dict;
  946. }
  947. }
  948. Array keys = added.keys();
  949. keys.sort();
  950. for (int i = 0; i < keys.size(); i++) {
  951. const Variant &key = keys.get(i);
  952. const Dictionary &value = (Dictionary)added[key];
  953. add_custom_type(value["name"], value["script"], value["description"], value["return_icon_type"], value["category"], value["highend"]);
  954. }
  955. _update_options_menu();
  956. }
  957. String VisualShaderEditor::_get_description(int p_idx) {
  958. return add_options[p_idx].description;
  959. }
  960. void VisualShaderEditor::_update_options_menu() {
  961. node_desc->set_text("");
  962. members_dialog->get_ok_button()->set_disabled(true);
  963. members->clear();
  964. TreeItem *root = members->create_item();
  965. String filter = node_filter->get_text().strip_edges();
  966. bool use_filter = !filter.is_empty();
  967. bool is_first_item = true;
  968. Color unsupported_color = get_theme_color("error_color", "Editor");
  969. Color supported_color = get_theme_color("warning_color", "Editor");
  970. static bool low_driver = ProjectSettings::get_singleton()->get("rendering/driver/driver_name") == "GLES2";
  971. Map<String, TreeItem *> folders;
  972. int current_func = -1;
  973. if (!visual_shader.is_null()) {
  974. current_func = visual_shader->get_mode();
  975. }
  976. Vector<AddOption> custom_options;
  977. Vector<AddOption> embedded_options;
  978. for (int i = 0; i < add_options.size(); i++) {
  979. if (!use_filter || add_options[i].name.findn(filter) != -1) {
  980. if ((add_options[i].func != current_func && add_options[i].func != -1) || !_is_available(add_options[i].mode)) {
  981. continue;
  982. }
  983. const_cast<AddOption &>(add_options[i]).temp_idx = i; // save valid id
  984. if (add_options[i].is_custom) {
  985. custom_options.push_back(add_options[i]);
  986. } else {
  987. embedded_options.push_back(add_options[i]);
  988. }
  989. }
  990. }
  991. Vector<AddOption> options;
  992. SortArray<AddOption, _OptionComparator> sorter;
  993. sorter.sort(custom_options.ptrw(), custom_options.size());
  994. options.append_array(custom_options);
  995. options.append_array(embedded_options);
  996. for (int i = 0; i < options.size(); i++) {
  997. String path = options[i].category;
  998. Vector<String> subfolders = path.split("/");
  999. TreeItem *category = nullptr;
  1000. if (!folders.has(path)) {
  1001. category = root;
  1002. String path_temp = "";
  1003. for (int j = 0; j < subfolders.size(); j++) {
  1004. path_temp += subfolders[j];
  1005. if (!folders.has(path_temp)) {
  1006. category = members->create_item(category);
  1007. category->set_selectable(0, false);
  1008. category->set_collapsed(!use_filter);
  1009. category->set_text(0, subfolders[j]);
  1010. folders.insert(path_temp, category);
  1011. } else {
  1012. category = folders[path_temp];
  1013. }
  1014. }
  1015. } else {
  1016. category = folders[path];
  1017. }
  1018. TreeItem *item = members->create_item(category);
  1019. if (options[i].highend && low_driver) {
  1020. item->set_custom_color(0, unsupported_color);
  1021. } else if (options[i].highend) {
  1022. item->set_custom_color(0, supported_color);
  1023. }
  1024. item->set_text(0, options[i].name);
  1025. if (is_first_item && use_filter) {
  1026. item->select(0);
  1027. node_desc->set_text(options[i].description);
  1028. is_first_item = false;
  1029. }
  1030. switch (options[i].return_type) {
  1031. case VisualShaderNode::PORT_TYPE_SCALAR:
  1032. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("float", "EditorIcons"));
  1033. break;
  1034. case VisualShaderNode::PORT_TYPE_SCALAR_INT:
  1035. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("int", "EditorIcons"));
  1036. break;
  1037. case VisualShaderNode::PORT_TYPE_VECTOR:
  1038. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Vector3", "EditorIcons"));
  1039. break;
  1040. case VisualShaderNode::PORT_TYPE_BOOLEAN:
  1041. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("bool", "EditorIcons"));
  1042. break;
  1043. case VisualShaderNode::PORT_TYPE_TRANSFORM:
  1044. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Transform3D", "EditorIcons"));
  1045. break;
  1046. case VisualShaderNode::PORT_TYPE_SAMPLER:
  1047. item->set_icon(0, EditorNode::get_singleton()->get_gui_base()->get_theme_icon("ImageTexture", "EditorIcons"));
  1048. break;
  1049. default:
  1050. break;
  1051. }
  1052. item->set_meta("id", options[i].temp_idx);
  1053. }
  1054. }
  1055. void VisualShaderEditor::_set_mode(int p_which) {
  1056. if (p_which == VisualShader::MODE_SKY) {
  1057. edit_type_standard->set_visible(false);
  1058. edit_type_particles->set_visible(false);
  1059. edit_type_sky->set_visible(true);
  1060. edit_type = edit_type_sky;
  1061. custom_mode_box->set_visible(false);
  1062. mode = MODE_FLAGS_SKY;
  1063. } else if (p_which == VisualShader::MODE_PARTICLES) {
  1064. edit_type_standard->set_visible(false);
  1065. edit_type_particles->set_visible(true);
  1066. edit_type_sky->set_visible(false);
  1067. edit_type = edit_type_particles;
  1068. if ((edit_type->get_selected() + 3) > VisualShader::TYPE_PROCESS) {
  1069. custom_mode_box->set_visible(false);
  1070. } else {
  1071. custom_mode_box->set_visible(true);
  1072. }
  1073. mode = MODE_FLAGS_PARTICLES;
  1074. } else {
  1075. edit_type_particles->set_visible(false);
  1076. edit_type_standard->set_visible(true);
  1077. edit_type_sky->set_visible(false);
  1078. edit_type = edit_type_standard;
  1079. custom_mode_box->set_visible(false);
  1080. mode = MODE_FLAGS_SPATIAL_CANVASITEM;
  1081. }
  1082. visual_shader->set_shader_type(get_current_shader_type());
  1083. }
  1084. Size2 VisualShaderEditor::get_minimum_size() const {
  1085. return Size2(10, 200);
  1086. }
  1087. void VisualShaderEditor::_draw_color_over_button(Object *obj, Color p_color) {
  1088. Button *button = Object::cast_to<Button>(obj);
  1089. if (!button) {
  1090. return;
  1091. }
  1092. Ref<StyleBox> normal = get_theme_stylebox("normal", "Button");
  1093. button->draw_rect(Rect2(normal->get_offset(), button->get_size() - normal->get_minimum_size()), p_color);
  1094. }
  1095. void VisualShaderEditor::_update_created_node(GraphNode *node) {
  1096. const Ref<StyleBoxFlat> sb = node->get_theme_stylebox("frame", "GraphNode");
  1097. Color c = sb->get_border_color();
  1098. const Color mono_color = ((c.r + c.g + c.b) / 3) < 0.7 ? Color(1.0, 1.0, 1.0, 0.85) : Color(0.0, 0.0, 0.0, 0.85);
  1099. c = mono_color;
  1100. node->add_theme_color_override("title_color", c);
  1101. c.a = 0.7;
  1102. node->add_theme_color_override("close_color", c);
  1103. node->add_theme_color_override("resizer_color", c);
  1104. }
  1105. void VisualShaderEditor::_update_uniforms(bool p_update_refs) {
  1106. VisualShaderNodeUniformRef::clear_uniforms();
  1107. for (int t = 0; t < VisualShader::TYPE_MAX; t++) {
  1108. Vector<int> tnodes = visual_shader->get_node_list((VisualShader::Type)t);
  1109. for (int i = 0; i < tnodes.size(); i++) {
  1110. Ref<VisualShaderNode> vsnode = visual_shader->get_node((VisualShader::Type)t, tnodes[i]);
  1111. Ref<VisualShaderNodeUniform> uniform = vsnode;
  1112. if (uniform.is_valid()) {
  1113. Ref<VisualShaderNodeFloatUniform> float_uniform = vsnode;
  1114. Ref<VisualShaderNodeIntUniform> int_uniform = vsnode;
  1115. Ref<VisualShaderNodeVec3Uniform> vec3_uniform = vsnode;
  1116. Ref<VisualShaderNodeColorUniform> color_uniform = vsnode;
  1117. Ref<VisualShaderNodeBooleanUniform> bool_uniform = vsnode;
  1118. Ref<VisualShaderNodeTransformUniform> transform_uniform = vsnode;
  1119. VisualShaderNodeUniformRef::UniformType uniform_type;
  1120. if (float_uniform.is_valid()) {
  1121. uniform_type = VisualShaderNodeUniformRef::UniformType::UNIFORM_TYPE_FLOAT;
  1122. } else if (int_uniform.is_valid()) {
  1123. uniform_type = VisualShaderNodeUniformRef::UniformType::UNIFORM_TYPE_INT;
  1124. } else if (bool_uniform.is_valid()) {
  1125. uniform_type = VisualShaderNodeUniformRef::UniformType::UNIFORM_TYPE_BOOLEAN;
  1126. } else if (vec3_uniform.is_valid()) {
  1127. uniform_type = VisualShaderNodeUniformRef::UniformType::UNIFORM_TYPE_VECTOR;
  1128. } else if (transform_uniform.is_valid()) {
  1129. uniform_type = VisualShaderNodeUniformRef::UniformType::UNIFORM_TYPE_TRANSFORM;
  1130. } else if (color_uniform.is_valid()) {
  1131. uniform_type = VisualShaderNodeUniformRef::UniformType::UNIFORM_TYPE_COLOR;
  1132. } else {
  1133. uniform_type = VisualShaderNodeUniformRef::UniformType::UNIFORM_TYPE_SAMPLER;
  1134. }
  1135. VisualShaderNodeUniformRef::add_uniform(uniform->get_uniform_name(), uniform_type);
  1136. }
  1137. }
  1138. }
  1139. if (p_update_refs) {
  1140. graph_plugin->update_uniform_refs();
  1141. }
  1142. }
  1143. void VisualShaderEditor::_update_uniform_refs(Set<String> &p_deleted_names) {
  1144. for (int i = 0; i < VisualShader::TYPE_MAX; i++) {
  1145. VisualShader::Type type = VisualShader::Type(i);
  1146. Vector<int> nodes = visual_shader->get_node_list(type);
  1147. for (int j = 0; j < nodes.size(); j++) {
  1148. if (j > 0) {
  1149. Ref<VisualShaderNodeUniformRef> ref = visual_shader->get_node(type, nodes[j]);
  1150. if (ref.is_valid()) {
  1151. if (p_deleted_names.has(ref->get_uniform_name())) {
  1152. undo_redo->add_do_method(ref.ptr(), "set_uniform_name", "[None]");
  1153. undo_redo->add_undo_method(ref.ptr(), "set_uniform_name", ref->get_uniform_name());
  1154. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", VisualShader::Type(i), nodes[j]);
  1155. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", VisualShader::Type(i), nodes[j]);
  1156. }
  1157. }
  1158. }
  1159. }
  1160. }
  1161. }
  1162. void VisualShaderEditor::_update_graph() {
  1163. if (updating) {
  1164. return;
  1165. }
  1166. if (visual_shader.is_null()) {
  1167. return;
  1168. }
  1169. graph->set_scroll_ofs(visual_shader->get_graph_offset() * EDSCALE);
  1170. VisualShader::Type type = get_current_shader_type();
  1171. graph->clear_connections();
  1172. //erase all nodes
  1173. for (int i = 0; i < graph->get_child_count(); i++) {
  1174. if (Object::cast_to<GraphNode>(graph->get_child(i))) {
  1175. Node *node = graph->get_child(i);
  1176. graph->remove_child(node);
  1177. memdelete(node);
  1178. i--;
  1179. }
  1180. }
  1181. List<VisualShader::Connection> connections;
  1182. visual_shader->get_node_connections(type, &connections);
  1183. graph_plugin->set_connections(connections);
  1184. Vector<int> nodes = visual_shader->get_node_list(type);
  1185. _update_uniforms(false);
  1186. graph_plugin->clear_links();
  1187. graph_plugin->make_dirty(true);
  1188. graph_plugin->update_theme();
  1189. for (int n_i = 0; n_i < nodes.size(); n_i++) {
  1190. graph_plugin->add_node(type, nodes[n_i]);
  1191. }
  1192. graph_plugin->make_dirty(false);
  1193. for (List<VisualShader::Connection>::Element *E = connections.front(); E; E = E->next()) {
  1194. int from = E->get().from_node;
  1195. int from_idx = E->get().from_port;
  1196. int to = E->get().to_node;
  1197. int to_idx = E->get().to_port;
  1198. graph->connect_node(itos(from), from_idx, itos(to), to_idx);
  1199. }
  1200. float graph_minimap_opacity = EditorSettings::get_singleton()->get("editors/visual_editors/minimap_opacity");
  1201. graph->set_minimap_opacity(graph_minimap_opacity);
  1202. }
  1203. VisualShader::Type VisualShaderEditor::get_current_shader_type() const {
  1204. VisualShader::Type type;
  1205. if (mode & MODE_FLAGS_PARTICLES) {
  1206. type = VisualShader::Type(edit_type->get_selected() + 3 + (custom_mode_enabled ? 3 : 0));
  1207. } else if (mode & MODE_FLAGS_SKY) {
  1208. type = VisualShader::Type(edit_type->get_selected() + 8);
  1209. } else {
  1210. type = VisualShader::Type(edit_type->get_selected());
  1211. }
  1212. return type;
  1213. }
  1214. void VisualShaderEditor::_add_input_port(int p_node, int p_port, int p_port_type, const String &p_name) {
  1215. VisualShader::Type type = get_current_shader_type();
  1216. Ref<VisualShaderNodeExpression> node = visual_shader->get_node(type, p_node);
  1217. if (node.is_null()) {
  1218. return;
  1219. }
  1220. undo_redo->create_action(TTR("Add Input Port"));
  1221. undo_redo->add_do_method(node.ptr(), "add_input_port", p_port, p_port_type, p_name);
  1222. undo_redo->add_undo_method(node.ptr(), "remove_input_port", p_port);
  1223. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node);
  1224. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node);
  1225. undo_redo->commit_action();
  1226. }
  1227. void VisualShaderEditor::_add_output_port(int p_node, int p_port, int p_port_type, const String &p_name) {
  1228. VisualShader::Type type = get_current_shader_type();
  1229. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  1230. if (node.is_null()) {
  1231. return;
  1232. }
  1233. undo_redo->create_action(TTR("Add Output Port"));
  1234. undo_redo->add_do_method(node.ptr(), "add_output_port", p_port, p_port_type, p_name);
  1235. undo_redo->add_undo_method(node.ptr(), "remove_output_port", p_port);
  1236. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node);
  1237. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node);
  1238. undo_redo->commit_action();
  1239. }
  1240. void VisualShaderEditor::_change_input_port_type(int p_type, int p_node, int p_port) {
  1241. VisualShader::Type type = get_current_shader_type();
  1242. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  1243. if (node.is_null()) {
  1244. return;
  1245. }
  1246. undo_redo->create_action(TTR("Change Input Port Type"));
  1247. undo_redo->add_do_method(node.ptr(), "set_input_port_type", p_port, p_type);
  1248. undo_redo->add_undo_method(node.ptr(), "set_input_port_type", p_port, node->get_input_port_type(p_port));
  1249. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node);
  1250. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node);
  1251. undo_redo->commit_action();
  1252. }
  1253. void VisualShaderEditor::_change_output_port_type(int p_type, int p_node, int p_port) {
  1254. VisualShader::Type type = get_current_shader_type();
  1255. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  1256. if (node.is_null()) {
  1257. return;
  1258. }
  1259. undo_redo->create_action(TTR("Change Output Port Type"));
  1260. undo_redo->add_do_method(node.ptr(), "set_output_port_type", p_port, p_type);
  1261. undo_redo->add_undo_method(node.ptr(), "set_output_port_type", p_port, node->get_output_port_type(p_port));
  1262. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node);
  1263. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node);
  1264. undo_redo->commit_action();
  1265. }
  1266. void VisualShaderEditor::_change_input_port_name(const String &p_text, Object *p_line_edit, int p_node_id, int p_port_id) {
  1267. VisualShader::Type type = get_current_shader_type();
  1268. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node_id);
  1269. ERR_FAIL_COND(!node.is_valid());
  1270. String prev_name = node->get_input_port_name(p_port_id);
  1271. if (prev_name == p_text) {
  1272. return;
  1273. }
  1274. LineEdit *line_edit = Object::cast_to<LineEdit>(p_line_edit);
  1275. ERR_FAIL_COND(!line_edit);
  1276. String validated_name = visual_shader->validate_port_name(p_text, node.ptr(), p_port_id, false);
  1277. if (validated_name == String() || prev_name == validated_name) {
  1278. line_edit->set_text(node->get_input_port_name(p_port_id));
  1279. return;
  1280. }
  1281. undo_redo->create_action(TTR("Change Input Port Name"));
  1282. undo_redo->add_do_method(node.ptr(), "set_input_port_name", p_port_id, validated_name);
  1283. undo_redo->add_undo_method(node.ptr(), "set_input_port_name", p_port_id, node->get_input_port_name(p_port_id));
  1284. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node_id);
  1285. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node_id);
  1286. undo_redo->commit_action();
  1287. }
  1288. void VisualShaderEditor::_change_output_port_name(const String &p_text, Object *p_line_edit, int p_node_id, int p_port_id) {
  1289. VisualShader::Type type = get_current_shader_type();
  1290. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node_id);
  1291. ERR_FAIL_COND(!node.is_valid());
  1292. String prev_name = node->get_output_port_name(p_port_id);
  1293. if (prev_name == p_text) {
  1294. return;
  1295. }
  1296. LineEdit *line_edit = Object::cast_to<LineEdit>(p_line_edit);
  1297. ERR_FAIL_COND(!line_edit);
  1298. String validated_name = visual_shader->validate_port_name(p_text, node.ptr(), p_port_id, true);
  1299. if (validated_name == String() || prev_name == validated_name) {
  1300. line_edit->set_text(node->get_output_port_name(p_port_id));
  1301. return;
  1302. }
  1303. undo_redo->create_action(TTR("Change Output Port Name"));
  1304. undo_redo->add_do_method(node.ptr(), "set_output_port_name", p_port_id, validated_name);
  1305. undo_redo->add_undo_method(node.ptr(), "set_output_port_name", p_port_id, prev_name);
  1306. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node_id);
  1307. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node_id);
  1308. undo_redo->commit_action();
  1309. }
  1310. void VisualShaderEditor::_expand_output_port(int p_node, int p_port, bool p_expand) {
  1311. VisualShader::Type type = get_current_shader_type();
  1312. Ref<VisualShaderNode> node = visual_shader->get_node(type, p_node);
  1313. ERR_FAIL_COND(!node.is_valid());
  1314. if (p_expand) {
  1315. undo_redo->create_action(TTR("Expand Output Port"));
  1316. } else {
  1317. undo_redo->create_action(TTR("Shrink Output Port"));
  1318. }
  1319. undo_redo->add_do_method(node.ptr(), "_set_output_port_expanded", p_port, p_expand);
  1320. undo_redo->add_undo_method(node.ptr(), "_set_output_port_expanded", p_port, !p_expand);
  1321. int type_size = 0;
  1322. if (node->get_output_port_type(p_port) == VisualShaderNode::PORT_TYPE_VECTOR) {
  1323. type_size = 3;
  1324. }
  1325. List<VisualShader::Connection> conns;
  1326. visual_shader->get_node_connections(type, &conns);
  1327. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  1328. int from_node = E->get().from_node;
  1329. int from_port = E->get().from_port;
  1330. int to_node = E->get().to_node;
  1331. int to_port = E->get().to_port;
  1332. if (from_node == p_node) {
  1333. if (p_expand) {
  1334. if (from_port > p_port) { // reconnect ports after expanded ports
  1335. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1336. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  1337. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1338. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port, to_node, to_port);
  1339. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port + type_size, to_node, to_port);
  1340. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port + type_size, to_node, to_port);
  1341. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port + type_size, to_node, to_port);
  1342. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port + type_size, to_node, to_port);
  1343. }
  1344. } else {
  1345. if (from_port > p_port + type_size) { // reconnect ports after expanded ports
  1346. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1347. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  1348. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1349. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port, to_node, to_port);
  1350. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, from_node, from_port - type_size, to_node, to_port);
  1351. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port - type_size, to_node, to_port);
  1352. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port - type_size, to_node, to_port);
  1353. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port - type_size, to_node, to_port);
  1354. } else if (from_port > p_port) { // disconnect component ports
  1355. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1356. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  1357. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1358. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port, to_node, to_port);
  1359. }
  1360. }
  1361. }
  1362. }
  1363. int preview_port = node->get_output_port_for_preview();
  1364. if (p_expand) {
  1365. if (preview_port > p_port) {
  1366. undo_redo->add_do_method(node.ptr(), "set_output_port_for_preview", preview_port + type_size);
  1367. undo_redo->add_undo_method(node.ptr(), "set_output_port_for_preview", preview_port);
  1368. }
  1369. } else {
  1370. if (preview_port > p_port + type_size) {
  1371. undo_redo->add_do_method(node.ptr(), "set_output_port_for_preview", preview_port - type_size);
  1372. undo_redo->add_undo_method(node.ptr(), "set_output_port_for_preview", preview_port);
  1373. }
  1374. }
  1375. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node);
  1376. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node);
  1377. undo_redo->commit_action();
  1378. }
  1379. void VisualShaderEditor::_remove_input_port(int p_node, int p_port) {
  1380. VisualShader::Type type = get_current_shader_type();
  1381. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  1382. if (node.is_null()) {
  1383. return;
  1384. }
  1385. undo_redo->create_action(TTR("Remove Input Port"));
  1386. List<VisualShader::Connection> conns;
  1387. visual_shader->get_node_connections(type, &conns);
  1388. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  1389. int from_node = E->get().from_node;
  1390. int from_port = E->get().from_port;
  1391. int to_node = E->get().to_node;
  1392. int to_port = E->get().to_port;
  1393. if (to_node == p_node) {
  1394. if (to_port == p_port) {
  1395. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1396. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  1397. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1398. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port, to_node, to_port);
  1399. } else if (to_port > p_port) {
  1400. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1401. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  1402. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1403. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port, to_node, to_port);
  1404. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port - 1);
  1405. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port - 1);
  1406. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port, to_node, to_port - 1);
  1407. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port - 1);
  1408. }
  1409. }
  1410. }
  1411. undo_redo->add_do_method(node.ptr(), "remove_input_port", p_port);
  1412. 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));
  1413. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node);
  1414. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node);
  1415. undo_redo->commit_action();
  1416. }
  1417. void VisualShaderEditor::_remove_output_port(int p_node, int p_port) {
  1418. VisualShader::Type type = get_current_shader_type();
  1419. Ref<VisualShaderNodeGroupBase> node = visual_shader->get_node(type, p_node);
  1420. if (node.is_null()) {
  1421. return;
  1422. }
  1423. undo_redo->create_action(TTR("Remove Output Port"));
  1424. List<VisualShader::Connection> conns;
  1425. visual_shader->get_node_connections(type, &conns);
  1426. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  1427. int from_node = E->get().from_node;
  1428. int from_port = E->get().from_port;
  1429. int to_node = E->get().to_node;
  1430. int to_port = E->get().to_port;
  1431. if (from_node == p_node) {
  1432. if (from_port == p_port) {
  1433. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1434. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  1435. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1436. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port, to_node, to_port);
  1437. } else if (from_port > p_port) {
  1438. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1439. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port, to_node, to_port);
  1440. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port, to_node, to_port);
  1441. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port, to_node, to_port);
  1442. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes_forced", type, from_node, from_port - 1, to_node, to_port);
  1443. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_port - 1, to_node, to_port);
  1444. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_port - 1, to_node, to_port);
  1445. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_port - 1, to_node, to_port);
  1446. }
  1447. }
  1448. }
  1449. int preview_port = node->get_output_port_for_preview();
  1450. if (preview_port != -1) {
  1451. if (preview_port == p_port) {
  1452. undo_redo->add_do_method(node.ptr(), "set_output_port_for_preview", -1);
  1453. undo_redo->add_undo_method(node.ptr(), "set_output_port_for_preview", preview_port);
  1454. } else if (preview_port > p_port) {
  1455. undo_redo->add_do_method(node.ptr(), "set_output_port_for_preview", preview_port - 1);
  1456. undo_redo->add_undo_method(node.ptr(), "set_output_port_for_preview", preview_port);
  1457. }
  1458. }
  1459. undo_redo->add_do_method(node.ptr(), "remove_output_port", p_port);
  1460. 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));
  1461. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, p_node);
  1462. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type, p_node);
  1463. undo_redo->commit_action();
  1464. }
  1465. void VisualShaderEditor::_expression_focus_out(Object *code_edit, int p_node) {
  1466. VisualShader::Type type = get_current_shader_type();
  1467. Ref<VisualShaderNodeExpression> node = visual_shader->get_node(type, p_node);
  1468. if (node.is_null()) {
  1469. return;
  1470. }
  1471. CodeEdit *expression_box = Object::cast_to<CodeEdit>(code_edit);
  1472. if (node->get_expression() == expression_box->get_text()) {
  1473. return;
  1474. }
  1475. undo_redo->create_action(TTR("Set VisualShader Expression"));
  1476. undo_redo->add_do_method(node.ptr(), "set_expression", expression_box->get_text());
  1477. undo_redo->add_undo_method(node.ptr(), "set_expression", node->get_expression());
  1478. undo_redo->add_do_method(graph_plugin.ptr(), "set_expression", type, p_node, expression_box->get_text());
  1479. undo_redo->add_undo_method(graph_plugin.ptr(), "set_expression", type, p_node, node->get_expression());
  1480. undo_redo->commit_action();
  1481. }
  1482. void VisualShaderEditor::_set_node_size(int p_type, int p_node, const Vector2 &p_size) {
  1483. VisualShader::Type type = VisualShader::Type(p_type);
  1484. Ref<VisualShaderNodeResizableBase> node = visual_shader->get_node(type, p_node);
  1485. if (node.is_null()) {
  1486. return;
  1487. }
  1488. Size2 size = p_size;
  1489. if (!node->is_allow_v_resize()) {
  1490. size.y = 0;
  1491. }
  1492. node->set_size(size);
  1493. if (get_current_shader_type() == type) {
  1494. Ref<VisualShaderNodeExpression> expression_node = Object::cast_to<VisualShaderNodeExpression>(node.ptr());
  1495. Control *text_box = nullptr;
  1496. if (!expression_node.is_null()) {
  1497. text_box = expression_node->is_ctrl_pressed(0);
  1498. if (text_box) {
  1499. text_box->set_custom_minimum_size(Size2(0, 0));
  1500. }
  1501. }
  1502. GraphNode *gn = nullptr;
  1503. Node *node2 = graph->get_node(itos(p_node));
  1504. gn = Object::cast_to<GraphNode>(node2);
  1505. if (!gn) {
  1506. return;
  1507. }
  1508. gn->set_custom_minimum_size(size);
  1509. gn->set_size(Size2(1, 1));
  1510. if (!expression_node.is_null() && text_box) {
  1511. Size2 box_size = size;
  1512. if (gn != nullptr) {
  1513. if (box_size.x < 150 * EDSCALE || box_size.y < 0) {
  1514. box_size.x = gn->get_size().x;
  1515. }
  1516. }
  1517. box_size.x -= text_box->get_offset(SIDE_LEFT);
  1518. box_size.x -= 28 * EDSCALE;
  1519. box_size.y -= text_box->get_offset(SIDE_TOP);
  1520. box_size.y -= 28 * EDSCALE;
  1521. text_box->set_custom_minimum_size(Size2(box_size.x, box_size.y));
  1522. text_box->set_size(Size2(1, 1));
  1523. }
  1524. }
  1525. }
  1526. void VisualShaderEditor::_node_resized(const Vector2 &p_new_size, int p_type, int p_node) {
  1527. Ref<VisualShaderNodeResizableBase> node = visual_shader->get_node(VisualShader::Type(p_type), p_node);
  1528. if (node.is_null()) {
  1529. return;
  1530. }
  1531. undo_redo->create_action(TTR("Resize VisualShader Node"), UndoRedo::MERGE_ENDS);
  1532. undo_redo->add_do_method(this, "_set_node_size", p_type, p_node, p_new_size);
  1533. undo_redo->add_undo_method(this, "_set_node_size", p_type, p_node, node->get_size());
  1534. undo_redo->commit_action();
  1535. }
  1536. void VisualShaderEditor::_preview_select_port(int p_node, int p_port) {
  1537. VisualShader::Type type = get_current_shader_type();
  1538. Ref<VisualShaderNode> node = visual_shader->get_node(type, p_node);
  1539. if (node.is_null()) {
  1540. return;
  1541. }
  1542. int prev_port = node->get_output_port_for_preview();
  1543. if (node->get_output_port_for_preview() == p_port) {
  1544. p_port = -1; //toggle it
  1545. }
  1546. undo_redo->create_action(p_port == -1 ? TTR("Hide Port Preview") : TTR("Show Port Preview"));
  1547. undo_redo->add_do_method(node.ptr(), "set_output_port_for_preview", p_port);
  1548. undo_redo->add_undo_method(node.ptr(), "set_output_port_for_preview", prev_port);
  1549. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", (int)type, p_node);
  1550. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", (int)type, p_node);
  1551. undo_redo->commit_action();
  1552. }
  1553. void VisualShaderEditor::_comment_title_popup_show(const Point2 &p_position, int p_node_id) {
  1554. VisualShader::Type type = get_current_shader_type();
  1555. Ref<VisualShaderNodeComment> node = visual_shader->get_node(type, p_node_id);
  1556. if (node.is_null()) {
  1557. return;
  1558. }
  1559. comment_title_change_edit->set_text(node->get_title());
  1560. comment_title_change_popup->set_meta("id", p_node_id);
  1561. comment_title_change_popup->popup();
  1562. comment_title_change_popup->set_position(p_position);
  1563. }
  1564. void VisualShaderEditor::_comment_title_text_changed(const String &p_new_text) {
  1565. comment_title_change_edit->set_size(Size2(-1, -1));
  1566. comment_title_change_popup->set_size(Size2(-1, -1));
  1567. }
  1568. void VisualShaderEditor::_comment_title_text_submitted(const String &p_new_text) {
  1569. comment_title_change_popup->hide();
  1570. }
  1571. void VisualShaderEditor::_comment_title_popup_focus_out() {
  1572. comment_title_change_popup->hide();
  1573. }
  1574. void VisualShaderEditor::_comment_title_popup_hide() {
  1575. ERR_FAIL_COND(!comment_title_change_popup->has_meta("id"));
  1576. int node_id = (int)comment_title_change_popup->get_meta("id");
  1577. VisualShader::Type type = get_current_shader_type();
  1578. Ref<VisualShaderNodeComment> node = visual_shader->get_node(type, node_id);
  1579. ERR_FAIL_COND(node.is_null());
  1580. if (node->get_title() == comment_title_change_edit->get_text()) {
  1581. return; // nothing changed - ignored
  1582. }
  1583. undo_redo->create_action(TTR("Set Comment Node Title"));
  1584. undo_redo->add_do_method(node.ptr(), "set_title", comment_title_change_edit->get_text());
  1585. undo_redo->add_undo_method(node.ptr(), "set_title", node->get_title());
  1586. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", (int)type, node_id);
  1587. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", (int)type, node_id);
  1588. undo_redo->commit_action();
  1589. }
  1590. void VisualShaderEditor::_comment_desc_popup_show(const Point2 &p_position, int p_node_id) {
  1591. VisualShader::Type type = get_current_shader_type();
  1592. Ref<VisualShaderNodeComment> node = visual_shader->get_node(type, p_node_id);
  1593. if (node.is_null()) {
  1594. return;
  1595. }
  1596. comment_desc_change_edit->set_text(node->get_description());
  1597. comment_desc_change_popup->set_meta("id", p_node_id);
  1598. comment_desc_change_popup->popup();
  1599. comment_desc_change_popup->set_position(p_position);
  1600. }
  1601. void VisualShaderEditor::_comment_desc_text_changed() {
  1602. comment_desc_change_edit->set_size(Size2(-1, -1));
  1603. comment_desc_change_popup->set_size(Size2(-1, -1));
  1604. }
  1605. void VisualShaderEditor::_comment_desc_confirm() {
  1606. comment_desc_change_popup->hide();
  1607. }
  1608. void VisualShaderEditor::_comment_desc_popup_hide() {
  1609. ERR_FAIL_COND(!comment_desc_change_popup->has_meta("id"));
  1610. int node_id = (int)comment_desc_change_popup->get_meta("id");
  1611. VisualShader::Type type = get_current_shader_type();
  1612. Ref<VisualShaderNodeComment> node = visual_shader->get_node(type, node_id);
  1613. ERR_FAIL_COND(node.is_null());
  1614. if (node->get_description() == comment_desc_change_edit->get_text()) {
  1615. return; // nothing changed - ignored
  1616. }
  1617. undo_redo->create_action(TTR("Set Comment Node Description"));
  1618. undo_redo->add_do_method(node.ptr(), "set_description", comment_desc_change_edit->get_text());
  1619. undo_redo->add_undo_method(node.ptr(), "set_description", node->get_title());
  1620. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", (int)type, node_id);
  1621. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", (int)type, node_id);
  1622. undo_redo->commit_action();
  1623. }
  1624. void VisualShaderEditor::_uniform_line_edit_changed(const String &p_text, int p_node_id) {
  1625. VisualShader::Type type = get_current_shader_type();
  1626. Ref<VisualShaderNodeUniform> node = visual_shader->get_node(type, p_node_id);
  1627. ERR_FAIL_COND(!node.is_valid());
  1628. String validated_name = visual_shader->validate_uniform_name(p_text, node);
  1629. if (validated_name == node->get_uniform_name()) {
  1630. return;
  1631. }
  1632. undo_redo->create_action(TTR("Set Uniform Name"));
  1633. undo_redo->add_do_method(node.ptr(), "set_uniform_name", validated_name);
  1634. undo_redo->add_undo_method(node.ptr(), "set_uniform_name", node->get_uniform_name());
  1635. undo_redo->add_do_method(graph_plugin.ptr(), "set_uniform_name", type, p_node_id, validated_name);
  1636. undo_redo->add_undo_method(graph_plugin.ptr(), "set_uniform_name", type, p_node_id, node->get_uniform_name());
  1637. undo_redo->add_do_method(this, "_update_uniforms", true);
  1638. undo_redo->add_undo_method(this, "_update_uniforms", true);
  1639. Set<String> changed_names;
  1640. changed_names.insert(node->get_uniform_name());
  1641. _update_uniform_refs(changed_names);
  1642. undo_redo->commit_action();
  1643. }
  1644. void VisualShaderEditor::_uniform_line_edit_focus_out(Object *line_edit, int p_node_id) {
  1645. _uniform_line_edit_changed(Object::cast_to<LineEdit>(line_edit)->get_text(), p_node_id);
  1646. }
  1647. void VisualShaderEditor::_port_name_focus_out(Object *line_edit, int p_node_id, int p_port_id, bool p_output) {
  1648. if (!p_output) {
  1649. _change_input_port_name(Object::cast_to<LineEdit>(line_edit)->get_text(), line_edit, p_node_id, p_port_id);
  1650. } else {
  1651. _change_output_port_name(Object::cast_to<LineEdit>(line_edit)->get_text(), line_edit, p_node_id, p_port_id);
  1652. }
  1653. }
  1654. void VisualShaderEditor::_port_edited() {
  1655. VisualShader::Type type = get_current_shader_type();
  1656. Variant value = property_editor->get_variant();
  1657. Ref<VisualShaderNode> vsn = visual_shader->get_node(type, editing_node);
  1658. ERR_FAIL_COND(!vsn.is_valid());
  1659. undo_redo->create_action(TTR("Set Input Default Port"));
  1660. Ref<VisualShaderNodeCustom> custom = Object::cast_to<VisualShaderNodeCustom>(vsn.ptr());
  1661. if (custom.is_valid()) {
  1662. undo_redo->add_do_method(custom.ptr(), "_set_input_port_default_value", editing_port, value);
  1663. undo_redo->add_undo_method(custom.ptr(), "_set_input_port_default_value", editing_port, vsn->get_input_port_default_value(editing_port));
  1664. } else {
  1665. undo_redo->add_do_method(vsn.ptr(), "set_input_port_default_value", editing_port, value);
  1666. undo_redo->add_undo_method(vsn.ptr(), "set_input_port_default_value", editing_port, vsn->get_input_port_default_value(editing_port));
  1667. }
  1668. undo_redo->add_do_method(graph_plugin.ptr(), "set_input_port_default_value", type, editing_node, editing_port, value);
  1669. undo_redo->add_undo_method(graph_plugin.ptr(), "set_input_port_default_value", type, editing_node, editing_port, vsn->get_input_port_default_value(editing_port));
  1670. undo_redo->commit_action();
  1671. property_editor->hide();
  1672. }
  1673. void VisualShaderEditor::_edit_port_default_input(Object *p_button, int p_node, int p_port) {
  1674. VisualShader::Type type = get_current_shader_type();
  1675. Ref<VisualShaderNode> vsn = visual_shader->get_node(type, p_node);
  1676. Button *button = Object::cast_to<Button>(p_button);
  1677. ERR_FAIL_COND(!button);
  1678. Variant value = vsn->get_input_port_default_value(p_port);
  1679. property_editor->set_position(button->get_screen_position() + Vector2(0, button->get_size().height));
  1680. property_editor->edit(nullptr, "", value.get_type(), value, 0, "");
  1681. property_editor->popup();
  1682. editing_node = p_node;
  1683. editing_port = p_port;
  1684. }
  1685. void VisualShaderEditor::_setup_node(VisualShaderNode *p_node, int p_op_idx) {
  1686. // FLOAT_OP
  1687. {
  1688. VisualShaderNodeFloatOp *floatOp = Object::cast_to<VisualShaderNodeFloatOp>(p_node);
  1689. if (floatOp) {
  1690. floatOp->set_operator((VisualShaderNodeFloatOp::Operator)p_op_idx);
  1691. return;
  1692. }
  1693. }
  1694. // FLOAT_FUNC
  1695. {
  1696. VisualShaderNodeFloatFunc *floatFunc = Object::cast_to<VisualShaderNodeFloatFunc>(p_node);
  1697. if (floatFunc) {
  1698. floatFunc->set_function((VisualShaderNodeFloatFunc::Function)p_op_idx);
  1699. return;
  1700. }
  1701. }
  1702. // VECTOR_OP
  1703. {
  1704. VisualShaderNodeVectorOp *vecOp = Object::cast_to<VisualShaderNodeVectorOp>(p_node);
  1705. if (vecOp) {
  1706. vecOp->set_operator((VisualShaderNodeVectorOp::Operator)p_op_idx);
  1707. return;
  1708. }
  1709. }
  1710. // VECTOR_FUNC
  1711. {
  1712. VisualShaderNodeVectorFunc *vecFunc = Object::cast_to<VisualShaderNodeVectorFunc>(p_node);
  1713. if (vecFunc) {
  1714. vecFunc->set_function((VisualShaderNodeVectorFunc::Function)p_op_idx);
  1715. return;
  1716. }
  1717. }
  1718. // COLOR_OP
  1719. {
  1720. VisualShaderNodeColorOp *colorOp = Object::cast_to<VisualShaderNodeColorOp>(p_node);
  1721. if (colorOp) {
  1722. colorOp->set_operator((VisualShaderNodeColorOp::Operator)p_op_idx);
  1723. return;
  1724. }
  1725. }
  1726. // COLOR_FUNC
  1727. {
  1728. VisualShaderNodeColorFunc *colorFunc = Object::cast_to<VisualShaderNodeColorFunc>(p_node);
  1729. if (colorFunc) {
  1730. colorFunc->set_function((VisualShaderNodeColorFunc::Function)p_op_idx);
  1731. return;
  1732. }
  1733. }
  1734. // INT_OP
  1735. {
  1736. VisualShaderNodeIntOp *intOp = Object::cast_to<VisualShaderNodeIntOp>(p_node);
  1737. if (intOp) {
  1738. intOp->set_operator((VisualShaderNodeIntOp::Operator)p_op_idx);
  1739. return;
  1740. }
  1741. }
  1742. // INT_FUNC
  1743. {
  1744. VisualShaderNodeIntFunc *intFunc = Object::cast_to<VisualShaderNodeIntFunc>(p_node);
  1745. if (intFunc) {
  1746. intFunc->set_function((VisualShaderNodeIntFunc::Function)p_op_idx);
  1747. return;
  1748. }
  1749. }
  1750. // TRANSFORM_FUNC
  1751. {
  1752. VisualShaderNodeTransformFunc *matFunc = Object::cast_to<VisualShaderNodeTransformFunc>(p_node);
  1753. if (matFunc) {
  1754. matFunc->set_function((VisualShaderNodeTransformFunc::Function)p_op_idx);
  1755. return;
  1756. }
  1757. }
  1758. //UV_FUNC
  1759. {
  1760. VisualShaderNodeUVFunc *uvFunc = Object::cast_to<VisualShaderNodeUVFunc>(p_node);
  1761. if (uvFunc) {
  1762. uvFunc->set_function((VisualShaderNodeUVFunc::Function)p_op_idx);
  1763. return;
  1764. }
  1765. }
  1766. // IS
  1767. {
  1768. VisualShaderNodeIs *is = Object::cast_to<VisualShaderNodeIs>(p_node);
  1769. if (is) {
  1770. is->set_function((VisualShaderNodeIs::Function)p_op_idx);
  1771. return;
  1772. }
  1773. }
  1774. // COMPARE
  1775. {
  1776. VisualShaderNodeCompare *cmp = Object::cast_to<VisualShaderNodeCompare>(p_node);
  1777. if (cmp) {
  1778. cmp->set_function((VisualShaderNodeCompare::Function)p_op_idx);
  1779. return;
  1780. }
  1781. }
  1782. // DERIVATIVE
  1783. {
  1784. VisualShaderNodeScalarDerivativeFunc *sderFunc = Object::cast_to<VisualShaderNodeScalarDerivativeFunc>(p_node);
  1785. if (sderFunc) {
  1786. sderFunc->set_function((VisualShaderNodeScalarDerivativeFunc::Function)p_op_idx);
  1787. return;
  1788. }
  1789. VisualShaderNodeVectorDerivativeFunc *vderFunc = Object::cast_to<VisualShaderNodeVectorDerivativeFunc>(p_node);
  1790. if (vderFunc) {
  1791. vderFunc->set_function((VisualShaderNodeVectorDerivativeFunc::Function)p_op_idx);
  1792. return;
  1793. }
  1794. }
  1795. // MIX
  1796. {
  1797. VisualShaderNodeMix *mix = Object::cast_to<VisualShaderNodeMix>(p_node);
  1798. if (mix) {
  1799. mix->set_op_type((VisualShaderNodeMix::OpType)p_op_idx);
  1800. return;
  1801. }
  1802. }
  1803. // CLAMP
  1804. {
  1805. VisualShaderNodeClamp *clampFunc = Object::cast_to<VisualShaderNodeClamp>(p_node);
  1806. if (clampFunc) {
  1807. clampFunc->set_op_type((VisualShaderNodeClamp::OpType)p_op_idx);
  1808. return;
  1809. }
  1810. }
  1811. // SWITCH
  1812. {
  1813. VisualShaderNodeSwitch *switchFunc = Object::cast_to<VisualShaderNodeSwitch>(p_node);
  1814. if (switchFunc) {
  1815. switchFunc->set_op_type((VisualShaderNodeSwitch::OpType)p_op_idx);
  1816. return;
  1817. }
  1818. }
  1819. // SMOOTHSTEP
  1820. {
  1821. VisualShaderNodeSmoothStep *smoothStepFunc = Object::cast_to<VisualShaderNodeSmoothStep>(p_node);
  1822. if (smoothStepFunc) {
  1823. smoothStepFunc->set_op_type((VisualShaderNodeSmoothStep::OpType)p_op_idx);
  1824. return;
  1825. }
  1826. }
  1827. // STEP
  1828. {
  1829. VisualShaderNodeStep *stepFunc = Object::cast_to<VisualShaderNodeStep>(p_node);
  1830. if (stepFunc) {
  1831. stepFunc->set_op_type((VisualShaderNodeStep::OpType)p_op_idx);
  1832. return;
  1833. }
  1834. }
  1835. // MULTIPLY_ADD
  1836. {
  1837. VisualShaderNodeMultiplyAdd *fmaFunc = Object::cast_to<VisualShaderNodeMultiplyAdd>(p_node);
  1838. if (fmaFunc) {
  1839. fmaFunc->set_op_type((VisualShaderNodeMultiplyAdd::OpType)p_op_idx);
  1840. }
  1841. }
  1842. }
  1843. void VisualShaderEditor::_add_node(int p_idx, int p_op_idx, String p_resource_path, int p_node_idx) {
  1844. ERR_FAIL_INDEX(p_idx, add_options.size());
  1845. Ref<VisualShaderNode> vsnode;
  1846. bool is_custom = add_options[p_idx].is_custom;
  1847. if (!is_custom && add_options[p_idx].type != String()) {
  1848. VisualShaderNode *vsn = Object::cast_to<VisualShaderNode>(ClassDB::instantiate(add_options[p_idx].type));
  1849. ERR_FAIL_COND(!vsn);
  1850. VisualShaderNodeFloatConstant *constant = Object::cast_to<VisualShaderNodeFloatConstant>(vsn);
  1851. if (constant) {
  1852. if ((int)add_options[p_idx].value != -1) {
  1853. constant->set_constant(add_options[p_idx].value);
  1854. }
  1855. } else {
  1856. if (p_op_idx != -1) {
  1857. VisualShaderNodeInput *input = Object::cast_to<VisualShaderNodeInput>(vsn);
  1858. if (input) {
  1859. input->set_input_name(add_options[p_idx].sub_func_str);
  1860. } else {
  1861. _setup_node(vsn, p_op_idx);
  1862. }
  1863. }
  1864. }
  1865. vsnode = Ref<VisualShaderNode>(vsn);
  1866. } else {
  1867. ERR_FAIL_COND(add_options[p_idx].script.is_null());
  1868. String base_type = add_options[p_idx].script->get_instance_base_type();
  1869. VisualShaderNode *vsn = Object::cast_to<VisualShaderNode>(ClassDB::instantiate(base_type));
  1870. ERR_FAIL_COND(!vsn);
  1871. vsnode = Ref<VisualShaderNode>(vsn);
  1872. vsnode->set_script(add_options[p_idx].script);
  1873. }
  1874. Point2 position = graph->get_scroll_ofs();
  1875. if (saved_node_pos_dirty) {
  1876. position += saved_node_pos;
  1877. } else {
  1878. position += graph->get_size() * 0.5;
  1879. position /= EDSCALE;
  1880. }
  1881. saved_node_pos_dirty = false;
  1882. VisualShader::Type type = get_current_shader_type();
  1883. int id_to_use = visual_shader->get_valid_node_id(type);
  1884. if (p_resource_path.is_empty()) {
  1885. undo_redo->create_action(TTR("Add Node to Visual Shader"));
  1886. } else {
  1887. id_to_use += p_node_idx;
  1888. }
  1889. undo_redo->add_do_method(visual_shader.ptr(), "add_node", type, vsnode, position, id_to_use);
  1890. undo_redo->add_undo_method(visual_shader.ptr(), "remove_node", type, id_to_use);
  1891. undo_redo->add_do_method(graph_plugin.ptr(), "add_node", type, id_to_use);
  1892. undo_redo->add_undo_method(graph_plugin.ptr(), "remove_node", type, id_to_use);
  1893. VisualShaderNodeExpression *expr = Object::cast_to<VisualShaderNodeExpression>(vsnode.ptr());
  1894. if (expr) {
  1895. undo_redo->add_do_method(expr, "set_size", Size2(250 * EDSCALE, 150 * EDSCALE));
  1896. }
  1897. bool created_expression_port = false;
  1898. if (to_node != -1 && to_slot != -1) {
  1899. VisualShaderNode::PortType input_port_type = visual_shader->get_node(type, to_node)->get_input_port_type(to_slot);
  1900. if (expr && expr->is_editable() && input_port_type != VisualShaderNode::PORT_TYPE_SAMPLER) {
  1901. undo_redo->add_do_method(expr, "add_output_port", 0, input_port_type, "output0");
  1902. undo_redo->add_undo_method(expr, "remove_output_port", 0);
  1903. String initial_expression_code;
  1904. switch (input_port_type) {
  1905. case VisualShaderNode::PORT_TYPE_SCALAR:
  1906. initial_expression_code = "output0 = 1.0;";
  1907. break;
  1908. case VisualShaderNode::PORT_TYPE_SCALAR_INT:
  1909. initial_expression_code = "output0 = 1;";
  1910. break;
  1911. case VisualShaderNode::PORT_TYPE_VECTOR:
  1912. initial_expression_code = "output0 = vec3(1.0, 1.0, 1.0);";
  1913. break;
  1914. case VisualShaderNode::PORT_TYPE_BOOLEAN:
  1915. initial_expression_code = "output0 = true;";
  1916. break;
  1917. case VisualShaderNode::PORT_TYPE_TRANSFORM:
  1918. initial_expression_code = "output0 = mat4(1.0);";
  1919. break;
  1920. default:
  1921. break;
  1922. }
  1923. undo_redo->add_do_method(expr, "set_expression", initial_expression_code);
  1924. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, id_to_use);
  1925. created_expression_port = true;
  1926. }
  1927. if (vsnode->get_output_port_count() > 0 || created_expression_port) {
  1928. int _from_node = id_to_use;
  1929. int _from_slot = 0;
  1930. if (created_expression_port) {
  1931. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, _from_node, _from_slot, to_node, to_slot);
  1932. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, _from_node, _from_slot, to_node, to_slot);
  1933. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, _from_node, _from_slot, to_node, to_slot);
  1934. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, _from_node, _from_slot, to_node, to_slot);
  1935. } else {
  1936. // Need to setting up Input node properly before committing since `is_port_types_compatible` (calling below) is using `mode` and `shader_type`.
  1937. VisualShaderNodeInput *input = Object::cast_to<VisualShaderNodeInput>(vsnode.ptr());
  1938. if (input) {
  1939. input->set_shader_mode(visual_shader->get_mode());
  1940. input->set_shader_type(visual_shader->get_shader_type());
  1941. }
  1942. // Attempting to connect to the first correct port.
  1943. for (int i = 0; i < vsnode->get_output_port_count(); i++) {
  1944. if (visual_shader->is_port_types_compatible(vsnode->get_output_port_type(i), input_port_type)) {
  1945. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, _from_node, i, to_node, to_slot);
  1946. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, _from_node, i, to_node, to_slot);
  1947. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, _from_node, i, to_node, to_slot);
  1948. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, _from_node, i, to_node, to_slot);
  1949. break;
  1950. }
  1951. }
  1952. }
  1953. }
  1954. } else if (from_node != -1 && from_slot != -1) {
  1955. VisualShaderNode::PortType output_port_type = visual_shader->get_node(type, from_node)->get_output_port_type(from_slot);
  1956. if (expr && expr->is_editable()) {
  1957. undo_redo->add_do_method(expr, "add_input_port", 0, output_port_type, "input0");
  1958. undo_redo->add_undo_method(expr, "remove_input_port", 0);
  1959. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type, id_to_use);
  1960. created_expression_port = true;
  1961. }
  1962. if (vsnode->get_input_port_count() > 0 || created_expression_port) {
  1963. int _to_node = id_to_use;
  1964. int _to_slot = 0;
  1965. if (created_expression_port) {
  1966. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_slot, _to_node, _to_slot);
  1967. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, from_node, from_slot, _to_node, _to_slot);
  1968. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_slot, _to_node, _to_slot);
  1969. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_slot, _to_node, _to_slot);
  1970. } else {
  1971. // Attempting to connect to the first correct port.
  1972. for (int i = 0; i < vsnode->get_input_port_count(); i++) {
  1973. if (visual_shader->is_port_types_compatible(output_port_type, vsnode->get_input_port_type(i))) {
  1974. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from_node, from_slot, _to_node, i);
  1975. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, from_node, from_slot, _to_node, i);
  1976. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, from_node, from_slot, _to_node, i);
  1977. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, from_node, from_slot, _to_node, i);
  1978. break;
  1979. }
  1980. }
  1981. }
  1982. }
  1983. }
  1984. VisualShaderNodeUniform *uniform = Object::cast_to<VisualShaderNodeUniform>(vsnode.ptr());
  1985. if (uniform) {
  1986. undo_redo->add_do_method(this, "_update_uniforms", true);
  1987. undo_redo->add_undo_method(this, "_update_uniforms", true);
  1988. }
  1989. VisualShaderNodeCurveTexture *curve = Object::cast_to<VisualShaderNodeCurveTexture>(vsnode.ptr());
  1990. if (curve) {
  1991. graph_plugin->call_deferred("update_curve", id_to_use);
  1992. }
  1993. if (p_resource_path.is_empty()) {
  1994. undo_redo->commit_action();
  1995. } else {
  1996. //post-initialization
  1997. VisualShaderNodeTexture *texture2d = Object::cast_to<VisualShaderNodeTexture>(vsnode.ptr());
  1998. VisualShaderNodeTexture3D *texture3d = Object::cast_to<VisualShaderNodeTexture3D>(vsnode.ptr());
  1999. if (texture2d || texture3d || curve) {
  2000. undo_redo->add_do_method(vsnode.ptr(), "set_texture", ResourceLoader::load(p_resource_path));
  2001. return;
  2002. }
  2003. VisualShaderNodeCubemap *cubemap = Object::cast_to<VisualShaderNodeCubemap>(vsnode.ptr());
  2004. if (cubemap) {
  2005. undo_redo->add_do_method(vsnode.ptr(), "set_cube_map", ResourceLoader::load(p_resource_path));
  2006. return;
  2007. }
  2008. VisualShaderNodeTexture2DArray *texture2d_array = Object::cast_to<VisualShaderNodeTexture2DArray>(vsnode.ptr());
  2009. if (texture2d_array) {
  2010. undo_redo->add_do_method(vsnode.ptr(), "set_texture_array", ResourceLoader::load(p_resource_path));
  2011. }
  2012. }
  2013. }
  2014. void VisualShaderEditor::_node_dragged(const Vector2 &p_from, const Vector2 &p_to, int p_node) {
  2015. VisualShader::Type type = get_current_shader_type();
  2016. drag_buffer.push_back({ type, p_node, p_from, p_to });
  2017. if (!drag_dirty) {
  2018. call_deferred("_nodes_dragged");
  2019. }
  2020. drag_dirty = true;
  2021. }
  2022. void VisualShaderEditor::_nodes_dragged() {
  2023. drag_dirty = false;
  2024. undo_redo->create_action(TTR("Node(s) Moved"));
  2025. for (List<DragOp>::Element *E = drag_buffer.front(); E; E = E->next()) {
  2026. undo_redo->add_do_method(visual_shader.ptr(), "set_node_position", E->get().type, E->get().node, E->get().to);
  2027. undo_redo->add_undo_method(visual_shader.ptr(), "set_node_position", E->get().type, E->get().node, E->get().from);
  2028. undo_redo->add_do_method(graph_plugin.ptr(), "set_node_position", E->get().type, E->get().node, E->get().to);
  2029. undo_redo->add_undo_method(graph_plugin.ptr(), "set_node_position", E->get().type, E->get().node, E->get().from);
  2030. }
  2031. drag_buffer.clear();
  2032. undo_redo->commit_action();
  2033. }
  2034. void VisualShaderEditor::_connection_request(const String &p_from, int p_from_index, const String &p_to, int p_to_index) {
  2035. VisualShader::Type type = get_current_shader_type();
  2036. int from = p_from.to_int();
  2037. int to = p_to.to_int();
  2038. if (!visual_shader->can_connect_nodes(type, from, p_from_index, to, p_to_index)) {
  2039. return;
  2040. }
  2041. undo_redo->create_action(TTR("Nodes Connected"));
  2042. List<VisualShader::Connection> conns;
  2043. visual_shader->get_node_connections(type, &conns);
  2044. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  2045. if (E->get().to_node == to && E->get().to_port == p_to_index) {
  2046. 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);
  2047. 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);
  2048. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2049. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2050. }
  2051. }
  2052. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, from, p_from_index, to, p_to_index);
  2053. undo_redo->add_undo_method(visual_shader.ptr(), "disconnect_nodes", type, from, p_from_index, to, p_to_index);
  2054. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, from, p_from_index, to, p_to_index);
  2055. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, from, p_from_index, to, p_to_index);
  2056. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", (int)type, to);
  2057. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", (int)type, to);
  2058. undo_redo->commit_action();
  2059. }
  2060. void VisualShaderEditor::_disconnection_request(const String &p_from, int p_from_index, const String &p_to, int p_to_index) {
  2061. graph->disconnect_node(p_from, p_from_index, p_to, p_to_index);
  2062. VisualShader::Type type = get_current_shader_type();
  2063. int from = p_from.to_int();
  2064. int to = p_to.to_int();
  2065. undo_redo->create_action(TTR("Nodes Disconnected"));
  2066. undo_redo->add_do_method(visual_shader.ptr(), "disconnect_nodes", type, from, p_from_index, to, p_to_index);
  2067. undo_redo->add_undo_method(visual_shader.ptr(), "connect_nodes", type, from, p_from_index, to, p_to_index);
  2068. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, from, p_from_index, to, p_to_index);
  2069. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, from, p_from_index, to, p_to_index);
  2070. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", (int)type, to);
  2071. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", (int)type, to);
  2072. undo_redo->commit_action();
  2073. }
  2074. void VisualShaderEditor::_connection_to_empty(const String &p_from, int p_from_slot, const Vector2 &p_release_position) {
  2075. from_node = p_from.to_int();
  2076. from_slot = p_from_slot;
  2077. _show_members_dialog(true);
  2078. }
  2079. void VisualShaderEditor::_connection_from_empty(const String &p_to, int p_to_slot, const Vector2 &p_release_position) {
  2080. to_node = p_to.to_int();
  2081. to_slot = p_to_slot;
  2082. _show_members_dialog(true);
  2083. }
  2084. void VisualShaderEditor::_delete_nodes(int p_type, const List<int> &p_nodes) {
  2085. VisualShader::Type type = VisualShader::Type(p_type);
  2086. List<VisualShader::Connection> conns;
  2087. visual_shader->get_node_connections(type, &conns);
  2088. for (const List<int>::Element *F = p_nodes.front(); F; F = F->next()) {
  2089. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  2090. if (E->get().from_node == F->get() || E->get().to_node == F->get()) {
  2091. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2092. }
  2093. }
  2094. }
  2095. Set<String> uniform_names;
  2096. for (const List<int>::Element *F = p_nodes.front(); F; F = F->next()) {
  2097. Ref<VisualShaderNode> node = visual_shader->get_node(type, F->get());
  2098. undo_redo->add_do_method(visual_shader.ptr(), "remove_node", type, F->get());
  2099. undo_redo->add_undo_method(visual_shader.ptr(), "add_node", type, node, visual_shader->get_node_position(type, F->get()), F->get());
  2100. undo_redo->add_undo_method(graph_plugin.ptr(), "add_node", type, F->get());
  2101. undo_redo->add_do_method(this, "_clear_buffer");
  2102. undo_redo->add_undo_method(this, "_clear_buffer");
  2103. // restore size, inputs and outputs if node is group
  2104. VisualShaderNodeGroupBase *group = Object::cast_to<VisualShaderNodeGroupBase>(node.ptr());
  2105. if (group) {
  2106. undo_redo->add_undo_method(group, "set_size", group->get_size());
  2107. undo_redo->add_undo_method(group, "set_inputs", group->get_inputs());
  2108. undo_redo->add_undo_method(group, "set_outputs", group->get_outputs());
  2109. }
  2110. // restore expression text if node is expression
  2111. VisualShaderNodeExpression *expression = Object::cast_to<VisualShaderNodeExpression>(node.ptr());
  2112. if (expression) {
  2113. undo_redo->add_undo_method(expression, "set_expression", expression->get_expression());
  2114. }
  2115. VisualShaderNodeUniform *uniform = Object::cast_to<VisualShaderNodeUniform>(node.ptr());
  2116. if (uniform) {
  2117. uniform_names.insert(uniform->get_uniform_name());
  2118. }
  2119. }
  2120. List<VisualShader::Connection> used_conns;
  2121. for (const List<int>::Element *F = p_nodes.front(); F; F = F->next()) {
  2122. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  2123. if (E->get().from_node == F->get() || E->get().to_node == F->get()) {
  2124. bool cancel = false;
  2125. for (List<VisualShader::Connection>::Element *R = used_conns.front(); R; R = R->next()) {
  2126. 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) {
  2127. cancel = true; // to avoid ERR_ALREADY_EXISTS warning
  2128. break;
  2129. }
  2130. }
  2131. if (!cancel) {
  2132. 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);
  2133. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2134. used_conns.push_back(E->get());
  2135. }
  2136. }
  2137. }
  2138. }
  2139. // delete nodes from the graph
  2140. for (const List<int>::Element *F = p_nodes.front(); F; F = F->next()) {
  2141. undo_redo->add_do_method(graph_plugin.ptr(), "remove_node", type, F->get());
  2142. }
  2143. // update uniform refs if any uniform has been deleted
  2144. if (uniform_names.size() > 0) {
  2145. undo_redo->add_do_method(this, "_update_uniforms", true);
  2146. undo_redo->add_undo_method(this, "_update_uniforms", true);
  2147. _update_uniform_refs(uniform_names);
  2148. }
  2149. }
  2150. void VisualShaderEditor::_replace_node(VisualShader::Type p_type_id, int p_node_id, const StringName &p_from, const StringName &p_to) {
  2151. undo_redo->add_do_method(visual_shader.ptr(), "replace_node", p_type_id, p_node_id, p_to);
  2152. undo_redo->add_undo_method(visual_shader.ptr(), "replace_node", p_type_id, p_node_id, p_from);
  2153. }
  2154. void VisualShaderEditor::_update_constant(VisualShader::Type p_type_id, int p_node_id, Variant p_var, int p_preview_port) {
  2155. Ref<VisualShaderNode> node = visual_shader->get_node(p_type_id, p_node_id);
  2156. ERR_FAIL_COND(!node.is_valid());
  2157. ERR_FAIL_COND(!node->has_method("set_constant"));
  2158. node->call("set_constant", p_var);
  2159. if (p_preview_port != -1) {
  2160. node->set_output_port_for_preview(p_preview_port);
  2161. }
  2162. }
  2163. void VisualShaderEditor::_update_uniform(VisualShader::Type p_type_id, int p_node_id, Variant p_var, int p_preview_port) {
  2164. Ref<VisualShaderNodeUniform> uniform = visual_shader->get_node(p_type_id, p_node_id);
  2165. ERR_FAIL_COND(!uniform.is_valid());
  2166. String valid_name = visual_shader->validate_uniform_name(uniform->get_uniform_name(), uniform);
  2167. uniform->set_uniform_name(valid_name);
  2168. graph_plugin->set_uniform_name(p_type_id, p_node_id, valid_name);
  2169. if (uniform->has_method("set_default_value_enabled")) {
  2170. uniform->call("set_default_value_enabled", true);
  2171. uniform->call("set_default_value", p_var);
  2172. }
  2173. if (p_preview_port != -1) {
  2174. uniform->set_output_port_for_preview(p_preview_port);
  2175. }
  2176. }
  2177. void VisualShaderEditor::_convert_constants_to_uniforms(bool p_vice_versa) {
  2178. VisualShader::Type type_id = get_current_shader_type();
  2179. if (!p_vice_versa) {
  2180. undo_redo->create_action(TTR("Convert Constant Node(s) To Uniform(s)"));
  2181. } else {
  2182. undo_redo->create_action(TTR("Convert Uniform Node(s) To Constant(s)"));
  2183. }
  2184. const Set<int> &current_set = p_vice_versa ? selected_uniforms : selected_constants;
  2185. Set<String> deleted_names;
  2186. for (Set<int>::Element *E = current_set.front(); E; E = E->next()) {
  2187. int node_id = E->get();
  2188. Ref<VisualShaderNode> node = visual_shader->get_node(type_id, node_id);
  2189. bool caught = false;
  2190. Variant var;
  2191. // float
  2192. if (!p_vice_versa) {
  2193. Ref<VisualShaderNodeFloatConstant> float_const = Object::cast_to<VisualShaderNodeFloatConstant>(node.ptr());
  2194. if (float_const.is_valid()) {
  2195. _replace_node(type_id, node_id, "VisualShaderNodeFloatConstant", "VisualShaderNodeFloatUniform");
  2196. var = float_const->get_constant();
  2197. caught = true;
  2198. }
  2199. } else {
  2200. Ref<VisualShaderNodeFloatUniform> float_uniform = Object::cast_to<VisualShaderNodeFloatUniform>(node.ptr());
  2201. if (float_uniform.is_valid()) {
  2202. _replace_node(type_id, node_id, "VisualShaderNodeFloatUniform", "VisualShaderNodeFloatConstant");
  2203. var = float_uniform->get_default_value();
  2204. caught = true;
  2205. }
  2206. }
  2207. // int
  2208. if (!caught) {
  2209. if (!p_vice_versa) {
  2210. Ref<VisualShaderNodeIntConstant> int_const = Object::cast_to<VisualShaderNodeIntConstant>(node.ptr());
  2211. if (int_const.is_valid()) {
  2212. _replace_node(type_id, node_id, "VisualShaderNodeIntConstant", "VisualShaderNodeIntUniform");
  2213. var = int_const->get_constant();
  2214. caught = true;
  2215. }
  2216. } else {
  2217. Ref<VisualShaderNodeIntUniform> int_uniform = Object::cast_to<VisualShaderNodeIntUniform>(node.ptr());
  2218. if (int_uniform.is_valid()) {
  2219. _replace_node(type_id, node_id, "VisualShaderNodeIntUniform", "VisualShaderNodeIntConstant");
  2220. var = int_uniform->get_default_value();
  2221. caught = true;
  2222. }
  2223. }
  2224. }
  2225. // boolean
  2226. if (!caught) {
  2227. if (!p_vice_versa) {
  2228. Ref<VisualShaderNodeBooleanConstant> boolean_const = Object::cast_to<VisualShaderNodeBooleanConstant>(node.ptr());
  2229. if (boolean_const.is_valid()) {
  2230. _replace_node(type_id, node_id, "VisualShaderNodeBooleanConstant", "VisualShaderNodeBooleanUniform");
  2231. var = boolean_const->get_constant();
  2232. caught = true;
  2233. }
  2234. } else {
  2235. Ref<VisualShaderNodeBooleanUniform> boolean_uniform = Object::cast_to<VisualShaderNodeBooleanUniform>(node.ptr());
  2236. if (boolean_uniform.is_valid()) {
  2237. _replace_node(type_id, node_id, "VisualShaderNodeBooleanUniform", "VisualShaderNodeBooleanConstant");
  2238. var = boolean_uniform->get_default_value();
  2239. caught = true;
  2240. }
  2241. }
  2242. }
  2243. // vec3
  2244. if (!caught) {
  2245. if (!p_vice_versa) {
  2246. Ref<VisualShaderNodeVec3Constant> vec3_const = Object::cast_to<VisualShaderNodeVec3Constant>(node.ptr());
  2247. if (vec3_const.is_valid()) {
  2248. _replace_node(type_id, node_id, "VisualShaderNodeVec3Constant", "VisualShaderNodeVec3Uniform");
  2249. var = vec3_const->get_constant();
  2250. caught = true;
  2251. }
  2252. } else {
  2253. Ref<VisualShaderNodeVec3Uniform> vec3_uniform = Object::cast_to<VisualShaderNodeVec3Uniform>(node.ptr());
  2254. if (vec3_uniform.is_valid()) {
  2255. _replace_node(type_id, node_id, "VisualShaderNodeVec3Uniform", "VisualShaderNodeVec3Constant");
  2256. var = vec3_uniform->get_default_value();
  2257. caught = true;
  2258. }
  2259. }
  2260. }
  2261. // color
  2262. if (!caught) {
  2263. if (!p_vice_versa) {
  2264. Ref<VisualShaderNodeColorConstant> color_const = Object::cast_to<VisualShaderNodeColorConstant>(node.ptr());
  2265. if (color_const.is_valid()) {
  2266. _replace_node(type_id, node_id, "VisualShaderNodeColorConstant", "VisualShaderNodeColorUniform");
  2267. var = color_const->get_constant();
  2268. caught = true;
  2269. }
  2270. } else {
  2271. Ref<VisualShaderNodeColorUniform> color_uniform = Object::cast_to<VisualShaderNodeColorUniform>(node.ptr());
  2272. if (color_uniform.is_valid()) {
  2273. _replace_node(type_id, node_id, "VisualShaderNodeColorUniform", "VisualShaderNodeColorConstant");
  2274. var = color_uniform->get_default_value();
  2275. caught = true;
  2276. }
  2277. }
  2278. }
  2279. // transform
  2280. if (!caught) {
  2281. if (!p_vice_versa) {
  2282. Ref<VisualShaderNodeTransformConstant> transform_const = Object::cast_to<VisualShaderNodeTransformConstant>(node.ptr());
  2283. if (transform_const.is_valid()) {
  2284. _replace_node(type_id, node_id, "VisualShaderNodeTransformConstant", "VisualShaderNodeTransformUniform");
  2285. var = transform_const->get_constant();
  2286. caught = true;
  2287. }
  2288. } else {
  2289. Ref<VisualShaderNodeTransformUniform> transform_uniform = Object::cast_to<VisualShaderNodeTransformUniform>(node.ptr());
  2290. if (transform_uniform.is_valid()) {
  2291. _replace_node(type_id, node_id, "VisualShaderNodeTransformUniform", "VisualShaderNodeTransformConstant");
  2292. var = transform_uniform->get_default_value();
  2293. caught = true;
  2294. }
  2295. }
  2296. }
  2297. ERR_CONTINUE(!caught);
  2298. int preview_port = node->get_output_port_for_preview();
  2299. if (!p_vice_versa) {
  2300. undo_redo->add_do_method(this, "_update_uniform", type_id, node_id, var, preview_port);
  2301. undo_redo->add_undo_method(this, "_update_constant", type_id, node_id, var, preview_port);
  2302. } else {
  2303. undo_redo->add_do_method(this, "_update_constant", type_id, node_id, var, preview_port);
  2304. undo_redo->add_undo_method(this, "_update_uniform", type_id, node_id, var, preview_port);
  2305. Ref<VisualShaderNodeUniform> uniform = Object::cast_to<VisualShaderNodeUniform>(node.ptr());
  2306. ERR_CONTINUE(!uniform.is_valid());
  2307. deleted_names.insert(uniform->get_uniform_name());
  2308. }
  2309. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type_id, node_id);
  2310. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type_id, node_id);
  2311. }
  2312. undo_redo->add_do_method(this, "_update_uniforms", true);
  2313. undo_redo->add_undo_method(this, "_update_uniforms", true);
  2314. if (deleted_names.size() > 0) {
  2315. _update_uniform_refs(deleted_names);
  2316. }
  2317. undo_redo->commit_action();
  2318. }
  2319. void VisualShaderEditor::_delete_node_request(int p_type, int p_node) {
  2320. List<int> to_erase;
  2321. to_erase.push_back(p_node);
  2322. undo_redo->create_action(TTR("Delete VisualShader Node"));
  2323. _delete_nodes(p_type, to_erase);
  2324. undo_redo->commit_action();
  2325. }
  2326. void VisualShaderEditor::_delete_nodes_request() {
  2327. List<int> to_erase;
  2328. for (int i = 0; i < graph->get_child_count(); i++) {
  2329. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  2330. if (gn) {
  2331. if (gn->is_selected() && gn->is_close_button_visible()) {
  2332. to_erase.push_back(gn->get_name().operator String().to_int());
  2333. }
  2334. }
  2335. }
  2336. if (to_erase.is_empty()) {
  2337. return;
  2338. }
  2339. undo_redo->create_action(TTR("Delete VisualShader Node(s)"));
  2340. _delete_nodes(get_current_shader_type(), to_erase);
  2341. undo_redo->commit_action();
  2342. }
  2343. void VisualShaderEditor::_node_selected(Object *p_node) {
  2344. VisualShader::Type type = get_current_shader_type();
  2345. GraphNode *gn = Object::cast_to<GraphNode>(p_node);
  2346. ERR_FAIL_COND(!gn);
  2347. int id = String(gn->get_name()).to_int();
  2348. Ref<VisualShaderNode> vsnode = visual_shader->get_node(type, id);
  2349. ERR_FAIL_COND(!vsnode.is_valid());
  2350. //do not rely on this, makes editor more complex
  2351. //EditorNode::get_singleton()->push_item(vsnode.ptr(), "", true);
  2352. }
  2353. void VisualShaderEditor::_graph_gui_input(const Ref<InputEvent> &p_event) {
  2354. Ref<InputEventMouseButton> mb = p_event;
  2355. VisualShader::Type type = get_current_shader_type();
  2356. if (mb.is_valid() && mb->is_pressed() && mb->get_button_index() == MOUSE_BUTTON_RIGHT) {
  2357. selected_constants.clear();
  2358. selected_uniforms.clear();
  2359. selected_comment = -1;
  2360. List<int> to_change;
  2361. for (int i = 0; i < graph->get_child_count(); i++) {
  2362. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  2363. if (gn) {
  2364. if (gn->is_selected() && gn->is_close_button_visible()) {
  2365. int id = gn->get_name().operator String().to_int();
  2366. to_change.push_back(id);
  2367. Ref<VisualShaderNode> node = visual_shader->get_node(type, id);
  2368. VisualShaderNodeComment *comment_node = Object::cast_to<VisualShaderNodeComment>(node.ptr());
  2369. if (comment_node != nullptr) {
  2370. selected_comment = id;
  2371. }
  2372. VisualShaderNodeConstant *constant_node = Object::cast_to<VisualShaderNodeConstant>(node.ptr());
  2373. if (constant_node != nullptr) {
  2374. selected_constants.insert(id);
  2375. }
  2376. VisualShaderNodeUniform *uniform_node = Object::cast_to<VisualShaderNodeUniform>(node.ptr());
  2377. if (uniform_node != nullptr && uniform_node->is_convertible_to_constant()) {
  2378. selected_uniforms.insert(id);
  2379. }
  2380. }
  2381. }
  2382. }
  2383. if (to_change.size() > 1) {
  2384. selected_comment = -1;
  2385. }
  2386. if (to_change.is_empty() && copy_nodes_buffer.is_empty()) {
  2387. _show_members_dialog(true);
  2388. } else {
  2389. popup_menu->set_item_disabled(NodeMenuOptions::COPY, to_change.is_empty());
  2390. popup_menu->set_item_disabled(NodeMenuOptions::PASTE, copy_nodes_buffer.is_empty());
  2391. popup_menu->set_item_disabled(NodeMenuOptions::DELETE, to_change.is_empty());
  2392. popup_menu->set_item_disabled(NodeMenuOptions::DUPLICATE, to_change.is_empty());
  2393. int temp = popup_menu->get_item_index(NodeMenuOptions::SEPARATOR2);
  2394. if (temp != -1) {
  2395. popup_menu->remove_item(temp);
  2396. }
  2397. temp = popup_menu->get_item_index(NodeMenuOptions::CONVERT_CONSTANTS_TO_UNIFORMS);
  2398. if (temp != -1) {
  2399. popup_menu->remove_item(temp);
  2400. }
  2401. temp = popup_menu->get_item_index(NodeMenuOptions::CONVERT_UNIFORMS_TO_CONSTANTS);
  2402. if (temp != -1) {
  2403. popup_menu->remove_item(temp);
  2404. }
  2405. temp = popup_menu->get_item_index(NodeMenuOptions::SEPARATOR3);
  2406. if (temp != -1) {
  2407. popup_menu->remove_item(temp);
  2408. }
  2409. temp = popup_menu->get_item_index(NodeMenuOptions::SET_COMMENT_TITLE);
  2410. if (temp != -1) {
  2411. popup_menu->remove_item(temp);
  2412. }
  2413. temp = popup_menu->get_item_index(NodeMenuOptions::SET_COMMENT_DESCRIPTION);
  2414. if (temp != -1) {
  2415. popup_menu->remove_item(temp);
  2416. }
  2417. if (selected_comment != -1) {
  2418. popup_menu->add_separator("", NodeMenuOptions::SEPARATOR2);
  2419. popup_menu->add_item(TTR("Set Comment Title"), NodeMenuOptions::SET_COMMENT_TITLE);
  2420. popup_menu->add_item(TTR("Set Comment Description"), NodeMenuOptions::SET_COMMENT_DESCRIPTION);
  2421. }
  2422. if (selected_constants.size() > 0 || selected_uniforms.size() > 0) {
  2423. popup_menu->add_separator("", NodeMenuOptions::SEPARATOR3);
  2424. if (selected_constants.size() > 0) {
  2425. popup_menu->add_item(TTR("Convert Constant(s) to Uniform(s)"), NodeMenuOptions::CONVERT_CONSTANTS_TO_UNIFORMS);
  2426. }
  2427. if (selected_uniforms.size() > 0) {
  2428. popup_menu->add_item(TTR("Convert Uniform(s) to Constant(s)"), NodeMenuOptions::CONVERT_UNIFORMS_TO_CONSTANTS);
  2429. }
  2430. }
  2431. menu_point = graph->get_local_mouse_position();
  2432. Point2 gpos = Input::get_singleton()->get_mouse_position();
  2433. popup_menu->set_position(gpos);
  2434. popup_menu->set_size(Size2(-1, -1));
  2435. popup_menu->popup();
  2436. }
  2437. }
  2438. }
  2439. void VisualShaderEditor::_show_members_dialog(bool at_mouse_pos) {
  2440. if (at_mouse_pos) {
  2441. saved_node_pos_dirty = true;
  2442. saved_node_pos = graph->get_local_mouse_position();
  2443. Point2 gpos = Input::get_singleton()->get_mouse_position();
  2444. members_dialog->popup();
  2445. members_dialog->set_position(gpos);
  2446. } else {
  2447. members_dialog->popup();
  2448. saved_node_pos_dirty = false;
  2449. members_dialog->set_position(graph->get_global_position() + Point2(5 * EDSCALE, 65 * EDSCALE));
  2450. }
  2451. // keep dialog within window bounds
  2452. Size2 window_size = DisplayServer::get_singleton()->window_get_size();
  2453. Rect2 dialog_rect = Rect2(members_dialog->get_position(), members_dialog->get_size());
  2454. if (dialog_rect.position.y + dialog_rect.size.y > window_size.y) {
  2455. int difference = dialog_rect.position.y + dialog_rect.size.y - window_size.y;
  2456. members_dialog->set_position(members_dialog->get_position() - Point2(0, difference));
  2457. }
  2458. if (dialog_rect.position.x + dialog_rect.size.x > window_size.x) {
  2459. int difference = dialog_rect.position.x + dialog_rect.size.x - window_size.x;
  2460. members_dialog->set_position(members_dialog->get_position() - Point2(difference, 0));
  2461. }
  2462. node_filter->call_deferred("grab_focus"); // still not visible
  2463. node_filter->select_all();
  2464. }
  2465. void VisualShaderEditor::_sbox_input(const Ref<InputEvent> &p_ie) {
  2466. Ref<InputEventKey> ie = p_ie;
  2467. if (ie.is_valid() && (ie->get_keycode() == KEY_UP ||
  2468. ie->get_keycode() == KEY_DOWN ||
  2469. ie->get_keycode() == KEY_ENTER ||
  2470. ie->get_keycode() == KEY_KP_ENTER)) {
  2471. members->call("_gui_input", ie);
  2472. node_filter->accept_event();
  2473. }
  2474. }
  2475. void VisualShaderEditor::_notification(int p_what) {
  2476. if (p_what == NOTIFICATION_ENTER_TREE) {
  2477. node_filter->set_clear_button_enabled(true);
  2478. // collapse tree by default
  2479. TreeItem *category = members->get_root()->get_first_child();
  2480. while (category) {
  2481. category->set_collapsed(true);
  2482. TreeItem *sub_category = category->get_first_child();
  2483. while (sub_category) {
  2484. sub_category->set_collapsed(true);
  2485. sub_category = sub_category->get_next();
  2486. }
  2487. category = category->get_next();
  2488. }
  2489. }
  2490. if (p_what == NOTIFICATION_DRAG_BEGIN) {
  2491. Dictionary dd = get_viewport()->gui_get_drag_data();
  2492. if (members->is_visible_in_tree() && dd.has("id")) {
  2493. members->set_drop_mode_flags(Tree::DROP_MODE_ON_ITEM);
  2494. }
  2495. } else if (p_what == NOTIFICATION_DRAG_END) {
  2496. members->set_drop_mode_flags(0);
  2497. }
  2498. if (p_what == NOTIFICATION_ENTER_TREE || p_what == NOTIFICATION_THEME_CHANGED) {
  2499. highend_label->set_modulate(get_theme_color("vulkan_color", "Editor"));
  2500. node_filter->set_right_icon(Control::get_theme_icon("Search", "EditorIcons"));
  2501. preview_shader->set_icon(Control::get_theme_icon("Shader", "EditorIcons"));
  2502. {
  2503. Color background_color = EDITOR_GET("text_editor/highlighting/background_color");
  2504. Color text_color = EDITOR_GET("text_editor/highlighting/text_color");
  2505. Color keyword_color = EDITOR_GET("text_editor/highlighting/keyword_color");
  2506. Color control_flow_keyword_color = EDITOR_GET("text_editor/highlighting/control_flow_keyword_color");
  2507. Color comment_color = EDITOR_GET("text_editor/highlighting/comment_color");
  2508. Color symbol_color = EDITOR_GET("text_editor/highlighting/symbol_color");
  2509. Color function_color = EDITOR_GET("text_editor/highlighting/function_color");
  2510. Color number_color = EDITOR_GET("text_editor/highlighting/number_color");
  2511. Color members_color = EDITOR_GET("text_editor/highlighting/member_variable_color");
  2512. preview_text->add_theme_color_override("background_color", background_color);
  2513. for (List<String>::Element *E = keyword_list.front(); E; E = E->next()) {
  2514. if (ShaderLanguage::is_control_flow_keyword(E->get())) {
  2515. syntax_highlighter->add_keyword_color(E->get(), control_flow_keyword_color);
  2516. } else {
  2517. syntax_highlighter->add_keyword_color(E->get(), keyword_color);
  2518. }
  2519. }
  2520. preview_text->add_theme_font_override("font", get_theme_font("expression", "EditorFonts"));
  2521. preview_text->add_theme_font_size_override("font_size", get_theme_font_size("expression_size", "EditorFonts"));
  2522. preview_text->add_theme_color_override("font_color", text_color);
  2523. syntax_highlighter->set_number_color(number_color);
  2524. syntax_highlighter->set_symbol_color(symbol_color);
  2525. syntax_highlighter->set_function_color(function_color);
  2526. syntax_highlighter->set_member_variable_color(members_color);
  2527. syntax_highlighter->clear_color_regions();
  2528. syntax_highlighter->add_color_region("/*", "*/", comment_color, false);
  2529. syntax_highlighter->add_color_region("//", "", comment_color, true);
  2530. preview_text->clear_comment_delimiters();
  2531. preview_text->add_comment_delimiter("/*", "*/", false);
  2532. preview_text->add_comment_delimiter("//", "", true);
  2533. error_panel->add_theme_style_override("panel", get_theme_stylebox("panel", "Panel"));
  2534. error_label->add_theme_font_override("font", get_theme_font("status_source", "EditorFonts"));
  2535. error_label->add_theme_font_size_override("font_size", get_theme_font_size("status_source_size", "EditorFonts"));
  2536. error_label->add_theme_color_override("font_color", get_theme_color("error_color", "Editor"));
  2537. }
  2538. tools->set_icon(EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Tools", "EditorIcons"));
  2539. if (p_what == NOTIFICATION_THEME_CHANGED && is_visible_in_tree()) {
  2540. _update_graph();
  2541. }
  2542. }
  2543. }
  2544. void VisualShaderEditor::_scroll_changed(const Vector2 &p_scroll) {
  2545. if (updating) {
  2546. return;
  2547. }
  2548. updating = true;
  2549. visual_shader->set_graph_offset(p_scroll / EDSCALE);
  2550. updating = false;
  2551. }
  2552. void VisualShaderEditor::_node_changed(int p_id) {
  2553. if (updating) {
  2554. return;
  2555. }
  2556. if (is_visible_in_tree()) {
  2557. _update_graph();
  2558. }
  2559. }
  2560. void VisualShaderEditor::_dup_update_excluded(int p_type, Set<int> &r_excluded) {
  2561. r_excluded.clear();
  2562. VisualShader::Type type = (VisualShader::Type)p_type;
  2563. for (int i = 0; i < graph->get_child_count(); i++) {
  2564. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  2565. if (gn) {
  2566. int id = String(gn->get_name()).to_int();
  2567. Ref<VisualShaderNode> node = visual_shader->get_node(type, id);
  2568. Ref<VisualShaderNodeOutput> output = node;
  2569. if (output.is_valid()) {
  2570. r_excluded.insert(id);
  2571. continue;
  2572. }
  2573. r_excluded.insert(id);
  2574. }
  2575. }
  2576. }
  2577. void VisualShaderEditor::_dup_copy_nodes(int p_type, List<int> &r_nodes, Set<int> &r_excluded) {
  2578. VisualShader::Type type = (VisualShader::Type)p_type;
  2579. selection_center.x = 0.0f;
  2580. selection_center.y = 0.0f;
  2581. for (int i = 0; i < graph->get_child_count(); i++) {
  2582. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  2583. if (gn) {
  2584. int id = String(gn->get_name()).to_int();
  2585. Ref<VisualShaderNode> node = visual_shader->get_node(type, id);
  2586. Ref<VisualShaderNodeOutput> output = node;
  2587. if (output.is_valid()) { // can't duplicate output
  2588. r_excluded.insert(id);
  2589. continue;
  2590. }
  2591. if (node.is_valid() && gn->is_selected()) {
  2592. Vector2 pos = visual_shader->get_node_position(type, id);
  2593. selection_center += pos;
  2594. r_nodes.push_back(id);
  2595. }
  2596. r_excluded.insert(id);
  2597. }
  2598. }
  2599. selection_center /= (float)r_nodes.size();
  2600. }
  2601. 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) {
  2602. VisualShader::Type type = (VisualShader::Type)p_type;
  2603. VisualShader::Type pasted_type = (VisualShader::Type)p_pasted_type;
  2604. int base_id = visual_shader->get_valid_node_id(type);
  2605. int id_from = base_id;
  2606. Map<int, int> connection_remap;
  2607. Set<int> unsupported_set;
  2608. for (List<int>::Element *E = r_nodes.front(); E; E = E->next()) {
  2609. connection_remap[E->get()] = id_from;
  2610. Ref<VisualShaderNode> node = visual_shader->get_node(pasted_type, E->get());
  2611. bool unsupported = false;
  2612. for (int i = 0; i < add_options.size(); i++) {
  2613. if (add_options[i].type == node->get_class_name()) {
  2614. if (!_is_available(add_options[i].mode)) {
  2615. unsupported = true;
  2616. }
  2617. break;
  2618. }
  2619. }
  2620. if (unsupported) {
  2621. unsupported_set.insert(E->get());
  2622. continue;
  2623. }
  2624. Ref<VisualShaderNode> dupli = node->duplicate();
  2625. 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);
  2626. undo_redo->add_do_method(graph_plugin.ptr(), "add_node", type, id_from);
  2627. // duplicate size, inputs and outputs if node is group
  2628. Ref<VisualShaderNodeGroupBase> group = Object::cast_to<VisualShaderNodeGroupBase>(node.ptr());
  2629. if (!group.is_null()) {
  2630. undo_redo->add_do_method(dupli.ptr(), "set_size", group->get_size());
  2631. undo_redo->add_do_method(graph_plugin.ptr(), "set_node_size", type, id_from, group->get_size());
  2632. undo_redo->add_do_method(dupli.ptr(), "set_inputs", group->get_inputs());
  2633. undo_redo->add_do_method(dupli.ptr(), "set_outputs", group->get_outputs());
  2634. }
  2635. // duplicate expression text if node is expression
  2636. Ref<VisualShaderNodeExpression> expression = Object::cast_to<VisualShaderNodeExpression>(node.ptr());
  2637. if (!expression.is_null()) {
  2638. undo_redo->add_do_method(dupli.ptr(), "set_expression", expression->get_expression());
  2639. }
  2640. id_from++;
  2641. }
  2642. List<VisualShader::Connection> conns;
  2643. visual_shader->get_node_connections(pasted_type, &conns);
  2644. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  2645. if (unsupported_set.has(E->get().from_node) || unsupported_set.has(E->get().to_node)) {
  2646. continue;
  2647. }
  2648. if (connection_remap.has(E->get().from_node) && connection_remap.has(E->get().to_node)) {
  2649. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, connection_remap[E->get().from_node], E->get().from_port, connection_remap[E->get().to_node], E->get().to_port);
  2650. undo_redo->add_do_method(graph_plugin.ptr(), "connect_nodes", type, connection_remap[E->get().from_node], E->get().from_port, connection_remap[E->get().to_node], E->get().to_port);
  2651. undo_redo->add_undo_method(graph_plugin.ptr(), "disconnect_nodes", type, connection_remap[E->get().from_node], E->get().from_port, connection_remap[E->get().to_node], E->get().to_port);
  2652. }
  2653. }
  2654. id_from = base_id;
  2655. for (List<int>::Element *E = r_nodes.front(); E; E = E->next()) {
  2656. undo_redo->add_undo_method(visual_shader.ptr(), "remove_node", type, id_from);
  2657. undo_redo->add_undo_method(graph_plugin.ptr(), "remove_node", type, id_from);
  2658. id_from++;
  2659. }
  2660. undo_redo->commit_action();
  2661. if (p_select) {
  2662. // reselect duplicated nodes by excluding the other ones
  2663. for (int i = 0; i < graph->get_child_count(); i++) {
  2664. GraphNode *gn = Object::cast_to<GraphNode>(graph->get_child(i));
  2665. if (gn) {
  2666. int id = String(gn->get_name()).to_int();
  2667. if (!r_excluded.has(id)) {
  2668. gn->set_selected(true);
  2669. } else {
  2670. gn->set_selected(false);
  2671. }
  2672. }
  2673. }
  2674. }
  2675. }
  2676. void VisualShaderEditor::_clear_buffer() {
  2677. copy_nodes_buffer.clear();
  2678. copy_nodes_excluded_buffer.clear();
  2679. }
  2680. void VisualShaderEditor::_duplicate_nodes() {
  2681. int type = get_current_shader_type();
  2682. List<int> nodes;
  2683. Set<int> excluded;
  2684. _dup_copy_nodes(type, nodes, excluded);
  2685. if (nodes.is_empty()) {
  2686. return;
  2687. }
  2688. undo_redo->create_action(TTR("Duplicate VisualShader Node(s)"));
  2689. _dup_paste_nodes(type, type, nodes, excluded, Vector2(10, 10) * EDSCALE, true);
  2690. }
  2691. void VisualShaderEditor::_copy_nodes() {
  2692. copy_type = get_current_shader_type();
  2693. _clear_buffer();
  2694. _dup_copy_nodes(copy_type, copy_nodes_buffer, copy_nodes_excluded_buffer);
  2695. }
  2696. void VisualShaderEditor::_paste_nodes(bool p_use_custom_position, const Vector2 &p_custom_position) {
  2697. if (copy_nodes_buffer.is_empty()) {
  2698. return;
  2699. }
  2700. int type = get_current_shader_type();
  2701. float scale = graph->get_zoom();
  2702. Vector2 mpos;
  2703. if (p_use_custom_position) {
  2704. mpos = p_custom_position;
  2705. } else {
  2706. mpos = graph->get_local_mouse_position();
  2707. }
  2708. undo_redo->create_action(TTR("Paste VisualShader Node(s)"));
  2709. _dup_paste_nodes(type, copy_type, copy_nodes_buffer, copy_nodes_excluded_buffer, (graph->get_scroll_ofs() / scale + mpos / scale - selection_center), false);
  2710. _dup_update_excluded(type, copy_nodes_excluded_buffer); // to prevent selection of previous copies at new paste
  2711. }
  2712. void VisualShaderEditor::_mode_selected(int p_id) {
  2713. int offset = 0;
  2714. if (mode & MODE_FLAGS_PARTICLES) {
  2715. offset = 3;
  2716. if (p_id + offset > VisualShader::TYPE_PROCESS) {
  2717. custom_mode_box->set_visible(false);
  2718. custom_mode_enabled = false;
  2719. } else {
  2720. custom_mode_box->set_visible(true);
  2721. if (custom_mode_box->is_pressed()) {
  2722. custom_mode_enabled = true;
  2723. offset += 3;
  2724. }
  2725. }
  2726. } else if (mode & MODE_FLAGS_SKY) {
  2727. offset = 8;
  2728. }
  2729. visual_shader->set_shader_type(VisualShader::Type(p_id + offset));
  2730. _update_options_menu();
  2731. _update_graph();
  2732. }
  2733. void VisualShaderEditor::_custom_mode_toggled(bool p_enabled) {
  2734. if (!(mode & MODE_FLAGS_PARTICLES)) {
  2735. return;
  2736. }
  2737. custom_mode_enabled = p_enabled;
  2738. int id = edit_type->get_selected() + 3;
  2739. if (p_enabled) {
  2740. visual_shader->set_shader_type(VisualShader::Type(id + 3));
  2741. } else {
  2742. visual_shader->set_shader_type(VisualShader::Type(id));
  2743. }
  2744. _update_options_menu();
  2745. _update_graph();
  2746. }
  2747. void VisualShaderEditor::_input_select_item(Ref<VisualShaderNodeInput> p_input, String p_name) {
  2748. String prev_name = p_input->get_input_name();
  2749. if (p_name == prev_name) {
  2750. return;
  2751. }
  2752. bool type_changed = p_input->get_input_type_by_name(p_name) != p_input->get_input_type_by_name(prev_name);
  2753. UndoRedo *undo_redo = EditorNode::get_singleton()->get_undo_redo();
  2754. undo_redo->create_action(TTR("Visual Shader Input Type Changed"));
  2755. undo_redo->add_do_method(p_input.ptr(), "set_input_name", p_name);
  2756. undo_redo->add_undo_method(p_input.ptr(), "set_input_name", prev_name);
  2757. // update output port
  2758. for (int type_id = 0; type_id < VisualShader::TYPE_MAX; type_id++) {
  2759. VisualShader::Type type = VisualShader::Type(type_id);
  2760. int id = visual_shader->find_node_id(type, p_input);
  2761. if (id != VisualShader::NODE_ID_INVALID) {
  2762. if (type_changed) {
  2763. List<VisualShader::Connection> conns;
  2764. visual_shader->get_node_connections(type, &conns);
  2765. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  2766. if (E->get().from_node == id) {
  2767. if (visual_shader->is_port_types_compatible(p_input->get_input_type_by_name(p_name), visual_shader->get_node(type, E->get().to_node)->get_input_port_type(E->get().to_port))) {
  2768. undo_redo->add_do_method(visual_shader.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2769. 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);
  2770. continue;
  2771. }
  2772. 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);
  2773. 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);
  2774. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2775. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2776. }
  2777. }
  2778. }
  2779. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type_id, id);
  2780. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type_id, id);
  2781. break;
  2782. }
  2783. }
  2784. undo_redo->commit_action();
  2785. }
  2786. void VisualShaderEditor::_uniform_select_item(Ref<VisualShaderNodeUniformRef> p_uniform_ref, String p_name) {
  2787. String prev_name = p_uniform_ref->get_uniform_name();
  2788. if (p_name == prev_name) {
  2789. return;
  2790. }
  2791. bool type_changed = p_uniform_ref->get_uniform_type_by_name(p_name) != p_uniform_ref->get_uniform_type_by_name(prev_name);
  2792. UndoRedo *undo_redo = EditorNode::get_singleton()->get_undo_redo();
  2793. undo_redo->create_action(TTR("UniformRef Name Changed"));
  2794. undo_redo->add_do_method(p_uniform_ref.ptr(), "set_uniform_name", p_name);
  2795. undo_redo->add_undo_method(p_uniform_ref.ptr(), "set_uniform_name", prev_name);
  2796. // update output port
  2797. for (int type_id = 0; type_id < VisualShader::TYPE_MAX; type_id++) {
  2798. VisualShader::Type type = VisualShader::Type(type_id);
  2799. int id = visual_shader->find_node_id(type, p_uniform_ref);
  2800. if (id != VisualShader::NODE_ID_INVALID) {
  2801. if (type_changed) {
  2802. List<VisualShader::Connection> conns;
  2803. visual_shader->get_node_connections(type, &conns);
  2804. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  2805. if (E->get().from_node == id) {
  2806. if (visual_shader->is_port_types_compatible(p_uniform_ref->get_uniform_type_by_name(p_name), visual_shader->get_node(type, E->get().to_node)->get_input_port_type(E->get().to_port))) {
  2807. continue;
  2808. }
  2809. 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);
  2810. 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);
  2811. undo_redo->add_do_method(graph_plugin.ptr(), "disconnect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2812. undo_redo->add_undo_method(graph_plugin.ptr(), "connect_nodes", type, E->get().from_node, E->get().from_port, E->get().to_node, E->get().to_port);
  2813. }
  2814. }
  2815. }
  2816. undo_redo->add_do_method(graph_plugin.ptr(), "update_node", type_id, id);
  2817. undo_redo->add_undo_method(graph_plugin.ptr(), "update_node", type_id, id);
  2818. break;
  2819. }
  2820. }
  2821. undo_redo->commit_action();
  2822. }
  2823. void VisualShaderEditor::_float_constant_selected(int p_index, int p_node) {
  2824. if (p_index == 0) {
  2825. graph_plugin->update_node_size(p_node);
  2826. return;
  2827. }
  2828. --p_index;
  2829. ERR_FAIL_INDEX(p_index, MAX_FLOAT_CONST_DEFS);
  2830. VisualShader::Type type = get_current_shader_type();
  2831. Ref<VisualShaderNodeFloatConstant> node = visual_shader->get_node(type, p_node);
  2832. if (!node.is_valid()) {
  2833. return;
  2834. }
  2835. undo_redo->create_action(TTR("Set constant"));
  2836. undo_redo->add_do_method(node.ptr(), "set_constant", float_constant_defs[p_index].value);
  2837. undo_redo->add_undo_method(node.ptr(), "set_constant", node->get_constant());
  2838. undo_redo->add_do_method(graph_plugin.ptr(), "update_constant", type, p_node);
  2839. undo_redo->add_undo_method(graph_plugin.ptr(), "update_constant", type, p_node);
  2840. undo_redo->commit_action();
  2841. }
  2842. void VisualShaderEditor::_member_filter_changed(const String &p_text) {
  2843. _update_options_menu();
  2844. }
  2845. void VisualShaderEditor::_member_selected() {
  2846. TreeItem *item = members->get_selected();
  2847. if (item != nullptr && item->has_meta("id")) {
  2848. members_dialog->get_ok_button()->set_disabled(false);
  2849. highend_label->set_visible(add_options[item->get_meta("id")].highend);
  2850. node_desc->set_text(_get_description(item->get_meta("id")));
  2851. } else {
  2852. highend_label->set_visible(false);
  2853. members_dialog->get_ok_button()->set_disabled(true);
  2854. node_desc->set_text("");
  2855. }
  2856. }
  2857. void VisualShaderEditor::_member_unselected() {
  2858. }
  2859. void VisualShaderEditor::_member_create() {
  2860. TreeItem *item = members->get_selected();
  2861. if (item != nullptr && item->has_meta("id")) {
  2862. int idx = members->get_selected()->get_meta("id");
  2863. _add_node(idx, add_options[idx].sub_func);
  2864. members_dialog->hide();
  2865. }
  2866. }
  2867. void VisualShaderEditor::_member_cancel() {
  2868. to_node = -1;
  2869. to_slot = -1;
  2870. from_node = -1;
  2871. from_slot = -1;
  2872. }
  2873. void VisualShaderEditor::_tools_menu_option(int p_idx) {
  2874. TreeItem *category = members->get_root()->get_first_child();
  2875. switch (p_idx) {
  2876. case EXPAND_ALL:
  2877. while (category) {
  2878. category->set_collapsed(false);
  2879. TreeItem *sub_category = category->get_first_child();
  2880. while (sub_category) {
  2881. sub_category->set_collapsed(false);
  2882. sub_category = sub_category->get_next();
  2883. }
  2884. category = category->get_next();
  2885. }
  2886. break;
  2887. case COLLAPSE_ALL:
  2888. while (category) {
  2889. category->set_collapsed(true);
  2890. TreeItem *sub_category = category->get_first_child();
  2891. while (sub_category) {
  2892. sub_category->set_collapsed(true);
  2893. sub_category = sub_category->get_next();
  2894. }
  2895. category = category->get_next();
  2896. }
  2897. break;
  2898. default:
  2899. break;
  2900. }
  2901. }
  2902. void VisualShaderEditor::_node_menu_id_pressed(int p_idx) {
  2903. switch (p_idx) {
  2904. case NodeMenuOptions::ADD:
  2905. _show_members_dialog(true);
  2906. break;
  2907. case NodeMenuOptions::COPY:
  2908. _copy_nodes();
  2909. break;
  2910. case NodeMenuOptions::PASTE:
  2911. _paste_nodes(true, menu_point);
  2912. break;
  2913. case NodeMenuOptions::DELETE:
  2914. _delete_nodes_request();
  2915. break;
  2916. case NodeMenuOptions::DUPLICATE:
  2917. _duplicate_nodes();
  2918. break;
  2919. case NodeMenuOptions::CONVERT_CONSTANTS_TO_UNIFORMS:
  2920. _convert_constants_to_uniforms(false);
  2921. break;
  2922. case NodeMenuOptions::CONVERT_UNIFORMS_TO_CONSTANTS:
  2923. _convert_constants_to_uniforms(true);
  2924. break;
  2925. case NodeMenuOptions::SET_COMMENT_TITLE:
  2926. _comment_title_popup_show(get_global_mouse_position(), selected_comment);
  2927. break;
  2928. case NodeMenuOptions::SET_COMMENT_DESCRIPTION:
  2929. _comment_desc_popup_show(get_global_mouse_position(), selected_comment);
  2930. break;
  2931. default:
  2932. break;
  2933. }
  2934. }
  2935. Variant VisualShaderEditor::get_drag_data_fw(const Point2 &p_point, Control *p_from) {
  2936. if (p_from == members) {
  2937. TreeItem *it = members->get_item_at_position(p_point);
  2938. if (!it) {
  2939. return Variant();
  2940. }
  2941. if (!it->has_meta("id")) {
  2942. return Variant();
  2943. }
  2944. int id = it->get_meta("id");
  2945. AddOption op = add_options[id];
  2946. Dictionary d;
  2947. d["id"] = id;
  2948. if (op.sub_func == -1) {
  2949. d["sub_func"] = op.sub_func_str;
  2950. } else {
  2951. d["sub_func"] = op.sub_func;
  2952. }
  2953. Label *label = memnew(Label);
  2954. label->set_text(it->get_text(0));
  2955. set_drag_preview(label);
  2956. return d;
  2957. }
  2958. return Variant();
  2959. }
  2960. bool VisualShaderEditor::can_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) const {
  2961. if (p_from == graph) {
  2962. Dictionary d = p_data;
  2963. if (d.has("id")) {
  2964. return true;
  2965. }
  2966. if (d.has("files")) {
  2967. return true;
  2968. }
  2969. }
  2970. return false;
  2971. }
  2972. void VisualShaderEditor::drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) {
  2973. if (p_from == graph) {
  2974. Dictionary d = p_data;
  2975. if (d.has("id")) {
  2976. int idx = d["id"];
  2977. saved_node_pos = p_point;
  2978. saved_node_pos_dirty = true;
  2979. _add_node(idx, add_options[idx].sub_func);
  2980. } else if (d.has("files")) {
  2981. undo_redo->create_action(TTR("Add Node(s) to Visual Shader"));
  2982. if (d["files"].get_type() == Variant::PACKED_STRING_ARRAY) {
  2983. PackedStringArray arr = d["files"];
  2984. for (int i = 0; i < arr.size(); i++) {
  2985. String type = ResourceLoader::get_resource_type(arr[i]);
  2986. if (type == "GDScript") {
  2987. Ref<Script> script = ResourceLoader::load(arr[i]);
  2988. if (script->get_instance_base_type() == "VisualShaderNodeCustom") {
  2989. saved_node_pos = p_point + Vector2(0, i * 250 * EDSCALE);
  2990. saved_node_pos_dirty = true;
  2991. int idx = -1;
  2992. for (int j = custom_node_option_idx; j < add_options.size(); j++) {
  2993. if (add_options[j].script.is_valid()) {
  2994. if (add_options[j].script->get_path() == arr[i]) {
  2995. idx = j;
  2996. break;
  2997. }
  2998. }
  2999. }
  3000. if (idx != -1) {
  3001. _add_node(idx, -1, arr[i], i);
  3002. }
  3003. }
  3004. } else if (type == "CurveTexture") {
  3005. saved_node_pos = p_point + Vector2(0, i * 250 * EDSCALE);
  3006. saved_node_pos_dirty = true;
  3007. _add_node(curve_node_option_idx, -1, arr[i], i);
  3008. } else if (ClassDB::get_parent_class(type) == "Texture2D") {
  3009. saved_node_pos = p_point + Vector2(0, i * 250 * EDSCALE);
  3010. saved_node_pos_dirty = true;
  3011. _add_node(texture2d_node_option_idx, -1, arr[i], i);
  3012. } else if (type == "Texture2DArray") {
  3013. saved_node_pos = p_point + Vector2(0, i * 250 * EDSCALE);
  3014. saved_node_pos_dirty = true;
  3015. _add_node(texture2d_array_node_option_idx, -1, arr[i], i);
  3016. } else if (ClassDB::get_parent_class(type) == "Texture3D") {
  3017. saved_node_pos = p_point + Vector2(0, i * 250 * EDSCALE);
  3018. saved_node_pos_dirty = true;
  3019. _add_node(texture3d_node_option_idx, -1, arr[i], i);
  3020. } else if (type == "Cubemap") {
  3021. saved_node_pos = p_point + Vector2(0, i * 250 * EDSCALE);
  3022. saved_node_pos_dirty = true;
  3023. _add_node(cubemap_node_option_idx, -1, arr[i], i);
  3024. }
  3025. }
  3026. }
  3027. undo_redo->commit_action();
  3028. }
  3029. }
  3030. }
  3031. void VisualShaderEditor::_show_preview_text() {
  3032. preview_showed = !preview_showed;
  3033. if (preview_showed) {
  3034. if (preview_first) {
  3035. preview_window->set_size(Size2(400 * EDSCALE, 600 * EDSCALE));
  3036. preview_window->popup_centered();
  3037. preview_first = false;
  3038. } else {
  3039. preview_window->popup();
  3040. }
  3041. _preview_size_changed();
  3042. if (pending_update_preview) {
  3043. _update_preview();
  3044. pending_update_preview = false;
  3045. }
  3046. } else {
  3047. preview_window->hide();
  3048. }
  3049. }
  3050. void VisualShaderEditor::_preview_close_requested() {
  3051. preview_showed = false;
  3052. preview_window->hide();
  3053. preview_shader->set_pressed(false);
  3054. }
  3055. void VisualShaderEditor::_preview_size_changed() {
  3056. preview_vbox->set_custom_minimum_size(preview_window->get_size());
  3057. }
  3058. static ShaderLanguage::DataType _get_global_variable_type(const StringName &p_variable) {
  3059. RS::GlobalVariableType gvt = RS::get_singleton()->global_variable_get_type(p_variable);
  3060. return RS::global_variable_type_get_shader_datatype(gvt);
  3061. }
  3062. void VisualShaderEditor::_update_preview() {
  3063. if (!preview_showed) {
  3064. pending_update_preview = true;
  3065. return;
  3066. }
  3067. String code = visual_shader->get_code();
  3068. preview_text->set_text(code);
  3069. ShaderLanguage sl;
  3070. 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())), ShaderLanguage::VaryingFunctionNames(), ShaderTypes::get_singleton()->get_types(), _get_global_variable_type);
  3071. for (int i = 0; i < preview_text->get_line_count(); i++) {
  3072. preview_text->set_line_background_color(i, Color(0, 0, 0, 0));
  3073. }
  3074. if (err != OK) {
  3075. Color error_line_color = EDITOR_GET("text_editor/highlighting/mark_color");
  3076. preview_text->set_line_background_color(sl.get_error_line() - 1, error_line_color);
  3077. error_panel->show();
  3078. String text = "error(" + itos(sl.get_error_line()) + "): " + sl.get_error_text();
  3079. error_label->set_text(text);
  3080. shader_error = true;
  3081. } else {
  3082. error_panel->hide();
  3083. shader_error = false;
  3084. }
  3085. }
  3086. void VisualShaderEditor::_visibility_changed() {
  3087. if (!is_visible()) {
  3088. if (preview_window->is_visible()) {
  3089. preview_shader->set_pressed(false);
  3090. preview_window->hide();
  3091. preview_showed = false;
  3092. }
  3093. }
  3094. }
  3095. void VisualShaderEditor::_bind_methods() {
  3096. ClassDB::bind_method("_update_graph", &VisualShaderEditor::_update_graph);
  3097. ClassDB::bind_method("_update_options_menu", &VisualShaderEditor::_update_options_menu);
  3098. ClassDB::bind_method("_add_node", &VisualShaderEditor::_add_node);
  3099. ClassDB::bind_method("_node_changed", &VisualShaderEditor::_node_changed);
  3100. ClassDB::bind_method("_input_select_item", &VisualShaderEditor::_input_select_item);
  3101. ClassDB::bind_method("_uniform_select_item", &VisualShaderEditor::_uniform_select_item);
  3102. ClassDB::bind_method("_set_node_size", &VisualShaderEditor::_set_node_size);
  3103. ClassDB::bind_method("_clear_buffer", &VisualShaderEditor::_clear_buffer);
  3104. ClassDB::bind_method("_update_uniforms", &VisualShaderEditor::_update_uniforms);
  3105. ClassDB::bind_method("_set_mode", &VisualShaderEditor::_set_mode);
  3106. ClassDB::bind_method("_nodes_dragged", &VisualShaderEditor::_nodes_dragged);
  3107. ClassDB::bind_method("_float_constant_selected", &VisualShaderEditor::_float_constant_selected);
  3108. ClassDB::bind_method("_update_constant", &VisualShaderEditor::_update_constant);
  3109. ClassDB::bind_method("_update_uniform", &VisualShaderEditor::_update_uniform);
  3110. ClassDB::bind_method("_expand_output_port", &VisualShaderEditor::_expand_output_port);
  3111. ClassDB::bind_method(D_METHOD("_get_drag_data_fw"), &VisualShaderEditor::get_drag_data_fw);
  3112. ClassDB::bind_method(D_METHOD("_can_drop_data_fw"), &VisualShaderEditor::can_drop_data_fw);
  3113. ClassDB::bind_method(D_METHOD("_drop_data_fw"), &VisualShaderEditor::drop_data_fw);
  3114. ClassDB::bind_method("_is_available", &VisualShaderEditor::_is_available);
  3115. }
  3116. VisualShaderEditor *VisualShaderEditor::singleton = nullptr;
  3117. VisualShaderEditor::VisualShaderEditor() {
  3118. singleton = this;
  3119. updating = false;
  3120. saved_node_pos_dirty = false;
  3121. saved_node_pos = Point2(0, 0);
  3122. ShaderLanguage::get_keyword_list(&keyword_list);
  3123. pending_update_preview = false;
  3124. shader_error = false;
  3125. to_node = -1;
  3126. to_slot = -1;
  3127. from_node = -1;
  3128. from_slot = -1;
  3129. graph = memnew(GraphEdit);
  3130. graph->get_zoom_hbox()->set_h_size_flags(SIZE_EXPAND_FILL);
  3131. graph->set_v_size_flags(SIZE_EXPAND_FILL);
  3132. graph->set_h_size_flags(SIZE_EXPAND_FILL);
  3133. add_child(graph);
  3134. graph->set_drag_forwarding(this);
  3135. float graph_minimap_opacity = EditorSettings::get_singleton()->get("editors/visual_editors/minimap_opacity");
  3136. graph->set_minimap_opacity(graph_minimap_opacity);
  3137. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_SCALAR);
  3138. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_SCALAR_INT);
  3139. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_BOOLEAN);
  3140. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_VECTOR);
  3141. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_TRANSFORM);
  3142. graph->add_valid_right_disconnect_type(VisualShaderNode::PORT_TYPE_SAMPLER);
  3143. //graph->add_valid_left_disconnect_type(0);
  3144. graph->set_v_size_flags(SIZE_EXPAND_FILL);
  3145. graph->connect("connection_request", callable_mp(this, &VisualShaderEditor::_connection_request), varray(), CONNECT_DEFERRED);
  3146. graph->connect("disconnection_request", callable_mp(this, &VisualShaderEditor::_disconnection_request), varray(), CONNECT_DEFERRED);
  3147. graph->connect("node_selected", callable_mp(this, &VisualShaderEditor::_node_selected));
  3148. graph->connect("scroll_offset_changed", callable_mp(this, &VisualShaderEditor::_scroll_changed));
  3149. graph->connect("duplicate_nodes_request", callable_mp(this, &VisualShaderEditor::_duplicate_nodes));
  3150. graph->connect("copy_nodes_request", callable_mp(this, &VisualShaderEditor::_copy_nodes));
  3151. graph->connect("paste_nodes_request", callable_mp(this, &VisualShaderEditor::_paste_nodes), varray(false, Point2()));
  3152. graph->connect("delete_nodes_request", callable_mp(this, &VisualShaderEditor::_delete_nodes_request));
  3153. graph->connect("gui_input", callable_mp(this, &VisualShaderEditor::_graph_gui_input));
  3154. graph->connect("connection_to_empty", callable_mp(this, &VisualShaderEditor::_connection_to_empty));
  3155. graph->connect("connection_from_empty", callable_mp(this, &VisualShaderEditor::_connection_from_empty));
  3156. graph->connect("visibility_changed", callable_mp(this, &VisualShaderEditor::_visibility_changed));
  3157. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR);
  3158. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR_INT);
  3159. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_VECTOR);
  3160. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_BOOLEAN);
  3161. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShaderNode::PORT_TYPE_SCALAR_INT);
  3162. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShaderNode::PORT_TYPE_SCALAR);
  3163. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShaderNode::PORT_TYPE_VECTOR);
  3164. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SCALAR_INT, VisualShaderNode::PORT_TYPE_BOOLEAN);
  3165. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_SCALAR);
  3166. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_SCALAR_INT);
  3167. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR);
  3168. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_BOOLEAN);
  3169. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_SCALAR);
  3170. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_SCALAR_INT);
  3171. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_VECTOR);
  3172. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_BOOLEAN);
  3173. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_TRANSFORM, VisualShaderNode::PORT_TYPE_TRANSFORM);
  3174. graph->add_valid_connection_type(VisualShaderNode::PORT_TYPE_SAMPLER, VisualShaderNode::PORT_TYPE_SAMPLER);
  3175. VSeparator *vs = memnew(VSeparator);
  3176. graph->get_zoom_hbox()->add_child(vs);
  3177. graph->get_zoom_hbox()->move_child(vs, 0);
  3178. custom_mode_box = memnew(CheckBox);
  3179. custom_mode_box->set_text(TTR("Custom"));
  3180. custom_mode_box->set_pressed(false);
  3181. custom_mode_box->set_visible(false);
  3182. custom_mode_box->connect("toggled", callable_mp(this, &VisualShaderEditor::_custom_mode_toggled));
  3183. edit_type_standard = memnew(OptionButton);
  3184. edit_type_standard->add_item(TTR("Vertex"));
  3185. edit_type_standard->add_item(TTR("Fragment"));
  3186. edit_type_standard->add_item(TTR("Light"));
  3187. edit_type_standard->select(1);
  3188. edit_type_standard->connect("item_selected", callable_mp(this, &VisualShaderEditor::_mode_selected));
  3189. edit_type_particles = memnew(OptionButton);
  3190. edit_type_particles->add_item(TTR("Start"));
  3191. edit_type_particles->add_item(TTR("Process"));
  3192. edit_type_particles->add_item(TTR("Collide"));
  3193. edit_type_particles->select(0);
  3194. edit_type_particles->connect("item_selected", callable_mp(this, &VisualShaderEditor::_mode_selected));
  3195. edit_type_sky = memnew(OptionButton);
  3196. edit_type_sky->add_item(TTR("Sky"));
  3197. edit_type_sky->select(0);
  3198. edit_type_sky->connect("item_selected", callable_mp(this, &VisualShaderEditor::_mode_selected));
  3199. edit_type = edit_type_standard;
  3200. graph->get_zoom_hbox()->add_child(custom_mode_box);
  3201. graph->get_zoom_hbox()->move_child(custom_mode_box, 0);
  3202. graph->get_zoom_hbox()->add_child(edit_type_standard);
  3203. graph->get_zoom_hbox()->move_child(edit_type_standard, 0);
  3204. graph->get_zoom_hbox()->add_child(edit_type_particles);
  3205. graph->get_zoom_hbox()->move_child(edit_type_particles, 0);
  3206. graph->get_zoom_hbox()->add_child(edit_type_sky);
  3207. graph->get_zoom_hbox()->move_child(edit_type_sky, 0);
  3208. add_node = memnew(Button);
  3209. add_node->set_flat(true);
  3210. graph->get_zoom_hbox()->add_child(add_node);
  3211. add_node->set_text(TTR("Add Node..."));
  3212. graph->get_zoom_hbox()->move_child(add_node, 0);
  3213. add_node->connect("pressed", callable_mp(this, &VisualShaderEditor::_show_members_dialog), varray(false));
  3214. preview_shader = memnew(Button);
  3215. preview_shader->set_flat(true);
  3216. preview_shader->set_toggle_mode(true);
  3217. preview_shader->set_tooltip(TTR("Show generated shader code."));
  3218. graph->get_zoom_hbox()->add_child(preview_shader);
  3219. preview_shader->connect("pressed", callable_mp(this, &VisualShaderEditor::_show_preview_text));
  3220. ///////////////////////////////////////
  3221. // PREVIEW WINDOW
  3222. ///////////////////////////////////////
  3223. preview_window = memnew(Window);
  3224. preview_window->set_title(TTR("Generated shader code"));
  3225. preview_window->set_visible(preview_showed);
  3226. preview_window->connect("close_requested", callable_mp(this, &VisualShaderEditor::_preview_close_requested));
  3227. preview_window->connect("size_changed", callable_mp(this, &VisualShaderEditor::_preview_size_changed));
  3228. add_child(preview_window);
  3229. preview_vbox = memnew(VBoxContainer);
  3230. preview_window->add_child(preview_vbox);
  3231. preview_vbox->add_theme_constant_override("separation", 0);
  3232. preview_text = memnew(CodeEdit);
  3233. syntax_highlighter.instantiate();
  3234. preview_vbox->add_child(preview_text);
  3235. preview_text->set_v_size_flags(Control::SIZE_EXPAND_FILL);
  3236. preview_text->set_syntax_highlighter(syntax_highlighter);
  3237. preview_text->set_draw_line_numbers(true);
  3238. preview_text->set_readonly(true);
  3239. error_panel = memnew(PanelContainer);
  3240. preview_vbox->add_child(error_panel);
  3241. error_panel->set_visible(false);
  3242. error_label = memnew(Label);
  3243. error_panel->add_child(error_label);
  3244. error_label->set_autowrap_mode(Label::AUTOWRAP_WORD_SMART);
  3245. ///////////////////////////////////////
  3246. // POPUP MENU
  3247. ///////////////////////////////////////
  3248. popup_menu = memnew(PopupMenu);
  3249. add_child(popup_menu);
  3250. popup_menu->add_item(TTR("Add Node"), NodeMenuOptions::ADD);
  3251. popup_menu->add_separator();
  3252. popup_menu->add_item(TTR("Copy"), NodeMenuOptions::COPY);
  3253. popup_menu->add_item(TTR("Paste"), NodeMenuOptions::PASTE);
  3254. popup_menu->add_item(TTR("Delete"), NodeMenuOptions::DELETE);
  3255. popup_menu->add_item(TTR("Duplicate"), NodeMenuOptions::DUPLICATE);
  3256. popup_menu->connect("id_pressed", callable_mp(this, &VisualShaderEditor::_node_menu_id_pressed));
  3257. ///////////////////////////////////////
  3258. // SHADER NODES TREE
  3259. ///////////////////////////////////////
  3260. VBoxContainer *members_vb = memnew(VBoxContainer);
  3261. members_vb->set_v_size_flags(SIZE_EXPAND_FILL);
  3262. HBoxContainer *filter_hb = memnew(HBoxContainer);
  3263. members_vb->add_child(filter_hb);
  3264. node_filter = memnew(LineEdit);
  3265. filter_hb->add_child(node_filter);
  3266. node_filter->connect("text_changed", callable_mp(this, &VisualShaderEditor::_member_filter_changed));
  3267. node_filter->connect("gui_input", callable_mp(this, &VisualShaderEditor::_sbox_input));
  3268. node_filter->set_h_size_flags(SIZE_EXPAND_FILL);
  3269. node_filter->set_placeholder(TTR("Search"));
  3270. tools = memnew(MenuButton);
  3271. filter_hb->add_child(tools);
  3272. tools->set_tooltip(TTR("Options"));
  3273. tools->get_popup()->connect("id_pressed", callable_mp(this, &VisualShaderEditor::_tools_menu_option));
  3274. tools->get_popup()->add_item(TTR("Expand All"), EXPAND_ALL);
  3275. tools->get_popup()->add_item(TTR("Collapse All"), COLLAPSE_ALL);
  3276. members = memnew(Tree);
  3277. members_vb->add_child(members);
  3278. members->set_drag_forwarding(this);
  3279. members->set_h_size_flags(SIZE_EXPAND_FILL);
  3280. members->set_v_size_flags(SIZE_EXPAND_FILL);
  3281. members->set_hide_root(true);
  3282. members->set_allow_reselect(true);
  3283. members->set_hide_folding(false);
  3284. members->set_custom_minimum_size(Size2(180 * EDSCALE, 200 * EDSCALE));
  3285. members->connect("item_activated", callable_mp(this, &VisualShaderEditor::_member_create));
  3286. members->connect("item_selected", callable_mp(this, &VisualShaderEditor::_member_selected));
  3287. members->connect("nothing_selected", callable_mp(this, &VisualShaderEditor::_member_unselected));
  3288. HBoxContainer *desc_hbox = memnew(HBoxContainer);
  3289. members_vb->add_child(desc_hbox);
  3290. Label *desc_label = memnew(Label);
  3291. desc_hbox->add_child(desc_label);
  3292. desc_label->set_text(TTR("Description:"));
  3293. desc_hbox->add_spacer();
  3294. highend_label = memnew(Label);
  3295. desc_hbox->add_child(highend_label);
  3296. highend_label->set_visible(false);
  3297. highend_label->set_text("Vulkan");
  3298. highend_label->set_mouse_filter(Control::MOUSE_FILTER_STOP);
  3299. highend_label->set_tooltip(TTR("High-end node"));
  3300. node_desc = memnew(RichTextLabel);
  3301. members_vb->add_child(node_desc);
  3302. node_desc->set_h_size_flags(SIZE_EXPAND_FILL);
  3303. node_desc->set_v_size_flags(SIZE_FILL);
  3304. node_desc->set_custom_minimum_size(Size2(0, 70 * EDSCALE));
  3305. members_dialog = memnew(ConfirmationDialog);
  3306. members_dialog->set_title(TTR("Create Shader Node"));
  3307. members_dialog->set_exclusive(false);
  3308. members_dialog->add_child(members_vb);
  3309. members_dialog->get_ok_button()->set_text(TTR("Create"));
  3310. members_dialog->get_ok_button()->connect("pressed", callable_mp(this, &VisualShaderEditor::_member_create));
  3311. members_dialog->get_ok_button()->set_disabled(true);
  3312. members_dialog->connect("cancelled", callable_mp(this, &VisualShaderEditor::_member_cancel));
  3313. add_child(members_dialog);
  3314. alert = memnew(AcceptDialog);
  3315. alert->get_label()->set_autowrap_mode(Label::AUTOWRAP_WORD);
  3316. alert->get_label()->set_align(Label::ALIGN_CENTER);
  3317. alert->get_label()->set_valign(Label::VALIGN_CENTER);
  3318. alert->get_label()->set_custom_minimum_size(Size2(400, 60) * EDSCALE);
  3319. add_child(alert);
  3320. comment_title_change_popup = memnew(PopupPanel);
  3321. comment_title_change_edit = memnew(LineEdit);
  3322. comment_title_change_edit->set_expand_to_text_length_enabled(true);
  3323. comment_title_change_edit->connect("text_changed", callable_mp(this, &VisualShaderEditor::_comment_title_text_changed));
  3324. comment_title_change_edit->connect("text_submitted", callable_mp(this, &VisualShaderEditor::_comment_title_text_submitted));
  3325. comment_title_change_popup->add_child(comment_title_change_edit);
  3326. comment_title_change_edit->set_size(Size2(-1, -1));
  3327. comment_title_change_popup->set_size(Size2(-1, -1));
  3328. comment_title_change_popup->connect("focus_exited", callable_mp(this, &VisualShaderEditor::_comment_title_popup_focus_out));
  3329. comment_title_change_popup->connect("popup_hide", callable_mp(this, &VisualShaderEditor::_comment_title_popup_hide));
  3330. add_child(comment_title_change_popup);
  3331. comment_desc_change_popup = memnew(PopupPanel);
  3332. VBoxContainer *comment_desc_vbox = memnew(VBoxContainer);
  3333. comment_desc_change_popup->add_child(comment_desc_vbox);
  3334. comment_desc_change_edit = memnew(TextEdit);
  3335. comment_desc_change_edit->connect("text_changed", callable_mp(this, &VisualShaderEditor::_comment_desc_text_changed));
  3336. comment_desc_vbox->add_child(comment_desc_change_edit);
  3337. comment_desc_change_edit->set_custom_minimum_size(Size2(300 * EDSCALE, 150 * EDSCALE));
  3338. comment_desc_change_edit->set_size(Size2(-1, -1));
  3339. comment_desc_change_popup->set_size(Size2(-1, -1));
  3340. comment_desc_change_popup->connect("focus_exited", callable_mp(this, &VisualShaderEditor::_comment_desc_confirm));
  3341. comment_desc_change_popup->connect("popup_hide", callable_mp(this, &VisualShaderEditor::_comment_desc_popup_hide));
  3342. Button *comment_desc_confirm_button = memnew(Button);
  3343. comment_desc_confirm_button->set_text(TTR("OK"));
  3344. comment_desc_vbox->add_child(comment_desc_confirm_button);
  3345. comment_desc_confirm_button->connect("pressed", callable_mp(this, &VisualShaderEditor::_comment_desc_confirm));
  3346. add_child(comment_desc_change_popup);
  3347. ///////////////////////////////////////
  3348. // SHADER NODES TREE OPTIONS
  3349. ///////////////////////////////////////
  3350. // COLOR
  3351. add_options.push_back(AddOption("ColorFunc", "Color", "Common", "VisualShaderNodeColorFunc", TTR("Color function."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3352. add_options.push_back(AddOption("ColorOp", "Color", "Common", "VisualShaderNodeColorOp", TTR("Color operator."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3353. add_options.push_back(AddOption("Grayscale", "Color", "Functions", "VisualShaderNodeColorFunc", TTR("Grayscale function."), VisualShaderNodeColorFunc::FUNC_GRAYSCALE, VisualShaderNode::PORT_TYPE_VECTOR));
  3354. add_options.push_back(AddOption("HSV2RGB", "Color", "Functions", "VisualShaderNodeVectorFunc", TTR("Converts HSV vector to RGB equivalent."), VisualShaderNodeVectorFunc::FUNC_HSV2RGB, VisualShaderNode::PORT_TYPE_VECTOR));
  3355. add_options.push_back(AddOption("RGB2HSV", "Color", "Functions", "VisualShaderNodeVectorFunc", TTR("Converts RGB vector to HSV equivalent."), VisualShaderNodeVectorFunc::FUNC_RGB2HSV, VisualShaderNode::PORT_TYPE_VECTOR));
  3356. add_options.push_back(AddOption("Sepia", "Color", "Functions", "VisualShaderNodeColorFunc", TTR("Sepia function."), VisualShaderNodeColorFunc::FUNC_SEPIA, VisualShaderNode::PORT_TYPE_VECTOR));
  3357. add_options.push_back(AddOption("Burn", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Burn operator."), VisualShaderNodeColorOp::OP_BURN, VisualShaderNode::PORT_TYPE_VECTOR));
  3358. add_options.push_back(AddOption("Darken", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Darken operator."), VisualShaderNodeColorOp::OP_DARKEN, VisualShaderNode::PORT_TYPE_VECTOR));
  3359. add_options.push_back(AddOption("Difference", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Difference operator."), VisualShaderNodeColorOp::OP_DIFFERENCE, VisualShaderNode::PORT_TYPE_VECTOR));
  3360. add_options.push_back(AddOption("Dodge", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Dodge operator."), VisualShaderNodeColorOp::OP_DODGE, VisualShaderNode::PORT_TYPE_VECTOR));
  3361. add_options.push_back(AddOption("HardLight", "Color", "Operators", "VisualShaderNodeColorOp", TTR("HardLight operator."), VisualShaderNodeColorOp::OP_HARD_LIGHT, VisualShaderNode::PORT_TYPE_VECTOR));
  3362. add_options.push_back(AddOption("Lighten", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Lighten operator."), VisualShaderNodeColorOp::OP_LIGHTEN, VisualShaderNode::PORT_TYPE_VECTOR));
  3363. add_options.push_back(AddOption("Overlay", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Overlay operator."), VisualShaderNodeColorOp::OP_OVERLAY, VisualShaderNode::PORT_TYPE_VECTOR));
  3364. add_options.push_back(AddOption("Screen", "Color", "Operators", "VisualShaderNodeColorOp", TTR("Screen operator."), VisualShaderNodeColorOp::OP_SCREEN, VisualShaderNode::PORT_TYPE_VECTOR));
  3365. add_options.push_back(AddOption("SoftLight", "Color", "Operators", "VisualShaderNodeColorOp", TTR("SoftLight operator."), VisualShaderNodeColorOp::OP_SOFT_LIGHT, VisualShaderNode::PORT_TYPE_VECTOR));
  3366. add_options.push_back(AddOption("ColorConstant", "Color", "Variables", "VisualShaderNodeColorConstant", TTR("Color constant."), -1, -1));
  3367. add_options.push_back(AddOption("ColorUniform", "Color", "Variables", "VisualShaderNodeColorUniform", TTR("Color uniform."), -1, -1));
  3368. // CONDITIONAL
  3369. const String &compare_func_desc = TTR("Returns the boolean result of the %s comparison between two parameters.");
  3370. add_options.push_back(AddOption("Equal", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Equal (==)")), VisualShaderNodeCompare::FUNC_EQUAL, VisualShaderNode::PORT_TYPE_BOOLEAN));
  3371. add_options.push_back(AddOption("GreaterThan", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Greater Than (>)")), VisualShaderNodeCompare::FUNC_GREATER_THAN, VisualShaderNode::PORT_TYPE_BOOLEAN));
  3372. 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));
  3373. 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));
  3374. 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));
  3375. 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));
  3376. add_options.push_back(AddOption("LessThan", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Less Than (<)")), VisualShaderNodeCompare::FUNC_LESS_THAN, VisualShaderNode::PORT_TYPE_BOOLEAN));
  3377. 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));
  3378. add_options.push_back(AddOption("NotEqual", "Conditional", "Functions", "VisualShaderNodeCompare", vformat(compare_func_desc, TTR("Not Equal (!=)")), VisualShaderNodeCompare::FUNC_NOT_EQUAL, VisualShaderNode::PORT_TYPE_BOOLEAN));
  3379. add_options.push_back(AddOption("Switch", "Conditional", "Functions", "VisualShaderNodeSwitch", TTR("Returns an associated vector if the provided boolean value is true or false."), VisualShaderNodeSwitch::OP_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR));
  3380. add_options.push_back(AddOption("SwitchBool", "Conditional", "Functions", "VisualShaderNodeSwitch", TTR("Returns an associated boolean if the provided boolean value is true or false."), VisualShaderNodeSwitch::OP_TYPE_BOOLEAN, VisualShaderNode::PORT_TYPE_BOOLEAN));
  3381. add_options.push_back(AddOption("SwitchFloat", "Conditional", "Functions", "VisualShaderNodeSwitch", TTR("Returns an associated floating-point scalar if the provided boolean value is true or false."), VisualShaderNodeSwitch::OP_TYPE_FLOAT, VisualShaderNode::PORT_TYPE_SCALAR));
  3382. add_options.push_back(AddOption("SwitchInt", "Conditional", "Functions", "VisualShaderNodeSwitch", TTR("Returns an associated integer scalar if the provided boolean value is true or false."), VisualShaderNodeSwitch::OP_TYPE_INT, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3383. add_options.push_back(AddOption("SwitchTransform", "Conditional", "Functions", "VisualShaderNodeSwitch", TTR("Returns an associated transform if the provided boolean value is true or false."), VisualShaderNodeSwitch::OP_TYPE_TRANSFORM, VisualShaderNode::PORT_TYPE_TRANSFORM));
  3384. 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));
  3385. 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));
  3386. add_options.push_back(AddOption("BooleanConstant", "Conditional", "Variables", "VisualShaderNodeBooleanConstant", TTR("Boolean constant."), -1, VisualShaderNode::PORT_TYPE_BOOLEAN));
  3387. add_options.push_back(AddOption("BooleanUniform", "Conditional", "Variables", "VisualShaderNodeBooleanUniform", TTR("Boolean uniform."), -1, VisualShaderNode::PORT_TYPE_BOOLEAN));
  3388. // INPUT
  3389. const String input_param_shader_modes = TTR("'%s' input parameter for all shader modes.");
  3390. // SPATIAL-FOR-ALL
  3391. add_options.push_back(AddOption("Camera", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "camera"), "camera", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_SPATIAL));
  3392. 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));
  3393. 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));
  3394. add_options.push_back(AddOption("Normal", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "normal"), "normal", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_SPATIAL));
  3395. 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));
  3396. add_options.push_back(AddOption("Projection", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "projection"), "projection", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_SPATIAL));
  3397. add_options.push_back(AddOption("Time", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "time"), "time", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_SPATIAL));
  3398. 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));
  3399. add_options.push_back(AddOption("World", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "world"), "world", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_SPATIAL));
  3400. // CANVASITEM-FOR-ALL
  3401. 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));
  3402. 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));
  3403. 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));
  3404. 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));
  3405. 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));
  3406. // PARTICLES-FOR-ALL
  3407. add_options.push_back(AddOption("Active", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "active"), "active", VisualShaderNode::PORT_TYPE_BOOLEAN, -1, Shader::MODE_PARTICLES));
  3408. add_options.push_back(AddOption("Alpha", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "alpha"), "alpha", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_PARTICLES));
  3409. add_options.push_back(AddOption("AttractorForce", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "attractor_force"), "attractor_force", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_PARTICLES));
  3410. add_options.push_back(AddOption("Color", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "color"), "color", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_PARTICLES));
  3411. add_options.push_back(AddOption("Custom", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "custom"), "custom", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_PARTICLES));
  3412. add_options.push_back(AddOption("CustomAlpha", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "custom_alpha"), "custom_alpha", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_PARTICLES));
  3413. add_options.push_back(AddOption("Delta", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "delta"), "delta", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_PARTICLES));
  3414. add_options.push_back(AddOption("EmissionTransform", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "emission_transform"), "emission_transform", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_PARTICLES));
  3415. add_options.push_back(AddOption("Index", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "index"), "index", VisualShaderNode::PORT_TYPE_SCALAR_INT, -1, Shader::MODE_PARTICLES));
  3416. add_options.push_back(AddOption("LifeTime", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "lifetime"), "lifetime", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_PARTICLES));
  3417. add_options.push_back(AddOption("Restart", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "restart"), "restart", VisualShaderNode::PORT_TYPE_BOOLEAN, -1, Shader::MODE_PARTICLES));
  3418. add_options.push_back(AddOption("Time", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "time"), "time", VisualShaderNode::PORT_TYPE_SCALAR, -1, Shader::MODE_PARTICLES));
  3419. add_options.push_back(AddOption("Transform", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "transform"), "transform", VisualShaderNode::PORT_TYPE_TRANSFORM, -1, Shader::MODE_PARTICLES));
  3420. add_options.push_back(AddOption("Velocity", "Input", "All", "VisualShaderNodeInput", vformat(input_param_shader_modes, "velocity"), "velocity", VisualShaderNode::PORT_TYPE_VECTOR, -1, Shader::MODE_PARTICLES));
  3421. /////////////////
  3422. add_options.push_back(AddOption("Input", "Input", "Common", "VisualShaderNodeInput", TTR("Input parameter.")));
  3423. // SPATIAL INPUTS
  3424. const String input_param_for_vertex_and_fragment_shader_modes = TTR("'%s' input parameter for vertex and fragment shader modes.");
  3425. const String input_param_for_fragment_and_light_shader_modes = TTR("'%s' input parameter for fragment and light shader modes.");
  3426. const String input_param_for_fragment_shader_mode = TTR("'%s' input parameter for fragment shader mode.");
  3427. const String input_param_for_sky_shader_mode = TTR("'%s' input parameter for sky shader mode.");
  3428. const String input_param_for_light_shader_mode = TTR("'%s' input parameter for light shader mode.");
  3429. const String input_param_for_vertex_shader_mode = TTR("'%s' input parameter for vertex shader mode.");
  3430. const String input_param_for_start_shader_mode = TTR("'%s' input parameter for start shader mode.");
  3431. const String input_param_for_process_shader_mode = TTR("'%s' input parameter for process shader mode.");
  3432. const String input_param_for_collide_shader_mode = TTR("'%s' input parameter for collide shader mode.");
  3433. const String input_param_for_start_and_process_shader_mode = TTR("'%s' input parameter for start and process shader modes.");
  3434. const String input_param_for_process_and_collide_shader_mode = TTR("'%s' input parameter for process and collide shader modes.");
  3435. const String input_param_for_vertex_and_fragment_shader_mode = TTR("'%s' input parameter for vertex and fragment shader modes.");
  3436. add_options.push_back(AddOption("Alpha", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "alpha"), "alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3437. add_options.push_back(AddOption("Binormal", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "binormal"), "binormal", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3438. add_options.push_back(AddOption("Color", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "color"), "color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3439. add_options.push_back(AddOption("DepthTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "depth_texture"), "depth_texture", VisualShaderNode::PORT_TYPE_SAMPLER, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3440. add_options.push_back(AddOption("FragCoord", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "fragcoord"), "fragcoord", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3441. add_options.push_back(AddOption("FrontFacing", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "front_facing"), "front_facing", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3442. add_options.push_back(AddOption("PointCoord", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "point_coord"), "point_coord", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3443. add_options.push_back(AddOption("ScreenTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "screen_texture"), "screen_texture", VisualShaderNode::PORT_TYPE_SAMPLER, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3444. add_options.push_back(AddOption("ScreenUV", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "screen_uv"), "screen_uv", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3445. add_options.push_back(AddOption("Side", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "side"), "side", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3446. add_options.push_back(AddOption("Tangent", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "tangent"), "tangent", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3447. add_options.push_back(AddOption("UV", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "uv"), "uv", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3448. add_options.push_back(AddOption("UV2", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "uv2"), "uv2", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3449. add_options.push_back(AddOption("Vertex", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "vertex"), "vertex", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3450. add_options.push_back(AddOption("View", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "view"), "view", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_SPATIAL));
  3451. add_options.push_back(AddOption("Albedo", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "albedo"), "albedo", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3452. add_options.push_back(AddOption("Attenuation", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "attenuation"), "attenuation", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3453. add_options.push_back(AddOption("Diffuse", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "diffuse"), "diffuse", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3454. add_options.push_back(AddOption("FragCoord", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "fragcoord"), "fragcoord", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3455. add_options.push_back(AddOption("Light", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light"), "light", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3456. add_options.push_back(AddOption("LightColor", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_color"), "light_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3457. add_options.push_back(AddOption("Metallic", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "metallic"), "metallic", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3458. add_options.push_back(AddOption("Roughness", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "roughness"), "roughness", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3459. add_options.push_back(AddOption("Specular", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "specular"), "specular", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3460. add_options.push_back(AddOption("Backlight", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "backlight"), "backlight", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3461. add_options.push_back(AddOption("View", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "view"), "view", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3462. add_options.push_back(AddOption("Alpha", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "alpha"), "alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3463. add_options.push_back(AddOption("Binormal", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "binormal"), "binormal", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3464. add_options.push_back(AddOption("Color", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "color"), "color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3465. add_options.push_back(AddOption("ModelView", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "modelview"), "modelview", VisualShaderNode::PORT_TYPE_TRANSFORM, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3466. add_options.push_back(AddOption("PointSize", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "point_size"), "point_size", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3467. add_options.push_back(AddOption("Tangent", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_mode, "tangent"), "tangent", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3468. add_options.push_back(AddOption("UV", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "uv"), "uv", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3469. add_options.push_back(AddOption("UV2", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "uv2"), "uv2", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3470. add_options.push_back(AddOption("Vertex", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "vertex"), "vertex", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3471. // CANVASITEM INPUTS
  3472. add_options.push_back(AddOption("AtLightPass", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "at_light_pass"), "at_light_pass", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3473. add_options.push_back(AddOption("FragCoord", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "fragcoord"), "fragcoord", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3474. add_options.push_back(AddOption("NormalTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "normal_texture"), "normal_texture", VisualShaderNode::PORT_TYPE_SAMPLER, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3475. 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, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3476. 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, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3477. add_options.push_back(AddOption("ScreenTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "screen_texture"), "screen_texture", VisualShaderNode::PORT_TYPE_SAMPLER, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3478. 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, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3479. add_options.push_back(AddOption("SpecularShininess", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "specular_shininess"), "specular_shininess", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3480. add_options.push_back(AddOption("SpecularShininessAlpha", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "specular_shininess_alpha"), "specular_shininess_alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3481. add_options.push_back(AddOption("SpecularShininessTexture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_shader_mode, "specular_shininess_texture"), "specular_shininess_texture", VisualShaderNode::PORT_TYPE_SAMPLER, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3482. add_options.push_back(AddOption("Texture", "Input", "Fragment", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "texture"), "texture", VisualShaderNode::PORT_TYPE_SAMPLER, TYPE_FLAGS_FRAGMENT, Shader::MODE_CANVAS_ITEM));
  3483. add_options.push_back(AddOption("FragCoord", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "fragcoord"), "fragcoord", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3484. add_options.push_back(AddOption("Light", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light"), "light", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3485. add_options.push_back(AddOption("LightAlpha", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_alpha"), "light_alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3486. add_options.push_back(AddOption("LightColor", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_color"), "light_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3487. add_options.push_back(AddOption("LightColorAlpha", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_color_alpha"), "light_color_alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3488. add_options.push_back(AddOption("LightPosition", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_position"), "light_position", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3489. add_options.push_back(AddOption("LightVertex", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "light_vertex"), "light_vertex", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3490. add_options.push_back(AddOption("Normal", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "normal"), "normal", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3491. 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, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3492. 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, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3493. add_options.push_back(AddOption("ShadowAlpha", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "shadow_alpha"), "shadow_alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3494. add_options.push_back(AddOption("ShadowColor", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_light_shader_mode, "shadow_color"), "shadow_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3495. add_options.push_back(AddOption("SpecularShininess", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "specular_shininess"), "specular_shininess", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3496. add_options.push_back(AddOption("SpecularShininessAlpha", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "specular_shininess_alpha"), "specular_shininess_alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3497. add_options.push_back(AddOption("Texture", "Input", "Light", "VisualShaderNodeInput", vformat(input_param_for_fragment_and_light_shader_modes, "texture"), "texture", VisualShaderNode::PORT_TYPE_SAMPLER, TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3498. add_options.push_back(AddOption("AtLightPass", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_and_fragment_shader_modes, "at_light_pass"), "at_light_pass", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_VERTEX, Shader::MODE_CANVAS_ITEM));
  3499. add_options.push_back(AddOption("Canvas", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "canvas"), "canvas", VisualShaderNode::PORT_TYPE_TRANSFORM, TYPE_FLAGS_VERTEX, Shader::MODE_CANVAS_ITEM));
  3500. add_options.push_back(AddOption("PointSize", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "point_size"), "point_size", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_VERTEX, Shader::MODE_CANVAS_ITEM));
  3501. add_options.push_back(AddOption("Screen", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "screen"), "screen", VisualShaderNode::PORT_TYPE_TRANSFORM, TYPE_FLAGS_VERTEX, Shader::MODE_CANVAS_ITEM));
  3502. add_options.push_back(AddOption("Vertex", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "vertex"), "vertex", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_VERTEX, Shader::MODE_CANVAS_ITEM));
  3503. add_options.push_back(AddOption("World", "Input", "Vertex", "VisualShaderNodeInput", vformat(input_param_for_vertex_shader_mode, "world"), "world", VisualShaderNode::PORT_TYPE_TRANSFORM, TYPE_FLAGS_VERTEX, Shader::MODE_CANVAS_ITEM));
  3504. // SKY INPUTS
  3505. add_options.push_back(AddOption("AtCubeMapPass", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "at_cubemap_pass"), "at_cubemap_pass", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3506. add_options.push_back(AddOption("AtHalfResPass", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "at_half_res_pass"), "at_half_res_pass", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3507. add_options.push_back(AddOption("AtQuarterResPass", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "at_quarter_res_pass"), "at_quarter_res_pass", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3508. add_options.push_back(AddOption("EyeDir", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "eyedir"), "eyedir", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3509. add_options.push_back(AddOption("HalfResColor", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "half_res_color"), "half_res_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3510. add_options.push_back(AddOption("HalfResAlpha", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "half_res_alpha"), "half_res_alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3511. add_options.push_back(AddOption("Light0Color", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light0_color"), "light0_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3512. add_options.push_back(AddOption("Light0Direction", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light0_direction"), "light0_direction", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3513. add_options.push_back(AddOption("Light0Enabled", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light0_enabled"), "light0_enabled", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3514. add_options.push_back(AddOption("Light0Energy", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light0_energy"), "light0_energy", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3515. add_options.push_back(AddOption("Light1Color", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light1_color"), "light1_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3516. add_options.push_back(AddOption("Light1Direction", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light1_direction"), "light1_direction", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3517. add_options.push_back(AddOption("Light1Enabled", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light1_enabled"), "light1_enabled", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3518. add_options.push_back(AddOption("Light1Energy", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light1_energy"), "light1_energy", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3519. add_options.push_back(AddOption("Light2Color", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light2_color"), "light2_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3520. add_options.push_back(AddOption("Light2Direction", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light2_direction"), "light2_direction", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3521. add_options.push_back(AddOption("Light2Enabled", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light2_enabled"), "light2_enabled", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3522. add_options.push_back(AddOption("Light2Energy", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light2_energy"), "light2_energy", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3523. add_options.push_back(AddOption("Light3Color", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light3_color"), "light3_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3524. add_options.push_back(AddOption("Light3Direction", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light3_direction"), "light3_direction", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3525. add_options.push_back(AddOption("Light3Enabled", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light3_enabled"), "light3_enabled", VisualShaderNode::PORT_TYPE_BOOLEAN, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3526. add_options.push_back(AddOption("Light3Energy", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "light3_energy"), "light3_energy", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3527. add_options.push_back(AddOption("Position", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "position"), "position", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3528. add_options.push_back(AddOption("QuarterResColor", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "quarter_res_color"), "quarter_res_color", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3529. add_options.push_back(AddOption("QuarterResAlpha", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "quarter_res_alpha"), "quarter_res_alpha", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3530. add_options.push_back(AddOption("Radiance", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "radiance"), "radiance", VisualShaderNode::PORT_TYPE_SAMPLER, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3531. add_options.push_back(AddOption("ScreenUV", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "screen_uv"), "screen_uv", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3532. add_options.push_back(AddOption("SkyCoords", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "sky_coords"), "sky_coords", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3533. add_options.push_back(AddOption("Time", "Input", "Sky", "VisualShaderNodeInput", vformat(input_param_for_sky_shader_mode, "time"), "time", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_SKY, Shader::MODE_SKY));
  3534. // PARTICLES
  3535. add_options.push_back(AddOption("CollisionDepth", "Input", "Collide", "VisualShaderNodeInput", vformat(input_param_for_collide_shader_mode, "collision_depth"), "collision_depth", VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_COLLIDE, Shader::MODE_PARTICLES));
  3536. add_options.push_back(AddOption("CollisionNormal", "Input", "Collide", "VisualShaderNodeInput", vformat(input_param_for_collide_shader_mode, "collision_normal"), "collision_normal", VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_COLLIDE, Shader::MODE_PARTICLES));
  3537. add_options.push_back(AddOption("EmitParticle", "Particles", "", "VisualShaderNodeParticleEmit", "", -1, -1, -1, Shader::MODE_PARTICLES));
  3538. add_options.push_back(AddOption("ParticleAccelerator", "Particles", "", "VisualShaderNodeParticleAccelerator", "", -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_PROCESS, Shader::MODE_PARTICLES));
  3539. add_options.push_back(AddOption("ParticleRandomness", "Particles", "", "VisualShaderNodeParticleRandomness", "", -1, VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_EMIT | TYPE_FLAGS_PROCESS | TYPE_FLAGS_COLLIDE, Shader::MODE_PARTICLES));
  3540. add_options.push_back(AddOption("MultiplyByAxisAngle", "Particles", "Transform", "VisualShaderNodeParticleMultiplyByAxisAngle", "A node for help to multiply a position input vector by rotation using specific axis. Intended to work with emitters.", -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_EMIT | TYPE_FLAGS_PROCESS | TYPE_FLAGS_COLLIDE, Shader::MODE_PARTICLES));
  3541. add_options.push_back(AddOption("BoxEmitter", "Particles", "Emitters", "VisualShaderNodeParticleBoxEmitter", "", -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_EMIT, Shader::MODE_PARTICLES));
  3542. add_options.push_back(AddOption("RingEmitter", "Particles", "Emitters", "VisualShaderNodeParticleRingEmitter", "", -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_EMIT, Shader::MODE_PARTICLES));
  3543. add_options.push_back(AddOption("SphereEmitter", "Particles", "Emitters", "VisualShaderNodeParticleSphereEmitter", "", -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_EMIT, Shader::MODE_PARTICLES));
  3544. add_options.push_back(AddOption("ConeVelocity", "Particles", "Velocity", "VisualShaderNodeParticleConeVelocity", "", -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_EMIT, Shader::MODE_PARTICLES));
  3545. // SCALAR
  3546. add_options.push_back(AddOption("FloatFunc", "Scalar", "Common", "VisualShaderNodeFloatFunc", TTR("Float function."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  3547. add_options.push_back(AddOption("IntFunc", "Scalar", "Common", "VisualShaderNodeIntFunc", TTR("Integer function."), -1, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3548. add_options.push_back(AddOption("FloatOp", "Scalar", "Common", "VisualShaderNodeFloatOp", TTR("Float operator."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  3549. add_options.push_back(AddOption("IntOp", "Scalar", "Common", "VisualShaderNodeIntOp", TTR("Integer operator."), -1, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3550. //CONSTANTS
  3551. for (int i = 0; i < MAX_FLOAT_CONST_DEFS; i++) {
  3552. add_options.push_back(AddOption(float_constant_defs[i].name, "Scalar", "Constants", "VisualShaderNodeFloatConstant", float_constant_defs[i].desc, -1, VisualShaderNode::PORT_TYPE_SCALAR, -1, -1, float_constant_defs[i].value));
  3553. }
  3554. // FUNCTIONS
  3555. add_options.push_back(AddOption("Abs", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the absolute value of the parameter."), VisualShaderNodeFloatFunc::FUNC_ABS, VisualShaderNode::PORT_TYPE_SCALAR));
  3556. 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));
  3557. add_options.push_back(AddOption("ACos", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the arc-cosine of the parameter."), VisualShaderNodeFloatFunc::FUNC_ACOS, VisualShaderNode::PORT_TYPE_SCALAR));
  3558. 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));
  3559. add_options.push_back(AddOption("ASin", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the arc-sine of the parameter."), VisualShaderNodeFloatFunc::FUNC_ASIN, VisualShaderNode::PORT_TYPE_SCALAR));
  3560. 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));
  3561. add_options.push_back(AddOption("ATan", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the arc-tangent of the parameter."), VisualShaderNodeFloatFunc::FUNC_ATAN, VisualShaderNode::PORT_TYPE_SCALAR));
  3562. add_options.push_back(AddOption("ATan2", "Scalar", "Functions", "VisualShaderNodeFloatOp", TTR("Returns the arc-tangent of the parameters."), VisualShaderNodeFloatOp::OP_ATAN2, VisualShaderNode::PORT_TYPE_SCALAR));
  3563. 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));
  3564. 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));
  3565. add_options.push_back(AddOption("Clamp", "Scalar", "Functions", "VisualShaderNodeClamp", TTR("Constrains a value to lie between two further values."), VisualShaderNodeClamp::OP_TYPE_FLOAT, VisualShaderNode::PORT_TYPE_SCALAR));
  3566. add_options.push_back(AddOption("Clamp", "Scalar", "Functions", "VisualShaderNodeClamp", TTR("Constrains a value to lie between two further values."), VisualShaderNodeClamp::OP_TYPE_INT, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3567. add_options.push_back(AddOption("Cos", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the cosine of the parameter."), VisualShaderNodeFloatFunc::FUNC_COS, VisualShaderNode::PORT_TYPE_SCALAR));
  3568. add_options.push_back(AddOption("CosH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the hyperbolic cosine of the parameter."), VisualShaderNodeFloatFunc::FUNC_COSH, VisualShaderNode::PORT_TYPE_SCALAR));
  3569. add_options.push_back(AddOption("Degrees", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Converts a quantity in radians to degrees."), VisualShaderNodeFloatFunc::FUNC_DEGREES, VisualShaderNode::PORT_TYPE_SCALAR));
  3570. add_options.push_back(AddOption("Exp", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Base-e Exponential."), VisualShaderNodeFloatFunc::FUNC_EXP, VisualShaderNode::PORT_TYPE_SCALAR));
  3571. add_options.push_back(AddOption("Exp2", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Base-2 Exponential."), VisualShaderNodeFloatFunc::FUNC_EXP2, VisualShaderNode::PORT_TYPE_SCALAR));
  3572. 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));
  3573. add_options.push_back(AddOption("Fract", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Computes the fractional part of the argument."), VisualShaderNodeFloatFunc::FUNC_FRAC, VisualShaderNode::PORT_TYPE_SCALAR));
  3574. 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));
  3575. add_options.push_back(AddOption("Log", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Natural logarithm."), VisualShaderNodeFloatFunc::FUNC_LOG, VisualShaderNode::PORT_TYPE_SCALAR));
  3576. add_options.push_back(AddOption("Log2", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Base-2 logarithm."), VisualShaderNodeFloatFunc::FUNC_LOG2, VisualShaderNode::PORT_TYPE_SCALAR));
  3577. add_options.push_back(AddOption("Max", "Scalar", "Functions", "VisualShaderNodeFloatOp", TTR("Returns the greater of two values."), VisualShaderNodeFloatOp::OP_MAX, VisualShaderNode::PORT_TYPE_SCALAR));
  3578. add_options.push_back(AddOption("Min", "Scalar", "Functions", "VisualShaderNodeFloatOp", TTR("Returns the lesser of two values."), VisualShaderNodeFloatOp::OP_MIN, VisualShaderNode::PORT_TYPE_SCALAR));
  3579. add_options.push_back(AddOption("Mix", "Scalar", "Functions", "VisualShaderNodeMix", TTR("Linear interpolation between two scalars."), VisualShaderNodeMix::OP_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR));
  3580. add_options.push_back(AddOption("MultiplyAdd", "Scalar", "Functions", "VisualShaderNodeMultiplyAdd", TTR("Performs a fused multiply-add operation (a * b + c) on scalars."), VisualShaderNodeMultiplyAdd::OP_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR));
  3581. add_options.push_back(AddOption("Negate", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the opposite value of the parameter."), VisualShaderNodeFloatFunc::FUNC_NEGATE, VisualShaderNode::PORT_TYPE_SCALAR));
  3582. 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));
  3583. add_options.push_back(AddOption("OneMinus", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("1.0 - scalar"), VisualShaderNodeFloatFunc::FUNC_ONEMINUS, VisualShaderNode::PORT_TYPE_SCALAR));
  3584. 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));
  3585. add_options.push_back(AddOption("Radians", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Converts a quantity in degrees to radians."), VisualShaderNodeFloatFunc::FUNC_RADIANS, VisualShaderNode::PORT_TYPE_SCALAR));
  3586. add_options.push_back(AddOption("Reciprocal", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("1.0 / scalar"), VisualShaderNodeFloatFunc::FUNC_RECIPROCAL, VisualShaderNode::PORT_TYPE_SCALAR));
  3587. add_options.push_back(AddOption("Round", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Finds the nearest integer to the parameter."), VisualShaderNodeFloatFunc::FUNC_ROUND, VisualShaderNode::PORT_TYPE_SCALAR));
  3588. 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));
  3589. 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));
  3590. add_options.push_back(AddOption("Sign", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Extracts the sign of the parameter."), VisualShaderNodeFloatFunc::FUNC_SIGN, VisualShaderNode::PORT_TYPE_SCALAR));
  3591. add_options.push_back(AddOption("Sign", "Scalar", "Functions", "VisualShaderNodeIntFunc", TTR("Extracts the sign of the parameter."), VisualShaderNodeIntFunc::FUNC_SIGN, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3592. add_options.push_back(AddOption("Sin", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the sine of the parameter."), VisualShaderNodeFloatFunc::FUNC_SIN, VisualShaderNode::PORT_TYPE_SCALAR));
  3593. add_options.push_back(AddOption("SinH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the hyperbolic sine of the parameter."), VisualShaderNodeFloatFunc::FUNC_SINH, VisualShaderNode::PORT_TYPE_SCALAR));
  3594. add_options.push_back(AddOption("Sqrt", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the square root of the parameter."), VisualShaderNodeFloatFunc::FUNC_SQRT, VisualShaderNode::PORT_TYPE_SCALAR));
  3595. add_options.push_back(AddOption("SmoothStep", "Scalar", "Functions", "VisualShaderNodeSmoothStep", 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."), VisualShaderNodeSmoothStep::OP_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR));
  3596. add_options.push_back(AddOption("Step", "Scalar", "Functions", "VisualShaderNodeStep", TTR("Step function( scalar(edge), scalar(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge' and otherwise 1.0."), VisualShaderNodeStep::OP_TYPE_SCALAR, VisualShaderNode::PORT_TYPE_SCALAR));
  3597. add_options.push_back(AddOption("Tan", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the tangent of the parameter."), VisualShaderNodeFloatFunc::FUNC_TAN, VisualShaderNode::PORT_TYPE_SCALAR));
  3598. add_options.push_back(AddOption("TanH", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Returns the hyperbolic tangent of the parameter."), VisualShaderNodeFloatFunc::FUNC_TANH, VisualShaderNode::PORT_TYPE_SCALAR));
  3599. add_options.push_back(AddOption("Trunc", "Scalar", "Functions", "VisualShaderNodeFloatFunc", TTR("Finds the truncated value of the parameter."), VisualShaderNodeFloatFunc::FUNC_TRUNC, VisualShaderNode::PORT_TYPE_SCALAR));
  3600. add_options.push_back(AddOption("Add", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Sums two floating-point scalars."), VisualShaderNodeFloatOp::OP_ADD, VisualShaderNode::PORT_TYPE_SCALAR));
  3601. add_options.push_back(AddOption("Add", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Sums two integer scalars."), VisualShaderNodeIntOp::OP_ADD, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3602. add_options.push_back(AddOption("Divide", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Divides two floating-point scalars."), VisualShaderNodeFloatOp::OP_DIV, VisualShaderNode::PORT_TYPE_SCALAR));
  3603. add_options.push_back(AddOption("Divide", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Divides two integer scalars."), VisualShaderNodeIntOp::OP_DIV, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3604. add_options.push_back(AddOption("Multiply", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Multiplies two floating-point scalars."), VisualShaderNodeFloatOp::OP_MUL, VisualShaderNode::PORT_TYPE_SCALAR));
  3605. add_options.push_back(AddOption("Multiply", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Multiplies two integer scalars."), VisualShaderNodeIntOp::OP_MUL, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3606. 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));
  3607. 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));
  3608. add_options.push_back(AddOption("Subtract", "Scalar", "Operators", "VisualShaderNodeFloatOp", TTR("Subtracts two floating-point scalars."), VisualShaderNodeFloatOp::OP_SUB, VisualShaderNode::PORT_TYPE_SCALAR));
  3609. add_options.push_back(AddOption("Subtract", "Scalar", "Operators", "VisualShaderNodeIntOp", TTR("Subtracts two integer scalars."), VisualShaderNodeIntOp::OP_SUB, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3610. add_options.push_back(AddOption("FloatConstant", "Scalar", "Variables", "VisualShaderNodeFloatConstant", TTR("Scalar floating-point constant."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  3611. add_options.push_back(AddOption("IntConstant", "Scalar", "Variables", "VisualShaderNodeIntConstant", TTR("Scalar integer constant."), -1, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3612. add_options.push_back(AddOption("FloatUniform", "Scalar", "Variables", "VisualShaderNodeFloatUniform", TTR("Scalar floating-point uniform."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  3613. add_options.push_back(AddOption("IntUniform", "Scalar", "Variables", "VisualShaderNodeIntUniform", TTR("Scalar integer uniform."), -1, VisualShaderNode::PORT_TYPE_SCALAR_INT));
  3614. // SDF
  3615. {
  3616. add_options.push_back(AddOption("ScreenUVToSDF", "SDF", "", "VisualShaderNodeScreenUVToSDF", TTR("Converts screen UV to a SDF."), -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3617. add_options.push_back(AddOption("SDFRaymarch", "SDF", "", "VisualShaderNodeSDFRaymarch", TTR("Casts a ray against the screen SDF and returns the distance travelled."), -1, -1, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3618. add_options.push_back(AddOption("SDFToScreenUV", "SDF", "", "VisualShaderNodeSDFToScreenUV", TTR("Converts a SDF to screen UV."), -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3619. add_options.push_back(AddOption("TextureSDF", "SDF", "", "VisualShaderNodeTextureSDF", TTR("Performs a SDF texture lookup."), -1, VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3620. add_options.push_back(AddOption("TextureSDFNormal", "SDF", "", "VisualShaderNodeTextureSDFNormal", TTR("Performs a SDF normal texture lookup."), -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, Shader::MODE_CANVAS_ITEM));
  3621. }
  3622. // TEXTURES
  3623. add_options.push_back(AddOption("UVFunc", "Textures", "Common", "VisualShaderNodeUVFunc", TTR("Function to be applied on texture coordinates."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3624. cubemap_node_option_idx = add_options.size();
  3625. add_options.push_back(AddOption("CubeMap", "Textures", "Functions", "VisualShaderNodeCubemap", TTR("Perform the cubic texture lookup."), -1, -1));
  3626. curve_node_option_idx = add_options.size();
  3627. add_options.push_back(AddOption("CurveTexture", "Textures", "Functions", "VisualShaderNodeCurveTexture", TTR("Perform the curve texture lookup."), -1, -1));
  3628. texture2d_node_option_idx = add_options.size();
  3629. add_options.push_back(AddOption("Texture2D", "Textures", "Functions", "VisualShaderNodeTexture", TTR("Perform the 2D texture lookup."), -1, -1));
  3630. texture2d_array_node_option_idx = add_options.size();
  3631. add_options.push_back(AddOption("Texture2DArray", "Textures", "Functions", "VisualShaderNodeTexture2DArray", TTR("Perform the 2D-array texture lookup."), -1, -1, -1, -1, -1));
  3632. texture3d_node_option_idx = add_options.size();
  3633. add_options.push_back(AddOption("Texture3D", "Textures", "Functions", "VisualShaderNodeTexture3D", TTR("Perform the 3D texture lookup."), -1, -1));
  3634. add_options.push_back(AddOption("UVPanning", "Textures", "Functions", "VisualShaderNodeUVFunc", TTR("Apply panning function on texture coordinates."), VisualShaderNodeUVFunc::FUNC_PANNING, VisualShaderNode::PORT_TYPE_VECTOR));
  3635. add_options.push_back(AddOption("UVScaling", "Textures", "Functions", "VisualShaderNodeUVFunc", TTR("Apply scaling function on texture coordinates."), VisualShaderNodeUVFunc::FUNC_SCALING, VisualShaderNode::PORT_TYPE_VECTOR));
  3636. add_options.push_back(AddOption("CubeMapUniform", "Textures", "Variables", "VisualShaderNodeCubemapUniform", TTR("Cubic texture uniform lookup."), -1, -1));
  3637. add_options.push_back(AddOption("TextureUniform", "Textures", "Variables", "VisualShaderNodeTextureUniform", TTR("2D texture uniform lookup."), -1, -1));
  3638. add_options.push_back(AddOption("TextureUniformTriplanar", "Textures", "Variables", "VisualShaderNodeTextureUniformTriplanar", TTR("2D texture uniform lookup with triplanar."), -1, -1, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, Shader::MODE_SPATIAL));
  3639. add_options.push_back(AddOption("Texture2DArrayUniform", "Textures", "Variables", "VisualShaderNodeTexture2DArrayUniform", TTR("2D array of textures uniform lookup."), -1, -1, -1, -1, -1));
  3640. add_options.push_back(AddOption("Texture3DUniform", "Textures", "Variables", "VisualShaderNodeTexture3DUniform", TTR("3D texture uniform lookup."), -1, -1, -1, -1, -1));
  3641. // TRANSFORM
  3642. add_options.push_back(AddOption("TransformFunc", "Transform", "Common", "VisualShaderNodeTransformFunc", TTR("Transform function."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  3643. 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));
  3644. add_options.push_back(AddOption("TransformCompose", "Transform", "Composition", "VisualShaderNodeTransformCompose", TTR("Composes transform from four vectors."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  3645. add_options.push_back(AddOption("TransformDecompose", "Transform", "Composition", "VisualShaderNodeTransformDecompose", TTR("Decomposes transform to four vectors.")));
  3646. add_options.push_back(AddOption("Determinant", "Transform", "Functions", "VisualShaderNodeDeterminant", TTR("Calculates the determinant of a transform."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  3647. add_options.push_back(AddOption("GetBillboardMatrix", "Transform", "Functions", "VisualShaderNodeBillboard", TTR("Calculates how the object should face the camera to be applied on Model View Matrix output port for 3D objects."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM, TYPE_FLAGS_VERTEX, Shader::MODE_SPATIAL));
  3648. add_options.push_back(AddOption("Inverse", "Transform", "Functions", "VisualShaderNodeTransformFunc", TTR("Calculates the inverse of a transform."), VisualShaderNodeTransformFunc::FUNC_INVERSE, VisualShaderNode::PORT_TYPE_TRANSFORM));
  3649. add_options.push_back(AddOption("Transpose", "Transform", "Functions", "VisualShaderNodeTransformFunc", TTR("Calculates the transpose of a transform."), VisualShaderNodeTransformFunc::FUNC_TRANSPOSE, VisualShaderNode::PORT_TYPE_TRANSFORM));
  3650. add_options.push_back(AddOption("TransformMult", "Transform", "Operators", "VisualShaderNodeTransformMult", TTR("Multiplies transform by transform."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  3651. add_options.push_back(AddOption("TransformVectorMult", "Transform", "Operators", "VisualShaderNodeTransformVecMult", TTR("Multiplies vector by transform."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3652. add_options.push_back(AddOption("TransformConstant", "Transform", "Variables", "VisualShaderNodeTransformConstant", TTR("Transform constant."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  3653. add_options.push_back(AddOption("TransformUniform", "Transform", "Variables", "VisualShaderNodeTransformUniform", TTR("Transform uniform."), -1, VisualShaderNode::PORT_TYPE_TRANSFORM));
  3654. // VECTOR
  3655. add_options.push_back(AddOption("VectorFunc", "Vector", "Common", "VisualShaderNodeVectorFunc", TTR("Vector function."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3656. add_options.push_back(AddOption("VectorOp", "Vector", "Common", "VisualShaderNodeVectorOp", TTR("Vector operator."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3657. add_options.push_back(AddOption("VectorCompose", "Vector", "Composition", "VisualShaderNodeVectorCompose", TTR("Composes vector from three scalars."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3658. add_options.push_back(AddOption("VectorDecompose", "Vector", "Composition", "VisualShaderNodeVectorDecompose", TTR("Decomposes vector to three scalars.")));
  3659. add_options.push_back(AddOption("Abs", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the absolute value of the parameter."), VisualShaderNodeVectorFunc::FUNC_ABS, VisualShaderNode::PORT_TYPE_VECTOR));
  3660. add_options.push_back(AddOption("ACos", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the arc-cosine of the parameter."), VisualShaderNodeVectorFunc::FUNC_ACOS, VisualShaderNode::PORT_TYPE_VECTOR));
  3661. 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));
  3662. add_options.push_back(AddOption("ASin", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the arc-sine of the parameter."), VisualShaderNodeVectorFunc::FUNC_ASIN, VisualShaderNode::PORT_TYPE_VECTOR));
  3663. 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));
  3664. add_options.push_back(AddOption("ATan", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the arc-tangent of the parameter."), VisualShaderNodeVectorFunc::FUNC_ATAN, VisualShaderNode::PORT_TYPE_VECTOR));
  3665. add_options.push_back(AddOption("ATan2", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Returns the arc-tangent of the parameters."), VisualShaderNodeVectorOp::OP_ATAN2, VisualShaderNode::PORT_TYPE_VECTOR));
  3666. 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));
  3667. 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));
  3668. add_options.push_back(AddOption("Clamp", "Vector", "Functions", "VisualShaderNodeClamp", TTR("Constrains a value to lie between two further values."), VisualShaderNodeClamp::OP_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR));
  3669. add_options.push_back(AddOption("Cos", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the cosine of the parameter."), VisualShaderNodeVectorFunc::FUNC_COS, VisualShaderNode::PORT_TYPE_VECTOR));
  3670. add_options.push_back(AddOption("CosH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the hyperbolic cosine of the parameter."), VisualShaderNodeVectorFunc::FUNC_COSH, VisualShaderNode::PORT_TYPE_VECTOR));
  3671. add_options.push_back(AddOption("Cross", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Calculates the cross product of two vectors."), VisualShaderNodeVectorOp::OP_CROSS, VisualShaderNode::PORT_TYPE_VECTOR));
  3672. add_options.push_back(AddOption("Degrees", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Converts a quantity in radians to degrees."), VisualShaderNodeVectorFunc::FUNC_DEGREES, VisualShaderNode::PORT_TYPE_VECTOR));
  3673. add_options.push_back(AddOption("Distance", "Vector", "Functions", "VisualShaderNodeVectorDistance", TTR("Returns the distance between two points."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  3674. add_options.push_back(AddOption("Dot", "Vector", "Functions", "VisualShaderNodeDotProduct", TTR("Calculates the dot product of two vectors."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  3675. add_options.push_back(AddOption("Exp", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Base-e Exponential."), VisualShaderNodeVectorFunc::FUNC_EXP, VisualShaderNode::PORT_TYPE_VECTOR));
  3676. add_options.push_back(AddOption("Exp2", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Base-2 Exponential."), VisualShaderNodeVectorFunc::FUNC_EXP2, VisualShaderNode::PORT_TYPE_VECTOR));
  3677. 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));
  3678. 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));
  3679. add_options.push_back(AddOption("Fract", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Computes the fractional part of the argument."), VisualShaderNodeVectorFunc::FUNC_FRAC, VisualShaderNode::PORT_TYPE_VECTOR));
  3680. 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));
  3681. add_options.push_back(AddOption("Length", "Vector", "Functions", "VisualShaderNodeVectorLen", TTR("Calculates the length of a vector."), -1, VisualShaderNode::PORT_TYPE_SCALAR));
  3682. add_options.push_back(AddOption("Log", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Natural logarithm."), VisualShaderNodeVectorFunc::FUNC_LOG, VisualShaderNode::PORT_TYPE_VECTOR));
  3683. add_options.push_back(AddOption("Log2", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Base-2 logarithm."), VisualShaderNodeVectorFunc::FUNC_LOG2, VisualShaderNode::PORT_TYPE_VECTOR));
  3684. add_options.push_back(AddOption("Max", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Returns the greater of two values."), VisualShaderNodeVectorOp::OP_MAX, VisualShaderNode::PORT_TYPE_VECTOR));
  3685. add_options.push_back(AddOption("Min", "Vector", "Functions", "VisualShaderNodeVectorOp", TTR("Returns the lesser of two values."), VisualShaderNodeVectorOp::OP_MIN, VisualShaderNode::PORT_TYPE_VECTOR));
  3686. add_options.push_back(AddOption("Mix", "Vector", "Functions", "VisualShaderNodeMix", TTR("Linear interpolation between two vectors."), VisualShaderNodeMix::OP_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR));
  3687. add_options.push_back(AddOption("MixS", "Vector", "Functions", "VisualShaderNodeMix", TTR("Linear interpolation between two vectors using scalar."), VisualShaderNodeMix::OP_TYPE_VECTOR_SCALAR, VisualShaderNode::PORT_TYPE_VECTOR));
  3688. add_options.push_back(AddOption("MultiplyAdd", "Vector", "Functions", "VisualShaderNodeMultiplyAdd", TTR("Performs a fused multiply-add operation (a * b + c) on vectors."), VisualShaderNodeMultiplyAdd::OP_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR));
  3689. add_options.push_back(AddOption("Negate", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the opposite value of the parameter."), VisualShaderNodeVectorFunc::FUNC_NEGATE, VisualShaderNode::PORT_TYPE_VECTOR));
  3690. add_options.push_back(AddOption("Normalize", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Calculates the normalize product of vector."), VisualShaderNodeVectorFunc::FUNC_NORMALIZE, VisualShaderNode::PORT_TYPE_VECTOR));
  3691. add_options.push_back(AddOption("OneMinus", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("1.0 - vector"), VisualShaderNodeVectorFunc::FUNC_ONEMINUS, VisualShaderNode::PORT_TYPE_VECTOR));
  3692. 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));
  3693. add_options.push_back(AddOption("Radians", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Converts a quantity in degrees to radians."), VisualShaderNodeVectorFunc::FUNC_RADIANS, VisualShaderNode::PORT_TYPE_VECTOR));
  3694. add_options.push_back(AddOption("Reciprocal", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("1.0 / vector"), VisualShaderNodeVectorFunc::FUNC_RECIPROCAL, VisualShaderNode::PORT_TYPE_VECTOR));
  3695. 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));
  3696. 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));
  3697. add_options.push_back(AddOption("Round", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Finds the nearest integer to the parameter."), VisualShaderNodeVectorFunc::FUNC_ROUND, VisualShaderNode::PORT_TYPE_VECTOR));
  3698. 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));
  3699. 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));
  3700. add_options.push_back(AddOption("Sign", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Extracts the sign of the parameter."), VisualShaderNodeVectorFunc::FUNC_SIGN, VisualShaderNode::PORT_TYPE_VECTOR));
  3701. add_options.push_back(AddOption("Sin", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the sine of the parameter."), VisualShaderNodeVectorFunc::FUNC_SIN, VisualShaderNode::PORT_TYPE_VECTOR));
  3702. add_options.push_back(AddOption("SinH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the hyperbolic sine of the parameter."), VisualShaderNodeVectorFunc::FUNC_SINH, VisualShaderNode::PORT_TYPE_VECTOR));
  3703. add_options.push_back(AddOption("Sqrt", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the square root of the parameter."), VisualShaderNodeVectorFunc::FUNC_SQRT, VisualShaderNode::PORT_TYPE_VECTOR));
  3704. add_options.push_back(AddOption("SmoothStep", "Vector", "Functions", "VisualShaderNodeSmoothStep", 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."), VisualShaderNodeSmoothStep::OP_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR));
  3705. add_options.push_back(AddOption("SmoothStepS", "Vector", "Functions", "VisualShaderNodeSmoothStep", 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."), VisualShaderNodeSmoothStep::OP_TYPE_VECTOR_SCALAR, VisualShaderNode::PORT_TYPE_VECTOR));
  3706. add_options.push_back(AddOption("Step", "Vector", "Functions", "VisualShaderNodeStep", TTR("Step function( vector(edge), vector(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge' and otherwise 1.0."), VisualShaderNodeStep::OP_TYPE_VECTOR, VisualShaderNode::PORT_TYPE_VECTOR));
  3707. add_options.push_back(AddOption("StepS", "Vector", "Functions", "VisualShaderNodeStep", TTR("Step function( scalar(edge), vector(x) ).\n\nReturns 0.0 if 'x' is smaller than 'edge' and otherwise 1.0."), VisualShaderNodeStep::OP_TYPE_VECTOR_SCALAR, VisualShaderNode::PORT_TYPE_VECTOR));
  3708. add_options.push_back(AddOption("Tan", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the tangent of the parameter."), VisualShaderNodeVectorFunc::FUNC_TAN, VisualShaderNode::PORT_TYPE_VECTOR));
  3709. add_options.push_back(AddOption("TanH", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Returns the hyperbolic tangent of the parameter."), VisualShaderNodeVectorFunc::FUNC_TANH, VisualShaderNode::PORT_TYPE_VECTOR));
  3710. add_options.push_back(AddOption("Trunc", "Vector", "Functions", "VisualShaderNodeVectorFunc", TTR("Finds the truncated value of the parameter."), VisualShaderNodeVectorFunc::FUNC_TRUNC, VisualShaderNode::PORT_TYPE_VECTOR));
  3711. add_options.push_back(AddOption("Add", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Adds vector to vector."), VisualShaderNodeVectorOp::OP_ADD, VisualShaderNode::PORT_TYPE_VECTOR));
  3712. add_options.push_back(AddOption("Divide", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Divides vector by vector."), VisualShaderNodeVectorOp::OP_DIV, VisualShaderNode::PORT_TYPE_VECTOR));
  3713. add_options.push_back(AddOption("Multiply", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Multiplies vector by vector."), VisualShaderNodeVectorOp::OP_MUL, VisualShaderNode::PORT_TYPE_VECTOR));
  3714. add_options.push_back(AddOption("Remainder", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Returns the remainder of the two vectors."), VisualShaderNodeVectorOp::OP_MOD, VisualShaderNode::PORT_TYPE_VECTOR));
  3715. add_options.push_back(AddOption("Subtract", "Vector", "Operators", "VisualShaderNodeVectorOp", TTR("Subtracts vector from vector."), VisualShaderNodeVectorOp::OP_SUB, VisualShaderNode::PORT_TYPE_VECTOR));
  3716. add_options.push_back(AddOption("VectorConstant", "Vector", "Variables", "VisualShaderNodeVec3Constant", TTR("Vector constant."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3717. add_options.push_back(AddOption("VectorUniform", "Vector", "Variables", "VisualShaderNodeVec3Uniform", TTR("Vector uniform."), -1, VisualShaderNode::PORT_TYPE_VECTOR));
  3718. // SPECIAL
  3719. add_options.push_back(AddOption("Comment", "Special", "", "VisualShaderNodeComment", TTR("A rectangular area with a description string for better graph organization.")));
  3720. 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.")));
  3721. 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));
  3722. 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.")));
  3723. add_options.push_back(AddOption("UniformRef", "Special", "", "VisualShaderNodeUniformRef", TTR("A reference to an existing uniform.")));
  3724. add_options.push_back(AddOption("ScalarDerivativeFunc", "Special", "Common", "VisualShaderNodeScalarDerivativeFunc", TTR("(Fragment/Light mode only) Scalar derivative function."), -1, VisualShaderNode::PORT_TYPE_SCALAR, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, -1, -1, true));
  3725. add_options.push_back(AddOption("VectorDerivativeFunc", "Special", "Common", "VisualShaderNodeVectorDerivativeFunc", TTR("(Fragment/Light mode only) Vector derivative function."), -1, VisualShaderNode::PORT_TYPE_VECTOR, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, -1, -1, true));
  3726. 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, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, -1, -1, true));
  3727. 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, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, -1, -1, true));
  3728. 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, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, -1, -1, true));
  3729. 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, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, -1, -1, true));
  3730. 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, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, -1, -1, true));
  3731. 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, TYPE_FLAGS_FRAGMENT | TYPE_FLAGS_LIGHT, -1, -1, true));
  3732. custom_node_option_idx = add_options.size();
  3733. /////////////////////////////////////////////////////////////////////
  3734. _update_options_menu();
  3735. undo_redo = EditorNode::get_singleton()->get_undo_redo();
  3736. Ref<VisualShaderNodePluginDefault> default_plugin;
  3737. default_plugin.instantiate();
  3738. add_plugin(default_plugin);
  3739. graph_plugin.instantiate();
  3740. property_editor = memnew(CustomPropertyEditor);
  3741. add_child(property_editor);
  3742. property_editor->connect("variant_changed", callable_mp(this, &VisualShaderEditor::_port_edited));
  3743. }
  3744. /////////////////
  3745. void VisualShaderEditorPlugin::edit(Object *p_object) {
  3746. visual_shader_editor->edit(Object::cast_to<VisualShader>(p_object));
  3747. }
  3748. bool VisualShaderEditorPlugin::handles(Object *p_object) const {
  3749. return p_object->is_class("VisualShader");
  3750. }
  3751. void VisualShaderEditorPlugin::make_visible(bool p_visible) {
  3752. if (p_visible) {
  3753. //editor->hide_animation_player_editors();
  3754. //editor->animation_panel_make_visible(true);
  3755. button->show();
  3756. editor->make_bottom_panel_item_visible(visual_shader_editor);
  3757. visual_shader_editor->update_custom_nodes();
  3758. visual_shader_editor->set_process_input(true);
  3759. //visual_shader_editor->set_process(true);
  3760. } else {
  3761. if (visual_shader_editor->is_visible_in_tree()) {
  3762. editor->hide_bottom_panel();
  3763. }
  3764. button->hide();
  3765. visual_shader_editor->set_process_input(false);
  3766. //visual_shader_editor->set_process(false);
  3767. }
  3768. }
  3769. VisualShaderEditorPlugin::VisualShaderEditorPlugin(EditorNode *p_node) {
  3770. editor = p_node;
  3771. visual_shader_editor = memnew(VisualShaderEditor);
  3772. visual_shader_editor->set_custom_minimum_size(Size2(0, 300) * EDSCALE);
  3773. button = editor->add_bottom_panel_item(TTR("VisualShader"), visual_shader_editor);
  3774. button->hide();
  3775. }
  3776. VisualShaderEditorPlugin::~VisualShaderEditorPlugin() {
  3777. }
  3778. ////////////////
  3779. class VisualShaderNodePluginInputEditor : public OptionButton {
  3780. GDCLASS(VisualShaderNodePluginInputEditor, OptionButton);
  3781. Ref<VisualShaderNodeInput> input;
  3782. public:
  3783. void _notification(int p_what) {
  3784. if (p_what == NOTIFICATION_READY) {
  3785. connect("item_selected", callable_mp(this, &VisualShaderNodePluginInputEditor::_item_selected));
  3786. }
  3787. }
  3788. void _item_selected(int p_item) {
  3789. VisualShaderEditor::get_singleton()->call_deferred("_input_select_item", input, get_item_text(p_item));
  3790. }
  3791. void setup(const Ref<VisualShaderNodeInput> &p_input) {
  3792. input = p_input;
  3793. Ref<Texture2D> type_icon[6] = {
  3794. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("float", "EditorIcons"),
  3795. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("int", "EditorIcons"),
  3796. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Vector3", "EditorIcons"),
  3797. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("bool", "EditorIcons"),
  3798. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Transform3D", "EditorIcons"),
  3799. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("ImageTexture", "EditorIcons"),
  3800. };
  3801. add_item("[None]");
  3802. int to_select = -1;
  3803. for (int i = 0; i < input->get_input_index_count(); i++) {
  3804. if (input->get_input_name() == input->get_input_index_name(i)) {
  3805. to_select = i + 1;
  3806. }
  3807. add_icon_item(type_icon[input->get_input_index_type(i)], input->get_input_index_name(i));
  3808. }
  3809. if (to_select >= 0) {
  3810. select(to_select);
  3811. }
  3812. }
  3813. };
  3814. ////////////////
  3815. class VisualShaderNodePluginUniformRefEditor : public OptionButton {
  3816. GDCLASS(VisualShaderNodePluginUniformRefEditor, OptionButton);
  3817. Ref<VisualShaderNodeUniformRef> uniform_ref;
  3818. public:
  3819. void _notification(int p_what) {
  3820. if (p_what == NOTIFICATION_READY) {
  3821. connect("item_selected", callable_mp(this, &VisualShaderNodePluginUniformRefEditor::_item_selected));
  3822. }
  3823. }
  3824. void _item_selected(int p_item) {
  3825. VisualShaderEditor::get_singleton()->call_deferred("_uniform_select_item", uniform_ref, get_item_text(p_item));
  3826. }
  3827. void setup(const Ref<VisualShaderNodeUniformRef> &p_uniform_ref) {
  3828. uniform_ref = p_uniform_ref;
  3829. Ref<Texture2D> type_icon[7] = {
  3830. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("float", "EditorIcons"),
  3831. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("int", "EditorIcons"),
  3832. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("bool", "EditorIcons"),
  3833. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Vector3", "EditorIcons"),
  3834. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Transform3D", "EditorIcons"),
  3835. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("Color", "EditorIcons"),
  3836. EditorNode::get_singleton()->get_gui_base()->get_theme_icon("ImageTexture", "EditorIcons"),
  3837. };
  3838. add_item("[None]");
  3839. int to_select = -1;
  3840. for (int i = 0; i < p_uniform_ref->get_uniforms_count(); i++) {
  3841. if (p_uniform_ref->get_uniform_name() == p_uniform_ref->get_uniform_name_by_index(i)) {
  3842. to_select = i + 1;
  3843. }
  3844. add_icon_item(type_icon[p_uniform_ref->get_uniform_type_by_index(i)], p_uniform_ref->get_uniform_name_by_index(i));
  3845. }
  3846. if (to_select >= 0) {
  3847. select(to_select);
  3848. }
  3849. }
  3850. };
  3851. ////////////////
  3852. class VisualShaderNodePluginDefaultEditor : public VBoxContainer {
  3853. GDCLASS(VisualShaderNodePluginDefaultEditor, VBoxContainer);
  3854. Ref<Resource> parent_resource;
  3855. int node_id;
  3856. VisualShader::Type shader_type;
  3857. public:
  3858. void _property_changed(const String &p_property, const Variant &p_value, const String &p_field = "", bool p_changing = false) {
  3859. if (p_changing) {
  3860. return;
  3861. }
  3862. UndoRedo *undo_redo = EditorNode::get_singleton()->get_undo_redo();
  3863. updating = true;
  3864. undo_redo->create_action(TTR("Edit Visual Property") + ": " + p_property, UndoRedo::MERGE_ENDS);
  3865. undo_redo->add_do_property(node.ptr(), p_property, p_value);
  3866. undo_redo->add_undo_property(node.ptr(), p_property, node->get(p_property));
  3867. if (p_value.get_type() == Variant::OBJECT) {
  3868. RES prev_res = node->get(p_property);
  3869. RES curr_res = p_value;
  3870. if (curr_res.is_null()) {
  3871. undo_redo->add_do_method(this, "_open_inspector", (RES)parent_resource.ptr());
  3872. } else {
  3873. undo_redo->add_do_method(this, "_open_inspector", (RES)curr_res.ptr());
  3874. }
  3875. if (!prev_res.is_null()) {
  3876. undo_redo->add_undo_method(this, "_open_inspector", (RES)prev_res.ptr());
  3877. } else {
  3878. undo_redo->add_undo_method(this, "_open_inspector", (RES)parent_resource.ptr());
  3879. }
  3880. }
  3881. if (p_property != "constant") {
  3882. undo_redo->add_do_method(VisualShaderEditor::get_singleton()->get_graph_plugin(), "update_node_deferred", shader_type, node_id);
  3883. undo_redo->add_undo_method(VisualShaderEditor::get_singleton()->get_graph_plugin(), "update_node_deferred", shader_type, node_id);
  3884. } else {
  3885. undo_redo->add_do_method(VisualShaderEditor::get_singleton()->get_graph_plugin(), "update_constant", shader_type, node_id);
  3886. undo_redo->add_undo_method(VisualShaderEditor::get_singleton()->get_graph_plugin(), "update_constant", shader_type, node_id);
  3887. }
  3888. undo_redo->commit_action();
  3889. updating = false;
  3890. }
  3891. void _node_changed() {
  3892. if (updating) {
  3893. return;
  3894. }
  3895. for (int i = 0; i < properties.size(); i++) {
  3896. properties[i]->update_property();
  3897. }
  3898. }
  3899. void _resource_selected(const String &p_path, RES p_resource) {
  3900. _open_inspector(p_resource);
  3901. }
  3902. void _open_inspector(RES p_resource) {
  3903. EditorNode::get_singleton()->get_inspector()->edit(p_resource.ptr());
  3904. }
  3905. bool updating;
  3906. Ref<VisualShaderNode> node;
  3907. Vector<EditorProperty *> properties;
  3908. Vector<Label *> prop_names;
  3909. void _show_prop_names(bool p_show) {
  3910. for (int i = 0; i < prop_names.size(); i++) {
  3911. prop_names[i]->set_visible(p_show);
  3912. }
  3913. }
  3914. void setup(Ref<Resource> p_parent_resource, Vector<EditorProperty *> p_properties, const Vector<StringName> &p_names, Ref<VisualShaderNode> p_node) {
  3915. parent_resource = p_parent_resource;
  3916. updating = false;
  3917. node = p_node;
  3918. properties = p_properties;
  3919. node_id = (int)p_node->get_meta("id");
  3920. shader_type = VisualShader::Type((int)p_node->get_meta("shader_type"));
  3921. for (int i = 0; i < p_properties.size(); i++) {
  3922. HBoxContainer *hbox = memnew(HBoxContainer);
  3923. hbox->set_h_size_flags(SIZE_EXPAND_FILL);
  3924. add_child(hbox);
  3925. Label *prop_name = memnew(Label);
  3926. String prop_name_str = p_names[i];
  3927. prop_name_str = prop_name_str.capitalize() + ":";
  3928. prop_name->set_text(prop_name_str);
  3929. prop_name->set_visible(false);
  3930. hbox->add_child(prop_name);
  3931. prop_names.push_back(prop_name);
  3932. p_properties[i]->set_h_size_flags(SIZE_EXPAND_FILL);
  3933. hbox->add_child(p_properties[i]);
  3934. bool res_prop = Object::cast_to<EditorPropertyResource>(p_properties[i]);
  3935. if (res_prop) {
  3936. p_properties[i]->connect("resource_selected", callable_mp(this, &VisualShaderNodePluginDefaultEditor::_resource_selected));
  3937. }
  3938. properties[i]->connect("property_changed", callable_mp(this, &VisualShaderNodePluginDefaultEditor::_property_changed));
  3939. properties[i]->set_object_and_property(node.ptr(), p_names[i]);
  3940. properties[i]->update_property();
  3941. properties[i]->set_name_split_ratio(0);
  3942. }
  3943. node->connect("changed", callable_mp(this, &VisualShaderNodePluginDefaultEditor::_node_changed));
  3944. }
  3945. static void _bind_methods() {
  3946. ClassDB::bind_method("_open_inspector", &VisualShaderNodePluginDefaultEditor::_open_inspector); // Used by UndoRedo.
  3947. ClassDB::bind_method("_show_prop_names", &VisualShaderNodePluginDefaultEditor::_show_prop_names); // Used with call_deferred.
  3948. }
  3949. };
  3950. Control *VisualShaderNodePluginDefault::create_editor(const Ref<Resource> &p_parent_resource, const Ref<VisualShaderNode> &p_node) {
  3951. if (p_node->is_class("VisualShaderNodeUniformRef")) {
  3952. //create input
  3953. VisualShaderNodePluginUniformRefEditor *uniform_editor = memnew(VisualShaderNodePluginUniformRefEditor);
  3954. uniform_editor->setup(p_node);
  3955. return uniform_editor;
  3956. }
  3957. if (p_node->is_class("VisualShaderNodeInput")) {
  3958. //create input
  3959. VisualShaderNodePluginInputEditor *input_editor = memnew(VisualShaderNodePluginInputEditor);
  3960. input_editor->setup(p_node);
  3961. return input_editor;
  3962. }
  3963. Vector<StringName> properties = p_node->get_editable_properties();
  3964. if (properties.size() == 0) {
  3965. return nullptr;
  3966. }
  3967. List<PropertyInfo> props;
  3968. p_node->get_property_list(&props);
  3969. Vector<PropertyInfo> pinfo;
  3970. for (List<PropertyInfo>::Element *E = props.front(); E; E = E->next()) {
  3971. for (int i = 0; i < properties.size(); i++) {
  3972. if (E->get().name == String(properties[i])) {
  3973. pinfo.push_back(E->get());
  3974. }
  3975. }
  3976. }
  3977. if (pinfo.size() == 0) {
  3978. return nullptr;
  3979. }
  3980. properties.clear();
  3981. Ref<VisualShaderNode> node = p_node;
  3982. Vector<EditorProperty *> editors;
  3983. for (int i = 0; i < pinfo.size(); i++) {
  3984. EditorProperty *prop = EditorInspector::instantiate_property_editor(node.ptr(), pinfo[i].type, pinfo[i].name, pinfo[i].hint, pinfo[i].hint_string, pinfo[i].usage);
  3985. if (!prop) {
  3986. return nullptr;
  3987. }
  3988. if (Object::cast_to<EditorPropertyResource>(prop)) {
  3989. Object::cast_to<EditorPropertyResource>(prop)->set_use_sub_inspector(false);
  3990. prop->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  3991. } else if (Object::cast_to<EditorPropertyTransform3D>(prop) || Object::cast_to<EditorPropertyVector3>(prop)) {
  3992. prop->set_custom_minimum_size(Size2(250 * EDSCALE, 0));
  3993. } else if (Object::cast_to<EditorPropertyFloat>(prop)) {
  3994. prop->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  3995. } else if (Object::cast_to<EditorPropertyEnum>(prop)) {
  3996. prop->set_custom_minimum_size(Size2(100 * EDSCALE, 0));
  3997. Object::cast_to<EditorPropertyEnum>(prop)->set_option_button_clip(false);
  3998. }
  3999. editors.push_back(prop);
  4000. properties.push_back(pinfo[i].name);
  4001. }
  4002. VisualShaderNodePluginDefaultEditor *editor = memnew(VisualShaderNodePluginDefaultEditor);
  4003. editor->setup(p_parent_resource, editors, properties, p_node);
  4004. return editor;
  4005. }
  4006. void EditorPropertyShaderMode::_option_selected(int p_which) {
  4007. //will not use this, instead will do all the logic setting manually
  4008. //emit_signal("property_changed", get_edited_property(), p_which);
  4009. Ref<VisualShader> visual_shader(Object::cast_to<VisualShader>(get_edited_object()));
  4010. if (visual_shader->get_mode() == p_which) {
  4011. return;
  4012. }
  4013. UndoRedo *undo_redo = EditorNode::get_singleton()->get_undo_redo();
  4014. undo_redo->create_action(TTR("Visual Shader Mode Changed"));
  4015. //do is easy
  4016. undo_redo->add_do_method(visual_shader.ptr(), "set_mode", p_which);
  4017. undo_redo->add_undo_method(visual_shader.ptr(), "set_mode", visual_shader->get_mode());
  4018. undo_redo->add_do_method(VisualShaderEditor::get_singleton(), "_set_mode", p_which);
  4019. undo_redo->add_undo_method(VisualShaderEditor::get_singleton(), "_set_mode", visual_shader->get_mode());
  4020. //now undo is hell
  4021. //1. restore connections to output
  4022. for (int i = 0; i < VisualShader::TYPE_MAX; i++) {
  4023. VisualShader::Type type = VisualShader::Type(i);
  4024. List<VisualShader::Connection> conns;
  4025. visual_shader->get_node_connections(type, &conns);
  4026. for (List<VisualShader::Connection>::Element *E = conns.front(); E; E = E->next()) {
  4027. if (E->get().to_node == VisualShader::NODE_ID_OUTPUT) {
  4028. 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);
  4029. }
  4030. }
  4031. }
  4032. //2. restore input indices
  4033. for (int i = 0; i < VisualShader::TYPE_MAX; i++) {
  4034. VisualShader::Type type = VisualShader::Type(i);
  4035. Vector<int> nodes = visual_shader->get_node_list(type);
  4036. for (int j = 0; j < nodes.size(); j++) {
  4037. Ref<VisualShaderNodeInput> input = visual_shader->get_node(type, nodes[j]);
  4038. if (!input.is_valid()) {
  4039. continue;
  4040. }
  4041. undo_redo->add_undo_method(input.ptr(), "set_input_name", input->get_input_name());
  4042. }
  4043. }
  4044. //3. restore enums and flags
  4045. List<PropertyInfo> props;
  4046. visual_shader->get_property_list(&props);
  4047. for (List<PropertyInfo>::Element *E = props.front(); E; E = E->next()) {
  4048. if (E->get().name.begins_with("flags/") || E->get().name.begins_with("modes/")) {
  4049. undo_redo->add_undo_property(visual_shader.ptr(), E->get().name, visual_shader->get(E->get().name));
  4050. }
  4051. }
  4052. undo_redo->add_do_method(VisualShaderEditor::get_singleton(), "_update_options_menu");
  4053. undo_redo->add_undo_method(VisualShaderEditor::get_singleton(), "_update_options_menu");
  4054. //update graph
  4055. undo_redo->add_do_method(VisualShaderEditor::get_singleton(), "_update_graph");
  4056. undo_redo->add_undo_method(VisualShaderEditor::get_singleton(), "_update_graph");
  4057. undo_redo->commit_action();
  4058. }
  4059. void EditorPropertyShaderMode::update_property() {
  4060. int which = get_edited_object()->get(get_edited_property());
  4061. options->select(which);
  4062. }
  4063. void EditorPropertyShaderMode::setup(const Vector<String> &p_options) {
  4064. for (int i = 0; i < p_options.size(); i++) {
  4065. options->add_item(p_options[i], i);
  4066. }
  4067. }
  4068. void EditorPropertyShaderMode::set_option_button_clip(bool p_enable) {
  4069. options->set_clip_text(p_enable);
  4070. }
  4071. void EditorPropertyShaderMode::_bind_methods() {
  4072. }
  4073. EditorPropertyShaderMode::EditorPropertyShaderMode() {
  4074. options = memnew(OptionButton);
  4075. options->set_clip_text(true);
  4076. add_child(options);
  4077. add_focusable(options);
  4078. options->connect("item_selected", callable_mp(this, &EditorPropertyShaderMode::_option_selected));
  4079. }
  4080. bool EditorInspectorShaderModePlugin::can_handle(Object *p_object) {
  4081. return true; //can handle everything
  4082. }
  4083. void EditorInspectorShaderModePlugin::parse_begin(Object *p_object) {
  4084. //do none
  4085. }
  4086. bool EditorInspectorShaderModePlugin::parse_property(Object *p_object, const Variant::Type p_type, const String &p_path, const PropertyHint p_hint, const String &p_hint_text, const uint32_t p_usage, const bool p_wide) {
  4087. if (p_path == "mode" && p_object->is_class("VisualShader") && p_type == Variant::INT) {
  4088. EditorPropertyShaderMode *editor = memnew(EditorPropertyShaderMode);
  4089. Vector<String> options = p_hint_text.split(",");
  4090. editor->setup(options);
  4091. add_property_editor(p_path, editor);
  4092. return true;
  4093. }
  4094. return false; //can be overridden, although it will most likely be last anyway
  4095. }
  4096. void EditorInspectorShaderModePlugin::parse_end() {
  4097. //do none
  4098. }
  4099. //////////////////////////////////
  4100. void VisualShaderNodePortPreview::_shader_changed() {
  4101. if (shader.is_null()) {
  4102. return;
  4103. }
  4104. Vector<VisualShader::DefaultTextureParam> default_textures;
  4105. String shader_code = shader->generate_preview_shader(type, node, port, default_textures);
  4106. Ref<Shader> preview_shader;
  4107. preview_shader.instantiate();
  4108. preview_shader->set_code(shader_code);
  4109. for (int i = 0; i < default_textures.size(); i++) {
  4110. preview_shader->set_default_texture_param(default_textures[i].name, default_textures[i].param);
  4111. }
  4112. Ref<ShaderMaterial> material;
  4113. material.instantiate();
  4114. material->set_shader(preview_shader);
  4115. //find if a material is also being edited and copy parameters to this one
  4116. for (int i = EditorNode::get_singleton()->get_editor_history()->get_path_size() - 1; i >= 0; i--) {
  4117. Object *object = ObjectDB::get_instance(EditorNode::get_singleton()->get_editor_history()->get_path_object(i));
  4118. ShaderMaterial *src_mat;
  4119. if (!object) {
  4120. continue;
  4121. }
  4122. if (object->has_method("get_material_override")) { // trying getting material from MeshInstance
  4123. src_mat = Object::cast_to<ShaderMaterial>(object->call("get_material_override"));
  4124. } else if (object->has_method("get_material")) { // from CanvasItem/Node2D
  4125. src_mat = Object::cast_to<ShaderMaterial>(object->call("get_material"));
  4126. } else {
  4127. src_mat = Object::cast_to<ShaderMaterial>(object);
  4128. }
  4129. if (src_mat && src_mat->get_shader().is_valid()) {
  4130. List<PropertyInfo> params;
  4131. src_mat->get_shader()->get_param_list(&params);
  4132. for (List<PropertyInfo>::Element *E = params.front(); E; E = E->next()) {
  4133. material->set(E->get().name, src_mat->get(E->get().name));
  4134. }
  4135. }
  4136. }
  4137. set_material(material);
  4138. }
  4139. void VisualShaderNodePortPreview::setup(const Ref<VisualShader> &p_shader, VisualShader::Type p_type, int p_node, int p_port) {
  4140. shader = p_shader;
  4141. shader->connect("changed", callable_mp(this, &VisualShaderNodePortPreview::_shader_changed));
  4142. type = p_type;
  4143. port = p_port;
  4144. node = p_node;
  4145. update();
  4146. _shader_changed();
  4147. }
  4148. Size2 VisualShaderNodePortPreview::get_minimum_size() const {
  4149. return Size2(100, 100) * EDSCALE;
  4150. }
  4151. void VisualShaderNodePortPreview::_notification(int p_what) {
  4152. if (p_what == NOTIFICATION_DRAW) {
  4153. Vector<Vector2> points;
  4154. Vector<Vector2> uvs;
  4155. Vector<Color> colors;
  4156. points.push_back(Vector2());
  4157. uvs.push_back(Vector2(0, 0));
  4158. colors.push_back(Color(1, 1, 1, 1));
  4159. points.push_back(Vector2(get_size().width, 0));
  4160. uvs.push_back(Vector2(1, 0));
  4161. colors.push_back(Color(1, 1, 1, 1));
  4162. points.push_back(get_size());
  4163. uvs.push_back(Vector2(1, 1));
  4164. colors.push_back(Color(1, 1, 1, 1));
  4165. points.push_back(Vector2(0, get_size().height));
  4166. uvs.push_back(Vector2(0, 1));
  4167. colors.push_back(Color(1, 1, 1, 1));
  4168. draw_primitive(points, colors, uvs);
  4169. }
  4170. }
  4171. void VisualShaderNodePortPreview::_bind_methods() {
  4172. }
  4173. //////////////////////////////////
  4174. String VisualShaderConversionPlugin::converts_to() const {
  4175. return "Shader";
  4176. }
  4177. bool VisualShaderConversionPlugin::handles(const Ref<Resource> &p_resource) const {
  4178. Ref<VisualShader> vshader = p_resource;
  4179. return vshader.is_valid();
  4180. }
  4181. Ref<Resource> VisualShaderConversionPlugin::convert(const Ref<Resource> &p_resource) const {
  4182. Ref<VisualShader> vshader = p_resource;
  4183. ERR_FAIL_COND_V(!vshader.is_valid(), Ref<Resource>());
  4184. Ref<Shader> shader;
  4185. shader.instantiate();
  4186. String code = vshader->get_code();
  4187. shader->set_code(code);
  4188. return shader;
  4189. }