visual_script_editor.cpp 101 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560
  1. /*************************************************************************/
  2. /* visual_script_editor.h */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* http://www.godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur. */
  9. /* */
  10. /* Permission is hereby granted, free of charge, to any person obtaining */
  11. /* a copy of this software and associated documentation files (the */
  12. /* "Software"), to deal in the Software without restriction, including */
  13. /* without limitation the rights to use, copy, modify, merge, publish, */
  14. /* distribute, sublicense, and/or sell copies of the Software, and to */
  15. /* permit persons to whom the Software is furnished to do so, subject to */
  16. /* the following conditions: */
  17. /* */
  18. /* The above copyright notice and this permission notice shall be */
  19. /* included in all copies or substantial portions of the Software. */
  20. /* */
  21. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  22. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  23. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  24. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  25. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  26. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  27. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  28. /*************************************************************************/
  29. #include "visual_script_editor.h"
  30. #include "editor/editor_node.h"
  31. #include "visual_script_nodes.h"
  32. #include "visual_script_flow_control.h"
  33. #include "visual_script_func_nodes.h"
  34. #include "visual_script_expression.h"
  35. #include "os/input.h"
  36. #include "editor/editor_resource_preview.h"
  37. #include "os/keyboard.h"
  38. #ifdef TOOLS_ENABLED
  39. class VisualScriptEditorSignalEdit : public Object {
  40. GDCLASS(VisualScriptEditorSignalEdit,Object)
  41. StringName sig;
  42. public:
  43. UndoRedo *undo_redo;
  44. Ref<VisualScript> script;
  45. protected:
  46. static void _bind_methods() {
  47. ClassDB::bind_method("_sig_changed",&VisualScriptEditorSignalEdit::_sig_changed);
  48. }
  49. void _sig_changed() {
  50. _change_notify();
  51. }
  52. bool _set(const StringName& p_name, const Variant& p_value) {
  53. if (sig==StringName())
  54. return false;
  55. if (p_name=="argument_count") {
  56. int new_argc=p_value;
  57. int argc = script->custom_signal_get_argument_count(sig);
  58. if (argc==new_argc)
  59. return true;
  60. undo_redo->create_action("Change Signal Arguments");
  61. if (new_argc < argc) {
  62. for(int i=new_argc;i<argc;i++) {
  63. undo_redo->add_do_method(script.ptr(),"custom_signal_remove_argument",sig,new_argc);
  64. undo_redo->add_undo_method(script.ptr(),"custom_signal_add_argument",sig,script->custom_signal_get_argument_name(sig,i),script->custom_signal_get_argument_type(sig,i),-1);
  65. }
  66. } else if (new_argc>argc) {
  67. for(int i=argc;i<new_argc;i++) {
  68. undo_redo->add_do_method(script.ptr(),"custom_signal_add_argument",sig,Variant::NIL,"arg"+itos(i+1),-1);
  69. undo_redo->add_undo_method(script.ptr(),"custom_signal_remove_argument",sig,argc);
  70. }
  71. }
  72. undo_redo->add_do_method(this,"_sig_changed");
  73. undo_redo->add_undo_method(this,"_sig_changed");
  74. undo_redo->commit_action();
  75. return true;
  76. }
  77. if (String(p_name).begins_with("argument/")) {
  78. int idx = String(p_name).get_slice("/",1).to_int()-1;
  79. ERR_FAIL_INDEX_V(idx,script->custom_signal_get_argument_count(sig),false);
  80. String what = String(p_name).get_slice("/",2);
  81. if (what=="type") {
  82. int old_type = script->custom_signal_get_argument_type(sig,idx);
  83. int new_type=p_value;
  84. undo_redo->create_action("Change Argument Type");
  85. undo_redo->add_do_method(script.ptr(),"custom_signal_set_argument_type",sig,idx,new_type);
  86. undo_redo->add_undo_method(script.ptr(),"custom_signal_set_argument_type",sig,idx,old_type);
  87. undo_redo->commit_action();
  88. return true;
  89. }
  90. if (what=="name") {
  91. String old_name = script->custom_signal_get_argument_name(sig,idx);
  92. String new_name=p_value;
  93. undo_redo->create_action("Change Argument name");
  94. undo_redo->add_do_method(script.ptr(),"custom_signal_set_argument_name",sig,idx,new_name);
  95. undo_redo->add_undo_method(script.ptr(),"custom_signal_set_argument_name",sig,idx,old_name);
  96. undo_redo->commit_action();
  97. return true;
  98. }
  99. }
  100. return false;
  101. }
  102. bool _get(const StringName& p_name,Variant &r_ret) const {
  103. if (sig==StringName())
  104. return false;
  105. if (p_name=="argument_count") {
  106. r_ret = script->custom_signal_get_argument_count(sig);
  107. return true;
  108. }
  109. if (String(p_name).begins_with("argument/")) {
  110. int idx = String(p_name).get_slice("/",1).to_int()-1;
  111. ERR_FAIL_INDEX_V(idx,script->custom_signal_get_argument_count(sig),false);
  112. String what = String(p_name).get_slice("/",2);
  113. if (what=="type") {
  114. r_ret = script->custom_signal_get_argument_type(sig,idx);
  115. return true;
  116. }
  117. if (what=="name") {
  118. r_ret = script->custom_signal_get_argument_name(sig,idx);
  119. return true;
  120. }
  121. }
  122. return false;
  123. }
  124. void _get_property_list( List<PropertyInfo> *p_list) const {
  125. if (sig==StringName())
  126. return;
  127. p_list->push_back(PropertyInfo(Variant::INT,"argument_count",PROPERTY_HINT_RANGE,"0,256"));
  128. String argt="Variant";
  129. for(int i=1;i<Variant::VARIANT_MAX;i++) {
  130. argt+=","+Variant::get_type_name(Variant::Type(i));
  131. }
  132. for(int i=0;i<script->custom_signal_get_argument_count(sig);i++) {
  133. p_list->push_back(PropertyInfo(Variant::INT,"argument/"+itos(i+1)+"/type",PROPERTY_HINT_ENUM,argt));
  134. p_list->push_back(PropertyInfo(Variant::STRING,"argument/"+itos(i+1)+"/name"));
  135. }
  136. }
  137. public:
  138. void edit(const StringName& p_sig) {
  139. sig=p_sig;
  140. _change_notify();
  141. }
  142. VisualScriptEditorSignalEdit() { undo_redo=NULL; }
  143. };
  144. class VisualScriptEditorVariableEdit : public Object {
  145. GDCLASS(VisualScriptEditorVariableEdit,Object)
  146. StringName var;
  147. public:
  148. UndoRedo *undo_redo;
  149. Ref<VisualScript> script;
  150. protected:
  151. static void _bind_methods() {
  152. ClassDB::bind_method("_var_changed",&VisualScriptEditorVariableEdit::_var_changed);
  153. ClassDB::bind_method("_var_value_changed",&VisualScriptEditorVariableEdit::_var_value_changed);
  154. }
  155. void _var_changed() {
  156. _change_notify();
  157. }
  158. void _var_value_changed() {
  159. _change_notify("value"); //so the whole tree is not redrawn, makes editing smoother in general
  160. }
  161. bool _set(const StringName& p_name, const Variant& p_value) {
  162. if (var==StringName())
  163. return false;
  164. if (String(p_name)=="value") {
  165. undo_redo->create_action("Set Variable Default Value");
  166. Variant current=script->get_variable_default_value(var);
  167. undo_redo->add_do_method(script.ptr(),"set_variable_default_value",var,p_value);
  168. undo_redo->add_undo_method(script.ptr(),"set_variable_default_value",var,current);
  169. undo_redo->add_do_method(this,"_var_value_changed");
  170. undo_redo->add_undo_method(this,"_var_value_changed");
  171. undo_redo->commit_action();
  172. return true;
  173. }
  174. Dictionary d = script->call("get_variable_info",var);
  175. if (String(p_name)=="type") {
  176. Dictionary dc=d.copy();
  177. dc["type"]=p_value;
  178. undo_redo->create_action("Set Variable Type");
  179. undo_redo->add_do_method(script.ptr(),"set_variable_info",var,dc);
  180. undo_redo->add_undo_method(script.ptr(),"set_variable_info",var,d);
  181. undo_redo->add_do_method(this,"_var_changed");
  182. undo_redo->add_undo_method(this,"_var_changed");
  183. undo_redo->commit_action();
  184. return true;
  185. }
  186. if (String(p_name)=="hint") {
  187. Dictionary dc=d.copy();
  188. dc["hint"]=p_value;
  189. undo_redo->create_action("Set Variable Type");
  190. undo_redo->add_do_method(script.ptr(),"set_variable_info",var,dc);
  191. undo_redo->add_undo_method(script.ptr(),"set_variable_info",var,d);
  192. undo_redo->add_do_method(this,"_var_changed");
  193. undo_redo->add_undo_method(this,"_var_changed");
  194. undo_redo->commit_action();
  195. return true;
  196. }
  197. if (String(p_name)=="hint_string") {
  198. Dictionary dc=d.copy();
  199. dc["hint_string"]=p_value;
  200. undo_redo->create_action("Set Variable Type");
  201. undo_redo->add_do_method(script.ptr(),"set_variable_info",var,dc);
  202. undo_redo->add_undo_method(script.ptr(),"set_variable_info",var,d);
  203. undo_redo->add_do_method(this,"_var_changed");
  204. undo_redo->add_undo_method(this,"_var_changed");
  205. undo_redo->commit_action();
  206. return true;
  207. }
  208. if (String(p_name)=="export") {
  209. script->set_variable_export(var,p_value);
  210. return true;
  211. }
  212. return false;
  213. }
  214. bool _get(const StringName& p_name,Variant &r_ret) const {
  215. if (var==StringName())
  216. return false;
  217. if (String(p_name)=="value") {
  218. r_ret=script->get_variable_default_value(var);
  219. return true;
  220. }
  221. PropertyInfo pinfo = script->get_variable_info(var);
  222. if (String(p_name)=="type") {
  223. r_ret=pinfo.type;
  224. return true;
  225. }
  226. if (String(p_name)=="hint") {
  227. r_ret=pinfo.hint;
  228. return true;
  229. }
  230. if (String(p_name)=="hint_string") {
  231. r_ret=pinfo.hint_string;
  232. return true;
  233. }
  234. if (String(p_name)=="export") {
  235. r_ret=script->get_variable_export(var);
  236. return true;
  237. }
  238. return false;
  239. }
  240. void _get_property_list( List<PropertyInfo> *p_list) const {
  241. if (var==StringName())
  242. return;
  243. String argt="Variant";
  244. for(int i=1;i<Variant::VARIANT_MAX;i++) {
  245. argt+=","+Variant::get_type_name(Variant::Type(i));
  246. }
  247. p_list->push_back(PropertyInfo(Variant::INT,"type",PROPERTY_HINT_ENUM,argt));
  248. p_list->push_back(PropertyInfo(script->get_variable_info(var).type,"value",script->get_variable_info(var).hint,script->get_variable_info(var).hint_string,PROPERTY_USAGE_DEFAULT));
  249. p_list->push_back(PropertyInfo(Variant::INT,"hint",PROPERTY_HINT_ENUM,"None,Range,ExpRange,Enum,ExpEasing,Length,SpriteFrame,KeyAccel,BitFlags,AllFlags,File,Dir,GlobalFile,GlobalDir,ResourceType,MultilineText"));
  250. p_list->push_back(PropertyInfo(Variant::STRING,"hint_string"));
  251. p_list->push_back(PropertyInfo(Variant::BOOL,"export"));
  252. }
  253. public:
  254. void edit(const StringName& p_var) {
  255. var=p_var;
  256. _change_notify();
  257. }
  258. VisualScriptEditorVariableEdit() { undo_redo=NULL; }
  259. };
  260. static Color _color_from_type(Variant::Type p_type) {
  261. Color color;
  262. switch(p_type) {
  263. case Variant::NIL: color = Color::html("69ecbd"); break;
  264. case Variant::BOOL: color = Color::html("8da6f0"); break;
  265. case Variant::INT: color = Color::html("7dc6ef"); break;
  266. case Variant::REAL: color = Color::html("61daf4"); break;
  267. case Variant::STRING: color = Color::html("6ba7ec"); break;
  268. case Variant::VECTOR2: color = Color::html("bd91f1"); break;
  269. case Variant::RECT2: color = Color::html("f191a5"); break;
  270. case Variant::VECTOR3: color = Color::html("d67dee"); break;
  271. case Variant::TRANSFORM2D: color = Color::html("c4ec69"); break;
  272. case Variant::PLANE: color = Color::html("f77070"); break;
  273. case Variant::QUAT: color = Color::html("ec69a3"); break;
  274. case Variant::RECT3: color = Color::html("ee7991"); break;
  275. case Variant::BASIS: color = Color::html("e3ec69"); break;
  276. case Variant::TRANSFORM: color = Color::html("f6a86e"); break;
  277. case Variant::COLOR: color = Color::html("9dff70"); break;
  278. case Variant::IMAGE: color = Color::html("93f1b9"); break;
  279. case Variant::NODE_PATH: color = Color::html("6993ec"); break;
  280. case Variant::_RID: color = Color::html("69ec9a"); break;
  281. case Variant::OBJECT: color = Color::html("79f3e8"); break;
  282. case Variant::INPUT_EVENT: color = Color::html("adf18f"); break;
  283. case Variant::DICTIONARY: color = Color::html("77edb1"); break;
  284. case Variant::ARRAY: color = Color::html("e0e0e0"); break;
  285. case Variant::POOL_BYTE_ARRAY: color = Color::html("aaf4c8"); break;
  286. case Variant::POOL_INT_ARRAY: color = Color::html("afdcf5"); break;
  287. case Variant::POOL_REAL_ARRAY: color = Color::html("97e7f8"); break;
  288. case Variant::POOL_STRING_ARRAY: color = Color::html("9dc4f2"); break;
  289. case Variant::POOL_VECTOR2_ARRAY: color = Color::html("d1b3f5"); break;
  290. case Variant::POOL_VECTOR3_ARRAY: color = Color::html("df9bf2"); break;
  291. case Variant::POOL_COLOR_ARRAY: color = Color::html("e9ff97"); break;
  292. default:
  293. color.set_hsv(p_type/float(Variant::VARIANT_MAX),0.7,0.7);
  294. }
  295. return color;
  296. }
  297. void VisualScriptEditor::_update_graph_connections() {
  298. graph->clear_connections();
  299. List<VisualScript::SequenceConnection> sequence_conns;
  300. script->get_sequence_connection_list(edited_func,&sequence_conns);
  301. for (List<VisualScript::SequenceConnection>::Element *E=sequence_conns.front();E;E=E->next()) {
  302. graph->connect_node(itos(E->get().from_node),E->get().from_output,itos(E->get().to_node),0);
  303. }
  304. List<VisualScript::DataConnection> data_conns;
  305. script->get_data_connection_list(edited_func,&data_conns);
  306. for (List<VisualScript::DataConnection>::Element *E=data_conns.front();E;E=E->next()) {
  307. VisualScript::DataConnection dc=E->get();
  308. Ref<VisualScriptNode> from_node = script->get_node(edited_func,E->get().from_node);
  309. Ref<VisualScriptNode> to_node = script->get_node(edited_func,E->get().to_node);
  310. if (to_node->has_input_sequence_port()) {
  311. dc.to_port++;
  312. }
  313. dc.from_port+=from_node->get_output_sequence_port_count();
  314. graph->connect_node(itos(E->get().from_node),dc.from_port,itos(E->get().to_node),dc.to_port);
  315. }
  316. }
  317. void VisualScriptEditor::_update_graph(int p_only_id) {
  318. if (updating_graph)
  319. return;
  320. updating_graph=true;
  321. //byebye all nodes
  322. if (p_only_id>=0) {
  323. if (graph->has_node(itos(p_only_id))) {
  324. Node* gid = graph->get_node(itos(p_only_id));
  325. if (gid)
  326. memdelete(gid);
  327. }
  328. } else {
  329. for(int i=0;i<graph->get_child_count();i++) {
  330. if (graph->get_child(i)->cast_to<GraphNode>()) {
  331. memdelete(graph->get_child(i));
  332. i--;
  333. }
  334. }
  335. }
  336. if (!script->has_function(edited_func)) {
  337. graph->hide();
  338. select_func_text->show();
  339. updating_graph=false;
  340. return;
  341. }
  342. graph->show();
  343. select_func_text->hide();
  344. Ref<Texture> type_icons[Variant::VARIANT_MAX]={
  345. Control::get_icon("MiniVariant","EditorIcons"),
  346. Control::get_icon("MiniBoolean","EditorIcons"),
  347. Control::get_icon("MiniInteger","EditorIcons"),
  348. Control::get_icon("MiniFloat","EditorIcons"),
  349. Control::get_icon("MiniString","EditorIcons"),
  350. Control::get_icon("MiniVector2","EditorIcons"),
  351. Control::get_icon("MiniRect2","EditorIcons"),
  352. Control::get_icon("MiniVector3","EditorIcons"),
  353. Control::get_icon("MiniTransform2D","EditorIcons"),
  354. Control::get_icon("MiniPlane","EditorIcons"),
  355. Control::get_icon("MiniQuat","EditorIcons"),
  356. Control::get_icon("MiniAabb","EditorIcons"),
  357. Control::get_icon("MiniBasis","EditorIcons"),
  358. Control::get_icon("MiniTransform","EditorIcons"),
  359. Control::get_icon("MiniColor","EditorIcons"),
  360. Control::get_icon("MiniImage","EditorIcons"),
  361. Control::get_icon("MiniPath","EditorIcons"),
  362. Control::get_icon("MiniRid","EditorIcons"),
  363. Control::get_icon("MiniObject","EditorIcons"),
  364. Control::get_icon("MiniInput","EditorIcons"),
  365. Control::get_icon("MiniDictionary","EditorIcons"),
  366. Control::get_icon("MiniArray","EditorIcons"),
  367. Control::get_icon("MiniRawArray","EditorIcons"),
  368. Control::get_icon("MiniIntArray","EditorIcons"),
  369. Control::get_icon("MiniFloatArray","EditorIcons"),
  370. Control::get_icon("MiniStringArray","EditorIcons"),
  371. Control::get_icon("MiniVector2Array","EditorIcons"),
  372. Control::get_icon("MiniVector3Array","EditorIcons"),
  373. Control::get_icon("MiniColorArray","EditorIcons")
  374. };
  375. Ref<Texture> seq_port = Control::get_icon("VisualShaderPort","EditorIcons");
  376. List<int> ids;
  377. script->get_node_list(edited_func,&ids);
  378. StringName editor_icons="EditorIcons";
  379. for(List<int>::Element *E=ids.front();E;E=E->next()) {
  380. if (p_only_id>=0 && p_only_id!=E->get())
  381. continue;
  382. Ref<VisualScriptNode> node = script->get_node(edited_func,E->get());
  383. Vector2 pos = script->get_node_pos(edited_func,E->get());
  384. GraphNode *gnode = memnew( GraphNode );
  385. gnode->set_title(node->get_caption());
  386. if (error_line==E->get()) {
  387. gnode->set_overlay(GraphNode::OVERLAY_POSITION);
  388. } else if (node->is_breakpoint()) {
  389. gnode->set_overlay(GraphNode::OVERLAY_BREAKPOINT);
  390. }
  391. if (EditorSettings::get_singleton()->has("editors/visual_script/color_"+node->get_category())) {
  392. gnode->set_modulate(EditorSettings::get_singleton()->get("editors/visual_script/color_"+node->get_category()));
  393. }
  394. gnode->set_meta("__vnode",node);
  395. gnode->set_name(itos(E->get()));
  396. gnode->connect("dragged",this,"_node_moved",varray(E->get()));
  397. gnode->connect("close_request",this,"_remove_node",varray(E->get()),CONNECT_DEFERRED);
  398. if (E->get()!=script->get_function_node_id(edited_func)) {
  399. //function can't be erased
  400. gnode->set_show_close_button(true);
  401. }
  402. if (node->cast_to<VisualScriptExpression>()) {
  403. LineEdit *line_edit = memnew( LineEdit );
  404. line_edit->set_text(node->get_text());
  405. line_edit->set_expand_to_text_length(true);
  406. line_edit->add_font_override("font",get_font("source","EditorFonts"));
  407. gnode->add_child(line_edit);
  408. line_edit->connect("text_changed",this,"_expression_text_changed",varray(E->get()));
  409. } else {
  410. Label *text = memnew( Label );
  411. text->set_text(node->get_text());
  412. gnode->add_child(text);
  413. }
  414. if (node->cast_to<VisualScriptComment>()) {
  415. Ref<VisualScriptComment> vsc=node;
  416. gnode->set_comment(true);
  417. gnode->set_resizeable(true);
  418. gnode->set_custom_minimum_size(vsc->get_size()*EDSCALE);
  419. gnode->connect("resize_request",this,"_comment_node_resized",varray(E->get()));
  420. }
  421. int slot_idx=0;
  422. bool single_seq_output = node->get_output_sequence_port_count()==1 && node->get_output_sequence_port_text(0)==String();
  423. gnode->set_slot(0,node->has_input_sequence_port(),TYPE_SEQUENCE,Color(1,1,1,1),single_seq_output,TYPE_SEQUENCE,Color(1,1,1,1),seq_port,seq_port);
  424. gnode->set_offset(pos*EDSCALE);
  425. slot_idx++;
  426. int mixed_seq_ports=0;
  427. if (!single_seq_output) {
  428. if (node->has_mixed_input_and_sequence_ports()) {
  429. mixed_seq_ports=node->get_output_sequence_port_count();
  430. } else {
  431. for(int i=0;i<node->get_output_sequence_port_count();i++) {
  432. Label *text2 = memnew( Label );
  433. text2->set_text(node->get_output_sequence_port_text(i));
  434. text2->set_align(Label::ALIGN_RIGHT);
  435. gnode->add_child(text2);
  436. gnode->set_slot(slot_idx,false,0,Color(),true,TYPE_SEQUENCE,Color(1,1,1,1),seq_port,seq_port);
  437. slot_idx++;
  438. }
  439. }
  440. }
  441. for(int i=0;i<MAX(node->get_output_value_port_count(),MAX(mixed_seq_ports,node->get_input_value_port_count()));i++) {
  442. bool left_ok=false;
  443. Variant::Type left_type=Variant::NIL;
  444. String left_name;
  445. if (i<node->get_input_value_port_count()) {
  446. PropertyInfo pi = node->get_input_value_port_info(i);
  447. left_ok=true;
  448. left_type=pi.type;
  449. left_name=pi.name;
  450. }
  451. bool right_ok=false;
  452. Variant::Type right_type=Variant::NIL;
  453. String right_name;
  454. if (i>=mixed_seq_ports && i<node->get_output_value_port_count()+mixed_seq_ports) {
  455. PropertyInfo pi = node->get_output_value_port_info(i-mixed_seq_ports);
  456. right_ok=true;
  457. right_type=pi.type;
  458. right_name=pi.name;
  459. }
  460. HBoxContainer *hbc = memnew( HBoxContainer);
  461. if (left_ok) {
  462. Ref<Texture> t;
  463. if (left_type>=0 && left_type<Variant::VARIANT_MAX) {
  464. t=type_icons[left_type];
  465. }
  466. if (t.is_valid()) {
  467. TextureRect *tf = memnew(TextureRect);
  468. tf->set_texture(t);
  469. tf->set_stretch_mode(TextureRect::STRETCH_KEEP_CENTERED);
  470. hbc->add_child(tf);
  471. }
  472. hbc->add_child(memnew(Label(left_name)));
  473. if (left_type!=Variant::NIL && !script->is_input_value_port_connected(edited_func,E->get(),i)) {
  474. PropertyInfo pi = node->get_input_value_port_info(i);
  475. Button *button = memnew( Button );
  476. Variant value = node->get_default_input_value(i);
  477. if (value.get_type()!=left_type) {
  478. //different type? for now convert
  479. //not the same, reconvert
  480. Variant::CallError ce;
  481. const Variant *existingp=&value;
  482. value = Variant::construct(left_type,&existingp,1,ce,false);
  483. }
  484. if (left_type==Variant::COLOR) {
  485. button->set_custom_minimum_size(Size2(30,0)*EDSCALE);
  486. button->connect("draw",this,"_draw_color_over_button",varray(button,value));
  487. } else if (left_type==Variant::OBJECT && Ref<Resource>(value).is_valid()) {
  488. Ref<Resource> res = value;
  489. Array arr;
  490. arr.push_back(button->get_instance_ID());
  491. arr.push_back(String(value));
  492. EditorResourcePreview::get_singleton()->queue_edited_resource_preview(res,this,"_button_resource_previewed",arr);
  493. } else if (pi.type==Variant::INT && pi.hint==PROPERTY_HINT_ENUM){
  494. button->set_text(pi.hint_string.get_slice(",",value));
  495. } else {
  496. button->set_text(value);
  497. }
  498. button->connect("pressed",this,"_default_value_edited",varray(button,E->get(),i));
  499. hbc->add_child(button);
  500. }
  501. } else {
  502. Control *c = memnew(Control);
  503. c->set_custom_minimum_size(Size2(10,0)*EDSCALE);
  504. hbc->add_child(c);
  505. }
  506. hbc->add_spacer();
  507. if (i<mixed_seq_ports) {
  508. Label *text2 = memnew( Label );
  509. text2->set_text(node->get_output_sequence_port_text(i));
  510. text2->set_align(Label::ALIGN_RIGHT);
  511. hbc->add_child(text2);
  512. }
  513. if (right_ok) {
  514. hbc->add_child(memnew(Label(right_name)));
  515. Ref<Texture> t;
  516. if (right_type>=0 && right_type<Variant::VARIANT_MAX) {
  517. t=type_icons[right_type];
  518. }
  519. if (t.is_valid()) {
  520. TextureRect *tf = memnew(TextureRect);
  521. tf->set_texture(t);
  522. tf->set_stretch_mode(TextureRect::STRETCH_KEEP_CENTERED);
  523. hbc->add_child(tf);
  524. }
  525. }
  526. gnode->add_child(hbc);
  527. if (i<mixed_seq_ports) {
  528. gnode->set_slot(slot_idx,left_ok,left_type,_color_from_type(left_type),true,TYPE_SEQUENCE,Color(1,1,1,1),Ref<Texture>(),seq_port);
  529. } else {
  530. gnode->set_slot(slot_idx,left_ok,left_type,_color_from_type(left_type),right_ok,right_type,_color_from_type(right_type));
  531. }
  532. slot_idx++;
  533. }
  534. graph->add_child(gnode);
  535. if (gnode->is_comment()) {
  536. graph->move_child(gnode,0);
  537. }
  538. }
  539. _update_graph_connections();
  540. graph->call_deferred("set_scroll_ofs",script->get_function_scroll(edited_func)*EDSCALE); //may need to adapt a bit, let it do so
  541. updating_graph=false;
  542. }
  543. void VisualScriptEditor::_update_members() {
  544. updating_members=true;
  545. members->clear();
  546. TreeItem *root = members->create_item();
  547. TreeItem *functions = members->create_item(root);
  548. functions->set_selectable(0,false);
  549. functions->set_text(0,TTR("Functions:"));
  550. functions->add_button(0,Control::get_icon("Override","EditorIcons"),1);
  551. functions->add_button(0,Control::get_icon("Add","EditorIcons"),0);
  552. functions->set_custom_bg_color(0,Control::get_color("prop_section","Editor"));
  553. List<StringName> func_names;
  554. script->get_function_list(&func_names);
  555. for (List<StringName>::Element *E=func_names.front();E;E=E->next()) {
  556. TreeItem *ti = members->create_item(functions) ;
  557. ti->set_text(0,E->get());
  558. ti->set_selectable(0,true);
  559. ti->set_editable(0,true);
  560. //ti->add_button(0,Control::get_icon("Edit","EditorIcons"),0); function arguments are in the node now
  561. ti->add_button(0,Control::get_icon("Del","EditorIcons"),1);
  562. ti->set_metadata(0,E->get());
  563. if (E->get()==edited_func) {
  564. ti->set_custom_bg_color(0,get_color("prop_category","Editor"));
  565. ti->set_custom_color(0,Color(1,1,1,1));
  566. }
  567. if (selected==E->get())
  568. ti->select(0);
  569. }
  570. TreeItem *variables = members->create_item(root);
  571. variables->set_selectable(0,false);
  572. variables->set_text(0,TTR("Variables:"));
  573. variables->add_button(0,Control::get_icon("Add","EditorIcons"));
  574. variables->set_custom_bg_color(0,Control::get_color("prop_section","Editor"));
  575. Ref<Texture> type_icons[Variant::VARIANT_MAX]={
  576. Control::get_icon("MiniVariant","EditorIcons"),
  577. Control::get_icon("MiniBoolean","EditorIcons"),
  578. Control::get_icon("MiniInteger","EditorIcons"),
  579. Control::get_icon("MiniFloat","EditorIcons"),
  580. Control::get_icon("MiniString","EditorIcons"),
  581. Control::get_icon("MiniVector2","EditorIcons"),
  582. Control::get_icon("MiniRect2","EditorIcons"),
  583. Control::get_icon("MiniVector3","EditorIcons"),
  584. Control::get_icon("MiniMatrix32","EditorIcons"),
  585. Control::get_icon("MiniPlane","EditorIcons"),
  586. Control::get_icon("MiniQuat","EditorIcons"),
  587. Control::get_icon("MiniAabb","EditorIcons"),
  588. Control::get_icon("MiniMatrix3","EditorIcons"),
  589. Control::get_icon("MiniTransform","EditorIcons"),
  590. Control::get_icon("MiniColor","EditorIcons"),
  591. Control::get_icon("MiniImage","EditorIcons"),
  592. Control::get_icon("MiniPath","EditorIcons"),
  593. Control::get_icon("MiniRid","EditorIcons"),
  594. Control::get_icon("MiniObject","EditorIcons"),
  595. Control::get_icon("MiniInput","EditorIcons"),
  596. Control::get_icon("MiniDictionary","EditorIcons"),
  597. Control::get_icon("MiniArray","EditorIcons"),
  598. Control::get_icon("MiniRawArray","EditorIcons"),
  599. Control::get_icon("MiniIntArray","EditorIcons"),
  600. Control::get_icon("MiniFloatArray","EditorIcons"),
  601. Control::get_icon("MiniStringArray","EditorIcons"),
  602. Control::get_icon("MiniVector2Array","EditorIcons"),
  603. Control::get_icon("MiniVector3Array","EditorIcons"),
  604. Control::get_icon("MiniColorArray","EditorIcons")
  605. };
  606. List<StringName> var_names;
  607. script->get_variable_list(&var_names);
  608. for (List<StringName>::Element *E=var_names.front();E;E=E->next()) {
  609. TreeItem *ti = members->create_item(variables);
  610. ti->set_text(0,E->get());
  611. Variant var = script->get_variable_default_value(E->get());
  612. ti->set_suffix(0,"="+String(var));
  613. ti->set_icon(0,type_icons[script->get_variable_info(E->get()).type]);
  614. ti->set_selectable(0,true);
  615. ti->set_editable(0,true);
  616. ti->add_button(0,Control::get_icon("Edit","EditorIcons"),0);
  617. ti->add_button(0,Control::get_icon("Del","EditorIcons"),1);
  618. ti->set_metadata(0,E->get());
  619. if (selected==E->get())
  620. ti->select(0);
  621. }
  622. TreeItem *_signals = members->create_item(root);
  623. _signals->set_selectable(0,false);
  624. _signals->set_text(0,TTR("Signals:"));
  625. _signals->add_button(0,Control::get_icon("Add","EditorIcons"));
  626. _signals->set_custom_bg_color(0,Control::get_color("prop_section","Editor"));
  627. List<StringName> signal_names;
  628. script->get_custom_signal_list(&signal_names);
  629. for (List<StringName>::Element *E=signal_names.front();E;E=E->next()) {
  630. TreeItem *ti = members->create_item(_signals);
  631. ti->set_text(0,E->get());
  632. ti->set_selectable(0,true);
  633. ti->set_editable(0,true);
  634. ti->add_button(0,Control::get_icon("Edit","EditorIcons"),0);
  635. ti->add_button(0,Control::get_icon("Del","EditorIcons"),1);
  636. ti->set_metadata(0,E->get());
  637. if (selected==E->get())
  638. ti->select(0);
  639. }
  640. String base_type=script->get_instance_base_type();
  641. String icon_type=base_type;
  642. if (!Control::has_icon(base_type,"EditorIcons")) {
  643. icon_type="Object";
  644. }
  645. base_type_select->set_text(base_type);
  646. base_type_select->set_icon(Control::get_icon(icon_type,"EditorIcons"));
  647. updating_members=false;
  648. }
  649. void VisualScriptEditor::_member_selected() {
  650. if (updating_members)
  651. return;
  652. TreeItem *ti=members->get_selected();
  653. ERR_FAIL_COND(!ti);
  654. selected=ti->get_metadata(0);
  655. //print_line("selected: "+String(selected));
  656. if (ti->get_parent()==members->get_root()->get_children()) {
  657. if (edited_func!=selected) {
  658. revert_on_drag=edited_func;
  659. edited_func=selected;
  660. _update_members();
  661. _update_graph();
  662. }
  663. return; //or crash because it will become invalid
  664. }
  665. }
  666. void VisualScriptEditor::_member_edited() {
  667. if (updating_members)
  668. return;
  669. TreeItem *ti=members->get_edited();
  670. ERR_FAIL_COND(!ti);
  671. String name = ti->get_metadata(0);
  672. String new_name = ti->get_text(0);
  673. if (name==new_name)
  674. return;
  675. if (!new_name.is_valid_identifier()) {
  676. EditorNode::get_singleton()->show_warning(TTR("Name is not a valid identifier:")+" "+new_name);
  677. updating_members=true;
  678. ti->set_text(0,name);
  679. updating_members=false;
  680. return;
  681. }
  682. if (script->has_function(new_name) || script->has_variable(new_name) || script->has_custom_signal(new_name)) {
  683. EditorNode::get_singleton()->show_warning(TTR("Name already in use by another func/var/signal:")+" "+new_name);
  684. updating_members=true;
  685. ti->set_text(0,name);
  686. updating_members=false;
  687. return;
  688. }
  689. TreeItem *root=members->get_root();
  690. if (ti->get_parent()==root->get_children()) {
  691. if (edited_func==selected) {
  692. edited_func=new_name;
  693. }
  694. selected=new_name;
  695. _update_graph();
  696. undo_redo->create_action(TTR("Rename Function"));
  697. undo_redo->add_do_method(script.ptr(),"rename_function",name,new_name);
  698. undo_redo->add_undo_method(script.ptr(),"rename_function",new_name,name);
  699. undo_redo->add_do_method(this,"_update_members");
  700. undo_redo->add_undo_method(this,"_update_members");
  701. undo_redo->commit_action();
  702. return; //or crash because it will become invalid
  703. }
  704. if (ti->get_parent()==root->get_children()->get_next()) {
  705. selected=new_name;
  706. undo_redo->create_action(TTR("Rename Variable"));
  707. undo_redo->add_do_method(script.ptr(),"rename_variable",name,new_name);
  708. undo_redo->add_undo_method(script.ptr(),"rename_variable",new_name,name);
  709. undo_redo->add_do_method(this,"_update_members");
  710. undo_redo->add_undo_method(this,"_update_members");
  711. undo_redo->commit_action();
  712. return; //or crash because it will become invalid
  713. }
  714. if (ti->get_parent()==root->get_children()->get_next()->get_next()) {
  715. selected=new_name;
  716. undo_redo->create_action(TTR("Rename Signal"));
  717. undo_redo->add_do_method(script.ptr(),"rename_custom_signal",name,new_name);
  718. undo_redo->add_undo_method(script.ptr(),"rename_custom_signal",new_name,name);
  719. undo_redo->add_do_method(this,"_update_members");
  720. undo_redo->add_undo_method(this,"_update_members");
  721. undo_redo->commit_action();
  722. return; //or crash because it will become invalid
  723. }
  724. }
  725. void VisualScriptEditor::_override_pressed(int p_id) {
  726. //override a virtual function or method from base type
  727. ERR_FAIL_COND(!virtuals_in_menu.has(p_id));
  728. VirtualInMenu vim=virtuals_in_menu[p_id];
  729. String name = _validate_name(vim.name);
  730. selected=name;
  731. edited_func=selected;
  732. Ref<VisualScriptFunction> func_node;
  733. func_node.instance();
  734. func_node->set_name(vim.name);
  735. undo_redo->create_action(TTR("Add Function"));
  736. undo_redo->add_do_method(script.ptr(),"add_function",name);
  737. for(int i=0;i<vim.args.size();i++) {
  738. func_node->add_argument(vim.args[i].first,vim.args[i].second);
  739. }
  740. undo_redo->add_do_method(script.ptr(),"add_node",name,script->get_available_id(),func_node);
  741. if (vim.ret!=Variant::NIL || vim.ret_variant) {
  742. Ref<VisualScriptReturn> ret_node;
  743. ret_node.instance();
  744. ret_node->set_return_type(vim.ret);
  745. ret_node->set_enable_return_value(true);
  746. ret_node->set_name(vim.name);
  747. undo_redo->add_do_method(script.ptr(),"add_node",name,script->get_available_id()+1,ret_node,Vector2(500,0));
  748. }
  749. undo_redo->add_undo_method(script.ptr(),"remove_function",name);
  750. undo_redo->add_do_method(this,"_update_members");
  751. undo_redo->add_undo_method(this,"_update_members");
  752. undo_redo->add_do_method(this,"_update_graph");
  753. undo_redo->add_undo_method(this,"_update_graph");
  754. undo_redo->commit_action();
  755. _update_graph();
  756. }
  757. void VisualScriptEditor::_member_button(Object *p_item, int p_column, int p_button) {
  758. TreeItem *ti=p_item->cast_to<TreeItem>();
  759. TreeItem *root=members->get_root();
  760. if (ti->get_parent()==root) {
  761. //main buttons
  762. if (ti==root->get_children()) {
  763. //add function, this one uses menu
  764. if (p_button==1) {
  765. new_function_menu->clear();
  766. new_function_menu->set_size(Size2(0,0));
  767. int idx=0;
  768. virtuals_in_menu.clear();
  769. List<MethodInfo> mi;
  770. ClassDB::get_method_list(script->get_instance_base_type(),&mi);
  771. for (List<MethodInfo>::Element *E=mi.front();E;E=E->next()) {
  772. MethodInfo mi=E->get();
  773. if (mi.flags&METHOD_FLAG_VIRTUAL) {
  774. VirtualInMenu vim;
  775. vim.name=mi.name;
  776. vim.ret=mi.return_val.type;
  777. if (mi.return_val.name!=String())
  778. vim.ret_variant=true;
  779. else
  780. vim.ret_variant=false;
  781. String desc;
  782. if (mi.return_val.type==Variant::NIL)
  783. desc="var";
  784. else
  785. desc=Variant::get_type_name(mi.return_val.type);
  786. desc+=" "+mi.name+" ( ";
  787. for(int i=0;i<mi.arguments.size();i++) {
  788. if (i>0)
  789. desc+=", ";
  790. if (mi.arguments[i].type==Variant::NIL)
  791. desc+="var ";
  792. else
  793. desc+=Variant::get_type_name(mi.arguments[i].type)+" ";
  794. desc+=mi.arguments[i].name;
  795. Pair<Variant::Type,String> p;
  796. p.first=mi.arguments[i].type;
  797. p.second=mi.arguments[i].name;
  798. vim.args.push_back( p );
  799. }
  800. desc+=" )";
  801. virtuals_in_menu[idx]=vim;
  802. new_function_menu->add_item(desc,idx);
  803. idx++;
  804. }
  805. }
  806. Rect2 pos = members->get_item_rect(ti);
  807. new_function_menu->set_pos(members->get_global_pos()+pos.pos+Vector2(0,pos.size.y));
  808. new_function_menu->popup();
  809. return;
  810. } else if (p_button==0) {
  811. String name = _validate_name("new_function");
  812. selected=name;
  813. edited_func=selected;
  814. Ref<VisualScriptFunction> func_node;
  815. func_node.instance();
  816. func_node->set_name(name);
  817. undo_redo->create_action(TTR("Add Function"));
  818. undo_redo->add_do_method(script.ptr(),"add_function",name);
  819. undo_redo->add_do_method(script.ptr(),"add_node",name,script->get_available_id(),func_node);
  820. undo_redo->add_undo_method(script.ptr(),"remove_function",name);
  821. undo_redo->add_do_method(this,"_update_members");
  822. undo_redo->add_undo_method(this,"_update_members");
  823. undo_redo->add_do_method(this,"_update_graph");
  824. undo_redo->add_undo_method(this,"_update_graph");
  825. undo_redo->commit_action();
  826. _update_graph();
  827. }
  828. return; //or crash because it will become invalid
  829. }
  830. if (ti==root->get_children()->get_next()) {
  831. //add variable
  832. String name = _validate_name("new_variable");
  833. selected=name;
  834. undo_redo->create_action(TTR("Add Variable"));
  835. undo_redo->add_do_method(script.ptr(),"add_variable",name);
  836. undo_redo->add_undo_method(script.ptr(),"remove_variable",name);
  837. undo_redo->add_do_method(this,"_update_members");
  838. undo_redo->add_undo_method(this,"_update_members");
  839. undo_redo->commit_action();
  840. return; //or crash because it will become invalid
  841. }
  842. if (ti==root->get_children()->get_next()->get_next()) {
  843. //add variable
  844. String name = _validate_name("new_signal");
  845. selected=name;
  846. undo_redo->create_action(TTR("Add Signal"));
  847. undo_redo->add_do_method(script.ptr(),"add_custom_signal",name);
  848. undo_redo->add_undo_method(script.ptr(),"remove_custom_signal",name);
  849. undo_redo->add_do_method(this,"_update_members");
  850. undo_redo->add_undo_method(this,"_update_members");
  851. undo_redo->commit_action();
  852. return; //or crash because it will become invalid
  853. }
  854. } else {
  855. if (ti->get_parent()==root->get_children()) {
  856. //edit/remove function
  857. String name = ti->get_metadata(0);
  858. if (p_button==1) {
  859. //delete the function
  860. undo_redo->create_action(TTR("Remove Function"));
  861. undo_redo->add_do_method(script.ptr(),"remove_function",name);
  862. undo_redo->add_undo_method(script.ptr(),"add_function",name);
  863. List<int> nodes;
  864. script->get_node_list(name,&nodes);
  865. for (List<int>::Element *E=nodes.front();E;E=E->next()) {
  866. undo_redo->add_undo_method(script.ptr(),"add_node",name,E->get(),script->get_node(name,E->get()),script->get_node_pos(name,E->get()));
  867. }
  868. List<VisualScript::SequenceConnection> seq_connections;
  869. script->get_sequence_connection_list(name,&seq_connections);
  870. for (List<VisualScript::SequenceConnection>::Element *E=seq_connections.front();E;E=E->next()) {
  871. undo_redo->add_undo_method(script.ptr(),"sequence_connect",name,E->get().from_node,E->get().from_output,E->get().to_node);
  872. }
  873. List<VisualScript::DataConnection> data_connections;
  874. script->get_data_connection_list(name,&data_connections);
  875. for (List<VisualScript::DataConnection>::Element *E=data_connections.front();E;E=E->next()) {
  876. undo_redo->add_undo_method(script.ptr(),"data_connect",name,E->get().from_node,E->get().from_port,E->get().to_node,E->get().to_port);
  877. }
  878. /*
  879. for(int i=0;i<script->function_get_argument_count(name);i++) {
  880. undo_redo->add_undo_method(script.ptr(),"function_add_argument",name,script->function_get_argument_name(name,i),script->function_get_argument_type(name,i));
  881. }
  882. */
  883. undo_redo->add_do_method(this,"_update_members");
  884. undo_redo->add_undo_method(this,"_update_members");
  885. undo_redo->add_do_method(this,"_update_graph");
  886. undo_redo->add_undo_method(this,"_update_graph");
  887. undo_redo->commit_action();
  888. } else if (p_button==0) {
  889. }
  890. return; //or crash because it will become invalid
  891. }
  892. if (ti->get_parent()==root->get_children()->get_next()) {
  893. //edit/remove variable
  894. String name = ti->get_metadata(0);
  895. if (p_button==1) {
  896. undo_redo->create_action(TTR("Remove Variable"));
  897. undo_redo->add_do_method(script.ptr(),"remove_variable",name);
  898. undo_redo->add_undo_method(script.ptr(),"add_variable",name,script->get_variable_default_value(name));
  899. undo_redo->add_undo_method(script.ptr(),"set_variable_info",name,script->call("get_variable_info",name)); //return as dict
  900. undo_redo->add_do_method(this,"_update_members");
  901. undo_redo->add_undo_method(this,"_update_members");
  902. undo_redo->commit_action();
  903. return; //or crash because it will become invalid
  904. } else if (p_button==0) {
  905. variable_editor->edit(name);
  906. edit_variable_dialog->set_title(TTR("Editing Variable:")+" "+name);
  907. edit_variable_dialog->popup_centered_minsize(Size2(400,200)*EDSCALE);
  908. }
  909. }
  910. if (ti->get_parent()==root->get_children()->get_next()->get_next()) {
  911. //edit/remove variable
  912. String name = ti->get_metadata(0);
  913. if (p_button==1) {
  914. undo_redo->create_action(TTR("Remove Signal"));
  915. undo_redo->add_do_method(script.ptr(),"remove_custom_signal",name);
  916. undo_redo->add_undo_method(script.ptr(),"add_custom_signal",name);
  917. for(int i=0;i<script->custom_signal_get_argument_count(name);i++) {
  918. undo_redo->add_undo_method(script.ptr(),"custom_signal_add_argument",name,script->custom_signal_get_argument_name(name,i),script->custom_signal_get_argument_type(name,i));
  919. }
  920. undo_redo->add_do_method(this,"_update_members");
  921. undo_redo->add_undo_method(this,"_update_members");
  922. undo_redo->commit_action();
  923. } else if (p_button==0) {
  924. signal_editor->edit(name);
  925. edit_signal_dialog->set_title(TTR("Editing Signal:")+" "+name);
  926. edit_signal_dialog->popup_centered_minsize(Size2(400,300)*EDSCALE);
  927. }
  928. return; //or crash because it will become invalid
  929. }
  930. }
  931. }
  932. void VisualScriptEditor::_expression_text_changed(const String& p_text,int p_id) {
  933. Ref<VisualScriptExpression> vse = script->get_node(edited_func,p_id);
  934. if (!vse.is_valid())
  935. return;
  936. updating_graph=true;
  937. undo_redo->create_action(TTR("Change Expression"),UndoRedo::MERGE_ENDS);
  938. undo_redo->add_do_property(vse.ptr(),"expression",p_text);
  939. undo_redo->add_undo_property(vse.ptr(),"expression",vse->get("expression"));
  940. undo_redo->add_do_method(this,"_update_graph",p_id);
  941. undo_redo->add_undo_method(this,"_update_graph",p_id);
  942. undo_redo->commit_action();
  943. Node *node = graph->get_node(itos(p_id));
  944. if (node->cast_to<Control>())
  945. node->cast_to<Control>()->set_size(Vector2(1,1)); //shrink if text is smaller
  946. updating_graph=false;
  947. }
  948. void VisualScriptEditor::_available_node_doubleclicked() {
  949. TreeItem *item = nodes->get_selected();
  950. if (!item)
  951. return;
  952. String which = item->get_metadata(0);
  953. if (which==String())
  954. return;
  955. Vector2 ofs = graph->get_scroll_ofs() + graph->get_size() * 0.5;
  956. if (graph->is_using_snap()) {
  957. int snap = graph->get_snap();
  958. ofs = ofs.snapped(Vector2(snap,snap));
  959. }
  960. ofs/=EDSCALE;
  961. while(true) {
  962. bool exists=false;
  963. List<int> existing;
  964. script->get_node_list(edited_func,&existing);
  965. for (List<int>::Element *E=existing.front();E;E=E->next()) {
  966. Point2 pos = script->get_node_pos(edited_func,E->get());
  967. if (pos.distance_to(ofs)<15) {
  968. ofs+=Vector2(graph->get_snap(),graph->get_snap());
  969. exists=true;
  970. break;
  971. }
  972. }
  973. if (exists)
  974. continue;
  975. break;
  976. }
  977. Ref<VisualScriptNode> vnode = VisualScriptLanguage::singleton->create_node_from_name(which);
  978. int new_id = script->get_available_id();
  979. undo_redo->create_action(TTR("Add Node"));
  980. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,vnode,ofs);
  981. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  982. undo_redo->add_do_method(this,"_update_graph");
  983. undo_redo->add_undo_method(this,"_update_graph");
  984. undo_redo->commit_action();
  985. Node* node = graph->get_node(itos(new_id));
  986. if (node) {
  987. graph->set_selected(node);
  988. _node_selected(node);
  989. }
  990. }
  991. void VisualScriptEditor::_update_available_nodes() {
  992. nodes->clear();
  993. TreeItem *root = nodes->create_item();
  994. Map<String,TreeItem*> path_cache;
  995. String filter = node_filter->get_text();
  996. List<String> fnodes;
  997. VisualScriptLanguage::singleton->get_registered_node_names(&fnodes);
  998. for (List<String>::Element *E=fnodes.front();E;E=E->next()) {
  999. Vector<String> path = E->get().split("/");
  1000. if (filter!=String() && path.size() && path[path.size()-1].findn(filter)==-1)
  1001. continue;
  1002. String sp;
  1003. TreeItem* parent=root;
  1004. for(int i=0;i<path.size()-1;i++) {
  1005. if (i>0)
  1006. sp+=",";
  1007. sp+=path[i];
  1008. if (!path_cache.has(sp)) {
  1009. TreeItem* pathn = nodes->create_item(parent);
  1010. pathn->set_selectable(0,false);
  1011. pathn->set_text(0,path[i].capitalize());
  1012. path_cache[sp]=pathn;
  1013. parent=pathn;
  1014. if (filter==String()) {
  1015. pathn->set_collapsed(true); //should remember state
  1016. }
  1017. } else {
  1018. parent=path_cache[sp];
  1019. }
  1020. }
  1021. TreeItem *item = nodes->create_item(parent);
  1022. item->set_text(0,path[path.size()-1].capitalize());
  1023. item->set_selectable(0,true);
  1024. item->set_metadata(0,E->get());
  1025. }
  1026. }
  1027. String VisualScriptEditor::_validate_name(const String& p_name) const {
  1028. String valid=p_name;
  1029. int counter=1;
  1030. while(true) {
  1031. bool exists = script->has_function(valid) || script->has_variable(valid) || script->has_custom_signal(valid);
  1032. if (exists) {
  1033. counter++;
  1034. valid=p_name+"_"+itos(counter);
  1035. continue;
  1036. }
  1037. break;
  1038. }
  1039. return valid;
  1040. }
  1041. void VisualScriptEditor::_on_nodes_delete() {
  1042. List<int> to_erase;
  1043. for(int i=0;i<graph->get_child_count();i++) {
  1044. GraphNode *gn = graph->get_child(i)->cast_to<GraphNode>();
  1045. if (gn) {
  1046. if (gn->is_selected() && gn->is_close_button_visible()) {
  1047. to_erase.push_back(gn->get_name().operator String().to_int());
  1048. }
  1049. }
  1050. }
  1051. if (to_erase.empty())
  1052. return;
  1053. undo_redo->create_action("Remove VisualScript Nodes");
  1054. for(List<int>::Element*F=to_erase.front();F;F=F->next()) {
  1055. undo_redo->add_do_method(script.ptr(),"remove_node",edited_func,F->get());
  1056. undo_redo->add_undo_method(script.ptr(),"add_node",edited_func,F->get(),script->get_node(edited_func,F->get()),script->get_node_pos(edited_func,F->get()));
  1057. List<VisualScript::SequenceConnection> sequence_conns;
  1058. script->get_sequence_connection_list(edited_func,&sequence_conns);
  1059. for (List<VisualScript::SequenceConnection>::Element *E=sequence_conns.front();E;E=E->next()) {
  1060. if (E->get().from_node==F->get() || E->get().to_node==F->get()) {
  1061. undo_redo->add_undo_method(script.ptr(),"sequence_connect",edited_func,E->get().from_node,E->get().from_output,E->get().to_node);
  1062. }
  1063. }
  1064. List<VisualScript::DataConnection> data_conns;
  1065. script->get_data_connection_list(edited_func,&data_conns);
  1066. for (List<VisualScript::DataConnection>::Element *E=data_conns.front();E;E=E->next()) {
  1067. if (E->get().from_node==F->get() || E->get().to_node==F->get()) {
  1068. undo_redo->add_undo_method(script.ptr(),"data_connect",edited_func,E->get().from_node,E->get().from_port,E->get().to_node,E->get().to_port);
  1069. }
  1070. }
  1071. }
  1072. undo_redo->add_do_method(this,"_update_graph");
  1073. undo_redo->add_undo_method(this,"_update_graph");
  1074. undo_redo->commit_action();
  1075. }
  1076. void VisualScriptEditor::_on_nodes_duplicate() {
  1077. List<int> to_duplicate;
  1078. for(int i=0;i<graph->get_child_count();i++) {
  1079. GraphNode *gn = graph->get_child(i)->cast_to<GraphNode>();
  1080. if (gn) {
  1081. if (gn->is_selected() && gn->is_close_button_visible()) {
  1082. to_duplicate.push_back(gn->get_name().operator String().to_int());
  1083. }
  1084. }
  1085. }
  1086. if (to_duplicate.empty())
  1087. return;
  1088. undo_redo->create_action("Duplicate VisualScript Nodes");
  1089. int idc=script->get_available_id()+1;
  1090. Set<int> to_select;
  1091. for(List<int>::Element*F=to_duplicate.front();F;F=F->next()) {
  1092. Ref<VisualScriptNode> node = script->get_node(edited_func,F->get());
  1093. Ref<VisualScriptNode> dupe = node->duplicate();
  1094. int new_id = idc++;
  1095. to_select.insert(new_id);
  1096. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,dupe,script->get_node_pos(edited_func,F->get())+Vector2(20,20));
  1097. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  1098. }
  1099. undo_redo->add_do_method(this,"_update_graph");
  1100. undo_redo->add_undo_method(this,"_update_graph");
  1101. undo_redo->commit_action();
  1102. for(int i=0;i<graph->get_child_count();i++) {
  1103. GraphNode *gn = graph->get_child(i)->cast_to<GraphNode>();
  1104. if (gn) {
  1105. int id = gn->get_name().operator String().to_int();
  1106. gn->set_selected(to_select.has(id));
  1107. }
  1108. }
  1109. if (to_select.size()) {
  1110. EditorNode::get_singleton()->push_item(script->get_node(edited_func,to_select.front()->get()).ptr());
  1111. }
  1112. }
  1113. void VisualScriptEditor::_input(const InputEvent& p_event) {
  1114. if (p_event.type==InputEvent::MOUSE_BUTTON && !p_event.mouse_button.pressed && p_event.mouse_button.button_index==BUTTON_LEFT) {
  1115. revert_on_drag=String(); //so we can still drag functions
  1116. }
  1117. }
  1118. Variant VisualScriptEditor::get_drag_data_fw(const Point2& p_point,Control* p_from) {
  1119. if (p_from==nodes) {
  1120. TreeItem *it = nodes->get_item_at_pos(p_point);
  1121. if (!it)
  1122. return Variant();
  1123. String type=it->get_metadata(0);
  1124. if (type==String())
  1125. return Variant();
  1126. Dictionary dd;
  1127. dd["type"]="visual_script_node_drag";
  1128. dd["node_type"]=type;
  1129. Label *label = memnew(Label);
  1130. label->set_text(it->get_text(0));
  1131. set_drag_preview(label);
  1132. return dd;
  1133. }
  1134. if (p_from==members) {
  1135. TreeItem *it = members->get_item_at_pos(p_point);
  1136. if (!it)
  1137. return Variant();
  1138. String type=it->get_metadata(0);
  1139. if (type==String())
  1140. return Variant();
  1141. Dictionary dd;
  1142. TreeItem *root=members->get_root();
  1143. if (it->get_parent()==root->get_children()) {
  1144. dd["type"]="visual_script_function_drag";
  1145. dd["function"]=type;
  1146. if (revert_on_drag!=String()) {
  1147. edited_func=revert_on_drag; //revert so function does not change
  1148. revert_on_drag=String();
  1149. _update_graph();
  1150. }
  1151. } else if (it->get_parent()==root->get_children()->get_next()) {
  1152. dd["type"]="visual_script_variable_drag";
  1153. dd["variable"]=type;
  1154. } else if (it->get_parent()==root->get_children()->get_next()->get_next()) {
  1155. dd["type"]="visual_script_signal_drag";
  1156. dd["signal"]=type;
  1157. } else {
  1158. return Variant();
  1159. }
  1160. Label *label = memnew(Label);
  1161. label->set_text(it->get_text(0));
  1162. set_drag_preview(label);
  1163. return dd;
  1164. }
  1165. return Variant();
  1166. }
  1167. bool VisualScriptEditor::can_drop_data_fw(const Point2& p_point,const Variant& p_data,Control* p_from) const{
  1168. if (p_from==graph) {
  1169. Dictionary d = p_data;
  1170. if (d.has("type") &&
  1171. (
  1172. String(d["type"])=="visual_script_node_drag" ||
  1173. String(d["type"])=="visual_script_function_drag" ||
  1174. String(d["type"])=="visual_script_variable_drag" ||
  1175. String(d["type"])=="visual_script_signal_drag" ||
  1176. String(d["type"])=="obj_property" ||
  1177. String(d["type"])=="resource" ||
  1178. String(d["type"])=="files" ||
  1179. String(d["type"])=="nodes"
  1180. ) ) {
  1181. if (String(d["type"])=="obj_property") {
  1182. #ifdef OSX_ENABLED
  1183. const_cast<VisualScriptEditor*>(this)->_show_hint(TTR("Hold Meta to drop a Getter. Hold Shift to drop a generic signature."));
  1184. #else
  1185. const_cast<VisualScriptEditor*>(this)->_show_hint(TTR("Hold Ctrl to drop a Getter. Hold Shift to drop a generic signature."));
  1186. #endif
  1187. }
  1188. if (String(d["type"])=="nodes") {
  1189. #ifdef OSX_ENABLED
  1190. const_cast<VisualScriptEditor*>(this)->_show_hint(TTR("Hold Meta to drop a simple reference to the node."));
  1191. #else
  1192. const_cast<VisualScriptEditor*>(this)->_show_hint(TTR("Hold Ctrl to drop a simple reference to the node."));
  1193. #endif
  1194. }
  1195. if (String(d["type"])=="visual_script_variable_drag") {
  1196. #ifdef OSX_ENABLED
  1197. const_cast<VisualScriptEditor*>(this)->_show_hint(TTR("Hold Meta to drop a Variable Setter."));
  1198. #else
  1199. const_cast<VisualScriptEditor*>(this)->_show_hint(TTR("Hold Ctrl to drop a Variable Setter."));
  1200. #endif
  1201. }
  1202. return true;
  1203. }
  1204. }
  1205. return false;
  1206. }
  1207. #ifdef TOOLS_ENABLED
  1208. static Node* _find_script_node(Node* p_edited_scene,Node* p_current_node,const Ref<Script> &script) {
  1209. if (p_edited_scene!=p_current_node && p_current_node->get_owner()!=p_edited_scene)
  1210. return NULL;
  1211. Ref<Script> scr = p_current_node->get_script();
  1212. if (scr.is_valid() && scr==script)
  1213. return p_current_node;
  1214. for(int i=0;i<p_current_node->get_child_count();i++) {
  1215. Node *n = _find_script_node(p_edited_scene,p_current_node->get_child(i),script);
  1216. if (n)
  1217. return n;
  1218. }
  1219. return NULL;
  1220. }
  1221. #endif
  1222. void VisualScriptEditor::drop_data_fw(const Point2& p_point,const Variant& p_data,Control* p_from){
  1223. if (p_from==graph) {
  1224. Dictionary d = p_data;
  1225. if (d.has("type") && String(d["type"])=="visual_script_node_drag") {
  1226. Vector2 ofs = graph->get_scroll_ofs() + p_point;
  1227. if (graph->is_using_snap()) {
  1228. int snap = graph->get_snap();
  1229. ofs = ofs.snapped(Vector2(snap,snap));
  1230. }
  1231. ofs/=EDSCALE;
  1232. Ref<VisualScriptNode> vnode = VisualScriptLanguage::singleton->create_node_from_name(d["node_type"]);
  1233. int new_id = script->get_available_id();
  1234. undo_redo->create_action(TTR("Add Node"));
  1235. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,vnode,ofs);
  1236. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  1237. undo_redo->add_do_method(this,"_update_graph");
  1238. undo_redo->add_undo_method(this,"_update_graph");
  1239. undo_redo->commit_action();
  1240. Node* node = graph->get_node(itos(new_id));
  1241. if (node) {
  1242. graph->set_selected(node);
  1243. _node_selected(node);
  1244. }
  1245. }
  1246. if (d.has("type") && String(d["type"])=="visual_script_variable_drag") {
  1247. #ifdef OSX_ENABLED
  1248. bool use_set = Input::get_singleton()->is_key_pressed(KEY_META);
  1249. #else
  1250. bool use_set = Input::get_singleton()->is_key_pressed(KEY_CONTROL);
  1251. #endif
  1252. Vector2 ofs = graph->get_scroll_ofs() + p_point;
  1253. if (graph->is_using_snap()) {
  1254. int snap = graph->get_snap();
  1255. ofs = ofs.snapped(Vector2(snap,snap));
  1256. }
  1257. ofs/=EDSCALE;
  1258. Ref<VisualScriptNode> vnode;
  1259. if (use_set) {
  1260. Ref<VisualScriptVariableSet> vnodes;
  1261. vnodes.instance();
  1262. vnodes->set_variable(d["variable"]);
  1263. vnode=vnodes;
  1264. } else {
  1265. Ref<VisualScriptVariableGet> vnodeg;
  1266. vnodeg.instance();
  1267. vnodeg->set_variable(d["variable"]);
  1268. vnode=vnodeg;
  1269. }
  1270. int new_id = script->get_available_id();
  1271. undo_redo->create_action(TTR("Add Node"));
  1272. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,vnode,ofs);
  1273. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  1274. undo_redo->add_do_method(this,"_update_graph");
  1275. undo_redo->add_undo_method(this,"_update_graph");
  1276. undo_redo->commit_action();
  1277. Node* node = graph->get_node(itos(new_id));
  1278. if (node) {
  1279. graph->set_selected(node);
  1280. _node_selected(node);
  1281. }
  1282. }
  1283. if (d.has("type") && String(d["type"])=="visual_script_function_drag") {
  1284. Vector2 ofs = graph->get_scroll_ofs() + p_point;
  1285. if (graph->is_using_snap()) {
  1286. int snap = graph->get_snap();
  1287. ofs = ofs.snapped(Vector2(snap,snap));
  1288. }
  1289. ofs/=EDSCALE;
  1290. Ref<VisualScriptFunctionCall> vnode;
  1291. vnode.instance();
  1292. vnode->set_call_mode(VisualScriptFunctionCall::CALL_MODE_SELF);
  1293. int new_id = script->get_available_id();
  1294. undo_redo->create_action(TTR("Add Node"));
  1295. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,vnode,ofs);
  1296. undo_redo->add_do_method(vnode.ptr(),"set_base_type",script->get_instance_base_type());
  1297. undo_redo->add_do_method(vnode.ptr(),"set_function",d["function"]);
  1298. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  1299. undo_redo->add_do_method(this,"_update_graph");
  1300. undo_redo->add_undo_method(this,"_update_graph");
  1301. undo_redo->commit_action();
  1302. Node* node = graph->get_node(itos(new_id));
  1303. if (node) {
  1304. graph->set_selected(node);
  1305. _node_selected(node);
  1306. }
  1307. }
  1308. if (d.has("type") && String(d["type"])=="visual_script_signal_drag") {
  1309. Vector2 ofs = graph->get_scroll_ofs() + p_point;
  1310. if (graph->is_using_snap()) {
  1311. int snap = graph->get_snap();
  1312. ofs = ofs.snapped(Vector2(snap,snap));
  1313. }
  1314. ofs/=EDSCALE;
  1315. Ref<VisualScriptEmitSignal> vnode;
  1316. vnode.instance();
  1317. vnode->set_signal(d["signal"]);
  1318. int new_id = script->get_available_id();
  1319. undo_redo->create_action(TTR("Add Node"));
  1320. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,vnode,ofs);
  1321. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  1322. undo_redo->add_do_method(this,"_update_graph");
  1323. undo_redo->add_undo_method(this,"_update_graph");
  1324. undo_redo->commit_action();
  1325. Node* node = graph->get_node(itos(new_id));
  1326. if (node) {
  1327. graph->set_selected(node);
  1328. _node_selected(node);
  1329. }
  1330. }
  1331. if (d.has("type") && String(d["type"])=="resource") {
  1332. Vector2 ofs = graph->get_scroll_ofs() + p_point;
  1333. if (graph->is_using_snap()) {
  1334. int snap = graph->get_snap();
  1335. ofs = ofs.snapped(Vector2(snap,snap));
  1336. }
  1337. ofs/=EDSCALE;
  1338. Ref<VisualScriptPreload> prnode;
  1339. prnode.instance();
  1340. prnode->set_preload(d["resource"]);
  1341. int new_id = script->get_available_id();
  1342. undo_redo->create_action(TTR("Add Preload Node"));
  1343. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,prnode,ofs);
  1344. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  1345. undo_redo->add_do_method(this,"_update_graph");
  1346. undo_redo->add_undo_method(this,"_update_graph");
  1347. undo_redo->commit_action();
  1348. Node* node = graph->get_node(itos(new_id));
  1349. if (node) {
  1350. graph->set_selected(node);
  1351. _node_selected(node);
  1352. }
  1353. }
  1354. if (d.has("type") && String(d["type"])=="files") {
  1355. Vector2 ofs = graph->get_scroll_ofs() + p_point;
  1356. if (graph->is_using_snap()) {
  1357. int snap = graph->get_snap();
  1358. ofs = ofs.snapped(Vector2(snap,snap));
  1359. }
  1360. ofs/=EDSCALE;
  1361. Array files = d["files"];
  1362. List<int> new_ids;
  1363. int new_id = script->get_available_id();
  1364. if (files.size()) {
  1365. undo_redo->create_action(TTR("Add Preload Node"));
  1366. for(int i=0;i<files.size();i++) {
  1367. Ref<Resource> res = ResourceLoader::load(files[i]);
  1368. if (!res.is_valid())
  1369. continue;
  1370. Ref<VisualScriptPreload> prnode;
  1371. prnode.instance();
  1372. prnode->set_preload(res);
  1373. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,prnode,ofs);
  1374. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  1375. new_ids.push_back(new_id);
  1376. new_id++;
  1377. ofs+=Vector2(20,20)*EDSCALE;
  1378. }
  1379. undo_redo->add_do_method(this,"_update_graph");
  1380. undo_redo->add_undo_method(this,"_update_graph");
  1381. undo_redo->commit_action();
  1382. }
  1383. for(List<int>::Element *E=new_ids.front();E;E=E->next()) {
  1384. Node* node = graph->get_node(itos(E->get()));
  1385. if (node) {
  1386. graph->set_selected(node);
  1387. _node_selected(node);
  1388. }
  1389. }
  1390. }
  1391. if (d.has("type") && String(d["type"])=="nodes") {
  1392. Node* sn = _find_script_node(get_tree()->get_edited_scene_root(),get_tree()->get_edited_scene_root(),script);
  1393. if (!sn) {
  1394. EditorNode::get_singleton()->show_warning("Can't drop nodes because script '"+get_name()+"' is not used in this scene.");
  1395. return;
  1396. }
  1397. #ifdef OSX_ENABLED
  1398. bool use_node = Input::get_singleton()->is_key_pressed(KEY_META);
  1399. #else
  1400. bool use_node = Input::get_singleton()->is_key_pressed(KEY_CONTROL);
  1401. #endif
  1402. Array nodes = d["nodes"];
  1403. Vector2 ofs = graph->get_scroll_ofs() + p_point;
  1404. if (graph->is_using_snap()) {
  1405. int snap = graph->get_snap();
  1406. ofs = ofs.snapped(Vector2(snap,snap));
  1407. }
  1408. ofs/=EDSCALE;
  1409. undo_redo->create_action(TTR("Add Node(s) From Tree"));
  1410. int base_id = script->get_available_id();
  1411. if (nodes.size()>1) {
  1412. use_node=true;
  1413. }
  1414. for(int i=0;i<nodes.size();i++) {
  1415. NodePath np = nodes[i];
  1416. Node *node = get_node(np);
  1417. if (!node) {
  1418. continue;
  1419. }
  1420. Ref<VisualScriptNode> n;
  1421. if (use_node) {
  1422. Ref<VisualScriptSceneNode> scene_node;
  1423. scene_node.instance();
  1424. scene_node->set_node_path(sn->get_path_to(node));
  1425. n=scene_node;
  1426. } else {
  1427. Ref<VisualScriptFunctionCall> call;
  1428. call.instance();
  1429. call->set_call_mode(VisualScriptFunctionCall::CALL_MODE_NODE_PATH);
  1430. call->set_base_path(sn->get_path_to(node));
  1431. call->set_base_type(node->get_class());
  1432. n=call;
  1433. method_select->select_method_from_instance(node);
  1434. selecting_method_id=base_id;
  1435. }
  1436. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,base_id,n,ofs);
  1437. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,base_id);
  1438. base_id++;
  1439. ofs+=Vector2(25,25);
  1440. }
  1441. undo_redo->add_do_method(this,"_update_graph");
  1442. undo_redo->add_undo_method(this,"_update_graph");
  1443. undo_redo->commit_action();
  1444. }
  1445. if (d.has("type") && String(d["type"])=="obj_property") {
  1446. Node* sn = _find_script_node(get_tree()->get_edited_scene_root(),get_tree()->get_edited_scene_root(),script);
  1447. if (!sn && !Input::get_singleton()->is_key_pressed(KEY_SHIFT)) {
  1448. EditorNode::get_singleton()->show_warning("Can't drop properties because script '"+get_name()+"' is not used in this scene.\nDrop holding 'Shift' to just copy the signature.");
  1449. return;
  1450. }
  1451. Object *obj=d["object"];
  1452. if (!obj)
  1453. return;
  1454. Node *node = obj->cast_to<Node>();
  1455. Vector2 ofs = graph->get_scroll_ofs() + p_point;
  1456. if (graph->is_using_snap()) {
  1457. int snap = graph->get_snap();
  1458. ofs = ofs.snapped(Vector2(snap,snap));
  1459. }
  1460. ofs/=EDSCALE;
  1461. #ifdef OSX_ENABLED
  1462. bool use_get = Input::get_singleton()->is_key_pressed(KEY_META);
  1463. #else
  1464. bool use_get = Input::get_singleton()->is_key_pressed(KEY_CONTROL);
  1465. #endif
  1466. if (!node || Input::get_singleton()->is_key_pressed(KEY_SHIFT)) {
  1467. if (use_get)
  1468. undo_redo->create_action(TTR("Add Getter Property"));
  1469. else
  1470. undo_redo->create_action(TTR("Add Setter Property"));
  1471. int base_id = script->get_available_id();
  1472. Ref<VisualScriptNode> vnode;
  1473. if (!use_get) {
  1474. Ref<VisualScriptPropertySet> pset;
  1475. pset.instance();
  1476. pset->set_call_mode(VisualScriptPropertySet::CALL_MODE_INSTANCE);
  1477. pset->set_base_type(obj->get_class());
  1478. /*if (use_value) {
  1479. pset->set_use_builtin_value(true);
  1480. pset->set_builtin_value(d["value"]);
  1481. }*/
  1482. vnode=pset;
  1483. } else {
  1484. Ref<VisualScriptPropertyGet> pget;
  1485. pget.instance();
  1486. pget->set_call_mode(VisualScriptPropertyGet::CALL_MODE_INSTANCE);
  1487. pget->set_base_type(obj->get_class());
  1488. vnode=pget;
  1489. }
  1490. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,base_id,vnode,ofs);
  1491. undo_redo->add_do_method(vnode.ptr(),"set_property",d["property"]);
  1492. if (!use_get) {
  1493. undo_redo->add_do_method(vnode.ptr(),"set_default_input_value",0,d["value"]);
  1494. }
  1495. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,base_id);
  1496. undo_redo->add_do_method(this,"_update_graph");
  1497. undo_redo->add_undo_method(this,"_update_graph");
  1498. undo_redo->commit_action();
  1499. } else {
  1500. if (use_get)
  1501. undo_redo->create_action(TTR("Add Getter Property"));
  1502. else
  1503. undo_redo->create_action(TTR("Add Setter Property"));
  1504. int base_id = script->get_available_id();
  1505. Ref<VisualScriptNode> vnode;
  1506. if (!use_get) {
  1507. Ref<VisualScriptPropertySet> pset;
  1508. pset.instance();
  1509. if (sn==node) {
  1510. pset->set_call_mode(VisualScriptPropertySet::CALL_MODE_SELF);
  1511. } else {
  1512. pset->set_call_mode(VisualScriptPropertySet::CALL_MODE_NODE_PATH);
  1513. pset->set_base_path(sn->get_path_to(node));
  1514. }
  1515. vnode=pset;
  1516. } else {
  1517. Ref<VisualScriptPropertyGet> pget;
  1518. pget.instance();
  1519. if (sn==node) {
  1520. pget->set_call_mode(VisualScriptPropertyGet::CALL_MODE_SELF);
  1521. } else {
  1522. pget->set_call_mode(VisualScriptPropertyGet::CALL_MODE_NODE_PATH);
  1523. pget->set_base_path(sn->get_path_to(node));
  1524. }
  1525. vnode=pget;
  1526. }
  1527. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,base_id,vnode,ofs);
  1528. undo_redo->add_do_method(vnode.ptr(),"set_property",d["property"]);
  1529. if (!use_get) {
  1530. undo_redo->add_do_method(vnode.ptr(),"set_default_input_value",0,d["value"]);
  1531. }
  1532. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,base_id);
  1533. undo_redo->add_do_method(this,"_update_graph");
  1534. undo_redo->add_undo_method(this,"_update_graph");
  1535. undo_redo->commit_action();
  1536. }
  1537. }
  1538. }
  1539. }
  1540. void VisualScriptEditor::_selected_method(const String& p_method) {
  1541. Ref<VisualScriptFunctionCall> vsfc = script->get_node(edited_func,selecting_method_id);
  1542. if (!vsfc.is_valid())
  1543. return;
  1544. vsfc->set_function(p_method);
  1545. }
  1546. void VisualScriptEditor::_draw_color_over_button(Object* obj,Color p_color) {
  1547. Button *button = obj->cast_to<Button>();
  1548. if (!button)
  1549. return;
  1550. Ref<StyleBox> normal = get_stylebox("normal","Button" );
  1551. button->draw_rect(Rect2(normal->get_offset(),button->get_size()-normal->get_minimum_size()),p_color);
  1552. }
  1553. void VisualScriptEditor::_button_resource_previewed(const String& p_path,const Ref<Texture>& p_preview,Variant p_ud) {
  1554. Array ud=p_ud;
  1555. ERR_FAIL_COND(ud.size()!=2);
  1556. ObjectID id = ud[0];
  1557. Object *obj = ObjectDB::get_instance(id);
  1558. if (!obj)
  1559. return;
  1560. Button *b = obj->cast_to<Button>();
  1561. ERR_FAIL_COND(!b);
  1562. if (p_preview.is_null()) {
  1563. b->set_text(ud[1]);
  1564. } else {
  1565. b->set_icon(p_preview);
  1566. }
  1567. }
  1568. /////////////////////////
  1569. void VisualScriptEditor::apply_code() {
  1570. }
  1571. Ref<Script> VisualScriptEditor::get_edited_script() const{
  1572. return script;
  1573. }
  1574. Vector<String> VisualScriptEditor::get_functions(){
  1575. return Vector<String>();
  1576. }
  1577. void VisualScriptEditor::set_edited_script(const Ref<Script>& p_script){
  1578. script=p_script;
  1579. signal_editor->script=p_script;
  1580. signal_editor->undo_redo=undo_redo;
  1581. variable_editor->script=p_script;
  1582. variable_editor->undo_redo=undo_redo;
  1583. script->connect("node_ports_changed",this,"_node_ports_changed");
  1584. _update_members();
  1585. _update_available_nodes();
  1586. }
  1587. void VisualScriptEditor::reload_text(){
  1588. }
  1589. String VisualScriptEditor::get_name(){
  1590. String name;
  1591. if (script->get_path().find("local://")==-1 && script->get_path().find("::")==-1) {
  1592. name=script->get_path().get_file();
  1593. if (is_unsaved()) {
  1594. name+="(*)";
  1595. }
  1596. } else if (script->get_name()!="")
  1597. name=script->get_name();
  1598. else
  1599. name=script->get_class()+"("+itos(script->get_instance_ID())+")";
  1600. return name;
  1601. }
  1602. Ref<Texture> VisualScriptEditor::get_icon(){
  1603. return Control::get_icon("VisualScript","EditorIcons");
  1604. }
  1605. bool VisualScriptEditor::is_unsaved(){
  1606. #ifdef TOOLS_ENABLED
  1607. return script->is_edited() || script->are_subnodes_edited();
  1608. #else
  1609. return false;
  1610. #endif
  1611. }
  1612. Variant VisualScriptEditor::get_edit_state(){
  1613. Dictionary d;
  1614. d["function"]=edited_func;
  1615. d["scroll"]=graph->get_scroll_ofs();
  1616. d["zoom"]=graph->get_zoom();
  1617. d["using_snap"]=graph->is_using_snap();
  1618. d["snap"]=graph->get_snap();
  1619. return d;
  1620. }
  1621. void VisualScriptEditor::set_edit_state(const Variant& p_state){
  1622. Dictionary d = p_state;
  1623. if (d.has("function")) {
  1624. edited_func=p_state;
  1625. selected=edited_func;
  1626. }
  1627. _update_graph();
  1628. _update_members();
  1629. if (d.has("scroll")) {
  1630. graph->set_scroll_ofs(d["scroll"]);
  1631. }
  1632. if (d.has("zoom")) {
  1633. graph->set_zoom(d["zoom"]);
  1634. }
  1635. if (d.has("snap")) {
  1636. graph->set_snap(d["snap"]);
  1637. }
  1638. if (d.has("snap_enabled")) {
  1639. graph->set_use_snap(d["snap_enabled"]);
  1640. }
  1641. }
  1642. void VisualScriptEditor::_center_on_node(int p_id) {
  1643. Node *n = graph->get_node(itos(p_id));
  1644. if (!n)
  1645. return;
  1646. GraphNode *gn = n->cast_to<GraphNode>();
  1647. if (gn) {
  1648. gn->set_selected(true);
  1649. Vector2 new_scroll = gn->get_offset() - graph->get_size()*0.5 + gn->get_size()*0.5;
  1650. graph->set_scroll_ofs( new_scroll );
  1651. script->set_function_scroll(edited_func,new_scroll/EDSCALE);
  1652. script->set_edited(true); //so it's saved
  1653. }
  1654. }
  1655. void VisualScriptEditor::goto_line(int p_line, bool p_with_error){
  1656. p_line+=1; //add one because script lines begin from 0.
  1657. if (p_with_error)
  1658. error_line=p_line;
  1659. List<StringName> functions;
  1660. script->get_function_list(&functions);
  1661. for (List<StringName>::Element *E=functions.front();E;E=E->next()) {
  1662. if (script->has_node(E->get(),p_line)) {
  1663. edited_func=E->get();
  1664. selected=edited_func;
  1665. _update_graph();
  1666. _update_members();
  1667. call_deferred("_center_on_node",p_line); //editor might be just created and size might not exist yet
  1668. return;
  1669. }
  1670. }
  1671. }
  1672. void VisualScriptEditor::trim_trailing_whitespace(){
  1673. }
  1674. void VisualScriptEditor::ensure_focus(){
  1675. graph->grab_focus();
  1676. }
  1677. void VisualScriptEditor::tag_saved_version(){
  1678. }
  1679. void VisualScriptEditor::reload(bool p_soft){
  1680. }
  1681. void VisualScriptEditor::get_breakpoints(List<int> *p_breakpoints){
  1682. List<StringName> functions;
  1683. script->get_function_list(&functions);
  1684. for (List<StringName>::Element *E=functions.front();E;E=E->next()) {
  1685. List<int> nodes;
  1686. script->get_node_list(E->get(),&nodes);
  1687. for (List<int>::Element *F=nodes.front();F;F=F->next()) {
  1688. Ref<VisualScriptNode> vsn = script->get_node(E->get(),F->get());
  1689. if (vsn->is_breakpoint()) {
  1690. p_breakpoints->push_back(F->get()-1); //subtract 1 because breakpoints in text start from zero
  1691. }
  1692. }
  1693. }
  1694. }
  1695. bool VisualScriptEditor::goto_method(const String& p_method){
  1696. if (!script->has_function(p_method))
  1697. return false;
  1698. edited_func=p_method;
  1699. selected=edited_func;
  1700. _update_members();
  1701. _update_graph();
  1702. return true;
  1703. }
  1704. void VisualScriptEditor::add_callback(const String& p_function,PoolStringArray p_args){
  1705. if (script->has_function(p_function)) {
  1706. edited_func=p_function;
  1707. selected=edited_func;
  1708. _update_members();
  1709. _update_graph();
  1710. return;
  1711. }
  1712. Ref<VisualScriptFunction> func;
  1713. func.instance();
  1714. for(int i=0;i<p_args.size();i++) {
  1715. String name = p_args[i];
  1716. Variant::Type type=Variant::NIL;
  1717. if (name.find(":")!=-1) {
  1718. String tt = name.get_slice(":",1);
  1719. name=name.get_slice(":",0);
  1720. for(int j=0;j<Variant::VARIANT_MAX;j++) {
  1721. String tname = Variant::get_type_name(Variant::Type(j));
  1722. if (tname==tt) {
  1723. type=Variant::Type(j);
  1724. break;
  1725. }
  1726. }
  1727. }
  1728. func->add_argument(type,name);
  1729. }
  1730. func->set_name(p_function);
  1731. script->add_function(p_function);
  1732. script->add_node(p_function,script->get_available_id(),func);
  1733. edited_func=p_function;
  1734. selected=edited_func;
  1735. _update_members();
  1736. _update_graph();
  1737. graph->call_deferred("set_scroll_ofs",script->get_function_scroll(edited_func)); //for first time it might need to be later
  1738. //undo_redo->clear_history();
  1739. }
  1740. void VisualScriptEditor::update_settings(){
  1741. _update_graph();
  1742. }
  1743. void VisualScriptEditor::set_debugger_active(bool p_active) {
  1744. if (!p_active) {
  1745. error_line=-1;
  1746. _update_graph(); //clear line break
  1747. }
  1748. }
  1749. void VisualScriptEditor::set_tooltip_request_func(String p_method,Object* p_obj){
  1750. }
  1751. Control *VisualScriptEditor::get_edit_menu(){
  1752. return edit_menu;
  1753. }
  1754. void VisualScriptEditor::_change_base_type() {
  1755. select_base_type->popup(true);
  1756. }
  1757. void VisualScriptEditor::_change_base_type_callback() {
  1758. String bt = select_base_type->get_selected_type();
  1759. ERR_FAIL_COND(bt==String());
  1760. undo_redo->create_action("Change Base Type");
  1761. undo_redo->add_do_method(script.ptr(),"set_instance_base_type",bt);
  1762. undo_redo->add_undo_method(script.ptr(),"set_instance_base_type",script->get_instance_base_type());
  1763. undo_redo->add_do_method(this,"_update_members");
  1764. undo_redo->add_undo_method(this,"_update_members");
  1765. undo_redo->commit_action();
  1766. }
  1767. void VisualScriptEditor::_node_selected(Node* p_node) {
  1768. Ref<VisualScriptNode> vnode = p_node->get_meta("__vnode");
  1769. if (vnode.is_null())
  1770. return;
  1771. EditorNode::get_singleton()->push_item(vnode.ptr()); //edit node in inspector
  1772. }
  1773. static bool _get_out_slot(const Ref<VisualScriptNode>& p_node,int p_slot,int& r_real_slot,bool& r_sequence) {
  1774. if (p_slot<p_node->get_output_sequence_port_count()) {
  1775. r_sequence=true;
  1776. r_real_slot=p_slot;
  1777. return true;
  1778. }
  1779. r_real_slot=p_slot-p_node->get_output_sequence_port_count();
  1780. r_sequence=false;
  1781. return (r_real_slot<p_node->get_output_value_port_count());
  1782. }
  1783. static bool _get_in_slot(const Ref<VisualScriptNode>& p_node,int p_slot,int& r_real_slot,bool& r_sequence) {
  1784. if (p_slot==0 && p_node->has_input_sequence_port()) {
  1785. r_sequence=true;
  1786. r_real_slot=0;
  1787. return true;
  1788. }
  1789. r_real_slot=p_slot-(p_node->has_input_sequence_port()?1:0);
  1790. r_sequence=false;
  1791. return r_real_slot<p_node->get_input_value_port_count();
  1792. }
  1793. void VisualScriptEditor::_begin_node_move() {
  1794. undo_redo->create_action("Move Node(s)");
  1795. }
  1796. void VisualScriptEditor::_end_node_move() {
  1797. undo_redo->commit_action();
  1798. }
  1799. void VisualScriptEditor::_move_node(String func,int p_id,const Vector2& p_to) {
  1800. if (func==String(edited_func)) {
  1801. Node* node = graph->get_node(itos(p_id));
  1802. if (node && node->cast_to<GraphNode>())
  1803. node->cast_to<GraphNode>()->set_offset(p_to);
  1804. }
  1805. script->set_node_pos(edited_func,p_id,p_to/EDSCALE);
  1806. }
  1807. void VisualScriptEditor::_node_moved(Vector2 p_from,Vector2 p_to, int p_id) {
  1808. undo_redo->add_do_method(this,"_move_node",String(edited_func),p_id,p_to);
  1809. undo_redo->add_undo_method(this,"_move_node",String(edited_func),p_id,p_from);
  1810. }
  1811. void VisualScriptEditor::_remove_node(int p_id) {
  1812. undo_redo->create_action("Remove VisualScript Node");
  1813. undo_redo->add_do_method(script.ptr(),"remove_node",edited_func,p_id);
  1814. undo_redo->add_undo_method(script.ptr(),"add_node",edited_func,p_id,script->get_node(edited_func,p_id),script->get_node_pos(edited_func,p_id));
  1815. List<VisualScript::SequenceConnection> sequence_conns;
  1816. script->get_sequence_connection_list(edited_func,&sequence_conns);
  1817. for (List<VisualScript::SequenceConnection>::Element *E=sequence_conns.front();E;E=E->next()) {
  1818. if (E->get().from_node==p_id || E->get().to_node==p_id) {
  1819. undo_redo->add_undo_method(script.ptr(),"sequence_connect",edited_func,E->get().from_node,E->get().from_output,E->get().to_node);
  1820. }
  1821. }
  1822. List<VisualScript::DataConnection> data_conns;
  1823. script->get_data_connection_list(edited_func,&data_conns);
  1824. for (List<VisualScript::DataConnection>::Element *E=data_conns.front();E;E=E->next()) {
  1825. if (E->get().from_node==p_id || E->get().to_node==p_id) {
  1826. undo_redo->add_undo_method(script.ptr(),"data_connect",edited_func,E->get().from_node,E->get().from_port,E->get().to_node,E->get().to_port);
  1827. }
  1828. }
  1829. undo_redo->add_do_method(this,"_update_graph");
  1830. undo_redo->add_undo_method(this,"_update_graph");
  1831. undo_redo->commit_action();
  1832. }
  1833. void VisualScriptEditor::_node_ports_changed(const String& p_func,int p_id) {
  1834. if (p_func!=String(edited_func))
  1835. return;
  1836. _update_graph(p_id);
  1837. }
  1838. void VisualScriptEditor::_graph_connected(const String& p_from,int p_from_slot,const String& p_to,int p_to_slot) {
  1839. Ref<VisualScriptNode> from_node = script->get_node(edited_func,p_from.to_int());
  1840. ERR_FAIL_COND(!from_node.is_valid());
  1841. bool from_seq;
  1842. int from_port;
  1843. if (!_get_out_slot(from_node,p_from_slot,from_port,from_seq))
  1844. return; //can't connect this, it' s invalid
  1845. Ref<VisualScriptNode> to_node = script->get_node(edited_func,p_to.to_int());
  1846. ERR_FAIL_COND(!to_node.is_valid());
  1847. bool to_seq;
  1848. int to_port;
  1849. if (!_get_in_slot(to_node,p_to_slot,to_port,to_seq))
  1850. return; //can't connect this, it' s invalid
  1851. ERR_FAIL_COND(from_seq!=to_seq);
  1852. undo_redo->create_action("Connect Nodes");
  1853. if (from_seq) {
  1854. undo_redo->add_do_method(script.ptr(),"sequence_connect",edited_func,p_from.to_int(),from_port,p_to.to_int());
  1855. undo_redo->add_undo_method(script.ptr(),"sequence_disconnect",edited_func,p_from.to_int(),from_port,p_to.to_int());
  1856. } else {
  1857. undo_redo->add_do_method(script.ptr(),"data_connect",edited_func,p_from.to_int(),from_port,p_to.to_int(),to_port);
  1858. undo_redo->add_undo_method(script.ptr(),"data_disconnect",edited_func,p_from.to_int(),from_port,p_to.to_int(),to_port);
  1859. //update nodes in sgraph
  1860. undo_redo->add_do_method(this,"_update_graph",p_from.to_int());
  1861. undo_redo->add_do_method(this,"_update_graph",p_to.to_int());
  1862. undo_redo->add_undo_method(this,"_update_graph",p_from.to_int());
  1863. undo_redo->add_undo_method(this,"_update_graph",p_to.to_int());
  1864. }
  1865. undo_redo->add_do_method(this,"_update_graph_connections");
  1866. undo_redo->add_undo_method(this,"_update_graph_connections");
  1867. undo_redo->commit_action();
  1868. }
  1869. void VisualScriptEditor::_graph_disconnected(const String& p_from,int p_from_slot,const String& p_to,int p_to_slot){
  1870. Ref<VisualScriptNode> from_node = script->get_node(edited_func,p_from.to_int());
  1871. ERR_FAIL_COND(!from_node.is_valid());
  1872. bool from_seq;
  1873. int from_port;
  1874. if (!_get_out_slot(from_node,p_from_slot,from_port,from_seq))
  1875. return; //can't connect this, it' s invalid
  1876. Ref<VisualScriptNode> to_node = script->get_node(edited_func,p_to.to_int());
  1877. ERR_FAIL_COND(!to_node.is_valid());
  1878. bool to_seq;
  1879. int to_port;
  1880. if (!_get_in_slot(to_node,p_to_slot,to_port,to_seq))
  1881. return; //can't connect this, it' s invalid
  1882. ERR_FAIL_COND(from_seq!=to_seq);
  1883. undo_redo->create_action("Connect Nodes");
  1884. if (from_seq) {
  1885. undo_redo->add_do_method(script.ptr(),"sequence_disconnect",edited_func,p_from.to_int(),from_port,p_to.to_int());
  1886. undo_redo->add_undo_method(script.ptr(),"sequence_connect",edited_func,p_from.to_int(),from_port,p_to.to_int());
  1887. } else {
  1888. undo_redo->add_do_method(script.ptr(),"data_disconnect",edited_func,p_from.to_int(),from_port,p_to.to_int(),to_port);
  1889. undo_redo->add_undo_method(script.ptr(),"data_connect",edited_func,p_from.to_int(),from_port,p_to.to_int(),to_port);
  1890. //update nodes in sgraph
  1891. undo_redo->add_do_method(this,"_update_graph",p_from.to_int());
  1892. undo_redo->add_do_method(this,"_update_graph",p_to.to_int());
  1893. undo_redo->add_undo_method(this,"_update_graph",p_from.to_int());
  1894. undo_redo->add_undo_method(this,"_update_graph",p_to.to_int());
  1895. }
  1896. undo_redo->add_do_method(this,"_update_graph_connections");
  1897. undo_redo->add_undo_method(this,"_update_graph_connections");
  1898. undo_redo->commit_action();
  1899. }
  1900. void VisualScriptEditor::_graph_connect_to_empty(const String& p_from,int p_from_slot,const Vector2& p_release_pos) {
  1901. Node* node = graph->get_node(p_from);
  1902. if (!node)
  1903. return;
  1904. GraphNode *gn = node->cast_to<GraphNode>();
  1905. if (!gn)
  1906. return;
  1907. Ref<VisualScriptNode> vsn = script->get_node(edited_func,p_from.to_int());
  1908. if (!vsn.is_valid())
  1909. return;
  1910. if (p_from_slot<vsn->get_output_sequence_port_count()) {
  1911. port_action_popup->clear();
  1912. port_action_popup->add_item(TTR("Condition"),CREATE_COND);
  1913. port_action_popup->add_item(TTR("Sequence"),CREATE_SEQUENCE);
  1914. port_action_popup->add_item(TTR("Switch"),CREATE_SWITCH);
  1915. port_action_popup->add_item(TTR("Iterator"),CREATE_ITERATOR);
  1916. port_action_popup->add_item(TTR("While"),CREATE_WHILE);
  1917. port_action_popup->add_item(TTR("Return"),CREATE_RETURN);
  1918. port_action_node=p_from.to_int();
  1919. port_action_output=p_from_slot;
  1920. } else {
  1921. port_action_popup->clear();
  1922. port_action_popup->add_item(TTR("Call"),CREATE_CALL);
  1923. port_action_popup->add_item(TTR("Get"),CREATE_GET);
  1924. port_action_popup->add_item(TTR("Set"),CREATE_SET);
  1925. port_action_output=p_from_slot-vsn->get_output_sequence_port_count();
  1926. port_action_node=p_from.to_int();
  1927. }
  1928. port_action_pos=p_release_pos;
  1929. port_action_popup->set_size(Size2(1,1));
  1930. port_action_popup->set_pos(graph->get_global_pos()+p_release_pos);
  1931. port_action_popup->popup();
  1932. }
  1933. VisualScriptNode::TypeGuess VisualScriptEditor::_guess_output_type(int p_node,int p_output,Set<int> &visited_nodes) {
  1934. VisualScriptNode::TypeGuess tg;
  1935. tg.type=Variant::NIL;
  1936. if (visited_nodes.has(p_node))
  1937. return tg; //no loop
  1938. visited_nodes.insert(p_node);
  1939. Ref<VisualScriptNode> node = script->get_node(edited_func,p_node);
  1940. if (!node.is_valid()) {
  1941. return tg;
  1942. }
  1943. Vector<VisualScriptNode::TypeGuess> in_guesses;
  1944. for(int i=0;i<node->get_input_value_port_count();i++) {
  1945. PropertyInfo pi = node->get_input_value_port_info(i);
  1946. VisualScriptNode::TypeGuess g;
  1947. g.type=pi.type;
  1948. if (g.type==Variant::NIL || g.type==Variant::OBJECT) {
  1949. //any or object input, must further guess what this is
  1950. int from_node;
  1951. int from_port;
  1952. if (script->get_input_value_port_connection_source(edited_func,p_node,i,&from_node,&from_port)) {
  1953. g = _guess_output_type(from_node,from_port,visited_nodes);
  1954. } else {
  1955. Variant defval = node->get_default_input_value(i);
  1956. if (defval.get_type()==Variant::OBJECT) {
  1957. Object *obj = defval;
  1958. if (obj) {
  1959. g.type=Variant::OBJECT;
  1960. g.GDCLASS=obj->get_class();
  1961. g.script=obj->get_script();
  1962. }
  1963. }
  1964. }
  1965. }
  1966. in_guesses.push_back(g);
  1967. }
  1968. return node->guess_output_type(in_guesses.ptr(),p_output);
  1969. }
  1970. void VisualScriptEditor::_port_action_menu(int p_option) {
  1971. Vector2 ofs = graph->get_scroll_ofs() + port_action_pos;
  1972. if (graph->is_using_snap()) {
  1973. int snap = graph->get_snap();
  1974. ofs = ofs.snapped(Vector2(snap,snap));
  1975. }
  1976. ofs/=EDSCALE;
  1977. bool seq_connect=false;
  1978. Ref<VisualScriptNode> vnode;
  1979. Set<int> vn;
  1980. switch(p_option) {
  1981. case CREATE_CALL: {
  1982. Ref<VisualScriptFunctionCall> n;
  1983. n.instance();
  1984. vnode=n;
  1985. VisualScriptNode::TypeGuess tg = _guess_output_type(port_action_node,port_action_output,vn);
  1986. if (tg.type==Variant::OBJECT) {
  1987. n->set_call_mode(VisualScriptFunctionCall::CALL_MODE_INSTANCE);
  1988. if (tg.GDCLASS!=StringName()) {
  1989. n->set_base_type(tg.GDCLASS);
  1990. } else {
  1991. n->set_base_type("Object");
  1992. }
  1993. if (tg.script.is_valid()) {
  1994. n->set_base_script(tg.script->get_path());
  1995. new_connect_node_select->select_method_from_script(tg.script);
  1996. } else {
  1997. new_connect_node_select->select_method_from_base_type(n->get_base_type());
  1998. }
  1999. } else {
  2000. n->set_call_mode(VisualScriptFunctionCall::CALL_MODE_BASIC_TYPE);
  2001. n->set_basic_type(tg.type);
  2002. new_connect_node_select->select_method_from_basic_type(tg.type);
  2003. }
  2004. } break;
  2005. case CREATE_SET: {
  2006. Ref<VisualScriptPropertySet> n;
  2007. n.instance();
  2008. vnode=n;
  2009. VisualScriptNode::TypeGuess tg = _guess_output_type(port_action_node,port_action_output,vn);
  2010. if (tg.type==Variant::OBJECT) {
  2011. n->set_call_mode(VisualScriptPropertySet::CALL_MODE_INSTANCE);
  2012. if (tg.GDCLASS!=StringName()) {
  2013. n->set_base_type(tg.GDCLASS);
  2014. } else {
  2015. n->set_base_type("Object");
  2016. }
  2017. if (tg.script.is_valid()) {
  2018. n->set_base_script(tg.script->get_path());
  2019. new_connect_node_select->select_property_from_script(tg.script);
  2020. } else {
  2021. new_connect_node_select->select_property_from_base_type(n->get_base_type());
  2022. }
  2023. } else {
  2024. n->set_call_mode(VisualScriptPropertySet::CALL_MODE_BASIC_TYPE);
  2025. n->set_basic_type(tg.type);
  2026. new_connect_node_select->select_property_from_basic_type(tg.type,tg.ev_type);
  2027. }
  2028. } break;
  2029. case CREATE_GET: {
  2030. Ref<VisualScriptPropertyGet> n;
  2031. n.instance();
  2032. vnode=n;
  2033. VisualScriptNode::TypeGuess tg = _guess_output_type(port_action_node,port_action_output,vn);
  2034. if (tg.type==Variant::OBJECT) {
  2035. n->set_call_mode(VisualScriptPropertyGet::CALL_MODE_INSTANCE);
  2036. if (tg.GDCLASS!=StringName()) {
  2037. n->set_base_type(tg.GDCLASS);
  2038. } else {
  2039. n->set_base_type("Object");
  2040. }
  2041. if (tg.script.is_valid()) {
  2042. n->set_base_script(tg.script->get_path());
  2043. new_connect_node_select->select_property_from_script(tg.script);
  2044. } else {
  2045. new_connect_node_select->select_property_from_base_type(n->get_base_type());
  2046. }
  2047. } else {
  2048. n->set_call_mode(VisualScriptPropertyGet::CALL_MODE_BASIC_TYPE);
  2049. n->set_basic_type(tg.type);
  2050. new_connect_node_select->select_property_from_basic_type(tg.type,tg.ev_type);
  2051. }
  2052. } break;
  2053. case CREATE_COND: {
  2054. Ref<VisualScriptCondition> n;
  2055. n.instance();
  2056. vnode=n;
  2057. seq_connect=true;
  2058. } break;
  2059. case CREATE_SEQUENCE: {
  2060. Ref<VisualScriptSequence> n;
  2061. n.instance();
  2062. vnode=n;
  2063. seq_connect=true;
  2064. } break;
  2065. case CREATE_SWITCH: {
  2066. Ref<VisualScriptSwitch> n;
  2067. n.instance();
  2068. vnode=n;
  2069. seq_connect=true;
  2070. } break;
  2071. case CREATE_ITERATOR: {
  2072. Ref<VisualScriptIterator> n;
  2073. n.instance();
  2074. vnode=n;
  2075. seq_connect=true;
  2076. } break;
  2077. case CREATE_WHILE: {
  2078. Ref<VisualScriptWhile> n;
  2079. n.instance();
  2080. vnode=n;
  2081. seq_connect=true;
  2082. } break;
  2083. case CREATE_RETURN: {
  2084. Ref<VisualScriptReturn> n;
  2085. n.instance();
  2086. vnode=n;
  2087. seq_connect=true;
  2088. } break;
  2089. }
  2090. int new_id = script->get_available_id();
  2091. undo_redo->create_action(TTR("Add Node"));
  2092. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,vnode,ofs);
  2093. if (seq_connect) {
  2094. undo_redo->add_do_method(script.ptr(),"sequence_connect",edited_func,port_action_node,port_action_output,new_id);
  2095. }
  2096. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  2097. undo_redo->add_do_method(this,"_update_graph",new_id);
  2098. undo_redo->add_undo_method(this,"_update_graph",new_id);
  2099. undo_redo->commit_action();
  2100. port_action_new_node=new_id;
  2101. }
  2102. void VisualScriptEditor::_selected_connect_node_method_or_setget(const String& p_text) {
  2103. Ref<VisualScriptNode> vsn = script->get_node(edited_func,port_action_new_node);
  2104. if (vsn->cast_to<VisualScriptFunctionCall>()) {
  2105. Ref<VisualScriptFunctionCall> vsfc = vsn;
  2106. vsfc->set_function(p_text);
  2107. script->data_connect(edited_func,port_action_node,port_action_output,port_action_new_node,0);
  2108. }
  2109. if (vsn->cast_to<VisualScriptPropertySet>()) {
  2110. Ref<VisualScriptPropertySet> vsp = vsn;
  2111. vsp->set_property(p_text);
  2112. script->data_connect(edited_func,port_action_node,port_action_output,port_action_new_node,0);
  2113. }
  2114. if (vsn->cast_to<VisualScriptPropertyGet>()) {
  2115. Ref<VisualScriptPropertyGet> vsp = vsn;
  2116. vsp->set_property(p_text);
  2117. script->data_connect(edited_func,port_action_node,port_action_output,port_action_new_node,0);
  2118. }
  2119. _update_graph(port_action_new_node);
  2120. _update_graph_connections();
  2121. }
  2122. void VisualScriptEditor::_default_value_changed() {
  2123. Ref<VisualScriptNode> vsn = script->get_node(edited_func,editing_id);
  2124. if (vsn.is_null())
  2125. return;
  2126. undo_redo->create_action("Change Input Value");
  2127. undo_redo->add_do_method(vsn.ptr(),"set_default_input_value",editing_input,default_value_edit->get_variant());
  2128. undo_redo->add_undo_method(vsn.ptr(),"set_default_input_value",editing_input,vsn->get_default_input_value(editing_input));
  2129. undo_redo->add_do_method(this,"_update_graph",editing_id);
  2130. undo_redo->add_undo_method(this,"_update_graph",editing_id);
  2131. undo_redo->commit_action();
  2132. }
  2133. void VisualScriptEditor::_default_value_edited(Node * p_button,int p_id,int p_input_port) {
  2134. Ref<VisualScriptNode> vsn = script->get_node(edited_func,p_id);
  2135. if (vsn.is_null())
  2136. return;
  2137. PropertyInfo pinfo = vsn->get_input_value_port_info(p_input_port);
  2138. Variant existing = vsn->get_default_input_value(p_input_port);
  2139. if (pinfo.type!=Variant::NIL && existing.get_type()!=pinfo.type) {
  2140. Variant::CallError ce;
  2141. const Variant *existingp=&existing;
  2142. existing = Variant::construct(pinfo.type,&existingp,1,ce,false);
  2143. }
  2144. default_value_edit->set_pos(p_button->cast_to<Control>()->get_global_pos()+Vector2(0,p_button->cast_to<Control>()->get_size().y));
  2145. default_value_edit->set_size(Size2(1,1));
  2146. if (default_value_edit->edit(NULL,pinfo.name,pinfo.type,existing,pinfo.hint,pinfo.hint_string)) {
  2147. if (pinfo.hint==PROPERTY_HINT_MULTILINE_TEXT)
  2148. default_value_edit->popup_centered_ratio();
  2149. else
  2150. default_value_edit->popup();
  2151. }
  2152. editing_id = p_id;
  2153. editing_input=p_input_port;
  2154. }
  2155. void VisualScriptEditor::_show_hint(const String& p_hint) {
  2156. hint_text->set_text(p_hint);
  2157. hint_text->show();
  2158. hint_text_timer->start();
  2159. }
  2160. void VisualScriptEditor::_hide_timer() {
  2161. hint_text->hide();
  2162. }
  2163. void VisualScriptEditor::_node_filter_changed(const String& p_text) {
  2164. _update_available_nodes();
  2165. }
  2166. void VisualScriptEditor::_notification(int p_what) {
  2167. if (p_what==NOTIFICATION_READY) {
  2168. node_filter_icon->set_texture(Control::get_icon("Zoom","EditorIcons"));
  2169. }
  2170. }
  2171. void VisualScriptEditor::_graph_ofs_changed(const Vector2& p_ofs) {
  2172. if (updating_graph)
  2173. return;
  2174. updating_graph=true;
  2175. if (script->has_function(edited_func)) {
  2176. script->set_function_scroll(edited_func,graph->get_scroll_ofs()/EDSCALE);
  2177. script->set_edited(true);
  2178. }
  2179. updating_graph=false;
  2180. }
  2181. void VisualScriptEditor::_comment_node_resized(const Vector2& p_new_size,int p_node) {
  2182. if (updating_graph)
  2183. return;
  2184. Ref<VisualScriptComment> vsc = script->get_node(edited_func,p_node);
  2185. if (vsc.is_null())
  2186. return;
  2187. Node *node = graph->get_node(itos(p_node));
  2188. if (!node)
  2189. return;
  2190. GraphNode *gn = node->cast_to<GraphNode>();
  2191. if (!gn)
  2192. return;
  2193. updating_graph=true;
  2194. graph->set_block_minimum_size_adjust(true); //faster resize
  2195. undo_redo->create_action("Resize Comment",UndoRedo::MERGE_ENDS);
  2196. undo_redo->add_do_method(vsc.ptr(),"set_size",p_new_size/EDSCALE);
  2197. undo_redo->add_undo_method(vsc.ptr(),"set_size",vsc->get_size());
  2198. undo_redo->commit_action();
  2199. gn->set_custom_minimum_size(p_new_size); //for this time since graph update is blocked
  2200. gn->set_size(Size2(1,1));
  2201. graph->set_block_minimum_size_adjust(false);
  2202. updating_graph=false;
  2203. }
  2204. void VisualScriptEditor::_menu_option(int p_what) {
  2205. switch(p_what) {
  2206. case EDIT_DELETE_NODES: {
  2207. _on_nodes_delete();
  2208. } break;
  2209. case EDIT_TOGGLE_BREAKPOINT: {
  2210. List<String> reselect;
  2211. for(int i=0;i<graph->get_child_count();i++) {
  2212. GraphNode *gn = graph->get_child(i)->cast_to<GraphNode>();
  2213. if (gn) {
  2214. if (gn->is_selected()) {
  2215. int id = String(gn->get_name()).to_int();
  2216. Ref<VisualScriptNode> vsn = script->get_node(edited_func,id);
  2217. if (vsn.is_valid()) {
  2218. vsn->set_breakpoint(!vsn->is_breakpoint());
  2219. reselect.push_back(gn->get_name());
  2220. }
  2221. }
  2222. }
  2223. }
  2224. _update_graph();
  2225. for(List<String>::Element *E=reselect.front();E;E=E->next()) {
  2226. GraphNode *gn = graph->get_node(E->get())->cast_to<GraphNode>();
  2227. gn->set_selected(true);
  2228. }
  2229. } break;
  2230. case EDIT_FIND_NODE_TYPE: {
  2231. //popup disappearing grabs focus to owner, so use call deferred
  2232. node_filter->call_deferred("grab_focus");
  2233. node_filter->call_deferred("select_all");
  2234. } break;
  2235. case EDIT_COPY_NODES:
  2236. case EDIT_CUT_NODES: {
  2237. if (!script->has_function(edited_func))
  2238. break;
  2239. clipboard->nodes.clear();
  2240. clipboard->data_connections.clear();
  2241. clipboard->sequence_connections.clear();
  2242. for(int i=0;i<graph->get_child_count();i++) {
  2243. GraphNode *gn = graph->get_child(i)->cast_to<GraphNode>();
  2244. if (gn) {
  2245. if (gn->is_selected()) {
  2246. int id = String(gn->get_name()).to_int();
  2247. Ref<VisualScriptNode> node = script->get_node(edited_func,id);
  2248. if (node->cast_to<VisualScriptFunction>()) {
  2249. EditorNode::get_singleton()->show_warning("Can't copy the function node.");
  2250. return;
  2251. }
  2252. if (node.is_valid()) {
  2253. clipboard->nodes[id]=node->duplicate();
  2254. clipboard->nodes_positions[id]=script->get_node_pos(edited_func,id);
  2255. }
  2256. }
  2257. }
  2258. }
  2259. if (clipboard->nodes.empty())
  2260. break;
  2261. List<VisualScript::SequenceConnection> sequence_connections;
  2262. script->get_sequence_connection_list(edited_func,&sequence_connections);
  2263. for (List<VisualScript::SequenceConnection>::Element *E=sequence_connections.front();E;E=E->next()) {
  2264. if (clipboard->nodes.has(E->get().from_node) && clipboard->nodes.has(E->get().to_node)) {
  2265. clipboard->sequence_connections.insert(E->get());
  2266. }
  2267. }
  2268. List<VisualScript::DataConnection> data_connections;
  2269. script->get_data_connection_list(edited_func,&data_connections);
  2270. for (List<VisualScript::DataConnection>::Element *E=data_connections.front();E;E=E->next()) {
  2271. if (clipboard->nodes.has(E->get().from_node) && clipboard->nodes.has(E->get().to_node)) {
  2272. clipboard->data_connections.insert(E->get());
  2273. }
  2274. }
  2275. if (p_what==EDIT_CUT_NODES) {
  2276. _on_nodes_delete(); // oh yeah, also delete on cut
  2277. }
  2278. } break;
  2279. case EDIT_PASTE_NODES: {
  2280. if (!script->has_function(edited_func))
  2281. break;
  2282. if (clipboard->nodes.empty()) {
  2283. EditorNode::get_singleton()->show_warning("Clipboard is empty!");
  2284. break;
  2285. }
  2286. Map<int,int> remap;
  2287. undo_redo->create_action("Paste VisualScript Nodes");
  2288. int idc=script->get_available_id()+1;
  2289. Set<int> to_select;
  2290. Set<Vector2> existing_positions;
  2291. {
  2292. List<int> nodes;
  2293. script->get_node_list(edited_func,&nodes);
  2294. for (List<int>::Element *E=nodes.front();E;E=E->next()) {
  2295. Vector2 pos = script->get_node_pos(edited_func,E->get()).snapped(Vector2(2,2));
  2296. existing_positions.insert(pos);
  2297. }
  2298. }
  2299. for (Map<int,Ref<VisualScriptNode> >::Element *E=clipboard->nodes.front();E;E=E->next()) {
  2300. Ref<VisualScriptNode> node = E->get()->duplicate();
  2301. int new_id = idc++;
  2302. to_select.insert(new_id);
  2303. remap[E->key()]=new_id;
  2304. Vector2 paste_pos = clipboard->nodes_positions[E->key()];
  2305. while(existing_positions.has(paste_pos.snapped(Vector2(2,2)))) {
  2306. paste_pos+=Vector2(20,20)*EDSCALE;
  2307. }
  2308. undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,node,paste_pos);
  2309. undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
  2310. }
  2311. for (Set<VisualScript::SequenceConnection>::Element *E=clipboard->sequence_connections.front();E;E=E->next()) {
  2312. undo_redo->add_do_method(script.ptr(),"sequence_connect",edited_func,remap[E->get().from_node],E->get().from_output,remap[E->get().to_node]);
  2313. undo_redo->add_undo_method(script.ptr(),"sequence_disconnect",edited_func,remap[E->get().from_node],E->get().from_output,remap[E->get().to_node]);
  2314. }
  2315. for (Set<VisualScript::DataConnection>::Element *E=clipboard->data_connections.front();E;E=E->next()) {
  2316. undo_redo->add_do_method(script.ptr(),"data_connect",edited_func,remap[E->get().from_node],E->get().from_port,remap[E->get().to_node],E->get().to_port);
  2317. undo_redo->add_undo_method(script.ptr(),"data_disconnect",edited_func,remap[E->get().from_node],E->get().from_port,remap[E->get().to_node],E->get().to_port);
  2318. }
  2319. undo_redo->add_do_method(this,"_update_graph");
  2320. undo_redo->add_undo_method(this,"_update_graph");
  2321. undo_redo->commit_action();
  2322. for(int i=0;i<graph->get_child_count();i++) {
  2323. GraphNode *gn = graph->get_child(i)->cast_to<GraphNode>();
  2324. if (gn) {
  2325. int id = gn->get_name().operator String().to_int();
  2326. gn->set_selected(to_select.has(id));
  2327. }
  2328. }
  2329. } break;
  2330. }
  2331. }
  2332. void VisualScriptEditor::_bind_methods() {
  2333. ClassDB::bind_method("_member_button",&VisualScriptEditor::_member_button);
  2334. ClassDB::bind_method("_member_edited",&VisualScriptEditor::_member_edited);
  2335. ClassDB::bind_method("_member_selected",&VisualScriptEditor::_member_selected);
  2336. ClassDB::bind_method("_update_members",&VisualScriptEditor::_update_members);
  2337. ClassDB::bind_method("_change_base_type",&VisualScriptEditor::_change_base_type);
  2338. ClassDB::bind_method("_change_base_type_callback",&VisualScriptEditor::_change_base_type_callback);
  2339. ClassDB::bind_method("_override_pressed",&VisualScriptEditor::_override_pressed);
  2340. ClassDB::bind_method("_node_selected",&VisualScriptEditor::_node_selected);
  2341. ClassDB::bind_method("_node_moved",&VisualScriptEditor::_node_moved);
  2342. ClassDB::bind_method("_move_node",&VisualScriptEditor::_move_node);
  2343. ClassDB::bind_method("_begin_node_move",&VisualScriptEditor::_begin_node_move);
  2344. ClassDB::bind_method("_end_node_move",&VisualScriptEditor::_end_node_move);
  2345. ClassDB::bind_method("_remove_node",&VisualScriptEditor::_remove_node);
  2346. ClassDB::bind_method("_update_graph",&VisualScriptEditor::_update_graph,DEFVAL(-1));
  2347. ClassDB::bind_method("_node_ports_changed",&VisualScriptEditor::_node_ports_changed);
  2348. ClassDB::bind_method("_available_node_doubleclicked",&VisualScriptEditor::_available_node_doubleclicked);
  2349. ClassDB::bind_method("_default_value_edited",&VisualScriptEditor::_default_value_edited);
  2350. ClassDB::bind_method("_default_value_changed",&VisualScriptEditor::_default_value_changed);
  2351. ClassDB::bind_method("_menu_option",&VisualScriptEditor::_menu_option);
  2352. ClassDB::bind_method("_graph_ofs_changed",&VisualScriptEditor::_graph_ofs_changed);
  2353. ClassDB::bind_method("_center_on_node",&VisualScriptEditor::_center_on_node);
  2354. ClassDB::bind_method("_comment_node_resized",&VisualScriptEditor::_comment_node_resized);
  2355. ClassDB::bind_method("_button_resource_previewed",&VisualScriptEditor::_button_resource_previewed);
  2356. ClassDB::bind_method("_port_action_menu",&VisualScriptEditor::_port_action_menu);
  2357. ClassDB::bind_method("_selected_connect_node_method_or_setget",&VisualScriptEditor::_selected_connect_node_method_or_setget);
  2358. ClassDB::bind_method("_expression_text_changed",&VisualScriptEditor::_expression_text_changed);
  2359. ClassDB::bind_method("get_drag_data_fw",&VisualScriptEditor::get_drag_data_fw);
  2360. ClassDB::bind_method("can_drop_data_fw",&VisualScriptEditor::can_drop_data_fw);
  2361. ClassDB::bind_method("drop_data_fw",&VisualScriptEditor::drop_data_fw);
  2362. ClassDB::bind_method("_input",&VisualScriptEditor::_input);
  2363. ClassDB::bind_method("_on_nodes_delete",&VisualScriptEditor::_on_nodes_delete);
  2364. ClassDB::bind_method("_on_nodes_duplicate",&VisualScriptEditor::_on_nodes_duplicate);
  2365. ClassDB::bind_method("_hide_timer",&VisualScriptEditor::_hide_timer);
  2366. ClassDB::bind_method("_graph_connected",&VisualScriptEditor::_graph_connected);
  2367. ClassDB::bind_method("_graph_disconnected",&VisualScriptEditor::_graph_disconnected);
  2368. ClassDB::bind_method("_graph_connect_to_empty",&VisualScriptEditor::_graph_connect_to_empty);
  2369. ClassDB::bind_method("_update_graph_connections",&VisualScriptEditor::_update_graph_connections);
  2370. ClassDB::bind_method("_node_filter_changed",&VisualScriptEditor::_node_filter_changed);
  2371. ClassDB::bind_method("_selected_method",&VisualScriptEditor::_selected_method);
  2372. ClassDB::bind_method("_draw_color_over_button",&VisualScriptEditor::_draw_color_over_button);
  2373. }
  2374. VisualScriptEditor::VisualScriptEditor() {
  2375. if (!clipboard) {
  2376. clipboard = memnew( Clipboard );
  2377. }
  2378. updating_graph=false;
  2379. edit_menu = memnew( MenuButton );
  2380. edit_menu->set_text(TTR("Edit"));
  2381. edit_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("visual_script_editor/delete_selected"), EDIT_DELETE_NODES);
  2382. edit_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("visual_script_editor/toggle_breakpoint"), EDIT_TOGGLE_BREAKPOINT);
  2383. edit_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("visual_script_editor/find_node_type"), EDIT_FIND_NODE_TYPE);
  2384. edit_menu->get_popup()->add_separator();
  2385. edit_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("visual_script_editor/copy_nodes"), EDIT_COPY_NODES);
  2386. edit_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("visual_script_editor/cut_nodes"), EDIT_CUT_NODES);
  2387. edit_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("visual_script_editor/paste_nodes"), EDIT_PASTE_NODES);
  2388. edit_menu->get_popup()->connect("id_pressed",this,"_menu_option");
  2389. main_hsplit = memnew( HSplitContainer );
  2390. add_child(main_hsplit);
  2391. main_hsplit->set_area_as_parent_rect();
  2392. left_vsplit = memnew( VSplitContainer );
  2393. main_hsplit->add_child(left_vsplit);
  2394. VBoxContainer *left_vb = memnew( VBoxContainer );
  2395. left_vsplit->add_child(left_vb);
  2396. left_vb->set_v_size_flags(SIZE_EXPAND_FILL);
  2397. left_vb->set_custom_minimum_size(Size2(230,1)*EDSCALE);
  2398. base_type_select = memnew( Button );
  2399. left_vb->add_margin_child(TTR("Base Type:"),base_type_select);
  2400. base_type_select->connect("pressed",this,"_change_base_type");
  2401. members = memnew( Tree );
  2402. left_vb->add_margin_child(TTR("Members:"),members,true);
  2403. members->set_hide_root(true);
  2404. members->connect("button_pressed",this,"_member_button");
  2405. members->connect("item_edited",this,"_member_edited");
  2406. members->connect("cell_selected",this,"_member_selected",varray(),CONNECT_DEFERRED);
  2407. members->set_single_select_cell_editing_only_when_already_selected(true);
  2408. members->set_hide_folding(true);
  2409. members->set_drag_forwarding(this);
  2410. VBoxContainer *left_vb2 = memnew( VBoxContainer );
  2411. left_vsplit->add_child(left_vb2);
  2412. left_vb2->set_v_size_flags(SIZE_EXPAND_FILL);
  2413. VBoxContainer *vbc_nodes = memnew( VBoxContainer );
  2414. HBoxContainer *hbc_nodes = memnew( HBoxContainer );
  2415. node_filter = memnew (LineEdit);
  2416. node_filter->connect("text_changed",this,"_node_filter_changed");
  2417. hbc_nodes->add_child(node_filter);
  2418. node_filter->set_h_size_flags(SIZE_EXPAND_FILL);
  2419. node_filter_icon = memnew( TextureRect );
  2420. node_filter_icon->set_stretch_mode(TextureRect::STRETCH_KEEP_CENTERED);
  2421. hbc_nodes->add_child(node_filter_icon);
  2422. vbc_nodes->add_child(hbc_nodes);
  2423. nodes = memnew( Tree );
  2424. vbc_nodes->add_child(nodes);
  2425. nodes->set_v_size_flags(SIZE_EXPAND_FILL);
  2426. left_vb2->add_margin_child(TTR("Available Nodes:"),vbc_nodes,true);
  2427. nodes->set_hide_root(true);
  2428. nodes->connect("item_activated",this,"_available_node_doubleclicked");
  2429. nodes->set_drag_forwarding(this);
  2430. graph = memnew( GraphEdit );
  2431. main_hsplit->add_child(graph);
  2432. graph->set_h_size_flags(SIZE_EXPAND_FILL);
  2433. graph->connect("node_selected",this,"_node_selected");
  2434. graph->connect("_begin_node_move",this,"_begin_node_move");
  2435. graph->connect("_end_node_move",this,"_end_node_move");
  2436. graph->connect("delete_nodes_request",this,"_on_nodes_delete");
  2437. graph->connect("duplicate_nodes_request",this,"_on_nodes_duplicate");
  2438. graph->set_drag_forwarding(this);
  2439. graph->hide();
  2440. graph->connect("scroll_offset_changed",this,"_graph_ofs_changed");
  2441. select_func_text = memnew( Label );
  2442. select_func_text->set_text(TTR("Select or create a function to edit graph"));
  2443. select_func_text->set_align(Label::ALIGN_CENTER);
  2444. select_func_text->set_valign(Label::VALIGN_CENTER);
  2445. select_func_text->set_h_size_flags(SIZE_EXPAND_FILL);
  2446. main_hsplit->add_child(select_func_text);
  2447. hint_text = memnew( Label );
  2448. hint_text->set_anchor_and_margin(MARGIN_TOP,ANCHOR_END,100);
  2449. hint_text->set_anchor_and_margin(MARGIN_BOTTOM,ANCHOR_END,0);
  2450. hint_text->set_anchor_and_margin(MARGIN_RIGHT,ANCHOR_END,0);
  2451. hint_text->set_align(Label::ALIGN_CENTER);
  2452. hint_text->set_valign(Label::VALIGN_CENTER);
  2453. graph->add_child(hint_text);
  2454. hint_text_timer = memnew( Timer );
  2455. hint_text_timer->set_wait_time(4);
  2456. hint_text_timer->connect("timeout",this,"_hide_timer");
  2457. add_child(hint_text_timer);
  2458. //allowed casts (connections)
  2459. for(int i=0;i<Variant::VARIANT_MAX;i++) {
  2460. graph->add_valid_connection_type(Variant::NIL,i);
  2461. graph->add_valid_connection_type(i,Variant::NIL);
  2462. for(int j=0;j<Variant::VARIANT_MAX;j++) {
  2463. if (Variant::can_convert(Variant::Type(i),Variant::Type(j))) {
  2464. graph->add_valid_connection_type(i,j);
  2465. }
  2466. }
  2467. graph->add_valid_right_disconnect_type(i);
  2468. }
  2469. graph->add_valid_left_disconnect_type(TYPE_SEQUENCE);
  2470. graph->connect("connection_request",this,"_graph_connected");
  2471. graph->connect("disconnection_request",this,"_graph_disconnected");
  2472. graph->connect("connection_to_empty",this,"_graph_connect_to_empty");
  2473. edit_signal_dialog = memnew( AcceptDialog );
  2474. edit_signal_dialog->get_ok()->set_text(TTR("Close"));
  2475. add_child(edit_signal_dialog);
  2476. edit_signal_dialog->set_title(TTR("Edit Signal Arguments:"));
  2477. signal_editor = memnew( VisualScriptEditorSignalEdit );
  2478. edit_signal_edit = memnew( PropertyEditor );
  2479. edit_signal_edit->hide_top_label();
  2480. edit_signal_dialog->add_child(edit_signal_edit);
  2481. edit_signal_edit->edit(signal_editor);
  2482. edit_variable_dialog = memnew( AcceptDialog );
  2483. edit_variable_dialog->get_ok()->set_text(TTR("Close"));
  2484. add_child(edit_variable_dialog);
  2485. edit_variable_dialog->set_title(TTR("Edit Variable:"));
  2486. variable_editor = memnew( VisualScriptEditorVariableEdit );
  2487. edit_variable_edit = memnew( PropertyEditor );
  2488. edit_variable_edit->hide_top_label();
  2489. edit_variable_dialog->add_child(edit_variable_edit);
  2490. edit_variable_edit->edit(variable_editor);
  2491. select_base_type=memnew(CreateDialog);
  2492. select_base_type->set_base_type("Object"); //anything goes
  2493. select_base_type->connect("create",this,"_change_base_type_callback");
  2494. select_base_type->get_ok()->set_text(TTR("Change"));
  2495. add_child(select_base_type);
  2496. undo_redo = EditorNode::get_singleton()->get_undo_redo();
  2497. new_function_menu = memnew( PopupMenu );
  2498. new_function_menu->connect("id_pressed",this,"_override_pressed");
  2499. add_child(new_function_menu);
  2500. updating_members=false;
  2501. set_process_input(true); //for revert on drag
  2502. set_process_unhandled_input(true); //for revert on drag
  2503. default_value_edit= memnew( CustomPropertyEditor);
  2504. add_child(default_value_edit);
  2505. default_value_edit->connect("variant_changed",this,"_default_value_changed");
  2506. method_select = memnew( PropertySelector );
  2507. add_child(method_select);
  2508. method_select->connect("selected",this,"_selected_method");
  2509. error_line=-1;
  2510. new_connect_node_select = memnew( PropertySelector );
  2511. add_child(new_connect_node_select);
  2512. new_connect_node_select->connect("selected",this,"_selected_connect_node_method_or_setget");
  2513. port_action_popup = memnew( PopupMenu );
  2514. add_child(port_action_popup);
  2515. port_action_popup->connect("id_pressed",this,"_port_action_menu");
  2516. }
  2517. VisualScriptEditor::~VisualScriptEditor() {
  2518. undo_redo->clear_history(); //avoid crashes
  2519. memdelete(signal_editor);
  2520. memdelete(variable_editor);
  2521. }
  2522. static ScriptEditorBase * create_editor(const Ref<Script>& p_script) {
  2523. if (p_script->cast_to<VisualScript>()) {
  2524. return memnew( VisualScriptEditor );
  2525. }
  2526. return NULL;
  2527. }
  2528. VisualScriptEditor::Clipboard *VisualScriptEditor::clipboard=NULL;
  2529. void VisualScriptEditor::free_clipboard() {
  2530. if (clipboard)
  2531. memdelete(clipboard);
  2532. }
  2533. static void register_editor_callback() {
  2534. ScriptEditor::register_create_script_editor_function(create_editor);
  2535. EditorSettings::get_singleton()->set("editors/visual_script/color_functions",Color(1,0.9,0.9));
  2536. EditorSettings::get_singleton()->set("editors/visual_script/color_data",Color(0.9,1.0,0.9));
  2537. EditorSettings::get_singleton()->set("editors/visual_script/color_operators",Color(0.9,0.9,1.0));
  2538. EditorSettings::get_singleton()->set("editors/visual_script/color_flow_control",Color(1.0,1.0,0.8));
  2539. EditorSettings::get_singleton()->set("editors/visual_script/color_custom",Color(0.8,1.0,1.0));
  2540. EditorSettings::get_singleton()->set("editors/visual_script/color_constants",Color(1.0,0.8,1.0));
  2541. ED_SHORTCUT("visual_script_editor/delete_selected", TTR("Delete Selected"));
  2542. ED_SHORTCUT("visual_script_editor/toggle_breakpoint", TTR("Toggle Breakpoint"), KEY_F9);
  2543. ED_SHORTCUT("visual_script_editor/find_node_type", TTR("Find Node Type"), KEY_MASK_CMD+KEY_F);
  2544. ED_SHORTCUT("visual_script_editor/copy_nodes", TTR("Copy Nodes"), KEY_MASK_CMD+KEY_C);
  2545. ED_SHORTCUT("visual_script_editor/cut_nodes", TTR("Cut Nodes"), KEY_MASK_CMD+KEY_X);
  2546. ED_SHORTCUT("visual_script_editor/paste_nodes", TTR("Paste Nodes"), KEY_MASK_CMD+KEY_V);
  2547. }
  2548. void VisualScriptEditor::register_editor() {
  2549. //too early to register stuff here, request a callback
  2550. EditorNode::add_plugin_init_callback(register_editor_callback);
  2551. }
  2552. #endif