#pragma once

#include "node.h"
#include "raymath.h"
#include <map>

#define R 14.0f

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

class MidiNode : public Node{

	public:

	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 = 25; // 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;

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

	void trigger(){
		triggered = true;
		start = GetTime();
		fade = midi_gate;
	}

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

	void draw(Vec2D<float> mouse, Camera2D cam) override {
		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(!dragging && !selected && !in_rect && !hovering) DrawCircle(position.x, position.y, radius, gate_color);
		if(selected || dragging || in_rect || hovering) DrawCircle(position.x, position.y, radius, (Color){108, 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(IsKeyDown(KEY_LEFT_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;}
			}
		}

		Vec2D<float> connector = position - ((position - mouse)).norm() * (radius + R);
		float opacity = 150.0f - Clamp(position.dist(mouse), 0.0f, 150.0f);
		
		if(hover_connect){
			DrawCircle(connector.x, connector.y, 8.0f, (Color){108, 153, 187, 255});
		} else {
			DrawCircle(connector.x, connector.y, 8.0f, (Color){108, 153, 187, (unsigned char)opacity});
		}
		if(highlighted) DrawCircleLinesV(position, radius+2, (Color){255, 0, 0, 200});
	}
};