fdiostream.h

// -*- mode: c++; c-basic-offset:4 -*-

// This file is part of libdap, A C++ implementation of the OPeNDAP Data
// Access Protocol.

// Copyright (c) 2009 OPeNDAP, Inc.
// Author: James Gallagher <jgallagher@opendap.org>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//
// You can contact OPeNDAP, Inc. at PO Box 112, Saunderstown, RI. 02874-0112.
//
// Portions of this code were taken verbatim from  Josuttis,
// "The C++ Standard Library," p.672

#ifndef _fdiostream_h
#define _fdiostream_h

#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif

#include <iostream>
#include <streambuf>
#include <algorithm>
#include <cstdio>

namespace libdap {

class fdoutbuf: public std::streambuf {
protected:
    int fd; // file descriptor
    bool close;
    static const int bufferSize = 4096; // Size of the data buffer
    char buffer[bufferSize];    // data buffer

public:
    fdoutbuf(int _fd, bool _close);
    virtual ~fdoutbuf();

protected:
    int flushBuffer();

    virtual int overflow(int c);
    virtual int sync();
    virtual std::streamsize xsputn(const char *s, std::streamsize num);
};

class fdostream: public std::ostream {
protected:
    fdoutbuf buf;
public:
    fdostream(int _fd, bool _close = false) :
            std::ostream(&buf), buf(_fd, _close)
    {
    }
};

class fdinbuf: public std::streambuf {
protected:
    int fd; // file descriptor
    bool close;
    static const int bufferSize = 4096; // Size of the data buffer
    static const int putBack = 128;
    char buffer[bufferSize];    // data buffer

public:
    fdinbuf(int _fd, bool close);
    virtual ~fdinbuf();

protected:
    virtual int underflow();
};

class fdistream: public std::istream {
protected:
    fdinbuf buf;
public:
    fdistream(int fd, bool close = false) :
            std::istream(&buf), buf(fd, close)
    {
    }
};

class fpinbuf: public std::streambuf {
protected:
    FILE *fp;   // FILE *
    bool close;
    static const int bufferSize = 4096; // Size of the data buffer
    static const int putBack = 128;
    char buffer[bufferSize];    // data buffer

public:
    fpinbuf(FILE *_fp, bool _close);
    virtual ~fpinbuf();

protected:
    virtual int underflow();
};

class fpistream: public std::istream {
protected:
    fpinbuf buf;
public:
    fpistream(FILE *fp, bool close = false) :
            std::istream(&buf), buf(fp, close)
    {
    }
};

}

#endif