#ifndef ROTOR_H
#define ROTOR_H

/*
ROTORD rewrite

TJR-Jan-2014

NB when plugging in to framework - convert all cerr messages to logged
*/

#define ENABLE_TYPENAME(A) template<> struct TypeName<A> { static const char *Get() { return #A; }};

#include <string>
#include <sstream>
#include <iostream>
#include <map>
#include <vector>
#include <unordered_map>
#include <json/json.h>
#include <typeinfo>

template <typename T>
struct TypeName
{
    static const char* Get()
    {
        return typeid(T).name();
    }
};

template <>
struct TypeName<int> {
	static const char* Get() {
		return "int";
	}
};
template <>
struct TypeName<double> {
	static const char* Get() {
		return "number";
	}
};
template <>
struct TypeName<std::string> {
	static const char* Get() {
		return "string";
	}
};



namespace Rotor {

	class Node;
	template <class NT> class Node_type;

	class Audio_frame{
		public:
			Audio_frame(uint16_t *_samples,int _channels,int _numsamples){
				samples=_samples;
				channels=_channels;
				numsamples=_numsamples;
			}
			uint16_t *samples;
			int channels,numsamples;
	};
	class Frame_parameters{
		public:
			Frame_parameters(double _time,double _framerate,double _duration,int _w,int _h,Audio_frame *_audio=nullptr)
			{ time=_time; framerate=_framerate; duration=_duration; w=_w; h=_h;audio=_audio;};
			Frame_parameters(int _frame,double _framerate,double _duration,int _w,int _h,Audio_frame *_audio=nullptr)
			{ time=((double)_frame)/_framerate; framerate=_framerate; duration=_duration; w=_w; h=_h;audio=_audio;};
			int h,w;
			Frame_parameters lastframe() const{
				return Frame_parameters(time-(1.0/framerate),framerate,duration,w,h);
			}
			Frame_parameters nextframe() const{
				return Frame_parameters(time+(1.0/framerate),framerate,duration,w,h);
			}
			double get_time() const{
				return time;
			}
		private:
			double time; //num/denom ?
			double framerate;
			double duration;
			Audio_frame *audio;
	};
	class Variable { //pure virtual base type for variable pointers
		public:
			Variable(){connection=nullptr;};
			virtual ~Variable(){};
			virtual Json::Value to_json()=0;
			virtual void init(Json::Value s)=0;
			virtual bool connect(Node* target)=0;
	    	virtual std::string get_type()=0;
	    	virtual void create_connection(std::unordered_map<std::string,Node*> &nodes)=0;
			bool is_connected(){
				if (connection) return true;
				return false;
			}
			std::string get_connection_id();
			std::string name;
		protected:
			Node* connection;
			bool connectable;
			std::string input;
	};
	template <class T> class Variable_type : public Variable {
		public:
			Variable_type(std::string _name="",std::string _input="",bool _connectable=true){name=_name;input=_input;connectable=_connectable;};
			void init(Json::Value s){
				std::istringstream cur(s["value"].asString());
				cur >> value;
				name=s["name"].asString();
				input=s["input"].asString();
			}
			void create_connection(std::unordered_map<std::string,Node*> &nodes){
				for (auto node:nodes){
					if (node.first==input){
						if (connect(node.second)) {
							std::cerr<<"connecting '"<<TypeName<T>::Get()<<"' input to '"<<input<<"'"<<std::endl;
						}
						else std::cerr<<"could not connect '"<<TypeName<T>::Get()<<"'' input to '"<<input<<"'"<<std::endl;
					}
				}
			}
			Json::Value to_json();
	    		std::string get_type(){ //need this to output node templates
	    	    		return TypeName<T>::Get();
	    		}
//have to cast connect and get_output to use templated return types
	    		bool connect(Node* target){
				if (connectable){
					if (dynamic_cast<Node_type<T>*>(target)){
						connection=target;
						return true;
					}
				}
				return false;
			}
			const T& get(const Frame_parameters &frame){
				if (connection){
					return (dynamic_cast<Node_type<T>*>(connection))->get_output(frame);
					}
				return value;
			}
		protected:
			T value;
	};
	class Variable_array: public Variable {
		public:
			Variable_array(){};
		protected:
			std::vector<Variable> values;
	};
	template <class T> class Variable_array_type: public Variable_array {
		public:
			Variable_array_type(std::string n){name=n;connectable=true;};
			void init(Json::Value s){
				name=s["name"].asString();
				if (!s["input"].empty()){
					for (uint32_t i=0;i<s["input"].size();i++){
						add("",s["input"][i].asString());
					}
				}
			}
			void create_connection(std::unordered_map<std::string,Node*> &nodes){
				for (auto v:values){
					v.create_connection(nodes);
				}
			}
			Json::Value to_json();
	    		std::string get_type(){
	    	    		return TypeName<T>::Get();
	    		}
	    		bool connect(Node* target){
				if (connectable){
					if (dynamic_cast<Node_type<T>*>(target)){
						connection=target;
						return true;
					}
				}
				return false;
			}
			void add(std::string _name,std::string _input,bool _connectable=true){
				values.push_back(Variable_type<T>(_name,_input,_connectable));
			}
	    	void add(std::string _name="",bool _connectable=true,int num=1){
				for (int i=0;i<num;i++) values.push_back(Variable_type<T>(_name,"",_connectable));
			}
	    	bool connect(uint32_t which,Node* target){
				if (values.size()>which){
					return values[which].connect(target);
				}
				return false;
			}
			const T& get(uint32_t which,const Frame_parameters &frame){
				if (values.size()>which){
					return values[which].get(frame);
				}
				return value;
			}
			const std::vector<Variable_type<T>>& get_values(){return values;};
		protected:
			std::vector<Variable_type<T>> values;
			T value;
	};
	//don't forget the dupliicate inlets
	//it needs to be a property of a var
	//vars need to be:
	//std::unordered_map<std::string,std::pair<bool,std::vector<Variable*>>>
	//??
	//or could it be that a var has properties
	//var needs to be a class?
	//or multi var could be a subclass of var?
	//with an extra accessor const T& get(int which,const Frame_parameters &frame)
	//in Node we could have create_multi_inlet() and add_multi_inlet()
	class Node { //base type for node pointers
		public:
			virtual ~Node(){
				for (auto v:vars){
					delete v.second;
				}
			}
			bool connect(std::string v,Node *t){
				auto var=vars.find(v);
				if (var!=vars.end()){
					if ((*var).second->connect(t)){
						return true;
					}
				}
				return false;
			}
			std::string get_type(){return type;};
			std::string& get_id(){return id;};
			Json::Value to_json();
			virtual Node* clone(Json::Value &_settings)=0;
			virtual std::string get_output_type()=0;
			void create_connections(std::unordered_map<std::string,Node*> &nodes){
				for (auto var:vars) var.second->create_connection(nodes);
			}
		protected:
			std::unordered_map<std::string,Variable*> vars;
			std::string type;
			std::string id;
			std::string type_id;
			std::string description;
			std::string title;
			std::string ui_type;
	};
	template <class NT> class Node_type : public Node {
		public:
			virtual const NT& get_output(const Frame_parameters &frame)=0;
			void init(Json::Value settings){
				if (!settings["vars"].empty()){
					for ( uint32_t i = 0; i < settings["vars"].size(); ++i ) {
						vars[settings["vars"][i]["name"].asString()]->init(settings["vars"][i]);
					}
				}
				if (!settings["id"].empty()) id=settings["id"].asString();
			}
			std::string get_output_type(){return TypeName<NT>::Get();};
			template <class IT> Variable_type<IT>* create_inlet(std::string name){
				vars[name]=new Variable_type<IT>(name,"",true);
				return (dynamic_cast<Variable_type<IT>*>(vars[name]));
			}
			template <class IT> Variable_type<IT>* create_attribute(std::string name){
				vars[name]=new Variable_type<IT>(name,"",false);
				return (dynamic_cast<Variable_type<IT>*>(vars[name]));
			}
			template <class IT> Variable_array_type<IT>* create_array(std::string name){
				vars[name]=new Variable_array_type<IT>(name);
				return (dynamic_cast<Variable_array_type<IT>*>(vars[name]));
			}
	};
}

#endif //ROTOR_H