RT_GPU/includes/RT/Clusterizer.hpp

/* ************************************************************************** */
/*                                                                            */
/*                                                        :::      ::::::::   */
/*   Clusterizer.hpp                                    :+:      :+:    :+:   */
/*                                                    +:+ +:+         +:+     */
/*   By: tomoron <tomoron@student.42angouleme.fr>   +#+  +:+       +#+        */
/*                                                +#+#+#+#+#+   +#+           */
/*   Created: 2025/02/20 18:25:18 by tomoron           #+#    #+#             */
/*   Updated: 2025/02/23 22:41:25 by tomoron          ###   ########.fr       */
/*                                                                            */
/* ************************************************************************** */

#ifndef CLUSTERIZER_HPP
# define CLUSTERIZER_HPP

# include "RT.hpp"

typedef enum e_job_status
{
	WAITING,
	IN_PROGRESS,
	DONE
} t_job_status;

typedef struct s_job
{
	std::string	scene;
	glm::vec3	pos;
	glm::vec2	dir;
	size_t		samples;

	size_t			id;
} t_job;

typedef struct s_client
{
	std::vector<uint8_t> buffer;
	t_job				*curJob;
	int					progress;
	bool				ready;
}	t_client;

typedef enum e_msg
{
	RDY,
	JOB,
	JOB_RES_RQ,
	ACK,
	WAIT,
	IMAGE,
	ERR,
	UNKNOWN
} t_msg;

class Clusterizer
{
	public:
		Clusterizer(Arguments &args);
		~Clusterizer();

		void	update(void);
		bool	getError(void);
		void	imguiRender(void);
		bool	isServer(void);	
		bool	hasJobs(void);

		void	addJob(glm::vec3 pos, glm::vec2 dir, size_t samples);

	private:
		bool				_isActive;
		bool				_isServer;
		bool				_error;
		std::string			_sceneName;

		std::vector<t_job *> _jobs[3];

		void	imguiJobStat(void);
		void	imguiClients(void);

	private: //client
		void initClient(std::string &dest);
		void openClientConnection(const char *ip, int port);
		void clientHandleBuffer(void);
		void updateClient(void);
		void clientGetJob(void);
		void clientReceive(void);

		int						_serverFd;
		std::string				_serverIp;
		int						_serverPort;
		std::vector<uint8_t>	_receiveBuffer;
		t_job					_currentJob;

	private: //server
		void	initServer(std::string port);
		void	updateServer(void);

		void	initServerSocket(int port);
		int		acceptClients(void);
		void	updatePollfds(void);
		int		updateBuffer(int fd);
		void	handleBuffer(int fd, std::vector<uint8_t> &buf);
		void	deleteClient(int fd);

		int		dispatchJobs(void);


		

		int							_serverSocket;
		struct pollfd				*_pollfds;
		std::map<int, t_client>		_clients;
		size_t						_curId;
};

#endif