#pragma once

#ifndef TFARMCONTROLLER_H

#define TFARMCONTROLLER_H

#include <vector></vector>

using std::vector;

#include "tfarmtask.h"

//#include "texception.h"

#include "tconvert.h"

//#include "tfilepath.h"

class TFilePath;

#ifdef TFARMAPI

#undef TFARMAPI

#endif

#ifdef _WIN32

#ifdef TFARM_EXPORTS

#define TFARMAPI __declspec(dllexport)

#else

#define TFARMAPI __declspec(dllimport)

#endif

#else

#define TFARMAPI

#endif

//------------------------------------------------------------------------------

enum ServerState {

	Ready,

	Busy,

	NotResponding,

	Down,

	Offline,

	ServerUnknown

};

class ServerIdentity

{

public:

	ServerIdentity(const QString &id, const QString &name)

		: m_id(id), m_name(name) {}

	QString m_id;

	QString m_name;

};

class ServerInfo

{

public:

	QString m_name;

	QString m_ipAddress;

	QString m_portNumber;

	ServerState m_state;

	QString m_platform;

	int m_cpuCount;

	unsigned int m_totPhysMem;

	unsigned int m_totVirtMem;

	// the following fields can be used just if the server state is not Down

	unsigned int m_availPhysMem;

	unsigned int m_availVirtMem;

	QString m_currentTaskId;

};

//------------------------------------------------------------------------------

class TaskShortInfo

{

public:

	TaskShortInfo(const QString &id, const QString &name, TaskState status)

		: m_id(id), m_name(name), m_status(status) {}

	QString m_id;

	QString m_name;

	TaskState m_status;

};

//------------------------------------------------------------------------------

class TFARMAPI TFarmController

{

public:

	virtual ~TFarmController() {}

	virtual QString addTask(const TFarmTask &task, bool suspended) = 0;

	virtual void removeTask(const QString &id) = 0;

	virtual void suspendTask(const QString &id) = 0;

	virtual void activateTask(const QString &id) = 0;

	virtual void restartTask(const QString &id) = 0;

	virtual void getTasks(vector<qstring> &tasks) = 0;</qstring>

	virtual void getTasks(const QString &parentId, vector<qstring> &tasks) = 0;</qstring>

	virtual void getTasks(const QString &parentId, vector<taskshortinfo> &tasks) = 0;</taskshortinfo>

	virtual void queryTaskInfo(const QString &id, TFarmTask &task) = 0;

	virtual void queryTaskShortInfo(

		const QString &id,

		QString &parentId,

		QString &name,

		TaskState &status) = 0;

	// used (by a server) to notify a server start

	virtual void attachServer(const QString &name, const QString &addr, int port) = 0;

	// used (by a server) to notify a server stop

	virtual void detachServer(const QString &name, const QString &addr, int port) = 0;

	// used by a server to notify a task submission error

	virtual void taskSubmissionError(const QString &taskId, int errCode) = 0;

	// used by a server to notify a task progress

	virtual void taskProgress(

		const QString &taskId,

		int step,

		int stepCount,

		int frameNumber,

		FrameState state) = 0;

	// used by a server to notify a task completion

	virtual void taskCompleted(const QString &taskId, int exitCode) = 0;

	// fills the servers vector with the identities of the servers

	virtual void getServers(vector<serveridentity> &servers) = 0;</serveridentity>

	// returns the state of the server whose id has been specified

	virtual ServerState queryServerState2(const QString &id) = 0;

	// fills info with the infoes about the server whose id is specified

	virtual void queryServerInfo(const QString &id, ServerInfo &info) = 0;

	// activates the server whose id has been specified

	virtual void activateServer(const QString &id) = 0;

	// deactivates the server whose id has been specified

	// once deactivated, a server is not available for task rendering

	virtual void deactivateServer(const QString &id, bool completeRunningTasks = true) = 0;

};

//------------------------------------------------------------------------------

class TFARMAPI ControllerData

{

public:

	ControllerData(const QString &hostName = "", const QString &ipAddr = "", int port = 0)

		: m_hostName(hostName), m_ipAddress(ipAddr), m_port(port)

	{

	}

	bool operator==(const ControllerData &rhs)

	{

		return m_hostName == rhs.m_hostName &&

			   m_ipAddress == rhs.m_ipAddress &&

			   m_port == rhs.m_port;

	}

	QString m_hostName;

	QString m_ipAddress;

	int m_port;

};

//------------------------------------------------------------------------------

class TFARMAPI TFarmControllerFactory

{

public:

	TFarmControllerFactory();

	int create(const ControllerData &data, TFarmController **controller);

	int create(const QString &hostname, int port, TFarmController **controller);

};

//------------------------------------------------------------------------------

void TFARMAPI loadControllerData(const TFilePath &fp, ControllerData &data);

#endif