src/lagrangian/dsmc/parcels/Templates/DsmcParcel/DsmcParcelIO.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 "DsmcParcel.H"
  28 #include "IOstreams.H"
  29 #include "IOField.H"
  30 #include "Cloud.H"
  31
  32 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  33
  34 template <class ParcelType>
  35 Foam::DsmcParcel<ParcelType>::DsmcParcel
  36 (
  37     const Cloud<ParcelType>& cloud,
  38     Istream& is,
  39     bool readFields
  40 )
  41 :
  42     Particle<ParcelType>(cloud, is, readFields),
  43     U_(vector::zero),
  44     Ei_(0.0),
  45     typeId_(-1)
  46 {
  47     if (readFields)
  48     {
  49         if (is.format() == IOstream::ASCII)
  50         {
  51             is >> U_;
  52             Ei_ = readScalar(is);
  53             typeId_ = readLabel(is);
  54         }
  55         else
  56         {
  57             is.read
  58             (
  59                 reinterpret_cast<char*>(&U_),
  60                 sizeof(U_)
  61               + sizeof(Ei_)
  62               + sizeof(typeId_)
  63             );
  64         }
  65     }
  66
  67     // Check state of Istream
  68     is.check
  69     (
  70         "DsmcParcel<ParcelType>::DsmcParcel"
  71         "(const Cloud<ParcelType>&, Istream&, bool)"
  72     );
  73 }
  74
  75
  76 template<class ParcelType>
  77 void Foam::DsmcParcel<ParcelType>::readFields(Cloud<ParcelType>& c)
  78 {
  79     if (!c.size())
  80     {
  81         return;
  82     }
  83
  84     Particle<ParcelType>::readFields(c);
  85
  86     IOField<vector> U(c.fieldIOobject("U", IOobject::MUST_READ));
  87     c.checkFieldIOobject(c, U);
  88
  89     IOField<scalar> Ei(c.fieldIOobject("Ei", IOobject::MUST_READ));
  90     c.checkFieldIOobject(c, Ei);
  91
  92     IOField<label> typeId(c.fieldIOobject("typeId", IOobject::MUST_READ));
  93     c.checkFieldIOobject(c, typeId);
  94
  95     label i = 0;
  96     forAllIter(typename Cloud<ParcelType>, c, iter)
  97     {
  98         ParcelType& p = iter();
  99
 100         p.U_ = U[i];
 101         p.Ei_ = Ei[i];
 102         p.typeId_ = typeId[i];
 103         i++;
 104     }
 105 }
 106
 107
 108 template<class ParcelType>
 109 void Foam::DsmcParcel<ParcelType>::writeFields(const Cloud<ParcelType>& c)
 110 {
 111     Particle<ParcelType>::writeFields(c);
 112
 113     label np =  c.size();
 114
 115     IOField<vector> U(c.fieldIOobject("U", IOobject::NO_READ), np);
 116     IOField<scalar> Ei(c.fieldIOobject("Ei", IOobject::NO_READ), np);
 117     IOField<label> typeId(c.fieldIOobject("typeId", IOobject::NO_READ), np);
 118
 119     label i = 0;
 120     forAllConstIter(typename Cloud<ParcelType>, c, iter)
 121     {
 122         const DsmcParcel<ParcelType>& p = iter();
 123
 124         U[i] = p.U();
 125         Ei[i] = p.Ei();
 126         typeId[i] = p.typeId();
 127         i++;
 128     }
 129
 130     U.write();
 131     Ei.write();
 132     typeId.write();
 133 }
 134
 135
 136 // * * * * * * * * * * * * * * * IOstream Operators  * * * * * * * * * * * * //
 137
 138 template<class ParcelType>
 139 Foam::Ostream& Foam::operator<<
 140 (
 141     Ostream& os,
 142     const DsmcParcel<ParcelType>& p
 143 )
 144 {
 145     if (os.format() == IOstream::ASCII)
 146     {
 147         os  << static_cast<const Particle<ParcelType>& >(p)
 148             << token::SPACE << p.U()
 149             << token::SPACE << p.Ei()
 150             << token::SPACE << p.typeId();
 151     }
 152     else
 153     {
 154         os  << static_cast<const Particle<ParcelType>& >(p);
 155         os.write
 156         (
 157             reinterpret_cast<const char*>(&p.U_),
 158             sizeof(p.U())
 159           + sizeof(p.Ei())
 160           + sizeof(p.typeId())
 161         );
 162     }
 163
 164     // Check state of Ostream
 165     os.check
 166     (
 167         "Ostream& operator<<(Ostream&, const DsmcParcel<ParcelType>&)"
 168     );
 169
 170     return os;
 171 }
 172
 173
 174 // ************************************************************************* //