59 lines
1.5 KiB
C++
59 lines
1.5 KiB
C++
#include <QtConcurrent>
|
|
#include "ipcpreviewworker.h"
|
|
#include "previewgeneratormapfunctor.h"
|
|
IPCPreviewWorker::IPCPreviewWorker(QLocalSocket *peer)
|
|
{
|
|
this->peer = peer;
|
|
this->connect(&previewWorkerWatcher, &QFutureWatcher<QByteArray>::resultReadyAt, this,
|
|
[this](int index)
|
|
{
|
|
if(this->peer != nullptr)
|
|
{
|
|
QDataStream stream{this->peer};
|
|
stream << previewWorkerWatcher.resultAt(index);
|
|
this->peer->flush();
|
|
}
|
|
});
|
|
connect(&previewWorkerWatcher, &QFutureWatcher<QByteArray>::finished, this, &IPCPreviewWorker::shutdownSocket);
|
|
connect(this->peer, &QLocalSocket::disconnected, this, &IPCPreviewWorker::shutdownSocket);
|
|
}
|
|
|
|
void IPCPreviewWorker::shutdownSocket()
|
|
{
|
|
if(cleaned)
|
|
{
|
|
return;
|
|
}
|
|
cleaned = true;
|
|
if(this->peer != nullptr)
|
|
{
|
|
if(this->peer->state() == QLocalSocket::ConnectedState)
|
|
{
|
|
this->peer->flush();
|
|
this->peer->waitForBytesWritten();
|
|
this->peer->disconnectFromServer();
|
|
if(this->peer->state() != QLocalSocket::UnconnectedState)
|
|
{
|
|
this->peer->waitForDisconnected();
|
|
}
|
|
}
|
|
delete this->peer;
|
|
this->peer = nullptr;
|
|
}
|
|
emit finished();
|
|
}
|
|
|
|
void IPCPreviewWorker::start(RenderConfig config, const QVector<RenderTarget> &targets)
|
|
{
|
|
auto mapFunctor = PreviewGeneratorMapFunctor();
|
|
mapFunctor.setRenderConfig(config);
|
|
|
|
previewWorkerWatcher.setFuture(QtConcurrent::mapped(targets, mapFunctor));
|
|
}
|
|
|
|
void IPCPreviewWorker::stop()
|
|
{
|
|
previewWorkerWatcher.cancel();
|
|
previewWorkerWatcher.waitForFinished();
|
|
}
|