#pragma once

#include "datatypes.h"
#include "graph-item.h"
#include "node.h"
#include "raymath.h"
#include <map>

#define RADIUS 14.0f

namespace VISEQ::GRAPH::NODES {

	class MidiNode : public Node{
	
		public:
		using GraphItem::state;

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

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

		NodeType nodeType = NodeType::MIDINODE;

		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;
		
		bool triggered = false;

		int mode = 0;

		int midi_note_channel = 1;
		int midi_note = 48;
		int midi_velocity = 60;
		int midi_gate = 10; // gate time in milliseconds

		int midi_cc_channel = 0;
		int midi_cc = 0;
		int midi_cc_value = 0;

		int midi_pc_channel = 0;
		int midi_pc = 0;
		int midi_pc_value = 0;
		
		int midi_pitch_bend_channel = 0;
		int midi_pitch_bend = 8192;

		float start = 0;
		float fade = 0;
		float time = 0;
		float last = 0;

		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 = midi_gate;
		}

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

		int Ggate(){
			return midi_gate;
		}
		void interact(){
			float wheel = GetMouseWheelMove();

			if(KEYS_DOWN.ctrl && !KEYS_DOWN.alt){
				switch(GraphItem::state->node_mode){
					case 0:{
						midi_note += wheel;
					break;}
					case 1:{
						midi_cc += wheel;
					break;}
					case 2:{
						midi_pitch_bend += wheel;
					break;}
				};
				if(midi_note < 0 ) midi_note = VISEQ::MIDI::Notes.size()-1;
				if(midi_note >= VISEQ::MIDI::Notes.size() ) midi_note = 0;

				if(midi_cc < 0 ) midi_cc = 127;
				if(midi_cc > 127 ) midi_cc = 0;

				if(midi_pitch_bend < 0 ) midi_pitch_bend = 0;
				if(midi_pitch_bend > 16383 ) midi_pitch_bend = 16383;

			} else if(KEYS_DOWN.shift && !KEYS_DOWN.alt) {
				midi_velocity += wheel;
				if(midi_velocity < 0 ) midi_velocity = 127;
				if(midi_velocity > 127 ) midi_velocity = 0; 
			} else if(KEYS_DOWN.alt) {
				switch(GraphItem::state->node_mode){
					case 0:{
						float incr = (float)wheel * (KEYS_DOWN.shift ? 100.0f : KEYS_DOWN.ctrl ? 1000.0f : 10.0f);
						midi_gate += incr;
						if(midi_gate < 0 ) midi_gate = 0;
					break;}
					case 1:{
						midi_cc_value += wheel;
						if(midi_cc_value < 0 ) midi_cc_value = 0;
						if(midi_cc_value > 127 ) midi_cc_value = 127;
					break;}
					case 2:{
						float incr = (float)wheel * (KEYS_DOWN.shift ? 100.0f : KEYS_DOWN.ctrl ? 1000.0f : 10.0f);
						midi_pitch_bend += incr;
						if(midi_pitch_bend < 0 ) midi_pitch_bend = 0;
						if(midi_pitch_bend > 16383 ) midi_pitch_bend = 16383;
					break;}
				};
			}
		}

		void draw(Vec2Df mouse, Camera2D cam) override {
			Node::draw();
			if(hovering() || selected) interact();

			last = time;
			time = GetTime();
			if(time - start >= (float)midi_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) * midi_velocity - 90.0f, 24, (Color){158, 134, 200, 255});
			DrawCircleLinesV(position, radius + 4, (Color){100, 100, 100, 255});

			float midi_gate_map = round((255.f / midi_gate) * fade);

			Color gate_color = VS_COLOR_GREEN;

			switch(mode){
				case 0:{
					gate_color = ColorAlphaBlend(VS_COLOR_RED, (Color){180, 210, 115, (unsigned char)Clamp(255 - midi_gate_map, 0, 255)}, WHITE);
					break;}
				case 1:{
					gate_color = ColorAlphaBlend(VS_COLOR_RED, (Color){229, 181, 103, (unsigned char)Clamp(255 - midi_gate_map, 0, 255)}, WHITE);
					break;}
				case 2:{
					gate_color = ColorAlphaBlend(VS_COLOR_RED, (Color){232, 125, 62, (unsigned char)Clamp(255 - midi_gate_map, 0, 255)}, WHITE);
					break;}
			}
			if(!picked_up && !selected && !in_selection_rect && !hovering()) DrawCircle(position.x, position.y, radius, gate_color);
			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(KEYS_DOWN.alt){
				float value = 0;
				int offset_gate = 0;
				switch(mode){
					case 0:
						value = (float)midi_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)midi_cc_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)midi_pitch_bend;
						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)midi_note].c_str(), 12);
						DrawText(MIDI::Notes[midi_note].c_str(), position.x-offset/2.0f, position.y-5, 12, WHITE );
						break;}
					case 1:{
						int offset = MeasureText(std::format("{}", midi_cc).c_str(), 12);
						DrawText(std::format("{}", midi_cc).c_str(), position.x-offset/2.0f, position.y-5, 12, WHITE );
						break;}
					case 2:{
						int offset = MeasureText(std::format("{}", midi_pitch_bend).c_str(), 12);
						DrawText(std::format("{}", midi_pitch_bend).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);
		}
	};
}