viseq/src/signal-manager.h

#pragma once

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

#include <chrono>
#include <memory>
#include <thread>

using namespace VISEQ::GRAPH;

namespace VISEQ::SIGNAL {

	typedef std::shared_ptr<Signal> SPTR_Signal;
	typedef std::vector<SPTR_Signal> VEC_SPTR_Signal;

	class SignalManager : public std::enable_shared_from_this<SignalManager>{

		public:
			SignalManager() {}
			SignalManager(std::shared_ptr<GraphManager> gm, std::mutex *mutex) : mutex(mutex), gm(gm) {}
			
			std::shared_ptr<SignalManager> getPtr(){
				return shared_from_this();
			}

			VISEQ::MIDI::Midi midi;
			
			std::mutex *mutex;
			std::shared_ptr<GraphManager> gm;

			VEC_SPTR_Signal signals;

			VEC_SPTR_Signal erase_signals;

			bool loop = true;
			bool play = false;

			void pause(){
				play = false;
			}

			void stop(){
				play = false;
				reset();
			}

			void start(){
				play = true;
			}
			
			void reset(){
				for(auto &s : signals){
					s->position = s->start.lock()->position + Vec2Df{0.1, 0.1};
					s->next = s->start;
					s->last = s->start;
				}
				gm->reset();
			}

			void startSignalThread(){
				std::thread t1(&SignalManager::updateSignals, this);
				t1.detach();
			}

			void stopSignalThread(){
				loop = false;
			}
		
			void addSignalAtNode(WPTR_Node node){
				Color c = {(unsigned char)(20 + (rand() % 200)), 100, (unsigned char)(20 + (int)(rand() % 200)), 200};
				signals.emplace_back(std::make_shared<Signal>(node.lock()->position + Vec2Df{0.5f, 0.5f}, 8.0f, 0, c)); // position + 1 or else it does not render. why? no idea!
				signals.back()->next = node;
				signals.back()->last = node;
				signals.back()->start = node;
			}

			void rerouteSignal(VEC_SPTR_Node nodes, VEC_WPTR_Node adjacent_nodes){
				auto reroute_signals = signals | RANGE_FILTER([nodes](SIGNAL::SPTR_Signal s){return std::ranges::contains(nodes, s->next.lock());});

				for(auto s : reroute_signals){
					float dist = 10000000;
					SPTR_Node next_node;
					for(auto n : adjacent_nodes){
						if(n.lock()->position.dist(s->position) < dist) {
							dist = n.lock()->position.dist(s->position);
							next_node = n.lock();
						}
					}
					s->next = next_node;
					s->last = next_node;
				}

			}

			void rebaseSignal(VEC_SPTR_Node nodes){
				auto rebase_signals = signals | RANGE_FILTER([nodes](SIGNAL::SPTR_Signal s){return std::ranges::contains(nodes, s->start.lock());});
				
				for(auto s : rebase_signals){
					s->start = s->next;
				}
			}

			void removeSignal(std::shared_ptr<Signal> s){
				erase_signals.push_back(s);
			}

			void updateSignals(){
				auto last = std::chrono::high_resolution_clock::now();
				while(loop){
					std::chrono::duration<double, std::micro> diff = std::chrono::high_resolution_clock::now() - last;
					if((int)diff.count() >= 250){
						mutex->lock();
						if(play){
							for(auto& s : signals){
								s->direction = (s->next.lock()->position - s->position).norm();
								s->update(diff.count());
								if(s->position.dist(s->next.lock()->position) < 0.6f) {
									WPTR_Node next_node = gm->next(s->next, s->last);
									s->next.lock()->trigger();

									if(next_node.lock() != s->next.lock() && !next_node.expired() && s->next.lock()->getNodeType() == NodeType::MIDINODE) {
										auto midinode = std::dynamic_pointer_cast<MidiNode>(s->next.lock());
										for(auto &m : midinode->midi_outs){
											if(m.second.enabled) {
												if(midinode->send_note) midi.note(m.second.i, midinode->midi_note_channel, (int)Clamp(midinode->midi_note + s->midi_note_offset, 0, 127), midinode->midi_velocity, midinode->midi_gate);	
												if(midinode->send_cc) midi.cc(m.second.i, midinode->midi_cc_channel, midinode->midi_cc, (int)Clamp(midinode->midi_cc_value + s->midi_cc_offset, 0, 127));	
												if(midinode->send_pitch_bend) midi.pitch_bend(m.second.i, midinode->midi_pitch_bend_channel, midinode->midi_pitch_bend);	
											}
										}
										s->last = s->next;
										s->next = next_node;
									}
								}
							}
						}
						midi.update();
						mutex->unlock();
						last = std::chrono::high_resolution_clock::now();
					}
				}
			}

			void drawSignals(Vec2Df mouse_pos, Camera2D camera){
				mutex->lock();
				if(!erase_signals.empty()){
					for(auto s: erase_signals){
						std::erase(signals, s); 	
					}
					erase_signals.clear();
				}
				for(auto s : signals){
					s->draw(mouse_pos, camera);
				}
				mutex->unlock();
			}

	        NLOHMANN_DEFINE_TYPE_INTRUSIVE(SignalManager, signals)
		
	};
}