VtkVisPipeline.h

Go to the documentation of this file.

 
#pragma once
 
#include <QMap>
#include <QVector>
 
#include "Applications/DataHolderLib/Color.h"
#include "GeoLib/GeoType.h"
#include "GeoLib/Point.h"
#include "MeshLib/MeshEnums.h"
#include "Base/TreeModel.h"
 
class vtkAlgorithm;
class vtkDataSet;
class vtkLight;
class vtkPointSet;
class vtkPolyDataAlgorithm;
class vtkProp3D;
class vtkRenderer;
class vtkUnstructuredGridAlgorithm;
 
class QModelIndex;
class QString;
 
class GeoTreeModel;
class ProcessModel;
class MeshModel;
class StationTreeModel;
class TreeModel;
class VtkVisPipelineItem;
 
namespace MeshLib
{
    class VtkMappedMeshSource;
}
 
class VtkVisPipeline : public TreeModel
{
    Q_OBJECT
 
public:
    explicit VtkVisPipeline(vtkRenderer* renderer, QObject* parent = nullptr);
 
    bool setData(const QModelIndex& index, const QVariant& value,
                 int role = Qt::EditRole) override;
 
    void addLight(const GeoLib::Point &pos);
 
    vtkLight* getLight(const GeoLib::Point &pos) const;
 
    void removeLight(const GeoLib::Point &pos);
 
    QColor getBGColor() const;
 
    void setBGColor(const QColor &color);
 
    QModelIndex getIndex(vtkProp3D* actor);
 
    Qt::ItemFlags flags(const QModelIndex& index) const override;
 
    void loadFromFile(QString filename);
 
    void resetCameraOnAddOrRemove(bool reset) { _resetCameraOnAddOrRemove = reset; }
 
    void setGlobalSuperelevation(double factor) const;
 
    void setGlobalBackfaceCulling(bool enable) const;
 
    void showMeshElementQuality(MeshLib::VtkMappedMeshSource* source,
                                MeshLib::MeshQualityType t,
                                std::vector<double> const& quality);
 
public slots:
    void addPipelineItem(MeshModel* model, const QModelIndex &idx);
    void addPipelineItem(GeoTreeModel* model, const std::string &name, GeoLib::GEOTYPE type);
    void addPipelineItem(StationTreeModel* model, const std::string &name);
    QModelIndex addPipelineItem(VtkVisPipelineItem* item, const QModelIndex &parent);
 
    QModelIndex addPipelineItem(vtkAlgorithm* source, QModelIndex parent = QModelIndex());
 
    void removeSourceItem(MeshModel* model, const QModelIndex &idx);
    void removeSourceItem(GeoTreeModel* model, const std::string &name, GeoLib::GEOTYPE type);
    void removeSourceItem(StationTreeModel* model, const std::string &name);
 
    void removePipelineItem(QModelIndex index);
 
    void highlightGeoObject(const vtkPolyDataAlgorithm* source, int index);
 
    void removeHighlightedGeoObject();
 
    void highlightMeshComponent(vtkUnstructuredGridAlgorithm const*const source, unsigned index, bool is_element);
 
    void removeHighlightedMeshComponent();
 
 
private:
    void listArrays(vtkDataSet* dataSet);
 
    vtkRenderer* _renderer;
    QVector<vtkAlgorithm*> _sources;
    std::list<vtkLight*> _lights;
    QMap<vtkProp3D*, QModelIndex> _actorMap;
    bool _resetCameraOnAddOrRemove;
 
    QModelIndex _highlighted_geo_index;
    QModelIndex _highlighted_mesh_component;
 
signals:
    void vtkVisPipelineChanged() const;
    void itemSelected(const QModelIndex&) const;
};