src/OCC2VTK/GEOM_VertexSource.cxx

   1 // Copyright (C) 2007-2023  CEA, EDF, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include "GEOM_VertexSource.h"
  21
  22 #include <vtkObjectFactory.h>
  23
  24 #include <vtkPoints.h>
  25 #include <vtkCellArray.h>
  26 #include <vtkPolyData.h>
  27 #include <vtkPolyDataMapper.h>
  28 #include <vtkInformation.h>
  29 #include <vtkInformationVector.h>
  30
  31 #include <gp_Pnt.hxx>
  32 #include <BRep_Tool.hxx>
  33
  34 vtkStandardNewMacro(GEOM_VertexSource)
  35
  36 GEOM_VertexSource::GEOM_VertexSource()
  37 {
  38   this->SetNumberOfInputPorts(0);
  39 }
  40
  41 GEOM_VertexSource::~GEOM_VertexSource()
  42 {
  43 }
  44
  45 void
  46 GEOM_VertexSource::
  47 AddVertex(const TopoDS_Vertex& theVertex)
  48 {
  49   myVertexSet.Add(theVertex);
  50 }
  51
  52 int GEOM_VertexSource::RequestData(vtkInformation *vtkNotUsed(request),
  53                                    vtkInformationVector **vtkNotUsed(inputVector),
  54                                    vtkInformationVector *outputVector)
  55 {
  56   vtkInformation *outInfo = outputVector->GetInformationObject(0);
  57   vtkPolyData *aPolyData = vtkPolyData::SafeDownCast(
  58     outInfo->Get(vtkDataObject::DATA_OBJECT()));
  59
  60   aPolyData->Allocate();
  61   vtkPoints* aPts = vtkPoints::New();
  62   aPolyData->SetPoints(aPts);
  63   aPts->Delete();
  64
  65   TVertexSet::Iterator anIter(myVertexSet);
  66   for(; anIter.More(); anIter.Next()){
  67     const TopoDS_Vertex& aVertex = anIter.Value();
  68     OCC2VTK(aVertex,aPolyData,aPts);
  69   }
  70   return 1;
  71 }
  72
  73 void
  74 GEOM_VertexSource::
  75 OCC2VTK(const TopoDS_Vertex& theVertex,
  76         vtkPolyData* thePolyData,
  77         vtkPoints* thePts)
  78 {
  79   gp_Pnt aPnt = BRep_Tool::Pnt(theVertex);
  80   vtkIdType anId = thePts->InsertNextPoint(aPnt.X(),aPnt.Y(),aPnt.Z());
  81   thePolyData->InsertNextCell(VTK_VERTEX,1,&anId);
  82 }