#include "rotor.h"

using namespace Rotor;
const string Graph::toString(){
	string xmlgraph;
	if (loaded) {
		xml.copyXmlToString(xmlgraph);
		return xmlgraph;
	}
	else return "";
}
vector<Node*> Graph::find_nodes(const string &type){
	vector<Node*> found;
	for (std::unordered_map<string,Node*>::iterator it=nodes.begin();it!=nodes.end();++it) {
		if (it->second->type==type) found.push_back(it->second);
	}
	return found;
};
Node* Graph::find_node(const string &type){
	for (std::unordered_map<string,Node*>::iterator it=nodes.begin();it!=nodes.end();++it) {
		if (it->second->type==type) return it->second;
	}
	return nullptr; //can be tested against
};
/*
bool Graph::signal_render(string &signal_xml,const float framerate) {
	if (find_node("signal_output")) {
		Signal_output *signal_output=dynamic_cast<Signal_output*>(find_node("signal_output"));
		return signal_output->render(duration,framerate,signal_xml);
	}
	cerr<<"Rotor: signal output node not found"<<endl;

	return false;
}
*/
bool Graph::print_features(xmlIO &XML,string &node){
	if (nodes.find(node)!=nodes.end()){
		if (dynamic_cast<Audio_processor*>(nodes[node])){
			XML.addValue("features",dynamic_cast<Audio_processor*>(nodes[node])->get_features());
			return true;
		}
		XML.addValue("error","node /"+node+"/ is not an Audio processor");
		return false;
	}
	XML.addValue("error","could not find node /"+node+"/");
	return false;
}
bool Graph::preview(xmlIO &XML,string &node,string &_format,int frame,int w,int h){
	if (nodes.find(node)!=nodes.end()){
		float t=frame/framerate;
		XML.addTag("preview");
		XML.addAttribute("preview","frame",ofToString(frame),0);
		XML.addAttribute("preview","nodeID",node,0);
		XML.pushTag("preview");
		if (dynamic_cast<Signal_node*>(nodes[node])){
			Time_spec ts=Time_spec(t,framerate,0.0f);
			XML.addValue("signal",dynamic_cast<Signal_node*>(nodes[node])->get_output(ts));
		}
		if (dynamic_cast<Image_node*>(nodes[node])){
			Frame_spec fs=Frame_spec(t,framerate,0.0f,w,h);
			Image *img=dynamic_cast<Image_node*>(nodes[node])->get_output(fs);
			vector<uchar> buf;
			string format=(_format==""?".png":_format);
			if (cv::imencode(format,img->rgb,buf)){ //, const vector<int>& params=vector<int>())
				stringstream output;
				Poco::Base64Encoder *enc=new Poco::Base64Encoder(output);
				enc->write(buf.data(),buf.size());
				enc->close();
				delete enc;
				XML.addValue("image",output.str());
				XML.addAttribute("image","format",format,0);
			}
		}
		XML.popTag();
		return true;	
	}
	return false;
}

bool Graph::video_render(const string &output_filename,const float framerate,float& progress) {
	//vector<Node*> loaders=find_nodes("video_loader");
	//for (auto i:loaders){
	//	if (!dynamic_cast<Video_loader*>(i)->isLoaded) {
	//		cerr<<"Rotor: all loaders must be populated before rendering"<<endl;
	//		return false;
	//	}
	//}
	if (find_node("video_output")) {
		Video_output *video_output=dynamic_cast<Video_output*>(find_node("video_output"));
		for (auto f: find_nodes("video_feedback")){
			(dynamic_cast<Video_feedback*>(f))->set_feedback(&(video_output->image));
		}
		return video_output->render(duration,framerate,output_filename,audio_filename,progress,outW,outH);
	}
	
	cerr<<"Rotor: video output node not found"<<endl;
	return false;
}

