visual_script_nodes.cpp 59 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348
  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. void register_visual_script_nodes() {
  1550. VisualScriptLanguage::singleton->add_register_func("data/set_variable",create_node_generic<VisualScriptVariableGet>);
  1551. VisualScriptLanguage::singleton->add_register_func("data/get_variable",create_node_generic<VisualScriptVariableSet>);
  1552. VisualScriptLanguage::singleton->add_register_func("data/constant",create_node_generic<VisualScriptConstant>);
  1553. VisualScriptLanguage::singleton->add_register_func("data/global_constant",create_node_generic<VisualScriptGlobalConstant>);
  1554. VisualScriptLanguage::singleton->add_register_func("data/math_constant",create_node_generic<VisualScriptMathConstant>);
  1555. VisualScriptLanguage::singleton->add_register_func("data/engine_singleton",create_node_generic<VisualScriptEngineSingleton>);
  1556. VisualScriptLanguage::singleton->add_register_func("data/scene_node",create_node_generic<VisualScriptSceneNode>);
  1557. VisualScriptLanguage::singleton->add_register_func("data/scene_tree",create_node_generic<VisualScriptSceneTree>);
  1558. VisualScriptLanguage::singleton->add_register_func("data/resource_path",create_node_generic<VisualScriptResourcePath>);
  1559. VisualScriptLanguage::singleton->add_register_func("data/self",create_node_generic<VisualScriptSelf>);
  1560. VisualScriptLanguage::singleton->add_register_func("custom/custom_node",create_node_generic<VisualScriptCustomNode>);
  1561. VisualScriptLanguage::singleton->add_register_func("index/get_index",create_node_generic<VisualScriptIndexGet>);
  1562. VisualScriptLanguage::singleton->add_register_func("index/set_index",create_node_generic<VisualScriptIndexSet>);
  1563. VisualScriptLanguage::singleton->add_register_func("operators/compare/equal",create_op_node<Variant::OP_EQUAL>);
  1564. VisualScriptLanguage::singleton->add_register_func("operators/compare/not_equal",create_op_node<Variant::OP_NOT_EQUAL>);
  1565. VisualScriptLanguage::singleton->add_register_func("operators/compare/less",create_op_node<Variant::OP_LESS>);
  1566. VisualScriptLanguage::singleton->add_register_func("operators/compare/less_equal",create_op_node<Variant::OP_LESS_EQUAL>);
  1567. VisualScriptLanguage::singleton->add_register_func("operators/compare/greater",create_op_node<Variant::OP_GREATER>);
  1568. VisualScriptLanguage::singleton->add_register_func("operators/compare/greater_equal",create_op_node<Variant::OP_GREATER_EQUAL>);
  1569. //mathematic
  1570. VisualScriptLanguage::singleton->add_register_func("operators/math/add",create_op_node<Variant::OP_ADD>);
  1571. VisualScriptLanguage::singleton->add_register_func("operators/math/subtract",create_op_node<Variant::OP_SUBSTRACT>);
  1572. VisualScriptLanguage::singleton->add_register_func("operators/math/multiply",create_op_node<Variant::OP_MULTIPLY>);
  1573. VisualScriptLanguage::singleton->add_register_func("operators/math/divide",create_op_node<Variant::OP_DIVIDE>);
  1574. VisualScriptLanguage::singleton->add_register_func("operators/math/negate",create_op_node<Variant::OP_NEGATE>);
  1575. VisualScriptLanguage::singleton->add_register_func("operators/math/remainder",create_op_node<Variant::OP_MODULE>);
  1576. VisualScriptLanguage::singleton->add_register_func("operators/math/string_concat",create_op_node<Variant::OP_STRING_CONCAT>);
  1577. //bitwise
  1578. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/shift_left",create_op_node<Variant::OP_SHIFT_LEFT>);
  1579. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/shift_right",create_op_node<Variant::OP_SHIFT_RIGHT>);
  1580. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_and",create_op_node<Variant::OP_BIT_AND>);
  1581. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_or",create_op_node<Variant::OP_BIT_OR>);
  1582. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_xor",create_op_node<Variant::OP_BIT_XOR>);
  1583. VisualScriptLanguage::singleton->add_register_func("operators/bitwise/bit_negate",create_op_node<Variant::OP_BIT_NEGATE>);
  1584. //logic
  1585. VisualScriptLanguage::singleton->add_register_func("operators/logic/and",create_op_node<Variant::OP_AND>);
  1586. VisualScriptLanguage::singleton->add_register_func("operators/logic/or",create_op_node<Variant::OP_OR>);
  1587. VisualScriptLanguage::singleton->add_register_func("operators/logic/xor",create_op_node<Variant::OP_XOR>);
  1588. VisualScriptLanguage::singleton->add_register_func("operators/logic/not",create_op_node<Variant::OP_NOT>);
  1589. VisualScriptLanguage::singleton->add_register_func("operators/logic/in",create_op_node<Variant::OP_IN>);
  1590. }