visual_script_nodes.cpp 67 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656
  1. #include "visual_script_nodes.h"
  2. #include "global_constants.h"
  3. #include "globals.h"
  4. #include "scene/main/scene_main_loop.h"
  5. #include "os/os.h"
  6. #include "scene/main/node.h"
  7. //////////////////////////////////////////
  8. ////////////////FUNCTION//////////////////
  9. //////////////////////////////////////////
  10. bool VisualScriptFunction::_set(const StringName& p_name, const Variant& p_value) {
  11. if (p_name=="argument_count") {
  12. int new_argc=p_value;
  13. int argc = arguments.size();
  14. if (argc==new_argc)
  15. return true;
  16. arguments.resize(new_argc);
  17. for(int i=argc;i<new_argc;i++) {
  18. arguments[i].name="arg"+itos(i+1);
  19. arguments[i].type=Variant::NIL;
  20. }
  21. ports_changed_notify();
  22. _change_notify();
  23. return true;
  24. }
  25. if (String(p_name).begins_with("argument/")) {
  26. int idx = String(p_name).get_slice("/",1).to_int()-1;
  27. ERR_FAIL_INDEX_V(idx,arguments.size(),false);
  28. String what = String(p_name).get_slice("/",2);
  29. if (what=="type") {
  30. Variant::Type new_type = Variant::Type(int(p_value));
  31. arguments[idx].type=new_type;
  32. ports_changed_notify();
  33. return true;
  34. }
  35. if (what=="name") {
  36. arguments[idx].name=p_value;
  37. ports_changed_notify();
  38. return true;
  39. }
  40. }
  41. if (p_name=="stack/stackless") {
  42. set_stack_less(p_value);
  43. return true;
  44. }
  45. if (p_name=="stack/size") {
  46. stack_size=p_value;
  47. return true;
  48. }
  49. if (p_name=="rpc/mode") {
  50. rpc_mode=ScriptInstance::RPCMode(int(p_value));
  51. return true;
  52. }
  53. return false;
  54. }
  55. bool VisualScriptFunction::_get(const StringName& p_name,Variant &r_ret) const {
  56. if (p_name=="argument_count") {
  57. r_ret = arguments.size();
  58. return true;
  59. }
  60. if (String(p_name).begins_with("argument/")) {
  61. int idx = String(p_name).get_slice("/",1).to_int()-1;
  62. ERR_FAIL_INDEX_V(idx,arguments.size(),false);
  63. String what = String(p_name).get_slice("/",2);
  64. if (what=="type") {
  65. r_ret = arguments[idx].type;
  66. return true;
  67. }
  68. if (what=="name") {
  69. r_ret = arguments[idx].name;
  70. return true;
  71. }
  72. }
  73. if (p_name=="stack/stackless") {
  74. r_ret=stack_less;
  75. return true;
  76. }
  77. if (p_name=="stack/size") {
  78. r_ret=stack_size;
  79. return true;
  80. }
  81. if (p_name=="rpc/mode") {
  82. r_ret=rpc_mode;
  83. return true;
  84. }
  85. return false;
  86. }
  87. void VisualScriptFunction::_get_property_list( List<PropertyInfo> *p_list) const {
  88. p_list->push_back(PropertyInfo(Variant::INT,"argument_count",PROPERTY_HINT_RANGE,"0,256"));
  89. String argt="Variant";
  90. for(int i=1;i<Variant::VARIANT_MAX;i++) {
  91. argt+=","+Variant::get_type_name(Variant::Type(i));
  92. }
  93. for(int i=0;i<arguments.size();i++) {
  94. p_list->push_back(PropertyInfo(Variant::INT,"argument/"+itos(i+1)+"/type",PROPERTY_HINT_ENUM,argt));
  95. p_list->push_back(PropertyInfo(Variant::STRING,"argument/"+itos(i+1)+"/name"));
  96. }
  97. if (!stack_less) {
  98. p_list->push_back(PropertyInfo(Variant::INT,"stack/size",PROPERTY_HINT_RANGE,"1,100000"));
  99. }
  100. p_list->push_back(PropertyInfo(Variant::BOOL,"stack/stackless"));
  101. p_list->push_back(PropertyInfo(Variant::INT,"rpc/mode",PROPERTY_HINT_ENUM,"Disabled,Remote,Sync,Master,Slave"));
  102. }
  103. int VisualScriptFunction::get_output_sequence_port_count() const {
  104. return 1;
  105. }
  106. bool VisualScriptFunction::has_input_sequence_port() const{
  107. return false;
  108. }
  109. int VisualScriptFunction::get_input_value_port_count() const{
  110. return 0;
  111. }
  112. int VisualScriptFunction::get_output_value_port_count() const{
  113. return arguments.size();
  114. }
  115. String VisualScriptFunction::get_output_sequence_port_text(int p_port) const {
  116. return String();
  117. }
  118. PropertyInfo VisualScriptFunction::get_input_value_port_info(int p_idx) const{
  119. ERR_FAIL_V(PropertyInfo());
  120. return PropertyInfo();
  121. }
  122. PropertyInfo VisualScriptFunction::get_output_value_port_info(int p_idx) const{
  123. ERR_FAIL_INDEX_V(p_idx,arguments.size(),PropertyInfo());
  124. PropertyInfo out;
  125. out.type=arguments[p_idx].type;
  126. out.name=arguments[p_idx].name;
  127. return out;
  128. }
  129. String VisualScriptFunction::get_caption() const {
  130. return "Function";
  131. }
  132. String VisualScriptFunction::get_text() const {
  133. return get_name(); //use name as function name I guess
  134. }
  135. void VisualScriptFunction::add_argument(Variant::Type p_type,const String& p_name,int p_index){
  136. Argument arg;
  137. arg.name=p_name;
  138. arg.type=p_type;
  139. if (p_index>=0)
  140. arguments.insert(p_index,arg);
  141. else
  142. arguments.push_back(arg);
  143. ports_changed_notify();
  144. }
  145. void VisualScriptFunction::set_argument_type(int p_argidx,Variant::Type p_type){
  146. ERR_FAIL_INDEX(p_argidx,arguments.size());
  147. arguments[p_argidx].type=p_type;
  148. ports_changed_notify();
  149. }
  150. Variant::Type VisualScriptFunction::get_argument_type(int p_argidx) const {
  151. ERR_FAIL_INDEX_V(p_argidx,arguments.size(),Variant::NIL);
  152. return arguments[p_argidx].type;
  153. }
  154. void VisualScriptFunction::set_argument_name(int p_argidx,const String& p_name) {
  155. ERR_FAIL_INDEX(p_argidx,arguments.size());
  156. arguments[p_argidx].name=p_name;
  157. ports_changed_notify();
  158. }
  159. String VisualScriptFunction::get_argument_name(int p_argidx) const {
  160. ERR_FAIL_INDEX_V(p_argidx,arguments.size(),String());
  161. return arguments[p_argidx].name;
  162. }
  163. void VisualScriptFunction::remove_argument(int p_argidx) {
  164. ERR_FAIL_INDEX(p_argidx,arguments.size());
  165. arguments.remove(p_argidx);
  166. ports_changed_notify();
  167. }
  168. int VisualScriptFunction::get_argument_count() const {
  169. return arguments.size();
  170. }
  171. void VisualScriptFunction::set_rpc_mode(ScriptInstance::RPCMode p_mode) {
  172. rpc_mode=p_mode;
  173. }
  174. ScriptInstance::RPCMode VisualScriptFunction::get_rpc_mode() const {
  175. return rpc_mode;
  176. }
  177. class VisualScriptNodeInstanceFunction : public VisualScriptNodeInstance {
  178. public:
  179. VisualScriptFunction *node;
  180. VisualScriptInstance *instance;
  181. //virtual int get_working_memory_size() const { return 0; }
  182. //virtual bool is_output_port_unsequenced(int p_idx) const { return false; }
  183. //virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return true; }
  184. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  185. int ac = node->get_argument_count();
  186. for(int i=0;i<ac;i++) {
  187. #ifdef DEBUG_ENABLED
  188. Variant::Type expected = node->get_argument_type(i);
  189. if (expected!=Variant::NIL) {
  190. if (!Variant::can_convert_strict(p_inputs[i]->get_type(),expected)) {
  191. r_error.error=Variant::CallError::CALL_ERROR_INVALID_ARGUMENT;
  192. r_error.expected=expected;
  193. r_error.argument=i;
  194. return 0;
  195. }
  196. }
  197. #endif
  198. *p_outputs[i]=*p_inputs[i];
  199. }
  200. return 0;
  201. }
  202. };
  203. VisualScriptNodeInstance* VisualScriptFunction::instance(VisualScriptInstance* p_instance) {
  204. VisualScriptNodeInstanceFunction * instance = memnew(VisualScriptNodeInstanceFunction );
  205. instance->node=this;
  206. instance->instance=p_instance;
  207. return instance;
  208. }
  209. VisualScriptFunction::VisualScriptFunction() {
  210. stack_size=256;
  211. stack_less=false;
  212. rpc_mode=ScriptInstance::RPC_MODE_DISABLED;
  213. }
  214. void VisualScriptFunction::set_stack_less(bool p_enable) {
  215. stack_less=p_enable;
  216. _change_notify();
  217. }
  218. bool VisualScriptFunction::is_stack_less() const {
  219. return stack_less;
  220. }
  221. void VisualScriptFunction::set_stack_size(int p_size) {
  222. ERR_FAIL_COND(p_size <1 || p_size>100000);
  223. stack_size=p_size;
  224. }
  225. int VisualScriptFunction::get_stack_size() const {
  226. return stack_size;
  227. }
  228. //////////////////////////////////////////
  229. ////////////////OPERATOR//////////////////
  230. //////////////////////////////////////////
  231. int VisualScriptOperator::get_output_sequence_port_count() const {
  232. return 1;
  233. }
  234. bool VisualScriptOperator::has_input_sequence_port() const{
  235. return true;
  236. }
  237. int VisualScriptOperator::get_input_value_port_count() const{
  238. return (op==Variant::OP_BIT_NEGATE || op==Variant::OP_NOT || op==Variant::OP_NEGATE) ? 1 : 2;
  239. }
  240. int VisualScriptOperator::get_output_value_port_count() const{
  241. return 1;
  242. }
  243. String VisualScriptOperator::get_output_sequence_port_text(int p_port) const {
  244. return String();
  245. }
  246. PropertyInfo VisualScriptOperator::get_input_value_port_info(int p_idx) const{
  247. static const Variant::Type port_types[Variant::OP_MAX][2]={
  248. {Variant::NIL,Variant::NIL}, //OP_EQUAL,
  249. {Variant::NIL,Variant::NIL}, //OP_NOT_EQUAL,
  250. {Variant::NIL,Variant::NIL}, //OP_LESS,
  251. {Variant::NIL,Variant::NIL}, //OP_LESS_EQUAL,
  252. {Variant::NIL,Variant::NIL}, //OP_GREATER,
  253. {Variant::NIL,Variant::NIL}, //OP_GREATER_EQUAL,
  254. //mathematic
  255. {Variant::NIL,Variant::NIL}, //OP_ADD,
  256. {Variant::NIL,Variant::NIL}, //OP_SUBSTRACT,
  257. {Variant::NIL,Variant::NIL}, //OP_MULTIPLY,
  258. {Variant::NIL,Variant::NIL}, //OP_DIVIDE,
  259. {Variant::NIL,Variant::NIL}, //OP_NEGATE,
  260. {Variant::INT,Variant::INT}, //OP_MODULE,
  261. {Variant::STRING,Variant::STRING}, //OP_STRING_CONCAT,
  262. //bitwise
  263. {Variant::INT,Variant::INT}, //OP_SHIFT_LEFT,
  264. {Variant::INT,Variant::INT}, //OP_SHIFT_RIGHT,
  265. {Variant::INT,Variant::INT}, //OP_BIT_AND,
  266. {Variant::INT,Variant::INT}, //OP_BIT_OR,
  267. {Variant::INT,Variant::INT}, //OP_BIT_XOR,
  268. {Variant::INT,Variant::INT}, //OP_BIT_NEGATE,
  269. //logic
  270. {Variant::BOOL,Variant::BOOL}, //OP_AND,
  271. {Variant::BOOL,Variant::BOOL}, //OP_OR,
  272. {Variant::BOOL,Variant::BOOL}, //OP_XOR,
  273. {Variant::BOOL,Variant::BOOL}, //OP_NOT,
  274. //containment
  275. {Variant::NIL,Variant::NIL} //OP_IN,
  276. };
  277. ERR_FAIL_INDEX_V(p_idx,Variant::OP_MAX,PropertyInfo());
  278. PropertyInfo pinfo;
  279. pinfo.name=p_idx==0?"A":"B";
  280. pinfo.type=port_types[op][p_idx];
  281. return pinfo;
  282. }
  283. PropertyInfo VisualScriptOperator::get_output_value_port_info(int p_idx) const{
  284. static const Variant::Type port_types[Variant::OP_MAX]={
  285. //comparation
  286. Variant::BOOL, //OP_EQUAL,
  287. Variant::BOOL, //OP_NOT_EQUAL,
  288. Variant::BOOL, //OP_LESS,
  289. Variant::BOOL, //OP_LESS_EQUAL,
  290. Variant::BOOL, //OP_GREATER,
  291. Variant::BOOL, //OP_GREATER_EQUAL,
  292. //mathematic
  293. Variant::NIL, //OP_ADD,
  294. Variant::NIL, //OP_SUBSTRACT,
  295. Variant::NIL, //OP_MULTIPLY,
  296. Variant::NIL, //OP_DIVIDE,
  297. Variant::NIL, //OP_NEGATE,
  298. Variant::INT, //OP_MODULE,
  299. Variant::STRING, //OP_STRING_CONCAT,
  300. //bitwise
  301. Variant::INT, //OP_SHIFT_LEFT,
  302. Variant::INT, //OP_SHIFT_RIGHT,
  303. Variant::INT, //OP_BIT_AND,
  304. Variant::INT, //OP_BIT_OR,
  305. Variant::INT, //OP_BIT_XOR,
  306. Variant::INT, //OP_BIT_NEGATE,
  307. //logic
  308. Variant::BOOL, //OP_AND,
  309. Variant::BOOL, //OP_OR,
  310. Variant::BOOL, //OP_XOR,
  311. Variant::BOOL, //OP_NOT,
  312. //containment
  313. Variant::BOOL //OP_IN,
  314. };
  315. PropertyInfo pinfo;
  316. pinfo.name="";
  317. pinfo.type=port_types[op];
  318. return pinfo;
  319. }
  320. static const char* op_names[]={
  321. //comparation
  322. "Equal", //OP_EQUAL,
  323. "NotEqual", //OP_NOT_EQUAL,
  324. "Less", //OP_LESS,
  325. "LessEqual", //OP_LESS_EQUAL,
  326. "Greater", //OP_GREATER,
  327. "GreaterEq", //OP_GREATER_EQUAL,
  328. //mathematic
  329. "Add", //OP_ADD,
  330. "Subtract", //OP_SUBSTRACT,
  331. "Multiply", //OP_MULTIPLY,
  332. "Divide", //OP_DIVIDE,
  333. "Negate", //OP_NEGATE,
  334. "Remainder", //OP_MODULE,
  335. "Concat", //OP_STRING_CONCAT,
  336. //bitwise
  337. "ShiftLeft", //OP_SHIFT_LEFT,
  338. "ShiftRight", //OP_SHIFT_RIGHT,
  339. "BitAnd", //OP_BIT_AND,
  340. "BitOr", //OP_BIT_OR,
  341. "BitXor", //OP_BIT_XOR,
  342. "BitNeg", //OP_BIT_NEGATE,
  343. //logic
  344. "And", //OP_AND,
  345. "Or", //OP_OR,
  346. "Xor", //OP_XOR,
  347. "Not", //OP_NOT,
  348. //containment
  349. "In", //OP_IN,
  350. };
  351. String VisualScriptOperator::get_caption() const {
  352. return op_names[op];
  353. }
  354. String VisualScriptOperator::get_text() const {
  355. static const wchar_t* op_names[]={
  356. //comparation
  357. L"A = B", //OP_EQUAL,
  358. L"A \u2260 B", //OP_NOT_EQUAL,
  359. L"A < B", //OP_LESS,
  360. L"A \u2264 B", //OP_LESS_EQUAL,
  361. L"A > B", //OP_GREATER,
  362. L"A \u2265 B", //OP_GREATER_EQUAL,
  363. //mathematic
  364. L"A + B", //OP_ADD,
  365. L"A - B", //OP_SUBSTRACT,
  366. L"A x B", //OP_MULTIPLY,
  367. L"A \u00F7 B", //OP_DIVIDE,
  368. L"\u00AC A", //OP_NEGATE,
  369. L"A mod B", //OP_MODULE,
  370. L"A .. B", //OP_STRING_CONCAT,
  371. //bitwise
  372. L"A << B", //OP_SHIFT_LEFT,
  373. L"A >> B", //OP_SHIFT_RIGHT,
  374. L"A & B", //OP_BIT_AND,
  375. L"A | B", //OP_BIT_OR,
  376. L"A ^ B", //OP_BIT_XOR,
  377. L"~A", //OP_BIT_NEGATE,
  378. //logic
  379. L"A and B", //OP_AND,
  380. L"A or B", //OP_OR,
  381. L"A xor B", //OP_XOR,
  382. L"not A", //OP_NOT,
  383. };
  384. return op_names[op];
  385. }
  386. void VisualScriptOperator::set_operator(Variant::Operator p_op) {
  387. if (op==p_op)
  388. return;
  389. op=p_op;
  390. ports_changed_notify();
  391. }
  392. Variant::Operator VisualScriptOperator::get_operator() const{
  393. return op;
  394. }
  395. void VisualScriptOperator::_bind_methods() {
  396. ObjectTypeDB::bind_method(_MD("set_operator","op"),&VisualScriptOperator::set_operator);
  397. ObjectTypeDB::bind_method(_MD("get_operator"),&VisualScriptOperator::get_operator);
  398. String types;
  399. for(int i=0;i<Variant::OP_MAX;i++) {
  400. if (i>0)
  401. types+=",";
  402. types+=op_names[i];
  403. }
  404. ADD_PROPERTY(PropertyInfo(Variant::INT,"operator_value/type",PROPERTY_HINT_ENUM,types),_SCS("set_operator"),_SCS("get_operator"));
  405. }
  406. class VisualScriptNodeInstanceOperator : public VisualScriptNodeInstance {
  407. public:
  408. bool unary;
  409. Variant::Operator op;
  410. //virtual int get_working_memory_size() const { return 0; }
  411. //virtual bool is_output_port_unsequenced(int p_idx) const { return false; }
  412. //virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return true; }
  413. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  414. bool valid;
  415. if (unary) {
  416. Variant::evaluate(op,*p_inputs[0],Variant(),*p_outputs[0],valid);
  417. } else {
  418. Variant::evaluate(op,*p_inputs[0],*p_inputs[1],*p_outputs[0],valid);
  419. }
  420. if (!valid) {
  421. r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
  422. if (p_outputs[0]->get_type()==Variant::STRING) {
  423. r_error_str=*p_outputs[0];
  424. } else {
  425. if (unary)
  426. r_error_str=String(op_names[op])+RTR(": Invalid argument of type: ")+Variant::get_type_name(p_inputs[0]->get_type());
  427. else
  428. r_error_str=String(op_names[op])+RTR(": Invalid arguments: ")+"A: "+Variant::get_type_name(p_inputs[0]->get_type())+" B: "+Variant::get_type_name(p_inputs[1]->get_type());
  429. }
  430. }
  431. return 0;
  432. }
  433. };
  434. VisualScriptNodeInstance* VisualScriptOperator::instance(VisualScriptInstance* p_instance) {
  435. VisualScriptNodeInstanceOperator * instance = memnew(VisualScriptNodeInstanceOperator );
  436. instance->unary=get_input_value_port_count()==1;
  437. instance->op=op;
  438. return instance;
  439. }
  440. VisualScriptOperator::VisualScriptOperator() {
  441. op=Variant::OP_ADD;
  442. }
  443. template<Variant::Operator OP>
  444. static Ref<VisualScriptNode> create_op_node(const String& p_name) {
  445. Ref<VisualScriptOperator> node;
  446. node.instance();
  447. node->set_operator(OP);
  448. return node;
  449. }
  450. //////////////////////////////////////////
  451. ////////////////VARIABLE GET//////////////////
  452. //////////////////////////////////////////
  453. int VisualScriptVariableGet::get_output_sequence_port_count() const {
  454. return 0;
  455. }
  456. bool VisualScriptVariableGet::has_input_sequence_port() const{
  457. return false;
  458. }
  459. int VisualScriptVariableGet::get_input_value_port_count() const{
  460. return 0;
  461. }
  462. int VisualScriptVariableGet::get_output_value_port_count() const{
  463. return 1;
  464. }
  465. String VisualScriptVariableGet::get_output_sequence_port_text(int p_port) const {
  466. return String();
  467. }
  468. PropertyInfo VisualScriptVariableGet::get_input_value_port_info(int p_idx) const{
  469. return PropertyInfo();
  470. }
  471. PropertyInfo VisualScriptVariableGet::get_output_value_port_info(int p_idx) const{
  472. PropertyInfo pinfo;
  473. pinfo.name="value";
  474. if (get_visual_script().is_valid() && get_visual_script()->has_variable(variable)) {
  475. PropertyInfo vinfo = get_visual_script()->get_variable_info(variable);
  476. pinfo.type=vinfo.type;
  477. pinfo.hint=vinfo.hint;
  478. pinfo.hint_string=vinfo.hint_string;
  479. }
  480. return pinfo;
  481. }
  482. String VisualScriptVariableGet::get_caption() const {
  483. return "Variable";
  484. }
  485. String VisualScriptVariableGet::get_text() const {
  486. return variable;
  487. }
  488. void VisualScriptVariableGet::set_variable(StringName p_variable) {
  489. if (variable==p_variable)
  490. return;
  491. variable=p_variable;
  492. ports_changed_notify();
  493. }
  494. StringName VisualScriptVariableGet::get_variable() const{
  495. return variable;
  496. }
  497. void VisualScriptVariableGet::_validate_property(PropertyInfo& property) const {
  498. if (property.name=="variable/name" && get_visual_script().is_valid()) {
  499. Ref<VisualScript> vs = get_visual_script();
  500. List<StringName> vars;
  501. vs->get_variable_list(&vars);
  502. String vhint;
  503. for (List<StringName>::Element *E=vars.front();E;E=E->next()) {
  504. if (vhint!=String())
  505. vhint+=",";
  506. vhint+=E->get().operator String();
  507. }
  508. property.hint=PROPERTY_HINT_ENUM;
  509. property.hint_string=vhint;
  510. }
  511. }
  512. void VisualScriptVariableGet::_bind_methods() {
  513. ObjectTypeDB::bind_method(_MD("set_variable","name"),&VisualScriptVariableGet::set_variable);
  514. ObjectTypeDB::bind_method(_MD("get_variable"),&VisualScriptVariableGet::get_variable);
  515. ADD_PROPERTY(PropertyInfo(Variant::STRING,"variable/name"),_SCS("set_variable"),_SCS("get_variable"));
  516. }
  517. class VisualScriptNodeInstanceVariableGet : public VisualScriptNodeInstance {
  518. public:
  519. VisualScriptVariableGet *node;
  520. VisualScriptInstance *instance;
  521. StringName variable;
  522. //virtual int get_working_memory_size() const { return 0; }
  523. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  524. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  525. if (instance->get_variable(variable,r_value)==false) {
  526. r_error=RTR("VariableGet not found in script: ")+"'"+String(variable)+"'";
  527. return false;
  528. } else {
  529. return true;
  530. }
  531. }
  532. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  533. return 0;
  534. }
  535. };
  536. VisualScriptNodeInstance* VisualScriptVariableGet::instance(VisualScriptInstance* p_instance) {
  537. VisualScriptNodeInstanceVariableGet * instance = memnew(VisualScriptNodeInstanceVariableGet );
  538. instance->node=this;
  539. instance->instance=p_instance;
  540. instance->variable=variable;
  541. return instance;
  542. }
  543. VisualScriptVariableGet::VisualScriptVariableGet() {
  544. }
  545. //////////////////////////////////////////
  546. ////////////////VARIABLE GET//////////////////
  547. //////////////////////////////////////////
  548. int VisualScriptVariableSet::get_output_sequence_port_count() const {
  549. return 1;
  550. }
  551. bool VisualScriptVariableSet::has_input_sequence_port() const{
  552. return true;
  553. }
  554. int VisualScriptVariableSet::get_input_value_port_count() const{
  555. return 1;
  556. }
  557. int VisualScriptVariableSet::get_output_value_port_count() const{
  558. return 0;
  559. }
  560. String VisualScriptVariableSet::get_output_sequence_port_text(int p_port) const {
  561. return String();
  562. }
  563. PropertyInfo VisualScriptVariableSet::get_input_value_port_info(int p_idx) const{
  564. PropertyInfo pinfo;
  565. pinfo.name="set";
  566. if (get_visual_script().is_valid() && get_visual_script()->has_variable(variable)) {
  567. PropertyInfo vinfo = get_visual_script()->get_variable_info(variable);
  568. pinfo.type=vinfo.type;
  569. pinfo.hint=vinfo.hint;
  570. pinfo.hint_string=vinfo.hint_string;
  571. }
  572. return pinfo;
  573. }
  574. PropertyInfo VisualScriptVariableSet::get_output_value_port_info(int p_idx) const{
  575. return PropertyInfo();
  576. }
  577. String VisualScriptVariableSet::get_caption() const {
  578. return "VariableSet";
  579. }
  580. String VisualScriptVariableSet::get_text() const {
  581. return variable;
  582. }
  583. void VisualScriptVariableSet::set_variable(StringName p_variable) {
  584. if (variable==p_variable)
  585. return;
  586. variable=p_variable;
  587. ports_changed_notify();
  588. }
  589. StringName VisualScriptVariableSet::get_variable() const{
  590. return variable;
  591. }
  592. void VisualScriptVariableSet::_validate_property(PropertyInfo& property) const {
  593. if (property.name=="variable/name" && get_visual_script().is_valid()) {
  594. Ref<VisualScript> vs = get_visual_script();
  595. List<StringName> vars;
  596. vs->get_variable_list(&vars);
  597. String vhint;
  598. for (List<StringName>::Element *E=vars.front();E;E=E->next()) {
  599. if (vhint!=String())
  600. vhint+=",";
  601. vhint+=E->get().operator String();
  602. }
  603. property.hint=PROPERTY_HINT_ENUM;
  604. property.hint_string=vhint;
  605. }
  606. }
  607. void VisualScriptVariableSet::_bind_methods() {
  608. ObjectTypeDB::bind_method(_MD("set_variable","name"),&VisualScriptVariableSet::set_variable);
  609. ObjectTypeDB::bind_method(_MD("get_variable"),&VisualScriptVariableSet::get_variable);
  610. ADD_PROPERTY(PropertyInfo(Variant::STRING,"variable/name"),_SCS("set_variable"),_SCS("get_variable"));
  611. }
  612. class VisualScriptNodeInstanceVariableSet : public VisualScriptNodeInstance {
  613. public:
  614. VisualScriptVariableSet *node;
  615. VisualScriptInstance *instance;
  616. StringName variable;
  617. //virtual int get_working_memory_size() const { return 0; }
  618. virtual bool is_output_port_unsequenced(int p_idx) const { return false; }
  619. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return false; }
  620. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  621. if (instance->set_variable(variable,*p_inputs[0])==false) {
  622. r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD ;
  623. r_error_str=RTR("VariableSet not found in script: ")+"'"+String(variable)+"'";
  624. }
  625. return 0;
  626. }
  627. };
  628. VisualScriptNodeInstance* VisualScriptVariableSet::instance(VisualScriptInstance* p_instance) {
  629. VisualScriptNodeInstanceVariableSet * instance = memnew(VisualScriptNodeInstanceVariableSet );
  630. instance->node=this;
  631. instance->instance=p_instance;
  632. instance->variable=variable;
  633. return instance;
  634. }
  635. VisualScriptVariableSet::VisualScriptVariableSet() {
  636. }
  637. //////////////////////////////////////////
  638. ////////////////CONSTANT//////////////////
  639. //////////////////////////////////////////
  640. int VisualScriptConstant::get_output_sequence_port_count() const {
  641. return 0;
  642. }
  643. bool VisualScriptConstant::has_input_sequence_port() const{
  644. return false;
  645. }
  646. int VisualScriptConstant::get_input_value_port_count() const{
  647. return 0;
  648. }
  649. int VisualScriptConstant::get_output_value_port_count() const{
  650. return 1;
  651. }
  652. String VisualScriptConstant::get_output_sequence_port_text(int p_port) const {
  653. return String();
  654. }
  655. PropertyInfo VisualScriptConstant::get_input_value_port_info(int p_idx) const{
  656. return PropertyInfo();
  657. }
  658. PropertyInfo VisualScriptConstant::get_output_value_port_info(int p_idx) const{
  659. PropertyInfo pinfo;
  660. pinfo.name="get";
  661. pinfo.type=type;
  662. return pinfo;
  663. }
  664. String VisualScriptConstant::get_caption() const {
  665. return "Constant";
  666. }
  667. String VisualScriptConstant::get_text() const {
  668. return String(value);
  669. }
  670. void VisualScriptConstant::set_constant_type(Variant::Type p_type) {
  671. if (type==p_type)
  672. return;
  673. type=p_type;
  674. ports_changed_notify();
  675. Variant::CallError ce;
  676. value=Variant::construct(type,NULL,0,ce);
  677. _change_notify();
  678. }
  679. Variant::Type VisualScriptConstant::get_constant_type() const{
  680. return type;
  681. }
  682. void VisualScriptConstant::set_constant_value(Variant p_value){
  683. if (value==p_value)
  684. return;
  685. value=p_value;
  686. ports_changed_notify();
  687. }
  688. Variant VisualScriptConstant::get_constant_value() const{
  689. return value;
  690. }
  691. void VisualScriptConstant::_validate_property(PropertyInfo& property) const {
  692. if (property.name=="constant/value") {
  693. property.type=type;
  694. if (type==Variant::NIL)
  695. property.usage=0; //do not save if nil
  696. }
  697. }
  698. void VisualScriptConstant::_bind_methods() {
  699. ObjectTypeDB::bind_method(_MD("set_constant_type","type"),&VisualScriptConstant::set_constant_type);
  700. ObjectTypeDB::bind_method(_MD("get_constant_type"),&VisualScriptConstant::get_constant_type);
  701. ObjectTypeDB::bind_method(_MD("set_constant_value","value"),&VisualScriptConstant::set_constant_value);
  702. ObjectTypeDB::bind_method(_MD("get_constant_value"),&VisualScriptConstant::get_constant_value);
  703. String argt="Null";
  704. for(int i=1;i<Variant::VARIANT_MAX;i++) {
  705. argt+=","+Variant::get_type_name(Variant::Type(i));
  706. }
  707. ADD_PROPERTY(PropertyInfo(Variant::INT,"constant/type",PROPERTY_HINT_ENUM,argt),_SCS("set_constant_type"),_SCS("get_constant_type"));
  708. ADD_PROPERTY(PropertyInfo(Variant::NIL,"constant/value"),_SCS("set_constant_value"),_SCS("get_constant_value"));
  709. }
  710. class VisualScriptNodeInstanceConstant : public VisualScriptNodeInstance {
  711. public:
  712. Variant constant;
  713. //virtual int get_working_memory_size() const { return 0; }
  714. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  715. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  716. *r_value=constant;
  717. return true;
  718. }
  719. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  720. return 0;
  721. }
  722. };
  723. VisualScriptNodeInstance* VisualScriptConstant::instance(VisualScriptInstance* p_instance) {
  724. VisualScriptNodeInstanceConstant * instance = memnew(VisualScriptNodeInstanceConstant );
  725. instance->constant=value;
  726. return instance;
  727. }
  728. VisualScriptConstant::VisualScriptConstant() {
  729. type=Variant::NIL;
  730. }
  731. //////////////////////////////////////////
  732. ////////////////INDEX////////////////////
  733. //////////////////////////////////////////
  734. int VisualScriptIndexGet::get_output_sequence_port_count() const {
  735. return 1;
  736. }
  737. bool VisualScriptIndexGet::has_input_sequence_port() const{
  738. return true;
  739. }
  740. int VisualScriptIndexGet::get_input_value_port_count() const{
  741. return 2;
  742. }
  743. int VisualScriptIndexGet::get_output_value_port_count() const{
  744. return 1;
  745. }
  746. String VisualScriptIndexGet::get_output_sequence_port_text(int p_port) const {
  747. return String();
  748. }
  749. PropertyInfo VisualScriptIndexGet::get_input_value_port_info(int p_idx) const{
  750. if (p_idx==0) {
  751. return PropertyInfo(Variant::NIL,"base");
  752. } else {
  753. return PropertyInfo(Variant::NIL,"index");
  754. }
  755. }
  756. PropertyInfo VisualScriptIndexGet::get_output_value_port_info(int p_idx) const{
  757. return PropertyInfo();
  758. }
  759. String VisualScriptIndexGet::get_caption() const {
  760. return "IndexGet";
  761. }
  762. String VisualScriptIndexGet::get_text() const {
  763. return String("get");
  764. }
  765. class VisualScriptNodeInstanceIndexGet : public VisualScriptNodeInstance {
  766. public:
  767. //virtual int get_working_memory_size() const { return 0; }
  768. //virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  769. //virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return false; }
  770. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  771. bool valid;
  772. *p_outputs[0] = p_inputs[0]->get(*p_inputs[1],&valid);
  773. if (!valid) {
  774. r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
  775. r_error_str="Invalid get: "+p_inputs[0]->get_construct_string();
  776. }
  777. return 0;
  778. }
  779. };
  780. VisualScriptNodeInstance* VisualScriptIndexGet::instance(VisualScriptInstance* p_instance) {
  781. VisualScriptNodeInstanceIndexGet * instance = memnew(VisualScriptNodeInstanceIndexGet );
  782. return instance;
  783. }
  784. VisualScriptIndexGet::VisualScriptIndexGet() {
  785. }
  786. //////////////////////////////////////////
  787. ////////////////INDEXSET//////////////////
  788. //////////////////////////////////////////
  789. int VisualScriptIndexSet::get_output_sequence_port_count() const {
  790. return 1;
  791. }
  792. bool VisualScriptIndexSet::has_input_sequence_port() const{
  793. return true;
  794. }
  795. int VisualScriptIndexSet::get_input_value_port_count() const{
  796. return 3;
  797. }
  798. int VisualScriptIndexSet::get_output_value_port_count() const{
  799. return 0;
  800. }
  801. String VisualScriptIndexSet::get_output_sequence_port_text(int p_port) const {
  802. return String();
  803. }
  804. PropertyInfo VisualScriptIndexSet::get_input_value_port_info(int p_idx) const{
  805. if (p_idx==0) {
  806. return PropertyInfo(Variant::NIL,"base");
  807. } else if (p_idx==1){
  808. return PropertyInfo(Variant::NIL,"index");
  809. } else {
  810. return PropertyInfo(Variant::NIL,"value");
  811. }
  812. }
  813. PropertyInfo VisualScriptIndexSet::get_output_value_port_info(int p_idx) const{
  814. return PropertyInfo();
  815. }
  816. String VisualScriptIndexSet::get_caption() const {
  817. return "IndexSet";
  818. }
  819. String VisualScriptIndexSet::get_text() const {
  820. return String("set");
  821. }
  822. class VisualScriptNodeInstanceIndexSet : public VisualScriptNodeInstance {
  823. public:
  824. //virtual int get_working_memory_size() const { return 0; }
  825. //virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  826. //virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return false; }
  827. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  828. bool valid;
  829. *p_outputs[0]=*p_inputs[0];
  830. p_outputs[0]->set(*p_inputs[1],*p_inputs[2],&valid);
  831. if (!valid) {
  832. r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
  833. r_error_str="Invalid set: "+p_inputs[1]->get_construct_string();
  834. }
  835. return 0;
  836. }
  837. };
  838. VisualScriptNodeInstance* VisualScriptIndexSet::instance(VisualScriptInstance* p_instance) {
  839. VisualScriptNodeInstanceIndexSet * instance = memnew(VisualScriptNodeInstanceIndexSet );
  840. return instance;
  841. }
  842. VisualScriptIndexSet::VisualScriptIndexSet() {
  843. }
  844. //////////////////////////////////////////
  845. ////////////////GLOBALCONSTANT///////////
  846. //////////////////////////////////////////
  847. int VisualScriptGlobalConstant::get_output_sequence_port_count() const {
  848. return 0;
  849. }
  850. bool VisualScriptGlobalConstant::has_input_sequence_port() const{
  851. return false;
  852. }
  853. int VisualScriptGlobalConstant::get_input_value_port_count() const{
  854. return 0;
  855. }
  856. int VisualScriptGlobalConstant::get_output_value_port_count() const{
  857. return 1;
  858. }
  859. String VisualScriptGlobalConstant::get_output_sequence_port_text(int p_port) const {
  860. return String();
  861. }
  862. PropertyInfo VisualScriptGlobalConstant::get_input_value_port_info(int p_idx) const{
  863. return PropertyInfo();
  864. }
  865. PropertyInfo VisualScriptGlobalConstant::get_output_value_port_info(int p_idx) const{
  866. return PropertyInfo(Variant::REAL,"value");
  867. }
  868. String VisualScriptGlobalConstant::get_caption() const {
  869. return "GlobalConst";
  870. }
  871. String VisualScriptGlobalConstant::get_text() const {
  872. return GlobalConstants::get_global_constant_name(index);
  873. }
  874. void VisualScriptGlobalConstant::set_global_constant(int p_which) {
  875. index=p_which;
  876. _change_notify();
  877. ports_changed_notify();
  878. }
  879. int VisualScriptGlobalConstant::get_global_constant() {
  880. return index;
  881. }
  882. class VisualScriptNodeInstanceGlobalConstant : public VisualScriptNodeInstance {
  883. public:
  884. int index;
  885. //virtual int get_working_memory_size() const { return 0; }
  886. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  887. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  888. *r_value = GlobalConstants::get_global_constant_value(index);
  889. return true;
  890. }
  891. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  892. return 0;
  893. }
  894. };
  895. VisualScriptNodeInstance* VisualScriptGlobalConstant::instance(VisualScriptInstance* p_instance) {
  896. VisualScriptNodeInstanceGlobalConstant * instance = memnew(VisualScriptNodeInstanceGlobalConstant );
  897. instance->index=index;
  898. return instance;
  899. }
  900. void VisualScriptGlobalConstant::_bind_methods() {
  901. ObjectTypeDB::bind_method(_MD("set_global_constant","index"),&VisualScriptGlobalConstant::set_global_constant);
  902. ObjectTypeDB::bind_method(_MD("get_global_constant"),&VisualScriptGlobalConstant::get_global_constant);
  903. String cc;
  904. for(int i=0;i<GlobalConstants::get_global_constant_count();i++) {
  905. if (i>0)
  906. cc+=",";
  907. cc+=GlobalConstants::get_global_constant_name(i);
  908. }
  909. ADD_PROPERTY(PropertyInfo(Variant::INT,"constant",PROPERTY_HINT_ENUM,cc),_SCS("set_global_constant"),_SCS("get_global_constant"));
  910. }
  911. VisualScriptGlobalConstant::VisualScriptGlobalConstant() {
  912. index=0;
  913. }
  914. //////////////////////////////////////////
  915. ////////////////MATHCONSTANT///////////
  916. //////////////////////////////////////////
  917. const char* VisualScriptMathConstant::const_name[MATH_CONSTANT_MAX]={
  918. "One",
  919. "PI",
  920. "PIx2",
  921. "PI/2",
  922. "E",
  923. "Sqrt2",
  924. };
  925. double VisualScriptMathConstant::const_value[MATH_CONSTANT_MAX]={
  926. 1.0,
  927. Math_PI,
  928. Math_PI*2,
  929. Math_PI*0.5,
  930. 2.71828182845904523536,
  931. Math::sqrt(2.0)
  932. };
  933. int VisualScriptMathConstant::get_output_sequence_port_count() const {
  934. return 0;
  935. }
  936. bool VisualScriptMathConstant::has_input_sequence_port() const{
  937. return false;
  938. }
  939. int VisualScriptMathConstant::get_input_value_port_count() const{
  940. return 0;
  941. }
  942. int VisualScriptMathConstant::get_output_value_port_count() const{
  943. return 1;
  944. }
  945. String VisualScriptMathConstant::get_output_sequence_port_text(int p_port) const {
  946. return String();
  947. }
  948. PropertyInfo VisualScriptMathConstant::get_input_value_port_info(int p_idx) const{
  949. return PropertyInfo();
  950. }
  951. PropertyInfo VisualScriptMathConstant::get_output_value_port_info(int p_idx) const{
  952. return PropertyInfo(Variant::REAL,"value");
  953. }
  954. String VisualScriptMathConstant::get_caption() const {
  955. return "MathConst";
  956. }
  957. String VisualScriptMathConstant::get_text() const {
  958. return const_name[constant];
  959. }
  960. void VisualScriptMathConstant::set_math_constant(MathConstant p_which) {
  961. constant=p_which;
  962. _change_notify();
  963. ports_changed_notify();
  964. }
  965. VisualScriptMathConstant::MathConstant VisualScriptMathConstant::get_math_constant() {
  966. return constant;
  967. }
  968. class VisualScriptNodeInstanceMathConstant : public VisualScriptNodeInstance {
  969. public:
  970. float value;
  971. //virtual int get_working_memory_size() const { return 0; }
  972. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  973. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  974. *r_value = value;
  975. return true;
  976. }
  977. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  978. return 0;
  979. }
  980. };
  981. VisualScriptNodeInstance* VisualScriptMathConstant::instance(VisualScriptInstance* p_instance) {
  982. VisualScriptNodeInstanceMathConstant * instance = memnew(VisualScriptNodeInstanceMathConstant );
  983. instance->value=const_value[constant];
  984. return instance;
  985. }
  986. void VisualScriptMathConstant::_bind_methods() {
  987. ObjectTypeDB::bind_method(_MD("set_math_constant","which"),&VisualScriptMathConstant::set_math_constant);
  988. ObjectTypeDB::bind_method(_MD("get_math_constant"),&VisualScriptMathConstant::get_math_constant);
  989. String cc;
  990. for(int i=0;i<MATH_CONSTANT_MAX;i++) {
  991. if (i>0)
  992. cc+=",";
  993. cc+=const_name[i];
  994. }
  995. ADD_PROPERTY(PropertyInfo(Variant::INT,"constant",PROPERTY_HINT_ENUM,cc),_SCS("set_math_constant"),_SCS("get_math_constant"));
  996. }
  997. VisualScriptMathConstant::VisualScriptMathConstant() {
  998. constant=MATH_CONSTANT_ONE;
  999. }
  1000. //////////////////////////////////////////
  1001. ////////////////GLOBALSINGLETON///////////
  1002. //////////////////////////////////////////
  1003. int VisualScriptEngineSingleton::get_output_sequence_port_count() const {
  1004. return 0;
  1005. }
  1006. bool VisualScriptEngineSingleton::has_input_sequence_port() const{
  1007. return false;
  1008. }
  1009. int VisualScriptEngineSingleton::get_input_value_port_count() const{
  1010. return 0;
  1011. }
  1012. int VisualScriptEngineSingleton::get_output_value_port_count() const{
  1013. return 1;
  1014. }
  1015. String VisualScriptEngineSingleton::get_output_sequence_port_text(int p_port) const {
  1016. return String();
  1017. }
  1018. PropertyInfo VisualScriptEngineSingleton::get_input_value_port_info(int p_idx) const{
  1019. return PropertyInfo();
  1020. }
  1021. PropertyInfo VisualScriptEngineSingleton::get_output_value_port_info(int p_idx) const{
  1022. return PropertyInfo(Variant::OBJECT,"instance");
  1023. }
  1024. String VisualScriptEngineSingleton::get_caption() const {
  1025. return "EngineSingleton";
  1026. }
  1027. String VisualScriptEngineSingleton::get_text() const {
  1028. return singleton;
  1029. }
  1030. void VisualScriptEngineSingleton::set_singleton(const String& p_string) {
  1031. singleton=p_string;
  1032. _change_notify();
  1033. ports_changed_notify();
  1034. }
  1035. String VisualScriptEngineSingleton::get_singleton() {
  1036. return singleton;
  1037. }
  1038. class VisualScriptNodeInstanceEngineSingleton : public VisualScriptNodeInstance {
  1039. public:
  1040. Object* singleton;
  1041. //virtual int get_working_memory_size() const { return 0; }
  1042. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  1043. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  1044. *r_value=singleton;
  1045. return true;
  1046. }
  1047. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  1048. return 0;
  1049. }
  1050. };
  1051. VisualScriptNodeInstance* VisualScriptEngineSingleton::instance(VisualScriptInstance* p_instance) {
  1052. VisualScriptNodeInstanceEngineSingleton * instance = memnew(VisualScriptNodeInstanceEngineSingleton );
  1053. instance->singleton=Globals::get_singleton()->get_singleton_object(singleton);
  1054. return instance;
  1055. }
  1056. void VisualScriptEngineSingleton::_bind_methods() {
  1057. ObjectTypeDB::bind_method(_MD("set_singleton","name"),&VisualScriptEngineSingleton::set_singleton);
  1058. ObjectTypeDB::bind_method(_MD("get_singleton"),&VisualScriptEngineSingleton::get_singleton);
  1059. String cc;
  1060. List<Globals::Singleton> singletons;
  1061. Globals::get_singleton()->get_singletons(&singletons);
  1062. for (List<Globals::Singleton>::Element *E=singletons.front();E;E=E->next()) {
  1063. if (E->get().name=="VS" || E->get().name=="PS" || E->get().name=="PS2D" || E->get().name=="AS" || E->get().name=="TS" || E->get().name=="SS" || E->get().name=="SS2D")
  1064. continue; //skip these, too simple named
  1065. if (cc!=String())
  1066. cc+=",";
  1067. cc+=E->get().name;
  1068. }
  1069. ADD_PROPERTY(PropertyInfo(Variant::STRING,"constant",PROPERTY_HINT_ENUM,cc),_SCS("set_singleton"),_SCS("get_singleton"));
  1070. }
  1071. VisualScriptEngineSingleton::VisualScriptEngineSingleton() {
  1072. singleton=String();
  1073. }
  1074. //////////////////////////////////////////
  1075. ////////////////GETNODE///////////
  1076. //////////////////////////////////////////
  1077. int VisualScriptSceneNode::get_output_sequence_port_count() const {
  1078. return 0;
  1079. }
  1080. bool VisualScriptSceneNode::has_input_sequence_port() const{
  1081. return false;
  1082. }
  1083. int VisualScriptSceneNode::get_input_value_port_count() const{
  1084. return 0;
  1085. }
  1086. int VisualScriptSceneNode::get_output_value_port_count() const{
  1087. return 1;
  1088. }
  1089. String VisualScriptSceneNode::get_output_sequence_port_text(int p_port) const {
  1090. return String();
  1091. }
  1092. PropertyInfo VisualScriptSceneNode::get_input_value_port_info(int p_idx) const{
  1093. return PropertyInfo();
  1094. }
  1095. PropertyInfo VisualScriptSceneNode::get_output_value_port_info(int p_idx) const{
  1096. return PropertyInfo(Variant::OBJECT,"node");
  1097. }
  1098. String VisualScriptSceneNode::get_caption() const {
  1099. return "SceneNode";
  1100. }
  1101. String VisualScriptSceneNode::get_text() const {
  1102. return path.simplified();
  1103. }
  1104. void VisualScriptSceneNode::set_node_path(const NodePath& p_path) {
  1105. path=p_path;
  1106. _change_notify();
  1107. ports_changed_notify();
  1108. }
  1109. NodePath VisualScriptSceneNode::get_node_path() {
  1110. return path;
  1111. }
  1112. class VisualScriptNodeInstanceSceneNode : public VisualScriptNodeInstance {
  1113. public:
  1114. VisualScriptSceneNode *node;
  1115. VisualScriptInstance *instance;
  1116. NodePath path;
  1117. //virtual int get_working_memory_size() const { return 0; }
  1118. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  1119. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  1120. Node* node = instance->get_owner_ptr()->cast_to<Node>();
  1121. if (!node) {
  1122. r_error="Base object is not a Node!";
  1123. return false;
  1124. }
  1125. Node* another = node->get_node(path);
  1126. if (!node) {
  1127. r_error="Path does not lead Node!";
  1128. return false;
  1129. }
  1130. *r_value=another;
  1131. return true;
  1132. }
  1133. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  1134. return 0;
  1135. }
  1136. };
  1137. VisualScriptNodeInstance* VisualScriptSceneNode::instance(VisualScriptInstance* p_instance) {
  1138. VisualScriptNodeInstanceSceneNode * instance = memnew(VisualScriptNodeInstanceSceneNode );
  1139. instance->node=this;
  1140. instance->instance=p_instance;
  1141. instance->path=path;
  1142. return instance;
  1143. }
  1144. #ifdef TOOLS_ENABLED
  1145. static Node* _find_script_node(Node* p_edited_scene,Node* p_current_node,const Ref<Script> &script) {
  1146. if (p_edited_scene!=p_current_node && p_current_node->get_owner()!=p_edited_scene)
  1147. return NULL;
  1148. Ref<Script> scr = p_current_node->get_script();
  1149. if (scr.is_valid() && scr==script)
  1150. return p_current_node;
  1151. for(int i=0;i<p_current_node->get_child_count();i++) {
  1152. Node *n = _find_script_node(p_edited_scene,p_current_node->get_child(i),script);
  1153. if (n)
  1154. return n;
  1155. }
  1156. return NULL;
  1157. }
  1158. #endif
  1159. void VisualScriptSceneNode::_validate_property(PropertyInfo& property) const {
  1160. #ifdef TOOLS_ENABLED
  1161. if (property.name=="node_path") {
  1162. Ref<Script> script = get_visual_script();
  1163. if (!script.is_valid())
  1164. return;
  1165. MainLoop * main_loop = OS::get_singleton()->get_main_loop();
  1166. if (!main_loop)
  1167. return;
  1168. SceneTree *scene_tree = main_loop->cast_to<SceneTree>();
  1169. if (!scene_tree)
  1170. return;
  1171. Node *edited_scene = scene_tree->get_edited_scene_root();
  1172. if (!edited_scene)
  1173. return;
  1174. Node* script_node = _find_script_node(edited_scene,edited_scene,script);
  1175. if (!script_node)
  1176. return;
  1177. property.hint_string=script_node->get_path();
  1178. }
  1179. #endif
  1180. }
  1181. void VisualScriptSceneNode::_bind_methods() {
  1182. ObjectTypeDB::bind_method(_MD("set_node_path","path"),&VisualScriptSceneNode::set_node_path);
  1183. ObjectTypeDB::bind_method(_MD("get_node_path"),&VisualScriptSceneNode::get_node_path);
  1184. ADD_PROPERTY(PropertyInfo(Variant::NODE_PATH,"node_path",PROPERTY_HINT_NODE_PATH_TO_EDITED_NODE),_SCS("set_node_path"),_SCS("get_node_path"));
  1185. }
  1186. VisualScriptSceneNode::VisualScriptSceneNode() {
  1187. path=String(".");
  1188. }
  1189. //////////////////////////////////////////
  1190. ////////////////SceneTree///////////
  1191. //////////////////////////////////////////
  1192. int VisualScriptSceneTree::get_output_sequence_port_count() const {
  1193. return 0;
  1194. }
  1195. bool VisualScriptSceneTree::has_input_sequence_port() const{
  1196. return false;
  1197. }
  1198. int VisualScriptSceneTree::get_input_value_port_count() const{
  1199. return 0;
  1200. }
  1201. int VisualScriptSceneTree::get_output_value_port_count() const{
  1202. return 1;
  1203. }
  1204. String VisualScriptSceneTree::get_output_sequence_port_text(int p_port) const {
  1205. return String();
  1206. }
  1207. PropertyInfo VisualScriptSceneTree::get_input_value_port_info(int p_idx) const{
  1208. return PropertyInfo();
  1209. }
  1210. PropertyInfo VisualScriptSceneTree::get_output_value_port_info(int p_idx) const{
  1211. return PropertyInfo(Variant::OBJECT,"instance");
  1212. }
  1213. String VisualScriptSceneTree::get_caption() const {
  1214. return "SceneTree";
  1215. }
  1216. String VisualScriptSceneTree::get_text() const {
  1217. return "";
  1218. }
  1219. class VisualScriptNodeInstanceSceneTree : public VisualScriptNodeInstance {
  1220. public:
  1221. VisualScriptSceneTree *node;
  1222. VisualScriptInstance *instance;
  1223. //virtual int get_working_memory_size() const { return 0; }
  1224. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  1225. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  1226. Node* node = instance->get_owner_ptr()->cast_to<Node>();
  1227. if (!node) {
  1228. r_error="Base object is not a Node!";
  1229. return false;
  1230. }
  1231. SceneTree* tree = node->get_tree();
  1232. if (!tree) {
  1233. r_error="Attempt to get SceneTree while node is not in the active tree.";
  1234. return false;
  1235. }
  1236. *r_value=tree;
  1237. return true;
  1238. }
  1239. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  1240. return 0;
  1241. }
  1242. };
  1243. VisualScriptNodeInstance* VisualScriptSceneTree::instance(VisualScriptInstance* p_instance) {
  1244. VisualScriptNodeInstanceSceneTree * instance = memnew(VisualScriptNodeInstanceSceneTree );
  1245. instance->node=this;
  1246. instance->instance=p_instance;
  1247. return instance;
  1248. }
  1249. void VisualScriptSceneTree::_validate_property(PropertyInfo& property) const {
  1250. }
  1251. void VisualScriptSceneTree::_bind_methods() {
  1252. }
  1253. VisualScriptSceneTree::VisualScriptSceneTree() {
  1254. }
  1255. //////////////////////////////////////////
  1256. ////////////////RESPATH///////////
  1257. //////////////////////////////////////////
  1258. int VisualScriptResourcePath::get_output_sequence_port_count() const {
  1259. return 0;
  1260. }
  1261. bool VisualScriptResourcePath::has_input_sequence_port() const{
  1262. return false;
  1263. }
  1264. int VisualScriptResourcePath::get_input_value_port_count() const{
  1265. return 0;
  1266. }
  1267. int VisualScriptResourcePath::get_output_value_port_count() const{
  1268. return 1;
  1269. }
  1270. String VisualScriptResourcePath::get_output_sequence_port_text(int p_port) const {
  1271. return String();
  1272. }
  1273. PropertyInfo VisualScriptResourcePath::get_input_value_port_info(int p_idx) const{
  1274. return PropertyInfo();
  1275. }
  1276. PropertyInfo VisualScriptResourcePath::get_output_value_port_info(int p_idx) const{
  1277. return PropertyInfo(Variant::STRING,"path");
  1278. }
  1279. String VisualScriptResourcePath::get_caption() const {
  1280. return "ResourcePath";
  1281. }
  1282. String VisualScriptResourcePath::get_text() const {
  1283. return path;
  1284. }
  1285. void VisualScriptResourcePath::set_resource_path(const String& p_path) {
  1286. path=p_path;
  1287. _change_notify();
  1288. ports_changed_notify();
  1289. }
  1290. String VisualScriptResourcePath::get_resource_path() {
  1291. return path;
  1292. }
  1293. class VisualScriptNodeInstanceResourcePath : public VisualScriptNodeInstance {
  1294. public:
  1295. String path;
  1296. //virtual int get_working_memory_size() const { return 0; }
  1297. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  1298. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  1299. *r_value = path;
  1300. return true;
  1301. }
  1302. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  1303. return 0;
  1304. }
  1305. };
  1306. VisualScriptNodeInstance* VisualScriptResourcePath::instance(VisualScriptInstance* p_instance) {
  1307. VisualScriptNodeInstanceResourcePath * instance = memnew(VisualScriptNodeInstanceResourcePath );
  1308. instance->path=path;
  1309. return instance;
  1310. }
  1311. void VisualScriptResourcePath::_bind_methods() {
  1312. ObjectTypeDB::bind_method(_MD("set_resource_path","path"),&VisualScriptResourcePath::set_resource_path);
  1313. ObjectTypeDB::bind_method(_MD("get_resource_path"),&VisualScriptResourcePath::get_resource_path);
  1314. ADD_PROPERTY(PropertyInfo(Variant::STRING,"path",PROPERTY_HINT_FILE),_SCS("set_resource_path"),_SCS("get_resource_path"));
  1315. }
  1316. VisualScriptResourcePath::VisualScriptResourcePath() {
  1317. path="";
  1318. }
  1319. //////////////////////////////////////////
  1320. ////////////////SELF///////////
  1321. //////////////////////////////////////////
  1322. int VisualScriptSelf::get_output_sequence_port_count() const {
  1323. return 0;
  1324. }
  1325. bool VisualScriptSelf::has_input_sequence_port() const{
  1326. return false;
  1327. }
  1328. int VisualScriptSelf::get_input_value_port_count() const{
  1329. return 0;
  1330. }
  1331. int VisualScriptSelf::get_output_value_port_count() const{
  1332. return 1;
  1333. }
  1334. String VisualScriptSelf::get_output_sequence_port_text(int p_port) const {
  1335. return String();
  1336. }
  1337. PropertyInfo VisualScriptSelf::get_input_value_port_info(int p_idx) const{
  1338. return PropertyInfo();
  1339. }
  1340. PropertyInfo VisualScriptSelf::get_output_value_port_info(int p_idx) const{
  1341. return PropertyInfo(Variant::OBJECT,"instance");
  1342. }
  1343. String VisualScriptSelf::get_caption() const {
  1344. return "Self";
  1345. }
  1346. String VisualScriptSelf::get_text() const {
  1347. if (get_visual_script().is_valid())
  1348. return get_visual_script()->get_instance_base_type();
  1349. else
  1350. return "";
  1351. }
  1352. class VisualScriptNodeInstanceSelf : public VisualScriptNodeInstance {
  1353. public:
  1354. VisualScriptInstance* instance;
  1355. //virtual int get_working_memory_size() const { return 0; }
  1356. virtual bool is_output_port_unsequenced(int p_idx) const { return true; }
  1357. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  1358. *r_value = instance->get_owner_ptr();
  1359. return true;
  1360. }
  1361. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  1362. return 0;
  1363. }
  1364. };
  1365. VisualScriptNodeInstance* VisualScriptSelf::instance(VisualScriptInstance* p_instance) {
  1366. VisualScriptNodeInstanceSelf * instance = memnew(VisualScriptNodeInstanceSelf );
  1367. instance->instance=p_instance;
  1368. return instance;
  1369. }
  1370. void VisualScriptSelf::_bind_methods() {
  1371. }
  1372. VisualScriptSelf::VisualScriptSelf() {
  1373. }
  1374. //////////////////////////////////////////
  1375. ////////////////CUSTOM (SCRIPTED)///////////
  1376. //////////////////////////////////////////
  1377. int VisualScriptCustomNode::get_output_sequence_port_count() const {
  1378. if (get_script_instance() && get_script_instance()->has_method("_get_output_sequence_port_count")) {
  1379. return get_script_instance()->call("_get_output_sequence_port_count");
  1380. }
  1381. return 0;
  1382. }
  1383. bool VisualScriptCustomNode::has_input_sequence_port() const{
  1384. if (get_script_instance() && get_script_instance()->has_method("_has_input_sequence_port")) {
  1385. return get_script_instance()->call("_has_input_sequence_port");
  1386. }
  1387. return false;
  1388. }
  1389. int VisualScriptCustomNode::get_input_value_port_count() const{
  1390. if (get_script_instance() && get_script_instance()->has_method("_get_input_value_port_count")) {
  1391. return get_script_instance()->call("_get_input_value_port_count");
  1392. }
  1393. return 0;
  1394. }
  1395. int VisualScriptCustomNode::get_output_value_port_count() const{
  1396. if (get_script_instance() && get_script_instance()->has_method("_get_output_value_port_count")) {
  1397. return get_script_instance()->call("_get_output_value_port_count");
  1398. }
  1399. return 0;
  1400. }
  1401. String VisualScriptCustomNode::get_output_sequence_port_text(int p_port) const {
  1402. if (get_script_instance() && get_script_instance()->has_method("_get_output_sequence_port_text")) {
  1403. return get_script_instance()->call("_get_output_sequence_port_text",p_port);
  1404. }
  1405. return String();
  1406. }
  1407. PropertyInfo VisualScriptCustomNode::get_input_value_port_info(int p_idx) const{
  1408. PropertyInfo info;
  1409. if (get_script_instance() && get_script_instance()->has_method("_get_input_value_port_type")) {
  1410. info.type=Variant::Type(int(get_script_instance()->call("_get_input_value_port_type",p_idx)));
  1411. }
  1412. if (get_script_instance() && get_script_instance()->has_method("_get_input_value_port_name")) {
  1413. info.name=get_script_instance()->call("_get_input_value_port_name",p_idx);
  1414. }
  1415. return info;
  1416. }
  1417. PropertyInfo VisualScriptCustomNode::get_output_value_port_info(int p_idx) const{
  1418. PropertyInfo info;
  1419. if (get_script_instance() && get_script_instance()->has_method("_get_output_value_port_type")) {
  1420. info.type=Variant::Type(int(get_script_instance()->call("_get_output_value_port_type",p_idx)));
  1421. }
  1422. if (get_script_instance() && get_script_instance()->has_method("_get_output_value_port_name")) {
  1423. info.name=get_script_instance()->call("_get_output_value_port_name",p_idx);
  1424. }
  1425. return info;
  1426. }
  1427. String VisualScriptCustomNode::get_caption() const {
  1428. if (get_script_instance() && get_script_instance()->has_method("_get_caption")) {
  1429. return get_script_instance()->call("_get_caption");
  1430. }
  1431. return "CustomNode";
  1432. }
  1433. String VisualScriptCustomNode::get_text() const {
  1434. if (get_script_instance() && get_script_instance()->has_method("_get_text")) {
  1435. return get_script_instance()->call("_get_text");
  1436. }
  1437. return "";
  1438. }
  1439. String VisualScriptCustomNode::get_category() const {
  1440. if (get_script_instance() && get_script_instance()->has_method("_get_category")) {
  1441. return get_script_instance()->call("_get_category");
  1442. }
  1443. return "custom";
  1444. }
  1445. class VisualScriptNodeInstanceCustomNode : public VisualScriptNodeInstance {
  1446. public:
  1447. VisualScriptInstance* instance;
  1448. VisualScriptCustomNode *node;
  1449. int in_count;
  1450. int out_count;
  1451. int work_mem_size;
  1452. Vector<bool> out_unsequenced;
  1453. virtual int get_working_memory_size() const { return work_mem_size; }
  1454. virtual bool is_output_port_unsequenced(int p_idx) const { return out_unsequenced[p_idx]; }
  1455. virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const {
  1456. if (!node->get_script_instance() || !node->get_script_instance()->has_method(VisualScriptLanguage::singleton->_get_output_port_unsequenced)) {
  1457. #ifdef DEBUG_ENABLED
  1458. r_error=RTR("Custom node has no _get_output_port_unsequenced(idx,wmem), but unsequenced ports were specified.");
  1459. return false;
  1460. }
  1461. #endif
  1462. Array work_mem(true);
  1463. work_mem.resize(work_mem_size);
  1464. *r_value = node->get_script_instance()->call(VisualScriptLanguage::singleton->_get_output_port_unsequenced,p_idx,work_mem);
  1465. for(int i=0;i<work_mem_size;i++) {
  1466. if (i<work_mem.size()) {
  1467. p_working_mem[i]=work_mem[i];
  1468. }
  1469. }
  1470. return true;
  1471. }
  1472. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  1473. if (node->get_script_instance()) {
  1474. #ifdef DEBUG_ENABLED
  1475. if (!node->get_script_instance()->has_method(VisualScriptLanguage::singleton->_step)) {
  1476. r_error_str=RTR("Custom node has no _step() method, can't process graph.");
  1477. r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
  1478. return 0;
  1479. }
  1480. #endif
  1481. Array in_values(true);
  1482. Array out_values(true);
  1483. Array work_mem(true);
  1484. in_values.resize(in_count);
  1485. for(int i=0;i<in_count;i++) {
  1486. in_values[i]=p_inputs[i];
  1487. }
  1488. out_values.resize(in_count);
  1489. work_mem.resize(work_mem_size);
  1490. for(int i=0;i<work_mem_size;i++) {
  1491. work_mem[i]=p_working_mem[i];
  1492. }
  1493. int ret_out;
  1494. Variant ret = node->get_script_instance()->call(VisualScriptLanguage::singleton->_step,in_values,out_values,p_start_mode,work_mem);
  1495. if (ret.get_type()==Variant::STRING) {
  1496. r_error_str=ret;
  1497. r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
  1498. return 0;
  1499. } else if (ret.is_num()) {
  1500. ret_out=ret;
  1501. } else {
  1502. r_error_str=RTR("Invalid return value from _step(), must be integer (seq out), or string (error).");
  1503. r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
  1504. return 0;
  1505. }
  1506. for(int i=0;i<out_count;i++) {
  1507. if (i<out_values.size()) {
  1508. *p_outputs[i]=out_values[i];
  1509. }
  1510. }
  1511. for(int i=0;i<work_mem_size;i++) {
  1512. if (i<work_mem.size()) {
  1513. p_working_mem[i]=work_mem[i];
  1514. }
  1515. }
  1516. return ret_out;
  1517. }
  1518. return 0;
  1519. }
  1520. };
  1521. VisualScriptNodeInstance* VisualScriptCustomNode::instance(VisualScriptInstance* p_instance) {
  1522. VisualScriptNodeInstanceCustomNode * instance = memnew(VisualScriptNodeInstanceCustomNode );
  1523. instance->instance=p_instance;
  1524. instance->in_count=get_input_value_port_count();
  1525. instance->out_count=get_output_value_port_count();
  1526. for(int i=0;i<instance->out_count;i++) {
  1527. bool unseq = get_script_instance() && get_script_instance()->has_method("_is_output_port_unsequenced") && bool(get_script_instance()->call("_is_output_port_unsequenced",i));
  1528. instance->out_unsequenced.push_back(unseq);
  1529. }
  1530. if (get_script_instance() && get_script_instance()->has_method("_get_working_memory_size")) {
  1531. instance->work_mem_size = get_script_instance()->call("_get_working_memory_size");
  1532. } else {
  1533. instance->work_mem_size=0;
  1534. }
  1535. return instance;
  1536. }
  1537. void VisualScriptCustomNode::_bind_methods() {
  1538. BIND_VMETHOD( MethodInfo(Variant::INT,"_get_output_sequence_port_count") );
  1539. BIND_VMETHOD( MethodInfo(Variant::BOOL,"_has_input_sequence_port") );
  1540. BIND_VMETHOD( MethodInfo(Variant::STRING,"_get_output_sequence_port_text",PropertyInfo(Variant::INT,"idx")) );
  1541. BIND_VMETHOD( MethodInfo(Variant::INT,"_get_input_value_port_count") );
  1542. BIND_VMETHOD( MethodInfo(Variant::INT,"_get_output_value_port_count") );
  1543. BIND_VMETHOD( MethodInfo(Variant::INT,"_get_input_value_port_type",PropertyInfo(Variant::INT,"idx")) );
  1544. BIND_VMETHOD( MethodInfo(Variant::STRING,"_get_input_value_port_name",PropertyInfo(Variant::INT,"idx")) );
  1545. BIND_VMETHOD( MethodInfo(Variant::INT,"_get_output_value_port_type",PropertyInfo(Variant::INT,"idx")) );
  1546. BIND_VMETHOD( MethodInfo(Variant::STRING,"_get_output_value_port_name",PropertyInfo(Variant::INT,"idx")) );
  1547. BIND_VMETHOD( MethodInfo(Variant::STRING,"_get_caption") );
  1548. BIND_VMETHOD( MethodInfo(Variant::STRING,"_get_text") );
  1549. BIND_VMETHOD( MethodInfo(Variant::STRING,"_get_category") );
  1550. BIND_VMETHOD( MethodInfo(Variant::INT,"_get_working_memory_size") );
  1551. BIND_VMETHOD( MethodInfo(Variant::INT,"_is_output_port_unsequenced",PropertyInfo(Variant::INT,"idx")) );
  1552. BIND_VMETHOD( MethodInfo(Variant::INT,"_get_output_port_unsequenced",PropertyInfo(Variant::INT,"idx"),PropertyInfo(Variant::ARRAY,"work_mem")) );
  1553. BIND_VMETHOD( MethodInfo(Variant::NIL,"_step:Variant",PropertyInfo(Variant::ARRAY,"inputs"),PropertyInfo(Variant::ARRAY,"outputs"),PropertyInfo(Variant::INT,"start_mode"),PropertyInfo(Variant::ARRAY,"working_mem")) );
  1554. BIND_CONSTANT( START_MODE_BEGIN_SEQUENCE );
  1555. BIND_CONSTANT( START_MODE_CONTINUE_SEQUENCE );
  1556. BIND_CONSTANT( START_MODE_RESUME_YIELD );
  1557. BIND_CONSTANT( STEP_PUSH_STACK_BIT );
  1558. BIND_CONSTANT( STEP_GO_BACK_BIT );
  1559. BIND_CONSTANT( STEP_NO_ADVANCE_BIT );
  1560. BIND_CONSTANT( STEP_EXIT_FUNCTION_BIT );
  1561. BIND_CONSTANT( STEP_YIELD_BIT );
  1562. }
  1563. VisualScriptCustomNode::VisualScriptCustomNode() {
  1564. }
  1565. //////////////////////////////////////////
  1566. ////////////////SUBCALL///////////
  1567. //////////////////////////////////////////
  1568. int VisualScriptSubCall::get_output_sequence_port_count() const {
  1569. return 1;
  1570. }
  1571. bool VisualScriptSubCall::has_input_sequence_port() const{
  1572. return true;
  1573. }
  1574. int VisualScriptSubCall::get_input_value_port_count() const{
  1575. Ref<Script> script = get_script();
  1576. if (script.is_valid() && script->has_method(VisualScriptLanguage::singleton->_subcall)) {
  1577. MethodInfo mi = script->get_method_info(VisualScriptLanguage::singleton->_subcall);
  1578. return mi.arguments.size();
  1579. }
  1580. return 0;
  1581. }
  1582. int VisualScriptSubCall::get_output_value_port_count() const{
  1583. return 1;
  1584. }
  1585. String VisualScriptSubCall::get_output_sequence_port_text(int p_port) const {
  1586. return String();
  1587. }
  1588. PropertyInfo VisualScriptSubCall::get_input_value_port_info(int p_idx) const{
  1589. Ref<Script> script = get_script();
  1590. if (script.is_valid() && script->has_method(VisualScriptLanguage::singleton->_subcall)) {
  1591. MethodInfo mi = script->get_method_info(VisualScriptLanguage::singleton->_subcall);
  1592. return mi.arguments[p_idx];
  1593. }
  1594. return PropertyInfo();
  1595. }
  1596. PropertyInfo VisualScriptSubCall::get_output_value_port_info(int p_idx) const{
  1597. Ref<Script> script = get_script();
  1598. if (script.is_valid() && script->has_method(VisualScriptLanguage::singleton->_subcall)) {
  1599. MethodInfo mi = script->get_method_info(VisualScriptLanguage::singleton->_subcall);
  1600. return mi.return_val;
  1601. }
  1602. return PropertyInfo();
  1603. }
  1604. String VisualScriptSubCall::get_caption() const {
  1605. return "SubCall";
  1606. }
  1607. String VisualScriptSubCall::get_text() const {
  1608. Ref<Script> script = get_script();
  1609. if (script.is_valid()) {
  1610. if (script->get_name()!=String())
  1611. return script->get_name();
  1612. if (script->get_path().is_resource_file())
  1613. return script->get_path().get_file();
  1614. return script->get_type();
  1615. }
  1616. return "";
  1617. }
  1618. String VisualScriptSubCall::get_category() const {
  1619. return "custom";
  1620. }
  1621. class VisualScriptNodeInstanceSubCall : public VisualScriptNodeInstance {
  1622. public:
  1623. VisualScriptInstance* instance;
  1624. VisualScriptSubCall *subcall;
  1625. int input_args;
  1626. bool valid;
  1627. //virtual int get_working_memory_size() const { return 0; }
  1628. //virtual bool is_output_port_unsequenced(int p_idx) const { return false; }
  1629. //virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return false; };
  1630. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  1631. if (!valid) {
  1632. r_error_str="Node requires a script with a _subcall(<args>) method to work.";
  1633. r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
  1634. return 0;
  1635. }
  1636. *p_outputs[0]=subcall->call(VisualScriptLanguage::singleton->_subcall,p_inputs,input_args,r_error_str);
  1637. return 0;
  1638. }
  1639. };
  1640. VisualScriptNodeInstance* VisualScriptSubCall::instance(VisualScriptInstance* p_instance) {
  1641. VisualScriptNodeInstanceSubCall * instance = memnew(VisualScriptNodeInstanceSubCall );
  1642. instance->instance=p_instance;
  1643. Ref<Script> script = get_script();
  1644. if (script.is_valid() && script->has_method(VisualScriptLanguage::singleton->_subcall)) {
  1645. instance->valid=true;
  1646. instance->input_args=get_input_value_port_count();
  1647. } else {
  1648. instance->valid=false;
  1649. }
  1650. return instance;
  1651. }
  1652. void VisualScriptSubCall::_bind_methods() {
  1653. BIND_VMETHOD( MethodInfo(Variant::NIL,"_subcall",PropertyInfo(Variant::NIL,"arguments:Variant")) );
  1654. }
  1655. VisualScriptSubCall::VisualScriptSubCall() {
  1656. }
  1657. //////////////////////////////////////////
  1658. ////////////////Comment///////////
  1659. //////////////////////////////////////////
  1660. int VisualScriptComment::get_output_sequence_port_count() const {
  1661. return 0;
  1662. }
  1663. bool VisualScriptComment::has_input_sequence_port() const{
  1664. return false;
  1665. }
  1666. int VisualScriptComment::get_input_value_port_count() const{
  1667. return 0;
  1668. }
  1669. int VisualScriptComment::get_output_value_port_count() const{
  1670. return 0;
  1671. }
  1672. String VisualScriptComment::get_output_sequence_port_text(int p_port) const {
  1673. return String();
  1674. }
  1675. PropertyInfo VisualScriptComment::get_input_value_port_info(int p_idx) const{
  1676. return PropertyInfo();
  1677. }
  1678. PropertyInfo VisualScriptComment::get_output_value_port_info(int p_idx) const{
  1679. return PropertyInfo();
  1680. }
  1681. String VisualScriptComment::get_caption() const {
  1682. return title;
  1683. }
  1684. String VisualScriptComment::get_text() const {
  1685. return description;
  1686. }
  1687. void VisualScriptComment::set_title(const String& p_title) {
  1688. if (title==p_title)
  1689. return;
  1690. title=p_title;
  1691. ports_changed_notify();
  1692. }
  1693. String VisualScriptComment::get_title() const{
  1694. return title;
  1695. }
  1696. void VisualScriptComment::set_description(const String& p_description){
  1697. if (description==p_description)
  1698. return;
  1699. description=p_description;
  1700. ports_changed_notify();
  1701. }
  1702. String VisualScriptComment::get_description() const{
  1703. return description;
  1704. }
  1705. void VisualScriptComment::set_size(const Size2& p_size){
  1706. if (size==p_size)
  1707. return;
  1708. size=p_size;
  1709. ports_changed_notify();
  1710. }
  1711. Size2 VisualScriptComment::get_size() const{
  1712. return size;
  1713. }
  1714. String VisualScriptComment::get_category() const {
  1715. return "data";
  1716. }
  1717. class VisualScriptNodeInstanceComment : public VisualScriptNodeInstance {
  1718. public:
  1719. VisualScriptInstance* instance;
  1720. //virtual int get_working_memory_size() const { return 0; }
  1721. //virtual bool is_output_port_unsequenced(int p_idx) const { return false; }
  1722. //virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return false; };
  1723. virtual int step(const Variant** p_inputs,Variant** p_outputs,StartMode p_start_mode,Variant* p_working_mem,Variant::CallError& r_error,String& r_error_str) {
  1724. return 0;
  1725. }
  1726. };
  1727. VisualScriptNodeInstance* VisualScriptComment::instance(VisualScriptInstance* p_instance) {
  1728. VisualScriptNodeInstanceComment * instance = memnew(VisualScriptNodeInstanceComment );
  1729. instance->instance=p_instance;
  1730. return instance;
  1731. }
  1732. void VisualScriptComment::_bind_methods() {
  1733. ObjectTypeDB::bind_method(_MD("set_title","title"),&VisualScriptComment::set_title);
  1734. ObjectTypeDB::bind_method(_MD("get_title"),&VisualScriptComment::get_title);
  1735. ObjectTypeDB::bind_method(_MD("set_description","description"),&VisualScriptComment::set_description);
  1736. ObjectTypeDB::bind_method(_MD("get_description"),&VisualScriptComment::get_description);
  1737. ObjectTypeDB::bind_method(_MD("set_size","size"),&VisualScriptComment::set_size);
  1738. ObjectTypeDB::bind_method(_MD("get_size"),&VisualScriptComment::get_size);
  1739. ADD_PROPERTY( PropertyInfo(Variant::STRING,"title"),_SCS("set_title"),_SCS("get_title"));
  1740. ADD_PROPERTY( PropertyInfo(Variant::STRING,"description",PROPERTY_HINT_MULTILINE_TEXT),_SCS("set_description"),_SCS("get_description"));
  1741. ADD_PROPERTY( PropertyInfo(Variant::VECTOR2,"size"),_SCS("set_size"),_SCS("get_size"));
  1742. }
  1743. VisualScriptComment::VisualScriptComment() {
  1744. title="Comment";
  1745. size=Size2(150,150);
  1746. }
  1747. void register_visual_script_nodes() {
  1748. VisualScriptLanguage::singleton->add_register_func("data/set_variable",create_node_generic<VisualScriptVariableGet>);
  1749. VisualScriptLanguage::singleton->add_register_func("data/get_variable",create_node_generic<VisualScriptVariableSet>);
  1750. VisualScriptLanguage::singleton->add_register_func("data/constant",create_node_generic<VisualScriptConstant>);
  1751. VisualScriptLanguage::singleton->add_register_func("data/global_constant",create_node_generic<VisualScriptGlobalConstant>);
  1752. VisualScriptLanguage::singleton->add_register_func("data/math_constant",create_node_generic<VisualScriptMathConstant>);
  1753. VisualScriptLanguage::singleton->add_register_func("data/engine_singleton",create_node_generic<VisualScriptEngineSingleton>);
  1754. VisualScriptLanguage::singleton->add_register_func("data/scene_node",create_node_generic<VisualScriptSceneNode>);
  1755. VisualScriptLanguage::singleton->add_register_func("data/scene_tree",create_node_generic<VisualScriptSceneTree>);
  1756. VisualScriptLanguage::singleton->add_register_func("data/resource_path",create_node_generic<VisualScriptResourcePath>);
  1757. VisualScriptLanguage::singleton->add_register_func("data/self",create_node_generic<VisualScriptSelf>);
  1758. VisualScriptLanguage::singleton->add_register_func("custom/custom_node",create_node_generic<VisualScriptCustomNode>);
  1759. VisualScriptLanguage::singleton->add_register_func("custom/sub_call",create_node_generic<VisualScriptSubCall>);
  1760. VisualScriptLanguage::singleton->add_register_func("data/comment",create_node_generic<VisualScriptComment>);
  1761. VisualScriptLanguage::singleton->add_register_func("index/get_index",create_node_generic<VisualScriptIndexGet>);
  1762. VisualScriptLanguage::singleton->add_register_func("index/set_index",create_node_generic<VisualScriptIndexSet>);
  1763. VisualScriptLanguage::singleton->add_register_func("operators/compare/equal",create_op_node<Variant::OP_EQUAL>);
  1764. VisualScriptLanguage::singleton->add_register_func("operators/compare/not_equal",create_op_node<Variant::OP_NOT_EQUAL>);
  1765. VisualScriptLanguage::singleton->add_register_func("operators/compare/less",create_op_node<Variant::OP_LESS>);
  1766. VisualScriptLanguage::singleton->add_register_func("operators/compare/less_equal",create_op_node<Variant::OP_LESS_EQUAL>);
  1767. VisualScriptLanguage::singleton->add_register_func("operators/compare/greater",create_op_node<Variant::OP_GREATER>);
  1768. VisualScriptLanguage::singleton->add_register_func("operators/compare/greater_equal",create_op_node<Variant::OP_GREATER_EQUAL>);
  1769. //mathematic
  1770. VisualScriptLanguage::singleton->add_register_func("operators/math/add",create_op_node<Variant::OP_ADD>);
  1771. VisualScriptLanguage::singleton->add_register_func("operators/math/subtract",create_op_node<Variant::OP_SUBSTRACT>);
  1772. VisualScriptLanguage::singleton->add_register_func("operators/math/multiply",create_op_node<Variant::OP_MULTIPLY>);
  1773. VisualScriptLanguage::singleton->add_register_func("operators/math/divide",create_op_node<Variant::OP_DIVIDE>);
  1774. VisualScriptLanguage::singleton->add_register_func("operators/math/negate",create_op_node<Variant::OP_NEGATE>);
  1775. VisualScriptLanguage::singleton->add_register_func("operators/math/remainder",create_op_node<Variant::OP_MODULE>);
  1776. VisualScriptLanguage::singleton->add_register_func("operators/math/string_concat",create_op_node<Variant::OP_STRING_CONCAT>);
  1777. //bitwise
  1778. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/shift_left",create_op_node<Variant::OP_SHIFT_LEFT>);
  1779. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/shift_right",create_op_node<Variant::OP_SHIFT_RIGHT>);
  1780. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_and",create_op_node<Variant::OP_BIT_AND>);
  1781. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_or",create_op_node<Variant::OP_BIT_OR>);
  1782. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_xor",create_op_node<Variant::OP_BIT_XOR>);
  1783. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_negate",create_op_node<Variant::OP_BIT_NEGATE>);
  1784. //logic
  1785. VisualScriptLanguage::singleton->add_register_func("operators/logic/and",create_op_node<Variant::OP_AND>);
  1786. VisualScriptLanguage::singleton->add_register_func("operators/logic/or",create_op_node<Variant::OP_OR>);
  1787. VisualScriptLanguage::singleton->add_register_func("operators/logic/xor",create_op_node<Variant::OP_XOR>);
  1788. VisualScriptLanguage::singleton->add_register_func("operators/logic/not",create_op_node<Variant::OP_NOT>);
  1789. VisualScriptLanguage::singleton->add_register_func("operators/logic/in",create_op_node<Variant::OP_IN>);
  1790. }