godot/modules/visual_script/visual_script_yield_nodes.cpp

#include "visual_script_yield_nodes.h"
#include "scene/main/scene_main_loop.h"
#include "os/os.h"
#include "scene/main/node.h"
#include "visual_script_nodes.h"

//////////////////////////////////////////
////////////////YIELD///////////
//////////////////////////////////////////

int VisualScriptYield::get_output_sequence_port_count() const {

	return 1;
}

bool VisualScriptYield::has_input_sequence_port() const{

	return true;
}

int VisualScriptYield::get_input_value_port_count() const{

	return 0;
}
int VisualScriptYield::get_output_value_port_count() const{

	return 0;
}

String VisualScriptYield::get_output_sequence_port_text(int p_port) const {

	return String();
}

PropertyInfo VisualScriptYield::get_input_value_port_info(int p_idx) const{

	return PropertyInfo();
}

PropertyInfo VisualScriptYield::get_output_value_port_info(int p_idx) const{

	return PropertyInfo();
}


String VisualScriptYield::get_caption() const {

	return yield_mode==YIELD_RETURN?"Yield":"Wait";
}

String VisualScriptYield::get_text() const {

	switch (yield_mode) {
		case YIELD_RETURN: return ""; break;
		case YIELD_FRAME: return "Next Frame"; break;
		case YIELD_FIXED_FRAME:  return "Next Fixed Frame"; break;
		case YIELD_WAIT:  return rtos(wait_time)+" sec(s)"; break;
	}

	return String();
}


class VisualScriptNodeInstanceYield : public VisualScriptNodeInstance {
public:

	VisualScriptYield::YieldMode mode;
	float wait_time;

	virtual int get_working_memory_size() const { return 1; } //yield needs at least 1
	//virtual bool is_output_port_unsequenced(int p_idx) const { return false; }
	//virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return false; }

	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) {

		if (p_start_mode==START_MODE_RESUME_YIELD) {
			return 0; //resuming yield
		} else {
			//yield


			SceneTree *tree = OS::get_singleton()->get_main_loop()->cast_to<SceneTree>();
			if (!tree) {
				r_error_str="Main Loop is not SceneTree";
				r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
				return 0;
			}

			Ref<VisualScriptFunctionState> state;
			state.instance();

			int ret = STEP_YIELD_BIT;
			switch(mode) {

				case VisualScriptYield::YIELD_RETURN:  ret=STEP_EXIT_FUNCTION_BIT; break; //return the yield
				case VisualScriptYield::YIELD_FRAME: state->connect_to_signal(tree,"idle_frame",Array()); break;
				case VisualScriptYield::YIELD_FIXED_FRAME:  state->connect_to_signal(tree,"fixed_frame",Array()); break;
				case VisualScriptYield::YIELD_WAIT:  state->connect_to_signal(tree->create_timer(wait_time).ptr(),"timeout",Array()); break;

			}

			*p_working_mem=state;

			return ret;
		}
	}

};

VisualScriptNodeInstance* VisualScriptYield::instance(VisualScriptInstance* p_instance) {

	VisualScriptNodeInstanceYield * instance = memnew(VisualScriptNodeInstanceYield );
	//instance->instance=p_instance;
	instance->mode=yield_mode;
	instance->wait_time=wait_time;
	return instance;
}

void VisualScriptYield::set_yield_mode(YieldMode p_mode) {

	if (yield_mode==p_mode)
		return;
	yield_mode=p_mode;
	ports_changed_notify();
	_change_notify();
}

VisualScriptYield::YieldMode VisualScriptYield::get_yield_mode(){

	return yield_mode;
}

void VisualScriptYield::set_wait_time(float p_time) {

	if (wait_time==p_time)
		return;
	wait_time=p_time;
	ports_changed_notify();

}

float VisualScriptYield::get_wait_time(){

	return wait_time;
}


void VisualScriptYield::_validate_property(PropertyInfo& property) const {


	if (property.name=="wait_time") {
		if (yield_mode!=YIELD_WAIT) {
			property.usage=0;
		}
	}
}

