electrodeobject.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ELECTRODEOBJECT_H
#define ELECTRODEOBJECT_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QColor>
#include <QMap>
#include <QString>
#include <QVector>
#include <QVector3D>
 
#include <memory>
 
// Forward-declare QRhi types so that this header stays QRhi-free
class QRhi;
class QRhiBuffer;
class QRhiResourceUpdateBatch;
 
//=============================================================================================================
// DEFINE NAMESPACE DISP3DLIB
//=============================================================================================================
 
namespace DISP3DLIB
{
 
//=============================================================================================================
struct DISP3DSHARED_EXPORT ElectrodeContact
{
    QString     name;           
    QVector3D   position;       
    float       radius = 0.5f;  
    QColor      color = Qt::yellow; 
    bool        selected = false;   
    float       value = 0.0f;   
};
 
//=============================================================================================================
enum class ElectrodeLayout {
    Depth = 0,  
    Strip,       
    Grid         
};
 
//=============================================================================================================
struct DISP3DSHARED_EXPORT ElectrodeArray
{
    QString                     label;                  
    ElectrodeLayout             layout = ElectrodeLayout::Depth; 
    int                         gridRows = 1;           
    int                         gridCols = 1;           
    QVector<ElectrodeContact>   contacts;               
    float                       shaftRadius = 0.4f;     
    QColor                      shaftColor = Qt::gray;  
};
 
//=============================================================================================================
class DISP3DSHARED_EXPORT ElectrodeObject
{
public:
    ElectrodeObject();
    ~ElectrodeObject();
 
    //=========================================================================================================
    void setArrays(const QVector<ElectrodeArray>& arrays);
 
    //=========================================================================================================
    const QVector<ElectrodeArray>& arrays() const;
 
    //=========================================================================================================
    int totalContactCount() const;
 
    //=========================================================================================================
    void setContactValues(const QMap<QString, float>& values,
                          const QColor& minColor = Qt::blue,
                          const QColor& maxColor = Qt::red);
 
    //=========================================================================================================
    void selectContact(const QString& name);
 
    //=========================================================================================================
    void clearSelection();
 
    //=========================================================================================================
    QString selectedContact() const;
 
    //=========================================================================================================
    void generateShaftGeometry(QVector<float>& vertices,
                               QVector<unsigned int>& indices,
                               int cylinderSides = 16) const;
 
    //=========================================================================================================
    void generateContactInstances(QVector<float>& instanceData) const;
 
    //=========================================================================================================
    QVector3D boundingBoxMin() const;
 
    //=========================================================================================================
    QVector3D boundingBoxMax() const;
 
    //=========================================================================================================
    void updateBuffers(QRhi *rhi, QRhiResourceUpdateBatch *u);
 
    //=========================================================================================================
    QRhiBuffer* vertexBuffer() const;
 
    //=========================================================================================================
    QRhiBuffer* indexBuffer() const;
 
    //=========================================================================================================
    QRhiBuffer* instanceBuffer() const;
 
    //=========================================================================================================
    uint32_t shaftIndexCount() const;
 
    //=========================================================================================================
    uint32_t contactInstanceCount() const;
 
private:
    QVector<ElectrodeArray> m_arrays;            
    QString                 m_selectedContact;   
    QVector3D               m_bbMin;             
    QVector3D               m_bbMax;             
 
    //=========================================================================================================
    void computeBoundingBox();
 
    //=========================================================================================================
    static QColor interpolateColor(float value, float minVal, float maxVal,
                                   const QColor& minColor, const QColor& maxColor);

    struct GpuBuffers;
    std::unique_ptr<GpuBuffers> m_gpu;
};
 
} // namespace DISP3DLIB
 
#endif // ELECTRODEOBJECT_H