gpuinterpolationmaterial.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================

#include "gpuinterpolationmaterial.h"

//=============================================================================================================
// QT INCLUDES
//=============================================================================================================

#include <Qt3DRender/QEffect>
#include <Qt3DRender/QParameter>
#include <Qt3DRender/QRenderPass>
#include <Qt3DRender/QFilterKey>
#include <Qt3DRender/QTechnique>
#include <Qt3DRender/QShaderProgram>
#include <Qt3DRender/QGraphicsApiFilter>
#include <QUrl>

//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================

//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================

using namespace DISP3DLIB;
using namespace Qt3DRender;

//=============================================================================================================
// DEFINE GLOBAL METHODS
//=============================================================================================================

//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================

GpuInterpolationMaterial::GpuInterpolationMaterial(bool bUseSortPolicy, Qt3DCore::QNode *parent)
    : AbstractPhongAlphaMaterial(bUseSortPolicy, parent)
    , m_pComputeShader(new QShaderProgram)
    , m_pComputeRenderPass(new QRenderPass)
    , m_pComputeFilterKey(new QFilterKey)
    , m_pComputeTechnique(new QTechnique)
    , m_pDrawShader(new QShaderProgram)
    , m_pDrawRenderPass(new QRenderPass)
    , m_pDrawTechnique(new QTechnique)
    , m_pSignalDataParameter(new QParameter)
    , m_pColsParameter(new QParameter)
    , m_pRowsParameter(new QParameter)
    , m_pInterpolationMatParameter(new QParameter)
    , m_pOutputColorParameter(new QParameter)
    , m_pThresholdXParameter(new QParameter(QStringLiteral("fThresholdX"), 1e-10f))
    , m_pThresholdZParameter(new QParameter(QStringLiteral("fThresholdZ"), 6e-6f))
    , m_pColormapParameter(new QParameter(QStringLiteral("ColormapType"), 3))
{
    init();
    setShaderCode();
}

//=============================================================================================================

void GpuInterpolationMaterial::init()
{
    //Compute part
    //Set OpenGL version
    m_pComputeTechnique->graphicsApiFilter()->setApi(QGraphicsApiFilter::OpenGL);
    m_pComputeTechnique->graphicsApiFilter()->setMajorVersion(4);
    m_pComputeTechnique->graphicsApiFilter()->setMinorVersion(3);
    m_pComputeTechnique->graphicsApiFilter()->setProfile(QGraphicsApiFilter::CoreProfile);

    //Set filter Keys
    m_pComputeFilterKey->setName(QStringLiteral("renderingStyle"));
    m_pComputeFilterKey->setValue(QStringLiteral("compute"));

    //Add to technique
    m_pComputeTechnique->addFilterKey(m_pComputeFilterKey);
    m_pComputeTechnique->addRenderPass(m_pComputeRenderPass);

    //Set default Interpolation matrix parameters
    m_pColsParameter->setName(QStringLiteral("cols"));
    m_pColsParameter->setValue(1);
    m_pRowsParameter->setName(QStringLiteral("rows"));
    m_pRowsParameter->setValue(1);
    m_pInterpolationMatParameter->setName(QStringLiteral("InterpolationMat"));

    //Set default output
    m_pOutputColorParameter->setName(QStringLiteral("OutputColor"));

    //Set default input
    m_pSignalDataParameter->setName(QStringLiteral("InputVec"));

    //Add compute Parameter
    m_pComputeRenderPass->addParameter(m_pColsParameter);
    m_pComputeRenderPass->addParameter(m_pRowsParameter);
    m_pComputeRenderPass->addParameter(m_pOutputColorParameter);
    m_pComputeRenderPass->addParameter(m_pInterpolationMatParameter);
    m_pComputeRenderPass->addParameter(m_pSignalDataParameter);

    //Add Threshold parameter
    m_pComputeRenderPass->addParameter(m_pThresholdXParameter);
    m_pComputeRenderPass->addParameter(m_pThresholdZParameter);

    //Add ColormapType
    m_pComputeRenderPass->addParameter(m_pColormapParameter);

    //Draw part
    //Add Phongalpha parameter
    m_pDrawRenderPass->addParameter(m_pDiffuseParameter);
    m_pDrawRenderPass->addParameter(m_pSpecularParameter);
    m_pDrawRenderPass->addParameter(m_pShininessParameter);
    m_pDrawRenderPass->addParameter(m_pAlphaParameter);

    //Set OpenGL version
    m_pDrawTechnique->graphicsApiFilter()->setApi(QGraphicsApiFilter::OpenGL);
    m_pDrawTechnique->graphicsApiFilter()->setMajorVersion(4);
    m_pDrawTechnique->graphicsApiFilter()->setMinorVersion(3);
    m_pDrawTechnique->graphicsApiFilter()->setProfile(QGraphicsApiFilter::CoreProfile);

    //Add to technique
    m_pDrawTechnique->addFilterKey(m_pDrawFilterKey);
    m_pDrawTechnique->addRenderPass(m_pDrawRenderPass);

    //Effect
    //Link shader and uniforms
    m_pEffect->addTechnique(m_pComputeTechnique);
    m_pEffect->addTechnique(m_pDrawTechnique);

    //Add to material
    this->setEffect(m_pEffect);
}

//=============================================================================================================

void GpuInterpolationMaterial::setShaderCode()
{
    m_pComputeShader->setComputeShaderCode(QShaderProgram::loadSource(QUrl(QStringLiteral("qrc:/engine/model/materials/shaders/compute/interpolation.csh"))));
    m_pComputeRenderPass->setShaderProgram(m_pComputeShader);

    m_pDrawShader->setVertexShaderCode(QShaderProgram::loadSource(QUrl(QStringLiteral("qrc:/engine/model/materials/shaders/compute/interpolation.vert"))));
    m_pDrawShader->setFragmentShaderCode(QShaderProgram::loadSource(QUrl(QStringLiteral("qrc:/engine/model/materials/shaders/compute/interpolation.frag"))));
    m_pDrawRenderPass->setShaderProgram(m_pDrawShader);
}

//=============================================================================================================