void VisualScriptYield::_bind_methods() {

	ClassDB::bind_method(D_METHOD("set_yield_mode","mode"),&VisualScriptYield::set_yield_mode);
	ClassDB::bind_method(D_METHOD("get_yield_mode"),&VisualScriptYield::get_yield_mode);

	ClassDB::bind_method(D_METHOD("set_wait_time","sec"),&VisualScriptYield::set_wait_time);
	ClassDB::bind_method(D_METHOD("get_wait_time"),&VisualScriptYield::get_wait_time);

	ADD_PROPERTY(PropertyInfo(Variant::INT,"mode",PROPERTY_HINT_ENUM,"Frame,FixedFrame,Time",PROPERTY_USAGE_NOEDITOR),"set_yield_mode","get_yield_mode");
	ADD_PROPERTY(PropertyInfo(Variant::REAL,"wait_time"),"set_wait_time","get_wait_time");


	BIND_CONSTANT( YIELD_FRAME );
	BIND_CONSTANT( YIELD_FIXED_FRAME );
	BIND_CONSTANT( YIELD_WAIT );

}

VisualScriptYield::VisualScriptYield() {

	yield_mode=YIELD_FRAME;
	wait_time=1;

}


template<VisualScriptYield::YieldMode MODE>
static Ref<VisualScriptNode> create_yield_node(const String& p_name) {

	Ref<VisualScriptYield> node;
	node.instance();
	node->set_yield_mode(MODE);
	return node;
}

///////////////////////////////////////////////////
////////////////YIELD SIGNAL//////////////////////
//////////////////////////////////////////////////

int VisualScriptYieldSignal::get_output_sequence_port_count() const {

	return 1;
}

bool VisualScriptYieldSignal::has_input_sequence_port() const{

	return true;
}
#ifdef TOOLS_ENABLED

static Node* _find_script_node(Node* p_edited_scene,Node* p_current_node,const Ref<Script> &script) {

	if (p_edited_scene!=p_current_node && p_current_node->get_owner()!=p_edited_scene)
		return NULL;

	Ref<Script> scr = p_current_node->get_script();

	if (scr.is_valid() && scr==script)
		return p_current_node;

	for(int i=0;i<p_current_node->get_child_count();i++) {
		Node *n = _find_script_node(p_edited_scene,p_current_node->get_child(i),script);
		if (n)
			return n;
	}

	return NULL;
}

#endif
Node *VisualScriptYieldSignal::_get_base_node() const {

#ifdef TOOLS_ENABLED
	Ref<Script> script = get_visual_script();
	if (!script.is_valid())
		return NULL;

	MainLoop * main_loop = OS::get_singleton()->get_main_loop();
	if (!main_loop)
		return NULL;

	SceneTree *scene_tree = main_loop->cast_to<SceneTree>();

	if (!scene_tree)
		return NULL;

	Node *edited_scene = scene_tree->get_edited_scene_root();

	if (!edited_scene)
		return NULL;

	Node* script_node = _find_script_node(edited_scene,edited_scene,script);

	if (!script_node)
		return NULL;

	if (!script_node->has_node(base_path))
		return NULL;

	Node *path_to = script_node->get_node(base_path);

	return path_to;
#else

	return NULL;
#endif
}

StringName VisualScriptYieldSignal::_get_base_type() const {

	if (call_mode==CALL_MODE_SELF && get_visual_script().is_valid())
		return get_visual_script()->get_instance_base_type();
	else if (call_mode==CALL_MODE_NODE_PATH && get_visual_script().is_valid()) {
		Node *path = _get_base_node();
		if (path)
			return path->get_class();

	}

	return base_type;
}

int VisualScriptYieldSignal::get_input_value_port_count() const{

	if (call_mode==CALL_MODE_INSTANCE)
		return 1;
	else
		return 0;

}
int VisualScriptYieldSignal::get_output_value_port_count() const{


	MethodInfo sr;

	if (!ClassDB::get_signal(_get_base_type(),signal,&sr))
		return 0;

	return sr.arguments.size();

}

String VisualScriptYieldSignal::get_output_sequence_port_text(int p_port) const {

	return String();
}

PropertyInfo VisualScriptYieldSignal::get_input_value_port_info(int p_idx) const{

	if (call_mode==CALL_MODE_INSTANCE)
		return PropertyInfo(Variant::OBJECT,"instance");
	else
		return PropertyInfo();

}

PropertyInfo VisualScriptYieldSignal::get_output_value_port_info(int p_idx) const{

	MethodInfo sr;

	if (!ClassDB::get_signal(_get_base_type(),signal,&sr))
		return PropertyInfo(); //no signal
	ERR_FAIL_INDEX_V(p_idx,sr.arguments.size(),PropertyInfo());
	return sr.arguments[p_idx];

}


