src/lagrangian/basic/IOPosition/IOPosition.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 2004-2011 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 "IOPosition.H"
  27
  28 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  29
  30 template<class CloudType>
  31 Foam::IOPosition<CloudType>::IOPosition(const CloudType& c)
  32 :
  33     regIOobject
  34     (
  35         IOobject
  36         (
  37             "positions",
  38             c.time().timeName(),
  39             c,
  40             IOobject::MUST_READ,
  41             IOobject::NO_WRITE
  42         )
  43     ),
  44     cloud_(c)
  45 {}
  46
  47
  48 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  49
  50 template<class CloudType>
  51 bool Foam::IOPosition<CloudType>::write() const
  52 {
  53     if (cloud_.size())
  54     {
  55         return regIOobject::write();
  56     }
  57     else
  58     {
  59         return true;
  60     }
  61 }
  62
  63
  64 template<class CloudType>
  65 bool Foam::IOPosition<CloudType>::writeData(Ostream& os) const
  66 {
  67     os  << cloud_.size() << nl << token::BEGIN_LIST << nl;
  68
  69     forAllConstIter(typename CloudType, cloud_, iter)
  70     {
  71         const typename CloudType::particleType& p = iter();
  72
  73         // Prevent writing additional fields
  74         p.write(os, false);
  75
  76         os  << nl;
  77     }
  78
  79     os  << token::END_LIST << endl;
  80
  81     return os.good();
  82 }
  83
  84
  85 template<class CloudType>
  86 void Foam::IOPosition<CloudType>::readData(CloudType& c, bool checkClass)
  87 {
  88     const polyMesh& mesh = c.pMesh();
  89
  90     Istream& is = readStream(checkClass ? typeName : "");
  91
  92     token firstToken(is);
  93
  94     if (firstToken.isLabel())
  95     {
  96         label s = firstToken.labelToken();
  97
  98         // Read beginning of contents
  99         is.readBeginList("IOPosition<CloudType>::readData(CloudType, bool)");
 100
 101         for (label i=0; i<s; i++)
 102         {
 103             // Do not read any fields, position only
 104             c.append(new typename CloudType::particleType(mesh, is, false));
 105         }
 106
 107         // Read end of contents
 108         is.readEndList("IOPosition<CloudType>::readData(CloudType, bool)");
 109     }
 110     else if (firstToken.isPunctuation())
 111     {
 112         if (firstToken.pToken() != token::BEGIN_LIST)
 113         {
 114             FatalIOErrorIn
 115             (
 116                 "void IOPosition<CloudType>::readData(CloudType&, bool)",
 117                 is
 118             )   << "incorrect first token, '(', found "
 119                 << firstToken.info() << exit(FatalIOError);
 120         }
 121
 122         token lastToken(is);
 123         while
 124         (
 125            !(
 126                 lastToken.isPunctuation()
 127              && lastToken.pToken() == token::END_LIST
 128             )
 129         )
 130         {
 131             is.putBack(lastToken);
 132             // Do not read any fields, position only
 133             c.append(new typename CloudType::particleType(mesh, is, false));
 134             is  >> lastToken;
 135         }
 136     }
 137     else
 138     {
 139         FatalIOErrorIn
 140         (
 141             "void IOPosition<ParticleType>::readData(CloudType&, bool)",
 142             is
 143         )   << "incorrect first token, expected <int> or '(', found "
 144             << firstToken.info() << exit(FatalIOError);
 145     }
 146
 147     // Check state of IOstream
 148     is.check
 149     (
 150         "void IOPosition<CloudType>::readData(CloudType&, bool)"
 151     );
 152 }
 153
 154
 155 // ************************************************************************* //