viseq/src/ui.h

#pragma once

#include "raylib.h"
#include "raygui/raygui.h"
#include "raygui/dark/style_dark.h"

#include "datatypes.h"
#include "graph-manager.h"
#include "signal-manager.h"

#include <ranges>

class AppUI{
	public:

	AppUI(AppSettings *settings, SignalManager *sm, GraphManager *gm) : settings(settings), gm(gm), sm(sm) {}

	AppSettings *settings;
	GraphManager *gm;
	SignalManager *sm;

	void init(){
		GuiLoadStyleDark();
	}

	void drawSideBar(){

		GuiPanel((Rectangle){-1,-30,280,(float)settings->height + 31}, "");
		const int s_controls = 25;
		//GuiSetStyle(BUTTON, STATE_FOCUSED, 0xFF0000FF);
		if(GuiButton((Rectangle){20, (float)s_controls, 80, 30}, "PLAY")) sm->play ? sm->pause() : sm->start();
		if(GuiButton((Rectangle){120, (float)s_controls, 60, 30}, "STOP")) sm->stop();
		if(GuiButton((Rectangle){200, (float)s_controls, 60, 30}, "RESET")) sm->reset();

		const int s_bpm = s_controls + 45;
		GuiLabel((Rectangle){20, (float)s_bpm, 120, 20}, "TIMING");
		GuiLabel((Rectangle){200, (float)s_bpm, 40, 20}, std::format("{}",settings->bpm).c_str());
		float _bpm = (float)settings->bpm;
		GuiSlider((Rectangle){20, (float)s_bpm + 25, 240, 20}, "", "", &_bpm, 1, 300);
		settings->bpm = _bpm;

		int s_midi = s_bpm + 75;
		GuiLabel((Rectangle){20, (float)s_midi, 240, 20}, "MIDI OUT");
		if(GuiButton((Rectangle){200, (float)s_midi, 60, 20}, "RELOAD")) sm->midi.enumerate();

		int i = 0;
		for(auto& mo : sm->midi.midi_out_ports){
			GuiCheckBox((Rectangle){20, (float)s_midi + 25 + i*20.0f, 20, 20}, mo.port.port_name.substr(0,32).c_str(), &mo.enabled);
			i++;
		}

		if(GuiButton((Rectangle){20, (float)s_midi + 45 + i*20.0f, 240, 20}, "ENABLE FOR ALL")) {
			for(auto &midis : sm->midi.midi_out_ports){
				if(midis.enabled){
					for(auto &n : gm->nodes){
						n.second.setMidiPort(midis.number, true);
					}
				}
			}
		}

		int s_signal = s_midi + 100 + i*20;
		GuiLabel((Rectangle){20, (float)s_signal, 240, 20}, "SIGNALS");
		if(GuiButton((Rectangle){20, (float)s_signal + 35, 240, 20}, "ADD RANDOM SIGNAL")){
			int ni = gm->getRandomNode(); 
			sm->addSignalAtNode(ni);
		}
		
		int signal_configs = 0;
		for(auto &s : sm->signals){
			if(GuiButton((Rectangle){20,(float)s_signal+70 + signal_configs*20, 20, 20},"X")) sm->removeSignal(signal_configs);
			DrawCircle(55, s_signal+80 + signal_configs*20, 6.0F, s.color);
			float s_multiplier = round(s.multiplier*10.0f)/10.0f;
			float signal_bpm = settings->bpm * s.multiplier;
			GuiSlider((Rectangle){70, (float)s_signal+70 + signal_configs*20, 160, 20}, "", std::format("{:.1f}",signal_bpm).c_str(), &s_multiplier, 0.1, 10);
			s.multiplier = s_multiplier;
			s.speed = settings->speed;
			signal_configs++;
		}
		
		int s_nodes = s_signal + 70 + signal_configs*20;
		GuiLabel((Rectangle){20, (float)s_nodes + 30, 240, 20}, "NODES");
		if(GuiButton((Rectangle){20, (float)s_nodes + 60, 240, 20}, "ADD RANDOM NODE")){
			gm->addNode(200.0f, 200.0f);
		}
		DrawText(std::format("{}", gm->nodes.size()).c_str(), 20, settings->height - 70, 15, VS_COLOR_WHITE);
		DrawFPS(20, settings->height - 30);
	}

