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

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 2011 OpenFOAM Foundation
   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
  13     the Free Software Foundation, either version 3 of the License, or
  14     (at your 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, 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     IOstream::compressionType compression
  41 )
  42 :
  43     ofPtr_(NULL)
  44 {
  45     if (pathname.empty())
  46     {
  47         if (OFstream::debug)
  48         {
  49             Info<< "OFstreamAllocator::OFstreamAllocator(const fileName&) : "
  50                    "cannot open null file " << endl;
  51         }
  52     }
  53
  54     if (compression == IOstream::COMPRESSED)
  55     {
  56         // get identically named uncompressed version out of the way
  57         if (isFile(pathname, false))
  58         {
  59             rm(pathname);
  60         }
  61
  62         ofPtr_ = new ogzstream((pathname + ".gz").c_str());
  63     }
  64     else
  65     {
  66         // get identically named compressed version out of the way
  67         if (isFile(pathname + ".gz", false))
  68         {
  69             rm(pathname + ".gz");
  70         }
  71
  72         ofPtr_ = new ofstream(pathname.c_str());
  73     }
  74 }
  75
  76
  77 Foam::OFstreamAllocator::~OFstreamAllocator()
  78 {
  79     delete ofPtr_;
  80 }
  81
  82
  83 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  84
  85 Foam::OFstream::OFstream
  86 (
  87     const fileName& pathname,
  88     streamFormat format,
  89     versionNumber version,
  90     compressionType compression
  91 )
  92 :
  93     OFstreamAllocator(pathname, compression),
  94     OSstream(*ofPtr_, "OFstream.sinkFile_", format, version, compression),
  95     pathname_(pathname)
  96 {
  97     setClosed();
  98     setState(ofPtr_->rdstate());
  99
 100     if (!good())
 101     {
 102         if (debug)
 103         {
 104             Info<< "IFstream::IFstream(const fileName&,"
 105                    "streamFormat format=ASCII,"
 106                    "versionNumber version=currentVersion) : "
 107                    "could not open file for input\n"
 108                    "in stream " << info() << Foam::endl;
 109         }
 110
 111         setBad();
 112     }
 113     else
 114     {
 115         setOpened();
 116     }
 117
 118     lineNumber_ = 1;
 119 }
 120
 121
 122 // * * * * * * * * * * * * * * * * Destructor  * * * * * * * * * * * * * * * //
 123
 124 Foam::OFstream::~OFstream()
 125 {}
 126
 127
 128 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 129
 130 std::ostream& Foam::OFstream::stdStream()
 131 {
 132     if (!ofPtr_)
 133     {
 134         FatalErrorIn("OFstream::stdStream()")
 135             << "No stream allocated." << abort(FatalError);
 136     }
 137     return *ofPtr_;
 138 }
 139
 140
 141 const std::ostream& Foam::OFstream::stdStream() const
 142 {
 143     if (!ofPtr_)
 144     {
 145         FatalErrorIn("OFstreamAllocator::stdStream() const")
 146             << "No stream allocated." << abort(FatalError);
 147     }
 148     return *ofPtr_;
 149 }
 150
 151
 152 void Foam::OFstream::print(Ostream& os) const
 153 {
 154     os  << "    OFstream: ";
 155     OSstream::print(os);
 156 }
 157
 158
 159 // ************************************************************************* //