1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
#pragma once
namespace nall {
string::string() : _data(nullptr), _refs(nullptr), _capacity(0), _size(0) {
}
auto string::get() -> char* {
static char _null[] = "";
if(!_data) return _null;
if(*_refs > 1) _data = _copy(); //make unique for write operations
return _data;
}
auto string::data() const -> const char* {
static const char _null[] = "";
if(!_data) return _null;
return _data;
}
auto string::reset() -> type& {
if(_data && !--*_refs) {
memory::free(_data);
_data = nullptr; //_refs = nullptr; is unnecessary
}
_capacity = 0;
_size = 0;
return *this;
}
auto string::reserve(uint capacity) -> type& {
if(capacity > _capacity) {
_capacity = bit::round(max(31u, capacity) + 1) - 1;
_data = _data ? _copy() : _allocate();
}
return *this;
}
auto string::resize(uint size) -> type& {
reserve(size);
get()[_size = size] = 0;
return *this;
}
auto string::operator=(const string& source) -> string& {
if(&source == this) return *this;
reset();
if(source._data) {
_data = source._data;
_refs = source._refs;
_capacity = source._capacity;
_size = source._size;
++*_refs;
}
return *this;
}
auto string::operator=(string&& source) -> string& {
if(&source == this) return *this;
reset();
_data = source._data;
_refs = source._refs;
_capacity = source._capacity;
_size = source._size;
source._data = nullptr;
source._refs = nullptr;
source._capacity = 0;
source._size = 0;
return *this;
}
auto string::_allocate() -> char* {
auto _temp = (char*)memory::allocate(_capacity + 1 + sizeof(uint));
*_temp = 0;
_refs = (uint*)(_temp + _capacity + 1); //this will always be aligned by 32 via reserve()
*_refs = 1;
return _temp;
}
auto string::_copy() -> char* {
auto _temp = (char*)memory::allocate(_capacity + 1 + sizeof(uint));
memory::copy(_temp, _data, _size = min(_capacity, _size));
_temp[_size] = 0;
--*_refs;
_refs = (uint*)(_temp + _capacity + 1);
*_refs = 1;
return _temp;
}
}
|
