78 lines
2.3 KiB
C++
78 lines
2.3 KiB
C++
#include <QDebug>
|
|
#include <QGuiApplication>
|
|
#include "normalanddepthmapsgenerator.h"
|
|
|
|
NormalAndDepthMapsGenerator::NormalAndDepthMapsGenerator(ModelWidget *modelWidget)
|
|
{
|
|
m_viewPortSize = QSize(modelWidget->widthInPixels(),
|
|
modelWidget->heightInPixels());
|
|
m_normalMapRender = createOfflineRender(modelWidget, 1);
|
|
m_depthMapRender = createOfflineRender(modelWidget, 2);
|
|
}
|
|
|
|
void NormalAndDepthMapsGenerator::updateMesh(Model *mesh)
|
|
{
|
|
if (nullptr == mesh) {
|
|
m_normalMapRender->updateMesh(nullptr);
|
|
m_depthMapRender->updateMesh(nullptr);
|
|
return;
|
|
}
|
|
m_normalMapRender->updateMesh(new Model(*mesh));
|
|
m_depthMapRender->updateMesh(mesh);
|
|
}
|
|
|
|
void NormalAndDepthMapsGenerator::setRenderThread(QThread *thread)
|
|
{
|
|
m_normalMapRender->setRenderThread(thread);
|
|
m_depthMapRender->setRenderThread(thread);
|
|
}
|
|
|
|
ModelOffscreenRender *NormalAndDepthMapsGenerator::createOfflineRender(ModelWidget *modelWidget, int purpose)
|
|
{
|
|
ModelOffscreenRender *offlineRender = new ModelOffscreenRender(modelWidget->format());
|
|
offlineRender->setXRotation(modelWidget->xRot());
|
|
offlineRender->setYRotation(modelWidget->yRot());
|
|
offlineRender->setZRotation(modelWidget->zRot());
|
|
offlineRender->setEyePosition(modelWidget->eyePosition());
|
|
offlineRender->setMoveToPosition(modelWidget->moveToPosition());
|
|
offlineRender->setRenderPurpose(purpose);
|
|
return offlineRender;
|
|
}
|
|
|
|
NormalAndDepthMapsGenerator::~NormalAndDepthMapsGenerator()
|
|
{
|
|
delete m_normalMapRender;
|
|
delete m_depthMapRender;
|
|
delete m_normalMap;
|
|
delete m_depthMap;
|
|
}
|
|
|
|
void NormalAndDepthMapsGenerator::generate()
|
|
{
|
|
m_normalMap = new QImage(m_normalMapRender->toImage(m_viewPortSize));
|
|
m_depthMap = new QImage(m_depthMapRender->toImage(m_viewPortSize));
|
|
}
|
|
|
|
void NormalAndDepthMapsGenerator::process()
|
|
{
|
|
generate();
|
|
m_normalMapRender->setRenderThread(QGuiApplication::instance()->thread());
|
|
m_depthMapRender->setRenderThread(QGuiApplication::instance()->thread());
|
|
this->moveToThread(QGuiApplication::instance()->thread());
|
|
emit finished();
|
|
}
|
|
|
|
QImage *NormalAndDepthMapsGenerator::takeNormalMap()
|
|
{
|
|
QImage *normalMap = m_normalMap;
|
|
m_normalMap = nullptr;
|
|
return normalMap;
|
|
}
|
|
|
|
QImage *NormalAndDepthMapsGenerator::takeDepthMap()
|
|
{
|
|
QImage *depthMap = m_depthMap;
|
|
m_depthMap = nullptr;
|
|
return depthMap;
|
|
}
|