visual_script_nodes.cpp 117 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076
  1. /*************************************************************************/
  2. /* visual_script_nodes.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_script_nodes.h"
  31. #include "core/config/engine.h"
  32. #include "core/config/project_settings.h"
  33. #include "core/core_constants.h"
  34. #include "core/input/input.h"
  35. #include "core/os/os.h"
  36. #include "scene/main/node.h"
  37. #include "scene/main/scene_tree.h"
  38. //////////////////////////////////////////
  39. ////////////////FUNCTION//////////////////
  40. //////////////////////////////////////////
  41. bool VisualScriptFunction::_set(const StringName &p_name, const Variant &p_value) {
  42. if (p_name == "argument_count") {
  43. int new_argc = p_value;
  44. int argc = arguments.size();
  45. if (argc == new_argc) {
  46. return true;
  47. }
  48. arguments.resize(new_argc);
  49. for (int i = argc; i < new_argc; i++) {
  50. arguments.write[i].name = "arg" + itos(i + 1);
  51. arguments.write[i].type = Variant::NIL;
  52. }
  53. ports_changed_notify();
  54. notify_property_list_changed();
  55. return true;
  56. }
  57. if (String(p_name).begins_with("argument_")) {
  58. int idx = String(p_name).get_slicec('_', 1).get_slicec('/', 0).to_int() - 1;
  59. ERR_FAIL_INDEX_V(idx, arguments.size(), false);
  60. String what = String(p_name).get_slice("/", 1);
  61. if (what == "type") {
  62. Variant::Type new_type = Variant::Type(int(p_value));
  63. arguments.write[idx].type = new_type;
  64. ports_changed_notify();
  65. return true;
  66. }
  67. if (what == "name") {
  68. arguments.write[idx].name = p_value;
  69. ports_changed_notify();
  70. return true;
  71. }
  72. }
  73. if (p_name == "stack/stackless") {
  74. set_stack_less(p_value);
  75. return true;
  76. }
  77. if (p_name == "stack/size") {
  78. stack_size = p_value;
  79. return true;
  80. }
  81. if (p_name == "rpc/mode") {
  82. rpc_mode = Multiplayer::RPCMode(int(p_value));
  83. return true;
  84. }
  85. if (p_name == "sequenced/sequenced") {
  86. sequenced = p_value;
  87. ports_changed_notify();
  88. return true;
  89. }
  90. return false;
  91. }
  92. bool VisualScriptFunction::_get(const StringName &p_name, Variant &r_ret) const {
  93. if (p_name == "argument_count") {
  94. r_ret = arguments.size();
  95. return true;
  96. }
  97. if (String(p_name).begins_with("argument_")) {
  98. int idx = String(p_name).get_slicec('_', 1).get_slicec('/', 0).to_int() - 1;
  99. ERR_FAIL_INDEX_V(idx, arguments.size(), false);
  100. String what = String(p_name).get_slice("/", 1);
  101. if (what == "type") {
  102. r_ret = arguments[idx].type;
  103. return true;
  104. }
  105. if (what == "name") {
  106. r_ret = arguments[idx].name;
  107. return true;
  108. }
  109. }
  110. if (p_name == "stack/stackless") {
  111. r_ret = stack_less;
  112. return true;
  113. }
  114. if (p_name == "stack/size") {
  115. r_ret = stack_size;
  116. return true;
  117. }
  118. if (p_name == "rpc/mode") {
  119. r_ret = rpc_mode;
  120. return true;
  121. }
  122. if (p_name == "sequenced/sequenced") {
  123. r_ret = sequenced;
  124. return true;
  125. }
  126. return false;
  127. }
  128. void VisualScriptFunction::_get_property_list(List<PropertyInfo> *p_list) const {
  129. p_list->push_back(PropertyInfo(Variant::INT, "argument_count", PROPERTY_HINT_RANGE, "0,256"));
  130. String argt = "Any";
  131. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  132. argt += "," + Variant::get_type_name(Variant::Type(i));
  133. }
  134. for (int i = 0; i < arguments.size(); i++) {
  135. p_list->push_back(PropertyInfo(Variant::INT, "argument_" + itos(i + 1) + "/type", PROPERTY_HINT_ENUM, argt));
  136. p_list->push_back(PropertyInfo(Variant::STRING, "argument_" + itos(i + 1) + "/name"));
  137. }
  138. p_list->push_back(PropertyInfo(Variant::BOOL, "sequenced/sequenced"));
  139. if (!stack_less) {
  140. p_list->push_back(PropertyInfo(Variant::INT, "stack/size", PROPERTY_HINT_RANGE, "1,100000"));
  141. }
  142. p_list->push_back(PropertyInfo(Variant::BOOL, "stack/stackless"));
  143. p_list->push_back(PropertyInfo(Variant::INT, "rpc/mode", PROPERTY_HINT_ENUM, "Disabled,Any,Authority"));
  144. }
  145. int VisualScriptFunction::get_output_sequence_port_count() const {
  146. return 1;
  147. }
  148. bool VisualScriptFunction::has_input_sequence_port() const {
  149. return false;
  150. }
  151. int VisualScriptFunction::get_input_value_port_count() const {
  152. return 0;
  153. }
  154. int VisualScriptFunction::get_output_value_port_count() const {
  155. return arguments.size();
  156. }
  157. String VisualScriptFunction::get_output_sequence_port_text(int p_port) const {
  158. return String();
  159. }
  160. PropertyInfo VisualScriptFunction::get_input_value_port_info(int p_idx) const {
  161. ERR_FAIL_V(PropertyInfo());
  162. }
  163. PropertyInfo VisualScriptFunction::get_output_value_port_info(int p_idx) const {
  164. // Need to check it without ERR_FAIL_COND, to prevent warnings from appearing on node creation via dragging.
  165. if (p_idx < 0 || p_idx >= arguments.size()) {
  166. return PropertyInfo();
  167. }
  168. PropertyInfo out;
  169. out.type = arguments[p_idx].type;
  170. out.name = arguments[p_idx].name;
  171. out.hint = arguments[p_idx].hint;
  172. out.hint_string = arguments[p_idx].hint_string;
  173. return out;
  174. }
  175. String VisualScriptFunction::get_caption() const {
  176. return "Function";
  177. }
  178. String VisualScriptFunction::get_text() const {
  179. return get_name(); //use name as function name I guess
  180. }
  181. void VisualScriptFunction::add_argument(Variant::Type p_type, const String &p_name, int p_index, const PropertyHint p_hint, const String &p_hint_string) {
  182. Argument arg;
  183. arg.name = p_name;
  184. arg.type = p_type;
  185. arg.hint = p_hint;
  186. arg.hint_string = p_hint_string;
  187. if (p_index >= 0) {
  188. arguments.insert(p_index, arg);
  189. } else {
  190. arguments.push_back(arg);
  191. }
  192. ports_changed_notify();
  193. }
  194. void VisualScriptFunction::set_argument_type(int p_argidx, Variant::Type p_type) {
  195. ERR_FAIL_INDEX(p_argidx, arguments.size());
  196. arguments.write[p_argidx].type = p_type;
  197. ports_changed_notify();
  198. }
  199. Variant::Type VisualScriptFunction::get_argument_type(int p_argidx) const {
  200. ERR_FAIL_INDEX_V(p_argidx, arguments.size(), Variant::NIL);
  201. return arguments[p_argidx].type;
  202. }
  203. void VisualScriptFunction::set_argument_name(int p_argidx, const String &p_name) {
  204. ERR_FAIL_INDEX(p_argidx, arguments.size());
  205. arguments.write[p_argidx].name = p_name;
  206. ports_changed_notify();
  207. }
  208. String VisualScriptFunction::get_argument_name(int p_argidx) const {
  209. ERR_FAIL_INDEX_V(p_argidx, arguments.size(), String());
  210. return arguments[p_argidx].name;
  211. }
  212. void VisualScriptFunction::remove_argument(int p_argidx) {
  213. ERR_FAIL_INDEX(p_argidx, arguments.size());
  214. arguments.remove_at(p_argidx);
  215. ports_changed_notify();
  216. }
  217. int VisualScriptFunction::get_argument_count() const {
  218. return arguments.size();
  219. }
  220. void VisualScriptFunction::set_rpc_mode(Multiplayer::RPCMode p_mode) {
  221. rpc_mode = p_mode;
  222. }
  223. Multiplayer::RPCMode VisualScriptFunction::get_rpc_mode() const {
  224. return rpc_mode;
  225. }
  226. class VisualScriptNodeInstanceFunction : public VisualScriptNodeInstance {
  227. public:
  228. VisualScriptFunction *node;
  229. VisualScriptInstance *instance;
  230. //virtual int get_working_memory_size() const { return 0; }
  231. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  232. int ac = node->get_argument_count();
  233. for (int i = 0; i < ac; i++) {
  234. #ifdef DEBUG_ENABLED
  235. Variant::Type expected = node->get_argument_type(i);
  236. if (expected != Variant::NIL) {
  237. if (!Variant::can_convert_strict(p_inputs[i]->get_type(), expected)) {
  238. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  239. r_error.expected = expected;
  240. r_error.argument = i;
  241. return 0;
  242. }
  243. }
  244. #endif
  245. *p_outputs[i] = *p_inputs[i];
  246. }
  247. return 0;
  248. }
  249. };
  250. VisualScriptNodeInstance *VisualScriptFunction::instantiate(VisualScriptInstance *p_instance) {
  251. VisualScriptNodeInstanceFunction *instance = memnew(VisualScriptNodeInstanceFunction);
  252. instance->node = this;
  253. instance->instance = p_instance;
  254. return instance;
  255. }
  256. void VisualScriptFunction::reset_state() {
  257. arguments.clear();
  258. stack_size = 256;
  259. stack_less = false;
  260. sequenced = true;
  261. rpc_mode = Multiplayer::RPC_MODE_DISABLED;
  262. }
  263. VisualScriptFunction::VisualScriptFunction() {
  264. stack_size = 256;
  265. stack_less = false;
  266. sequenced = true;
  267. rpc_mode = Multiplayer::RPC_MODE_DISABLED;
  268. }
  269. void VisualScriptFunction::set_stack_less(bool p_enable) {
  270. stack_less = p_enable;
  271. notify_property_list_changed();
  272. }
  273. bool VisualScriptFunction::is_stack_less() const {
  274. return stack_less;
  275. }
  276. void VisualScriptFunction::set_sequenced(bool p_enable) {
  277. sequenced = p_enable;
  278. }
  279. bool VisualScriptFunction::is_sequenced() const {
  280. return sequenced;
  281. }
  282. void VisualScriptFunction::set_stack_size(int p_size) {
  283. ERR_FAIL_COND(p_size < 1 || p_size > 100000);
  284. stack_size = p_size;
  285. }
  286. int VisualScriptFunction::get_stack_size() const {
  287. return stack_size;
  288. }
  289. //////////////////////////////////////////
  290. /////////////////LISTS////////////////////
  291. //////////////////////////////////////////
  292. int VisualScriptLists::get_output_sequence_port_count() const {
  293. if (sequenced) {
  294. return 1;
  295. }
  296. return 0;
  297. }
  298. bool VisualScriptLists::has_input_sequence_port() const {
  299. return sequenced;
  300. }
  301. String VisualScriptLists::get_output_sequence_port_text(int p_port) const {
  302. return "";
  303. }
  304. int VisualScriptLists::get_input_value_port_count() const {
  305. return inputports.size();
  306. }
  307. int VisualScriptLists::get_output_value_port_count() const {
  308. return outputports.size();
  309. }
  310. PropertyInfo VisualScriptLists::get_input_value_port_info(int p_idx) const {
  311. ERR_FAIL_INDEX_V(p_idx, inputports.size(), PropertyInfo());
  312. PropertyInfo pi;
  313. pi.name = inputports[p_idx].name;
  314. pi.type = inputports[p_idx].type;
  315. return pi;
  316. }
  317. PropertyInfo VisualScriptLists::get_output_value_port_info(int p_idx) const {
  318. ERR_FAIL_INDEX_V(p_idx, outputports.size(), PropertyInfo());
  319. PropertyInfo pi;
  320. pi.name = outputports[p_idx].name;
  321. pi.type = outputports[p_idx].type;
  322. return pi;
  323. }
  324. bool VisualScriptLists::is_input_port_editable() const {
  325. return ((flags & INPUT_EDITABLE) == INPUT_EDITABLE);
  326. }
  327. bool VisualScriptLists::is_input_port_name_editable() const {
  328. return ((flags & INPUT_NAME_EDITABLE) == INPUT_NAME_EDITABLE);
  329. }
  330. bool VisualScriptLists::is_input_port_type_editable() const {
  331. return ((flags & INPUT_TYPE_EDITABLE) == INPUT_TYPE_EDITABLE);
  332. }
  333. bool VisualScriptLists::is_output_port_editable() const {
  334. return ((flags & OUTPUT_EDITABLE) == OUTPUT_EDITABLE);
  335. }
  336. bool VisualScriptLists::is_output_port_name_editable() const {
  337. return ((flags & INPUT_NAME_EDITABLE) == INPUT_NAME_EDITABLE);
  338. }
  339. bool VisualScriptLists::is_output_port_type_editable() const {
  340. return ((flags & INPUT_TYPE_EDITABLE) == INPUT_TYPE_EDITABLE);
  341. }
  342. // for the inspector
  343. bool VisualScriptLists::_set(const StringName &p_name, const Variant &p_value) {
  344. if (p_name == "input_count" && is_input_port_editable()) {
  345. int new_argc = p_value;
  346. int argc = inputports.size();
  347. if (argc == new_argc) {
  348. return true;
  349. }
  350. inputports.resize(new_argc);
  351. for (int i = argc; i < new_argc; i++) {
  352. inputports.write[i].name = "arg" + itos(i + 1);
  353. inputports.write[i].type = Variant::NIL;
  354. }
  355. ports_changed_notify();
  356. notify_property_list_changed();
  357. return true;
  358. }
  359. if (String(p_name).begins_with("input_") && is_input_port_editable()) {
  360. int idx = String(p_name).get_slicec('_', 1).get_slicec('/', 0).to_int() - 1;
  361. ERR_FAIL_INDEX_V(idx, inputports.size(), false);
  362. String what = String(p_name).get_slice("/", 1);
  363. if (what == "type") {
  364. Variant::Type new_type = Variant::Type(int(p_value));
  365. inputports.write[idx].type = new_type;
  366. ports_changed_notify();
  367. return true;
  368. }
  369. if (what == "name") {
  370. inputports.write[idx].name = p_value;
  371. ports_changed_notify();
  372. return true;
  373. }
  374. }
  375. if (p_name == "output_count" && is_output_port_editable()) {
  376. int new_argc = p_value;
  377. int argc = outputports.size();
  378. if (argc == new_argc) {
  379. return true;
  380. }
  381. outputports.resize(new_argc);
  382. for (int i = argc; i < new_argc; i++) {
  383. outputports.write[i].name = "arg" + itos(i + 1);
  384. outputports.write[i].type = Variant::NIL;
  385. }
  386. ports_changed_notify();
  387. notify_property_list_changed();
  388. return true;
  389. }
  390. if (String(p_name).begins_with("output_") && is_output_port_editable()) {
  391. int idx = String(p_name).get_slicec('_', 1).get_slicec('/', 0).to_int() - 1;
  392. ERR_FAIL_INDEX_V(idx, outputports.size(), false);
  393. String what = String(p_name).get_slice("/", 1);
  394. if (what == "type") {
  395. Variant::Type new_type = Variant::Type(int(p_value));
  396. outputports.write[idx].type = new_type;
  397. ports_changed_notify();
  398. return true;
  399. }
  400. if (what == "name") {
  401. outputports.write[idx].name = p_value;
  402. ports_changed_notify();
  403. return true;
  404. }
  405. }
  406. if (p_name == "sequenced/sequenced") {
  407. sequenced = p_value;
  408. ports_changed_notify();
  409. return true;
  410. }
  411. return false;
  412. }
  413. bool VisualScriptLists::_get(const StringName &p_name, Variant &r_ret) const {
  414. if (p_name == "input_count" && is_input_port_editable()) {
  415. r_ret = inputports.size();
  416. return true;
  417. }
  418. if (String(p_name).begins_with("input_") && is_input_port_editable()) {
  419. int idx = String(p_name).get_slicec('_', 1).get_slicec('/', 0).to_int() - 1;
  420. ERR_FAIL_INDEX_V(idx, inputports.size(), false);
  421. String what = String(p_name).get_slice("/", 1);
  422. if (what == "type") {
  423. r_ret = inputports[idx].type;
  424. return true;
  425. }
  426. if (what == "name") {
  427. r_ret = inputports[idx].name;
  428. return true;
  429. }
  430. }
  431. if (p_name == "output_count" && is_output_port_editable()) {
  432. r_ret = outputports.size();
  433. return true;
  434. }
  435. if (String(p_name).begins_with("output_") && is_output_port_editable()) {
  436. int idx = String(p_name).get_slicec('_', 1).get_slicec('/', 0).to_int() - 1;
  437. ERR_FAIL_INDEX_V(idx, outputports.size(), false);
  438. String what = String(p_name).get_slice("/", 1);
  439. if (what == "type") {
  440. r_ret = outputports[idx].type;
  441. return true;
  442. }
  443. if (what == "name") {
  444. r_ret = outputports[idx].name;
  445. return true;
  446. }
  447. }
  448. if (p_name == "sequenced/sequenced") {
  449. r_ret = sequenced;
  450. return true;
  451. }
  452. return false;
  453. }
  454. void VisualScriptLists::_get_property_list(List<PropertyInfo> *p_list) const {
  455. if (is_input_port_editable()) {
  456. p_list->push_back(PropertyInfo(Variant::INT, "input_count", PROPERTY_HINT_RANGE, "0,256"));
  457. String argt = "Any";
  458. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  459. argt += "," + Variant::get_type_name(Variant::Type(i));
  460. }
  461. for (int i = 0; i < inputports.size(); i++) {
  462. p_list->push_back(PropertyInfo(Variant::INT, "input_" + itos(i + 1) + "/type", PROPERTY_HINT_ENUM, argt));
  463. p_list->push_back(PropertyInfo(Variant::STRING, "input_" + itos(i + 1) + "/name"));
  464. }
  465. }
  466. if (is_output_port_editable()) {
  467. p_list->push_back(PropertyInfo(Variant::INT, "output_count", PROPERTY_HINT_RANGE, "0,256"));
  468. String argt = "Any";
  469. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  470. argt += "," + Variant::get_type_name(Variant::Type(i));
  471. }
  472. for (int i = 0; i < outputports.size(); i++) {
  473. p_list->push_back(PropertyInfo(Variant::INT, "output_" + itos(i + 1) + "/type", PROPERTY_HINT_ENUM, argt));
  474. p_list->push_back(PropertyInfo(Variant::STRING, "output_" + itos(i + 1) + "/name"));
  475. }
  476. }
  477. p_list->push_back(PropertyInfo(Variant::BOOL, "sequenced/sequenced"));
  478. }
  479. // input data port interaction
  480. void VisualScriptLists::add_input_data_port(Variant::Type p_type, const String &p_name, int p_index) {
  481. if (!is_input_port_editable()) {
  482. return;
  483. }
  484. Port inp;
  485. inp.name = p_name;
  486. inp.type = p_type;
  487. if (p_index >= 0) {
  488. inputports.insert(p_index, inp);
  489. } else {
  490. inputports.push_back(inp);
  491. }
  492. ports_changed_notify();
  493. notify_property_list_changed();
  494. }
  495. void VisualScriptLists::set_input_data_port_type(int p_idx, Variant::Type p_type) {
  496. if (!is_input_port_type_editable()) {
  497. return;
  498. }
  499. ERR_FAIL_INDEX(p_idx, inputports.size());
  500. inputports.write[p_idx].type = p_type;
  501. ports_changed_notify();
  502. notify_property_list_changed();
  503. }
  504. void VisualScriptLists::set_input_data_port_name(int p_idx, const String &p_name) {
  505. if (!is_input_port_name_editable()) {
  506. return;
  507. }
  508. ERR_FAIL_INDEX(p_idx, inputports.size());
  509. inputports.write[p_idx].name = p_name;
  510. ports_changed_notify();
  511. notify_property_list_changed();
  512. }
  513. void VisualScriptLists::remove_input_data_port(int p_argidx) {
  514. if (!is_input_port_editable()) {
  515. return;
  516. }
  517. ERR_FAIL_INDEX(p_argidx, inputports.size());
  518. inputports.remove_at(p_argidx);
  519. ports_changed_notify();
  520. notify_property_list_changed();
  521. }
  522. // output data port interaction
  523. void VisualScriptLists::add_output_data_port(Variant::Type p_type, const String &p_name, int p_index) {
  524. if (!is_output_port_editable()) {
  525. return;
  526. }
  527. Port out;
  528. out.name = p_name;
  529. out.type = p_type;
  530. if (p_index >= 0) {
  531. outputports.insert(p_index, out);
  532. } else {
  533. outputports.push_back(out);
  534. }
  535. ports_changed_notify();
  536. notify_property_list_changed();
  537. }
  538. void VisualScriptLists::set_output_data_port_type(int p_idx, Variant::Type p_type) {
  539. if (!is_output_port_type_editable()) {
  540. return;
  541. }
  542. ERR_FAIL_INDEX(p_idx, outputports.size());
  543. outputports.write[p_idx].type = p_type;
  544. ports_changed_notify();
  545. notify_property_list_changed();
  546. }
  547. void VisualScriptLists::set_output_data_port_name(int p_idx, const String &p_name) {
  548. if (!is_output_port_name_editable()) {
  549. return;
  550. }
  551. ERR_FAIL_INDEX(p_idx, outputports.size());
  552. outputports.write[p_idx].name = p_name;
  553. ports_changed_notify();
  554. notify_property_list_changed();
  555. }
  556. void VisualScriptLists::remove_output_data_port(int p_argidx) {
  557. if (!is_output_port_editable()) {
  558. return;
  559. }
  560. ERR_FAIL_INDEX(p_argidx, outputports.size());
  561. outputports.remove_at(p_argidx);
  562. ports_changed_notify();
  563. notify_property_list_changed();
  564. }
  565. // sequences
  566. void VisualScriptLists::set_sequenced(bool p_enable) {
  567. if (sequenced == p_enable) {
  568. return;
  569. }
  570. sequenced = p_enable;
  571. ports_changed_notify();
  572. }
  573. bool VisualScriptLists::is_sequenced() const {
  574. return sequenced;
  575. }
  576. void VisualScriptLists::reset_state() {
  577. inputports.clear();
  578. outputports.clear();
  579. sequenced = false;
  580. flags = 0;
  581. }
  582. VisualScriptLists::VisualScriptLists() {
  583. // initialize
  584. sequenced = false;
  585. flags = 0;
  586. }
  587. void VisualScriptLists::_bind_methods() {
  588. ClassDB::bind_method(D_METHOD("add_input_data_port", "type", "name", "index"), &VisualScriptLists::add_input_data_port);
  589. ClassDB::bind_method(D_METHOD("set_input_data_port_name", "index", "name"), &VisualScriptLists::set_input_data_port_name);
  590. ClassDB::bind_method(D_METHOD("set_input_data_port_type", "index", "type"), &VisualScriptLists::set_input_data_port_type);
  591. ClassDB::bind_method(D_METHOD("remove_input_data_port", "index"), &VisualScriptLists::remove_input_data_port);
  592. ClassDB::bind_method(D_METHOD("add_output_data_port", "type", "name", "index"), &VisualScriptLists::add_output_data_port);
  593. ClassDB::bind_method(D_METHOD("set_output_data_port_name", "index", "name"), &VisualScriptLists::set_output_data_port_name);
  594. ClassDB::bind_method(D_METHOD("set_output_data_port_type", "index", "type"), &VisualScriptLists::set_output_data_port_type);
  595. ClassDB::bind_method(D_METHOD("remove_output_data_port", "index"), &VisualScriptLists::remove_output_data_port);
  596. }
  597. //////////////////////////////////////////
  598. //////////////COMPOSEARRAY////////////////
  599. //////////////////////////////////////////
  600. int VisualScriptComposeArray::get_output_sequence_port_count() const {
  601. if (sequenced) {
  602. return 1;
  603. }
  604. return 0;
  605. }
  606. bool VisualScriptComposeArray::has_input_sequence_port() const {
  607. return sequenced;
  608. }
  609. String VisualScriptComposeArray::get_output_sequence_port_text(int p_port) const {
  610. return "";
  611. }
  612. int VisualScriptComposeArray::get_input_value_port_count() const {
  613. return inputports.size();
  614. }
  615. int VisualScriptComposeArray::get_output_value_port_count() const {
  616. return 1;
  617. }
  618. PropertyInfo VisualScriptComposeArray::get_input_value_port_info(int p_idx) const {
  619. ERR_FAIL_INDEX_V(p_idx, inputports.size(), PropertyInfo());
  620. PropertyInfo pi;
  621. pi.name = inputports[p_idx].name;
  622. pi.type = inputports[p_idx].type;
  623. return pi;
  624. }
  625. PropertyInfo VisualScriptComposeArray::get_output_value_port_info(int p_idx) const {
  626. PropertyInfo pi;
  627. pi.name = "out";
  628. pi.type = Variant::ARRAY;
  629. return pi;
  630. }
  631. String VisualScriptComposeArray::get_caption() const {
  632. return "Compose Array";
  633. }
  634. String VisualScriptComposeArray::get_text() const {
  635. return "";
  636. }
  637. class VisualScriptComposeArrayNode : public VisualScriptNodeInstance {
  638. public:
  639. int input_count = 0;
  640. virtual int get_working_memory_size() const { return 0; }
  641. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  642. if (input_count > 0) {
  643. Array arr;
  644. for (int i = 0; i < input_count; i++) {
  645. arr.push_back((*p_inputs[i]));
  646. }
  647. Variant va = Variant(arr);
  648. *p_outputs[0] = va;
  649. }
  650. return 0;
  651. }
  652. };
  653. VisualScriptNodeInstance *VisualScriptComposeArray::instantiate(VisualScriptInstance *p_instance) {
  654. VisualScriptComposeArrayNode *instance = memnew(VisualScriptComposeArrayNode);
  655. instance->input_count = inputports.size();
  656. return instance;
  657. }
  658. VisualScriptComposeArray::VisualScriptComposeArray() {
  659. // initialize stuff here
  660. sequenced = false;
  661. flags = INPUT_EDITABLE;
  662. }
  663. //////////////////////////////////////////
  664. ////////////////OPERATOR//////////////////
  665. //////////////////////////////////////////
  666. int VisualScriptOperator::get_output_sequence_port_count() const {
  667. return 0;
  668. }
  669. bool VisualScriptOperator::has_input_sequence_port() const {
  670. return false;
  671. }
  672. int VisualScriptOperator::get_input_value_port_count() const {
  673. return (op == Variant::OP_BIT_NEGATE || op == Variant::OP_NOT || op == Variant::OP_NEGATE || op == Variant::OP_POSITIVE) ? 1 : 2;
  674. }
  675. int VisualScriptOperator::get_output_value_port_count() const {
  676. return 1;
  677. }
  678. String VisualScriptOperator::get_output_sequence_port_text(int p_port) const {
  679. return String();
  680. }
  681. PropertyInfo VisualScriptOperator::get_input_value_port_info(int p_idx) const {
  682. static const Variant::Type port_types[Variant::OP_MAX][2] = {
  683. { Variant::NIL, Variant::NIL }, //OP_EQUAL,
  684. { Variant::NIL, Variant::NIL }, //OP_NOT_EQUAL,
  685. { Variant::NIL, Variant::NIL }, //OP_LESS,
  686. { Variant::NIL, Variant::NIL }, //OP_LESS_EQUAL,
  687. { Variant::NIL, Variant::NIL }, //OP_GREATER,
  688. { Variant::NIL, Variant::NIL }, //OP_GREATER_EQUAL,
  689. //mathematic
  690. { Variant::NIL, Variant::NIL }, //OP_ADD,
  691. { Variant::NIL, Variant::NIL }, //OP_SUBTRACT,
  692. { Variant::NIL, Variant::NIL }, //OP_MULTIPLY,
  693. { Variant::NIL, Variant::NIL }, //OP_DIVIDE,
  694. { Variant::NIL, Variant::NIL }, //OP_NEGATE,
  695. { Variant::NIL, Variant::NIL }, //OP_POSITIVE,
  696. { Variant::INT, Variant::INT }, //OP_MODULE,
  697. //bitwise
  698. { Variant::INT, Variant::INT }, //OP_SHIFT_LEFT,
  699. { Variant::INT, Variant::INT }, //OP_SHIFT_RIGHT,
  700. { Variant::INT, Variant::INT }, //OP_BIT_AND,
  701. { Variant::INT, Variant::INT }, //OP_BIT_OR,
  702. { Variant::INT, Variant::INT }, //OP_BIT_XOR,
  703. { Variant::INT, Variant::INT }, //OP_BIT_NEGATE,
  704. //logic
  705. { Variant::BOOL, Variant::BOOL }, //OP_AND,
  706. { Variant::BOOL, Variant::BOOL }, //OP_OR,
  707. { Variant::BOOL, Variant::BOOL }, //OP_XOR,
  708. { Variant::BOOL, Variant::BOOL }, //OP_NOT,
  709. //containment
  710. { Variant::NIL, Variant::NIL } //OP_IN,
  711. };
  712. ERR_FAIL_INDEX_V(p_idx, 2, PropertyInfo());
  713. PropertyInfo pinfo;
  714. pinfo.name = p_idx == 0 ? "A" : "B";
  715. pinfo.type = port_types[op][p_idx];
  716. if (pinfo.type == Variant::NIL) {
  717. pinfo.type = typed;
  718. }
  719. return pinfo;
  720. }
  721. PropertyInfo VisualScriptOperator::get_output_value_port_info(int p_idx) const {
  722. static const Variant::Type port_types[Variant::OP_MAX] = {
  723. //comparison
  724. Variant::BOOL, //OP_EQUAL,
  725. Variant::BOOL, //OP_NOT_EQUAL,
  726. Variant::BOOL, //OP_LESS,
  727. Variant::BOOL, //OP_LESS_EQUAL,
  728. Variant::BOOL, //OP_GREATER,
  729. Variant::BOOL, //OP_GREATER_EQUAL,
  730. //mathematic
  731. Variant::NIL, //OP_ADD,
  732. Variant::NIL, //OP_SUBTRACT,
  733. Variant::NIL, //OP_MULTIPLY,
  734. Variant::NIL, //OP_DIVIDE,
  735. Variant::NIL, //OP_NEGATE,
  736. Variant::NIL, //OP_POSITIVE,
  737. Variant::INT, //OP_MODULE,
  738. //bitwise
  739. Variant::INT, //OP_SHIFT_LEFT,
  740. Variant::INT, //OP_SHIFT_RIGHT,
  741. Variant::INT, //OP_BIT_AND,
  742. Variant::INT, //OP_BIT_OR,
  743. Variant::INT, //OP_BIT_XOR,
  744. Variant::INT, //OP_BIT_NEGATE,
  745. //logic
  746. Variant::BOOL, //OP_AND,
  747. Variant::BOOL, //OP_OR,
  748. Variant::BOOL, //OP_XOR,
  749. Variant::BOOL, //OP_NOT,
  750. //containment
  751. Variant::BOOL //OP_IN,
  752. };
  753. PropertyInfo pinfo;
  754. pinfo.name = "";
  755. pinfo.type = port_types[op];
  756. if (pinfo.type == Variant::NIL) {
  757. pinfo.type = typed;
  758. }
  759. return pinfo;
  760. }
  761. String VisualScriptOperator::get_caption() const {
  762. switch (op) {
  763. // comparison
  764. case Variant::OP_EQUAL:
  765. return U"A = B";
  766. case Variant::OP_NOT_EQUAL:
  767. return U"A \u2260 B";
  768. case Variant::OP_LESS:
  769. return U"A < B";
  770. case Variant::OP_LESS_EQUAL:
  771. return U"A \u2264 B";
  772. case Variant::OP_GREATER:
  773. return U"A > B";
  774. case Variant::OP_GREATER_EQUAL:
  775. return U"A \u2265 B";
  776. // mathematic
  777. case Variant::OP_ADD:
  778. return U"A + B";
  779. case Variant::OP_SUBTRACT:
  780. return U"A - B";
  781. case Variant::OP_MULTIPLY:
  782. return U"A \u00D7 B";
  783. case Variant::OP_DIVIDE:
  784. return U"A \u00F7 B";
  785. case Variant::OP_NEGATE:
  786. return U"\u00AC A";
  787. case Variant::OP_POSITIVE:
  788. return U"+ A";
  789. case Variant::OP_MODULE:
  790. return U"A mod B";
  791. // bitwise
  792. case Variant::OP_SHIFT_LEFT:
  793. return U"A << B";
  794. case Variant::OP_SHIFT_RIGHT:
  795. return U"A >> B";
  796. case Variant::OP_BIT_AND:
  797. return U"A & B";
  798. case Variant::OP_BIT_OR:
  799. return U"A | B";
  800. case Variant::OP_BIT_XOR:
  801. return U"A ^ B";
  802. case Variant::OP_BIT_NEGATE:
  803. return U"~A";
  804. // logic
  805. case Variant::OP_AND:
  806. return U"A and B";
  807. case Variant::OP_OR:
  808. return U"A or B";
  809. case Variant::OP_XOR:
  810. return U"A xor B";
  811. case Variant::OP_NOT:
  812. return U"not A";
  813. case Variant::OP_IN:
  814. return U"A in B";
  815. default: {
  816. ERR_FAIL_V_MSG(
  817. U"Unknown node",
  818. U"Unknown node type encountered, caption not available.");
  819. }
  820. }
  821. }
  822. String VisualScriptOperator::get_operator_name(Variant::Operator p_op) {
  823. switch (p_op) {
  824. // comparison
  825. case Variant::OP_EQUAL:
  826. return "Are Equal";
  827. case Variant::OP_NOT_EQUAL:
  828. return "Are Not Equal";
  829. case Variant::OP_LESS:
  830. return "Less Than";
  831. case Variant::OP_LESS_EQUAL:
  832. return "Less Than or Equal";
  833. case Variant::OP_GREATER:
  834. return "Greater Than";
  835. case Variant::OP_GREATER_EQUAL:
  836. return "Greater Than or Equal";
  837. // mathematic
  838. case Variant::OP_ADD:
  839. return "Add";
  840. case Variant::OP_SUBTRACT:
  841. return "Subtract";
  842. case Variant::OP_MULTIPLY:
  843. return "Multiply";
  844. case Variant::OP_DIVIDE:
  845. return "Divide";
  846. case Variant::OP_NEGATE:
  847. return "Negate";
  848. case Variant::OP_POSITIVE:
  849. return "Positive";
  850. case Variant::OP_MODULE:
  851. return "Remainder";
  852. // bitwise
  853. case Variant::OP_SHIFT_LEFT:
  854. return "Bit Shift Left";
  855. case Variant::OP_SHIFT_RIGHT:
  856. return "Bit Shift Right";
  857. case Variant::OP_BIT_AND:
  858. return "Bit And";
  859. case Variant::OP_BIT_OR:
  860. return "Bit Or";
  861. case Variant::OP_BIT_XOR:
  862. return "Bit Xor";
  863. case Variant::OP_BIT_NEGATE:
  864. return "Bit Negate";
  865. // logic
  866. case Variant::OP_AND:
  867. return "And";
  868. case Variant::OP_OR:
  869. return "Or";
  870. case Variant::OP_XOR:
  871. return "Xor";
  872. case Variant::OP_NOT:
  873. return "Not";
  874. case Variant::OP_IN:
  875. return "In";
  876. default: {
  877. ERR_FAIL_INDEX_V(p_op, Variant::OP_MAX, "");
  878. return "Unknown Operator";
  879. }
  880. }
  881. }
  882. void VisualScriptOperator::set_operator(Variant::Operator p_op) {
  883. if (op == p_op) {
  884. return;
  885. }
  886. op = p_op;
  887. ports_changed_notify();
  888. }
  889. Variant::Operator VisualScriptOperator::get_operator() const {
  890. return op;
  891. }
  892. void VisualScriptOperator::set_typed(Variant::Type p_op) {
  893. if (typed == p_op) {
  894. return;
  895. }
  896. typed = p_op;
  897. ports_changed_notify();
  898. }
  899. Variant::Type VisualScriptOperator::get_typed() const {
  900. return typed;
  901. }
  902. void VisualScriptOperator::_bind_methods() {
  903. ClassDB::bind_method(D_METHOD("set_operator", "op"), &VisualScriptOperator::set_operator);
  904. ClassDB::bind_method(D_METHOD("get_operator"), &VisualScriptOperator::get_operator);
  905. ClassDB::bind_method(D_METHOD("set_typed", "type"), &VisualScriptOperator::set_typed);
  906. ClassDB::bind_method(D_METHOD("get_typed"), &VisualScriptOperator::get_typed);
  907. String types;
  908. for (int i = 0; i < Variant::OP_MAX; i++) {
  909. if (i > 0) {
  910. types += ",";
  911. }
  912. types += get_operator_name(static_cast<Variant::Operator>(i));
  913. }
  914. String argt = "Any";
  915. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  916. argt += "," + Variant::get_type_name(Variant::Type(i));
  917. }
  918. ADD_PROPERTY(PropertyInfo(Variant::INT, "operator", PROPERTY_HINT_ENUM, types), "set_operator", "get_operator");
  919. ADD_PROPERTY(PropertyInfo(Variant::INT, "type", PROPERTY_HINT_ENUM, argt), "set_typed", "get_typed");
  920. }
  921. class VisualScriptNodeInstanceOperator : public VisualScriptNodeInstance {
  922. public:
  923. bool unary;
  924. Variant::Operator op;
  925. //virtual int get_working_memory_size() const { return 0; }
  926. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  927. bool valid;
  928. if (unary) {
  929. Variant::evaluate(op, *p_inputs[0], Variant(), *p_outputs[0], valid);
  930. } else {
  931. Variant::evaluate(op, *p_inputs[0], *p_inputs[1], *p_outputs[0], valid);
  932. }
  933. if (!valid) {
  934. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  935. if (p_outputs[0]->get_type() == Variant::STRING) {
  936. r_error_str = *p_outputs[0];
  937. } else {
  938. if (unary) {
  939. r_error_str = String(Variant::get_operator_name(op)) + RTR(": Invalid argument of type: ") + Variant::get_type_name(p_inputs[0]->get_type());
  940. } else {
  941. r_error_str = String(Variant::get_operator_name(op)) + RTR(": Invalid arguments: ") + "A: " + Variant::get_type_name(p_inputs[0]->get_type()) + " B: " + Variant::get_type_name(p_inputs[1]->get_type());
  942. }
  943. }
  944. }
  945. return 0;
  946. }
  947. };
  948. VisualScriptNodeInstance *VisualScriptOperator::instantiate(VisualScriptInstance *p_instance) {
  949. VisualScriptNodeInstanceOperator *instance = memnew(VisualScriptNodeInstanceOperator);
  950. instance->unary = get_input_value_port_count() == 1;
  951. instance->op = op;
  952. return instance;
  953. }
  954. VisualScriptOperator::VisualScriptOperator() {
  955. op = Variant::OP_ADD;
  956. typed = Variant::NIL;
  957. }
  958. template <Variant::Operator OP>
  959. static Ref<VisualScriptNode> create_op_node(const String &p_name) {
  960. Ref<VisualScriptOperator> node;
  961. node.instantiate();
  962. node->set_operator(OP);
  963. return node;
  964. }
  965. //////////////////////////////////////////
  966. ////////////////OPERATOR//////////////////
  967. //////////////////////////////////////////
  968. int VisualScriptSelect::get_output_sequence_port_count() const {
  969. return 0;
  970. }
  971. bool VisualScriptSelect::has_input_sequence_port() const {
  972. return false;
  973. }
  974. int VisualScriptSelect::get_input_value_port_count() const {
  975. return 3;
  976. }
  977. int VisualScriptSelect::get_output_value_port_count() const {
  978. return 1;
  979. }
  980. String VisualScriptSelect::get_output_sequence_port_text(int p_port) const {
  981. return String();
  982. }
  983. PropertyInfo VisualScriptSelect::get_input_value_port_info(int p_idx) const {
  984. if (p_idx == 0) {
  985. return PropertyInfo(Variant::BOOL, "cond");
  986. } else if (p_idx == 1) {
  987. return PropertyInfo(typed, "a");
  988. } else {
  989. return PropertyInfo(typed, "b");
  990. }
  991. }
  992. PropertyInfo VisualScriptSelect::get_output_value_port_info(int p_idx) const {
  993. return PropertyInfo(typed, "out");
  994. }
  995. String VisualScriptSelect::get_caption() const {
  996. return "Select";
  997. }
  998. String VisualScriptSelect::get_text() const {
  999. return "a if cond, else b";
  1000. }
  1001. void VisualScriptSelect::set_typed(Variant::Type p_op) {
  1002. if (typed == p_op) {
  1003. return;
  1004. }
  1005. typed = p_op;
  1006. ports_changed_notify();
  1007. }
  1008. Variant::Type VisualScriptSelect::get_typed() const {
  1009. return typed;
  1010. }
  1011. void VisualScriptSelect::_bind_methods() {
  1012. ClassDB::bind_method(D_METHOD("set_typed", "type"), &VisualScriptSelect::set_typed);
  1013. ClassDB::bind_method(D_METHOD("get_typed"), &VisualScriptSelect::get_typed);
  1014. String argt = "Any";
  1015. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  1016. argt += "," + Variant::get_type_name(Variant::Type(i));
  1017. }
  1018. ADD_PROPERTY(PropertyInfo(Variant::INT, "type", PROPERTY_HINT_ENUM, argt), "set_typed", "get_typed");
  1019. }
  1020. class VisualScriptNodeInstanceSelect : public VisualScriptNodeInstance {
  1021. public:
  1022. //virtual int get_working_memory_size() const { return 0; }
  1023. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1024. bool cond = *p_inputs[0];
  1025. if (cond) {
  1026. *p_outputs[0] = *p_inputs[1];
  1027. } else {
  1028. *p_outputs[0] = *p_inputs[2];
  1029. }
  1030. return 0;
  1031. }
  1032. };
  1033. VisualScriptNodeInstance *VisualScriptSelect::instantiate(VisualScriptInstance *p_instance) {
  1034. VisualScriptNodeInstanceSelect *instance = memnew(VisualScriptNodeInstanceSelect);
  1035. return instance;
  1036. }
  1037. VisualScriptSelect::VisualScriptSelect() {
  1038. typed = Variant::NIL;
  1039. }
  1040. //////////////////////////////////////////
  1041. ////////////////VARIABLE GET//////////////////
  1042. //////////////////////////////////////////
  1043. int VisualScriptVariableGet::get_output_sequence_port_count() const {
  1044. return 0;
  1045. }
  1046. bool VisualScriptVariableGet::has_input_sequence_port() const {
  1047. return false;
  1048. }
  1049. int VisualScriptVariableGet::get_input_value_port_count() const {
  1050. return 0;
  1051. }
  1052. int VisualScriptVariableGet::get_output_value_port_count() const {
  1053. return 1;
  1054. }
  1055. String VisualScriptVariableGet::get_output_sequence_port_text(int p_port) const {
  1056. return String();
  1057. }
  1058. PropertyInfo VisualScriptVariableGet::get_input_value_port_info(int p_idx) const {
  1059. return PropertyInfo();
  1060. }
  1061. PropertyInfo VisualScriptVariableGet::get_output_value_port_info(int p_idx) const {
  1062. PropertyInfo pinfo;
  1063. pinfo.name = "value";
  1064. if (get_visual_script().is_valid() && get_visual_script()->has_variable(variable)) {
  1065. PropertyInfo vinfo = get_visual_script()->get_variable_info(variable);
  1066. pinfo.type = vinfo.type;
  1067. pinfo.hint = vinfo.hint;
  1068. pinfo.hint_string = vinfo.hint_string;
  1069. }
  1070. return pinfo;
  1071. }
  1072. String VisualScriptVariableGet::get_caption() const {
  1073. return "Get " + variable;
  1074. }
  1075. void VisualScriptVariableGet::set_variable(StringName p_variable) {
  1076. if (variable == p_variable) {
  1077. return;
  1078. }
  1079. variable = p_variable;
  1080. ports_changed_notify();
  1081. }
  1082. StringName VisualScriptVariableGet::get_variable() const {
  1083. return variable;
  1084. }
  1085. void VisualScriptVariableGet::_validate_property(PropertyInfo &property) const {
  1086. if (property.name == "var_name" && get_visual_script().is_valid()) {
  1087. Ref<VisualScript> vs = get_visual_script();
  1088. List<StringName> vars;
  1089. vs->get_variable_list(&vars);
  1090. String vhint;
  1091. for (const StringName &E : vars) {
  1092. if (vhint != String()) {
  1093. vhint += ",";
  1094. }
  1095. vhint += E.operator String();
  1096. }
  1097. property.hint = PROPERTY_HINT_ENUM;
  1098. property.hint_string = vhint;
  1099. }
  1100. }
  1101. void VisualScriptVariableGet::_bind_methods() {
  1102. ClassDB::bind_method(D_METHOD("set_variable", "name"), &VisualScriptVariableGet::set_variable);
  1103. ClassDB::bind_method(D_METHOD("get_variable"), &VisualScriptVariableGet::get_variable);
  1104. ADD_PROPERTY(PropertyInfo(Variant::STRING, "var_name"), "set_variable", "get_variable");
  1105. }
  1106. class VisualScriptNodeInstanceVariableGet : public VisualScriptNodeInstance {
  1107. public:
  1108. VisualScriptVariableGet *node;
  1109. VisualScriptInstance *instance;
  1110. StringName variable;
  1111. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1112. if (!instance->get_variable(variable, p_outputs[0])) {
  1113. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  1114. r_error_str = RTR("VariableGet not found in script: ") + "'" + String(variable) + "'";
  1115. return 0;
  1116. }
  1117. return 0;
  1118. }
  1119. };
  1120. VisualScriptNodeInstance *VisualScriptVariableGet::instantiate(VisualScriptInstance *p_instance) {
  1121. VisualScriptNodeInstanceVariableGet *instance = memnew(VisualScriptNodeInstanceVariableGet);
  1122. instance->node = this;
  1123. instance->instance = p_instance;
  1124. instance->variable = variable;
  1125. return instance;
  1126. }
  1127. VisualScriptVariableGet::VisualScriptVariableGet() {
  1128. }
  1129. //////////////////////////////////////////
  1130. ////////////////VARIABLE SET//////////////////
  1131. //////////////////////////////////////////
  1132. int VisualScriptVariableSet::get_output_sequence_port_count() const {
  1133. return 1;
  1134. }
  1135. bool VisualScriptVariableSet::has_input_sequence_port() const {
  1136. return true;
  1137. }
  1138. int VisualScriptVariableSet::get_input_value_port_count() const {
  1139. return 1;
  1140. }
  1141. int VisualScriptVariableSet::get_output_value_port_count() const {
  1142. return 0;
  1143. }
  1144. String VisualScriptVariableSet::get_output_sequence_port_text(int p_port) const {
  1145. return String();
  1146. }
  1147. PropertyInfo VisualScriptVariableSet::get_input_value_port_info(int p_idx) const {
  1148. PropertyInfo pinfo;
  1149. pinfo.name = "set";
  1150. if (get_visual_script().is_valid() && get_visual_script()->has_variable(variable)) {
  1151. PropertyInfo vinfo = get_visual_script()->get_variable_info(variable);
  1152. pinfo.type = vinfo.type;
  1153. pinfo.hint = vinfo.hint;
  1154. pinfo.hint_string = vinfo.hint_string;
  1155. }
  1156. return pinfo;
  1157. }
  1158. PropertyInfo VisualScriptVariableSet::get_output_value_port_info(int p_idx) const {
  1159. return PropertyInfo();
  1160. }
  1161. String VisualScriptVariableSet::get_caption() const {
  1162. return "Set " + variable;
  1163. }
  1164. void VisualScriptVariableSet::set_variable(StringName p_variable) {
  1165. if (variable == p_variable) {
  1166. return;
  1167. }
  1168. variable = p_variable;
  1169. ports_changed_notify();
  1170. }
  1171. StringName VisualScriptVariableSet::get_variable() const {
  1172. return variable;
  1173. }
  1174. void VisualScriptVariableSet::_validate_property(PropertyInfo &property) const {
  1175. if (property.name == "var_name" && get_visual_script().is_valid()) {
  1176. Ref<VisualScript> vs = get_visual_script();
  1177. List<StringName> vars;
  1178. vs->get_variable_list(&vars);
  1179. String vhint;
  1180. for (const StringName &E : vars) {
  1181. if (vhint != String()) {
  1182. vhint += ",";
  1183. }
  1184. vhint += E.operator String();
  1185. }
  1186. property.hint = PROPERTY_HINT_ENUM;
  1187. property.hint_string = vhint;
  1188. }
  1189. }
  1190. void VisualScriptVariableSet::_bind_methods() {
  1191. ClassDB::bind_method(D_METHOD("set_variable", "name"), &VisualScriptVariableSet::set_variable);
  1192. ClassDB::bind_method(D_METHOD("get_variable"), &VisualScriptVariableSet::get_variable);
  1193. ADD_PROPERTY(PropertyInfo(Variant::STRING, "var_name"), "set_variable", "get_variable");
  1194. }
  1195. class VisualScriptNodeInstanceVariableSet : public VisualScriptNodeInstance {
  1196. public:
  1197. VisualScriptVariableSet *node;
  1198. VisualScriptInstance *instance;
  1199. StringName variable;
  1200. //virtual int get_working_memory_size() const { return 0; }
  1201. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1202. if (!instance->set_variable(variable, *p_inputs[0])) {
  1203. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  1204. r_error_str = RTR("VariableSet not found in script: ") + "'" + String(variable) + "'";
  1205. }
  1206. return 0;
  1207. }
  1208. };
  1209. VisualScriptNodeInstance *VisualScriptVariableSet::instantiate(VisualScriptInstance *p_instance) {
  1210. VisualScriptNodeInstanceVariableSet *instance = memnew(VisualScriptNodeInstanceVariableSet);
  1211. instance->node = this;
  1212. instance->instance = p_instance;
  1213. instance->variable = variable;
  1214. return instance;
  1215. }
  1216. VisualScriptVariableSet::VisualScriptVariableSet() {
  1217. }
  1218. //////////////////////////////////////////
  1219. ////////////////CONSTANT//////////////////
  1220. //////////////////////////////////////////
  1221. int VisualScriptConstant::get_output_sequence_port_count() const {
  1222. return 0;
  1223. }
  1224. bool VisualScriptConstant::has_input_sequence_port() const {
  1225. return false;
  1226. }
  1227. int VisualScriptConstant::get_input_value_port_count() const {
  1228. return 0;
  1229. }
  1230. int VisualScriptConstant::get_output_value_port_count() const {
  1231. return 1;
  1232. }
  1233. String VisualScriptConstant::get_output_sequence_port_text(int p_port) const {
  1234. return String();
  1235. }
  1236. PropertyInfo VisualScriptConstant::get_input_value_port_info(int p_idx) const {
  1237. return PropertyInfo();
  1238. }
  1239. PropertyInfo VisualScriptConstant::get_output_value_port_info(int p_idx) const {
  1240. PropertyInfo pinfo;
  1241. pinfo.name = String(value);
  1242. pinfo.type = type;
  1243. return pinfo;
  1244. }
  1245. String VisualScriptConstant::get_caption() const {
  1246. return "Constant";
  1247. }
  1248. void VisualScriptConstant::set_constant_type(Variant::Type p_type) {
  1249. if (type == p_type) {
  1250. return;
  1251. }
  1252. type = p_type;
  1253. Callable::CallError ce;
  1254. Variant::construct(type, value, nullptr, 0, ce);
  1255. ports_changed_notify();
  1256. notify_property_list_changed();
  1257. }
  1258. Variant::Type VisualScriptConstant::get_constant_type() const {
  1259. return type;
  1260. }
  1261. void VisualScriptConstant::set_constant_value(Variant p_value) {
  1262. if (value == p_value) {
  1263. return;
  1264. }
  1265. value = p_value;
  1266. ports_changed_notify();
  1267. }
  1268. Variant VisualScriptConstant::get_constant_value() const {
  1269. return value;
  1270. }
  1271. void VisualScriptConstant::_validate_property(PropertyInfo &property) const {
  1272. if (property.name == "value") {
  1273. property.type = type;
  1274. if (type == Variant::NIL) {
  1275. property.usage = PROPERTY_USAGE_NONE; //do not save if nil
  1276. }
  1277. }
  1278. }
  1279. void VisualScriptConstant::_bind_methods() {
  1280. ClassDB::bind_method(D_METHOD("set_constant_type", "type"), &VisualScriptConstant::set_constant_type);
  1281. ClassDB::bind_method(D_METHOD("get_constant_type"), &VisualScriptConstant::get_constant_type);
  1282. ClassDB::bind_method(D_METHOD("set_constant_value", "value"), &VisualScriptConstant::set_constant_value);
  1283. ClassDB::bind_method(D_METHOD("get_constant_value"), &VisualScriptConstant::get_constant_value);
  1284. String argt = "Null";
  1285. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  1286. argt += "," + Variant::get_type_name(Variant::Type(i));
  1287. }
  1288. ADD_PROPERTY(PropertyInfo(Variant::INT, "type", PROPERTY_HINT_ENUM, argt), "set_constant_type", "get_constant_type");
  1289. ADD_PROPERTY(PropertyInfo(Variant::NIL, "value", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NIL_IS_VARIANT | PROPERTY_USAGE_DEFAULT), "set_constant_value", "get_constant_value");
  1290. }
  1291. class VisualScriptNodeInstanceConstant : public VisualScriptNodeInstance {
  1292. public:
  1293. Variant constant;
  1294. //virtual int get_working_memory_size() const { return 0; }
  1295. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1296. *p_outputs[0] = constant;
  1297. return 0;
  1298. }
  1299. };
  1300. VisualScriptNodeInstance *VisualScriptConstant::instantiate(VisualScriptInstance *p_instance) {
  1301. VisualScriptNodeInstanceConstant *instance = memnew(VisualScriptNodeInstanceConstant);
  1302. instance->constant = value;
  1303. return instance;
  1304. }
  1305. VisualScriptConstant::VisualScriptConstant() {
  1306. type = Variant::NIL;
  1307. }
  1308. //////////////////////////////////////////
  1309. ////////////////PRELOAD//////////////////
  1310. //////////////////////////////////////////
  1311. int VisualScriptPreload::get_output_sequence_port_count() const {
  1312. return 0;
  1313. }
  1314. bool VisualScriptPreload::has_input_sequence_port() const {
  1315. return false;
  1316. }
  1317. int VisualScriptPreload::get_input_value_port_count() const {
  1318. return 0;
  1319. }
  1320. int VisualScriptPreload::get_output_value_port_count() const {
  1321. return 1;
  1322. }
  1323. String VisualScriptPreload::get_output_sequence_port_text(int p_port) const {
  1324. return String();
  1325. }
  1326. PropertyInfo VisualScriptPreload::get_input_value_port_info(int p_idx) const {
  1327. return PropertyInfo();
  1328. }
  1329. PropertyInfo VisualScriptPreload::get_output_value_port_info(int p_idx) const {
  1330. PropertyInfo pinfo;
  1331. pinfo.type = Variant::OBJECT;
  1332. if (preload.is_valid()) {
  1333. pinfo.hint = PROPERTY_HINT_RESOURCE_TYPE;
  1334. pinfo.hint_string = preload->get_class();
  1335. if (preload->get_path().is_resource_file()) {
  1336. pinfo.name = preload->get_path();
  1337. } else if (preload->get_name() != String()) {
  1338. pinfo.name = preload->get_name();
  1339. } else {
  1340. pinfo.name = preload->get_class();
  1341. }
  1342. } else {
  1343. pinfo.name = "<empty>";
  1344. }
  1345. return pinfo;
  1346. }
  1347. String VisualScriptPreload::get_caption() const {
  1348. return "Preload";
  1349. }
  1350. void VisualScriptPreload::set_preload(const Ref<Resource> &p_preload) {
  1351. if (preload == p_preload) {
  1352. return;
  1353. }
  1354. preload = p_preload;
  1355. ports_changed_notify();
  1356. }
  1357. Ref<Resource> VisualScriptPreload::get_preload() const {
  1358. return preload;
  1359. }
  1360. void VisualScriptPreload::_bind_methods() {
  1361. ClassDB::bind_method(D_METHOD("set_preload", "resource"), &VisualScriptPreload::set_preload);
  1362. ClassDB::bind_method(D_METHOD("get_preload"), &VisualScriptPreload::get_preload);
  1363. ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "resource", PROPERTY_HINT_RESOURCE_TYPE, "Resource"), "set_preload", "get_preload");
  1364. }
  1365. class VisualScriptNodeInstancePreload : public VisualScriptNodeInstance {
  1366. public:
  1367. Ref<Resource> preload;
  1368. //virtual int get_working_memory_size() const { return 0; }
  1369. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1370. *p_outputs[0] = preload;
  1371. return 0;
  1372. }
  1373. };
  1374. VisualScriptNodeInstance *VisualScriptPreload::instantiate(VisualScriptInstance *p_instance) {
  1375. VisualScriptNodeInstancePreload *instance = memnew(VisualScriptNodeInstancePreload);
  1376. instance->preload = preload;
  1377. return instance;
  1378. }
  1379. VisualScriptPreload::VisualScriptPreload() {
  1380. }
  1381. //////////////////////////////////////////
  1382. ////////////////INDEX////////////////////
  1383. //////////////////////////////////////////
  1384. int VisualScriptIndexGet::get_output_sequence_port_count() const {
  1385. return 0;
  1386. }
  1387. bool VisualScriptIndexGet::has_input_sequence_port() const {
  1388. return false;
  1389. }
  1390. int VisualScriptIndexGet::get_input_value_port_count() const {
  1391. return 2;
  1392. }
  1393. int VisualScriptIndexGet::get_output_value_port_count() const {
  1394. return 1;
  1395. }
  1396. String VisualScriptIndexGet::get_output_sequence_port_text(int p_port) const {
  1397. return String();
  1398. }
  1399. PropertyInfo VisualScriptIndexGet::get_input_value_port_info(int p_idx) const {
  1400. if (p_idx == 0) {
  1401. return PropertyInfo(Variant::NIL, "base");
  1402. } else {
  1403. return PropertyInfo(Variant::NIL, "index");
  1404. }
  1405. }
  1406. PropertyInfo VisualScriptIndexGet::get_output_value_port_info(int p_idx) const {
  1407. return PropertyInfo();
  1408. }
  1409. String VisualScriptIndexGet::get_caption() const {
  1410. return "Get Index";
  1411. }
  1412. class VisualScriptNodeInstanceIndexGet : public VisualScriptNodeInstance {
  1413. public:
  1414. //virtual int get_working_memory_size() const { return 0; }
  1415. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1416. bool valid;
  1417. *p_outputs[0] = p_inputs[0]->get(*p_inputs[1], &valid);
  1418. if (!valid) {
  1419. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  1420. r_error_str = "Invalid get: " + p_inputs[0]->get_construct_string();
  1421. }
  1422. return 0;
  1423. }
  1424. };
  1425. VisualScriptNodeInstance *VisualScriptIndexGet::instantiate(VisualScriptInstance *p_instance) {
  1426. VisualScriptNodeInstanceIndexGet *instance = memnew(VisualScriptNodeInstanceIndexGet);
  1427. return instance;
  1428. }
  1429. VisualScriptIndexGet::VisualScriptIndexGet() {
  1430. }
  1431. //////////////////////////////////////////
  1432. ////////////////INDEXSET//////////////////
  1433. //////////////////////////////////////////
  1434. int VisualScriptIndexSet::get_output_sequence_port_count() const {
  1435. return 1;
  1436. }
  1437. bool VisualScriptIndexSet::has_input_sequence_port() const {
  1438. return true;
  1439. }
  1440. int VisualScriptIndexSet::get_input_value_port_count() const {
  1441. return 3;
  1442. }
  1443. int VisualScriptIndexSet::get_output_value_port_count() const {
  1444. return 0;
  1445. }
  1446. String VisualScriptIndexSet::get_output_sequence_port_text(int p_port) const {
  1447. return String();
  1448. }
  1449. PropertyInfo VisualScriptIndexSet::get_input_value_port_info(int p_idx) const {
  1450. if (p_idx == 0) {
  1451. return PropertyInfo(Variant::NIL, "base");
  1452. } else if (p_idx == 1) {
  1453. return PropertyInfo(Variant::NIL, "index");
  1454. } else {
  1455. return PropertyInfo(Variant::NIL, "value");
  1456. }
  1457. }
  1458. PropertyInfo VisualScriptIndexSet::get_output_value_port_info(int p_idx) const {
  1459. return PropertyInfo();
  1460. }
  1461. String VisualScriptIndexSet::get_caption() const {
  1462. return "Set Index";
  1463. }
  1464. class VisualScriptNodeInstanceIndexSet : public VisualScriptNodeInstance {
  1465. public:
  1466. //virtual int get_working_memory_size() const { return 0; }
  1467. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1468. bool valid;
  1469. // *p_output[0] points to the same place as *p_inputs[2] so we need a temp to store the value before the change in the next line
  1470. Variant temp = *p_inputs[2];
  1471. *p_outputs[0] = *p_inputs[0];
  1472. p_outputs[0]->set(*p_inputs[1], temp, &valid);
  1473. if (!valid) {
  1474. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  1475. r_error_str = "Invalid set: " + p_inputs[1]->get_construct_string();
  1476. }
  1477. return 0;
  1478. }
  1479. };
  1480. VisualScriptNodeInstance *VisualScriptIndexSet::instantiate(VisualScriptInstance *p_instance) {
  1481. VisualScriptNodeInstanceIndexSet *instance = memnew(VisualScriptNodeInstanceIndexSet);
  1482. return instance;
  1483. }
  1484. VisualScriptIndexSet::VisualScriptIndexSet() {
  1485. }
  1486. //////////////////////////////////////////
  1487. ////////////////GLOBALCONSTANT///////////
  1488. //////////////////////////////////////////
  1489. int VisualScriptGlobalConstant::get_output_sequence_port_count() const {
  1490. return 0;
  1491. }
  1492. bool VisualScriptGlobalConstant::has_input_sequence_port() const {
  1493. return false;
  1494. }
  1495. int VisualScriptGlobalConstant::get_input_value_port_count() const {
  1496. return 0;
  1497. }
  1498. int VisualScriptGlobalConstant::get_output_value_port_count() const {
  1499. return 1;
  1500. }
  1501. String VisualScriptGlobalConstant::get_output_sequence_port_text(int p_port) const {
  1502. return String();
  1503. }
  1504. PropertyInfo VisualScriptGlobalConstant::get_input_value_port_info(int p_idx) const {
  1505. return PropertyInfo();
  1506. }
  1507. PropertyInfo VisualScriptGlobalConstant::get_output_value_port_info(int p_idx) const {
  1508. String name = CoreConstants::get_global_constant_name(index);
  1509. return PropertyInfo(Variant::INT, name);
  1510. }
  1511. String VisualScriptGlobalConstant::get_caption() const {
  1512. return "Global Constant";
  1513. }
  1514. void VisualScriptGlobalConstant::set_global_constant(int p_which) {
  1515. index = p_which;
  1516. notify_property_list_changed();
  1517. ports_changed_notify();
  1518. }
  1519. int VisualScriptGlobalConstant::get_global_constant() {
  1520. return index;
  1521. }
  1522. class VisualScriptNodeInstanceGlobalConstant : public VisualScriptNodeInstance {
  1523. public:
  1524. int index;
  1525. //virtual int get_working_memory_size() const { return 0; }
  1526. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1527. *p_outputs[0] = CoreConstants::get_global_constant_value(index);
  1528. return 0;
  1529. }
  1530. };
  1531. VisualScriptNodeInstance *VisualScriptGlobalConstant::instantiate(VisualScriptInstance *p_instance) {
  1532. VisualScriptNodeInstanceGlobalConstant *instance = memnew(VisualScriptNodeInstanceGlobalConstant);
  1533. instance->index = index;
  1534. return instance;
  1535. }
  1536. void VisualScriptGlobalConstant::_bind_methods() {
  1537. ClassDB::bind_method(D_METHOD("set_global_constant", "index"), &VisualScriptGlobalConstant::set_global_constant);
  1538. ClassDB::bind_method(D_METHOD("get_global_constant"), &VisualScriptGlobalConstant::get_global_constant);
  1539. String cc;
  1540. for (int i = 0; i < CoreConstants::get_global_constant_count(); i++) {
  1541. if (i > 0) {
  1542. cc += ",";
  1543. }
  1544. cc += CoreConstants::get_global_constant_name(i);
  1545. }
  1546. ADD_PROPERTY(PropertyInfo(Variant::INT, "constant", PROPERTY_HINT_ENUM, cc), "set_global_constant", "get_global_constant");
  1547. }
  1548. VisualScriptGlobalConstant::VisualScriptGlobalConstant() {
  1549. index = 0;
  1550. }
  1551. //////////////////////////////////////////
  1552. ////////////////CLASSCONSTANT///////////
  1553. //////////////////////////////////////////
  1554. int VisualScriptClassConstant::get_output_sequence_port_count() const {
  1555. return 0;
  1556. }
  1557. bool VisualScriptClassConstant::has_input_sequence_port() const {
  1558. return false;
  1559. }
  1560. int VisualScriptClassConstant::get_input_value_port_count() const {
  1561. return 0;
  1562. }
  1563. int VisualScriptClassConstant::get_output_value_port_count() const {
  1564. return 1;
  1565. }
  1566. String VisualScriptClassConstant::get_output_sequence_port_text(int p_port) const {
  1567. return String();
  1568. }
  1569. PropertyInfo VisualScriptClassConstant::get_input_value_port_info(int p_idx) const {
  1570. return PropertyInfo();
  1571. }
  1572. PropertyInfo VisualScriptClassConstant::get_output_value_port_info(int p_idx) const {
  1573. if (name == "") {
  1574. return PropertyInfo(Variant::INT, String(base_type));
  1575. } else {
  1576. return PropertyInfo(Variant::INT, String(base_type) + "." + String(name));
  1577. }
  1578. }
  1579. String VisualScriptClassConstant::get_caption() const {
  1580. return "Class Constant";
  1581. }
  1582. void VisualScriptClassConstant::set_class_constant(const StringName &p_which) {
  1583. name = p_which;
  1584. notify_property_list_changed();
  1585. ports_changed_notify();
  1586. }
  1587. StringName VisualScriptClassConstant::get_class_constant() {
  1588. return name;
  1589. }
  1590. void VisualScriptClassConstant::set_base_type(const StringName &p_which) {
  1591. base_type = p_which;
  1592. List<String> constants;
  1593. ClassDB::get_integer_constant_list(base_type, &constants, true);
  1594. if (constants.size() > 0) {
  1595. bool found_name = false;
  1596. for (const String &E : constants) {
  1597. if (E == name) {
  1598. found_name = true;
  1599. break;
  1600. }
  1601. }
  1602. if (!found_name) {
  1603. name = constants[0];
  1604. }
  1605. } else {
  1606. name = "";
  1607. }
  1608. notify_property_list_changed();
  1609. ports_changed_notify();
  1610. }
  1611. StringName VisualScriptClassConstant::get_base_type() {
  1612. return base_type;
  1613. }
  1614. class VisualScriptNodeInstanceClassConstant : public VisualScriptNodeInstance {
  1615. public:
  1616. int value;
  1617. bool valid;
  1618. //virtual int get_working_memory_size() const { return 0; }
  1619. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1620. if (!valid) {
  1621. r_error_str = "Invalid constant name, pick a valid class constant.";
  1622. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  1623. }
  1624. *p_outputs[0] = value;
  1625. return 0;
  1626. }
  1627. };
  1628. VisualScriptNodeInstance *VisualScriptClassConstant::instantiate(VisualScriptInstance *p_instance) {
  1629. VisualScriptNodeInstanceClassConstant *instance = memnew(VisualScriptNodeInstanceClassConstant);
  1630. instance->value = ClassDB::get_integer_constant(base_type, name, &instance->valid);
  1631. return instance;
  1632. }
  1633. void VisualScriptClassConstant::_validate_property(PropertyInfo &property) const {
  1634. if (property.name == "constant") {
  1635. List<String> constants;
  1636. ClassDB::get_integer_constant_list(base_type, &constants, true);
  1637. property.hint_string = "";
  1638. for (const String &E : constants) {
  1639. if (property.hint_string != String()) {
  1640. property.hint_string += ",";
  1641. }
  1642. property.hint_string += E;
  1643. }
  1644. }
  1645. }
  1646. void VisualScriptClassConstant::_bind_methods() {
  1647. ClassDB::bind_method(D_METHOD("set_class_constant", "name"), &VisualScriptClassConstant::set_class_constant);
  1648. ClassDB::bind_method(D_METHOD("get_class_constant"), &VisualScriptClassConstant::get_class_constant);
  1649. ClassDB::bind_method(D_METHOD("set_base_type", "name"), &VisualScriptClassConstant::set_base_type);
  1650. ClassDB::bind_method(D_METHOD("get_base_type"), &VisualScriptClassConstant::get_base_type);
  1651. ADD_PROPERTY(PropertyInfo(Variant::STRING, "base_type", PROPERTY_HINT_TYPE_STRING, "Object"), "set_base_type", "get_base_type");
  1652. ADD_PROPERTY(PropertyInfo(Variant::STRING, "constant", PROPERTY_HINT_ENUM, ""), "set_class_constant", "get_class_constant");
  1653. }
  1654. VisualScriptClassConstant::VisualScriptClassConstant() {
  1655. base_type = "Object";
  1656. }
  1657. //////////////////////////////////////////
  1658. ////////////////BASICTYPECONSTANT///////////
  1659. //////////////////////////////////////////
  1660. int VisualScriptBasicTypeConstant::get_output_sequence_port_count() const {
  1661. return 0;
  1662. }
  1663. bool VisualScriptBasicTypeConstant::has_input_sequence_port() const {
  1664. return false;
  1665. }
  1666. int VisualScriptBasicTypeConstant::get_input_value_port_count() const {
  1667. return 0;
  1668. }
  1669. int VisualScriptBasicTypeConstant::get_output_value_port_count() const {
  1670. return 1;
  1671. }
  1672. String VisualScriptBasicTypeConstant::get_output_sequence_port_text(int p_port) const {
  1673. return String();
  1674. }
  1675. PropertyInfo VisualScriptBasicTypeConstant::get_input_value_port_info(int p_idx) const {
  1676. return PropertyInfo();
  1677. }
  1678. PropertyInfo VisualScriptBasicTypeConstant::get_output_value_port_info(int p_idx) const {
  1679. return PropertyInfo(type, "value");
  1680. }
  1681. String VisualScriptBasicTypeConstant::get_caption() const {
  1682. return "Basic Constant";
  1683. }
  1684. String VisualScriptBasicTypeConstant::get_text() const {
  1685. if (name == "") {
  1686. return Variant::get_type_name(type);
  1687. } else {
  1688. return Variant::get_type_name(type) + "." + String(name);
  1689. }
  1690. }
  1691. void VisualScriptBasicTypeConstant::set_basic_type_constant(const StringName &p_which) {
  1692. name = p_which;
  1693. notify_property_list_changed();
  1694. ports_changed_notify();
  1695. }
  1696. StringName VisualScriptBasicTypeConstant::get_basic_type_constant() const {
  1697. return name;
  1698. }
  1699. void VisualScriptBasicTypeConstant::set_basic_type(Variant::Type p_which) {
  1700. type = p_which;
  1701. List<StringName> constants;
  1702. Variant::get_constants_for_type(type, &constants);
  1703. if (constants.size() > 0) {
  1704. bool found_name = false;
  1705. for (const StringName &E : constants) {
  1706. if (E == name) {
  1707. found_name = true;
  1708. break;
  1709. }
  1710. }
  1711. if (!found_name) {
  1712. name = constants[0];
  1713. }
  1714. } else {
  1715. name = "";
  1716. }
  1717. notify_property_list_changed();
  1718. ports_changed_notify();
  1719. }
  1720. Variant::Type VisualScriptBasicTypeConstant::get_basic_type() const {
  1721. return type;
  1722. }
  1723. class VisualScriptNodeInstanceBasicTypeConstant : public VisualScriptNodeInstance {
  1724. public:
  1725. Variant value;
  1726. bool valid;
  1727. //virtual int get_working_memory_size() const { return 0; }
  1728. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1729. if (!valid) {
  1730. r_error_str = "Invalid constant name, pick a valid basic type constant.";
  1731. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  1732. }
  1733. *p_outputs[0] = value;
  1734. return 0;
  1735. }
  1736. };
  1737. VisualScriptNodeInstance *VisualScriptBasicTypeConstant::instantiate(VisualScriptInstance *p_instance) {
  1738. VisualScriptNodeInstanceBasicTypeConstant *instance = memnew(VisualScriptNodeInstanceBasicTypeConstant);
  1739. instance->value = Variant::get_constant_value(type, name, &instance->valid);
  1740. return instance;
  1741. }
  1742. void VisualScriptBasicTypeConstant::_validate_property(PropertyInfo &property) const {
  1743. if (property.name == "constant") {
  1744. List<StringName> constants;
  1745. Variant::get_constants_for_type(type, &constants);
  1746. if (constants.size() == 0) {
  1747. property.usage = PROPERTY_USAGE_NONE;
  1748. return;
  1749. }
  1750. property.hint_string = "";
  1751. for (const StringName &E : constants) {
  1752. if (property.hint_string != String()) {
  1753. property.hint_string += ",";
  1754. }
  1755. property.hint_string += String(E);
  1756. }
  1757. }
  1758. }
  1759. void VisualScriptBasicTypeConstant::_bind_methods() {
  1760. ClassDB::bind_method(D_METHOD("set_basic_type", "name"), &VisualScriptBasicTypeConstant::set_basic_type);
  1761. ClassDB::bind_method(D_METHOD("get_basic_type"), &VisualScriptBasicTypeConstant::get_basic_type);
  1762. ClassDB::bind_method(D_METHOD("set_basic_type_constant", "name"), &VisualScriptBasicTypeConstant::set_basic_type_constant);
  1763. ClassDB::bind_method(D_METHOD("get_basic_type_constant"), &VisualScriptBasicTypeConstant::get_basic_type_constant);
  1764. String argt = "Null";
  1765. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  1766. argt += "," + Variant::get_type_name(Variant::Type(i));
  1767. }
  1768. ADD_PROPERTY(PropertyInfo(Variant::INT, "basic_type", PROPERTY_HINT_ENUM, argt), "set_basic_type", "get_basic_type");
  1769. ADD_PROPERTY(PropertyInfo(Variant::STRING, "constant", PROPERTY_HINT_ENUM, ""), "set_basic_type_constant", "get_basic_type_constant");
  1770. }
  1771. VisualScriptBasicTypeConstant::VisualScriptBasicTypeConstant() {
  1772. type = Variant::NIL;
  1773. }
  1774. //////////////////////////////////////////
  1775. ////////////////MATHCONSTANT///////////
  1776. //////////////////////////////////////////
  1777. const char *VisualScriptMathConstant::const_name[MATH_CONSTANT_MAX] = {
  1778. "One",
  1779. "PI",
  1780. "PI/2",
  1781. "TAU",
  1782. "E",
  1783. "Sqrt2",
  1784. "INF",
  1785. "NAN"
  1786. };
  1787. double VisualScriptMathConstant::const_value[MATH_CONSTANT_MAX] = {
  1788. 1.0,
  1789. Math_PI,
  1790. Math_PI * 0.5,
  1791. Math_TAU,
  1792. 2.71828182845904523536,
  1793. Math::sqrt(2.0),
  1794. INFINITY,
  1795. NAN
  1796. };
  1797. int VisualScriptMathConstant::get_output_sequence_port_count() const {
  1798. return 0;
  1799. }
  1800. bool VisualScriptMathConstant::has_input_sequence_port() const {
  1801. return false;
  1802. }
  1803. int VisualScriptMathConstant::get_input_value_port_count() const {
  1804. return 0;
  1805. }
  1806. int VisualScriptMathConstant::get_output_value_port_count() const {
  1807. return 1;
  1808. }
  1809. String VisualScriptMathConstant::get_output_sequence_port_text(int p_port) const {
  1810. return String();
  1811. }
  1812. PropertyInfo VisualScriptMathConstant::get_input_value_port_info(int p_idx) const {
  1813. return PropertyInfo();
  1814. }
  1815. PropertyInfo VisualScriptMathConstant::get_output_value_port_info(int p_idx) const {
  1816. return PropertyInfo(Variant::FLOAT, const_name[constant]);
  1817. }
  1818. String VisualScriptMathConstant::get_caption() const {
  1819. return "Math Constant";
  1820. }
  1821. void VisualScriptMathConstant::set_math_constant(MathConstant p_which) {
  1822. constant = p_which;
  1823. notify_property_list_changed();
  1824. ports_changed_notify();
  1825. }
  1826. VisualScriptMathConstant::MathConstant VisualScriptMathConstant::get_math_constant() {
  1827. return constant;
  1828. }
  1829. class VisualScriptNodeInstanceMathConstant : public VisualScriptNodeInstance {
  1830. public:
  1831. float value;
  1832. //virtual int get_working_memory_size() const { return 0; }
  1833. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1834. *p_outputs[0] = value;
  1835. return 0;
  1836. }
  1837. };
  1838. VisualScriptNodeInstance *VisualScriptMathConstant::instantiate(VisualScriptInstance *p_instance) {
  1839. VisualScriptNodeInstanceMathConstant *instance = memnew(VisualScriptNodeInstanceMathConstant);
  1840. instance->value = const_value[constant];
  1841. return instance;
  1842. }
  1843. void VisualScriptMathConstant::_bind_methods() {
  1844. ClassDB::bind_method(D_METHOD("set_math_constant", "which"), &VisualScriptMathConstant::set_math_constant);
  1845. ClassDB::bind_method(D_METHOD("get_math_constant"), &VisualScriptMathConstant::get_math_constant);
  1846. String cc;
  1847. for (int i = 0; i < MATH_CONSTANT_MAX; i++) {
  1848. if (i > 0) {
  1849. cc += ",";
  1850. }
  1851. cc += const_name[i];
  1852. }
  1853. ADD_PROPERTY(PropertyInfo(Variant::INT, "constant", PROPERTY_HINT_ENUM, cc), "set_math_constant", "get_math_constant");
  1854. BIND_ENUM_CONSTANT(MATH_CONSTANT_ONE);
  1855. BIND_ENUM_CONSTANT(MATH_CONSTANT_PI);
  1856. BIND_ENUM_CONSTANT(MATH_CONSTANT_HALF_PI);
  1857. BIND_ENUM_CONSTANT(MATH_CONSTANT_TAU);
  1858. BIND_ENUM_CONSTANT(MATH_CONSTANT_E);
  1859. BIND_ENUM_CONSTANT(MATH_CONSTANT_SQRT2);
  1860. BIND_ENUM_CONSTANT(MATH_CONSTANT_INF);
  1861. BIND_ENUM_CONSTANT(MATH_CONSTANT_NAN);
  1862. BIND_ENUM_CONSTANT(MATH_CONSTANT_MAX);
  1863. }
  1864. VisualScriptMathConstant::VisualScriptMathConstant() {
  1865. constant = MATH_CONSTANT_ONE;
  1866. }
  1867. //////////////////////////////////////////
  1868. ////////////////ENGINESINGLETON///////////
  1869. //////////////////////////////////////////
  1870. int VisualScriptEngineSingleton::get_output_sequence_port_count() const {
  1871. return 0;
  1872. }
  1873. bool VisualScriptEngineSingleton::has_input_sequence_port() const {
  1874. return false;
  1875. }
  1876. int VisualScriptEngineSingleton::get_input_value_port_count() const {
  1877. return 0;
  1878. }
  1879. int VisualScriptEngineSingleton::get_output_value_port_count() const {
  1880. return 1;
  1881. }
  1882. String VisualScriptEngineSingleton::get_output_sequence_port_text(int p_port) const {
  1883. return String();
  1884. }
  1885. PropertyInfo VisualScriptEngineSingleton::get_input_value_port_info(int p_idx) const {
  1886. return PropertyInfo();
  1887. }
  1888. PropertyInfo VisualScriptEngineSingleton::get_output_value_port_info(int p_idx) const {
  1889. return PropertyInfo(Variant::OBJECT, singleton);
  1890. }
  1891. String VisualScriptEngineSingleton::get_caption() const {
  1892. return "Get Engine Singleton";
  1893. }
  1894. void VisualScriptEngineSingleton::set_singleton(const String &p_string) {
  1895. singleton = p_string;
  1896. notify_property_list_changed();
  1897. ports_changed_notify();
  1898. }
  1899. String VisualScriptEngineSingleton::get_singleton() {
  1900. return singleton;
  1901. }
  1902. class VisualScriptNodeInstanceEngineSingleton : public VisualScriptNodeInstance {
  1903. public:
  1904. Object *singleton;
  1905. //virtual int get_working_memory_size() const { return 0; }
  1906. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1907. *p_outputs[0] = singleton;
  1908. return 0;
  1909. }
  1910. };
  1911. VisualScriptNodeInstance *VisualScriptEngineSingleton::instantiate(VisualScriptInstance *p_instance) {
  1912. VisualScriptNodeInstanceEngineSingleton *instance = memnew(VisualScriptNodeInstanceEngineSingleton);
  1913. instance->singleton = Engine::get_singleton()->get_singleton_object(singleton);
  1914. return instance;
  1915. }
  1916. VisualScriptEngineSingleton::TypeGuess VisualScriptEngineSingleton::guess_output_type(TypeGuess *p_inputs, int p_output) const {
  1917. Object *obj = Engine::get_singleton()->get_singleton_object(singleton);
  1918. TypeGuess tg;
  1919. tg.type = Variant::OBJECT;
  1920. if (obj) {
  1921. tg.gdclass = obj->get_class();
  1922. tg.script = obj->get_script();
  1923. }
  1924. return tg;
  1925. }
  1926. void VisualScriptEngineSingleton::_validate_property(PropertyInfo &property) const {
  1927. String cc;
  1928. List<Engine::Singleton> singletons;
  1929. Engine::get_singleton()->get_singletons(&singletons);
  1930. for (const Engine::Singleton &E : singletons) {
  1931. if (E.name == "VS" || E.name == "PS" || E.name == "PS2D" || E.name == "AS" || E.name == "TS" || E.name == "SS" || E.name == "SS2D") {
  1932. continue; //skip these, too simple named
  1933. }
  1934. if (cc != String()) {
  1935. cc += ",";
  1936. }
  1937. cc += E.name;
  1938. }
  1939. property.hint = PROPERTY_HINT_ENUM;
  1940. property.hint_string = cc;
  1941. }
  1942. void VisualScriptEngineSingleton::_bind_methods() {
  1943. ClassDB::bind_method(D_METHOD("set_singleton", "name"), &VisualScriptEngineSingleton::set_singleton);
  1944. ClassDB::bind_method(D_METHOD("get_singleton"), &VisualScriptEngineSingleton::get_singleton);
  1945. ADD_PROPERTY(PropertyInfo(Variant::STRING, "constant"), "set_singleton", "get_singleton");
  1946. }
  1947. VisualScriptEngineSingleton::VisualScriptEngineSingleton() {
  1948. singleton = String();
  1949. }
  1950. //////////////////////////////////////////
  1951. ////////////////GETNODE///////////
  1952. //////////////////////////////////////////
  1953. int VisualScriptSceneNode::get_output_sequence_port_count() const {
  1954. return 0;
  1955. }
  1956. bool VisualScriptSceneNode::has_input_sequence_port() const {
  1957. return false;
  1958. }
  1959. int VisualScriptSceneNode::get_input_value_port_count() const {
  1960. return 0;
  1961. }
  1962. int VisualScriptSceneNode::get_output_value_port_count() const {
  1963. return 1;
  1964. }
  1965. String VisualScriptSceneNode::get_output_sequence_port_text(int p_port) const {
  1966. return String();
  1967. }
  1968. PropertyInfo VisualScriptSceneNode::get_input_value_port_info(int p_idx) const {
  1969. return PropertyInfo();
  1970. }
  1971. PropertyInfo VisualScriptSceneNode::get_output_value_port_info(int p_idx) const {
  1972. return PropertyInfo(Variant::OBJECT, path.simplified());
  1973. }
  1974. String VisualScriptSceneNode::get_caption() const {
  1975. return "Get Scene Node";
  1976. }
  1977. void VisualScriptSceneNode::set_node_path(const NodePath &p_path) {
  1978. path = p_path;
  1979. notify_property_list_changed();
  1980. ports_changed_notify();
  1981. }
  1982. NodePath VisualScriptSceneNode::get_node_path() {
  1983. return path;
  1984. }
  1985. class VisualScriptNodeInstanceSceneNode : public VisualScriptNodeInstance {
  1986. public:
  1987. VisualScriptSceneNode *node;
  1988. VisualScriptInstance *instance;
  1989. NodePath path;
  1990. //virtual int get_working_memory_size() const { return 0; }
  1991. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  1992. Node *node = Object::cast_to<Node>(instance->get_owner_ptr());
  1993. if (!node) {
  1994. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  1995. r_error_str = "Base object is not a Node!";
  1996. return 0;
  1997. }
  1998. Node *another = node->get_node(path);
  1999. if (!another) {
  2000. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  2001. r_error_str = "Path does not lead Node!";
  2002. return 0;
  2003. }
  2004. *p_outputs[0] = another;
  2005. return 0;
  2006. }
  2007. };
  2008. VisualScriptNodeInstance *VisualScriptSceneNode::instantiate(VisualScriptInstance *p_instance) {
  2009. VisualScriptNodeInstanceSceneNode *instance = memnew(VisualScriptNodeInstanceSceneNode);
  2010. instance->node = this;
  2011. instance->instance = p_instance;
  2012. instance->path = path;
  2013. return instance;
  2014. }
  2015. #ifdef TOOLS_ENABLED
  2016. static Node *_find_script_node(Node *p_edited_scene, Node *p_current_node, const Ref<Script> &script) {
  2017. if (p_edited_scene != p_current_node && p_current_node->get_owner() != p_edited_scene) {
  2018. return nullptr;
  2019. }
  2020. Ref<Script> scr = p_current_node->get_script();
  2021. if (scr.is_valid() && scr == script) {
  2022. return p_current_node;
  2023. }
  2024. for (int i = 0; i < p_current_node->get_child_count(); i++) {
  2025. Node *n = _find_script_node(p_edited_scene, p_current_node->get_child(i), script);
  2026. if (n) {
  2027. return n;
  2028. }
  2029. }
  2030. return nullptr;
  2031. }
  2032. #endif
  2033. VisualScriptSceneNode::TypeGuess VisualScriptSceneNode::guess_output_type(TypeGuess *p_inputs, int p_output) const {
  2034. VisualScriptSceneNode::TypeGuess tg;
  2035. tg.type = Variant::OBJECT;
  2036. tg.gdclass = "Node";
  2037. #ifdef TOOLS_ENABLED
  2038. Ref<Script> script = get_visual_script();
  2039. if (!script.is_valid()) {
  2040. return tg;
  2041. }
  2042. MainLoop *main_loop = OS::get_singleton()->get_main_loop();
  2043. SceneTree *scene_tree = Object::cast_to<SceneTree>(main_loop);
  2044. if (!scene_tree) {
  2045. return tg;
  2046. }
  2047. Node *edited_scene = scene_tree->get_edited_scene_root();
  2048. if (!edited_scene) {
  2049. return tg;
  2050. }
  2051. Node *script_node = _find_script_node(edited_scene, edited_scene, script);
  2052. if (!script_node) {
  2053. return tg;
  2054. }
  2055. Node *another = script_node->get_node(path);
  2056. if (another) {
  2057. tg.gdclass = another->get_class();
  2058. tg.script = another->get_script();
  2059. }
  2060. #endif
  2061. return tg;
  2062. }
  2063. void VisualScriptSceneNode::_validate_property(PropertyInfo &property) const {
  2064. #ifdef TOOLS_ENABLED
  2065. if (property.name == "node_path") {
  2066. Ref<Script> script = get_visual_script();
  2067. if (!script.is_valid()) {
  2068. return;
  2069. }
  2070. MainLoop *main_loop = OS::get_singleton()->get_main_loop();
  2071. SceneTree *scene_tree = Object::cast_to<SceneTree>(main_loop);
  2072. if (!scene_tree) {
  2073. return;
  2074. }
  2075. Node *edited_scene = scene_tree->get_edited_scene_root();
  2076. if (!edited_scene) {
  2077. return;
  2078. }
  2079. Node *script_node = _find_script_node(edited_scene, edited_scene, script);
  2080. if (!script_node) {
  2081. return;
  2082. }
  2083. property.hint_string = script_node->get_path();
  2084. }
  2085. #endif
  2086. }
  2087. void VisualScriptSceneNode::_bind_methods() {
  2088. ClassDB::bind_method(D_METHOD("set_node_path", "path"), &VisualScriptSceneNode::set_node_path);
  2089. ClassDB::bind_method(D_METHOD("get_node_path"), &VisualScriptSceneNode::get_node_path);
  2090. ADD_PROPERTY(PropertyInfo(Variant::NODE_PATH, "node_path", PROPERTY_HINT_NODE_PATH_TO_EDITED_NODE), "set_node_path", "get_node_path");
  2091. }
  2092. VisualScriptSceneNode::VisualScriptSceneNode() {
  2093. path = String(".");
  2094. }
  2095. //////////////////////////////////////////
  2096. ////////////////SceneTree///////////
  2097. //////////////////////////////////////////
  2098. int VisualScriptSceneTree::get_output_sequence_port_count() const {
  2099. return 0;
  2100. }
  2101. bool VisualScriptSceneTree::has_input_sequence_port() const {
  2102. return false;
  2103. }
  2104. int VisualScriptSceneTree::get_input_value_port_count() const {
  2105. return 0;
  2106. }
  2107. int VisualScriptSceneTree::get_output_value_port_count() const {
  2108. return 1;
  2109. }
  2110. String VisualScriptSceneTree::get_output_sequence_port_text(int p_port) const {
  2111. return String();
  2112. }
  2113. PropertyInfo VisualScriptSceneTree::get_input_value_port_info(int p_idx) const {
  2114. return PropertyInfo();
  2115. }
  2116. PropertyInfo VisualScriptSceneTree::get_output_value_port_info(int p_idx) const {
  2117. return PropertyInfo(Variant::OBJECT, "Scene Tree", PROPERTY_HINT_TYPE_STRING, "SceneTree");
  2118. }
  2119. String VisualScriptSceneTree::get_caption() const {
  2120. return "Get Scene Tree";
  2121. }
  2122. class VisualScriptNodeInstanceSceneTree : public VisualScriptNodeInstance {
  2123. public:
  2124. VisualScriptSceneTree *node;
  2125. VisualScriptInstance *instance;
  2126. //virtual int get_working_memory_size() const { return 0; }
  2127. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2128. Node *node = Object::cast_to<Node>(instance->get_owner_ptr());
  2129. if (!node) {
  2130. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  2131. r_error_str = "Base object is not a Node!";
  2132. return 0;
  2133. }
  2134. SceneTree *tree = node->get_tree();
  2135. if (!tree) {
  2136. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  2137. r_error_str = "Attempt to get SceneTree while node is not in the active tree.";
  2138. return 0;
  2139. }
  2140. *p_outputs[0] = tree;
  2141. return 0;
  2142. }
  2143. };
  2144. VisualScriptNodeInstance *VisualScriptSceneTree::instantiate(VisualScriptInstance *p_instance) {
  2145. VisualScriptNodeInstanceSceneTree *instance = memnew(VisualScriptNodeInstanceSceneTree);
  2146. instance->node = this;
  2147. instance->instance = p_instance;
  2148. return instance;
  2149. }
  2150. VisualScriptSceneTree::TypeGuess VisualScriptSceneTree::guess_output_type(TypeGuess *p_inputs, int p_output) const {
  2151. TypeGuess tg;
  2152. tg.type = Variant::OBJECT;
  2153. tg.gdclass = "SceneTree";
  2154. return tg;
  2155. }
  2156. void VisualScriptSceneTree::_validate_property(PropertyInfo &property) const {
  2157. }
  2158. void VisualScriptSceneTree::_bind_methods() {
  2159. }
  2160. VisualScriptSceneTree::VisualScriptSceneTree() {
  2161. }
  2162. //////////////////////////////////////////
  2163. ////////////////RESPATH///////////
  2164. //////////////////////////////////////////
  2165. int VisualScriptResourcePath::get_output_sequence_port_count() const {
  2166. return 0;
  2167. }
  2168. bool VisualScriptResourcePath::has_input_sequence_port() const {
  2169. return false;
  2170. }
  2171. int VisualScriptResourcePath::get_input_value_port_count() const {
  2172. return 0;
  2173. }
  2174. int VisualScriptResourcePath::get_output_value_port_count() const {
  2175. return 1;
  2176. }
  2177. String VisualScriptResourcePath::get_output_sequence_port_text(int p_port) const {
  2178. return String();
  2179. }
  2180. PropertyInfo VisualScriptResourcePath::get_input_value_port_info(int p_idx) const {
  2181. return PropertyInfo();
  2182. }
  2183. PropertyInfo VisualScriptResourcePath::get_output_value_port_info(int p_idx) const {
  2184. return PropertyInfo(Variant::STRING, path);
  2185. }
  2186. String VisualScriptResourcePath::get_caption() const {
  2187. return "Resource Path";
  2188. }
  2189. void VisualScriptResourcePath::set_resource_path(const String &p_path) {
  2190. path = p_path;
  2191. notify_property_list_changed();
  2192. ports_changed_notify();
  2193. }
  2194. String VisualScriptResourcePath::get_resource_path() {
  2195. return path;
  2196. }
  2197. class VisualScriptNodeInstanceResourcePath : public VisualScriptNodeInstance {
  2198. public:
  2199. String path;
  2200. //virtual int get_working_memory_size() const { return 0; }
  2201. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2202. *p_outputs[0] = path;
  2203. return 0;
  2204. }
  2205. };
  2206. VisualScriptNodeInstance *VisualScriptResourcePath::instantiate(VisualScriptInstance *p_instance) {
  2207. VisualScriptNodeInstanceResourcePath *instance = memnew(VisualScriptNodeInstanceResourcePath);
  2208. instance->path = path;
  2209. return instance;
  2210. }
  2211. void VisualScriptResourcePath::_bind_methods() {
  2212. ClassDB::bind_method(D_METHOD("set_resource_path", "path"), &VisualScriptResourcePath::set_resource_path);
  2213. ClassDB::bind_method(D_METHOD("get_resource_path"), &VisualScriptResourcePath::get_resource_path);
  2214. ADD_PROPERTY(PropertyInfo(Variant::STRING, "path", PROPERTY_HINT_FILE), "set_resource_path", "get_resource_path");
  2215. }
  2216. VisualScriptResourcePath::VisualScriptResourcePath() {
  2217. path = "";
  2218. }
  2219. //////////////////////////////////////////
  2220. ////////////////SELF///////////
  2221. //////////////////////////////////////////
  2222. int VisualScriptSelf::get_output_sequence_port_count() const {
  2223. return 0;
  2224. }
  2225. bool VisualScriptSelf::has_input_sequence_port() const {
  2226. return false;
  2227. }
  2228. int VisualScriptSelf::get_input_value_port_count() const {
  2229. return 0;
  2230. }
  2231. int VisualScriptSelf::get_output_value_port_count() const {
  2232. return 1;
  2233. }
  2234. String VisualScriptSelf::get_output_sequence_port_text(int p_port) const {
  2235. return String();
  2236. }
  2237. PropertyInfo VisualScriptSelf::get_input_value_port_info(int p_idx) const {
  2238. return PropertyInfo();
  2239. }
  2240. PropertyInfo VisualScriptSelf::get_output_value_port_info(int p_idx) const {
  2241. String type_name;
  2242. if (get_visual_script().is_valid()) {
  2243. type_name = get_visual_script()->get_instance_base_type();
  2244. } else {
  2245. type_name = "instance";
  2246. }
  2247. return PropertyInfo(Variant::OBJECT, type_name);
  2248. }
  2249. String VisualScriptSelf::get_caption() const {
  2250. return "Get Self";
  2251. }
  2252. class VisualScriptNodeInstanceSelf : public VisualScriptNodeInstance {
  2253. public:
  2254. VisualScriptInstance *instance;
  2255. //virtual int get_working_memory_size() const { return 0; }
  2256. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2257. *p_outputs[0] = instance->get_owner_ptr();
  2258. return 0;
  2259. }
  2260. };
  2261. VisualScriptNodeInstance *VisualScriptSelf::instantiate(VisualScriptInstance *p_instance) {
  2262. VisualScriptNodeInstanceSelf *instance = memnew(VisualScriptNodeInstanceSelf);
  2263. instance->instance = p_instance;
  2264. return instance;
  2265. }
  2266. VisualScriptSelf::TypeGuess VisualScriptSelf::guess_output_type(TypeGuess *p_inputs, int p_output) const {
  2267. VisualScriptSceneNode::TypeGuess tg;
  2268. tg.type = Variant::OBJECT;
  2269. tg.gdclass = "Object";
  2270. Ref<Script> script = get_visual_script();
  2271. if (!script.is_valid()) {
  2272. return tg;
  2273. }
  2274. tg.gdclass = script->get_instance_base_type();
  2275. tg.script = script;
  2276. return tg;
  2277. }
  2278. void VisualScriptSelf::_bind_methods() {
  2279. }
  2280. VisualScriptSelf::VisualScriptSelf() {
  2281. }
  2282. //////////////////////////////////////////
  2283. ////////////////CUSTOM (SCRIPTED)///////////
  2284. //////////////////////////////////////////
  2285. int VisualScriptCustomNode::get_output_sequence_port_count() const {
  2286. int ret;
  2287. if (GDVIRTUAL_CALL(_get_output_sequence_port_count, ret)) {
  2288. return ret;
  2289. }
  2290. return 0;
  2291. }
  2292. bool VisualScriptCustomNode::has_input_sequence_port() const {
  2293. bool ret;
  2294. if (GDVIRTUAL_CALL(_has_input_sequence_port, ret)) {
  2295. return ret;
  2296. }
  2297. return false;
  2298. }
  2299. int VisualScriptCustomNode::get_input_value_port_count() const {
  2300. int ret;
  2301. if (GDVIRTUAL_CALL(_get_input_value_port_count, ret)) {
  2302. return ret;
  2303. }
  2304. return 0;
  2305. }
  2306. int VisualScriptCustomNode::get_output_value_port_count() const {
  2307. int ret;
  2308. if (GDVIRTUAL_CALL(_get_output_value_port_count, ret)) {
  2309. return ret;
  2310. }
  2311. return 0;
  2312. }
  2313. String VisualScriptCustomNode::get_output_sequence_port_text(int p_port) const {
  2314. String ret;
  2315. if (GDVIRTUAL_CALL(_get_output_sequence_port_text, p_port, ret)) {
  2316. return ret;
  2317. }
  2318. return String();
  2319. }
  2320. PropertyInfo VisualScriptCustomNode::get_input_value_port_info(int p_idx) const {
  2321. PropertyInfo info;
  2322. {
  2323. int type;
  2324. if (GDVIRTUAL_CALL(_get_input_value_port_type, p_idx, type)) {
  2325. info.type = Variant::Type(type);
  2326. }
  2327. }
  2328. {
  2329. String name;
  2330. if (GDVIRTUAL_CALL(_get_input_value_port_name, p_idx, name)) {
  2331. info.name = name;
  2332. }
  2333. }
  2334. {
  2335. int hint;
  2336. if (GDVIRTUAL_CALL(_get_input_value_port_hint, p_idx, hint)) {
  2337. info.hint = PropertyHint(hint);
  2338. }
  2339. }
  2340. {
  2341. String hint_string;
  2342. if (GDVIRTUAL_CALL(_get_input_value_port_hint_string, p_idx, hint_string)) {
  2343. info.hint_string = hint_string;
  2344. }
  2345. }
  2346. return info;
  2347. }
  2348. PropertyInfo VisualScriptCustomNode::get_output_value_port_info(int p_idx) const {
  2349. PropertyInfo info;
  2350. {
  2351. int type;
  2352. if (GDVIRTUAL_CALL(_get_output_value_port_type, p_idx, type)) {
  2353. info.type = Variant::Type(type);
  2354. }
  2355. }
  2356. {
  2357. String name;
  2358. if (GDVIRTUAL_CALL(_get_output_value_port_name, p_idx, name)) {
  2359. info.name = name;
  2360. }
  2361. }
  2362. {
  2363. int hint;
  2364. if (GDVIRTUAL_CALL(_get_output_value_port_hint, p_idx, hint)) {
  2365. info.hint = PropertyHint(hint);
  2366. }
  2367. }
  2368. {
  2369. String hint_string;
  2370. if (GDVIRTUAL_CALL(_get_output_value_port_hint_string, p_idx, hint_string)) {
  2371. info.hint_string = hint_string;
  2372. }
  2373. }
  2374. return info;
  2375. }
  2376. VisualScriptCustomNode::TypeGuess VisualScriptCustomNode::guess_output_type(TypeGuess *p_inputs, int p_output) const {
  2377. TypeGuess tg;
  2378. PropertyInfo pi = VisualScriptCustomNode::get_output_value_port_info(p_output);
  2379. tg.type = pi.type;
  2380. if (pi.type == Variant::OBJECT) {
  2381. if (pi.hint == PROPERTY_HINT_RESOURCE_TYPE) {
  2382. if (pi.hint_string.is_resource_file()) {
  2383. tg.script = ResourceLoader::load(pi.hint_string);
  2384. } else if (ClassDB::class_exists(pi.hint_string)) {
  2385. tg.gdclass = pi.hint_string;
  2386. }
  2387. }
  2388. }
  2389. return tg;
  2390. }
  2391. String VisualScriptCustomNode::get_caption() const {
  2392. String ret;
  2393. if (GDVIRTUAL_CALL(_get_caption, ret)) {
  2394. return ret;
  2395. }
  2396. return "CustomNode";
  2397. }
  2398. String VisualScriptCustomNode::get_text() const {
  2399. String ret;
  2400. if (GDVIRTUAL_CALL(_get_text, ret)) {
  2401. return ret;
  2402. }
  2403. return "";
  2404. }
  2405. String VisualScriptCustomNode::get_category() const {
  2406. String ret;
  2407. if (GDVIRTUAL_CALL(_get_category, ret)) {
  2408. return ret;
  2409. }
  2410. return "Custom";
  2411. }
  2412. class VisualScriptNodeInstanceCustomNode : public VisualScriptNodeInstance {
  2413. public:
  2414. VisualScriptInstance *instance;
  2415. VisualScriptCustomNode *node;
  2416. int in_count;
  2417. int out_count;
  2418. int work_mem_size;
  2419. virtual int get_working_memory_size() const { return work_mem_size; }
  2420. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2421. if (GDVIRTUAL_IS_OVERRIDDEN_PTR(node, _step)) {
  2422. Array in_values;
  2423. Array out_values;
  2424. Array work_mem;
  2425. in_values.resize(in_count);
  2426. for (int i = 0; i < in_count; i++) {
  2427. in_values[i] = *p_inputs[i];
  2428. }
  2429. out_values.resize(out_count);
  2430. work_mem.resize(work_mem_size);
  2431. for (int i = 0; i < work_mem_size; i++) {
  2432. work_mem[i] = p_working_mem[i];
  2433. }
  2434. int ret_out;
  2435. Variant ret;
  2436. GDVIRTUAL_CALL_PTR(node, _step, in_values, out_values, p_start_mode, work_mem, ret);
  2437. if (ret.get_type() == Variant::STRING) {
  2438. r_error_str = ret;
  2439. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  2440. return 0;
  2441. } else if (ret.is_num()) {
  2442. ret_out = ret;
  2443. } else {
  2444. r_error_str = RTR("Invalid return value from _step(), must be integer (seq out), or string (error).");
  2445. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  2446. return 0;
  2447. }
  2448. for (int i = 0; i < out_count; i++) {
  2449. if (i < out_values.size()) {
  2450. *p_outputs[i] = out_values[i];
  2451. }
  2452. }
  2453. for (int i = 0; i < work_mem_size; i++) {
  2454. if (i < work_mem.size()) {
  2455. p_working_mem[i] = work_mem[i];
  2456. }
  2457. }
  2458. return ret_out;
  2459. } else {
  2460. r_error_str = RTR("Custom node has no _step() method, can't process graph.");
  2461. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  2462. }
  2463. return 0;
  2464. }
  2465. };
  2466. VisualScriptNodeInstance *VisualScriptCustomNode::instantiate(VisualScriptInstance *p_instance) {
  2467. VisualScriptNodeInstanceCustomNode *instance = memnew(VisualScriptNodeInstanceCustomNode);
  2468. instance->instance = p_instance;
  2469. instance->node = this;
  2470. instance->in_count = get_input_value_port_count();
  2471. instance->out_count = get_output_value_port_count();
  2472. instance->work_mem_size = 0;
  2473. GDVIRTUAL_CALL(_get_working_memory_size, instance->work_mem_size);
  2474. return instance;
  2475. }
  2476. void VisualScriptCustomNode::_script_changed() {
  2477. call_deferred(SNAME("ports_changed_notify"));
  2478. }
  2479. void VisualScriptCustomNode::_bind_methods() {
  2480. GDVIRTUAL_BIND(_get_output_sequence_port_count);
  2481. GDVIRTUAL_BIND(_has_input_sequence_port);
  2482. GDVIRTUAL_BIND(_get_output_sequence_port_text, "seq_idx");
  2483. GDVIRTUAL_BIND(_get_input_value_port_count);
  2484. GDVIRTUAL_BIND(_get_input_value_port_type, "input_idx");
  2485. GDVIRTUAL_BIND(_get_input_value_port_name, "input_idx");
  2486. GDVIRTUAL_BIND(_get_input_value_port_hint, "input_idx");
  2487. GDVIRTUAL_BIND(_get_input_value_port_hint_string, "input_idx");
  2488. GDVIRTUAL_BIND(_get_output_value_port_count);
  2489. GDVIRTUAL_BIND(_get_output_value_port_type, "output_idx");
  2490. GDVIRTUAL_BIND(_get_output_value_port_name, "output_idx");
  2491. GDVIRTUAL_BIND(_get_output_value_port_hint, "output_idx");
  2492. GDVIRTUAL_BIND(_get_output_value_port_hint_string, "output_idx");
  2493. GDVIRTUAL_BIND(_get_caption);
  2494. GDVIRTUAL_BIND(_get_text);
  2495. GDVIRTUAL_BIND(_get_category);
  2496. GDVIRTUAL_BIND(_get_working_memory_size);
  2497. GDVIRTUAL_BIND(_step, "inputs", "outputs", "start_mode", "working_mem");
  2498. BIND_ENUM_CONSTANT(START_MODE_BEGIN_SEQUENCE);
  2499. BIND_ENUM_CONSTANT(START_MODE_CONTINUE_SEQUENCE);
  2500. BIND_ENUM_CONSTANT(START_MODE_RESUME_YIELD);
  2501. BIND_CONSTANT(STEP_PUSH_STACK_BIT);
  2502. BIND_CONSTANT(STEP_GO_BACK_BIT);
  2503. BIND_CONSTANT(STEP_NO_ADVANCE_BIT);
  2504. BIND_CONSTANT(STEP_EXIT_FUNCTION_BIT);
  2505. BIND_CONSTANT(STEP_YIELD_BIT);
  2506. }
  2507. VisualScriptCustomNode::VisualScriptCustomNode() {
  2508. connect("script_changed", callable_mp(this, &VisualScriptCustomNode::_script_changed));
  2509. }
  2510. //////////////////////////////////////////
  2511. ////////////////SUBCALL///////////
  2512. //////////////////////////////////////////
  2513. int VisualScriptSubCall::get_output_sequence_port_count() const {
  2514. return 1;
  2515. }
  2516. bool VisualScriptSubCall::has_input_sequence_port() const {
  2517. return true;
  2518. }
  2519. int VisualScriptSubCall::get_input_value_port_count() const {
  2520. Ref<Script> script = get_script();
  2521. if (script.is_valid() && script->has_method(VisualScriptLanguage::singleton->_subcall)) {
  2522. MethodInfo mi = script->get_method_info(VisualScriptLanguage::singleton->_subcall);
  2523. return mi.arguments.size();
  2524. }
  2525. return 0;
  2526. }
  2527. int VisualScriptSubCall::get_output_value_port_count() const {
  2528. return 1;
  2529. }
  2530. String VisualScriptSubCall::get_output_sequence_port_text(int p_port) const {
  2531. return String();
  2532. }
  2533. PropertyInfo VisualScriptSubCall::get_input_value_port_info(int p_idx) const {
  2534. Ref<Script> script = get_script();
  2535. if (script.is_valid() && script->has_method(VisualScriptLanguage::singleton->_subcall)) {
  2536. MethodInfo mi = script->get_method_info(VisualScriptLanguage::singleton->_subcall);
  2537. return mi.arguments[p_idx];
  2538. }
  2539. return PropertyInfo();
  2540. }
  2541. PropertyInfo VisualScriptSubCall::get_output_value_port_info(int p_idx) const {
  2542. Ref<Script> script = get_script();
  2543. if (script.is_valid() && script->has_method(VisualScriptLanguage::singleton->_subcall)) {
  2544. MethodInfo mi = script->get_method_info(VisualScriptLanguage::singleton->_subcall);
  2545. return mi.return_val;
  2546. }
  2547. return PropertyInfo();
  2548. }
  2549. String VisualScriptSubCall::get_caption() const {
  2550. return "SubCall";
  2551. }
  2552. String VisualScriptSubCall::get_text() const {
  2553. Ref<Script> script = get_script();
  2554. if (script.is_valid()) {
  2555. if (script->get_name() != String()) {
  2556. return script->get_name();
  2557. }
  2558. if (script->get_path().is_resource_file()) {
  2559. return script->get_path().get_file();
  2560. }
  2561. return script->get_class();
  2562. }
  2563. return "";
  2564. }
  2565. String VisualScriptSubCall::get_category() const {
  2566. return "custom";
  2567. }
  2568. class VisualScriptNodeInstanceSubCall : public VisualScriptNodeInstance {
  2569. public:
  2570. VisualScriptInstance *instance;
  2571. VisualScriptSubCall *subcall;
  2572. int input_args;
  2573. bool valid;
  2574. //virtual int get_working_memory_size() const { return 0; }
  2575. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2576. if (!valid) {
  2577. r_error_str = "Node requires a script with a _subcall(<args>) method to work.";
  2578. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  2579. return 0;
  2580. }
  2581. *p_outputs[0] = subcall->call(VisualScriptLanguage::singleton->_subcall, p_inputs, input_args, r_error);
  2582. return 0;
  2583. }
  2584. };
  2585. VisualScriptNodeInstance *VisualScriptSubCall::instantiate(VisualScriptInstance *p_instance) {
  2586. VisualScriptNodeInstanceSubCall *instance = memnew(VisualScriptNodeInstanceSubCall);
  2587. instance->instance = p_instance;
  2588. Ref<Script> script = get_script();
  2589. if (script.is_valid() && script->has_method(VisualScriptLanguage::singleton->_subcall)) {
  2590. instance->valid = true;
  2591. instance->input_args = get_input_value_port_count();
  2592. } else {
  2593. instance->valid = false;
  2594. }
  2595. return instance;
  2596. }
  2597. void VisualScriptSubCall::_bind_methods() {
  2598. // Since this is script only, registering virtual function is no longer valid. Will have to go in docs.
  2599. }
  2600. VisualScriptSubCall::VisualScriptSubCall() {
  2601. }
  2602. //////////////////////////////////////////
  2603. ////////////////Comment///////////
  2604. //////////////////////////////////////////
  2605. int VisualScriptComment::get_output_sequence_port_count() const {
  2606. return 0;
  2607. }
  2608. bool VisualScriptComment::has_input_sequence_port() const {
  2609. return false;
  2610. }
  2611. int VisualScriptComment::get_input_value_port_count() const {
  2612. return 0;
  2613. }
  2614. int VisualScriptComment::get_output_value_port_count() const {
  2615. return 0;
  2616. }
  2617. String VisualScriptComment::get_output_sequence_port_text(int p_port) const {
  2618. return String();
  2619. }
  2620. PropertyInfo VisualScriptComment::get_input_value_port_info(int p_idx) const {
  2621. return PropertyInfo();
  2622. }
  2623. PropertyInfo VisualScriptComment::get_output_value_port_info(int p_idx) const {
  2624. return PropertyInfo();
  2625. }
  2626. String VisualScriptComment::get_caption() const {
  2627. return title;
  2628. }
  2629. String VisualScriptComment::get_text() const {
  2630. return description;
  2631. }
  2632. void VisualScriptComment::set_title(const String &p_title) {
  2633. if (title == p_title) {
  2634. return;
  2635. }
  2636. title = p_title;
  2637. ports_changed_notify();
  2638. }
  2639. String VisualScriptComment::get_title() const {
  2640. return title;
  2641. }
  2642. void VisualScriptComment::set_description(const String &p_description) {
  2643. if (description == p_description) {
  2644. return;
  2645. }
  2646. description = p_description;
  2647. ports_changed_notify();
  2648. }
  2649. String VisualScriptComment::get_description() const {
  2650. return description;
  2651. }
  2652. void VisualScriptComment::set_size(const Size2 &p_size) {
  2653. if (size == p_size) {
  2654. return;
  2655. }
  2656. size = p_size;
  2657. ports_changed_notify();
  2658. }
  2659. Size2 VisualScriptComment::get_size() const {
  2660. return size;
  2661. }
  2662. String VisualScriptComment::get_category() const {
  2663. return "data";
  2664. }
  2665. class VisualScriptNodeInstanceComment : public VisualScriptNodeInstance {
  2666. public:
  2667. VisualScriptInstance *instance;
  2668. //virtual int get_working_memory_size() const { return 0; }
  2669. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2670. return 0;
  2671. }
  2672. };
  2673. VisualScriptNodeInstance *VisualScriptComment::instantiate(VisualScriptInstance *p_instance) {
  2674. VisualScriptNodeInstanceComment *instance = memnew(VisualScriptNodeInstanceComment);
  2675. instance->instance = p_instance;
  2676. return instance;
  2677. }
  2678. void VisualScriptComment::_bind_methods() {
  2679. ClassDB::bind_method(D_METHOD("set_title", "title"), &VisualScriptComment::set_title);
  2680. ClassDB::bind_method(D_METHOD("get_title"), &VisualScriptComment::get_title);
  2681. ClassDB::bind_method(D_METHOD("set_description", "description"), &VisualScriptComment::set_description);
  2682. ClassDB::bind_method(D_METHOD("get_description"), &VisualScriptComment::get_description);
  2683. ClassDB::bind_method(D_METHOD("set_size", "size"), &VisualScriptComment::set_size);
  2684. ClassDB::bind_method(D_METHOD("get_size"), &VisualScriptComment::get_size);
  2685. ADD_PROPERTY(PropertyInfo(Variant::STRING, "title"), "set_title", "get_title");
  2686. ADD_PROPERTY(PropertyInfo(Variant::STRING, "description", PROPERTY_HINT_MULTILINE_TEXT), "set_description", "get_description");
  2687. ADD_PROPERTY(PropertyInfo(Variant::VECTOR2, "size"), "set_size", "get_size");
  2688. }
  2689. VisualScriptComment::VisualScriptComment() {
  2690. title = "Comment";
  2691. size = Size2(150, 150);
  2692. }
  2693. //////////////////////////////////////////
  2694. ////////////////Constructor///////////
  2695. //////////////////////////////////////////
  2696. int VisualScriptConstructor::get_output_sequence_port_count() const {
  2697. return 0;
  2698. }
  2699. bool VisualScriptConstructor::has_input_sequence_port() const {
  2700. return false;
  2701. }
  2702. int VisualScriptConstructor::get_input_value_port_count() const {
  2703. return constructor.arguments.size();
  2704. }
  2705. int VisualScriptConstructor::get_output_value_port_count() const {
  2706. return 1;
  2707. }
  2708. String VisualScriptConstructor::get_output_sequence_port_text(int p_port) const {
  2709. return "";
  2710. }
  2711. PropertyInfo VisualScriptConstructor::get_input_value_port_info(int p_idx) const {
  2712. return constructor.arguments[p_idx];
  2713. }
  2714. PropertyInfo VisualScriptConstructor::get_output_value_port_info(int p_idx) const {
  2715. return PropertyInfo(type, "value");
  2716. }
  2717. String VisualScriptConstructor::get_caption() const {
  2718. return "Construct " + Variant::get_type_name(type);
  2719. }
  2720. String VisualScriptConstructor::get_category() const {
  2721. return "functions";
  2722. }
  2723. void VisualScriptConstructor::set_constructor_type(Variant::Type p_type) {
  2724. if (type == p_type) {
  2725. return;
  2726. }
  2727. type = p_type;
  2728. ports_changed_notify();
  2729. }
  2730. Variant::Type VisualScriptConstructor::get_constructor_type() const {
  2731. return type;
  2732. }
  2733. void VisualScriptConstructor::set_constructor(const Dictionary &p_info) {
  2734. constructor = MethodInfo::from_dict(p_info);
  2735. ports_changed_notify();
  2736. }
  2737. Dictionary VisualScriptConstructor::get_constructor() const {
  2738. return constructor;
  2739. }
  2740. class VisualScriptNodeInstanceConstructor : public VisualScriptNodeInstance {
  2741. public:
  2742. VisualScriptInstance *instance;
  2743. Variant::Type type;
  2744. int argcount;
  2745. //virtual int get_working_memory_size() const { return 0; }
  2746. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2747. Callable::CallError ce;
  2748. Variant::construct(type, *p_outputs[0], p_inputs, argcount, ce);
  2749. if (ce.error != Callable::CallError::CALL_OK) {
  2750. r_error_str = "Invalid arguments for constructor";
  2751. }
  2752. return 0;
  2753. }
  2754. };
  2755. VisualScriptNodeInstance *VisualScriptConstructor::instantiate(VisualScriptInstance *p_instance) {
  2756. VisualScriptNodeInstanceConstructor *instance = memnew(VisualScriptNodeInstanceConstructor);
  2757. instance->instance = p_instance;
  2758. instance->type = type;
  2759. instance->argcount = constructor.arguments.size();
  2760. return instance;
  2761. }
  2762. void VisualScriptConstructor::_bind_methods() {
  2763. ClassDB::bind_method(D_METHOD("set_constructor_type", "type"), &VisualScriptConstructor::set_constructor_type);
  2764. ClassDB::bind_method(D_METHOD("get_constructor_type"), &VisualScriptConstructor::get_constructor_type);
  2765. ClassDB::bind_method(D_METHOD("set_constructor", "constructor"), &VisualScriptConstructor::set_constructor);
  2766. ClassDB::bind_method(D_METHOD("get_constructor"), &VisualScriptConstructor::get_constructor);
  2767. ADD_PROPERTY(PropertyInfo(Variant::INT, "type", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR | PROPERTY_USAGE_INTERNAL), "set_constructor_type", "get_constructor_type");
  2768. ADD_PROPERTY(PropertyInfo(Variant::DICTIONARY, "constructor", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR | PROPERTY_USAGE_INTERNAL), "set_constructor", "get_constructor");
  2769. }
  2770. VisualScriptConstructor::VisualScriptConstructor() {
  2771. type = Variant::NIL;
  2772. }
  2773. static Map<String, Pair<Variant::Type, MethodInfo>> constructor_map;
  2774. static Ref<VisualScriptNode> create_constructor_node(const String &p_name) {
  2775. ERR_FAIL_COND_V(!constructor_map.has(p_name), Ref<VisualScriptNode>());
  2776. Ref<VisualScriptConstructor> vsc;
  2777. vsc.instantiate();
  2778. vsc->set_constructor_type(constructor_map[p_name].first);
  2779. vsc->set_constructor(constructor_map[p_name].second);
  2780. return vsc;
  2781. }
  2782. //////////////////////////////////////////
  2783. ////////////////LocalVar///////////
  2784. //////////////////////////////////////////
  2785. int VisualScriptLocalVar::get_output_sequence_port_count() const {
  2786. return 0;
  2787. }
  2788. bool VisualScriptLocalVar::has_input_sequence_port() const {
  2789. return false;
  2790. }
  2791. int VisualScriptLocalVar::get_input_value_port_count() const {
  2792. return 0;
  2793. }
  2794. int VisualScriptLocalVar::get_output_value_port_count() const {
  2795. return 1;
  2796. }
  2797. String VisualScriptLocalVar::get_output_sequence_port_text(int p_port) const {
  2798. return "";
  2799. }
  2800. PropertyInfo VisualScriptLocalVar::get_input_value_port_info(int p_idx) const {
  2801. return PropertyInfo();
  2802. }
  2803. PropertyInfo VisualScriptLocalVar::get_output_value_port_info(int p_idx) const {
  2804. return PropertyInfo(type, name);
  2805. }
  2806. String VisualScriptLocalVar::get_caption() const {
  2807. return "Get Local Var";
  2808. }
  2809. String VisualScriptLocalVar::get_category() const {
  2810. return "data";
  2811. }
  2812. void VisualScriptLocalVar::set_var_name(const StringName &p_name) {
  2813. if (name == p_name) {
  2814. return;
  2815. }
  2816. name = p_name;
  2817. ports_changed_notify();
  2818. }
  2819. StringName VisualScriptLocalVar::get_var_name() const {
  2820. return name;
  2821. }
  2822. void VisualScriptLocalVar::set_var_type(Variant::Type p_type) {
  2823. type = p_type;
  2824. ports_changed_notify();
  2825. }
  2826. Variant::Type VisualScriptLocalVar::get_var_type() const {
  2827. return type;
  2828. }
  2829. class VisualScriptNodeInstanceLocalVar : public VisualScriptNodeInstance {
  2830. public:
  2831. VisualScriptInstance *instance;
  2832. StringName name;
  2833. virtual int get_working_memory_size() const { return 1; }
  2834. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2835. *p_outputs[0] = *p_working_mem;
  2836. return 0;
  2837. }
  2838. };
  2839. VisualScriptNodeInstance *VisualScriptLocalVar::instantiate(VisualScriptInstance *p_instance) {
  2840. VisualScriptNodeInstanceLocalVar *instance = memnew(VisualScriptNodeInstanceLocalVar);
  2841. instance->instance = p_instance;
  2842. instance->name = name;
  2843. return instance;
  2844. }
  2845. void VisualScriptLocalVar::_bind_methods() {
  2846. ClassDB::bind_method(D_METHOD("set_var_name", "name"), &VisualScriptLocalVar::set_var_name);
  2847. ClassDB::bind_method(D_METHOD("get_var_name"), &VisualScriptLocalVar::get_var_name);
  2848. ClassDB::bind_method(D_METHOD("set_var_type", "type"), &VisualScriptLocalVar::set_var_type);
  2849. ClassDB::bind_method(D_METHOD("get_var_type"), &VisualScriptLocalVar::get_var_type);
  2850. String argt = "Any";
  2851. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  2852. argt += "," + Variant::get_type_name(Variant::Type(i));
  2853. }
  2854. ADD_PROPERTY(PropertyInfo(Variant::STRING, "var_name"), "set_var_name", "get_var_name");
  2855. ADD_PROPERTY(PropertyInfo(Variant::INT, "type", PROPERTY_HINT_ENUM, argt), "set_var_type", "get_var_type");
  2856. }
  2857. VisualScriptLocalVar::VisualScriptLocalVar() {
  2858. name = "new_local";
  2859. type = Variant::NIL;
  2860. }
  2861. //////////////////////////////////////////
  2862. ////////////////LocalVar///////////
  2863. //////////////////////////////////////////
  2864. int VisualScriptLocalVarSet::get_output_sequence_port_count() const {
  2865. return 1;
  2866. }
  2867. bool VisualScriptLocalVarSet::has_input_sequence_port() const {
  2868. return true;
  2869. }
  2870. int VisualScriptLocalVarSet::get_input_value_port_count() const {
  2871. return 1;
  2872. }
  2873. int VisualScriptLocalVarSet::get_output_value_port_count() const {
  2874. return 1;
  2875. }
  2876. String VisualScriptLocalVarSet::get_output_sequence_port_text(int p_port) const {
  2877. return "";
  2878. }
  2879. PropertyInfo VisualScriptLocalVarSet::get_input_value_port_info(int p_idx) const {
  2880. return PropertyInfo(type, "set");
  2881. }
  2882. PropertyInfo VisualScriptLocalVarSet::get_output_value_port_info(int p_idx) const {
  2883. return PropertyInfo(type, "get");
  2884. }
  2885. String VisualScriptLocalVarSet::get_caption() const {
  2886. return "Set Local Var";
  2887. }
  2888. String VisualScriptLocalVarSet::get_text() const {
  2889. return name;
  2890. }
  2891. String VisualScriptLocalVarSet::get_category() const {
  2892. return "data";
  2893. }
  2894. void VisualScriptLocalVarSet::set_var_name(const StringName &p_name) {
  2895. if (name == p_name) {
  2896. return;
  2897. }
  2898. name = p_name;
  2899. ports_changed_notify();
  2900. }
  2901. StringName VisualScriptLocalVarSet::get_var_name() const {
  2902. return name;
  2903. }
  2904. void VisualScriptLocalVarSet::set_var_type(Variant::Type p_type) {
  2905. type = p_type;
  2906. ports_changed_notify();
  2907. }
  2908. Variant::Type VisualScriptLocalVarSet::get_var_type() const {
  2909. return type;
  2910. }
  2911. class VisualScriptNodeInstanceLocalVarSet : public VisualScriptNodeInstance {
  2912. public:
  2913. VisualScriptInstance *instance;
  2914. StringName name;
  2915. virtual int get_working_memory_size() const { return 1; }
  2916. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  2917. *p_working_mem = *p_inputs[0];
  2918. *p_outputs[0] = *p_working_mem;
  2919. return 0;
  2920. }
  2921. };
  2922. VisualScriptNodeInstance *VisualScriptLocalVarSet::instantiate(VisualScriptInstance *p_instance) {
  2923. VisualScriptNodeInstanceLocalVarSet *instance = memnew(VisualScriptNodeInstanceLocalVarSet);
  2924. instance->instance = p_instance;
  2925. instance->name = name;
  2926. return instance;
  2927. }
  2928. void VisualScriptLocalVarSet::_bind_methods() {
  2929. ClassDB::bind_method(D_METHOD("set_var_name", "name"), &VisualScriptLocalVarSet::set_var_name);
  2930. ClassDB::bind_method(D_METHOD("get_var_name"), &VisualScriptLocalVarSet::get_var_name);
  2931. ClassDB::bind_method(D_METHOD("set_var_type", "type"), &VisualScriptLocalVarSet::set_var_type);
  2932. ClassDB::bind_method(D_METHOD("get_var_type"), &VisualScriptLocalVarSet::get_var_type);
  2933. String argt = "Any";
  2934. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  2935. argt += "," + Variant::get_type_name(Variant::Type(i));
  2936. }
  2937. ADD_PROPERTY(PropertyInfo(Variant::STRING, "var_name"), "set_var_name", "get_var_name");
  2938. ADD_PROPERTY(PropertyInfo(Variant::INT, "type", PROPERTY_HINT_ENUM, argt), "set_var_type", "get_var_type");
  2939. }
  2940. VisualScriptLocalVarSet::VisualScriptLocalVarSet() {
  2941. name = "new_local";
  2942. type = Variant::NIL;
  2943. }
  2944. //////////////////////////////////////////
  2945. ////////////////LocalVar///////////
  2946. //////////////////////////////////////////
  2947. int VisualScriptInputAction::get_output_sequence_port_count() const {
  2948. return 0;
  2949. }
  2950. bool VisualScriptInputAction::has_input_sequence_port() const {
  2951. return false;
  2952. }
  2953. int VisualScriptInputAction::get_input_value_port_count() const {
  2954. return 0;
  2955. }
  2956. int VisualScriptInputAction::get_output_value_port_count() const {
  2957. return 1;
  2958. }
  2959. String VisualScriptInputAction::get_output_sequence_port_text(int p_port) const {
  2960. return "";
  2961. }
  2962. PropertyInfo VisualScriptInputAction::get_input_value_port_info(int p_idx) const {
  2963. return PropertyInfo();
  2964. }
  2965. PropertyInfo VisualScriptInputAction::get_output_value_port_info(int p_idx) const {
  2966. String mstr;
  2967. switch (mode) {
  2968. case MODE_PRESSED: {
  2969. mstr = "pressed";
  2970. } break;
  2971. case MODE_RELEASED: {
  2972. mstr = "not pressed";
  2973. } break;
  2974. case MODE_JUST_PRESSED: {
  2975. mstr = "just pressed";
  2976. } break;
  2977. case MODE_JUST_RELEASED: {
  2978. mstr = "just released";
  2979. } break;
  2980. }
  2981. return PropertyInfo(Variant::BOOL, mstr);
  2982. }
  2983. String VisualScriptInputAction::get_caption() const {
  2984. return "Action " + name;
  2985. }
  2986. String VisualScriptInputAction::get_category() const {
  2987. return "data";
  2988. }
  2989. void VisualScriptInputAction::set_action_name(const StringName &p_name) {
  2990. if (name == p_name) {
  2991. return;
  2992. }
  2993. name = p_name;
  2994. ports_changed_notify();
  2995. }
  2996. StringName VisualScriptInputAction::get_action_name() const {
  2997. return name;
  2998. }
  2999. void VisualScriptInputAction::set_action_mode(Mode p_mode) {
  3000. if (mode == p_mode) {
  3001. return;
  3002. }
  3003. mode = p_mode;
  3004. ports_changed_notify();
  3005. }
  3006. VisualScriptInputAction::Mode VisualScriptInputAction::get_action_mode() const {
  3007. return mode;
  3008. }
  3009. class VisualScriptNodeInstanceInputAction : public VisualScriptNodeInstance {
  3010. public:
  3011. VisualScriptInstance *instance;
  3012. StringName action;
  3013. VisualScriptInputAction::Mode mode;
  3014. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  3015. switch (mode) {
  3016. case VisualScriptInputAction::MODE_PRESSED: {
  3017. *p_outputs[0] = Input::get_singleton()->is_action_pressed(action);
  3018. } break;
  3019. case VisualScriptInputAction::MODE_RELEASED: {
  3020. *p_outputs[0] = !Input::get_singleton()->is_action_pressed(action);
  3021. } break;
  3022. case VisualScriptInputAction::MODE_JUST_PRESSED: {
  3023. *p_outputs[0] = Input::get_singleton()->is_action_just_pressed(action);
  3024. } break;
  3025. case VisualScriptInputAction::MODE_JUST_RELEASED: {
  3026. *p_outputs[0] = Input::get_singleton()->is_action_just_released(action);
  3027. } break;
  3028. }
  3029. return 0;
  3030. }
  3031. };
  3032. VisualScriptNodeInstance *VisualScriptInputAction::instantiate(VisualScriptInstance *p_instance) {
  3033. VisualScriptNodeInstanceInputAction *instance = memnew(VisualScriptNodeInstanceInputAction);
  3034. instance->instance = p_instance;
  3035. instance->action = name;
  3036. instance->mode = mode;
  3037. return instance;
  3038. }
  3039. void VisualScriptInputAction::_validate_property(PropertyInfo &property) const {
  3040. if (property.name == "action") {
  3041. property.hint = PROPERTY_HINT_ENUM;
  3042. String actions;
  3043. List<PropertyInfo> pinfo;
  3044. ProjectSettings::get_singleton()->get_property_list(&pinfo);
  3045. Vector<String> al;
  3046. for (const PropertyInfo &pi : pinfo) {
  3047. if (!pi.name.begins_with("input/")) {
  3048. continue;
  3049. }
  3050. String name = pi.name.substr(pi.name.find("/") + 1, pi.name.length());
  3051. al.push_back(name);
  3052. }
  3053. al.sort();
  3054. for (int i = 0; i < al.size(); i++) {
  3055. if (actions != String()) {
  3056. actions += ",";
  3057. }
  3058. actions += al[i];
  3059. }
  3060. property.hint_string = actions;
  3061. }
  3062. }
  3063. void VisualScriptInputAction::_bind_methods() {
  3064. ClassDB::bind_method(D_METHOD("set_action_name", "name"), &VisualScriptInputAction::set_action_name);
  3065. ClassDB::bind_method(D_METHOD("get_action_name"), &VisualScriptInputAction::get_action_name);
  3066. ClassDB::bind_method(D_METHOD("set_action_mode", "mode"), &VisualScriptInputAction::set_action_mode);
  3067. ClassDB::bind_method(D_METHOD("get_action_mode"), &VisualScriptInputAction::get_action_mode);
  3068. ADD_PROPERTY(PropertyInfo(Variant::STRING, "action"), "set_action_name", "get_action_name");
  3069. ADD_PROPERTY(PropertyInfo(Variant::INT, "mode", PROPERTY_HINT_ENUM, "Pressed,Released,JustPressed,JustReleased"), "set_action_mode", "get_action_mode");
  3070. BIND_ENUM_CONSTANT(MODE_PRESSED);
  3071. BIND_ENUM_CONSTANT(MODE_RELEASED);
  3072. BIND_ENUM_CONSTANT(MODE_JUST_PRESSED);
  3073. BIND_ENUM_CONSTANT(MODE_JUST_RELEASED);
  3074. }
  3075. VisualScriptInputAction::VisualScriptInputAction() {
  3076. name = "";
  3077. mode = MODE_PRESSED;
  3078. }
  3079. //////////////////////////////////////////
  3080. ////////////////Constructor///////////
  3081. //////////////////////////////////////////
  3082. int VisualScriptDeconstruct::get_output_sequence_port_count() const {
  3083. return 0;
  3084. }
  3085. bool VisualScriptDeconstruct::has_input_sequence_port() const {
  3086. return false;
  3087. }
  3088. int VisualScriptDeconstruct::get_input_value_port_count() const {
  3089. return 1;
  3090. }
  3091. int VisualScriptDeconstruct::get_output_value_port_count() const {
  3092. return elements.size();
  3093. }
  3094. String VisualScriptDeconstruct::get_output_sequence_port_text(int p_port) const {
  3095. return "";
  3096. }
  3097. PropertyInfo VisualScriptDeconstruct::get_input_value_port_info(int p_idx) const {
  3098. return PropertyInfo(type, "value");
  3099. }
  3100. PropertyInfo VisualScriptDeconstruct::get_output_value_port_info(int p_idx) const {
  3101. return PropertyInfo(elements[p_idx].type, elements[p_idx].name);
  3102. }
  3103. String VisualScriptDeconstruct::get_caption() const {
  3104. return "Deconstruct " + Variant::get_type_name(type);
  3105. }
  3106. String VisualScriptDeconstruct::get_category() const {
  3107. return "functions";
  3108. }
  3109. void VisualScriptDeconstruct::_update_elements() {
  3110. elements.clear();
  3111. Variant v;
  3112. Callable::CallError ce;
  3113. Variant::construct(type, v, nullptr, 0, ce);
  3114. List<PropertyInfo> pinfo;
  3115. v.get_property_list(&pinfo);
  3116. for (const PropertyInfo &E : pinfo) {
  3117. Element e;
  3118. e.name = E.name;
  3119. e.type = E.type;
  3120. elements.push_back(e);
  3121. }
  3122. }
  3123. void VisualScriptDeconstruct::set_deconstruct_type(Variant::Type p_type) {
  3124. if (type == p_type) {
  3125. return;
  3126. }
  3127. type = p_type;
  3128. _update_elements();
  3129. ports_changed_notify();
  3130. notify_property_list_changed(); //to make input appear/disappear
  3131. }
  3132. Variant::Type VisualScriptDeconstruct::get_deconstruct_type() const {
  3133. return type;
  3134. }
  3135. void VisualScriptDeconstruct::_set_elem_cache(const Array &p_elements) {
  3136. ERR_FAIL_COND(p_elements.size() % 2 == 1);
  3137. elements.resize(p_elements.size() / 2);
  3138. for (int i = 0; i < elements.size(); i++) {
  3139. elements.write[i].name = p_elements[i * 2 + 0];
  3140. elements.write[i].type = Variant::Type(int(p_elements[i * 2 + 1]));
  3141. }
  3142. }
  3143. Array VisualScriptDeconstruct::_get_elem_cache() const {
  3144. Array ret;
  3145. for (int i = 0; i < elements.size(); i++) {
  3146. ret.push_back(elements[i].name);
  3147. ret.push_back(elements[i].type);
  3148. }
  3149. return ret;
  3150. }
  3151. class VisualScriptNodeInstanceDeconstruct : public VisualScriptNodeInstance {
  3152. public:
  3153. VisualScriptInstance *instance;
  3154. Vector<StringName> outputs;
  3155. //virtual int get_working_memory_size() const { return 0; }
  3156. virtual int step(const Variant **p_inputs, Variant **p_outputs, StartMode p_start_mode, Variant *p_working_mem, Callable::CallError &r_error, String &r_error_str) {
  3157. Variant in = *p_inputs[0];
  3158. for (int i = 0; i < outputs.size(); i++) {
  3159. bool valid;
  3160. *p_outputs[i] = in.get(outputs[i], &valid);
  3161. if (!valid) {
  3162. r_error_str = "Can't obtain element '" + String(outputs[i]) + "' from " + Variant::get_type_name(in.get_type());
  3163. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  3164. return 0;
  3165. }
  3166. }
  3167. return 0;
  3168. }
  3169. };
  3170. VisualScriptNodeInstance *VisualScriptDeconstruct::instantiate(VisualScriptInstance *p_instance) {
  3171. VisualScriptNodeInstanceDeconstruct *instance = memnew(VisualScriptNodeInstanceDeconstruct);
  3172. instance->instance = p_instance;
  3173. instance->outputs.resize(elements.size());
  3174. for (int i = 0; i < elements.size(); i++) {
  3175. instance->outputs.write[i] = elements[i].name;
  3176. }
  3177. return instance;
  3178. }
  3179. void VisualScriptDeconstruct::_validate_property(PropertyInfo &property) const {
  3180. }
  3181. void VisualScriptDeconstruct::_bind_methods() {
  3182. ClassDB::bind_method(D_METHOD("set_deconstruct_type", "type"), &VisualScriptDeconstruct::set_deconstruct_type);
  3183. ClassDB::bind_method(D_METHOD("get_deconstruct_type"), &VisualScriptDeconstruct::get_deconstruct_type);
  3184. ClassDB::bind_method(D_METHOD("_set_elem_cache", "_cache"), &VisualScriptDeconstruct::_set_elem_cache);
  3185. ClassDB::bind_method(D_METHOD("_get_elem_cache"), &VisualScriptDeconstruct::_get_elem_cache);
  3186. String argt = "Any";
  3187. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  3188. argt += "," + Variant::get_type_name(Variant::Type(i));
  3189. }
  3190. ADD_PROPERTY(PropertyInfo(Variant::INT, "type", PROPERTY_HINT_ENUM, argt), "set_deconstruct_type", "get_deconstruct_type");
  3191. ADD_PROPERTY(PropertyInfo(Variant::ARRAY, "elem_cache", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR | PROPERTY_USAGE_INTERNAL), "_set_elem_cache", "_get_elem_cache");
  3192. }
  3193. VisualScriptDeconstruct::VisualScriptDeconstruct() {
  3194. type = Variant::NIL;
  3195. }
  3196. template <Variant::Type T>
  3197. static Ref<VisualScriptNode> create_node_deconst_typed(const String &p_name) {
  3198. Ref<VisualScriptDeconstruct> node;
  3199. node.instantiate();
  3200. node->set_deconstruct_type(T);
  3201. return node;
  3202. }
  3203. void register_visual_script_nodes() {
  3204. VisualScriptLanguage::singleton->add_register_func("data/set_variable", create_node_generic<VisualScriptVariableSet>);
  3205. VisualScriptLanguage::singleton->add_register_func("data/get_variable", create_node_generic<VisualScriptVariableGet>);
  3206. VisualScriptLanguage::singleton->add_register_func("data/engine_singleton", create_node_generic<VisualScriptEngineSingleton>);
  3207. VisualScriptLanguage::singleton->add_register_func("data/scene_node", create_node_generic<VisualScriptSceneNode>);
  3208. VisualScriptLanguage::singleton->add_register_func("data/scene_tree", create_node_generic<VisualScriptSceneTree>);
  3209. VisualScriptLanguage::singleton->add_register_func("data/resource_path", create_node_generic<VisualScriptResourcePath>);
  3210. VisualScriptLanguage::singleton->add_register_func("data/self", create_node_generic<VisualScriptSelf>);
  3211. VisualScriptLanguage::singleton->add_register_func("data/comment", create_node_generic<VisualScriptComment>);
  3212. VisualScriptLanguage::singleton->add_register_func("data/get_local_variable", create_node_generic<VisualScriptLocalVar>);
  3213. VisualScriptLanguage::singleton->add_register_func("data/set_local_variable", create_node_generic<VisualScriptLocalVarSet>);
  3214. VisualScriptLanguage::singleton->add_register_func("data/preload", create_node_generic<VisualScriptPreload>);
  3215. VisualScriptLanguage::singleton->add_register_func("data/action", create_node_generic<VisualScriptInputAction>);
  3216. VisualScriptLanguage::singleton->add_register_func("constants/constant", create_node_generic<VisualScriptConstant>);
  3217. VisualScriptLanguage::singleton->add_register_func("constants/math_constant", create_node_generic<VisualScriptMathConstant>);
  3218. VisualScriptLanguage::singleton->add_register_func("constants/class_constant", create_node_generic<VisualScriptClassConstant>);
  3219. VisualScriptLanguage::singleton->add_register_func("constants/global_constant", create_node_generic<VisualScriptGlobalConstant>);
  3220. VisualScriptLanguage::singleton->add_register_func("constants/basic_type_constant", create_node_generic<VisualScriptBasicTypeConstant>);
  3221. VisualScriptLanguage::singleton->add_register_func("custom/custom_node", create_node_generic<VisualScriptCustomNode>);
  3222. VisualScriptLanguage::singleton->add_register_func("custom/sub_call", create_node_generic<VisualScriptSubCall>);
  3223. VisualScriptLanguage::singleton->add_register_func("index/get_index", create_node_generic<VisualScriptIndexGet>);
  3224. VisualScriptLanguage::singleton->add_register_func("index/set_index", create_node_generic<VisualScriptIndexSet>);
  3225. VisualScriptLanguage::singleton->add_register_func("operators/compare/equal", create_op_node<Variant::OP_EQUAL>);
  3226. VisualScriptLanguage::singleton->add_register_func("operators/compare/not_equal", create_op_node<Variant::OP_NOT_EQUAL>);
  3227. VisualScriptLanguage::singleton->add_register_func("operators/compare/less", create_op_node<Variant::OP_LESS>);
  3228. VisualScriptLanguage::singleton->add_register_func("operators/compare/less_equal", create_op_node<Variant::OP_LESS_EQUAL>);
  3229. VisualScriptLanguage::singleton->add_register_func("operators/compare/greater", create_op_node<Variant::OP_GREATER>);
  3230. VisualScriptLanguage::singleton->add_register_func("operators/compare/greater_equal", create_op_node<Variant::OP_GREATER_EQUAL>);
  3231. //mathematic
  3232. VisualScriptLanguage::singleton->add_register_func("operators/math/add", create_op_node<Variant::OP_ADD>);
  3233. VisualScriptLanguage::singleton->add_register_func("operators/math/subtract", create_op_node<Variant::OP_SUBTRACT>);
  3234. VisualScriptLanguage::singleton->add_register_func("operators/math/multiply", create_op_node<Variant::OP_MULTIPLY>);
  3235. VisualScriptLanguage::singleton->add_register_func("operators/math/divide", create_op_node<Variant::OP_DIVIDE>);
  3236. VisualScriptLanguage::singleton->add_register_func("operators/math/negate", create_op_node<Variant::OP_NEGATE>);
  3237. VisualScriptLanguage::singleton->add_register_func("operators/math/positive", create_op_node<Variant::OP_POSITIVE>);
  3238. VisualScriptLanguage::singleton->add_register_func("operators/math/remainder", create_op_node<Variant::OP_MODULE>);
  3239. //bitwise
  3240. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/shift_left", create_op_node<Variant::OP_SHIFT_LEFT>);
  3241. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/shift_right", create_op_node<Variant::OP_SHIFT_RIGHT>);
  3242. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_and", create_op_node<Variant::OP_BIT_AND>);
  3243. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_or", create_op_node<Variant::OP_BIT_OR>);
  3244. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_xor", create_op_node<Variant::OP_BIT_XOR>);
  3245. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_negate", create_op_node<Variant::OP_BIT_NEGATE>);
  3246. //logic
  3247. VisualScriptLanguage::singleton->add_register_func("operators/logic/and", create_op_node<Variant::OP_AND>);
  3248. VisualScriptLanguage::singleton->add_register_func("operators/logic/or", create_op_node<Variant::OP_OR>);
  3249. VisualScriptLanguage::singleton->add_register_func("operators/logic/xor", create_op_node<Variant::OP_XOR>);
  3250. VisualScriptLanguage::singleton->add_register_func("operators/logic/not", create_op_node<Variant::OP_NOT>);
  3251. VisualScriptLanguage::singleton->add_register_func("operators/logic/in", create_op_node<Variant::OP_IN>);
  3252. VisualScriptLanguage::singleton->add_register_func("operators/logic/select", create_node_generic<VisualScriptSelect>);
  3253. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::VECTOR2), create_node_deconst_typed<Variant::Type::VECTOR2>);
  3254. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::VECTOR2I), create_node_deconst_typed<Variant::Type::VECTOR2I>);
  3255. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::VECTOR3), create_node_deconst_typed<Variant::Type::VECTOR3>);
  3256. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::VECTOR3I), create_node_deconst_typed<Variant::Type::VECTOR3I>);
  3257. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::COLOR), create_node_deconst_typed<Variant::Type::COLOR>);
  3258. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::RECT2), create_node_deconst_typed<Variant::Type::RECT2>);
  3259. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::RECT2I), create_node_deconst_typed<Variant::Type::RECT2I>);
  3260. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::TRANSFORM2D), create_node_deconst_typed<Variant::Type::TRANSFORM2D>);
  3261. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::PLANE), create_node_deconst_typed<Variant::Type::PLANE>);
  3262. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::QUATERNION), create_node_deconst_typed<Variant::Type::QUATERNION>);
  3263. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::AABB), create_node_deconst_typed<Variant::Type::AABB>);
  3264. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::BASIS), create_node_deconst_typed<Variant::Type::BASIS>);
  3265. VisualScriptLanguage::singleton->add_register_func("functions/deconstruct/" + Variant::get_type_name(Variant::Type::TRANSFORM3D), create_node_deconst_typed<Variant::Type::TRANSFORM3D>);
  3266. VisualScriptLanguage::singleton->add_register_func("functions/compose_array", create_node_generic<VisualScriptComposeArray>);
  3267. for (int i = 1; i < Variant::VARIANT_MAX; i++) {
  3268. List<MethodInfo> constructors;
  3269. Variant::get_constructor_list(Variant::Type(i), &constructors);
  3270. for (const MethodInfo &E : constructors) {
  3271. if (E.arguments.size() > 0) {
  3272. String name = "functions/constructors/" + Variant::get_type_name(Variant::Type(i)) + "(";
  3273. for (int j = 0; j < E.arguments.size(); j++) {
  3274. if (j > 0) {
  3275. name += ", ";
  3276. }
  3277. if (E.arguments.size() == 1) {
  3278. name += Variant::get_type_name(E.arguments[j].type);
  3279. } else {
  3280. name += E.arguments[j].name;
  3281. }
  3282. }
  3283. name += ")";
  3284. VisualScriptLanguage::singleton->add_register_func(name, create_constructor_node);
  3285. Pair<Variant::Type, MethodInfo> pair;
  3286. pair.first = Variant::Type(i);
  3287. pair.second = E;
  3288. constructor_map[name] = pair;
  3289. }
  3290. }
  3291. }
  3292. }
  3293. void unregister_visual_script_nodes() {
  3294. constructor_map.clear();
  3295. }