electrodeobject.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ELECTRODEOBJECT_H
#define ELECTRODEOBJECT_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
#include "../core/rendertypes.h"
 
#include <ieeg/ieeg_implantation.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QByteArray>
#include <QColor>
#include <QMap>
#include <QString>
#include <QVector>
#include <QVector3D>
 
//=============================================================================================================
// DEFINE NAMESPACE DISP3DLIB
//=============================================================================================================
 
namespace DISP3DLIB
{
 
//=============================================================================================================
class DISP3DSHARED_EXPORT ElectrodeObject
{
public:
    // Drawn sizes in metres. Real contacts are about 0.8 mm (depth) and 4 mm (subdural) across;
    // both are enlarged to stay visible and pickable, keeping their ratio.
    static constexpr float kDepthContactRadius = 0.0011f;
    static constexpr float kSubduralContactRadius = 0.0019f;
    static constexpr float kShaftRadius = 0.00045f;
    static constexpr float kShaftOvershoot = 0.004f; 
    static constexpr float kTieRadius = 0.00035f;    
    static constexpr float kMaxTieLength = 0.015f;   
 
    ElectrodeObject();
    ~ElectrodeObject();

    static QColor leadColor(int index);
 
    //=========================================================================================================
    void setImplantation(const IEEGLIB::IeegImplantation& implantation);

    const IEEGLIB::IeegImplantation& implantation() const
    {
        return m_implantation;
    }

    int totalContactCount() const;
 
    //=========================================================================================================
    void setContactValues(const QMap<QString, float>& values, const QString& colormap = QStringLiteral("Hot"));

    void clearContactValues();

    QVector<uint32_t> surfaceColors(const Eigen::MatrixX3f& vertices, double distance = 0.01) const;
 
    //=========================================================================================================
    void selectContact(const QString& name);

    void clearSelection();

    QString selectedContact() const;
 
    //=========================================================================================================
    void setHoveredContact(int index);

    int hoveredContact() const
    {
        return m_hoveredContact;
    }
 
    //=========================================================================================================
    int intersect(const QVector3D& rayOrigin, const QVector3D& rayDir, float& dist) const;

    const IEEGLIB::IeegContact* contactAt(int index) const;

    QString contactLabel(int index) const;
 
    //=========================================================================================================
    InstancedStreams contactStreams() const;

    InstancedStreams bodyStreams() const;

    const RenderKey& contactRenderKey() const
    {
        return m_contactRenderKey;
    }

    const RenderKey& bodyRenderKey() const
    {
        return m_bodyRenderKey;
    }
 
    bool isVisible() const
    {
        return m_visible;
    }
    void setVisible(bool visible)
    {
        m_visible = visible;
    }
 
    //=========================================================================================================
    QVector3D boundingBoxMin() const;

    QVector3D boundingBoxMax() const;
 
private:
    struct InstanceData
    {
        float model[16];
        float color[4];
        float isSelected;
    };
 
    void buildBodies();
    void buildContactInstances();
    float contactRadius(int contact) const;
 
    IEEGLIB::IeegImplantation m_implantation;
    QVector<int> m_leadOfContact;    
    QVector<QColor> m_contactColors; 
    QMap<QString, float> m_contactValues;
    QString m_colormap;
    float m_valueMin = 0.0f;
    float m_valueMax = 0.0f;
    int m_selectedContact = -1;
    int m_hoveredContact = -1;
    QVector3D m_bbMin;
    QVector3D m_bbMax;
    bool m_visible = true;
 
    // Unit sphere and unit cylinder (along +y, centred) shared by all instances
    QByteArray m_sphereVertices;
    QByteArray m_sphereIndices;
    QByteArray m_cylinderVertices;
    QByteArray m_cylinderIndices;
    int m_sphereIndexCount = 0;
    int m_cylinderIndexCount = 0;
 
    QByteArray m_contactInstances;
    QByteArray m_bodyInstances;
    int m_bodyCount = 0;
    quint64 m_contactGeneration = 0;
    quint64 m_bodyGeneration = 0;
    RenderKey m_contactRenderKey;
    RenderKey m_bodyRenderKey;
 
    void computeBoundingBox();
 
    float normalized(float value) const;
};
 
} // namespace DISP3DLIB
 
#endif // ELECTRODEOBJECT_H