visual_script_nodes.cpp 63 KB

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