applications/utilities/postProcessing/graphics/PV3FoamReader/vtkPV3Foam/vtkPV3FoamMeshSet.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright held by original author
   6      \\/     M anipulation  |
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of OpenFOAM.
  10
  11     OpenFOAM is free software; you can redistribute it and/or modify it
  12     under the terms of the GNU General Public License as published by the
  13     Free Software Foundation; either version 2 of the License, or (at your
  14     option) any later version.
  15
  16     OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
  17     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  18     FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  19     for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with OpenFOAM; if not, write to the Free Software Foundation,
  23     Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  24
  25 Description
  26
  27 \*---------------------------------------------------------------------------*/
  28
  29 #include "vtkPV3Foam.H"
  30
  31 // Foam includes
  32 #include "faceSet.H"
  33 #include "pointSet.H"
  34 #include "vtkPV3FoamPoints.H"
  35
  36 // VTK includes
  37 #include "vtkPoints.h"
  38 #include "vtkPolyData.h"
  39 #include "vtkCellArray.h"
  40
  41 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  42
  43 vtkPolyData* Foam::vtkPV3Foam::faceSetVTKMesh
  44 (
  45     const fvMesh& mesh,
  46     const faceSet& fSet
  47 )
  48 {
  49     vtkPolyData* vtkmesh = vtkPolyData::New();
  50
  51     if (debug)
  52     {
  53         Info<< "<beg> Foam::vtkPV3Foam::faceSetVTKMesh" << endl;
  54         printMemory();
  55     }
  56
  57     // Construct primitivePatch of faces in fSet.
  58
  59     const faceList& meshFaces = mesh.faces();
  60     faceList patchFaces(fSet.size());
  61     label faceI = 0;
  62     forAllConstIter(faceSet, fSet, iter)
  63     {
  64         patchFaces[faceI++] = meshFaces[iter.key()];
  65     }
  66     primitiveFacePatch p(patchFaces, mesh.points());
  67
  68
  69     // The balance of this routine should be identical to patchVTKMesh
  70
  71     // Convert Foam mesh vertices to VTK
  72     const pointField& points = p.localPoints();
  73
  74     vtkPoints *vtkpoints = vtkPoints::New();
  75     vtkpoints->Allocate( points.size() );
  76     forAll(points, i)
  77     {
  78         vtkPV3FoamInsertNextPoint(vtkpoints, points[i]);
  79     }
  80     vtkmesh->SetPoints(vtkpoints);
  81     vtkpoints->Delete();
  82
  83     // Add faces as polygons
  84     const faceList& faces = p.localFaces();
  85
  86     vtkCellArray* vtkcells = vtkCellArray::New();
  87     vtkcells->Allocate( faces.size() );
  88
  89     forAll(faces, faceI)
  90     {
  91         const face& f = faces[faceI];
  92         vtkIdType nodeIds[f.size()];
  93
  94         forAll(f, fp)
  95         {
  96             nodeIds[fp] = f[fp];
  97         }
  98         vtkcells->InsertNextCell(f.size(), nodeIds);
  99     }
 100
 101     vtkmesh->SetPolys(vtkcells);
 102     vtkcells->Delete();
 103
 104     if (debug)
 105     {
 106         Info<< "<end> Foam::vtkPV3Foam::faceSetVTKMesh" << endl;
 107         printMemory();
 108     }
 109
 110     return vtkmesh;
 111 }
 112
 113
 114 vtkPolyData* Foam::vtkPV3Foam::pointSetVTKMesh
 115 (
 116     const fvMesh& mesh,
 117     const pointSet& pSet
 118 )
 119 {
 120     vtkPolyData* vtkmesh = vtkPolyData::New();
 121
 122     if (debug)
 123     {
 124         Info<< "<beg> Foam::vtkPV3Foam::pointSetVTKMesh" << endl;
 125         printMemory();
 126     }
 127
 128     const pointField& meshPoints = mesh.points();
 129
 130     vtkPoints *vtkpoints = vtkPoints::New();
 131     vtkpoints->Allocate( pSet.size() );
 132
 133     forAllConstIter(pointSet, pSet, iter)
 134     {
 135         vtkPV3FoamInsertNextPoint(vtkpoints, meshPoints[iter.key()]);
 136     }
 137
 138     vtkmesh->SetPoints(vtkpoints);
 139     vtkpoints->Delete();
 140
 141     if (debug)
 142     {
 143         Info<< "<end> Foam::vtkPV3Foam::pointSetVTKMesh" << endl;
 144         printMemory();
 145     }
 146
 147     return vtkmesh;
 148 }
 149
 150
 151 // ************************************************************************* //