BAN: Add basic String implementation
This commit is contained in:
parent
19307ce197
commit
00f230fcb2
|
@ -0,0 +1,184 @@
|
|||
#pragma
|
||||
|
||||
#include <BAN/Errors.h>
|
||||
#include <BAN/Formatter.h>
|
||||
#include <BAN/Memory.h>
|
||||
|
||||
#include <assert.h>
|
||||
#include <string.h>
|
||||
#include <sys/param.h>
|
||||
|
||||
namespace BAN
|
||||
{
|
||||
|
||||
class String
|
||||
{
|
||||
public:
|
||||
using size_type = size_t;
|
||||
|
||||
public:
|
||||
String() = default;
|
||||
String(const char*);
|
||||
~String();
|
||||
|
||||
ErrorOr<void> PushBack(char);
|
||||
ErrorOr<void> Insert(char, size_type);
|
||||
ErrorOr<void> Append(const char*);
|
||||
ErrorOr<void> Append(const String&);
|
||||
|
||||
void PopBack();
|
||||
void Remove(size_type);
|
||||
|
||||
char operator[](size_type) const;
|
||||
char& operator[](size_type);
|
||||
|
||||
ErrorOr<void> Resize(size_type, char = '\0');
|
||||
ErrorOr<void> Reserve(size_type);
|
||||
|
||||
size_type Size() const;
|
||||
size_type Capasity() const;
|
||||
|
||||
const char* Data() const;
|
||||
|
||||
private:
|
||||
ErrorOr<void> EnsureCapasity(size_type);
|
||||
|
||||
private:
|
||||
char* m_data = nullptr;
|
||||
size_type m_capasity = 0;
|
||||
size_type m_size = 0;
|
||||
};
|
||||
|
||||
String::String(const char* string)
|
||||
{
|
||||
size_type len = strlen(string);
|
||||
MUST(EnsureCapasity(len + 1));
|
||||
memcpy(m_data, string, len);
|
||||
m_data[len] = '\0';
|
||||
m_size = len;
|
||||
}
|
||||
|
||||
String::~String()
|
||||
{
|
||||
BAN::deallocator(m_data);
|
||||
}
|
||||
|
||||
ErrorOr<void> String::PushBack(char ch)
|
||||
{
|
||||
TRY(EnsureCapasity(m_size + 2));
|
||||
m_data[m_size] = ch;
|
||||
m_data[m_size + 1] = '\0';
|
||||
m_size++;
|
||||
return {};
|
||||
}
|
||||
|
||||
ErrorOr<void> String::Insert(char ch, size_type index)
|
||||
{
|
||||
assert(index <= m_size);
|
||||
TRY(EnsureCapasity(m_size + 2));
|
||||
memmove(m_data + index + 1, m_data + index, m_size - index);
|
||||
m_data[index] = ch;
|
||||
m_data[m_size + 1] = '\0';
|
||||
m_size++;
|
||||
return {};
|
||||
}
|
||||
|
||||
ErrorOr<void> String::Append(const char* string)
|
||||
{
|
||||
size_t len = strlen(string);
|
||||
TRY(EnsureCapasity(m_size + len + 1));
|
||||
memcpy(m_data + m_size, string, len);
|
||||
m_data[m_size + len] = '\0';
|
||||
m_size += len;
|
||||
return {};
|
||||
}
|
||||
|
||||
ErrorOr<void> String::Append(const String& string)
|
||||
{
|
||||
TRY(Append(string.Data()));
|
||||
return {};
|
||||
}
|
||||
|
||||
void String::PopBack()
|
||||
{
|
||||
assert(m_size > 0);
|
||||
m_data[m_size - 1] = '\0';
|
||||
m_size--;
|
||||
}
|
||||
|
||||
void String::Remove(size_type index)
|
||||
{
|
||||
assert(index < m_size);
|
||||
memmove(m_data + index, m_data + index + 1, m_size - index - 1);
|
||||
m_data[m_size - 1] = '\0';
|
||||
m_size--;
|
||||
}
|
||||
|
||||
char String::operator[](size_type index) const
|
||||
{
|
||||
assert(index < m_size);
|
||||
return m_data[index];
|
||||
}
|
||||
|
||||
char& String::operator[](size_type index)
|
||||
{
|
||||
assert(index < m_size);
|
||||
return m_data[index];
|
||||
}
|
||||
|
||||
ErrorOr<void> String::Resize(size_type size, char ch)
|
||||
{
|
||||
if (size < m_size)
|
||||
{
|
||||
m_data[size] = '\0';
|
||||
m_size = size;
|
||||
}
|
||||
else if (size > m_size)
|
||||
{
|
||||
TRY(EnsureCapasity(size + 1));
|
||||
for (size_type i = m_size; i < size; i++)
|
||||
m_data[i] = ch;
|
||||
m_data[size] = '\0';
|
||||
m_size = size;
|
||||
}
|
||||
m_size = size;
|
||||
return {};
|
||||
}
|
||||
|
||||
ErrorOr<void> String::Reserve(size_type size)
|
||||
{
|
||||
TRY(EnsureCapasity(size + 1));
|
||||
return {};
|
||||
}
|
||||
|
||||
String::size_type String::Size() const
|
||||
{
|
||||
return m_size;
|
||||
}
|
||||
|
||||
String::size_type String::Capasity() const
|
||||
{
|
||||
return m_capasity;
|
||||
}
|
||||
|
||||
const char* String::Data() const
|
||||
{
|
||||
return m_data;
|
||||
}
|
||||
|
||||
ErrorOr<void> String::EnsureCapasity(size_type size)
|
||||
{
|
||||
if (m_capasity >= size)
|
||||
return {};
|
||||
size_type new_cap = MAX(size, m_capasity * 1.5f);
|
||||
void* new_data = BAN::allocator(new_cap);
|
||||
if (new_data == nullptr)
|
||||
return Error::FromString("String: Could not allocate memory");
|
||||
memcpy(new_data, m_data, m_size + 1);
|
||||
BAN::deallocator(m_data);
|
||||
m_data = (char*)new_data;
|
||||
m_capasity = new_cap;
|
||||
return {};
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in New Issue