Commit 0ce72890a8fe0ed29f6beb5d515868c67c89be14
1 parent
dea455b3
Refactored vector
Showing
1 changed file
with
44 additions
and
49 deletions
examples11/04-vectortmpl/vector.h
... | ... | @@ -18,6 +18,42 @@ private: |
18 | 18 | return ++w; |
19 | 19 | }; |
20 | 20 | |
21 | + void destroy_array(C *ptr, unsigned int s) { | |
22 | + for (unsigned j = 0; j < s; ++j) | |
23 | + (ptr + j)->~C(); | |
24 | + free(ptr); | |
25 | + } | |
26 | + | |
27 | + void resize_before_push() { | |
28 | + if (size >= cap) { | |
29 | + if (cap > std::numeric_limits<decltype(size)>::max() / (2 * sizeof(C))) | |
30 | + throw bad_alloc(); | |
31 | + unsigned int newcap = (cap == 0) ? 1 : 2 * cap; | |
32 | + C *newdata; | |
33 | + if constexpr (!std::is_trivially_copyable<C>()) { | |
34 | + newdata = static_cast<C *>( | |
35 | + std::aligned_alloc(alignof(C), newcap * sizeof(C))); | |
36 | + if (!newdata) | |
37 | + throw bad_alloc(); | |
38 | + unsigned i; | |
39 | + try { | |
40 | + for (i = 0; i < size; i++) | |
41 | + new (newdata + i) C(std::move(data[i])); | |
42 | + } catch (...) { | |
43 | + destroy_array(newdata, i); | |
44 | + throw; | |
45 | + } | |
46 | + destroy_array(data, size); | |
47 | + } else { | |
48 | + newdata = static_cast<C *>(std::realloc(data, newcap * sizeof(C))); | |
49 | + if (!newdata) | |
50 | + throw bad_alloc(); | |
51 | + } | |
52 | + data = newdata; | |
53 | + cap = newcap; | |
54 | + } | |
55 | + } | |
56 | + | |
21 | 57 | public: |
22 | 58 | class index_out_of_range {}; |
23 | 59 | explicit vector(unsigned s) { |
... | ... | @@ -34,23 +70,19 @@ public: |
34 | 70 | for (i = 0; i < size; i++) |
35 | 71 | new (data + i) C(); |
36 | 72 | } catch (...) { |
37 | - for (unsigned j = 0; j < i; ++j) | |
38 | - (data + j)->~C(); | |
39 | - free(data); | |
73 | + destroy_array(data, i); | |
40 | 74 | throw; |
41 | 75 | } |
42 | 76 | } |
43 | 77 | |
44 | - ~vector() { | |
45 | - for (unsigned j = 0; j < size; ++j) | |
46 | - (data + j)->~C(); | |
47 | - free(data); | |
48 | - } | |
78 | + ~vector() { destroy_array(data, size); } | |
79 | + | |
49 | 80 | C &operator[](unsigned int pos) { |
50 | 81 | if (pos >= size) |
51 | 82 | throw index_out_of_range(); |
52 | 83 | return data[pos]; |
53 | 84 | } |
85 | + | |
54 | 86 | C operator[](unsigned int pos) const { |
55 | 87 | if (pos >= size) |
56 | 88 | throw index_out_of_range(); |
... | ... | @@ -68,9 +100,7 @@ public: |
68 | 100 | for (i = 0; i < size; i++) |
69 | 101 | new (data + i) C(s.data[i]); |
70 | 102 | } catch (...) { |
71 | - for (unsigned j = 0; j < i; ++j) | |
72 | - (data + j)->~C(); | |
73 | - free(data); | |
103 | + destroy_array(data, i); | |
74 | 104 | throw; |
75 | 105 | } |
76 | 106 | } |
... | ... | @@ -105,44 +135,9 @@ public: |
105 | 135 | } |
106 | 136 | |
107 | 137 | void push_back(const C &s) { |
108 | - if (size < cap) { | |
109 | - new (data + size) C(s); | |
110 | - ++size; | |
111 | - } else { | |
112 | - if (cap > std::numeric_limits<decltype(size)>::max() / (2*sizeof(C))) | |
113 | - throw bad_alloc(); | |
114 | - unsigned int newcap = (cap == 0) ? 1 : 2 * cap; | |
115 | - C *newdata; | |
116 | - if constexpr (!std::is_trivially_copyable<C>()) { | |
117 | - newdata = static_cast<C *>( | |
118 | - std::aligned_alloc(alignof(C), newcap * sizeof(C))); | |
119 | - if (!newdata) | |
120 | - throw bad_alloc(); | |
121 | - unsigned i; | |
122 | - try { | |
123 | - for (i = 0; i < size; i++) | |
124 | - new (newdata + i) C(std::move(data[i])); | |
125 | - } catch (...) { | |
126 | - for (unsigned j = 0; j < i; ++j) | |
127 | - (newdata + j)->~C(); | |
128 | - free(newdata); | |
129 | - throw; | |
130 | - } | |
131 | - new (newdata + size) C(s); | |
132 | - for (unsigned j = 0; j < size; ++j) | |
133 | - (data + j)->~C(); | |
134 | - free(data); | |
135 | - } else { | |
136 | - newdata = static_cast<C *>(std::realloc(data, newcap * sizeof(C))); | |
137 | - if (!newdata) | |
138 | - throw bad_alloc(); | |
139 | - new (newdata + size) C(s); | |
140 | - } | |
141 | - | |
142 | - ++size; | |
143 | - data = newdata; | |
144 | - cap = newcap; | |
145 | - } | |
138 | + resize_before_push(); | |
139 | + new (data + size) C(s); | |
140 | + ++size; | |
146 | 141 | } |
147 | 142 | }; |
148 | 143 | #endif /* __VECTOR_H__ */ | ... | ... |