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 (C) 2009-2009 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 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(DsmcCloud<ParcelType>& c)
  78 {
  79     if (!c.size())
  80     {
  81         return;
  82     }
  83
  84     IOField<vector> U(c.fieldIOobject("U", IOobject::MUST_READ));
  85     c.checkFieldIOobject(c, U);
  86
  87     IOField<scalar> Ei(c.fieldIOobject("Ei", IOobject::MUST_READ));
  88     c.checkFieldIOobject(c, Ei);
  89
  90     IOField<label> typeId(c.fieldIOobject("typeId", IOobject::MUST_READ));
  91     c.checkFieldIOobject(c, typeId);
  92
  93     label i = 0;
  94     forAllIter(typename Cloud<ParcelType>, c, iter)
  95     {
  96         ParcelType& p = iter();
  97
  98         p.U_ = U[i];
  99         p.Ei_ = Ei[i];
 100         p.typeId_ = typeId[i];
 101         i++;
 102     }
 103 }
 104
 105
 106 template<class ParcelType>
 107 void Foam::DsmcParcel<ParcelType>::writeFields(const DsmcCloud<ParcelType>& c)
 108 {
 109     Particle<ParcelType>::writeFields(c);
 110
 111     label np =  c.size();
 112
 113     IOField<vector> U(c.fieldIOobject("U", IOobject::NO_READ), np);
 114     IOField<scalar> Ei(c.fieldIOobject("Ei", IOobject::NO_READ), np);
 115     IOField<label> typeId(c.fieldIOobject("typeId", IOobject::NO_READ), np);
 116
 117     label i = 0;
 118     forAllConstIter(typename Cloud<ParcelType>, c, iter)
 119     {
 120         const DsmcParcel<ParcelType>& p = iter();
 121
 122         U[i] = p.U();
 123         Ei[i] = p.Ei();
 124         typeId[i] = p.typeId();
 125         i++;
 126     }
 127
 128     U.write();
 129     Ei.write();
 130     typeId.write();
 131 }
 132
 133
 134 // * * * * * * * * * * * * * * * IOstream Operators  * * * * * * * * * * * * //
 135
 136 template<class ParcelType>
 137 Foam::Ostream& Foam::operator<<
 138 (
 139     Ostream& os,
 140     const DsmcParcel<ParcelType>& p
 141 )
 142 {
 143     if (os.format() == IOstream::ASCII)
 144     {
 145         os  << static_cast<const Particle<ParcelType>& >(p)
 146             << token::SPACE << p.U()
 147             << token::SPACE << p.Ei()
 148             << token::SPACE << p.typeId();
 149     }
 150     else
 151     {
 152         os  << static_cast<const Particle<ParcelType>& >(p);
 153         os.write
 154         (
 155             reinterpret_cast<const char*>(&p.U_),
 156             sizeof(p.U())
 157           + sizeof(p.Ei())
 158           + sizeof(p.typeId())
 159         );
 160     }
 161
 162     // Check state of Ostream
 163     os.check
 164     (
 165         "Ostream& operator<<(Ostream&, const DsmcParcel<ParcelType>&)"
 166     );
 167
 168     return os;
 169 }
 170
 171
 172 // ************************************************************************* //