MeshModel.h

Go to the documentation of this file.

 
#pragma once
 
#include <memory>
 
#include "MeshLib/MeshEnums.h"
 
#include "Base/TreeModel.h"
 
namespace DataHolderLib
{
class Project;
}
 
namespace MeshLib {
class Mesh;
}
 
class vtkUnstructuredGridAlgorithm;
 
class MeshModel : public TreeModel
{
    Q_OBJECT
 
public:
    explicit MeshModel(DataHolderLib::Project& project,
                       QObject* parent = nullptr);
 
    void addMesh(std::unique_ptr<MeshLib::Mesh> mesh);
 
    int columnCount(const QModelIndex& parent = QModelIndex()) const override;
 
public slots:
    void addMesh(MeshLib::Mesh* mesh);
    const MeshLib::Mesh* getMesh(const QModelIndex &idx) const;
    const MeshLib::Mesh* getMesh(const std::string &name) const;
    bool removeMesh(const QModelIndex &idx);
    bool removeMesh(const std::string &name);
    void updateMesh(MeshLib::Mesh* /*mesh*/);
    void updateModel();
    vtkUnstructuredGridAlgorithm* vtkSource(const QModelIndex &idx) const;
    vtkUnstructuredGridAlgorithm* vtkSource(const std::string &name) const;
 
private:
    void addMeshObject(const MeshLib::Mesh* mesh);
 
    //bool isUniqueMeshName(std::string &name);
    DataHolderLib::Project& _project;
 
    static std::map<MeshLib::MeshElemType, QVariant> createMeshElemTypeMap();
 
    static const std::map<MeshLib::MeshElemType, QVariant> elem_type_map;
    static const QVariant element_str;
 
signals:
    void meshAdded(MeshModel*, const QModelIndex&);
    void meshRemoved(MeshModel*, const QModelIndex&);
};