String VisualScriptYieldSignal::get_caption() const {

	static const char*cname[3]= {
		"WaitSignal",
		"WaitNodeSignal",
		"WaitInstanceSigna;",
	};

	return cname[call_mode];
}

String VisualScriptYieldSignal::get_text() const {

	if (call_mode==CALL_MODE_SELF)
		return "  "+String(signal)+"()";
	else
		return "  "+_get_base_type()+"."+String(signal)+"()";

}


void VisualScriptYieldSignal::set_base_type(const StringName& p_type) {

	if (base_type==p_type)
		return;

	base_type=p_type;

	_change_notify();
	ports_changed_notify();
}

StringName VisualScriptYieldSignal::get_base_type() const{

	return base_type;
}

void VisualScriptYieldSignal::set_signal(const StringName& p_type){

	if (signal==p_type)
		return;

	signal=p_type;

	_change_notify();
	ports_changed_notify();
}
StringName VisualScriptYieldSignal::get_signal() const {


	return signal;
}

void VisualScriptYieldSignal::set_base_path(const NodePath& p_type) {

	if (base_path==p_type)
		return;

	base_path=p_type;

	_change_notify();
	ports_changed_notify();
}

NodePath VisualScriptYieldSignal::get_base_path() const {

	return base_path;
}


void VisualScriptYieldSignal::set_call_mode(CallMode p_mode) {

	if (call_mode==p_mode)
		return;

	call_mode=p_mode;

	_change_notify();
	ports_changed_notify();

}

VisualScriptYieldSignal::CallMode VisualScriptYieldSignal::get_call_mode() const {

	return call_mode;
}


void VisualScriptYieldSignal::_validate_property(PropertyInfo& property) const {

	if (property.name=="signal/base_type") {
		if (call_mode!=CALL_MODE_INSTANCE) {
			property.usage=PROPERTY_USAGE_NOEDITOR;
		}
	}


	if (property.name=="signal/node_path") {
		if (call_mode!=CALL_MODE_NODE_PATH) {
			property.usage=0;
		} else {

			Node *bnode = _get_base_node();
			if (bnode) {
				property.hint_string=bnode->get_path(); //convert to loong string
			} else {

			}
		}
	}

	if (property.name=="signal/signal") {
		property.hint=PROPERTY_HINT_ENUM;


		List<MethodInfo> methods;

		ClassDB::get_signal_list(_get_base_type(),&methods);

		List<String> mstring;
		for (List<MethodInfo>::Element *E=methods.front();E;E=E->next()) {
			if (E->get().name.begins_with("_"))
				continue;
			mstring.push_back(E->get().name.get_slice(":",0));
		}

		mstring.sort();

		String ml;
		for (List<String>::Element *E=mstring.front();E;E=E->next()) {

			if (ml!=String())
				ml+=",";
			ml+=E->get();
		}

		property.hint_string=ml;
	}


}


void VisualScriptYieldSignal::_bind_methods() {

	ClassDB::bind_method(D_METHOD("set_base_type","base_type"),&VisualScriptYieldSignal::set_base_type);
	ClassDB::bind_method(D_METHOD("get_base_type"),&VisualScriptYieldSignal::get_base_type);

	ClassDB::bind_method(D_METHOD("set_signal","signal"),&VisualScriptYieldSignal::set_signal);
	ClassDB::bind_method(D_METHOD("get_signal"),&VisualScriptYieldSignal::get_signal);

	ClassDB::bind_method(D_METHOD("set_call_mode","mode"),&VisualScriptYieldSignal::set_call_mode);
	ClassDB::bind_method(D_METHOD("get_call_mode"),&VisualScriptYieldSignal::get_call_mode);

	ClassDB::bind_method(D_METHOD("set_base_path","base_path"),&VisualScriptYieldSignal::set_base_path);
	ClassDB::bind_method(D_METHOD("get_base_path"),&VisualScriptYieldSignal::get_base_path);



	String bt;
	for(int i=0;i<Variant::VARIANT_MAX;i++) {
		if (i>0)
			bt+=",";

		bt+=Variant::get_type_name(Variant::Type(i));
	}

	ADD_PROPERTY(PropertyInfo(Variant::INT,"signal/call_mode",PROPERTY_HINT_ENUM,"Self,Node Path,Instance"),"set_call_mode","get_call_mode");
	ADD_PROPERTY(PropertyInfo(Variant::STRING,"signal/base_type",PROPERTY_HINT_TYPE_STRING,"Object"),"set_base_type","get_base_type");
	ADD_PROPERTY(PropertyInfo(Variant::NODE_PATH,"signal/node_path",PROPERTY_HINT_NODE_PATH_TO_EDITED_NODE),"set_base_path","get_base_path");
	ADD_PROPERTY(PropertyInfo(Variant::STRING,"signal/signal"),"set_signal","get_signal");


	BIND_CONSTANT( CALL_MODE_SELF );
	BIND_CONSTANT( CALL_MODE_NODE_PATH);
	BIND_CONSTANT( CALL_MODE_INSTANCE);

}

