forked from Bananymous/banan-os
BAN: Implement basic vector
This commit is contained in:
parent
171a33bbcd
commit
07f61d9b82
|
@ -0,0 +1,169 @@
|
|||
#pragma
|
||||
|
||||
#include <BAN/Errors.h>
|
||||
#include <BAN/Memory.h>
|
||||
|
||||
#include <assert.h>
|
||||
#include <string.h>
|
||||
#include <sys/param.h>
|
||||
|
||||
namespace BAN
|
||||
{
|
||||
|
||||
template<typename T>
|
||||
class Vector
|
||||
{
|
||||
public:
|
||||
using size_type = size_t;
|
||||
using value_type = T;
|
||||
|
||||
public:
|
||||
Vector() = default;
|
||||
~Vector();
|
||||
|
||||
ErrorOr<void> PushBack(const T&);
|
||||
ErrorOr<void> Insert(const T&, size_type);
|
||||
|
||||
void PopBack();
|
||||
void Remove(size_type);
|
||||
|
||||
const T& operator[](size_type) const;
|
||||
T& operator[](size_type);
|
||||
|
||||
ErrorOr<void> Resize(size_type);
|
||||
ErrorOr<void> Reserve(size_type);
|
||||
|
||||
bool Empty() const;
|
||||
size_type Size() const;
|
||||
size_type Capasity() const;
|
||||
|
||||
private:
|
||||
ErrorOr<void> EnsureCapasity(size_type);
|
||||
|
||||
private:
|
||||
T* m_data = nullptr;
|
||||
size_type m_capasity = 0;
|
||||
size_type m_size = 0;
|
||||
};
|
||||
|
||||
template<typename T>
|
||||
Vector<T>::~Vector()
|
||||
{
|
||||
for (size_type i = 0; i < m_size; i++)
|
||||
m_data[i].~T();
|
||||
BAN::deallocator(m_data);
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
ErrorOr<void> Vector<T>::PushBack(const T& value)
|
||||
{
|
||||
TRY(EnsureCapasity(m_size + 1));
|
||||
m_data[m_size] = value;
|
||||
m_size++;
|
||||
return {};
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
ErrorOr<void> Vector<T>::Insert(const T& value, size_type index)
|
||||
{
|
||||
assert(index <= m_size);
|
||||
TRY(EnsureCapasity(m_size + 1));
|
||||
memmove(m_data + index + 1, m_data + index, (m_size - index) * sizeof(T));
|
||||
m_data[index] = value;
|
||||
m_size++;
|
||||
return {};
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
void Vector<T>::PopBack()
|
||||
{
|
||||
assert(m_size > 0);
|
||||
m_data[m_size - 1].~T();
|
||||
m_size--;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
void Vector<T>::Remove(size_type index)
|
||||
{
|
||||
assert(index < m_size);
|
||||
m_data[index].~T();
|
||||
memmove(m_data + index, m_data + index + 1, (m_size - index - 1) * sizeof(T));
|
||||
m_size--;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
const T& Vector<T>::operator[](size_type index) const
|
||||
{
|
||||
assert(index < m_size);
|
||||
return m_data[index];
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
T& Vector<T>::operator[](size_type index)
|
||||
{
|
||||
assert(index < m_size);
|
||||
return m_data[index];
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
ErrorOr<void> Vector<T>::Resize(size_type size)
|
||||
{
|
||||
if (size < m_size)
|
||||
{
|
||||
for (size_type i = size; i < m_size; i++)
|
||||
m_data[i].~T();
|
||||
m_size = size;
|
||||
}
|
||||
else if (size > m_size)
|
||||
{
|
||||
TRY(EnsureCapasity(size));
|
||||
for (size_type i = m_size; i < size; i++)
|
||||
m_data[i] = T();
|
||||
m_size = size;
|
||||
}
|
||||
m_size = size;
|
||||
return {};
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
ErrorOr<void> Vector<T>::Reserve(size_type size)
|
||||
{
|
||||
TRY(EnsureCapasity(size));
|
||||
return {};
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
bool Vector<T>::Empty() const
|
||||
{
|
||||
return m_size == 0;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
typename Vector<T>::size_type Vector<T>::Size() const
|
||||
{
|
||||
return m_size;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
typename Vector<T>::size_type Vector<T>::Capasity() const
|
||||
{
|
||||
return m_capasity;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
ErrorOr<void> Vector<T>::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 * sizeof(T));
|
||||
if (new_data == nullptr)
|
||||
return Error::FromString("Vector: Could not allocate memory");
|
||||
memcpy(new_data, m_data, m_size * sizeof(T));
|
||||
BAN::deallocator(m_data);
|
||||
m_data = (T*)new_data;
|
||||
m_capasity = new_cap;
|
||||
return {};
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in New Issue