	void drawPerNodeConfig(Camera2D *camera){
		for(auto& n : gm->nodes | std::ranges::views::values | std::views::filter([](Node n){ return n.config; })){
			Vector2 np = GetWorldToScreen2D((Vector2){n.position.x, n.position.y}, *camera);
			
			n.config = !GuiWindowBox((Rectangle){np.x + 10, np.y + 10, 240, 440}, "config");

			int option = n.distributionMode;
			GuiToggleGroup((Rectangle){ np.x + 20, np.y + 40, 73, 20 }, "GEO;RR;RAND", &option);
			n.distributionMode = (Distribution)option;
		
			GuiLabel((Rectangle){np.x + 20, np.y + 65, 220, 20}, "SIGNALS");
			if(GuiButton((Rectangle){np.x + 20, np.y + 90, 220, 20}, "ADD SIGNAL")){
				sm->addSignalAtNode(n.id);
			}

			int y_midi = 120;
			GuiLabel((Rectangle){np.x + 20, np.y + y_midi, 220, 20}, "MIDI OUT");

			int i = 0;
			for(auto& mo : n.midi_outs){
				GuiCheckBox((Rectangle){np.x + 20, np.y + y_midi + 20 + i*20.0f, 20, 20}, mo.second.name.c_str(), &mo.second.enabled);
				i++;
			}
			
			int y_channel = y_midi + 30 + i*20;
			GuiCheckBox((Rectangle){np.x + 20, np.y + y_channel, 20, 20}, "NOTE", &n.send_note);
			GuiCheckBox((Rectangle){np.x + 90, np.y + y_channel, 20, 20}, "CC", &n.send_cc);
			GuiCheckBox((Rectangle){np.x + 160, np.y + y_channel, 20, 20}, "PITCH", &n.send_pitch_bend);
			
			int y_cc_note = y_channel + 120;
			switch(settings->mode){
				case 0:{
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note - 75, 220, 20}, "NOTE CHANNEL");
					GuiToggleGroup((Rectangle){np.x + 20, np.y + y_cc_note - 50, 25, 20}, "1;2;3;4;5;6;7;8\n9;10;11;12;13;14;15;16", &n.midi_note_channel);
		
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note , 100, 20}, "NOTE");
					float v = n.midi_note;
					GuiLabel((Rectangle){np.x + 210, np.y + y_cc_note , 20, 20}, sm->midi.midiNotes[(int)round(v)].c_str());
					GuiLabel((Rectangle){np.x + 160, np.y + y_cc_note , 40, 20}, std::format("{}",n.midi_note).c_str());
					GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 30, 220, 20}, "", "", &v, 0, 127);
					n.midi_note = round(v);
					
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note + 60, 100, 20}, "VELOCITY");
					v = n.midi_velocity;
					GuiLabel((Rectangle){np.x + 210, np.y + y_cc_note + 60, 20, 20}, std::format("{}",n.midi_velocity).c_str());
					GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 90, 220, 20}, "", "", &v, 0, 127);
					n.midi_velocity = round(v);
					
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note + 120, 100, 20}, "GATE");
					v = n.midi_gate;
					GuiLabel((Rectangle){np.x + 210, np.y + y_cc_note + 120, 20, 20}, std::format("{}",(float)n.midi_gate/1000.0f).c_str());
					GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 150, 220, 20}, "", "", &v, 0, 30000);
					n.midi_gate = round(v);
				break;}
				case 1:
					{
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note - 75, 220, 20}, "CC CHANNEL");
					GuiToggleGroup((Rectangle){np.x + 20, np.y + y_cc_note - 50, 25, 20}, "1;2;3;4;5;6;7;8\n9;10;11;12;13;14;15;16", &n.midi_cc_channel);
					
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note , 100, 20}, "CC");
					float v = n.midi_cc;
					GuiLabel((Rectangle){np.x + 160, np.y + y_cc_note , 40, 20}, std::format("{}",n.midi_cc).c_str());
					GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 30, 220, 20}, "", "", &v, 0, 127);
					n.midi_cc = round(v);
					
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note + 60, 100, 20}, "CC VALUE");
					v = n.midi_cc_value;
					GuiLabel((Rectangle){np.x + 210, np.y + y_cc_note + 60, 20, 20}, std::format("{}",n.midi_cc_value).c_str());
					GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 90, 220, 20}, "", "", &v, 0, 127);
					n.midi_cc_value = round(v);
					
					break;}
				case 2:
					{
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note - 75, 220, 20}, "PITCH BEND CHANNEL");
					GuiToggleGroup((Rectangle){np.x + 20, np.y + y_cc_note - 50, 25, 20}, "1;2;3;4;5;6;7;8\n9;10;11;12;13;14;15;16", &n.midi_pitch_bend_channel);
					
					GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note , 100, 20}, "CC");
					float v = n.midi_pitch_bend;
					GuiLabel((Rectangle){np.x + 160, np.y + y_cc_note , 40, 20}, std::format("{}",n.midi_pitch_bend).c_str());
					GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 30, 220, 20}, "", "", &v, 0, 16383);
					n.midi_pitch_bend = round(v);
					break;}
			}
		}
	}

	void draw(Vec2D<float> *mouse_pos, Camera2D *camera){
		drawSideBar();
		drawPerNodeConfig(camera);
	}

};