networkobject.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef NETWORKOBJECT_H
#define NETWORKOBJECT_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include "../core/rendertypes.h"
 
#include <QMatrix4x4>
#include <QVector3D>
#include <QColor>
#include <vector>
#include <memory>
 
#include <connectivity/network/network.h>
 
 
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);
 
    //=========================================================================================================
    InstancedStreams nodeStreams() const;

    InstancedStreams edgeStreams() const;

    int nodeInstanceCount() const
    {
        return m_nodeInstanceCount;
    }

    int edgeInstanceCount() const
    {
        return m_edgeInstanceCount;
    }

    const RenderKey& nodeRenderKey() const
    {
        return m_nodeRenderKey;
    }
    const RenderKey& edgeRenderKey() const
    {
        return m_edgeRenderKey;
    }
 
    // ── 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 ──────────────────────────────────────────────
    RenderKey m_nodeRenderKey;
    RenderKey m_edgeRenderKey;
 
    QByteArray m_nodeVertexData;
    QByteArray m_nodeIndexData;
    QByteArray m_nodeInstanceData;
    int m_nodeIndexCount = 0;
    int m_nodeInstanceCount = 0;
    quint64 m_nodeGeometryGeneration = 0;
    quint64 m_nodeInstanceGeneration = 0;
 
    // ── Edge GPU resources ──────────────────────────────────────────────
    QByteArray m_edgeVertexData;
    QByteArray m_edgeIndexData;
    QByteArray m_edgeInstanceData;
    int m_edgeIndexCount = 0;
    int m_edgeInstanceCount = 0;
    quint64 m_edgeGeometryGeneration = 0;
    quint64 m_edgeInstanceGeneration = 0;
 
    bool m_visible = true;
};
 
} // namespace DISP3DLIB
 
#endif // NETWORKOBJECT_H