66 lines
1.6 KiB
C
66 lines
1.6 KiB
C
|
#ifndef _STREAM_FIX_H_
|
||
|
|
#define _STREAM_FIX_H_
|
||
|
|
#include <condition_variable>
|
||
|
|
#include <iostream>
|
||
|
|
#include <mutex>
|
||
|
|
#include <queue>
|
||
|
|
#include <sstream>
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
inline std::string stringify(const T& input) {
|
||
|
|
std::ostringstream output; // from www .ja va 2s . com
|
||
|
|
output << input;
|
||
|
|
return output.str();
|
||
|
|
}
|
||
|
|
|
||
|
|
class FakeStream {
|
||
|
|
public:
|
||
|
|
template <typename T>
|
||
|
|
FakeStream& operator<<(const T& val) {
|
||
|
|
if (closed) return *this;
|
||
|
|
std::lock_guard<std::mutex> lock(mutex_guard);
|
||
|
|
string_queue.push(stringify(val));
|
||
|
|
mutex_signal.notify_one();
|
||
|
|
return *this;
|
||
|
|
};
|
||
|
|
template <typename T>
|
||
|
|
FakeStream& operator>>(T& val) {
|
||
|
|
if (closed) return *this;
|
||
|
|
std::unique_lock<std::mutex> lock(mutex_guard);
|
||
|
|
mutex_signal.wait(lock,
|
||
|
|
[this] { return !string_queue.empty() || closed; });
|
||
|
|
if (closed) return *this;
|
||
|
|
val = string_queue.front();
|
||
|
|
string_queue.pop();
|
||
|
|
return *this;
|
||
|
|
};
|
||
|
|
|
||
|
|
bool try_get_line(std::string& val);
|
||
|
|
|
||
|
|
void close();
|
||
|
|
bool is_closed();
|
||
|
|
|
||
|
|
std::streambuf* rdbuf();
|
||
|
|
std::streambuf* rdbuf(std::streambuf* __sb);
|
||
|
|
|
||
|
|
private:
|
||
|
|
bool closed = false;
|
||
|
|
std::queue<std::string> string_queue;
|
||
|
|
std::string line;
|
||
|
|
std::mutex mutex_guard;
|
||
|
|
std::condition_variable mutex_signal;
|
||
|
|
};
|
||
|
|
|
||
|
|
namespace std {
|
||
|
|
bool getline(FakeStream& is, std::string& str);
|
||
|
|
} // namespace std
|
||
|
|
|
||
|
|
// #define endl fakeendl
|
||
|
|
// #define cout fakeout
|
||
|
|
// #define cin fakein
|
||
|
|
|
||
|
|
extern FakeStream fakeout;
|
||
|
|
extern FakeStream fakein;
|
||
|
|
extern std::string fakeendl;
|
||
|
|
|
||
|
|
#endif
|