diff options
Diffstat (limited to 'src/io/parallelIO.h')
-rw-r--r-- | src/io/parallelIO.h | 108 |
1 files changed, 108 insertions, 0 deletions
diff --git a/src/io/parallelIO.h b/src/io/parallelIO.h new file mode 100644 index 0000000..bec560e --- /dev/null +++ b/src/io/parallelIO.h @@ -0,0 +1,108 @@ +/* This file is part of the OpenLB library + * + * Copyright (C) 2007 Jonas Latt + * E-mail contact: info@openlb.net + * The most recent release of OpenLB can be downloaded at + * <http://www.openlb.net/> + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public + * License along with this program; if not, write to the Free + * Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + * Boston, MA 02110-1301, USA. +*/ + +#ifndef PARALLEL_IO_H +#define PARALLEL_IO_H + +#include <streambuf> +#include <istream> +#include <ostream> +#include <fstream> +#include <iostream> + +namespace olb { + +class ParBuf : public std::streambuf { +public: + typedef enum {normal, serial} Modes; + // There are two modes for proceeding of parallel i/o: + // - normal: the default mode. all nodes write data, and all + // nodes receive data; this mode requests broadcasting. + // - serial: only node 0 writes and receives data. +public: + ParBuf(std::streambuf* _originalBuf); + Modes getMode() const; + void setMode(Modes _mode); +protected: + int_type overflow (int_type c) override; + std::streamsize xsputn(const char* s, std::streamsize num) override; + + int_type uflow() override; + int_type underflow() override; + std::streamsize xsgetn (char* s, std::streamsize num) override; +private: + std::streambuf* originalBuf; + Modes mode; +}; + +class olb_ofstream : public std::ostream { +public: + olb_ofstream(); + explicit olb_ofstream(const char * filename, + openmode mode = out | trunc ); + ~olb_ofstream() override; + + std::streambuf* rdbuf() const; + bool is_open(); + void open(const char* filename, openmode mode = out | trunc); + void close(); +private: + std::filebuf fbuf; + ParBuf mybuf; +}; + +class olb_ifstream : public std::istream { +public: + olb_ifstream(); + explicit olb_ifstream(const char * filename, + openmode mode = in ); + ~olb_ifstream() override; + + std::streambuf* rdbuf() const; + bool is_open(); + void open(const char* filename, openmode mode = in); + void close(); +private: + std::filebuf fbuf; + ParBuf mybuf; +}; + +class olb_fstream : public std::iostream { +public: + olb_fstream(); + explicit olb_fstream(const char * filename, + openmode mode = in | out ); + ~olb_fstream() override; + + std::streambuf* rdbuf() const; + bool is_open(); + void open(const char* filename, openmode mode = in | out); + void close(); +private: + std::filebuf fbuf; + ParBuf mybuf; +}; + +} // namespace olb + +#endif |