1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
/*************************************************************************
* odil - Copyright (C) Universite de Strasbourg
* Distributed under the terms of the CeCILL-B license, as published by
* the CEA-CNRS-INRIA. Refer to the LICENSE file or to
* http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
* for details.
************************************************************************/
#ifndef _fc7ce52d_b32d_43c4_87fc_465b674e6726
#define _fc7ce52d_b32d_43c4_87fc_465b674e6726
#include <ios>
#include <iostream>
#include <streambuf>
#include <string>
#include <pybind11/pybind11.h>
namespace odil
{
namespace wrappers
{
namespace python
{
/// @brief streambuf wrapper around Python byte-based file-like objects.
class streambuf : public std::streambuf
{
public:
streambuf(pybind11::object object, std::string::size_type buffer_size=4096);
~streambuf() =default;
streambuf(streambuf const &) =delete;
streambuf(streambuf &&) =default;
streambuf & operator=(streambuf const &) =delete;
streambuf & operator=(streambuf &&) =default;
protected:
std::streambuf::pos_type
seekoff(
std::streambuf::off_type off, std::ios_base::seekdir dir,
std::ios_base::openmode which = std::ios_base::in | std::ios_base::out);
std::streambuf::pos_type
seekpos(
std::streambuf::pos_type pos,
std::ios_base::openmode which = std::ios_base::in | std::ios_base::out);
int underflow();
int uflow();
int overflow(int ch=std::char_traits<char>::eof());
private:
pybind11::object _object;
std::string::size_type _buffer_size;
std::string _buffer;
std::string::size_type _current;
void _update_buffer();
};
class iostream: public std::iostream
{
public:
iostream(pybind11::object file_like)
: std::iostream(nullptr), _streambuf(file_like)
{
this->rdbuf(&this->_streambuf);
}
private:
streambuf _streambuf;
};
}
}
}
#endif // _fc7ce52d_b32d_43c4_87fc_465b674e6726
|
