godot/servers/visual/shader_graph.cpp

/*************************************************************************/
/*  shader_graph.cpp                                                     */
/*************************************************************************/
/*                       This file is part of:                           */
/*                           GODOT ENGINE                                */
/*                    http://www.godotengine.org                         */
/*************************************************************************/
/* Copyright (c) 2007-2015 Juan Linietsky, Ariel Manzur.                 */
/*                                                                       */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the       */
/* "Software"), to deal in the Software without restriction, including   */
/* without limitation the rights to use, copy, modify, merge, publish,   */
/* distribute, sublicense, and/or sell copies of the Software, and to    */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions:                                             */
/*                                                                       */
/* The above copyright notice and this permission notice shall be        */
/* included in all copies or substantial portions of the Software.       */
/*                                                                       */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,       */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF    */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
/*************************************************************************/
#include "shader_graph.h"

#if 0


struct _ConnectionKey {

	int node;
	int slot;	
		
	_FORCE_INLINE_ _ConnectionKey(int p_node=0,int p_slot=0) { node=p_node; slot=p_slot; }
	
	_FORCE_INLINE_ bool operator<(const _ConnectionKey& p_other) const {
	
		if (node<p_other.node)
			return true;
		else if (node>p_other.node)
			return false;			
		else
			return slot<p_other.slot;
	}
};

