dust3d/src/normalanddepthmapsgenerator.cpp

#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;
}