GeoTreeModel.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 <vtkPolyDataAlgorithm.h>
#include <vector>
 
#include "GeoLib/GeoType.h"
#include "GeoLib/PointVec.h"
#include "GeoLib/PolylineVec.h"
#include "GeoLib/SurfaceVec.h"
#include "Base/TreeModel.h"
 
namespace GeoLib
{
struct GeoObject;
}
 
class QString;
class QModelIndex;
class GeoTreeItem;
class GeoObjectListItem;

class GeoTreeModel : public TreeModel
{
    Q_OBJECT
 
public:
    explicit GeoTreeModel(QObject* parent = nullptr);
    ~GeoTreeModel() override;

    void addPointList(QString geoName, GeoLib::PointVec const& pointVec);
    void addPolylineList(QString geoName, GeoLib::PolylineVec const& polylineVec);
    void appendPolylines(const std::string &name, GeoLib::PolylineVec const& polylineVec);
    void addSurfaceList(QString geoName, GeoLib::SurfaceVec const& surfaceVec);
    void appendSurfaces(const std::string &name, GeoLib::SurfaceVec const& surfaceVec);
    const std::vector<GeoTreeItem*> &getLists() { return _lists; }
    void removeGeoList(const std::string &name, GeoLib::GEOTYPE type);
 
    void renameGeometry(std::string const& old_name,
                        std::string const& new_name);
 
    void setNameForItem(const std::string &name, GeoLib::GEOTYPE type, std::size_t id, std::string item_name);
 
    /*
     * Returns the geo-object specified by the given index.
     * \param index Index of the requested item
     * \param listName Here, the method will put the name of the geometry this object belongs to.
     * \return A geo-object (Point / Polyline / Surface)
     */
    //const GeoLib::GeoObject* objectFromIndex( const QModelIndex& index, QString &geoName ) const;

    vtkPolyDataAlgorithm* vtkSource(const std::string &name, GeoLib::GEOTYPE type) const;
 
private:
    void addChildren(GeoObjectListItem* plyList,
                     GeoLib::PolylineVec const& polyline_vec,
                     std::size_t start_index,
                     std::size_t end_index);

    void addChildren(GeoObjectListItem* sfcList,
                     GeoLib::SurfaceVec const& surface_vec,
                     std::size_t start_index,
                     std::size_t end_index);
 
    std::vector<GeoTreeItem*> _lists;
};