src/foam/db/IOstreams/Fstreams/OFstream.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | foam-extend: Open Source CFD
   4    \\    /   O peration     | Version:     3.2
   5     \\  /    A nd           | Web:         http://www.foam-extend.org
   6      \\/     M anipulation  | For copyright notice see file Copyright
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of foam-extend.
  10
  11     foam-extend 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 3 of the License, or (at your
  14     option) any later version.
  15
  16     foam-extend is distributed in the hope that it will be useful, but
  17     WITHOUT ANY WARRANTY; without even the implied warranty of
  18     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19     General Public License for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with foam-extend.  If not, see <http://www.gnu.org/licenses/>.
  23
  24 \*---------------------------------------------------------------------------*/
  25
  26 #include "OFstream.H"
  27 #include "OSspecific.H"
  28 #include "gzstream.h"
  29
  30 // * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
  31
  32 defineTypeNameAndDebug(Foam::OFstream, 0);
  33
  34
  35 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  36
  37 Foam::OFstreamAllocator::OFstreamAllocator
  38 (
  39     const fileName& pathname,
  40     ios_base::openmode mode,
  41     IOstream::compressionType compression
  42 )
  43 :
  44     ofPtr_(NULL)
  45 {
  46     if (pathname.empty())
  47     {
  48         if (OFstream::debug)
  49         {
  50             Info<< "OFstreamAllocator::OFstreamAllocator(const fileName&) : "
  51                    "cannot open null file " << endl;
  52         }
  53     }
  54
  55     if (compression == IOstream::COMPRESSED)
  56     {
  57         // get identically named uncompressed version out of the way
  58         if (isFile(pathname, false))
  59         {
  60             rm(pathname);
  61         }
  62
  63         ofPtr_ = new ogzstream((pathname + ".gz").c_str(), mode);
  64     }
  65     else
  66     {
  67         // get identically named compressed version out of the way
  68         if (isFile(pathname + ".gz", false))
  69         {
  70             rm(pathname + ".gz");
  71         }
  72
  73         ofPtr_ = new ofstream(pathname.c_str(), mode);
  74     }
  75 }
  76
  77
  78 Foam::OFstreamAllocator::~OFstreamAllocator()
  79 {
  80     delete ofPtr_;
  81 }
  82
  83
  84 std::ostream& Foam::OFstreamAllocator::stdStream()
  85 {
  86     if (!ofPtr_)
  87     {
  88         FatalErrorIn("OFstreamAllocator::stdStream()")
  89             << "No stream allocated." << abort(FatalError);
  90     }
  91     return *ofPtr_;
  92 }
  93
  94
  95 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  96
  97 Foam::OFstream::OFstream
  98 (
  99     const fileName& pathname,
 100     ios_base::openmode mode,
 101     streamFormat format,
 102     versionNumber version,
 103     compressionType compression
 104 )
 105 :
 106     OFstreamAllocator(pathname, mode, compression),
 107     OSstream(*ofPtr_, "OFstream.sinkFile_", format, version, compression),
 108     pathname_(pathname)
 109 {
 110     setClosed();
 111     setState(ofPtr_->rdstate());
 112
 113     if (!good())
 114     {
 115         if (debug)
 116         {
 117             Info<< "IFstream::IFstream(const fileName&,"
 118                    "streamFormat format=ASCII,"
 119                    "versionNumber version=currentVersion) : "
 120                    "could not open file for input\n"
 121                    "in stream " << info() << Foam::endl;
 122         }
 123
 124         setBad();
 125     }
 126     else
 127     {
 128         setOpened();
 129     }
 130
 131     lineNumber_ = 1;
 132 }
 133
 134
 135 // * * * * * * * * * * * * * * * * Destructors * * * * * * * * * * * * * * * //
 136
 137 Foam::OFstream::~OFstream()
 138 {}
 139
 140
 141 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 142
 143 void Foam::OFstream::print(Ostream& os) const
 144 {
 145     os  << "    OFstream: ";
 146     OSstream::print(os);
 147 }
 148
 149
 150 // ************************************************************************* //