VtkSurfacesSource.cpp

Go to the documentation of this file.

 
// ** INCLUDES **
#include "VtkSurfacesSource.h"
 
#include <vtkCellArray.h>
#include <vtkCellData.h>
#include <vtkInformation.h>
#include <vtkInformationVector.h>
#include <vtkObjectFactory.h>
#include <vtkPolyData.h>
#include <vtkProperty.h>
#include <vtkSmartPointer.h>
#include <vtkStreamingDemandDrivenPipeline.h>
#include <vtkTriangle.h>
 
#include <limits>
 
#include "Applications/DataHolderLib/Color.h"
#include "GeoLib/Triangle.h"
 
vtkStandardNewMacro(VtkSurfacesSource);
 
VtkSurfacesSource::VtkSurfacesSource()
{
    _removable = false;  // From VtkAlgorithmProperties
    this->SetNumberOfInputPorts(0);
    // this->SetColorBySurface(true);
 
    const DataHolderLib::Color c = DataHolderLib::getRandomColor();
    vtkProperty* vtkProps = GetProperties();
    vtkProps->SetColor(c[0] / 255.0, c[1] / 255.0, c[2] / 255.0);
    vtkProps->SetEdgeVisibility(0);
}
 
void VtkSurfacesSource::PrintSelf(ostream& os, vtkIndent indent)
{
    this->Superclass::PrintSelf(os, indent);
 
    if (_surfaces->empty())
    {
        return;
    }
 
    os << indent << "== VtkSurfacesSource =="
       << "\n";
}
 
int VtkSurfacesSource::RequestData(vtkInformation* request,
                                   vtkInformationVector** inputVector,
                                   vtkInformationVector* outputVector)
{
    (void)request;
    (void)inputVector;
 
    if (_surfaces->empty())
    {
        return 0;
    }
 
    const std::vector<GeoLib::Point*>* surfacePoints =
        (*_surfaces)[0]->getPointVec();
    std::size_t const nPoints = surfacePoints->size();
 
    vtkSmartPointer<vtkInformation> outInfo =
        outputVector->GetInformationObject(0);
    vtkSmartPointer<vtkPolyData> output =
        vtkPolyData::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT()));
    if (outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_PIECE_NUMBER()) >
        0)
    {
        return 1;
    }
 
    vtkSmartPointer<vtkPoints> newPoints = vtkSmartPointer<vtkPoints>::New();
    newPoints->SetNumberOfPoints(nPoints);
 
    vtkSmartPointer<vtkCellArray> newPolygons =
        vtkSmartPointer<vtkCellArray>::New();
    // newPolygons->Allocate(nSurfaces);
 
    vtkSmartPointer<vtkIntArray> sfcIDs = vtkSmartPointer<vtkIntArray>::New();
    sfcIDs->SetNumberOfComponents(1);
    sfcIDs->SetName("SurfaceIDs");
 
    for (std::size_t i = 0; i < nPoints; ++i)
    {
        newPoints->SetPoint(i, (*surfacePoints)[i]->data());
    }
 
    int count(0);
    for (auto surface : *_surfaces)
    {
        const std::size_t nTriangles = surface->getNumberOfTriangles();
 
        for (std::size_t i = 0; i < nTriangles; ++i)
        {
            vtkTriangle* new_tri = vtkTriangle::New();
            new_tri->GetPointIds()->SetNumberOfIds(3);
 
            const GeoLib::Triangle* triangle = (*surface)[i];
            for (std::size_t j = 0; j < 3; ++j)
            {
                new_tri->GetPointIds()->SetId(j, ((*triangle)[j]));
            }
            newPolygons->InsertNextCell(new_tri);
            sfcIDs->InsertNextValue(count);
            new_tri->Delete();
        }
        count++;
    }
 
    output->SetPoints(newPoints);
    output->SetPolys(newPolygons);
    output->GetCellData()->AddArray(sfcIDs);
    output->GetCellData()->SetActiveAttribute("SurfaceIDs",
                                              vtkDataSetAttributes::SCALARS);
    output->Squeeze();
 
    return 1;
}
 
int VtkSurfacesSource::RequestInformation(
    vtkInformation* /*request*/,
    vtkInformationVector** /*inputVector*/,
    vtkInformationVector* /*outputVector*/)
{
    return 1;
}
 
void VtkSurfacesSource::SetUserProperty(QString name, QVariant value)
{
    VtkAlgorithmProperties::SetUserProperty(name, value);
    (*_algorithmUserProperties)[name] = value;
}