ostream_wrapper.cpp 1.84 KB
Newer Older
1
#include "yaml-cpp/ostream_wrapper.h"
2
3
4
5
#include <cstring>

namespace YAML
{
6
	ostream_wrapper::ostream_wrapper(): m_buffer(0), m_pos(0), m_size(0), m_row(0), m_col(0), m_comment(false)
7
8
9
10
	{
		reserve(1024);
	}
	
11
	ostream_wrapper::~ostream_wrapper()
12
13
14
15
	{
		delete [] m_buffer;
	}
	
16
	void ostream_wrapper::reserve(std::size_t size)
17
18
19
20
21
22
23
24
25
26
27
	{
		if(size <= m_size)
			return;
		
		char *newBuffer = new char[size];
		std::memset(newBuffer, 0, size * sizeof(char));
		std::memcpy(newBuffer, m_buffer, m_size * sizeof(char));
		delete [] m_buffer;
		m_buffer = newBuffer;
		m_size = size;
	}
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
    
    void ostream_wrapper::write(const std::string& str)
    {
        while(m_pos + str.size() + 1 >= m_size)
            reserve(m_size * 2);
        
        std::copy(str.begin(), str.end(), m_buffer + m_pos);
        
        for(std::size_t i=0;i<str.size();i++)
            update_pos(str[i]);
    }

    void ostream_wrapper::write(const char *str, std::size_t size)
    {
        while(m_pos + size + 1 >= m_size)
            reserve(m_size * 2);

        std::copy(str, str + size, m_buffer + m_pos);
        
        for(std::size_t i=0;i<size;i++)
            update_pos(str[i]);
    }
50
	
51
	void ostream_wrapper::put(char ch)
52
53
54
55
56
	{
		if(m_pos >= m_size - 1)   // an extra space for the NULL terminator
			reserve(m_size * 2);
		
		m_buffer[m_pos] = ch;
57
58
59
60
61
        update_pos(ch);
	}

    void ostream_wrapper::update_pos(char ch)
    {
62
		m_pos++;
63
        m_col++;
64
65
66
67
		
		if(ch == '\n') {
			m_row++;
			m_col = 0;
68
            m_comment = false;
69
70
		}
    }
71

72
	ostream_wrapper& operator << (ostream_wrapper& out, const char *str)
73
	{
74
        out.write(str, std::strlen(str));
75
76
77
		return out;
	}
	
78
	ostream_wrapper& operator << (ostream_wrapper& out, const std::string& str)
79
	{
80
		out.write(str);
81
82
83
		return out;
	}
	
84
	ostream_wrapper& operator << (ostream_wrapper& out, char ch)
85
86
87
88
89
	{
		out.put(ch);
		return out;
	}
}