VtkPolylinesSource.cpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
// ** INCLUDES **
#include "VtkPolylinesSource.h"
 
#include <vtkCellArray.h>
#include <vtkCellData.h>
#include <vtkInformation.h>
#include <vtkInformationVector.h>
#include <vtkObjectFactory.h>
#include <vtkPointData.h>
#include <vtkPoints.h>
#include <vtkPolyData.h>
#include <vtkProperty.h>
#include <vtkSmartPointer.h>
#include <vtkStreamingDemandDrivenPipeline.h>
 
#include "Applications/DataHolderLib/Color.h"
#include "BaseLib/Logging.h"
#include "GeoLib/Polyline.h"
 
vtkStandardNewMacro(VtkPolylinesSource);
 
VtkPolylinesSource::VtkPolylinesSource()
{
    _removable = false;  // From VtkAlgorithmProperties
    this->SetNumberOfInputPorts(0);
 
    const DataHolderLib::Color c = DataHolderLib::getRandomColor();
    GetProperties()->SetColor(c[0] / 255.0, c[1] / 255.0, c[2] / 255.0);
}
 
VtkPolylinesSource::~VtkPolylinesSource() = default;
 
void VtkPolylinesSource::PrintSelf(ostream& os, vtkIndent indent)
{
    this->Superclass::PrintSelf(os, indent);
 
    if (_polylines->empty())
    {
        return;
    }
 
    for (auto polyline : *_polylines)
    {
        os << indent << "== Polyline =="
           << "\n";
        int numPoints = polyline->getNumberOfPoints();
        for (int i = 0; i < numPoints; i++)
        {
            const GeoLib::Point& point = *polyline->getPoint(i);
            os << indent << "Point " << i << " (" << point[0] << ", "
               << point[1] << ", " << point[2] << ")\n";
        }
    }
}
 
int VtkPolylinesSource::RequestData(vtkInformation* request,
                                    vtkInformationVector** inputVector,
                                    vtkInformationVector* outputVector)
{
    (void)request;
    (void)inputVector;
 
    if (!_polylines)
    {
        return 0;
    }
    if (_polylines->empty())
    {
        ERR("VtkPolylineSource::RequestData(): Size of polyline vector is 0");
        return 0;
    }
 
    vtkSmartPointer<vtkInformation> outInfo =
        outputVector->GetInformationObject(0);
    vtkSmartPointer<vtkPolyData> output =
        vtkPolyData::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT()));
 
    vtkSmartPointer<vtkPoints> newPoints = vtkSmartPointer<vtkPoints>::New();
    vtkSmartPointer<vtkCellArray> newLines =
        vtkSmartPointer<vtkCellArray>::New();
 
    // newPoints->Allocate(numPoints);
    // newLines->Allocate(newLines->EstimateSize(numLines, 2));
 
    if (outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_PIECE_NUMBER()) >
        0)
    {
        return 1;
    }
 
    vtkSmartPointer<vtkIntArray> plyIDs = vtkSmartPointer<vtkIntArray>::New();
    plyIDs->SetNumberOfComponents(1);
    plyIDs->SetName("PolylineIDs");
 
    unsigned lastMaxIndex(0);
    const std::size_t nPolylines(_polylines->size());
    for (std::size_t j = 0; j < nPolylines; j++)
    {
        const int numPoints = (*_polylines)[j]->getNumberOfPoints();
        const bool isClosed = (*_polylines)[j]->isClosed();
 
        // Generate points
        const int numVerts = (isClosed) ? numPoints - 1 : numPoints;
        for (int i = 0; i < numVerts; i++)
        {
            const GeoLib::Point* point = (*_polylines)[j]->getPoint(i);
            const double* coords = point->data();
            newPoints->InsertNextPoint(coords);
        }
 
        // Generate lines
        newLines->InsertNextCell(numPoints);
        plyIDs->InsertNextValue(j);
        for (int i = 0; i < numVerts; i++)
        {
            newLines->InsertCellPoint(i + lastMaxIndex);
        }
 
        if (isClosed)
        {
            newLines->InsertCellPoint(lastMaxIndex);
        }
 
        lastMaxIndex += numVerts;
    }
 
    output->SetPoints(newPoints);
    output->SetLines(newLines);
    output->GetCellData()->AddArray(plyIDs);
    output->GetCellData()->SetActiveAttribute("PolylineIDs",
                                              vtkDataSetAttributes::SCALARS);
    output->Squeeze();
 
    return 1;
}
 
int VtkPolylinesSource::RequestInformation(
    vtkInformation* /*request*/,
    vtkInformationVector** /*inputVector*/,
    vtkInformationVector* /*outputVector*/)
{
    return 1;
}
 
void VtkPolylinesSource::SetUserProperty(QString name, QVariant value)
{
    Q_UNUSED(name);
    Q_UNUSED(value);
}