dipoleobject.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef DIPOLEOBJECT_H
#define DIPOLEOBJECT_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include <QMatrix4x4>
#include <QVector>
#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;
 
namespace DISP3DLIB
{
 
//=============================================================================================================
class DISP3DSHARED_EXPORT DipoleObject
{
public:
    DipoleObject();
    ~DipoleObject();
 
    void load(const INVLIB::InvEcdSet& ecdSet);
 
    // Place all loaded dipoles with this transform (replaces any previous one)
    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;
    }
 
    //=========================================================================================================
    bool boundingBox(QVector3D& min, QVector3D& max) const;
 
    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;
    QVector<QMatrix4x4> m_loadedModels; 
 
    bool m_geometryDirty = false;
    bool m_instancesDirty = false;
 
    std::vector<QVector4D> m_originalColors;
};
 
} // namespace DISP3DLIB
 
#endif // DIPOLEOBJECT_H