Vector.vala 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217
  1. namespace Invercargill.DataStructures {
  2. public class Vector<T> : Enumerable<T>, Lot<T>, ReadOnlyCollection<T>, ReadOnlyAddressable<T>, Collection<T>, Addressable<T> {
  3. private T[] array;
  4. private int n_items = 0;
  5. private SafeReadFunc<T>? safe_read;
  6. private SafeWriteFunc<T>? safe_write;
  7. private const int INITIAL_SIZE = 2;
  8. public Vector() {
  9. array = new T[INITIAL_SIZE];
  10. safe_read = get_safe_read_function_for<T>();
  11. safe_write = get_safe_write_function_for<T>();
  12. }
  13. public override Tracker<T> get_tracker() {
  14. return new AddressableTracker<T>(this);
  15. }
  16. public override T[] to_array () {
  17. lock(array) {
  18. var a2 = new T[n_items];
  19. if(safe_write != null) {
  20. for(var i = 0; i < n_items; i++) {
  21. safe_write(a2, i, array_read(i));
  22. }
  23. }
  24. else {
  25. for(var i = 0; i < n_items; i++) {
  26. a2[i] = array[i];
  27. }
  28. }
  29. return a2;
  30. }
  31. }
  32. public void add(T item) {
  33. lock(array) {
  34. ensure_room(1);
  35. array_write(n_items, item);
  36. n_items++;
  37. }
  38. }
  39. public override void add_all(Enumerable<T> items) {
  40. lock(array) {
  41. foreach (var item in items) {
  42. ensure_room(1);
  43. array_write(n_items, item);
  44. n_items++;
  45. }
  46. }
  47. }
  48. public new T @get(uint index) throws IndexError {
  49. IndexError? e = null;
  50. lock(array) {
  51. if(index < 0) {
  52. e = new IndexError.INDEX_EXCEEDS_LOWER_BOUNDS("Index is less than 0");
  53. }
  54. else if(index >= n_items) {
  55. e = new IndexError.INDEX_EXCEEDS_UPPER_BOUNDS(@"Tried to access index $(index) on a vector with $(n_items) item(s)");
  56. }
  57. else {
  58. return array_read(index);
  59. }
  60. }
  61. throw e;
  62. }
  63. public bool try_get(uint index, out T value) {
  64. lock(array) {
  65. if(index >= 0 && index < n_items) {
  66. value = array_read(index);
  67. return true;
  68. }
  69. }
  70. value = null;
  71. return false;
  72. }
  73. public new void @set(int index, T value) throws IndexError {
  74. IndexError? e = null;
  75. lock(array) {
  76. if(index < 0) {
  77. e = new IndexError.INDEX_EXCEEDS_LOWER_BOUNDS("Index is less than 0");
  78. }
  79. else if(index >= n_items) {
  80. e = new IndexError.INDEX_EXCEEDS_UPPER_BOUNDS(@"Tried to set index $(index) on a vector with $(n_items) item(s)");
  81. }
  82. else {
  83. array_write(index, value);
  84. return;
  85. }
  86. }
  87. throw e;
  88. }
  89. private void ensure_room(int items) {
  90. if(array.length <= n_items + items) {
  91. array.resize(array.length * 2);
  92. }
  93. }
  94. public override int count() {
  95. return n_items;
  96. }
  97. public void remove_at(int index) throws IndexError {
  98. var e = remove_internal(index);
  99. if(e != null) {
  100. throw e;
  101. }
  102. }
  103. private IndexError? remove_internal(uint index) {
  104. IndexError? e = null;
  105. lock(array) {
  106. if(index < 0) {
  107. e = new IndexError.INDEX_EXCEEDS_LOWER_BOUNDS("Index is less than 0");
  108. }
  109. else if(index >= n_items) {
  110. e = new IndexError.INDEX_EXCEEDS_UPPER_BOUNDS(@"Tried to set index $(index) on a vector with $(n_items) item(s)");
  111. }
  112. else {
  113. for(uint i = index; i < n_items; i++) {
  114. if(i+1 < n_items) {
  115. array_write(i, array_read(i+1));
  116. continue;
  117. }
  118. array[i] = null;
  119. }
  120. n_items --;
  121. }
  122. }
  123. return e;
  124. }
  125. public override T last(owned PredicateDelegate<T>? predicate = null) throws SequenceError {
  126. if(predicate != null) {
  127. return base.last((owned)predicate);
  128. }
  129. SequenceError? e = null;
  130. lock(array) {
  131. if(n_items == 0) {
  132. e = new SequenceError.NO_ELEMENTS("The sequence contains no elements");
  133. }
  134. else {
  135. return array_read(n_items -1);
  136. }
  137. }
  138. throw e;
  139. }
  140. public override T? last_or_default(owned PredicateDelegate<T>? predicate = null) {
  141. if(predicate != null) {
  142. return base.last_or_default((owned)predicate);
  143. }
  144. lock(array) {
  145. if(n_items == 0) {
  146. return null;
  147. }
  148. else {
  149. return array_read(n_items -1);
  150. }
  151. }
  152. }
  153. private void array_write(uint index, T item) {
  154. if(safe_write != null) {
  155. safe_write(array, index, item);
  156. return;
  157. }
  158. array[index] = item;
  159. }
  160. private T array_read(uint index) {
  161. if(safe_read != null) {
  162. return safe_read(array, index);
  163. }
  164. return array[index];
  165. }
  166. public uint? first_index_of(PredicateDelegate<T> predicate) {
  167. var i = -1;
  168. foreach (var item in this) {
  169. i++;
  170. if(predicate(item)) {
  171. return i;
  172. }
  173. }
  174. return null;
  175. }
  176. public void remove_first_where(Invercargill.PredicateDelegate<T> predicate) {
  177. remove_internal(first_index_of(predicate));
  178. }
  179. public void remove_all_where(Invercargill.PredicateDelegate<T> predicate) {
  180. with_positions()
  181. .where(i => predicate(i.item))
  182. .select<int>(i => i.position)
  183. .iterate(i => remove_internal(i));
  184. }
  185. public void clear() {
  186. n_items = 0;
  187. array = new T[INITIAL_SIZE];
  188. }
  189. }
  190. }