| #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(); |
| |
| 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 0; |
| } 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; |
| } |
| } |
| |
| #endif // OSTREAM_WRAPPER_H_62B23520_7C8E_11DE_8A39_0800200C9A66 |