77 lines
2.0 KiB
C++
77 lines
2.0 KiB
C++
#ifndef OSTREAM_WRAPPER_H_62B23520_7C8E_11DE_8A39_0800200C9A66
|
|
#define OSTREAM_WRAPPER_H_62B23520_7C8E_11DE_8A39_0800200C9A66
|
|
|
|
#if defined(_MSC_VER) || \
|
|
(defined(__GNUC__) && (__GNUC__ == 3 && __GNUC_MINOR__ >= 4) || \
|
|
(__GNUC__ >= 4)) // GCC supports "pragma once" correctly since 3.4
|
|
#pragma once
|
|
#endif
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "yaml-cpp/dll.h"
|
|
|
|
namespace YAML {
|
|
class YAML_CPP_API ostream_wrapper {
|
|
public:
|
|
ostream_wrapper();
|
|
explicit ostream_wrapper(std::ostream& stream);
|
|
ostream_wrapper(const ostream_wrapper&) = delete;
|
|
ostream_wrapper(ostream_wrapper&&) = delete;
|
|
ostream_wrapper& operator=(const ostream_wrapper&) = delete;
|
|
ostream_wrapper& operator=(ostream_wrapper&&) = delete;
|
|
~ostream_wrapper();
|
|
|
|
void write(const std::string& str);
|
|
void write(const char* str, std::size_t size);
|
|
|
|
void set_comment() { m_comment = true; }
|
|
|
|
const char* str() const {
|
|
if (m_pStream) {
|
|
return nullptr;
|
|
} else {
|
|
m_buffer[m_pos] = '\0';
|
|
return &m_buffer[0];
|
|
}
|
|
}
|
|
|
|
std::size_t row() const { return m_row; }
|
|
std::size_t col() const { return m_col; }
|
|
std::size_t pos() const { return m_pos; }
|
|
bool comment() const { return m_comment; }
|
|
|
|
private:
|
|
void update_pos(char ch);
|
|
|
|
private:
|
|
mutable std::vector<char> m_buffer;
|
|
std::ostream* const m_pStream;
|
|
|
|
std::size_t m_pos;
|
|
std::size_t m_row, m_col;
|
|
bool m_comment;
|
|
};
|
|
|
|
template <std::size_t N>
|
|
inline ostream_wrapper& operator<<(ostream_wrapper& stream,
|
|
const char (&str)[N]) {
|
|
stream.write(str, N - 1);
|
|
return stream;
|
|
}
|
|
|
|
inline ostream_wrapper& operator<<(ostream_wrapper& stream,
|
|
const std::string& str) {
|
|
stream.write(str);
|
|
return stream;
|
|
}
|
|
|
|
inline ostream_wrapper& operator<<(ostream_wrapper& stream, char ch) {
|
|
stream.write(&ch, 1);
|
|
return stream;
|
|
}
|
|
} // namespace YAML
|
|
|
|
#endif // OSTREAM_WRAPPER_H_62B23520_7C8E_11DE_8A39_0800200C9A66
|