#if (defined(x64) || defined(__LP64__))

//Toonz includes
#include "tfilepath.h"
#include "trasterimage.h"
#include "tstream.h"
#include "timageinfo.h"
#include "trop.h"
#include "tsound.h"

//tipc includes
#include "tipc.h"
#include "t32bitsrv_wrap.h"

//Qt includes
#include <QSharedMemory>
#include <QMutexLocker>
#include <QDataStream>

#include "tiio_3gp_proxy.h"

/*
  For a list of supported commands through the 32-bit background server,
  see the related "t32libserver" project.
*/

//******************************************************************************
//    Generic stuff implementation
//******************************************************************************

/*NOT PRESENT: Inherits the MOV stuff*/

//******************************************************************************
//    TImageWriter3gp Proxy implementation
//******************************************************************************

class TImageWriter3gpProxy : public TImageWriter
{
	TLevelWriter3gp *m_lw;

public:
	int m_frameIndex;

public:
	TImageWriter3gpProxy(const TFilePath &fp, int frameIndex, TLevelWriter3gp *lw);
	~TImageWriter3gpProxy();

	bool is64bitOutputSupported() { return false; }
	void save(const TImageP &);

private:
	//not implemented
	TImageWriter3gpProxy(const TImageWriter3gpProxy &);
	TImageWriter3gpProxy &operator=(const TImageWriter3gpProxy &src);
};

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

TImageWriter3gpProxy::TImageWriter3gpProxy(const TFilePath &fp, int frameIndex, TLevelWriter3gp *lw)
	: TImageWriter(fp), m_lw(lw), m_frameIndex(frameIndex)
{
	m_lw->addRef();
}

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

TImageWriter3gpProxy::~TImageWriter3gpProxy()
{
	m_lw->release();
}

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

void TImageWriter3gpProxy::save(const TImageP &img)
{
	m_lw->save(img, m_frameIndex);
}

//******************************************************************************
//    TLevelWriter3gp Proxy implementation
//******************************************************************************

TLevelWriter3gp::TLevelWriter3gp(const TFilePath &path, TPropertyGroup *winfo)
	: TLevelWriter(path, winfo)
{
	static TAtomicVar count;
	unsigned int currCount = ++count;
	m_id = currCount;

	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;

	QString res, propsFp;
	if (winfo) {
		//Request a temporary file to store the infos to
		stream << (msg << QString("$tmpfile_request") << QString("initLW3") + QString::number(currCount));
		if (tipc::readMessage(stream, msg) != "ok")
			goto err;

		msg >> propsFp >> tipc::clr;
		assert(!propsFp.isEmpty());

		TFilePath propsTfp(propsFp.toStdWString());
		{
			TOStream os(propsTfp);
			winfo->saveData(os);
		}
	}

	//Pass fp to the server
	stream << (msg << QString("$initLW3gp") << m_id << QString::fromStdWString(path.getWideString()) << propsFp);
	if (tipc::readMessage(stream, msg) != "ok")
		goto err;

	if (winfo) {
		stream << (msg << tipc::clr << QString("$tmpfile_release")
					   << QString("initLW3gp") + QString::number(currCount));
		if (tipc::readMessage(stream, msg) != "ok")
			goto err;
	}

	return;

err:

	throw TException("Unable to write file");
}

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

TLevelWriter3gp::~TLevelWriter3gp()
{
	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;
	QString res;

	stream << (msg << QString("$closeLW3gp") << m_id);
	if (tipc::readMessage(stream, msg) != "ok")
		throw TException("Unable to write file");
}

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

void TLevelWriter3gp::setFrameRate(double fps)
{
	TLevelWriter::setFrameRate(fps);

	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;
	QString res;

	stream << (msg << QString("$LW3gpSetFrameRate") << m_id << fps);
	if (tipc::readMessage(stream, msg) != "ok")
		throw TException("Unexpected error");
}

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

TImageWriterP TLevelWriter3gp::getFrameWriter(TFrameId fid)
{
	if (fid.getLetter() != 0)
		return TImageWriterP(0);

	int index = fid.getNumber() - 1;
	return new TImageWriter3gpProxy(m_path, index, this);
}

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

void TLevelWriter3gp::save(const TImageP &img, int frameIndex)
{
	TRasterImageP ri(img);
	if (!img)
		throw TImageException(getFilePath(), "Unsupported image type");

	TRasterP ras(ri->getRaster());

	int lx = ras->getLx(), ly = ras->getLy(), pixSize = ras->getPixelSize();
	if (pixSize != 4)
		throw TImageException(getFilePath(), "Unsupported pixel type");

	int size = lx * ly * pixSize;

	//Send messages
	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;

	//Send the write message.
	stream << (msg << QString("$LW3gpImageWrite") << m_id << frameIndex << lx << ly);

	//Send the data through a shared memory segment
	{
		t32bitsrv::RasterExchanger<TPixel32> exch(ras);
		tipc::writeShMemBuffer(stream, msg << tipc::clr, size, &exch);
	}

	if (tipc::readMessage(stream, msg) != "ok")
		throw TImageException(getFilePath(), "Couln't save image");
}

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

void TLevelWriter3gp::saveSoundTrack(TSoundTrack *st)
{
	if (st == 0)
		return;

	//Prepare connection
	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	unsigned int size = st->getSampleSize() * st->getSampleCount();

	//Send the saveSoundTract command to the server
	tipc::Stream stream(&socket);
	tipc::Message msg;

	stream << (msg << QString("$LW3gpSaveSoundTrack") << m_id << st->getSampleRate() << st->getBitPerSample()
				   << st->getChannelCount() << st->getSampleCount() << st->getFormat().m_signedSample);

	t32bitsrv::BufferExchanger exch((UCHAR *)st->getRawData());
	tipc::writeShMemBuffer(stream, msg << tipc::clr, size, &exch);

	QString res(tipc::readMessage(stream, msg));
	assert(res == "ok");
}