Error ShaderGraph::generate(ShaderCodeGenerator * p_generator) const {
	
	Map<int,Node>::Element *E = node_map.front();
	int i=0;	
	while(E) {
	
		E->get().order=i++;
		E->get().out_valid=false;
		E->get().in_valid=false;
		E=E->next();
	}	

	int worst_case=connections.size() * connections.size(); // worst bubble case
	int iterations=0;
	int swaps;
	
	do {	
		swaps=0;
		const List<Connection>::Element *E=connections.front();

		while(E) {

			const Connection &c = E->get();
			
			const Node *src = &node_map[c.src_id];
			const Node *dst = &node_map[c.dst_id];
						
			if (src->order > dst->order) {
			
				SWAP(src->order, dst->order);
				swaps++;
			}
			
			E=E->next();
		}
		
	
		iterations++;
		
	} while (iterations<=worst_case && swaps>0);
	
	ERR_FAIL_COND_V( swaps != 0 , ERR_CYCLIC_LINK );
	
	//node array
	Vector<const Node*> nodes;
	nodes.resize(node_map.size());
	
	E = node_map.front();
	while(E) {
	
		ERR_FAIL_INDEX_V( E->get().order, nodes.size(), ERR_BUG);
		nodes[E->get().order]=&E->get();
		E=E->next();
	}
	
	//connection set
	
	Map<_ConnectionKey,int> in_connection_map;
	Map<_ConnectionKey,List<int> > out_connection_map;
	Map<_ConnectionKey,int> in_node_map;
	Map<_ConnectionKey,List<int> > out_node_map;

	const List<Connection>::Element *CE=connections.front();
	i=0;
	while(CE) {
		const Connection &c = CE->get();
		
		_ConnectionKey in_k;
		in_k.node=node_map[c.dst_id].order;
		in_k.slot=c.dst_slot;
		in_connection_map[in_k]=i;
		in_node_map[in_k]=node_map[c.src_id].order;

		_ConnectionKey out_k;
		out_k.node=node_map[c.src_id].order;
		out_k.slot=c.src_slot;
		if (!out_connection_map.has(out_k))
			out_connection_map[out_k]=List<int>();
		out_connection_map[out_k].push_back(i);
		if(!out_node_map.has(out_k))
			out_node_map[out_k]=List<int>();
		out_node_map[out_k].push_back(node_map[c.dst_id].order);
		
		i++;
		CE=CE->next();
	}
	
	// validate nodes if they are connected to an output

	for(int i=nodes.size()-1;i>=0;i--) {

		if (VisualServer::shader_get_output_count(nodes[i]->type)==0) {
			// an actual graph output

			_ConnectionKey in_k;
			in_k.node=nodes[i]->order;
			in_k.slot=0;

			if (in_node_map.has(in_k)) {
				nodes[i]->out_valid=true;
			}
		} else {
			// regular node

			bool valid=false;
			for(int j=0;j<VS::shader_get_output_count(nodes[i]->type);j++) {

				_ConnectionKey key(nodes[i]->order,j);

				if (out_node_map.has(key)) {
					for(List<int>::Element *CE=out_node_map[key].front();CE;CE=CE->next()) {

						int to_node=CE->get();
						ERR_CONTINUE(to_node<0 || to_node >=nodes.size());
						if (nodes[to_node]->out_valid) {
							valid=true;
							break;
						}


					}
				}
				if (valid)
					break;

			}

			nodes[i]->out_valid=valid;
		}
	}

	// validate nodes if they are connected to an input

	for(int i=0;i<nodes.size();i++) {

		if (VisualServer::shader_get_input_count(nodes[i]->type)==0) {
			// an actual graph input

			int out_count=VisualServer::shader_get_output_count(nodes[i]->type);


			for(int j=0;j<out_count;j++) {

				_ConnectionKey out_k;
				out_k.node=nodes[i]->order;
				out_k.slot=j;
				if (out_node_map.has(out_k)) {
					nodes[i]->in_valid=true;
					break;
				}
			}

		} else {
			// regular node
			// this is very important.. for a node to be valid, all its inputs need to be valid
			bool valid=true;
			for(int j=0;j<VS::shader_get_input_count(nodes[i]->type);j++) {


				bool in_valid=false;
				_ConnectionKey key(nodes[i]->order,j);
				if (in_node_map.has(key)) {

					int from_node=in_node_map[key];
					ERR_CONTINUE(from_node<0 || from_node>=nodes.size());
					if (nodes[from_node]->in_valid)
						in_valid=true;

				}

				if (!in_valid) {
					valid=false;
					break;
				}

			}

			nodes[i]->in_valid=valid;
		}
	}

	// write code
		
	p_generator->begin();
		
	for(int i=0;i<nodes.size();i++) {


		if (!nodes[i]->out_valid || !nodes[i]->in_valid) // valid in both ways
			continue; // skip node

		Vector<int> in_indices;
		in_indices.resize(VS::shader_get_input_count(nodes[i]->type));
		Vector<int> out_indices;
		Vector<int> out_slot_indices;
		
		for(int j=0;j<in_indices.size();j++) {
		
			_ConnectionKey key(nodes[i]->order,j);
			if (in_connection_map.has(key))
				in_indices[j]=in_connection_map[key];
			else
				in_indices[j]=-1;			
		}
		
		for(int j=0;j<VS::shader_get_output_count(nodes[i]->type);j++) {
		
			_ConnectionKey key(nodes[i]->order,j);
			if (out_connection_map.has(key)) {
				for(List<int>::Element *CE=out_connection_map[key].front();CE;CE=CE->next()) {
				
					out_indices.push_back(CE->get());
					out_slot_indices.push_back(j);
				}
			}
		}
		
		Error err = p_generator->add_node(nodes[i]->type,i,nodes[i]->id,nodes[i]->param,in_indices,out_indices,out_slot_indices);
		ERR_FAIL_COND_V( err, err );
	}
		
	p_generator->end();
		
	
	return OK;
}

void ShaderGraph::node_add(VS::ShaderNodeType p_type,int p_id) {


	ERR_FAIL_COND( node_map.has(p_id ) );
	ERR_FAIL_INDEX( p_type, VS::NODE_TYPE_MAX );
	Node node;
	
	node.type=p_type;
	node.id=p_id;
	node.x=0;
	node.y=0;

	node_map[p_id]=node;

}

void ShaderGraph::node_set_pos(int p_id, int p_x,int p_y) {

	ERR_FAIL_COND(!node_map.has(p_id));
	node_map[p_id].x=p_x;
	node_map[p_id].y=p_y;
}
int ShaderGraph::node_get_pos_x(int p_id) const {

	ERR_FAIL_COND_V(!node_map.has(p_id),-1);
	return node_map[p_id].x;
}
int ShaderGraph::node_get_pos_y(int p_id) const {

	ERR_FAIL_COND_V(!node_map.has(p_id),-1);
	return node_map[p_id].y;
}

