dipoleobject.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef DIPOLEOBJECT_H
#define DIPOLEOBJECT_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include <QMatrix4x4>
#include <QVector3D>
#include <vector>
#include <memory>
#include <inv/dipole_fit/inv_ecd_set.h>
 
// Forward-declare QRhi types so that this header stays QRhi-free
class QRhi;
class QRhiBuffer;
class QRhiResourceUpdateBatch;
 
//=============================================================================================================
class DISP3DSHARED_EXPORT DipoleObject
{
public:
    DipoleObject();
    ~DipoleObject();
 
    void load(const INVLIB::InvEcdSet &ecdSet);
    
    // Apply a transformation matrix to all dipoles
    void applyTransform(const QMatrix4x4 &trans);
 
    void updateBuffers(QRhi *rhi, QRhiResourceUpdateBatch *u);
 
    QRhiBuffer* vertexBuffer() const;
    QRhiBuffer* indexBuffer() const;
    QRhiBuffer* instanceBuffer() const;
    
    int indexCount() const { return m_indexCount; }
    int instanceCount() const { return m_instanceCount; }
    
    bool isVisible() const { return m_visible; }
    void setVisible(bool visible) { m_visible = visible; }
    
    QVector3D debugFirstDipolePosition() const; // For debugging
 
    void setSelected(int index, bool selected);
 
    //=========================================================================================================
    int intersect(const QVector3D &rayOrigin, const QVector3D &rayDir, float &dist) const;
 
private:
    void createGeometry();
    
    struct GpuBuffers;
    std::unique_ptr<GpuBuffers> m_gpu;
 
    int m_indexCount = 0;
    int m_instanceCount = 0;
    bool m_visible = true;
    
    struct VertexData {
        float x, y, z;
        float nx, ny, nz;
    };
    
    // Instance data: Model Matrix (4x4) + Color (vec4)
    struct InstanceData {
        float model[16]; 
        float color[4];
        float isSelected; // 1.0 = selected, 0.0 = not
    };
    
    QByteArray m_vertexData;
    QByteArray m_indexData;
    QByteArray m_instanceData;
    
    bool m_geometryDirty = false;
    bool m_instancesDirty = false;
    
    std::vector<QVector4D> m_originalColors;
};
 
#endif // DIPOLEOBJECT_H