applications/utilities/postProcessing/dataConversion/foamToVTK/faMeshWriter.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 \*---------------------------------------------------------------------------*/
  26
  27 #include "faMeshWriter.H"
  28 #include "writeFuns.H"
  29
  30 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  31
  32 // Construct from components
  33 Foam::faMeshWriter::faMeshWriter
  34 (
  35     const faMesh& aMesh,
  36     const bool binary,
  37     const fileName& fName
  38 )
  39 :
  40     aMesh_(aMesh),
  41     binary_(binary),
  42     os_(fName.c_str())
  43 {
  44     // Write header
  45     writeFuns::writeHeader(os_, binary_, faMesh::typeName);
  46     os_ << "DATASET UNSTRUCTURED_GRID" << std::endl;
  47
  48     // Write topology
  49     label nFaceVerts = 0;
  50
  51     const faceList& pp = aMesh.faces();
  52
  53     forAll(pp, faceI)
  54     {
  55         nFaceVerts += pp[faceI].size() + 1;
  56     }
  57
  58     os_ << "POINTS " << aMesh.nPoints() << " float" << std::endl;
  59
  60     DynamicList<floatScalar> ptField(3*aMesh.nPoints());
  61
  62     writeFuns::insert(aMesh.points(), ptField);
  63     writeFuns::write(os_, binary_, ptField);
  64
  65     os_ << "CELLS " << aMesh.nFaces() << ' ' << nFaceVerts
  66         << std::endl;
  67
  68     DynamicList<label> vertLabels(nFaceVerts);
  69     DynamicList<label> faceTypes(nFaceVerts);
  70
  71     forAll(pp, faceI)
  72     {
  73         const face& f = pp[faceI];
  74
  75         const label fSize = f.size();
  76         vertLabels.append(fSize);
  77
  78         writeFuns::insert(f, vertLabels);
  79
  80         if (fSize == 3)
  81         {
  82             faceTypes.append(vtkTopo::VTK_TRIANGLE);
  83         }
  84         else if (fSize == 4)
  85         {
  86             faceTypes.append(vtkTopo::VTK_QUAD);
  87         }
  88         else
  89         {
  90             faceTypes.append(vtkTopo::VTK_POLYGON);
  91         }
  92     }
  93
  94     writeFuns::write(os_, binary_, vertLabels);
  95
  96     os_ << "CELL_TYPES " << aMesh.nFaces() << std::endl;
  97
  98     writeFuns::write(os_, binary_, faceTypes);
  99 }
 100
 101
 102 // ************************************************************************* //