VtkStationSource.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 "Applications/DataHolderLib/Color.h"
#include "GeoLib/StationBorehole.h"
#include "GeoLib/Station.h"
#include "VtkAlgorithmProperties.h"

class VtkStationSource : public vtkPolyDataAlgorithm, public VtkAlgorithmProperties
{
public:
    static VtkStationSource* New();
 
    vtkTypeMacro(VtkStationSource,vtkPolyDataAlgorithm);

    const std::map<std::string, DataHolderLib::Color>& getColorLookupTable() const
        { return _colorLookupTable; }

//    int setColorLookupTable(const std::string &filename)
//        { return GeoLib::readColorLookupTable(_colorLookupTable, filename); }

    void setStations(const std::vector<GeoLib::Point*>* stations) { _stations = stations; }

    void PrintSelf(ostream& os, vtkIndent indent) override;
 
    void SetUserProperty(QString name, QVariant value) override;
 
protected:
    VtkStationSource();

    int RequestData(vtkInformation* request,
                    vtkInformationVector** inputVector,
                    vtkInformationVector* outputVector) override;
 
    int RequestInformation(vtkInformation* request,
                           vtkInformationVector** inputVector,
                           vtkInformationVector* outputVector) override;

    const std::vector<GeoLib::Point*>* _stations{nullptr};

    std::map<std::string, DataHolderLib::Color> _colorLookupTable;
 
private:
    std::size_t GetIndexByName( std::string const& name );
 
    std::map<std::string, vtkIdType> _id_map;
};