VtkStationSource.h

Go to the documentation of this file.

 
#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;
};