polylineobject.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef DISP3DLIB_POLYLINEOBJECT_H
#define DISP3DLIB_POLYLINEOBJECT_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QByteArray>
#include <QColor>
#include <QString>
#include <QVector>
 
#include <memory>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
class QRhi;
class QRhiBuffer;
class QRhiResourceUpdateBatch;
 
//=============================================================================================================
// DEFINE CLASS
//=============================================================================================================
 
// Deliberately at global scope, matching DipoleObject and NetworkObject. The
// other renderables in this directory are not namespaced either.
 
namespace DISP3DLIB
{
 
//=============================================================================================================
class DISP3DSHARED_EXPORT PolylineObject
{
public:
    //=========================================================================================================
    PolylineObject();
 
    //=========================================================================================================
    ~PolylineObject();
 
    //=========================================================================================================
    void setPoints(const QVector<Eigen::Vector3f>& vecPoints);
 
    //=========================================================================================================
    void clear();
 
    //=========================================================================================================
    void setRadius(float fRadius);
 
    //=========================================================================================================
    void setGradient(const QColor& startColor, const QColor& endColor);
 
    //=========================================================================================================
    void updateBuffers(QRhi* rhi, QRhiResourceUpdateBatch* u);
 
    //=========================================================================================================
    QRhiBuffer* vertexBuffer() const;
 
    //=========================================================================================================
    QRhiBuffer* indexBuffer() const;
 
    //=========================================================================================================
    QRhiBuffer* instanceBuffer() const;
 
    //=========================================================================================================
    int indexCount() const
    {
        return m_iIndexCount;
    }
 
    //=========================================================================================================
    int instanceCount() const
    {
        return m_iInstanceCount;
    }
 
    //=========================================================================================================
    bool isVisible() const
    {
        return m_bVisible;
    }
 
    //=========================================================================================================
    void setVisible(bool bVisible)
    {
        m_bVisible = bVisible;
    }
 
    //=========================================================================================================
    bool hasData() const
    {
        return m_iInstanceCount > 0;
    }
 
private:
    //=========================================================================================================
    void createSegmentGeometry();
 
    //=========================================================================================================
    void buildInstances();
 
    // Instance layout shared with DipoleObject and NetworkObject so that the
    // Dipole pipeline can render this object unchanged.
    struct InstanceData
    {
        float model[16];
        float color[4];
        float isSelected; 
    };
 
    struct VertexData
    {
        float x, y, z;
        float nx, ny, nz;
    };
 
    QVector<Eigen::Vector3f> m_vecPoints; 
 
    QByteArray m_vertexData;   
    QByteArray m_indexData;    
    QByteArray m_instanceData; 
 
    int m_iIndexCount;    
    int m_iInstanceCount; 
 
    float m_fRadius;     
    QColor m_startColor; 
    QColor m_endColor;   
 
    bool m_bVisible;        
    bool m_bGeometryDirty;  
    bool m_bInstancesDirty; 
 
    struct GpuResources;
    std::unique_ptr<GpuResources> m_gpu; 
};
 
} // namespace DISP3DLIB
 
#endif // DISP3DLIB_POLYLINEOBJECT_H