networkobject.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef NETWORKOBJECT_H
#define NETWORKOBJECT_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_rhi_global.h"
 
#include <QMatrix4x4>
#include <QVector3D>
#include <QColor>
#include <vector>
#include <memory>
 
#include <connectivity/network/network.h>
 
// Forward-declare QRhi types so that this header stays QRhi-free
class QRhi;
class QRhiBuffer;
class QRhiResourceUpdateBatch;
 
//=============================================================================================================
class DISP3DRHISHARED_EXPORT NetworkObject
{
public:
    //=========================================================================================================
    NetworkObject();
 
    //=========================================================================================================
    ~NetworkObject();
 
    //=========================================================================================================
    void load(const CONNECTIVITYLIB::Network &network, const QString &sColormap = "Viridis");
 
    //=========================================================================================================
    void setThreshold(double dThreshold);
 
    //=========================================================================================================
    void setColormap(const QString &sColormap);
 
    //=========================================================================================================
    void updateNodeBuffers(QRhi *rhi, QRhiResourceUpdateBatch *u);
 
    //=========================================================================================================
    void updateEdgeBuffers(QRhi *rhi, QRhiResourceUpdateBatch *u);
 
    // ── Node accessors ──────────────────────────────────────────────────
    QRhiBuffer* nodeVertexBuffer() const;
    QRhiBuffer* nodeIndexBuffer() const;
    QRhiBuffer* nodeInstanceBuffer() const;
    int nodeIndexCount() const { return m_nodeIndexCount; }
    int nodeInstanceCount() const { return m_nodeInstanceCount; }
 
    // ── Edge accessors ──────────────────────────────────────────────────
    QRhiBuffer* edgeVertexBuffer() const;
    QRhiBuffer* edgeIndexBuffer() const;
    QRhiBuffer* edgeInstanceBuffer() const;
    int edgeIndexCount() const { return m_edgeIndexCount; }
    int edgeInstanceCount() const { return m_edgeInstanceCount; }
 
    // ── Visibility ──────────────────────────────────────────────────────
    bool isVisible() const { return m_visible; }
    void setVisible(bool visible) { m_visible = visible; }
 
    bool hasData() const { return !m_network.isEmpty(); }
 
private:
    // Instance data: identical layout to DipoleObject for shader compatibility
    // Model Matrix (4x4) + Color (vec4) + isSelected (float)
    struct InstanceData {
        float model[16];
        float color[4];
        float isSelected;   // Always 0.0 for networks
    };

    struct VertexData {
        float x, y, z;
        float nx, ny, nz;
    };
 
    void createNodeGeometry();
    void createEdgeGeometry();
    void buildNodeInstances();
    void buildEdgeInstances();
 
    // ── Network data ────────────────────────────────────────────────────
    CONNECTIVITYLIB::Network m_network;
    QString m_colormap = "Viridis";
 
    // ── Node GPU resources ──────────────────────────────────────────────
    struct GpuBuffers;
    std::unique_ptr<GpuBuffers> m_gpu;
 
    QByteArray m_nodeVertexData;
    QByteArray m_nodeIndexData;
    QByteArray m_nodeInstanceData;
    int m_nodeIndexCount = 0;
    int m_nodeInstanceCount = 0;
    bool m_nodeGeometryDirty = false;
    bool m_nodeInstancesDirty = false;
 
    // ── Edge GPU resources ──────────────────────────────────────────────
    QByteArray m_edgeVertexData;
    QByteArray m_edgeIndexData;
    QByteArray m_edgeInstanceData;
    int m_edgeIndexCount = 0;
    int m_edgeInstanceCount = 0;
    bool m_edgeGeometryDirty = false;
    bool m_edgeInstancesDirty = false;
 
    bool m_visible = true;
};
 
#endif // NETWORKOBJECT_H