#pragma once

#include "config-node.h"
#include "datatypes.h"
#include "edge.h"
#include "node-handle.h"
#include "node.h"
#include "raylib.h"
#include "state.h"
#include <map>
#include <memory>

namespace VISEQ::APP::UI{
	class AppUI;
}

namespace VISEQ::GRAPH::NODES {

	class MidiNode : public Node{
		friend class APP::UI::AppUI;

		private:
		MIDI::MidiNote midiNote;
		MIDI::MidiCC midiCC;
		MIDI::MidiPC midiPC;
		MIDI::MidiPitchBend midiPitchBend;
		MIDI::MidiAftertouch midiAftertouch;
		MIDI::MidiPolyPressure midiPolyPressure;

		std::vector<std::shared_ptr<NodeHandle<int>>> midiNoteHandles;
		std::vector<std::shared_ptr<NodeHandle<int>>> midiCCHandles;
		std::vector<std::shared_ptr<NodeHandle<int>>> midiPCHandles;
		std::vector<std::shared_ptr<NodeHandle<int>>> midiPitchBendHandles;
		std::vector<std::shared_ptr<NodeHandle<int>>> midiAftertouchHandles;
		std::vector<std::shared_ptr<NodeHandle<int>>> midiPolyPressureHandles;

		public:

		struct NodeMidiOut{
			int i;
			std::string name;
			bool enabled;
		};

		MidiNode() {init();}
		MidiNode(Vec2Df position) : Node(position) {init();}

		NodeType nodeType = NodeType::MIDINODE;
		
		//std::vector<std::shared_ptr<ControlNode>> controlNodes;

		bool send_note = true;
		bool send_cc = false;
		bool send_pc = false;
		bool send_pitch_bend = false;

		int midi_out = -1;
		
		std::map<int, NodeMidiOut> midi_outs;

		bool gui_midi_channel_edit = false;
		bool gui_midi_note_edit = false;
		bool gui_midi_velocity_edit = false;
		bool gui_midi_gate_edit = false;
		
		float start = 0;
		float fade = 0;
		float time = 0;
		float last = 0;
		
