mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-09-11 11:25:40 +03:00
176 lines
3.6 KiB
C++
176 lines
3.6 KiB
C++
/*
|
|
* This class creates c++ like stream from file descriptor
|
|
* It uses gcc-specific functions, therefore is not portable
|
|
*
|
|
* Jeroen Vermeulen reckons that it can be replaced with Boost's io::stream_buffer
|
|
*
|
|
*/
|
|
|
|
|
|
#ifndef _FDSTREAM_
|
|
#define _FDSTREAM_
|
|
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
#if defined(__GLIBCXX__) || defined(__GLIBCPP__)
|
|
#include <ext/stdio_filebuf.h>
|
|
|
|
#include "util/file.hh"
|
|
|
|
#define BUFFER_SIZE (32768)
|
|
|
|
namespace MosesTuning
|
|
{
|
|
|
|
class _fdstream
|
|
{
|
|
protected:
|
|
_fdstream() :
|
|
_file_descriptor(), _filebuf(NULL) {
|
|
}
|
|
|
|
_fdstream(int file_descriptor, std::ios_base::openmode openmode) :
|
|
_file_descriptor(-1), _openmode(openmode) {
|
|
_filebuf = NULL;
|
|
open(file_descriptor, openmode);
|
|
_file_descriptor.reset(file_descriptor);
|
|
}
|
|
|
|
std::ios_base::openmode openmode() const {
|
|
return _openmode;
|
|
}
|
|
|
|
void open(int file_descriptor, std::ios_base::openmode openmode) {
|
|
// TODO: How does file_descriptor relate to the one we already have?
|
|
// Should we reset our own _file_descriptor to match it?
|
|
if (!_filebuf) {
|
|
// We create a C++ stream from a file descriptor
|
|
// stdio_filebuf is not synced with stdio.
|
|
// From GCC 3.4.0 on exists in addition stdio_sync_filebuf
|
|
// You can also create the filebuf from a FILE* with
|
|
// FILE* f = fdopen(file_descriptor, mode);
|
|
_filebuf = new __gnu_cxx::stdio_filebuf<char> (file_descriptor,
|
|
openmode);
|
|
}
|
|
}
|
|
|
|
virtual ~_fdstream() {
|
|
delete _filebuf;
|
|
_filebuf = NULL;
|
|
}
|
|
|
|
private:
|
|
util::scoped_fd _file_descriptor;
|
|
__gnu_cxx::stdio_filebuf<char>* _filebuf;
|
|
std::ios_base::openmode _openmode;
|
|
|
|
protected:
|
|
/// For child classes only: retrieve filebuf.
|
|
__gnu_cxx::stdio_filebuf<char> *get_filebuf() {
|
|
return _filebuf;
|
|
}
|
|
};
|
|
|
|
class ifdstream : public _fdstream
|
|
{
|
|
public:
|
|
ifdstream() :
|
|
_fdstream(), _stream(NULL) {
|
|
}
|
|
|
|
ifdstream(int file_descriptor) :
|
|
_fdstream(file_descriptor, std::ios_base::in) {
|
|
_stream = new std::istream(get_filebuf());
|
|
}
|
|
|
|
void open(int file_descriptor) {
|
|
if (!_stream) {
|
|
_fdstream::open(file_descriptor, std::ios_base::in);
|
|
_stream = new std::istream(get_filebuf());
|
|
}
|
|
}
|
|
|
|
ifdstream& operator>> (std::string& str) {
|
|
(*_stream) >> str;
|
|
|
|
return *this;
|
|
}
|
|
|
|
std::size_t getline(std::string& str) {
|
|
char tmp[BUFFER_SIZE];
|
|
std::size_t ret = getline(tmp, BUFFER_SIZE);
|
|
str = tmp;
|
|
return ret;
|
|
}
|
|
|
|
std::size_t getline(char* s, std::streamsize n) {
|
|
return (getline(s, n, '\n'));
|
|
}
|
|
|
|
std::size_t getline(char* s, std::streamsize n, char delim) {
|
|
int i = 0;
|
|
do {
|
|
s[i] = _stream->get();
|
|
i++;
|
|
} while(i < n-1 && s[i-1] != delim && s[i-1] != '\0');
|
|
|
|
s[i-1] = '\0'; // overwrite the delimiter given with string end
|
|
|
|
return i-1;
|
|
}
|
|
|
|
~ifdstream() {
|
|
//this->~_fdstream();
|
|
delete _stream;
|
|
}
|
|
|
|
private:
|
|
std::istream* _stream;
|
|
};
|
|
|
|
class ofdstream : public _fdstream
|
|
{
|
|
public:
|
|
ofdstream() :
|
|
_fdstream(), _stream(NULL) {
|
|
}
|
|
|
|
ofdstream(int file_descriptor) :
|
|
_fdstream(file_descriptor, std::ios_base::out) {
|
|
_stream = new std::ostream(get_filebuf());
|
|
}
|
|
|
|
void open(int file_descriptor) {
|
|
if (!_stream) {
|
|
_fdstream::open(file_descriptor, std::ios_base::out);
|
|
_stream = new std::ostream(get_filebuf());
|
|
}
|
|
}
|
|
|
|
|
|
ofdstream& operator<< (const std::string& str) {
|
|
if (_stream->good())
|
|
(*_stream) << str;
|
|
|
|
_stream->flush();
|
|
return *this;
|
|
}
|
|
|
|
~ofdstream() {
|
|
//this->~_fdstream();
|
|
delete _stream;
|
|
}
|
|
|
|
private:
|
|
std::ostream* _stream;
|
|
};
|
|
|
|
#else
|
|
#error "Not supported"
|
|
#endif
|
|
|
|
}
|
|
|
|
#endif // _FDSTREAM_
|