src/OpenFOAM/primitives/bools/Switch/SwitchIO.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 1991-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 "Switch.H"
  28 #include "IOstreams.H"
  29
  30 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  31
  32 Foam::Switch::Switch(Istream& is)
  33 {
  34     is >> *this;
  35 }
  36
  37
  38 // * * * * * * * * * * * * * * * IOstream Operators  * * * * * * * * * * * * //
  39
  40 Foam::Istream& Foam::operator>>(Istream& is, Switch& s)
  41 {
  42     token t(is);
  43
  44     if (!t.good())
  45     {
  46         is.setBad();
  47         return is;
  48     }
  49
  50     if (t.isLabel())
  51     {
  52         s.switch_ = Switch::asEnum(bool(t.labelToken()));
  53     }
  54     else if (t.isWord())
  55     {
  56         // allow invalid values, but catch after for correct error message
  57         Switch::switchType sw = Switch::asEnum(t.wordToken(), true);
  58
  59         if (sw == Switch::INVALID)
  60         {
  61             is.setBad();
  62             FatalIOErrorIn("operator>>(Istream&, Switch&)", is)
  63                 << "expected 'true/false', 'on/off' ... found " << t.wordToken()
  64                 << exit(FatalIOError);
  65
  66             return is;
  67         }
  68         else
  69         {
  70             s.switch_ = sw;
  71         }
  72     }
  73     else
  74     {
  75         is.setBad();
  76         FatalIOErrorIn("operator>>(Istream&, bool/Switch&)", is)
  77             << "wrong token type - expected bool found " << t
  78             << exit(FatalIOError);
  79
  80         return is;
  81     }
  82
  83
  84     // Check state of Istream
  85     is.check("Istream& operator>>(Istream&, Switch&)");
  86
  87     return is;
  88 }
  89
  90
  91 Foam::Ostream& Foam::operator<<(Ostream& os, const Switch& s)
  92 {
  93     os << Switch::names[s.switch_];
  94     os.check("Ostream& operator<<(Ostream&, const Switch&)");
  95     return os;
  96 }
  97
  98
  99 // ************************************************************************* //