		void init(){
			float orbit_1 = radius + 25.0f;
			float orbit_2 = orbit_1 + 20.0f;
			float orbit_3 = orbit_2 + 20.0f;
			float orbit_4 = orbit_3 + 20.0f;
			float orbit_5 = orbit_4 + 30.0f;

			midiNoteHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiNote.note, position, VS_COLOR_GREEN, "NOTE", orbit_5, 12));
			midiNoteHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiNote.channel, position, VS_COLOR_PINK, "CHANNEL", orbit_1, 1, 1, 16, 40, 150));
			midiNoteHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiNote.gate, position, VS_COLOR_RED, "GATE", orbit_3, 30, 10, 30000, 40, 300));
			midiNoteHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiNote.velocity, position, VS_COLOR_PURPLE, "VELOCITY", orbit_4, 32));
			
			midiCCHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiCC.control, position, VS_COLOR_SUN, "CONTROL", orbit_3, 32));
			midiCCHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiCC.channel, position, VS_COLOR_PINK, "CHANNEL", orbit_1, 1, 1, 16, 40, 150));
			midiCCHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiCC.value, position, VS_COLOR_GREEN, "VALUE", orbit_4, 32));
			
			midiPCHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiPC.program, position, VS_COLOR_SUN, "PROGRAM", orbit_3, 32));
			midiPCHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiPC.channel, position, VS_COLOR_PINK, "CHANNEL", orbit_1, 1, 1, 16, 40, 150));
			midiPCHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiPC.value, position, VS_COLOR_GREEN, "VALUE", orbit_4, 32));

			midiPitchBendHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiPitchBend.channel, position, VS_COLOR_PINK, "CHANNEL", orbit_1, 1, 1, 16, 40, 150));
			midiPitchBendHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiPitchBend.value, position, VS_COLOR_GREEN, "VALUE", orbit_3, 256, -8192, 8192, 40, 300));
			
			midiAftertouchHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiAftertouch.channel, position, VS_COLOR_PINK, "CHANNEL", orbit_1, 1, 1, 16, 40, 150));
			midiAftertouchHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiAftertouch.value, position, VS_COLOR_GREEN, "VALUE", orbit_3, 32));
			
			midiPolyPressureHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiPolyPressure.channel, position, VS_COLOR_PINK, "CHANNEL", orbit_1, 1, 1, 16, 40, 150));
			midiPolyPressureHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiPolyPressure.value, position, VS_COLOR_GREEN, "VALUE", orbit_4, 32));
			midiPolyPressureHandles.emplace_back(std::make_shared<NodeHandle<int>>(midiPolyPressure.note, position, VS_COLOR_SUN, "NOTE", orbit_5, 12));
		}

		std::shared_ptr<MidiNode> getPtr(){
			return std::static_pointer_cast<MidiNode>(shared_from_this());
		}

		virtual NodeType getNodeType() const override {return nodeType;}

		void setMidiPort(int midi_port, bool _state){
			midi_outs[midi_port].enabled = _state;
		}

		void trigger() override {
			triggered = true;
			start = GetTime();
			fade = getNotes().gate;
		}

		void setMode(int _mode) override {
			mode = _mode;
		}

		MIDI::MidiNote getNotes() const {
			MIDI::MidiNote cNote;
			for(auto cn : getConnectedConfigNodes()){
				cNote += cn.lock()->getNotes();
			}
			return midiNote + cNote;
		}
		
		MIDI::MidiCC getCC() const {
			MIDI::MidiCC cCC;
			for(auto cn : getConnectedConfigNodes()){
				cCC += cn.lock()->getCC();
			}
			return midiCC + cCC;
		}
		
		MIDI::MidiPC getPC() const {
			MIDI::MidiPC cPC;
			for(auto cn : getConnectedConfigNodes()){
				cPC += cn.lock()->getPC();
			}
			return midiPC + cPC;
		}
		
		MIDI::MidiPitchBend getPitchBend() const {
			MIDI::MidiPitchBend cPitchBend;
			for(auto cn : getConnectedConfigNodes()){
				cPitchBend += cn.lock()->getPitchBend();
			}
			return midiPitchBend + cPitchBend;
		}
		
		MIDI::MidiAftertouch getAftertouch() const {
			MIDI::MidiAftertouch cAftertouch;
			for(auto cn : getConnectedConfigNodes()){
				cAftertouch += cn.lock()->getAftertouch();
			}
			return midiAftertouch + cAftertouch;
		}
		
		MIDI::MidiPolyPressure getPolyPressure() const {
			MIDI::MidiPolyPressure cPolyPressure;
			for(auto cn : getConnectedConfigNodes()){
				cPolyPressure += cn.lock()->getPolyPressure();
			}
			return midiPolyPressure + cPolyPressure;
		}
		
		std::vector<std::weak_ptr<ConfigNode>> getConnectedConfigNodes() const {
			auto config_conns = connections | RANGE_FILTER([](EDGES::SPTR_Edge c){ return c->start.lock()->getNodeType() == NodeType::CONFIGNODE || c->end.lock()->getNodeType() == NodeType::CONFIGNODE;}) | std::ranges::to<std::vector<std::shared_ptr<EDGES::Edge>>>();
			std::vector<std::weak_ptr<ConfigNode>> config_nodes;
			for(auto e : config_conns){
				if(e->start.lock()->getNodeType() == NodeType::CONFIGNODE) config_nodes.push_back(std::dynamic_pointer_cast<ConfigNode>(e->start.lock()));
				if(e->end.lock()->getNodeType() == NodeType::CONFIGNODE) config_nodes.push_back(std::dynamic_pointer_cast<ConfigNode>(e->end.lock()));
			}
			/*std::cout << "id: " << id << " n cn: " << config_nodes.size() << std::endl;*/
			return config_nodes;
		}

		void update() override {
			Node::update();
			mode = state->node_mode;	
			for(auto mh : midiNoteHandles){
				mh->update(position);
			}
			for(auto mh : midiCCHandles){
				mh->update(position);
			}
			for(auto mh : midiPCHandles){
				mh->update(position);
			}
			for(auto mh : midiPitchBendHandles){
				mh->update(position);
			}
			for(auto mh : midiAftertouchHandles){
				mh->update(position);
			}
			for(auto mh : midiPolyPressureHandles){
				mh->update(position);
			}
		}
		
		void modMidiNote(int incr){
			midiNote.note += incr;
			if(midiNote.note < 0 ) midiNote.note = VISEQ::MIDI::Notes.size()-1;
			if(midiNote.note >= (int)VISEQ::MIDI::Notes.size()) midiNote.note = 0;	
		}

		void modMidiCC(int incr){
			midiCC.control += incr;
			if(midiCC.control < 0 ) midiCC.control = 127;
			if(midiCC.control > 127 ) midiCC.control = 0;
		}
		
		void modMidiPC(int incr){
			midiPC.program += incr;
			if(midiPC.program < 0 ) midiPC.program = 127;
			if(midiPC.program > 127 ) midiPC.program = 0;
		}
		
		void modMidiPitchBend(int incr){
			midiPitchBend.value += incr;
			if(midiPitchBend.value < 0 ) midiPitchBend.value = 0;
			if(midiPitchBend.value > 16383 ) midiPitchBend.value = 16383;
		}
		
		void modMidiAftertouch(int incr){
			midiAftertouch.value += incr;
			if(midiAftertouch.value < 0 ) midiAftertouch.value = 0;
			if(midiAftertouch.value > 127 ) midiAftertouch.value = 127;
		}
		
		void modMidiPolyPressure(int incr){
			midiPolyPressure.value += incr;
			if(midiPolyPressure.value < 0 ) midiPolyPressure.value = 0;
			if(midiPolyPressure.value > 127 ) midiPolyPressure.value = 127;
		}

		void interact(){
			float wheel = GetMouseWheelMove();

			if(KEYS_DOWN.ctrl && !KEYS_DOWN.alt){
				switch(state->node_mode){
					case 0:{
						modMidiNote(wheel);
					break;}
					case 1:{
						modMidiCC(wheel);
					break;}
					case 2:{
						modMidiPC(wheel);
					break;}
					case 3:{
						modMidiPitchBend(wheel);
					break;}
					case 4:{
						modMidiAftertouch(wheel);
					break;}
					case 5:{
						modMidiPolyPressure(wheel);
					break;}
				};
			} else if(KEYS_DOWN.shift && !KEYS_DOWN.alt) {
				midiNote.velocity += wheel;
				if(midiNote.velocity < 0 ) midiNote.velocity = 127;
				if(midiNote.velocity > 127 ) midiNote.velocity = 0; 
			} else if(KEYS_DOWN.alt) {
				switch(state->node_mode){
					case 0:{
						float incr = (float)wheel * (KEYS_DOWN.shift ? 100.0f : KEYS_DOWN.ctrl ? 1000.0f : 10.0f);
						midiNote.gate += incr;
						if(midiNote.gate < 0 ) midiNote.gate = 0;
					break;}
					case 1:{
						midiCC.value += wheel;
						if(midiCC.value < 0 ) midiCC.value = 0;
						if(midiCC.value > 127 ) midiCC.value = 127;
					break;}
					case 2:{
						midiPC.value += wheel;
						if(midiPC.value < 0 ) midiPC.value = 0;
						if(midiPC.value > 127 ) midiPC.value = 127;
					break;}
					case 3:{
						float incr = (float)wheel * (KEYS_DOWN.shift ? 100.0f : KEYS_DOWN.ctrl ? 1000.0f : 10.0f);
						midiPitchBend.value += incr;
						if(midiPitchBend.value < 0 ) midiPitchBend.value = 0;
						if(midiPitchBend.value > 16383 ) midiPitchBend.value = 16383;
					break;}
					case 4:{
						float incr = (float)wheel * (KEYS_DOWN.shift ? 100.0f : KEYS_DOWN.ctrl ? 1000.0f : 10.0f);
						midiAftertouch.value += incr;
						if(midiAftertouch.value < 0 ) midiAftertouch.value = 0;
						if(midiAftertouch.value > 127 ) midiAftertouch.value = 127;
					break;}
					case 5:{
						float incr = (float)wheel * (KEYS_DOWN.shift ? 100.0f : KEYS_DOWN.ctrl ? 1000.0f : 10.0f);
						midiPolyPressure.value += incr;
						if(midiPolyPressure.value < 0 ) midiPolyPressure.value = 0;
						if(midiPolyPressure.value > 127 ) midiPolyPressure.value = 127;
					break;}
				};
			}
		}

		void draw(Vec2Df mouse, Camera2D cam) override {
			Node::draw();
			for(auto mh : midiNoteHandles){
				if(mode == 0) mh->unlock();
				else mh->lock();
				mh->draw();
			}
			for(auto mh : midiCCHandles){
				if(mode == 1) mh->unlock();
				else mh->lock();
				mh->draw();
			}
			for(auto mh : midiPCHandles){
				if(mode == 2) mh->unlock();
				else mh->lock();
				mh->draw();
			}
			for(auto mh : midiPitchBendHandles){
				if(mode == 3) mh->unlock();
				else mh->lock();
				mh->draw();
			}
			for(auto mh : midiAftertouchHandles){
				if(mode == 4) mh->unlock();
				else mh->lock();
				mh->draw();
			}
			for(auto mh : midiPolyPressureHandles){
				if(mode == 5) mh->unlock();
				else mh->lock();
				mh->draw();
			}
			
			if(hovering() || selected) interact();

			last = time;
			time = GetTime();
			if(time - start >= (float)getNotes().gate / 1000.0f){
				triggered = false;
				fade = 0;
			} else {
				fade -= (time - last)*1000.0f;
			}

			DrawRing(position, radius, radius + 4, 0, 360, 24, (Color){60, 60, 60, 255});
			DrawRing(position, radius, radius + 4, 0.0f - 90.0f, (360.0f/127.0f) * getNotes().velocity - 90.0f, 24, (Color){158, 134, 200, 255});
			DrawCircleLinesV(position, radius + 4, (Color){100, 100, 100, 255});

			float trigger_alpha = (1.f / getNotes().gate) * fade;
			
			if(!picked_up && !selected && !in_selection_rect && !hovering()) DrawCircle(position.x, position.y, radius, ColorLerp(state->mode_color, VS_COLOR_RED, trigger_alpha));
			if(picked_up || in_selection_rect || hovering()) DrawCircle(position.x, position.y, radius, (Color){108, 153, 187, 255});
			if(selected) DrawCircle(position.x, position.y, radius, (Color){255, 153, 187, 255});

			if(distributionMode == Distribution::RANDOM) DrawCircle(position.x, position.y, radius/2.0f, VS_COLOR_PINK);
			if(distributionMode == Distribution::ROUNDROBIN) DrawCircle(position.x, position.y, radius/2.0f, VS_COLOR_YELLOW);
			
			if(selected) DrawCircleLinesV(position, radius+8, (Color){158, 255, 210, 255});
			
			/*if(fade > 0) DrawCircle(position.x, position.y, radius, ColorAlpha(VS_COLOR_RED, trigger_alpha));*/

			if(KEYS_DOWN.alt){
				float value = 0;
				int offset_gate = 0;
				switch(mode){
					case 0:
						value = (float)getNotes().gate/1000.0f;
						offset_gate = MeasureText(std::format("{:.2f}", value).c_str(), 12);
						DrawText(std::format("{:.2f}",value).c_str(), position.x-offset_gate/2.0f, position.y-5, 12, WHITE );
						break;
					case 1:
						value = (float)getCC().value;
						offset_gate = MeasureText(std::format("{}", value).c_str(), 12);
						DrawText(std::format("{}",value).c_str(), position.x-offset_gate/2.0f, position.y-5, 12, WHITE );
						break;
					case 2:
						value = (float)getPitchBend().value;
						offset_gate = MeasureText(std::format("{}", value).c_str(), 12);
						DrawText(std::format("{}",value).c_str(), position.x-offset_gate/2.0f, position.y-5, 12, WHITE );
						break;
				}
			} else {
				switch(mode){
					case 0:{
						int offset = MeasureText(MIDI::Notes[(int)getNotes().note].c_str(), 12);
						DrawText(MIDI::Notes[getNotes().note].c_str(), position.x-offset/2.0f, position.y-5, 12, WHITE );
						break;}
					case 1:{
						int offset = MeasureText(std::format("{}", getCC().control).c_str(), 12);
						DrawText(std::format("{}", getCC().control).c_str(), position.x-offset/2.0f, position.y-5, 12, WHITE );
						break;}
					case 2:{
						int offset = MeasureText(std::format("{}", getPitchBend().value).c_str(), 12);
						DrawText(std::format("{}", getPitchBend().value).c_str(), position.x-offset/2.0f, position.y-5, 12, WHITE );
						break;}
				}
			}

			if(highlighted) DrawCircleLinesV(position, radius+2, (Color){255, 0, 0, 200});
			//DrawText(std::format("{}", id).c_str(), position.x + 20.0f, position.y + 20.0f, 10, WHITE);
		}
	
		friend void to_json(json& j, const MidiNode& N){
			std::vector<int> conns;		
			std::ranges::transform(N.connections.begin(), N.connections.end(), std::back_inserter(conns), [](VISEQ::GRAPH::EDGES::SPTR_Edge e){ return e->id;});
			std::map<std::string, Vec2Df> noteHandles;
			std::map<std::string, Vec2Df> CCHandles;
			std::map<std::string, Vec2Df> PCHandles;
			std::map<std::string, Vec2Df> pitchBendHandles;
			std::map<std::string, Vec2Df> aftertouchHandles;
			std::map<std::string, Vec2Df> polyPressureHandles;

			for(auto i : N.midiNoteHandles){
				noteHandles.insert(std::make_pair(i->name, i->position));
			}
			for(auto i : N.midiCCHandles){
				CCHandles.insert(std::make_pair(i->name, i->position));
			}
			for(auto i : N.midiPCHandles){
				PCHandles.insert(std::make_pair(i->name, i->position));
			}
			for(auto i : N.midiPitchBendHandles){
				pitchBendHandles.insert(std::make_pair(i->name, i->position));
			}
			for(auto i : N.midiAftertouchHandles){
				pitchBendHandles.insert(std::make_pair(i->name, i->position));
			}
			for(auto i : N.midiPolyPressureHandles){
				polyPressureHandles.insert(std::make_pair(i->name, i->position));
			}

			j = json{
				{"id", N.id},
				{"position", N.position},
				{"distributionMode", N.distributionMode},
				{"nodeType", N.nodeType},
				{"midiNote", N.midiNote},
				{"midiCC", N.midiCC},
				{"midiPC", N.midiPC},
				{"midiPitchBend", N.midiPitchBend},
				{"midiAftertouch", N.midiAftertouch},
				{"midiPolyPressure", N.midiPolyPressure},
				{"midiNoteHandle", noteHandles},
				{"midiCCHandle", CCHandles},
				{"midiPCHandle", PCHandles},
				{"midiPitchBendHandle", pitchBendHandles},
				{"midiAftertouchHandle", aftertouchHandles},
				{"midiPolyPressureHandle", polyPressureHandles},
				{"connections",conns}};

		}

		friend void from_json(const json& j, MidiNode& N){
			std::cout << "MidiNode" << std::endl;

			j.at("id").get_to(N.id);
			j.at("position").get_to<Vec2Df>(N.position);
			j.at("distributionMode").get_to(N.distributionMode);
			j.at("nodeType").get_to(N.nodeType);
			j.at("midiNote").get_to(N.midiNote);
			j.at("midiCC").get_to(N.midiCC);
			j.at("midiPC").get_to(N.midiPC);
			j.at("midiPitchBend").get_to(N.midiPitchBend);
			j.at("midiAftertouch").get_to(N.midiAftertouch);
			j.at("midiPolyPressure").get_to(N.midiPolyPressure);

			std::map<std::string, Vec2Df> noteHandles;
			std::map<std::string, Vec2Df> CCHandles;
			std::map<std::string, Vec2Df> PCHandles;
			std::map<std::string, Vec2Df> pitchBendHandles;
			std::map<std::string, Vec2Df> aftertouchHandles;
			std::map<std::string, Vec2Df> polyPressureHandles;

			j.at("midiNoteHandle").get_to(noteHandles);
			j.at("midiCCHandle").get_to(CCHandles);
			j.at("midiPCHandle").get_to(PCHandles);
			j.at("midiPitchBendHandle").get_to(pitchBendHandles);
			j.at("midiAftertouchHandle").get_to(aftertouchHandles);
			j.at("midiPolyPressureHandle").get_to(polyPressureHandles);

			N.midiNoteHandles[0]->position = noteHandles["NOTE"];
			N.midiNoteHandles[1]->position = noteHandles["CHANNEL"];
			N.midiNoteHandles[2]->position = noteHandles["GATE"];
			N.midiNoteHandles[3]->position = noteHandles["VELOCITY"];

			N.midiCCHandles[0]->position = CCHandles["CONTROL"];
			N.midiCCHandles[1]->position = CCHandles["CHANNEL"];
			N.midiCCHandles[2]->position = CCHandles["VALUE"];
			
			N.midiPCHandles[0]->position = PCHandles["PROGRAM"];
			N.midiPCHandles[1]->position = PCHandles["CHANNEL"];
			N.midiPCHandles[2]->position = PCHandles["VALUE"];

			N.midiPitchBendHandles[0]->position = pitchBendHandles["CHANNEL"];
			N.midiPitchBendHandles[1]->position = pitchBendHandles["VALUE"];
			
			N.midiAftertouchHandles[0]->position = aftertouchHandles["CHANNEL"];
			N.midiAftertouchHandles[1]->position = aftertouchHandles["VALUE"];

			N.midiPolyPressureHandles[0]->position = polyPressureHandles["CHANNEL"];
			N.midiPolyPressureHandles[1]->position = polyPressureHandles["VALUE"];
			N.midiPolyPressureHandles[2]->position = polyPressureHandles["NOTE"];
			//j.at("connections").get_to(N.connections);
		}
	};
}