src/edgeMesh/edgeFormats/vtk/VTKedgeFormat.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 2009-2010 OpenCFD Ltd.
   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
  13     the Free Software Foundation, either version 3 of the License, or
  14     (at your 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, see <http://www.gnu.org/licenses/>.
  23
  24 \*---------------------------------------------------------------------------*/
  25
  26 #include "VTKedgeFormat.H"
  27 #include "OFstream.H"
  28 #include "clock.H"
  29
  30 // * * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * //
  31
  32 void Foam::fileFormats::VTKedgeFormat::writeHeader
  33 (
  34     Ostream& os,
  35     const pointField& pointLst
  36 )
  37 {
  38     // Write header
  39     os  << "# vtk DataFile Version 2.0" << nl
  40         << "featureEdgeMesh written " << clock::dateTime().c_str() << nl
  41         << "ASCII" << nl
  42         << nl
  43         << "DATASET POLYDATA" << nl;
  44
  45     // Write vertex coords
  46     os  << "POINTS " << pointLst.size() << " float" << nl;
  47     forAll(pointLst, ptI)
  48     {
  49         const point& pt = pointLst[ptI];
  50
  51         os  << pt.x() << ' ' << pt.y() << ' ' << pt.z() << nl;
  52     }
  53 }
  54
  55
  56 void Foam::fileFormats::VTKedgeFormat::writeEdges
  57 (
  58     Ostream& os,
  59     const UList<edge>& edgeLst
  60 )
  61 {
  62     os  << "LINES " << edgeLst.size() << ' ' << 3*edgeLst.size() << nl;
  63
  64     forAll(edgeLst, edgeI)
  65     {
  66         const edge& e = edgeLst[edgeI];
  67
  68         os  << "2 " << e[0] << ' ' << e[1] << nl;
  69     }
  70 }
  71
  72
  73 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  74
  75 Foam::fileFormats::VTKedgeFormat::VTKedgeFormat()
  76 {}
  77
  78
  79 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  80
  81 void Foam::fileFormats::VTKedgeFormat::write
  82 (
  83     const fileName& filename,
  84     const edgeMesh& eMesh
  85 )
  86 {
  87     OFstream os(filename);
  88     if (!os.good())
  89     {
  90         FatalErrorIn
  91         (
  92             "fileFormats::VTKedgeFormat::write"
  93             "(const fileName&, const edgeMesh&)"
  94         )
  95             << "Cannot open file for writing " << filename
  96             << exit(FatalError);
  97     }
  98
  99     writeHeader(os, eMesh.points());
 100     writeEdges(os, eMesh.edges());
 101 }
 102
 103
 104 // ************************************************************************* //