bool Graph::set_resolution(int w,int h){
	if (w>64&&h>48){
		outW=w;
		outH=h;
		return true;
	}
	else return false;
}
bool Graph::load(string data,string media_path){
	if (xml.loadFromBuffer(data)){
		return parseXml(media_path);
	}
	return parseJson(data,media_path);
	return false;
}
bool Graph::loadFile(string &filename,string media_path){
	//if (loaded) 
	printf("loading graph: %s\n",(filename).c_str());
	if (xml.loadFile(filename)){
		return parseXml(media_path);
	}
	Poco::FileInputStream fis(filename);
	Poco::CountingInputStream countingIstr(fis);
	std::string str;
	Poco::StreamCopier::copyToString(countingIstr, str);
	return parseJson(str,media_path);
}
bool Graph::check_audio(string audio,string path){
	if (audio!="")	{
		Poco::File f=Poco::File(path+audio);
		if (f.exists()) {
			audio_filename=path+audio;
			audio_loaded=true;
			cerr<<"Rotor: loading "<<path+audio<<" from graph"<<endl;
			return true;
		}
		cerr<<"Rotor: audio file "<<path+audio<<" not found"<<endl;
	}
	return false;
}
bool Graph::parseJson(string &data,string &media_path){
	//cerr<<data<<endl;
	//cerr<<"Trying to load JSON"<<endl;
	Json::Value root;   // will contains the root value after parsing.
	Json::Reader reader;
	bool parsingSuccessful = reader.parse( data, root );
	if ( !parsingSuccessful )
	{
	    // report to the user the failure and their locations in the document.
	    std::cout  << "Failed to parse configuration\n"
	               << reader.getFormattedErrorMessages();
	    return false;
	}
	//we know the json validates so clear the existing graph
	nodes.clear();
	check_audio(root["audio"].asString(),media_path);
	init(root["ID"].asString(),root["description"].asString());
	Json::Value jnodes = root["nodes"];
	for ( int i = 0; i < jnodes.size(); ++i ) {
	   //cerr<<"json found node: "<<jnodes[i]["type"].asString()<<endl;
	   map<string,string> settings;
	   vector<string> attrs;
	   //iterate members
	   settings["type"]=jnodes[i]["type"].asString();
	   settings["title"]=jnodes[i]["title"].asString();
	   settings["description"]=jnodes[i]["description"].asString();
	   Node* node=factory.create(settings);
	   	if (node) {
			string nodeID=jnodes[i]["id"].asString();
			if (nodes.find(nodeID)==nodes.end()){
				cerr << "Rotor: creating node '"<<nodeID<<"': '"<< settings["type"] << "'" << endl;
				nodes[nodeID]=node;
				//signal inputs
			   	for (int j=0;j< jnodes[i]["signal_inputs"].size();j++){
			   		if ((nodes[nodeID])->inputs.size()>j) {
						string fromID=jnodes[i]["signal_inputs"][j]["from"].asString();
						if (fromID!=""){
							if (nodes.find(fromID)!=nodes.end()) {
								if (!nodes[nodeID]->inputs[j]->connect((Signal_node*)nodes[fromID])){
									cerr << "Rotor: graph loader cannot connect input " << j << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
									return false;
								}
								else cerr << "Rotor: linked input " << j << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
							}
							else cerr << "Rotor: linking input " << j << " of node: '" << nodeID << "', cannot find target '" << fromID << "'" << endl;
						}
					}
					else cerr << "Rotor: input " << j << " of node: '" << nodeID << "' does not exist" << endl;
				}
				//image inputs
				if (dynamic_cast<Image_node*>(nodes[nodeID])!=nullptr) {
					for (int k=0;k<jnodes[i]["image_inputs"].size();k++){
						if (((Image_node*)nodes[nodeID])->image_inputs.size()<=k) {
							if (nodes[nodeID]->duplicate_inputs) {
								while(((Image_node*)nodes[nodeID])->image_inputs.size()<=k){
									((Image_node*)nodes[nodeID])->create_image_input(settings["description"],settings["title"]);
								}
							}
						}
						if (((Image_node*)nodes[nodeID])->image_inputs.size()>k) {
							string fromID=jnodes[i]["image_inputs"][k]["from"].asString();
							if (fromID!=""){
								if (nodes.find(fromID)!=nodes.end()) {
									if (dynamic_cast<Image_node*>(nodes[fromID])!=nullptr) {
										if (!dynamic_cast<Image_node*>(nodes[nodeID])->image_inputs[k]->connect((Image_node*)nodes[fromID])){
											cerr << "Rotor: graph loader cannot connect image input " << k << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
											return false;
										}
										else cerr << "Rotor: linked image input " << k << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
									}
									else cerr << "Rotor: cannot link image input "<< k << " of node '" << nodeID << "' to node '" << fromID << "' : not an image node" << endl;
								}
								else cerr << "Rotor: linking image input " << k << " of node: '" << nodeID << "', cannot find target '" << fromID << "'" << endl;
							}
						}
						else cerr << "Rotor: image input number " << k << " of node: '" << nodeID << "' does not exist" << endl;
					}
				}
				//parameters
				for (int l=0;l<jnodes[i]["parameters"].size();l++){
					
					string parameter=jnodes[i]["parameters"][l]["name"].asString();
					
					if (nodes[nodeID]->parameters.find(parameter)!=nodes[nodeID]->parameters.end()) {
						float val=jnodes[i]["parameters"][l]["value"].asFloat();
						nodes[nodeID]->parameters.find(parameter)->second->value=val;
						string fromID=jnodes[i]["parameters"][l]["from"].asString();
						if (nodes.find(fromID)!=nodes.end()) {
							if (!nodes[nodeID]->parameters[parameter]->connect(nodes[fromID])){
								cerr << "Rotor: graph loader cannot connect parameter " << parameter << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
								return false;
							}
							else cerr << "Rotor: linked parameter " << parameter << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
						}
						else if (fromID!="") cerr << "Rotor: linking parameter " << parameter << " of node: '" << nodeID << "', cannot find target '" << fromID << "'" << endl;
					}
					else cerr << "Rotor: cannot find parameter '" << parameter << "' of "<<settings["type"]<<" "<< nodeID << endl;
					
				}
				//attributes
				for (int m=0;m<jnodes[i]["attributes"].size();m++){
					string attribute=jnodes[i]["attributes"][m]["name"].asString();
					if (nodes[nodeID]->attributes.find(attribute)!=nodes[nodeID]->attributes.end()) {
						string val=jnodes[i]["attributes"][m]["value"].asString();
						nodes[nodeID]->attributes.find(attribute)->second->value=val;
					}
					else cerr << "Rotor: cannot find attribute '" << attribute << "' of "<<settings["type"]<<" "<< nodeID << endl;
				}

			}
			else cerr << "Rotor: duplicate node '"<<nodeID<<"' "<< endl;
   		}
	   	else {
			cerr << "Rotor: graph loader cannot find node '" << settings["type"] << "'" << endl;
			return false;
		}

	}
	loaded=true;
	return true;
}
bool Graph::parseXml(string media_path){
	nodes.clear();
	check_audio(xml.getAttribute("patchbay","audio","",0),media_path);
	init(xml.getAttribute("patchbay","ID","",0),xml.getValue("patchbay","",0));
		if(xml.pushTag("patchbay")) {
			int n1=xml.getNumTags("node");
			for (int i1=0;i1<n1;i1++){
				map<string,string> settings;
				vector<string> attrs;
				xml.getAttributeNames("node",attrs,i1);
				for (auto& attr: attrs) {
					settings[attr]=xml.getAttribute("node",attr,"",i1);
					//cerr << "Got attribute: " << attr << ":" << xml.getAttribute("node",attr,"",i1) << endl;
				}
				settings["media_path"]=media_path;
				Node* node=factory.create(settings);
				if (node) {
					string nodeID=xml.getAttribute("node","ID","",i1);
					if (nodes.find(nodeID)==nodes.end()){
						string nodetype=xml.getAttribute("node","type","",i1);
						cerr << "Rotor: creating node '"<<nodeID<<"': '"<< xml.getAttribute("node","type","",i1) << "'" << endl;
						nodes[nodeID]=node;
						if(xml.pushTag("node",i1)) {
							int n2=xml.getNumTags("signal_input");
							for (int i2=0;i2<n2;i2++){
								//TODO expand
								//nodes[nodeID]->create_signal_input(xml.getAttribute("signal_input","description","",i2),xml.getAttribute("signal_input","title","",i2));
								if ((nodes[nodeID])->inputs.size()>i2) {
									string fromID=xml.getAttribute("signal_input","from","",i2);
									if (nodes.find(fromID)!=nodes.end()) {
										if (!nodes[nodeID]->inputs[i2]->connect((Signal_node*)nodes[fromID])){
											cerr << "Rotor: graph loader cannot connect input " << i2 << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
											return false;
										}
										else cerr << "Rotor: linked input " << i2 << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
									}
									else cerr << "Rotor: linking input " << i2 << " of node: '" << nodeID << "', cannot find target '" << fromID << "'" << endl;
								}
								else cerr << "Rotor: input " << i2 << " of node: '" << nodeID << "' does not exist" << endl;
								
							}
							int n3=xml.getNumTags("image_input");
							for (int i3=0;i3<n3;i3++){
								//handle expandable inputs
								if (((Image_node*)nodes[nodeID])->image_inputs.size()<=i3&((Image_node*)nodes[nodeID])->duplicate_inputs){
									string desc=((Image_node*)nodes[nodeID])->image_inputs[0]->description;
									string title=((Image_node*)nodes[nodeID])->image_inputs[0]->title;
									while(((Image_node*)nodes[nodeID])->image_inputs.size()<=i3){
										((Image_node*)nodes[nodeID])->create_image_input(desc,title);
									}
								}
								//((Image_node*)nodes[nodeID])->create_image_input(xml.getValue("image_input","",i3));
								if (((Image_node*)nodes[nodeID])->image_inputs.size()>i3) {
									string fromID=xml.getAttribute("image_input","from","",i3);
									if (nodes.find(fromID)!=nodes.end()) {
										if (!(((Image_node*)nodes[nodeID])->image_inputs[i3]->connect((Image_node*)nodes[fromID]))){
											cerr << "Rotor: graph loader cannot connect image input " << i3 << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
											return false;
										}
										else cerr << "Rotor: linked image input " << i3 << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
									}
									else cerr << "Rotor: linking image input " << i3 << " of node: '" << nodeID << "', cannot find target '" << fromID << "'" << endl;
								}
								else cerr << "Rotor: image input number " << i3 << " of node: '" << nodeID << "' does not exist" << endl;
							}
							int n4=xml.getNumTags("parameter");
							for (int i4=0;i4<n4;i4++){
								string parameter=xml.getAttribute("parameter","name","",i4);
								if (nodes[nodeID]->parameters.find(parameter)!=nodes[nodeID]->parameters.end()) {
									string val=xml.getAttribute("parameter","value","",i4);
									if (val!="") nodes[nodeID]->parameters.find(parameter)->second->value=ofToFloat(val);
									string fromID=xml.getAttribute("parameter","from","",i4);
									if (nodes.find(fromID)!=nodes.end()) {
										if (!nodes[nodeID]->parameters[parameter]->connect(nodes[fromID])){
											cerr << "Rotor: graph loader cannot connect parameter input " << i4 << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
											return false;
										}
										else cerr << "Rotor: linked parameter input " << i4 << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
									}
									else if (fromID!="") cerr << "Rotor: linking parameter input " << i4 << " of node: '" << nodeID << "', cannot find target '" << fromID << "'" << endl;
								}
								else cerr << "Rotor: cannot find parameter input '" << parameter << "' of "<<nodetype<<" "<< nodeID << endl;
							}
							//still support old parameter_input
							n4=xml.getNumTags("parameter_input");
							for (int i4=0;i4<n4;i4++){
								string parameter=xml.getAttribute("parameter_input","name","",i4);
								if (nodes[nodeID]->parameters.find(parameter)!=nodes[nodeID]->parameters.end()) {
									string val=xml.getAttribute("parameter_input","value","",i4);
									if (val!="") nodes[nodeID]->parameters.find(parameter)->second->value=ofToFloat(val);
									string fromID=xml.getAttribute("parameter_input","from","",i4);
									if (nodes.find(fromID)!=nodes.end()) {
										if (!nodes[nodeID]->parameters[parameter]->connect(nodes[fromID])){
											cerr << "Rotor: graph loader cannot connect parameter input " << i4 << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
											return false;
										}
										else cerr << "Rotor: linked parameter input " << i4 << " of node '" << nodeID << "' to node '" << fromID << "'" << endl;
									}
									else cerr << "Rotor: linking parameter input " << i4 << " of node: '" << nodeID << "', cannot find target '" << fromID << "'" << endl;
								}
								else cerr << "Rotor: cannot find parameter input '" << parameter << "' of "<<nodetype<<" "<< nodeID << endl;
							}
							//extra key/value pairs that can be specific to sub-settings - WHAT WAS THIS
							//int n5=xml.getNumTags("parameter");
							//for (int i5=0;i5<n5;i5++){
							//	nodes[nodeID]->set_parameter(xml.getAttribute("parameter","name","",i5),xml.getAttribute("parameter","value","",i5));
							//}
							//if (n5>0) cerr << "Rotor: found " << n5 << " extra parameters for node '" << nodeID << "'" << endl;
							
							xml.popTag();
						}
					}
					else cerr << "Rotor: duplicate node '"<<nodeID<<"' "<< endl;
				}
				else {
					cerr << "Rotor: graph loader cannot find node '" << xml.getAttribute("node","type","",i1) << "'" << endl;
					return false;
				}
			}
			xml.popTag();
		}
		loaded=true;
		return true;
}