class VisualScriptNodeInstanceYieldSignal : public VisualScriptNodeInstance {
public:


	VisualScriptYieldSignal::CallMode call_mode;
	NodePath node_path;
	int output_args;
	StringName signal;

	VisualScriptYieldSignal *node;
	VisualScriptInstance *instance;



	virtual int get_working_memory_size() const { return 1; }
	//virtual bool is_output_port_unsequenced(int p_idx) const { return false; }
	//virtual bool get_output_port_unsequenced(int p_idx,Variant* r_value,Variant* p_working_mem,String &r_error) const { return true; }

	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) {

		if (p_start_mode==START_MODE_RESUME_YIELD) {
			return 0; //resuming yield
		} else {
			//yield

			Object * object;

			switch(call_mode) {

				case VisualScriptYieldSignal::CALL_MODE_SELF: {

					object=instance->get_owner_ptr();

				} break;
				case VisualScriptYieldSignal::CALL_MODE_NODE_PATH: {

					Node* node = instance->get_owner_ptr()->cast_to<Node>();
					if (!node) {
						r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
						r_error_str="Base object is not a Node!";
						return 0;
					}

					Node* another = node->get_node(node_path);
					if (!node) {
						r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
						r_error_str="Path does not lead Node!";
						return 0;
					}

					object=another;

				} break;
				case VisualScriptYieldSignal::CALL_MODE_INSTANCE: {

					object = *p_inputs[0];
					if (!object) {
						r_error.error=Variant::CallError::CALL_ERROR_INVALID_METHOD;
						r_error_str="Supplied instance input is null.";
						return 0;

					}

				} break;

			}

			Ref<VisualScriptFunctionState> state;
			state.instance();

			state->connect_to_signal(object,signal,Array());

			*p_working_mem=state;

			return STEP_YIELD_BIT;
		}


	}


};

VisualScriptNodeInstance* VisualScriptYieldSignal::instance(VisualScriptInstance* p_instance) {

	VisualScriptNodeInstanceYieldSignal * instance = memnew(VisualScriptNodeInstanceYieldSignal );
	instance->node=this;
	instance->instance=p_instance;
	instance->signal=signal;
	instance->call_mode=call_mode;
	instance->node_path=base_path;
	instance->output_args = get_output_value_port_count();
	return instance;
}
VisualScriptYieldSignal::VisualScriptYieldSignal() {

	call_mode=CALL_MODE_SELF;
	base_type="Object";

}

template<VisualScriptYieldSignal::CallMode cmode>
static Ref<VisualScriptNode> create_yield_signal_node(const String& p_name) {

	Ref<VisualScriptYieldSignal> node;
	node.instance();
	node->set_call_mode(cmode);
	return node;
}

void register_visual_script_yield_nodes() {

	VisualScriptLanguage::singleton->add_register_func("functions/wait/wait_frame",create_yield_node<VisualScriptYield::YIELD_FRAME>);
	VisualScriptLanguage::singleton->add_register_func("functions/wait/wait_fixed_frame",create_yield_node<VisualScriptYield::YIELD_FIXED_FRAME>);
	VisualScriptLanguage::singleton->add_register_func("functions/wait/wait_time",create_yield_node<VisualScriptYield::YIELD_WAIT>);


	VisualScriptLanguage::singleton->add_register_func("functions/yield",create_yield_node<VisualScriptYield::YIELD_RETURN>);
	VisualScriptLanguage::singleton->add_register_func("functions/yield_signal",create_node_generic<VisualScriptYieldSignal>);

}