mirror/blah
1
0
Fork 0
mirror of https://github.com/NoelFB/blah.git synced 2024-11-25 16:18:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

added resize method to bufferstream

Browse Source
This commit is contained in:
Noel Berry 2021-05-06 20:47:52 -07:00
parent 5f9df4beee
commit 60f3939ff7
2 changed files with 27 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/blah/streams/bufferstream.h
View File

@ -19,6 +19,8 @@ namespace Blah
virtual bool is_readable() const override { return true; } virtual bool is_readable() const override { return true; }
virtual bool is_writable() const override { return true; } virtual bool is_writable() const override { return true; }
virtual void close() override; virtual void close() override;
void resize(i64 length);
void clear() { m_length = m_position = 0; } void clear() { m_length = m_position = 0; }
char* data() { return m_buffer; } char* data() { return m_buffer; }

38
src/streams/bufferstream.cpp
View File

@ -1,4 +1,4 @@
#include <blah/streams/bufferstream.h> #include "blah/streams/bufferstream.h"
#include <string.h> #include <string.h>
using namespace Blah; using namespace Blah;
@ -64,13 +64,33 @@ i64 BufferStream::write_from(const void* ptr, i64 len)
return 0; return 0;
// resize // resize
if (m_position + len >= m_capacity) if (m_position + len > m_length)
resize(m_position + len);
// copy data
if (ptr != nullptr)
memcpy(m_buffer + m_position, ptr, (size_t)len);
// increment position
m_position += len;
// return the amount we wrote
return len;
}
void BufferStream::resize(i64 length)
{
if (m_capacity > length)
{
m_length = length;
}
else
{ {
auto last_capacity = m_capacity; auto last_capacity = m_capacity;
if (m_capacity <= 0) if (m_capacity <= 0)
m_capacity = 16; m_capacity = 16;
while (m_position + len >= m_capacity) while (length >= m_capacity)
m_capacity *= 2; m_capacity *= 2;
char* new_buffer = new char[m_capacity]; char* new_buffer = new char[m_capacity];
@ -84,17 +104,7 @@ i64 BufferStream::write_from(const void* ptr, i64 len)
m_buffer = new_buffer; m_buffer = new_buffer;
} }
// copy data m_length = length;
if (ptr != nullptr)
memcpy(m_buffer + m_position, ptr, (size_t)len);
// increment position
m_position += len;
if (m_position > m_length)
m_length = m_position;
// return the amount we wrote
return len;
} }
void BufferStream::close() void BufferStream::close()