VtkVisPipeline.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
#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;
};