GeoObjectListItem.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 "Base/TreeItem.h"
 
#include "GeoLib/GeoType.h"
 
#include "VtkVis/VtkPointsSource.h"
#include "VtkVis/VtkPolylinesSource.h"
#include "VtkVis/VtkSurfacesSource.h"
#include <QModelIndex>
#include <vtkPolyDataAlgorithm.h>

class GeoObjectListItem : public TreeItem
{
public:
    GeoObjectListItem(const QList<QVariant>& data,
                      TreeItem* parent,
                      const std::vector<GeoLib::Point*>* geo_data = nullptr,
                      GeoLib::GEOTYPE type = GeoLib::GEOTYPE::POINT)
        : TreeItem(data, parent),
          _vtkSource(VtkPointsSource::New()),
          _type(type)
    {
        QString geo_name = parent->data(0).toString();
        static_cast<VtkPointsSource*>(_vtkSource)->setPoints(geo_data);
        static_cast<VtkPointsSource*>(_vtkSource)->SetName(geo_name + " - Points");
    }

    GeoObjectListItem(const QList<QVariant>& data,
                      TreeItem* parent,
                      const std::vector<GeoLib::Polyline*>* geo_data = nullptr,
                      GeoLib::GEOTYPE type = GeoLib::GEOTYPE::POLYLINE)
        : TreeItem(data, parent),
          _vtkSource(VtkPolylinesSource::New()),
          _type(type)
    {
        QString geo_name = parent->data(0).toString();
        static_cast<VtkPolylinesSource*>(_vtkSource)->setPolylines(geo_data);
        static_cast<VtkPolylinesSource*>(_vtkSource)->SetName(geo_name + " - Polylines");
    }

    GeoObjectListItem(const QList<QVariant>& data,
                      TreeItem* parent,
                      const std::vector<GeoLib::Surface*>* geo_data = nullptr,
                      GeoLib::GEOTYPE type = GeoLib::GEOTYPE::SURFACE)
        : TreeItem(data, parent),
          _vtkSource(VtkSurfacesSource::New()),
          _type(type)
    {
        QString geo_name = parent->data(0).toString();
        static_cast<VtkSurfacesSource*>(_vtkSource)->setSurfaces(geo_data);
        static_cast<VtkSurfacesSource*>(_vtkSource)->SetName(geo_name + " - Surfaces");
    }
 
    ~GeoObjectListItem() override { _vtkSource->Delete(); }
    GeoLib::GEOTYPE getType() { return _type; }

    vtkPolyDataAlgorithm* vtkSource() const { return _vtkSource; }
 
private:
    vtkPolyDataAlgorithm* _vtkSource;
 
    GeoLib::GEOTYPE _type;
};