tahoma2d/toonz/sources/common/tipc/tipcmsg.cpp

//Toonz includes
#include "traster.h"
#include "timage_io.h"

//Qt includes
#include <QIODevice>
#include <QSharedMemory>
#include <QTemporaryFile>
#include <QDir>
#include <QFile>
#include <QCoreApplication>

//tipc includes
#include "tipc.h"

#include "tipcmsg.h"

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

//  Local stuff

namespace
{
QHash<QString, QSharedMemory *> sharedMemories;
QHash<QString, QString> temporaryFiles;
}

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

namespace tipc
{

//*******************************************************************************
//    Shared Memory Request
//*******************************************************************************

template <>
QString DefaultMessageParser<SHMEM_REQUEST>::header() const { return QString("$shmem_request"); }

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

template <>
void DefaultMessageParser<SHMEM_REQUEST>::operator()(Message &msg)
{
	int size;
	QString id;
	msg >> id >> size >> clr;

	QSharedMemory *mem = new QSharedMemory(id);

	bool ok = (tipc::create(*mem, size) > 0);
	if (!ok) {
		msg << QString("err");
		delete mem;
		return;
	}

	sharedMemories.insert(id, mem);
	msg << QString("ok");
}

//*******************************************************************************
//    Shared Memory Release
//*******************************************************************************

template <>
QString DefaultMessageParser<SHMEM_RELEASE>::header() const { return QString("$shmem_release"); }

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

template <>
void DefaultMessageParser<SHMEM_RELEASE>::operator()(Message &msg)
{
	QString id;
	msg >> id >> clr;
	QSharedMemory *mem = sharedMemories.take(id);
	if (mem)
		delete mem;
	msg << QString("ok");
}

//*******************************************************************************
//    Temporary File Request
//*******************************************************************************

template <>
QString DefaultMessageParser<TMPFILE_REQUEST>::header() const { return QString("$tmpfile_request"); }

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

template <>
void DefaultMessageParser<TMPFILE_REQUEST>::operator()(Message &msg)
{
	QString id;
	msg >> id >> clr;

	//Build a temporary file with passed id group.
	//The created QTemporaryFile CANNOT be stored directly, as it internally
	//keeps the file open until the object is destroyed. Instead, we store its
	//filePath and manually remove it upon release.

	QTemporaryFile tmp(QDir::temp().filePath(id));
	tmp.setAutoRemove(false);
	if (!tmp.open()) {
		msg << QString("err");
		return;
	}

	temporaryFiles.insert(id, tmp.fileName());
	msg << QString("ok") << tmp.fileName();
}

//*******************************************************************************
//    Temporary File Release
//*******************************************************************************

template <>
QString DefaultMessageParser<TMPFILE_RELEASE>::header() const { return QString("$tmpfile_release"); }

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

template <>
void DefaultMessageParser<TMPFILE_RELEASE>::operator()(Message &msg)
{
	QString id;
	msg >> id >> clr;

	QString tmpPath = temporaryFiles.take(id);
	if (!tmpPath.isEmpty()) {
		QFile file(tmpPath);
		file.remove();
	}

	msg << QString("ok");
}

//*******************************************************************************
//    Quit On Error
//*******************************************************************************

template <>
QString DefaultMessageParser<QUIT_ON_ERROR>::header() const { return QString("$quit_on_error"); }

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

template <>
void DefaultMessageParser<QUIT_ON_ERROR>::operator()(Message &msg)
{
	QObject::connect(socket(), SIGNAL(error(QLocalSocket::LocalSocketError)), QCoreApplication::instance(), SLOT(quit()));
	//In Qt 5.5 originating process's termination emits 'disconnected' instead of 'error'
	QObject::connect(socket(), SIGNAL(disconnected()), QCoreApplication::instance(), SLOT(quit()));

	msg << clr << QString("ok");
}

//*******************************************************************************
//    Explicit template instantiation for all basic datatypes
//*******************************************************************************

//-------------------------- Default Messages ---------------------------

template class DefaultMessageParser<SHMEM_REQUEST>;
template class DefaultMessageParser<SHMEM_RELEASE>;
template class DefaultMessageParser<TMPFILE_REQUEST>;
template class DefaultMessageParser<TMPFILE_RELEASE>;
template class DefaultMessageParser<QUIT_ON_ERROR>;

} //namespace tipc