VtkCompositeLineToTubeFilter.cpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
// ** INCLUDES **
#include "VtkCompositeLineToTubeFilter.h"
 
#include <vtkCleanPolyData.h>
#include <vtkSmartPointer.h>
#include <vtkTubeFilter.h>
 
VtkCompositeLineToTubeFilter::VtkCompositeLineToTubeFilter(
    vtkAlgorithm* inputAlgorithm)
    : VtkCompositeFilter(inputAlgorithm)
{
    this->init();
}
 
VtkCompositeLineToTubeFilter::~VtkCompositeLineToTubeFilter() = default;
 
void VtkCompositeLineToTubeFilter::init()
{
    this->_inputDataObjectType = VTK_DATA_SET;
    this->_outputDataObjectType = VTK_POLY_DATA;
 
    // collapse coincident points
    vtkSmartPointer<vtkCleanPolyData> mergePoints =
        vtkSmartPointer<vtkCleanPolyData>::New();
    mergePoints->SetInputConnection(0, _inputAlgorithm->GetOutputPort(0));
    mergePoints->SetTolerance(0.0);
    mergePoints->ConvertLinesToPointsOn();
 
    double default_radius(GetInitialRadius());
    int default_number_of_sides(8);
    vtkTubeFilter* tubes = vtkTubeFilter::New();
    tubes->SetInputConnection(0, mergePoints->GetOutputPort(0));
 
    // tubes->SetInputArrayToProcess(1,0,0,vtkDataObject::FIELD_ASSOCIATION_CELLS,"StationValue");
    // tubes->SetVaryRadiusToVaryRadiusByScalar(); // KR radius changes with
    // scalar
 
    tubes->SetInputArrayToProcess(
        1, 0, 0, vtkDataObject::FIELD_ASSOCIATION_CELLS, "Stratigraphies");
    tubes->SetRadius(default_radius);
    tubes->SetNumberOfSides(default_number_of_sides);
    tubes->SetCapping(1);
 
    (*_algorithmUserProperties)["Radius"] = default_radius;
    (*_algorithmUserProperties)["NumberOfSides"] = default_number_of_sides;
    (*_algorithmUserProperties)["Capping"] = true;
 
    _outputAlgorithm = tubes;
}
 
void VtkCompositeLineToTubeFilter::SetUserProperty(QString name, QVariant value)
{
    VtkAlgorithmProperties::SetUserProperty(name, value);
 
    if (name.compare("Radius") == 0)
    {
        static_cast<vtkTubeFilter*>(_outputAlgorithm)
            ->SetRadius(value.toDouble());
    }
    else if (name.compare("NumberOfSides") == 0)
    {
        static_cast<vtkTubeFilter*>(_outputAlgorithm)
            ->SetNumberOfSides(value.toInt());
    }
    else if (name.compare("Capping") == 0)
    {
        static_cast<vtkTubeFilter*>(_outputAlgorithm)
            ->SetCapping(value.toBool());
    }
}