VisualNodeSocket.h

#pragma once

#include "VisualNodeCore.h"

namespace VisNodeSys
{
#define NODE_SOCKET_SIZE 5.0f
#define DEFAULT_NODE_SOCKET_COLOR ImColor(150, 150, 150)
#define DEFAULT_CONNECTION_COLOR ImColor(200, 200, 200)
#define DEFAULT_NODE_SOCKET_MOUSE_HOVERED_CONNECTION_COLOR ImColor(220, 220, 220)

	class Node;

	class VISUAL_NODE_SYSTEM_API NodeSocket
	{
		friend class NodeSystem;
		friend class NodeArea;
		friend Node;

		std::string ID;
		bool bOutput = false;
		std::string Type;
		std::string Name;
		std::vector<NodeSocket*> ConnectedSockets;

		static std::unordered_map<std::string, ImColor> SocketTypeToColorAssosiations;

		std::function<void* ()> OutputData = []() { return nullptr; };
	protected:
		Node* Parent = nullptr;
	public:
		NodeSocket(Node* Parent, std::string Type, std::string Name, bool bOutput = false, std::function<void* ()> OutputDataFunction = []() { return nullptr; });

		Node* GetParent() const;
		std::vector<NodeSocket*> GetConnectedSockets();

		std::string GetID() const;
		std::string GetName() const;

		std::string GetType() const;

		bool isOutput() const { return bOutput; }
		bool isInput() const { return !bOutput; }

		void SetFunctionToOutputData(std::function<void* ()> NewFunction);
		void* GetData() { return OutputData(); }
	};

	struct ConnectionStyle
	{
		ImColor ForceColor = DEFAULT_CONNECTION_COLOR;

		bool bMarchingAntsEffect = false;
		float MarchingAntsSpeed = 1.0f;
		bool bMarchingAntsReverseDirection = false;

		bool bPulseEffect = false;
		float PulseSpeed = 1.0f;
		float PulseMin = 0.1f;
		float PulseMax = 1.0f;
	};

	class NodeArea;
	class Connection;

	class RerouteNode
	{
		friend class NodeSystem;
		friend class NodeArea;
		friend class Connection;
		friend Node;

		std::string ID;
		Connection* Parent = nullptr;
		ImVec2 Position;

		NodeSocket* BeginSocket = nullptr;
		RerouteNode* BeginReroute = nullptr;

		NodeSocket* EndSocket = nullptr;
		RerouteNode* EndReroute = nullptr;

		bool bHovered = false;
		bool bSelected = false;

		RerouteNode() {};
	};

	struct ConnectionSegment
	{
		ImVec2 Begin;
		ImVec2 End;

		NodeSocket* BeginSocket = nullptr;
		RerouteNode* BeginReroute = nullptr;

		NodeSocket* EndSocket = nullptr;
		RerouteNode* EndReroute = nullptr;
	};

	class Connection
	{
		friend class NodeSystem;
		friend class NodeArea;
		friend Node;

		NodeSocket* Out = nullptr;
		NodeSocket* In = nullptr;

		bool bHovered = false;
		bool bSelected = false;
		ConnectionStyle Style;

		std::vector<RerouteNode*> RerouteNodes;

		Connection(NodeSocket* Out, NodeSocket* In);
		~Connection();
	};
}