StationTreeModel.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/Point.h"
 
#include "ModelTreeItem.h"
#include "Base/TreeModel.h"
 
namespace GeoLib
{
class Station;
class StationBorehole;
}
 
class QString;
class QModelIndex;
class PropertyBounds;

class StationTreeModel : public TreeModel
{
    Q_OBJECT
 
public:
    explicit StationTreeModel(QObject* parent = nullptr);
    ~StationTreeModel() override;
 
    void addStationList(QString listName, const std::vector<GeoLib::Point*>* stations);
    const std::vector<ModelTreeItem*> &getLists() { return _lists; }
    QModelIndex index(int row, int column,
                      const QModelIndex& parent = QModelIndex()) const override;
    //BaseItem* itemFromIndex( const QModelIndex& index ) const;
    void removeStationList(QModelIndex index);
    void removeStationList(const std::string &name);
    void setNameForItem(const std::string& stn_vec_name, std::size_t id,
                        std::string const& item_name);
    GeoLib::Station* stationFromIndex( const QModelIndex& index, QString &listName ) const;
    vtkPolyDataAlgorithm* vtkSource(const std::string &name) const;
 
private:
    std::vector<ModelTreeItem*> _lists;
};