tahoma2d/toonz/sources/image/3gp/tiio_3gp_proxy.cpp
Shinya Kitaoka d1f6c4e95b REFACTORING: Add final specifiers (#537)
* add final specifiers

* apply clang-format

* fix for macOS
2016-06-29 15:17:12 +09:00

435 lines
13 KiB
C++

#if (defined(x64) || defined(__LP64__) || defined(LINUX))
// 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 final : 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 final : 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