src/OpenFOAM/db/IOstreams/Pstreams/OPwrite.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 1991-2008 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 Description
  26     Write primitive and binary block from OPstream
  27
  28 \*---------------------------------------------------------------------------*/
  29
  30 #include "error.H"
  31
  32 #include "OPstream.H"
  33 #include "int.H"
  34 #include "token.H"
  35
  36 #include <cctype>
  37
  38 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  39
  40 namespace Foam
  41 {
  42
  43 // * * * * * * * * * * * * * * * * Constructor * * * * * * * * * * * * * * * //
  44
  45 OPstream::OPstream
  46 (
  47     const commsTypes commsType,
  48     const int toProcNo,
  49     const label bufSize,
  50     streamFormat format,
  51     versionNumber version
  52 )
  53 :
  54     Pstream(commsType, bufSize),
  55     Ostream(format, version),
  56     toProcNo_(toProcNo)
  57 {
  58     setOpened();
  59     setGood();
  60
  61     if (!bufSize)
  62     {
  63         buf_.setSize(1000);
  64     }
  65 }
  66
  67
  68 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  69
  70 Ostream& OPstream::write(const token&)
  71 {
  72     notImplemented("Ostream& OPstream::write(const token&)");
  73     setBad();
  74     return *this;
  75 }
  76
  77
  78 Ostream& OPstream::write(const char c)
  79 {
  80     if (!isspace(c))
  81     {
  82         writeToBuffer(c);
  83     }
  84
  85     return *this;
  86 }
  87
  88
  89 Ostream& OPstream::write(const char* s)
  90 {
  91     word nonWhiteChars(string::validate<word>(s));
  92
  93     if (nonWhiteChars.size() == 0)
  94     {
  95         return *this;
  96     }
  97     else if (nonWhiteChars.size() == 1)
  98     {
  99         return write(nonWhiteChars.c_str()[1]);
 100     }
 101     else
 102     {
 103         return write(nonWhiteChars);
 104     }
 105 }
 106
 107
 108 Ostream& OPstream::write(const word& w)
 109 {
 110     write(char(token::WORD));
 111
 112     size_t ws = w.size();
 113     writeToBuffer(ws);
 114     writeToBuffer(w.c_str(), ws + 1, 1);
 115
 116     return *this;
 117 }
 118
 119
 120 Ostream& OPstream::write(const string& s)
 121 {
 122     write(char(token::STRING));
 123
 124     size_t ss = s.size();
 125     writeToBuffer(ss);
 126     writeToBuffer(s.c_str(), ss + 1, 1);
 127
 128     return *this;
 129 }
 130
 131
 132 Ostream& OPstream::write(const label l)
 133 {
 134     write(char(token::LABEL));
 135
 136     writeToBuffer(l);
 137
 138     return *this;
 139 }
 140
 141
 142 Ostream& OPstream::write(const floatScalar s)
 143 {
 144     write(char(token::FLOAT_SCALAR));
 145
 146     writeToBuffer(s);
 147
 148     return *this;
 149 }
 150
 151
 152 Ostream& OPstream::write(const doubleScalar s)
 153 {
 154     write(char(token::DOUBLE_SCALAR));
 155
 156     writeToBuffer(s);
 157
 158     return *this;
 159 }
 160
 161
 162 Ostream& OPstream::write(const char* data, std::streamsize count)
 163 {
 164     if (format() != BINARY)
 165     {
 166         FatalErrorIn("Ostream::write(const char*, std::streamsize)")
 167             << "stream format not binary"
 168             << Foam::abort(FatalError);
 169     }
 170
 171     writeToBuffer(data, count, 8);
 172
 173     return *this;
 174 }
 175
 176
 177 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 178
 179 } // End namespace Foam
 180
 181 // ************************************************************************* //