tahoma2d/toonz/sources/toonzfarm/tfarmclient/application.cpp

#include "application.h"
#include "tfarmcontroller.h"
#include "ttcpip.h"
#include "tfilepath.h"
#include "tsystem.h"
#include "tthread.h"
#include "tenv.h"
#include "tconvert.h"
#include "tfilepath_io.h"
#include <vector>
#include <set>
#include <assert.h>
#include <fstream>
using namespace std;

//===================================================================

namespace
{

TFilePath getGlobalRoot()
{
#ifdef WIN32
	TFilePath groot(TSystem::getSystemValue(
		TFilePath("SOFTWARE\\Digital Video\\ToonzFarm\\1.0\\GLOBALROOT")));

	return groot;
#else
	TFilePath name = "TFARMGLOBALROOT";
	char *s = getenv(name.getFullPath().c_str());
	TFilePath rootDir = string(s ? s : "");
	return rootDir;
#endif
}
};

class Application::Imp
{
public:
	Imp() : m_farmController(0) {}
	~Imp() {}

	void loadControllerData();

	TFarmController *m_farmController;
	TFilePath m_currentFolder;

	ControllerData m_controllerData;
};

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

void Application::Imp::loadControllerData()
{
	TFilePath groot = getGlobalRoot();
	TFilePath fp = groot + "config" + "controller.txt";

	::loadControllerData(fp, m_controllerData);
}

//===================================================================

namespace
{

//===================================================================

Application *theApp;
bool programEnded = false;

class Cleanup
{
public:
	~Cleanup()
	{
		delete theApp;
		theApp = 0;
		programEnded = true;
	}
} cleanup;

//===================================================================

} // end of anonymous namespace

//===================================================================

Application::Application() : m_imp(new Imp)
{
}

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

Application::~Application()
{
	delete m_imp;
}

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

Application *Application::instance()
{
	assert(!programEnded);
	if (!theApp) {
		static TThread::Mutex AppMutex;
		TThread::ScopedLock sl(AppMutex);
		if (programEnded)
			return 0;

		if (!theApp)
			theApp = new Application;
	}
	return theApp;
}

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

TFarmController *Application::getController()
{
	return m_imp->m_farmController;
}

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

bool Application::testControllerConnection() const
{
	TTcpIpClient client;

	int sock;
	int ret = client.connect(m_imp->m_controllerData.m_hostName,
							 m_imp->m_controllerData.m_ipAddress,
							 m_imp->m_controllerData.m_port,
							 sock);
	if (ret == OK) {
		client.disconnect(sock);
		return true;
	}
	return false;
}

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

void Application::getControllerData(string &hostName, string &ipAddr, int &port) const
{
	hostName = m_imp->m_controllerData.m_hostName;
	ipAddr = m_imp->m_controllerData.m_ipAddress;
	port = m_imp->m_controllerData.m_port;
}

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

void Application::init()
{
	m_imp->loadControllerData();

	TFarmControllerFactory factory;
	int ret = factory.create(m_imp->m_controllerData, &m_imp->m_farmController);
}

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

void Application::setCurrentFolder(const TFilePath &fp)
{
	m_imp->m_currentFolder = fp;
}

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

TFilePath Application::getCurrentFolder()
{
	return m_imp->m_currentFolder;
}