networkobject.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef NETWORKOBJECT_H
#define NETWORKOBJECT_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_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;
 
namespace DISP3DLIB
{
 
//=============================================================================================================
class DISP3DSHARED_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;
};
 
} // namespace DISP3DLIB
 
#endif // NETWORKOBJECT_H