peisongxiao/yaml-cpp
1
0
Fork 0
mirror of https://github.com/jbeder/yaml-cpp.git synced 2025-09-09 12:41:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated ostream wrapper with a write() and update_pos

Browse Source
This commit is contained in:
Jesse Beder
2012-05-25 17:39:14 -05:00
parent 1602f78974
commit 7c85e9d5de
2 changed files with 46 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
src/ostream_wrapper.cpp
View File

@@ -13,7 +13,7 @@ namespace YAML
delete [] m_buffer;
}
void ostream_wrapper::reserve(unsigned size)
void ostream_wrapper::reserve(std::size_t size)
{
if(size <= m_size)
return;
@@ -25,6 +25,28 @@ namespace YAML
m_buffer = newBuffer;
m_size = size;
}
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]);
}
void ostream_wrapper::put(char ch)
{
@@ -32,27 +54,30 @@ namespace YAML
reserve(m_size * 2);
m_buffer[m_pos] = ch;
update_pos(ch);
}
void ostream_wrapper::update_pos(char ch)
{
m_pos++;
m_col++;
if(ch == '\n') {
m_row++;
m_col = 0;
m_comment = false;
} else
m_col++;
}
}
}
ostream_wrapper& operator << (ostream_wrapper& out, const char *str)
{
std::size_t length = std::strlen(str);
for(std::size_t i=0;i<length;i++)
out.put(str[i]);
out.write(str, std::strlen(str));
return out;
}
ostream_wrapper& operator << (ostream_wrapper& out, const std::string& str)
{
out << str.c_str();
out.write(str);
return out;
}