mirror of
https://github.com/geode-sdk/geode.git
synced 2024-11-23 07:57:51 -05:00
762 lines
20 KiB
C++
762 lines
20 KiB
C++
#pragma once
|
|
|
|
#include <algorithm>
|
|
#include <map>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
#include <set>
|
|
#include <vector>
|
|
|
|
namespace geode::base {
|
|
uintptr_t get();
|
|
}
|
|
|
|
#if defined(GEODE_IS_ANDROID)
|
|
|
|
#include "gnustl-map.hpp"
|
|
|
|
namespace gd {
|
|
using namespace geode::stl;
|
|
|
|
void* operatorNew(size_t size);
|
|
void operatorDelete(void* ptr);
|
|
|
|
template <typename T>
|
|
class allocator : public std::allocator<T> {
|
|
public:
|
|
typedef size_t size_type;
|
|
typedef T* pointer;
|
|
typedef const T* const_pointer;
|
|
|
|
template<typename _Tp1>
|
|
struct rebind {
|
|
typedef allocator<_Tp1> other;
|
|
};
|
|
|
|
pointer allocate(size_type n, const void *hint=0) {
|
|
return (pointer)operatorNew(n * sizeof(T));
|
|
}
|
|
|
|
void deallocate(pointer p, size_type n) {
|
|
return operatorDelete(p);
|
|
}
|
|
|
|
allocator() throw(): std::allocator<T>() { }
|
|
allocator(const allocator &a) throw(): std::allocator<T>(a) { }
|
|
template <class U>
|
|
allocator(const allocator<U> &a) throw(): std::allocator<T>(a) { }
|
|
~allocator() throw() { }
|
|
};
|
|
|
|
template <typename K, typename V>
|
|
class GEODE_DLL map {
|
|
protected:
|
|
std::less<K> compare;
|
|
_rb_tree_base m_header;
|
|
size_t m_nodecount;
|
|
|
|
public:
|
|
typedef _rb_tree_node<std::pair<K, V>>* _tree_node;
|
|
typedef _rb_tree_iterator<std::pair<K, V>> iterator;
|
|
|
|
std::map<K, V> std() {
|
|
return (std::map<K, V>)(*this);
|
|
}
|
|
|
|
operator std::map<K, V>() {
|
|
auto iter_node = static_cast<_tree_node>(m_header.m_left);
|
|
auto end_node = static_cast<_tree_node>(&m_header);
|
|
std::map<K, V> out;
|
|
for (; iter_node != end_node;
|
|
iter_node = static_cast<_tree_node>(_rb_increment(iter_node))) {
|
|
out[iter_node->m_value.first] = iter_node->m_value.second;
|
|
}
|
|
|
|
return out;
|
|
}
|
|
|
|
operator std::map<K, V>() const {
|
|
auto iter_node = static_cast<_tree_node>(m_header.m_left);
|
|
auto end_node = (_tree_node)(&m_header);
|
|
std::map<K, V> out;
|
|
for (; iter_node != end_node;
|
|
iter_node = static_cast<_tree_node>(_rb_increment(iter_node))) {
|
|
out[iter_node->m_value.first] = iter_node->m_value.second;
|
|
}
|
|
|
|
return out;
|
|
}
|
|
|
|
void insert(_tree_node x, _tree_node p, std::pair<K, V> const& val) {
|
|
bool insert_left =
|
|
(x != 0 || p == static_cast<_tree_node>(&m_header) || val.first < p->m_value.first);
|
|
|
|
_tree_node z = new _rb_tree_node<std::pair<K, V>>();
|
|
z->m_value = val;
|
|
|
|
_rb_insert_rebalance(insert_left, z, p, m_header);
|
|
++m_nodecount;
|
|
}
|
|
|
|
void insert(std::pair<K, V> const& val) {
|
|
insert_pair(val);
|
|
}
|
|
|
|
void insert_pair(std::pair<K, V> const& val) {
|
|
_tree_node x = static_cast<_tree_node>(m_header.m_parent);
|
|
_tree_node y = static_cast<_tree_node>(&m_header);
|
|
bool comp = true;
|
|
while (x != 0) {
|
|
y = x;
|
|
comp = val.first < x->m_value.first;
|
|
x = comp ? static_cast<_tree_node>(x->m_left) : static_cast<_tree_node>(x->m_right);
|
|
}
|
|
auto iter = y;
|
|
|
|
if (comp) {
|
|
if (iter == static_cast<_tree_node>(m_header.m_left)) {
|
|
insert(x, y, val);
|
|
}
|
|
else {
|
|
iter = static_cast<_tree_node>(_rb_decrement(iter));
|
|
}
|
|
}
|
|
if (iter->m_value.first < val.first) {
|
|
insert(x, y, val);
|
|
}
|
|
}
|
|
|
|
map(std::map<K, V> input) {
|
|
m_header.m_isblack = false;
|
|
m_header.m_parent = 0;
|
|
m_header.m_left = &m_header;
|
|
m_header.m_right = &m_header;
|
|
|
|
for (auto i : input) {
|
|
insert_pair(i);
|
|
}
|
|
}
|
|
|
|
void erase(_tree_node x) {
|
|
while (x != 0) {
|
|
erase(static_cast<_tree_node>(x->m_right));
|
|
auto y = static_cast<_tree_node>(x->m_left);
|
|
delete y;
|
|
x = y;
|
|
}
|
|
}
|
|
|
|
std::pair<iterator, iterator> equal_range(K const& __k) {
|
|
return std::pair<iterator, iterator>(lower_bound(__k), upper_bound(__k));
|
|
}
|
|
|
|
size_t erase(K const& __x) {
|
|
std::pair<iterator, iterator> __p = equal_range(__x);
|
|
size_t __old = size();
|
|
erase(__p.first, __p.second);
|
|
return __old - size();
|
|
}
|
|
|
|
void clear() {
|
|
erase(static_cast<_tree_node>(m_header.m_parent));
|
|
m_header.m_parent = 0;
|
|
m_header.m_left = &m_header;
|
|
m_header.m_right = &m_header;
|
|
m_nodecount = 0;
|
|
}
|
|
|
|
void erase(iterator __first, iterator __last) {
|
|
if (__first == begin() && __last == end()) {
|
|
clear();
|
|
}
|
|
else {
|
|
while (__first != __last) {
|
|
erase(__first++);
|
|
}
|
|
}
|
|
}
|
|
|
|
void erase(iterator __pos) {
|
|
_tree_node __y = static_cast<_tree_node>(_rb_rebalance_for_erase(
|
|
__pos.m_node, m_header
|
|
));
|
|
delete __y;
|
|
--m_nodecount;
|
|
}
|
|
|
|
V& operator[](K const& __k) {
|
|
iterator __i = lower_bound(__k);
|
|
if (__i == end() || compare(__k, (*__i).first)) {
|
|
insert_pair(std::pair<K, V>(__k, V()));
|
|
__i = lower_bound(__k);
|
|
}
|
|
return (*__i).second;
|
|
}
|
|
|
|
V& at(K const& __k) {
|
|
iterator __i = lower_bound(__k);
|
|
if (__i == end() || compare(__k, (*__i).first)) {
|
|
throw std::out_of_range("map::at");
|
|
}
|
|
|
|
return (*__i).second;
|
|
}
|
|
|
|
const V& at(K const& __k) const {
|
|
iterator __i = lower_bound(__k);
|
|
if (__i == end() || compare(__k, (*__i).first)) {
|
|
throw std::out_of_range("map::at");
|
|
}
|
|
|
|
return (*__i).second;
|
|
}
|
|
|
|
iterator begin() noexcept {
|
|
return iterator(m_header.m_left);
|
|
}
|
|
|
|
iterator end() noexcept {
|
|
return iterator(&m_header);
|
|
}
|
|
|
|
bool empty() const noexcept {
|
|
return m_nodecount == 0;
|
|
}
|
|
|
|
size_t size() const noexcept {
|
|
return m_nodecount;
|
|
}
|
|
|
|
iterator lower_bound(K const& __x) {
|
|
_tree_node __j = static_cast<_tree_node>(m_header.m_parent);
|
|
_tree_node __k = static_cast<_tree_node>(&m_header);
|
|
while (__j != nullptr) {
|
|
if (!compare(__j->m_value.first, __x)) {
|
|
__k = __j;
|
|
__j = static_cast<_tree_node>(__j->m_left);
|
|
}
|
|
else {
|
|
__j = static_cast<_tree_node>(__j->m_right);
|
|
}
|
|
}
|
|
return iterator(__k);
|
|
}
|
|
|
|
iterator upper_bound(K const& __x) {
|
|
_tree_node __j = static_cast<_tree_node>(m_header.m_parent);
|
|
_tree_node __k = static_cast<_tree_node>(&m_header);
|
|
while (__j != nullptr) {
|
|
if (compare(__x, __j->m_value.first)) {
|
|
__k = __j;
|
|
__j = static_cast<_tree_node>(__j->m_left);
|
|
}
|
|
else {
|
|
__j = static_cast<_tree_node>(__j->m_right);
|
|
}
|
|
}
|
|
return iterator(__k);
|
|
}
|
|
|
|
iterator find(K const& __x) {
|
|
iterator __j = lower_bound(__x);
|
|
return (__j == end() || compare(__x, (*__j).first)) ? end() : __j;
|
|
}
|
|
|
|
size_t count(K const& __x) {
|
|
return find(__x) != end() ? 1 : 0;
|
|
}
|
|
|
|
bool contains(K const& __x) {
|
|
return count(__x) > 0;
|
|
}
|
|
|
|
map(map const& lol) : map(std::map<K, V>(lol)) {}
|
|
|
|
map() : map(std::map<K, V>()) {}
|
|
|
|
~map() {
|
|
erase(static_cast<_tree_node>(m_header.m_parent));
|
|
}
|
|
};
|
|
|
|
// template <class Type>
|
|
// using vector = std::vector<Type>;
|
|
|
|
template <typename T>
|
|
class GEODE_DLL vector {
|
|
public:
|
|
using value_type = T;
|
|
|
|
auto allocator() const {
|
|
return gd::allocator<T>();
|
|
}
|
|
|
|
operator std::vector<T>() const {
|
|
return std::vector<T>(m_start, m_finish);
|
|
}
|
|
|
|
vector() {
|
|
m_start = nullptr;
|
|
m_finish = nullptr;
|
|
m_reserveEnd = nullptr;
|
|
}
|
|
|
|
size_t nextCapacity(size_t x) {
|
|
// minimum 16, powers of 2, don't use builtins
|
|
if (x < 16) return 16;
|
|
size_t out = 16;
|
|
while (out < x) {
|
|
out *= 2;
|
|
}
|
|
return out;
|
|
}
|
|
|
|
vector(std::vector<T> const& input) : vector() {
|
|
if (input.size()) {
|
|
auto capacity = nextCapacity(input.size());
|
|
m_start = this->allocator().allocate(capacity);
|
|
m_finish = m_start + input.size();
|
|
m_reserveEnd = m_start + capacity;
|
|
|
|
std::uninitialized_default_construct(m_start, m_finish);
|
|
std::copy(input.begin(), input.end(), m_start);
|
|
}
|
|
}
|
|
|
|
vector(gd::vector<T> const& input) : vector() {
|
|
if (input.size()) {
|
|
auto capacity = nextCapacity(input.size());
|
|
m_start = this->allocator().allocate(capacity);
|
|
m_finish = m_start + input.size();
|
|
m_reserveEnd = m_start + capacity;
|
|
|
|
std::uninitialized_default_construct(m_start, m_finish);
|
|
std::copy(input.begin(), input.end(), m_start);
|
|
}
|
|
}
|
|
|
|
vector(gd::vector<T>&& input) : vector() {
|
|
m_start = input.m_start;
|
|
m_finish = input.m_finish;
|
|
m_reserveEnd = input.m_reserveEnd;
|
|
|
|
input.m_start = nullptr;
|
|
input.m_finish = nullptr;
|
|
input.m_reserveEnd = nullptr;
|
|
}
|
|
|
|
vector& operator=(gd::vector<T> const& input) {
|
|
this->clear();
|
|
|
|
if (input.size()) {
|
|
auto capacity = nextCapacity(input.size());
|
|
m_start = this->allocator().allocate(capacity);
|
|
m_finish = m_start + input.size();
|
|
m_reserveEnd = m_start + capacity;
|
|
|
|
std::uninitialized_default_construct(m_start, m_finish);
|
|
std::copy(input.begin(), input.end(), m_start);
|
|
}
|
|
|
|
return *this;
|
|
}
|
|
|
|
vector& operator=(gd::vector<T>&& input) {
|
|
m_start = input.m_start;
|
|
m_finish = input.m_finish;
|
|
m_reserveEnd = input.m_reserveEnd;
|
|
|
|
input.m_start = nullptr;
|
|
input.m_finish = nullptr;
|
|
input.m_reserveEnd = nullptr;
|
|
|
|
return *this;
|
|
}
|
|
|
|
void grow() {
|
|
if (m_finish == m_reserveEnd) {
|
|
auto newSize = this->capacity() * 2;
|
|
auto newStart = this->allocator().allocate(newSize);
|
|
auto newFinish = newStart + this->size();
|
|
|
|
std::uninitialized_default_construct(newStart, newFinish);
|
|
std::copy(m_start, m_finish, newStart);
|
|
|
|
std::destroy(m_start, m_finish);
|
|
this->allocator().deallocate(m_start, this->capacity());
|
|
|
|
m_start = newStart;
|
|
m_finish = newFinish;
|
|
m_reserveEnd = newStart + newSize;
|
|
}
|
|
}
|
|
|
|
void shrink() {
|
|
if (m_finish < m_reserveEnd / 2 && this->capacity() > 16) {
|
|
auto newSize = this->capacity() / 2;
|
|
auto newStart = this->allocator().allocate(newSize);
|
|
auto newFinish = newStart + this->size();
|
|
|
|
std::uninitialized_default_construct(newStart, newFinish);
|
|
std::copy(m_start, m_finish, newStart);
|
|
|
|
std::destroy(m_start, m_finish);
|
|
this->allocator().deallocate(m_start, this->capacity());
|
|
|
|
m_start = newStart;
|
|
m_finish = newFinish;
|
|
m_reserveEnd = newStart + newSize;
|
|
}
|
|
}
|
|
|
|
void push_back(T const& input) {
|
|
this->grow();
|
|
|
|
new (m_finish) T(input);
|
|
++m_finish;
|
|
}
|
|
|
|
void push_back(T&& input) {
|
|
this->grow();
|
|
|
|
new (m_finish) T(std::move(input));
|
|
++m_finish;
|
|
}
|
|
|
|
void pop_back() {
|
|
if (m_finish != m_start) {
|
|
--m_finish;
|
|
m_finish->~T();
|
|
}
|
|
|
|
this->shrink();
|
|
}
|
|
|
|
vector(std::initializer_list<T> const& input) : vector() {
|
|
if (input.size()) {
|
|
auto capacity = nextCapacity(input.size());
|
|
m_start = this->allocator().allocate(capacity);
|
|
m_finish = m_start + input.size();
|
|
m_reserveEnd = m_start + capacity;
|
|
|
|
std::uninitialized_default_construct(m_start, m_finish);
|
|
std::copy(input.begin(), input.end(), m_start);
|
|
}
|
|
}
|
|
|
|
void clear() {
|
|
if (m_start) {
|
|
std::destroy(m_start, m_finish);
|
|
|
|
this->allocator().deallocate(m_start, this->capacity());
|
|
}
|
|
|
|
m_start = nullptr;
|
|
m_finish = nullptr;
|
|
m_reserveEnd = nullptr;
|
|
}
|
|
|
|
T& operator[](size_t index) {
|
|
return m_start[index];
|
|
}
|
|
|
|
T const& operator[](size_t index) const {
|
|
return m_start[index];
|
|
}
|
|
|
|
T& at(size_t index) {
|
|
if (index >= this->size()) {
|
|
throw std::out_of_range("gd::vector::at");
|
|
}
|
|
return m_start[index];
|
|
}
|
|
|
|
T const& at(size_t index) const {
|
|
if (index >= this->size()) {
|
|
throw std::out_of_range("gd::vector::at");
|
|
}
|
|
return m_start[index];
|
|
}
|
|
|
|
T& front() {
|
|
return *m_start;
|
|
}
|
|
|
|
T* begin() {
|
|
return m_start;
|
|
}
|
|
|
|
T* end() {
|
|
return m_finish;
|
|
}
|
|
|
|
T const* begin() const {
|
|
return m_start;
|
|
}
|
|
|
|
T const* end() const {
|
|
return m_finish;
|
|
}
|
|
|
|
~vector() {
|
|
if (m_start) {
|
|
for (auto& x : *this) {
|
|
x.~T();
|
|
}
|
|
delete m_start;
|
|
}
|
|
}
|
|
|
|
size_t size() const {
|
|
return m_finish - m_start;
|
|
}
|
|
|
|
size_t capacity() const {
|
|
return m_reserveEnd - m_start;
|
|
}
|
|
|
|
protected:
|
|
T* m_start;
|
|
T* m_finish;
|
|
T* m_reserveEnd;
|
|
};
|
|
|
|
struct _bit_reference {
|
|
uintptr_t* m_bitptr;
|
|
uintptr_t m_mask;
|
|
|
|
_bit_reference(uintptr_t* x, uintptr_t y) : m_bitptr(x), m_mask(y) {}
|
|
|
|
_bit_reference() : m_bitptr(0), m_mask(0) {}
|
|
|
|
operator bool() const {
|
|
return !!(*m_bitptr & m_mask);
|
|
}
|
|
|
|
_bit_reference& operator=(bool x) {
|
|
if (x) *m_bitptr |= m_mask;
|
|
else *m_bitptr &= ~m_mask;
|
|
return *this;
|
|
}
|
|
|
|
_bit_reference& operator=(_bit_reference const& x) {
|
|
return *this = bool(x);
|
|
}
|
|
|
|
bool operator==(_bit_reference const& x) const {
|
|
return bool(*this) == bool(x);
|
|
}
|
|
|
|
bool operator<(_bit_reference const& x) const {
|
|
return !bool(*this) && bool(x);
|
|
}
|
|
|
|
void flip() {
|
|
*m_bitptr ^= m_mask;
|
|
}
|
|
};
|
|
|
|
struct _bit_iterator {
|
|
uintptr_t* m_bitptr;
|
|
unsigned int m_offset;
|
|
|
|
_bit_iterator(uintptr_t* x) : m_bitptr(x), m_offset(0) {}
|
|
|
|
_bit_iterator(uintptr_t* x, unsigned o) : m_bitptr(x), m_offset(o) {}
|
|
|
|
_bit_reference operator*() const {
|
|
return _bit_reference(m_bitptr, 1UL << m_offset);
|
|
}
|
|
|
|
_bit_iterator& operator++() {
|
|
if (m_offset++ == sizeof(uintptr_t) - 1) {
|
|
m_offset = 0;
|
|
m_bitptr++;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
bool operator!=(_bit_iterator const& b) {
|
|
return !(m_bitptr == b.m_bitptr && m_offset == b.m_offset);
|
|
}
|
|
};
|
|
|
|
template <>
|
|
class vector<bool> {
|
|
protected:
|
|
_bit_iterator m_start;
|
|
_bit_iterator m_end;
|
|
uintptr_t* m_capacity_end;
|
|
|
|
public:
|
|
auto allocator() const {
|
|
return gd::allocator<uintptr_t>();
|
|
}
|
|
|
|
vector() : m_start(nullptr), m_end(nullptr), m_capacity_end(nullptr) {}
|
|
|
|
// vector(std::vector<bool> input) : vector() {
|
|
// auto realsize = input.size() / int(sizeof(uintptr_t));
|
|
// auto start = this->allocator().allocate(realsize);
|
|
|
|
// m_start = _bit_iterator(start);
|
|
// m_end = _bit_iterator(start + realsize, input.size() % sizeof(uintptr_t));
|
|
// m_capacity_end = start + realsize;
|
|
|
|
// auto itmp = m_start;
|
|
// for (auto i : input) {
|
|
// *itmp = i;
|
|
// ++itmp;
|
|
// }
|
|
// }
|
|
|
|
// vector(vector<bool> const& input) : vector() {
|
|
|
|
// }
|
|
|
|
// vector() : vector(std::vector<bool>()) {}
|
|
|
|
~vector() {
|
|
delete[] m_start.m_bitptr;
|
|
}
|
|
|
|
operator std::vector<bool>() const {
|
|
std::vector<bool> out;
|
|
for (auto i = m_start; i != m_end; ++i) {
|
|
out.push_back(*i);
|
|
}
|
|
return out;
|
|
}
|
|
|
|
_bit_reference operator[](size_t index) {
|
|
auto const real_index = index / sizeof(uintptr_t);
|
|
auto const offset = index % sizeof(uintptr_t);
|
|
return _bit_reference(&m_start.m_bitptr[real_index], 1UL << offset);
|
|
}
|
|
|
|
bool operator[](size_t index) const {
|
|
return const_cast<vector&>(*this)[index];
|
|
}
|
|
|
|
_bit_reference at(size_t index) {
|
|
// TODO: bounds checking
|
|
return this->operator[](index);
|
|
}
|
|
|
|
bool at(size_t index) const {
|
|
// TODO: bounds checking
|
|
return this->operator[](index);
|
|
}
|
|
};
|
|
|
|
// 2.2 TODO: Implement set, unordered_map and unordered_set
|
|
|
|
// the sizes of these are always the same, no matter the type
|
|
template <class V>
|
|
using set = void*[6];
|
|
|
|
template <class K, class V>
|
|
using unordered_map = void*[7];
|
|
|
|
template <class V>
|
|
using unordered_set = void*[7];
|
|
};
|
|
|
|
#elif defined(GEODE_IS_IOS)
|
|
namespace gd {
|
|
class GEODE_DLL string {
|
|
public:
|
|
string() {}
|
|
|
|
string(char const* ok) : m_internal(ok) {}
|
|
|
|
string(std::string ok) : m_internal(ok) {}
|
|
|
|
operator std::string() {
|
|
return m_internal;
|
|
}
|
|
|
|
operator std::string() const {
|
|
return m_internal;
|
|
}
|
|
|
|
string(string const& ok) : m_internal(ok) {}
|
|
|
|
string& operator=(char const* ok) {
|
|
m_internal = ok;
|
|
return *this;
|
|
}
|
|
|
|
string& operator=(string const& ok) {
|
|
m_internal = ok;
|
|
return *this;
|
|
}
|
|
|
|
~string() {}
|
|
|
|
char const* c_str() const {
|
|
return m_internal.c_str();
|
|
}
|
|
|
|
protected:
|
|
std::string m_internal;
|
|
};
|
|
|
|
template <typename T>
|
|
class GEODE_DLL vector {
|
|
public:
|
|
using value_type = T;
|
|
|
|
operator std::vector<T>() {
|
|
return m_internal;
|
|
}
|
|
|
|
void clear() {
|
|
m_internal.clear();
|
|
}
|
|
|
|
operator std::vector<T>() const {
|
|
return m_internal;
|
|
}
|
|
|
|
vector(std::vector<T> input) : m_internal(input) {}
|
|
|
|
T& front() {
|
|
return m_internal.front();
|
|
}
|
|
|
|
vector(vector const& lol) : m_internal(lol) {}
|
|
|
|
vector() : m_internal() {}
|
|
|
|
~vector() {}
|
|
|
|
protected:
|
|
std::vector<T> m_internal;
|
|
};
|
|
|
|
template <typename K, typename V>
|
|
class GEODE_DLL map {
|
|
protected:
|
|
std::map<K, V> m_internal;
|
|
|
|
public:
|
|
operator std::map<K, V>() {
|
|
return m_internal;
|
|
}
|
|
|
|
operator std::map<K, V>() const {
|
|
return m_internal;
|
|
}
|
|
|
|
map(std::map<K, V> input) : m_internal(input) {}
|
|
|
|
map(map const& lol) : m_internal(lol) {}
|
|
|
|
map() {}
|
|
|
|
~map() {}
|
|
};
|
|
}
|
|
#endif
|