Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix dmlc io

Browse Source
This commit is contained in:
tqchen 2015-04-11 18:16:52 -07:00
parent ad383b084d
commit bcfbe51e7e
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
include/dmlc/io.h
View File

@ -150,6 +150,10 @@ class ostream : public std::basic_ostream<char> {
: basic_ostream<char>(NULL), buf_(buffer_size) { : basic_ostream<char>(NULL), buf_(buffer_size) {
this->set_stream(stream); this->set_stream(stream);
} }
// explictly synchronize the buffer
virtual ~ostream() {
buf_.pubsync();
}
/*! /*!
* \brief set internal stream to be stream, reset states * \brief set internal stream to be stream, reset states
* \param stream new stream as output * \param stream new stream as output
@ -209,6 +213,7 @@ class istream : public std::basic_istream<char> {
: basic_istream<char>(NULL), buf_(buffer_size) { : basic_istream<char>(NULL), buf_(buffer_size) {
this->set_stream(stream); this->set_stream(stream);
} }
virtual ~istream() {}
/*! /*!
* \brief set internal stream to be stream, reset states * \brief set internal stream to be stream, reset states
* \param stream new stream as output * \param stream new stream as output
@ -281,6 +286,7 @@ inline bool Stream::Read(std::string *out_str) {
// implementations for ostream // implementations for ostream
inline void ostream::OutBuf::set_stream(Stream *stream) { inline void ostream::OutBuf::set_stream(Stream *stream) {
if (stream_ != NULL) this->pubsync();
this->stream_ = stream; this->stream_ = stream;
this->setp(&buffer_[0], &buffer_[0] + buffer_.size() - 1); this->setp(&buffer_[0], &buffer_[0] + buffer_.size() - 1);
} }