brainrenderer.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BRAINRENDERER_H
#define BRAINRENDERER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include "../core/rendertypes.h"
 
#include <QMatrix4x4>
#include <QVector3D>
#include <memory>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
class QRhi;
class QRhiCommandBuffer;
class QRhiRenderTarget;
class QRhiRenderPassDescriptor;
class QRhiResourceUpdateBatch;
class QRhiTexture;
class QRhiRenderBuffer;
class QRhiTextureRenderTarget;
 
namespace DISP3DLIB
{
 
class BrainSurface;
class DipoleObject;
class ElectrodeObject;
class NetworkObject;
class PolylineObject;
class VideoOverlay;
class SliceObject;
 
//=============================================================================================================
class DISP3DSHARED_EXPORT BrainRenderer
{
public:
    //=========================================================================================================
    BrainRenderer();
 
    //=========================================================================================================
    ~BrainRenderer();
 
    // ShaderMode is defined in core/rendertypes.h for lightweight inclusion.
    // These aliases preserve backward compatibility.
    using ShaderMode = DISP3DLIB::ShaderMode;
    static constexpr ShaderMode Standard = DISP3DLIB::Standard;
    static constexpr ShaderMode Holographic = DISP3DLIB::Holographic;
    static constexpr ShaderMode Anatomical = DISP3DLIB::Anatomical;
    static constexpr ShaderMode Dipole = DISP3DLIB::Dipole;
    static constexpr ShaderMode XRay = DISP3DLIB::XRay;
    static constexpr ShaderMode ShowNormals = DISP3DLIB::ShowNormals;

    struct SceneData
    {
        QMatrix4x4 mvp;
        QVector3D cameraPos;
        QVector3D lightDir;
        bool lightingEnabled = true;
        float overlayMode = 0.0f; // 0=FsSurface, 1=FsAnnotation, 2=Scientific, 3=STC
    };
 
    //=========================================================================================================
    void initialize(QRhi* rhi, QRhiRenderPassDescriptor* rp, int sampleCount);
 
    //=========================================================================================================
    void ensureRenderTargets(QRhi* rhi, QRhiTexture* colorTex, const QSize& pixelSize);
 
    //=========================================================================================================
    QRhiRenderTarget* rtClear() const;

    QRhiRenderTarget* rtPreserve() const;
 
    //=========================================================================================================
    void newFrame();

    int cachedMeshCount() const;
 
    //=========================================================================================================
    void beginFrame(QRhiCommandBuffer* cb, QRhiResourceUpdateBatch* updates = nullptr);
 
    //=========================================================================================================
    void beginPreservingPass(QRhiCommandBuffer* cb, QRhiResourceUpdateBatch* updates = nullptr);
 
    //=========================================================================================================
    int prepareSurfaceDraw(QRhiResourceUpdateBatch* u, const SceneData& data, BrainSurface* surface);

    void issueSurfaceDraw(QRhiCommandBuffer* cb, BrainSurface* surface, ShaderMode mode, int uniformOffset);
 
    //=========================================================================================================
    int prepareDipoleDraw(QRhiResourceUpdateBatch* u, const SceneData& data, DipoleObject* dipoles);

    void issueDipoleDraw(QRhiCommandBuffer* cb, DipoleObject* dipoles, int uniformOffset);

    int prepareNetworkDraw(QRhiResourceUpdateBatch* u, const SceneData& data, NetworkObject* network);

    void issueNetworkDraw(QRhiCommandBuffer* cb, NetworkObject* network, int uniformOffset);

    int preparePolylineDraw(QRhiResourceUpdateBatch* u, const SceneData& data, PolylineObject* polyline);

    void issuePolylineDraw(QRhiCommandBuffer* cb, PolylineObject* polyline, int uniformOffset);

    int prepareElectrodeDraw(QRhiResourceUpdateBatch* u, const SceneData& data, ElectrodeObject* electrodes);

    void issueElectrodeDraw(QRhiCommandBuffer* cb, ElectrodeObject* electrodes, int uniformOffset);
 
    //=========================================================================================================
    void prepareVideoOverlay(QRhi* rhi,
                             QRhiResourceUpdateBatch* u,
                             DISP3DLIB::VideoOverlay* overlay);
 
    //=========================================================================================================
    int prepareVideoOverlayDraw(QRhiResourceUpdateBatch* u,
                                const SceneData& data,
                                DISP3DLIB::VideoOverlay* overlay,
                                BrainSurface* surface);

    void issueVideoOverlayDraw(QRhiCommandBuffer* cb,
                               DISP3DLIB::VideoOverlay* overlay,
                               BrainSurface* surface,
                               int uniformOffset);
 
    //=========================================================================================================
    void prepareSlice(QRhi* rhi,
                      QRhiResourceUpdateBatch* u,
                      DISP3DLIB::SliceObject* slice,
                      int slotIndex);
 
    //=========================================================================================================
    int prepareSliceDraw(QRhiResourceUpdateBatch* u,
                         const SceneData& data,
                         int slotIndex);
 
    //=========================================================================================================
    void issueSliceDraw(QRhiCommandBuffer* cb,
                        int slotIndex,
                        int uniformOffset);
 
    //=========================================================================================================
    void endPass(QRhiCommandBuffer* cb);
 
private:
    struct Impl;
    std::unique_ptr<Impl> d;
};
 
} // namespace DISP3DLIB
 
#endif // BRAINRENDERER_H