src/OpenFOAM/db/IOstreams/Fstreams/IFstream.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 \*---------------------------------------------------------------------------*/
  26
  27 #include "IFstream.H"
  28 #include "OSspecific.H"
  29 #include "gzstream.h"
  30
  31 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  32
  33 namespace Foam
  34 {
  35
  36 // * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
  37
  38 defineTypeNameAndDebug(IFstream, 0);
  39
  40 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  41
  42 IFstreamAllocator::IFstreamAllocator(const fileName& pathname)
  43 :
  44     ifPtr_(NULL),
  45     compression_(IOstream::UNCOMPRESSED)
  46 {
  47     if (!pathname.size())
  48     {
  49         if (IFstream::debug)
  50         {
  51             Info<< "IFstreamAllocator::IFstreamAllocator"
  52                    "(const fileName& pathname) : "
  53                    "can't open null file "
  54                 << endl;
  55         }
  56     }
  57
  58     ifPtr_ = new ifstream(pathname.c_str());
  59
  60     // If the file is compressed, decompress it before reading.
  61     if (!ifPtr_->good() && file(pathname + ".gz"))
  62     {
  63         if (IFstream::debug)
  64         {
  65             Info<< "IFstreamAllocator::IFstreamAllocator"
  66                    "(const fileName& pathname) : "
  67                    "decompressing " << pathname + ".gz"
  68                 << endl;
  69         }
  70
  71         delete ifPtr_;
  72
  73         ifPtr_ = new igzstream((pathname + ".gz").c_str());
  74
  75         if (ifPtr_->good())
  76         {
  77             compression_ = IOstream::COMPRESSED;
  78         }
  79     }
  80 }
  81
  82
  83 IFstreamAllocator::~IFstreamAllocator()
  84 {
  85     delete ifPtr_;
  86 }
  87
  88
  89 istream& IFstreamAllocator::stdStream()
  90 {
  91     if (!ifPtr_)
  92     {
  93         FatalErrorIn("IFstreamAllocator::stdStream()")
  94             << "No stream allocated." << abort(FatalError);
  95     }
  96     return *ifPtr_;
  97 }
  98
  99
 100 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
 101
 102 IFstream::IFstream
 103 (
 104     const fileName& pathname,
 105     streamFormat format,
 106     versionNumber version
 107 )
 108 :
 109     IFstreamAllocator(pathname),
 110     ISstream
 111     (
 112         *ifPtr_,
 113         "IFstream.sourceFile_",
 114         format,
 115         version,
 116         IFstreamAllocator::compression_
 117     ),
 118     pathname_(pathname)
 119 {
 120     setClosed();
 121
 122     setState(ifPtr_->rdstate());
 123
 124     if (!good())
 125     {
 126         if (debug)
 127         {
 128             Info<< "IFstream::IFstream(const fileName& pathname,"
 129                    "streamFormat format=ASCII,"
 130                    "versionNumber version=currentVersion) : "
 131                    "couldn't open File for input"
 132                 << endl << info() << endl;
 133         }
 134
 135         setBad();
 136     }
 137     else
 138     {
 139         setOpened();
 140     }
 141
 142     lineNumber_ = 1;
 143 }
 144
 145
 146 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
 147
 148 IFstream::~IFstream()
 149 {}
 150
 151
 152 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 153
 154 void IFstream::print(Ostream& os) const
 155 {
 156     // Print File data
 157     os  << "IFstream: ";
 158     ISstream::print(os);
 159 }
 160
 161
 162 //- Return a non-const reference to const Istream
 163 //  Needed for read-constructors where the stream argument is temporary:
 164 //  e.g. thing thisThing(IFstream("thingFileName")());
 165 IFstream& IFstream::operator()() const
 166 {
 167     if (!good())
 168     {
 169         if (!file(pathname_) && !file(pathname_ + ".gz"))
 170         {
 171             FatalIOErrorIn("IFstream::operator()", *this)
 172                 << "file " << pathname_ << " does not exist"
 173                 << exit(FatalIOError);
 174         }
 175         else
 176         {
 177             check("IFstream::operator()");
 178             FatalIOError.exit();
 179         }
 180     }
 181
 182     return const_cast<IFstream&>(*this);
 183 }
 184
 185
 186 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 187
 188 } // End namespace Foam
 189
 190 // ************************************************************************* //