void ShaderGraph::node_remove(int p_id) {

	ERR_FAIL_COND(!node_map.has(p_id));
	
	//erase connections associated with node
	List<Connection>::Element *N,*E=connections.front();
	while(E) {
		N=E->next();
		const Connection &c = E->get();
		if (c.src_id==p_id || c.dst_id==p_id) {
		
			connections.erase(E);
		}
		E=N;
	}

	node_map.erase(p_id);
}

void ShaderGraph::node_change_type(int p_id, VS::ShaderNodeType p_type) {

	ERR_FAIL_COND(!node_map.has(p_id));
	node_map[p_id].type=p_type;
	node_map[p_id].param=Variant();

}

void ShaderGraph::node_set_param(int p_id, const Variant& p_value) {
	
	ERR_FAIL_COND(!node_map.has(p_id));
	node_map[p_id].param=p_value;	
}

void ShaderGraph::get_node_list(List<int> *p_node_list) const {

	Map<int,Node>::Element *E = node_map.front();
	
	while(E) {
	
		p_node_list->push_back(E->key());
		E=E->next();
	}
}


VS::ShaderNodeType ShaderGraph::node_get_type(int p_id) const {

	ERR_FAIL_COND_V(!node_map.has(p_id),VS::NODE_TYPE_MAX);
	return node_map[p_id].type;
}

Variant ShaderGraph::node_get_param(int p_id) const {

	ERR_FAIL_COND_V(!node_map.has(p_id),Variant());
	return node_map[p_id].param;	
}


Error ShaderGraph::connect(int p_src_id,int p_src_slot, int p_dst_id,int p_dst_slot) {

	ERR_FAIL_COND_V(p_src_id==p_dst_id, ERR_INVALID_PARAMETER);
	ERR_FAIL_COND_V(!node_map.has(p_src_id), ERR_INVALID_PARAMETER);
	ERR_FAIL_COND_V(!node_map.has(p_dst_id), ERR_INVALID_PARAMETER);
	VisualServer::ShaderNodeType type_src=node_map[p_src_id].type;
	VisualServer::ShaderNodeType type_dst=node_map[p_dst_id].type;
	ERR_FAIL_INDEX_V( p_src_slot, VisualServer::shader_get_output_count(type_src), ERR_INVALID_PARAMETER );
	ERR_FAIL_INDEX_V( p_dst_slot, VisualServer::shader_get_input_count(type_dst), ERR_INVALID_PARAMETER );
	ERR_FAIL_COND_V(VisualServer::shader_is_output_vector(type_src,p_src_slot) != VisualServer::shader_is_input_vector(type_dst,p_dst_slot), ERR_INVALID_PARAMETER );


	List<Connection>::Element *E=connections.front();
	while(E) {
		const Connection &c = E->get();
		ERR_FAIL_COND_V(c.dst_slot==p_dst_slot && c.dst_id == p_dst_id, ERR_ALREADY_EXISTS);
		
		E=E->next();
	}
	
	Connection c;
	c.src_slot=p_src_slot;
	c.src_id=p_src_id;
	c.dst_slot=p_dst_slot;
	c.dst_id=p_dst_id;
		
	connections.push_back(c);

	return OK;
}

bool ShaderGraph::is_connected(int p_src_id,int p_src_slot, int p_dst_id,int p_dst_slot) const {

	const List<Connection>::Element *E=connections.front();
	while(E) {
		const Connection &c = E->get();
		if (c.dst_slot==p_dst_slot && c.dst_id == p_dst_id && c.src_slot==p_src_slot && c.src_id == p_src_id)
			return true;

		E=E->next();
	}

	return false;
}

void ShaderGraph::disconnect(int p_src_id,int p_src_slot, int p_dst_id,int p_dst_slot) {

	List<Connection>::Element *N,*E=connections.front();
	while(E) {
		N=E->next();
		const Connection &c = E->get();
		if (c.src_slot==p_src_slot && c.src_id==p_src_id && c.dst_slot==p_dst_slot && c.dst_id == p_dst_id) {
		
			connections.erase(E);
		}
		E=N;
	}
	

}


void ShaderGraph::clear() {

	connections.clear();
	node_map.clear();
}

List<ShaderGraph::Connection> ShaderGraph::get_connection_list() const {

	return connections;

}

ShaderGraph::ShaderGraph() {


}


ShaderGraph::~ShaderGraph() {

}


#endif