applications/utilities/postProcessing/dataConversion/foamToVTK/readFields.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 "readFields.H"
  28 #include "IOobjectList.H"
  29
  30 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  31
  32 namespace Foam
  33 {
  34
  35 // * * * * * * * * * * * * * * * Global Functions  * * * * * * * * * * * * * //
  36
  37 template<class GeoField>
  38 void readFields
  39 (
  40     const vtkMesh& vMesh,
  41     const typename GeoField::Mesh& mesh,
  42     const IOobjectList& objects,
  43     const HashSet<word>& selectedFields,
  44     PtrList<GeoField>& fields
  45 )
  46 {
  47     // Search list of objects for volScalarFields
  48     IOobjectList fieldObjects(objects.lookupClass(GeoField::typeName));
  49
  50     // Construct the vol scalar fields
  51     fields.setSize(fieldObjects.size());
  52     label nFields = 0;
  53
  54     for
  55     (
  56         IOobjectList::iterator iter = fieldObjects.begin();
  57         iter != fieldObjects.end();
  58         ++iter
  59     )
  60     {
  61         if (selectedFields.empty() || selectedFields.found(iter()->name()))
  62         {
  63             fields.set
  64             (
  65                 nFields,
  66                 vMesh.interpolate
  67                 (
  68                     GeoField
  69                     (
  70                         *iter(),
  71                         mesh
  72                     )
  73                 )
  74             );
  75             nFields++;
  76         }
  77     }
  78
  79     fields.setSize(nFields);
  80 }
  81
  82
  83 template<class GeoField>
  84 void readFieldsNoSubset
  85 (
  86     const vtkMesh& vMesh,
  87     const typename GeoField::Mesh& mesh,
  88     const IOobjectList& objects,
  89     const HashSet<word>& selectedFields,
  90     PtrList<GeoField>& fields
  91 )
  92 {
  93     // Search list of objects for volScalarFields
  94     IOobjectList fieldObjects(objects.lookupClass(GeoField::typeName));
  95
  96     // Construct the vol scalar fields
  97     fields.setSize(fieldObjects.size());
  98     label nFields = 0;
  99
 100     for
 101     (
 102         IOobjectList::iterator iter = fieldObjects.begin();
 103         iter != fieldObjects.end();
 104         ++iter
 105     )
 106     {
 107         if (!selectedFields.size() || selectedFields.found(iter()->name()))
 108         {
 109             fields.set
 110             (
 111                 nFields,
 112                 GeoField
 113                 (
 114                     *iter(),
 115                     mesh
 116                 )
 117             );
 118             nFields++;
 119         }
 120     }
 121
 122     fields.setSize(nFields);
 123 }
 124
 125
 126 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 127
 128 } // End namespace Foam
 129
 130 // ************************************************************************* //