visual_shader_editor_plugin.cpp 255 KB

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