//******************************************************************************
//    TImageReaderMov Proxy implementation
//******************************************************************************

class TImageReader3gpProxy : public TImageReader
{
	TLevelReader3gp *m_lr;
	TImageInfo *m_info;

public:
	int m_frameIndex;

public:
	TImageReader3gpProxy(const TFilePath &fp, int frameIndex, TLevelReader3gp *lr, TImageInfo *info);
	~TImageReader3gpProxy() { m_lr->release(); }

	TImageP load();
	void load(const TRasterP &rasP, const TPoint &pos = TPoint(0, 0), int shrinkX = 1, int shrinkY = 1);

	TDimension getSize() const { return m_lr->getSize(); }
	TRect getBBox() const { return m_lr->getBBox(); }

	const TImageInfo *getImageInfo() const { return m_info; }

private:
	//not implemented
	TImageReader3gpProxy(const TImageReader3gpProxy &);
	TImageReader3gpProxy &operator=(const TImageReader3gpProxy &src);
};

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

TImageReader3gpProxy::TImageReader3gpProxy(const TFilePath &fp, int frameIndex,
										   TLevelReader3gp *lr, TImageInfo *info)
	: TImageReader(fp), m_lr(lr), m_frameIndex(frameIndex), m_info(info)
{
	m_lr->addRef();
}

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

TImageP TImageReader3gpProxy::load()
{
	TRaster32P ras(m_lr->getSize());
	m_lr->load(ras, m_frameIndex, TPointI(), 1, 1);
	return TRasterImageP(ras);
}

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

void TImageReader3gpProxy::load(const TRasterP &rasP, const TPoint &pos, int shrinkX, int shrinkY)
{
	//NOTE: The original implementation is different. But is also does not make sense...
	//I've substituted it with the lrm plain call.
	m_lr->load(rasP, m_frameIndex, pos, shrinkX, shrinkY);
}

//******************************************************************************
//    TLevelReader3gp Proxy implementation
//******************************************************************************

TLevelReader3gp::TLevelReader3gp(const TFilePath &path)
	: TLevelReader(path)
{
	static TAtomicVar count;
	unsigned int currCount = ++count;
	m_id = currCount;

	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;

	stream << (msg << QString("$initLR3gp") << m_id << QString::fromStdWString(path.getWideString()));
	if (tipc::readMessage(stream, msg) != "ok")
		throw TImageException(path, "Couldn't open file");

	double frameRate;
	msg >> m_lx >> m_ly >> frameRate >> tipc::clr;

	m_info = new TImageInfo;
	m_info->m_lx = m_lx;
	m_info->m_ly = m_ly;
	m_info->m_frameRate = frameRate;
}

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

TLevelReader3gp::~TLevelReader3gp()
{
	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;

	stream << (msg << QString("$closeLR3gp") << m_id);
	QString res(tipc::readMessage(stream, msg));
	assert(res == "ok");
}

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

TImageReaderP TLevelReader3gp::getFrameReader(TFrameId fid)
{
	if (fid.getLetter() != 0)
		return TImageReaderP(0);

	int index = fid.getNumber() - 1;
	return new TImageReader3gpProxy(m_path, index, this, m_info);
}

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

TLevelP TLevelReader3gp::loadInfo()
{
	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;

	TLevelP level;
	{
		QString shMemId(tipc::uniqueId());

		//Send the appropriate command
		stream << (msg << QString("$LR3gpLoadInfo") << m_id << shMemId);
		if (tipc::readMessage(stream, msg) != "ok")
			goto err;

		int frameCount;

		msg >> frameCount >> tipc::clr;

		//Read the data in the shared memory segment
		QSharedMemory shmem(shMemId);
		shmem.attach();
		shmem.lock();

		int *f, *fBegin = (int *)shmem.data(), *fEnd = fBegin + frameCount;
		assert(fBegin);

		for (f = fBegin; f < fEnd; ++f)
			level->setFrame(*f, TImageP());

		shmem.unlock();
		shmem.detach();

		//Release the shared memory segment
		stream << (msg << QString("$shmem_release") << shMemId);
		if (tipc::readMessage(stream, msg) != "ok")
			goto err;
	}

	return level;

err:

	throw TException("Couldn't read movie data");
	return TLevelP();
}

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

void TLevelReader3gp::enableRandomAccessRead(bool enable)
{
	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;

	stream << (msg << QString("$LR3gpEnableRandomAccessRead") << m_id << QString(enable ? "true" : "false"));
	QString res(tipc::readMessage(stream, msg));
	assert(res == "ok");
}

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

void TLevelReader3gp::load(const TRasterP &ras, int frameIndex, const TPoint &pos,
						   int shrinkX, int shrinkY)
{
	QLocalSocket socket;
	tipc::startSlaveConnection(&socket, t32bitsrv::srvName(), -1, t32bitsrv::srvCmdline());

	tipc::Stream stream(&socket);
	tipc::Message msg;

	unsigned int size = ras->getLx() * ras->getLy() * ras->getPixelSize();

	//Send the appropriate command to the 32-bit server
	stream << (msg << QString("$LR3gpImageRead") << m_id << ras->getLx() << ras->getLy() << ras->getPixelSize()
				   << frameIndex << pos.x << pos.y << shrinkX << shrinkY);

	t32bitsrv::RasterExchanger<TPixel32> exch(ras);
	if (!tipc::readShMemBuffer(stream, msg << tipc::clr, &exch))
		throw TException("Couldn't load image");
}

#endif //x64