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