| [44856ed] | 1 | // | 
|---|
|  | 2 | // Cforall Version 1.0.0 Copyright (C) 2016 University of Waterloo | 
|---|
|  | 3 | // | 
|---|
|  | 4 | // The contents of this file are covered under the licence agreement in the | 
|---|
|  | 5 | // file "LICENCE" distributed with Cforall. | 
|---|
|  | 6 | // | 
|---|
|  | 7 | // vector -- A growable array, with full-service iterators | 
|---|
|  | 8 | // | 
|---|
|  | 9 | // Author           : Michael Brooks | 
|---|
|  | 10 | // Created On       : Thu Jun 23 22:00:00 2021 | 
|---|
| [5e4a830] | 11 | // Last Modified By : Peter A. Buhr | 
|---|
|  | 12 | // Last Modified On : Tue Mar 14 08:40:53 2023 | 
|---|
|  | 13 | // Update Count     : 2 | 
|---|
| [44856ed] | 14 | // | 
|---|
|  | 15 |  | 
|---|
| [5e4a830] | 16 | #pragma once | 
|---|
|  | 17 |  | 
|---|
| [44856ed] | 18 | #include <stdlib.hfa> | 
|---|
|  | 19 | #include "list.hfa" | 
|---|
|  | 20 |  | 
|---|
|  | 21 | forall( T ) { | 
|---|
|  | 22 | struct vector; | 
|---|
|  | 23 |  | 
|---|
|  | 24 | struct vector_transit { | 
|---|
|  | 25 | vector(T) * col_$; | 
|---|
|  | 26 | ptrdiff_t idx_$; | 
|---|
|  | 27 | }; | 
|---|
|  | 28 |  | 
|---|
|  | 29 | struct vector_exit { | 
|---|
|  | 30 | vector(T) * invec_$; | 
|---|
|  | 31 | T * item_$; | 
|---|
|  | 32 | }; | 
|---|
|  | 33 |  | 
|---|
|  | 34 | struct vector_permit { | 
|---|
|  | 35 | vector(T) * invec_$; | 
|---|
|  | 36 | T * item_$; | 
|---|
|  | 37 | inline dlink(vector_permit(T)); | 
|---|
|  | 38 | }; | 
|---|
|  | 39 | P9_EMBEDDED(vector_permit(T), dlink(vector_permit(T))) | 
|---|
|  | 40 |  | 
|---|
|  | 41 | struct vector { | 
|---|
|  | 42 | T * buffer_first_$; | 
|---|
|  | 43 | T * buffer_end_$; | 
|---|
|  | 44 | T * elems_first_$; | 
|---|
|  | 45 | T * elems_end_$; // wrapped before storing, never == buffer_end_$ | 
|---|
|  | 46 | size_t exit_refcount_$; | 
|---|
|  | 47 | dlist(vector_permit(T)) live_iters_$; | 
|---|
|  | 48 | }; | 
|---|
|  | 49 | } | 
|---|
|  | 50 |  | 
|---|
|  | 51 | static inline | 
|---|
|  | 52 | forall( T ) { | 
|---|
|  | 53 |  | 
|---|
|  | 54 | // vector | 
|---|
|  | 55 |  | 
|---|
|  | 56 | void ?{}( vector( T ) &, size_t capacity ); | 
|---|
|  | 57 | void ^?{}( vector( T ) & ); | 
|---|
|  | 58 |  | 
|---|
|  | 59 | void ?{}( vector( T ) & ) = void; | 
|---|
|  | 60 | void ?{}( vector( T ) &, vector( T ) & ) = void; | 
|---|
|  | 61 | vector( T ) & ?=?( vector( T ) &, vector( T ) & ) = void; | 
|---|
|  | 62 |  | 
|---|
|  | 63 | // transit | 
|---|
|  | 64 |  | 
|---|
|  | 65 | void ?{}( vector_transit(T) & ) = void; | 
|---|
|  | 66 | void ?{}( vector_transit(T) &, vector_transit(T) & ); | 
|---|
|  | 67 | void ^?{}( vector_transit(T) & ); | 
|---|
|  | 68 |  | 
|---|
|  | 69 | T ?`val( vector_transit(T) & src ); | 
|---|
|  | 70 | void ?=?( vector_transit(T) & dst, T val ); | 
|---|
|  | 71 |  | 
|---|
|  | 72 | // exit | 
|---|
|  | 73 |  | 
|---|
|  | 74 | void ?{}( vector_exit(T) & ) = void; | 
|---|
|  | 75 | void ?{}( vector_exit(T) &, vector(T) * ) = void; | 
|---|
|  | 76 |  | 
|---|
|  | 77 | void ^?{}( vector_exit(T) & ); | 
|---|
|  | 78 | void ?{}( vector_exit(T) &, vector_transit(T) & ); | 
|---|
|  | 79 | void ?{}( vector_exit(T) &, vector_exit(T) & ); | 
|---|
|  | 80 |  | 
|---|
|  | 81 | T ?`val( vector_exit(T) & src ); | 
|---|
|  | 82 | void ?=?( vector_exit(T) & dst, T val ); | 
|---|
|  | 83 | T & ?=?( T & dst, vector_exit(T) & src ); | 
|---|
|  | 84 | void ?*=?( T & dst, vector_exit(T) & src ); | 
|---|
|  | 85 |  | 
|---|
|  | 86 | bool ?`moveNext( vector_exit(T) & it ); | 
|---|
|  | 87 |  | 
|---|
|  | 88 | // permit | 
|---|
|  | 89 |  | 
|---|
|  | 90 | void ?{}( vector_permit(T) & ) = void; | 
|---|
|  | 91 |  | 
|---|
|  | 92 | void ^?{}( vector_permit(T) & ); | 
|---|
|  | 93 | void ?{}( vector_permit(T) &, vector_transit(T) & ); | 
|---|
|  | 94 | void ?{}( vector_permit(T) &, vector_exit(T) & ); | 
|---|
|  | 95 | void ?{}( vector_permit(T) &, vector_permit(T) & ) = void; | 
|---|
|  | 96 |  | 
|---|
|  | 97 | T ?`val( vector_permit(T) & src ); | 
|---|
|  | 98 |  | 
|---|
|  | 99 | // api | 
|---|
|  | 100 |  | 
|---|
|  | 101 | vector_transit(T) push_last( vector( T ) & col, T val ); | 
|---|
|  | 102 | vector_transit(T) ?[?]( vector( T ) &, ptrdiff_t ); | 
|---|
|  | 103 | vector_exit(T) ?`origin( vector( T ) & ); | 
|---|
|  | 104 | size_t ?`capacity( vector(T) & ); | 
|---|
|  | 105 | size_t ?`length( vector(T) & ); | 
|---|
|  | 106 |  | 
|---|
|  | 107 | void insert_before( vector( T ) & col, ptrdiff_t idx, T val ); | 
|---|
|  | 108 |  | 
|---|
|  | 109 | } | 
|---|
|  | 110 |  | 
|---|
|  | 111 | static inline | 
|---|
|  | 112 | forall( T ) { | 
|---|
|  | 113 |  | 
|---|
|  | 114 | // vector | 
|---|
|  | 115 |  | 
|---|
|  | 116 | void ?{}( vector( T ) & this, size_t capacity ) { | 
|---|
|  | 117 | (this.buffer_first_$){ aalloc( capacity ) }; | 
|---|
|  | 118 | (this.buffer_end_$){ this.buffer_first_$ + capacity}; | 
|---|
|  | 119 | (this.elems_first_$){ 0p }; | 
|---|
|  | 120 | (this.elems_end_$){ this.buffer_first_$ }; | 
|---|
|  | 121 | (this.exit_refcount_$){ 0 }; | 
|---|
|  | 122 | (this.live_iters_$){}; | 
|---|
|  | 123 | } | 
|---|
|  | 124 |  | 
|---|
|  | 125 | void ^?{}( vector( T ) & this ) { | 
|---|
|  | 126 | assert( this.exit_refcount_$ == 0 ); | 
|---|
|  | 127 | free( this.buffer_first_$ ); | 
|---|
|  | 128 | this.buffer_first_$ = 0p; | 
|---|
|  | 129 | this.buffer_end_$ = 0p; | 
|---|
|  | 130 | this.elems_first_$ = 0p; | 
|---|
|  | 131 | this.elems_end_$ = 0p; | 
|---|
|  | 132 | } | 
|---|
|  | 133 |  | 
|---|
|  | 134 | // transit | 
|---|
|  | 135 |  | 
|---|
|  | 136 | void ?{}( vector_transit(T) & this, vector_transit(T) & other ) { | 
|---|
|  | 137 | // call autogen constructor deleted at end of hfa | 
|---|
|  | 138 | (this){ other.col_$, other.idx_$ }; | 
|---|
|  | 139 | } | 
|---|
|  | 140 |  | 
|---|
|  | 141 | void ^?{}( vector_transit(T) & ) {} | 
|---|
|  | 142 |  | 
|---|
|  | 143 |  | 
|---|
|  | 144 | vector_transit(T) ?[?]( vector( T ) & vec, ptrdiff_t idx ) { | 
|---|
|  | 145 | // call autogen constructor deleted at end of hfa | 
|---|
|  | 146 | vector_transit(T) ret = { & vec, idx }; | 
|---|
|  | 147 | return ret; | 
|---|
|  | 148 | } | 
|---|
|  | 149 |  | 
|---|
|  | 150 | T & findElemMem_$( vector(T) & v, ptrdiff_t idx ) { | 
|---|
|  | 151 | size_t len = v`length; | 
|---|
|  | 152 | while (idx > len) idx -= len; | 
|---|
|  | 153 | while (idx < 0  ) idx += len; | 
|---|
|  | 154 | T * ret = v.elems_first_$ + idx; | 
|---|
|  | 155 | if (ret < v.buffer_end_$) return *ret; | 
|---|
|  | 156 | ret -= (v.buffer_end_$ - v.buffer_first_$); | 
|---|
|  | 157 | assert( v.buffer_first_$ <= ret && ret < v.elems_end_$ ); | 
|---|
|  | 158 | return *ret; | 
|---|
|  | 159 | } | 
|---|
|  | 160 |  | 
|---|
|  | 161 | T ?`val( vector_transit(T) & src ) { | 
|---|
|  | 162 | T ret = findElemMem_$( *src.col_$, src.idx_$ ); | 
|---|
|  | 163 | return ret; | 
|---|
|  | 164 | } | 
|---|
|  | 165 |  | 
|---|
|  | 166 | void ?=?( vector_transit(T) & src, T val ) { | 
|---|
|  | 167 | findElemMem_$( *src.col_$, src.idx_$ ) = val; | 
|---|
|  | 168 | } | 
|---|
|  | 169 |  | 
|---|
|  | 170 | // exit | 
|---|
|  | 171 |  | 
|---|
|  | 172 | void ?{}( vector_exit(T) & this, vector_transit(T) & src ) { | 
|---|
|  | 173 | ( this.invec_$ ){ src.col_$ }; | 
|---|
|  | 174 | ( this.item_$ ){ & findElemMem_$( *src.col_$, src.idx_$ ) }; | 
|---|
|  | 175 |  | 
|---|
|  | 176 | this.invec_$->exit_refcount_$ ++; | 
|---|
|  | 177 | } | 
|---|
|  | 178 | void ?{}( vector_exit(T) & this, vector_exit(T) & src ){ | 
|---|
|  | 179 | ( this.invec_$ ){ src.invec_$ }; | 
|---|
|  | 180 | ( this.item_$ ){ src.item_$ }; | 
|---|
|  | 181 |  | 
|---|
|  | 182 | this.invec_$->exit_refcount_$ ++; | 
|---|
|  | 183 | } | 
|---|
|  | 184 |  | 
|---|
|  | 185 | void ^?{}( vector_exit(T) & it ) { | 
|---|
|  | 186 | it.invec_$->exit_refcount_$ --; | 
|---|
|  | 187 | } | 
|---|
|  | 188 |  | 
|---|
|  | 189 | T ?`val( vector_exit(T) & src ) { | 
|---|
|  | 190 | return *src.item_$; | 
|---|
|  | 191 | } | 
|---|
|  | 192 |  | 
|---|
|  | 193 | void ?*=?( T & dst, vector_exit(T) & src ) { | 
|---|
|  | 194 | dst = *src.item_$; | 
|---|
|  | 195 | } | 
|---|
|  | 196 |  | 
|---|
|  | 197 | bool ?`moveNext( vector_exit(T) & it ) { | 
|---|
|  | 198 | if (it.invec_$->elems_first_$ == 0p) { | 
|---|
|  | 199 | // vector is empty | 
|---|
|  | 200 | assert ( it.item_$ == 0p ); // it was at origin | 
|---|
|  | 201 | return false; | 
|---|
|  | 202 | } | 
|---|
|  | 203 | assert( it.invec_$->elems_first_$ < it.invec_$->elems_end_$ && "can't handle wraparound yet" ); // temporary: must implement | 
|---|
|  | 204 | if( it.item_$ == 0p ) { | 
|---|
|  | 205 | // moving from origin | 
|---|
|  | 206 | it.item_$ = it.invec_$->elems_first_$; | 
|---|
|  | 207 | } else { | 
|---|
|  | 208 | it.item_$ += 1; | 
|---|
|  | 209 | if( it.item_$ > it.invec_$->buffer_end_$ ) | 
|---|
|  | 210 | it.item_$ = it.invec_$->buffer_first_$; | 
|---|
|  | 211 | } | 
|---|
|  | 212 | if ( it.item_$ >= it.invec_$->elems_end_$ ) { | 
|---|
|  | 213 | // moving to origin | 
|---|
|  | 214 | it.item_$ = 0p; | 
|---|
|  | 215 | return false; | 
|---|
|  | 216 | } else { | 
|---|
|  | 217 | return true; | 
|---|
|  | 218 | } | 
|---|
|  | 219 | } | 
|---|
|  | 220 |  | 
|---|
|  | 221 | // permit | 
|---|
|  | 222 |  | 
|---|
|  | 223 | void ^?{}( vector_permit(T) & this ) { | 
|---|
|  | 224 | remove(this); | 
|---|
|  | 225 | } | 
|---|
|  | 226 |  | 
|---|
|  | 227 | void ?{}( vector_permit(T) & this, vector_transit(T) & src ) { | 
|---|
|  | 228 | ( this.invec_$ ){ src.col_$ }; | 
|---|
|  | 229 | ( this.item_$ ){ & findElemMem_$( *src.col_$, src.idx_$ ) }; | 
|---|
|  | 230 | insert_first( src.col_$->live_iters_$, this ); | 
|---|
|  | 231 | } | 
|---|
|  | 232 |  | 
|---|
|  | 233 | void ?{}( vector_permit(T) & this, vector_exit(T) & src ) { | 
|---|
|  | 234 | ( this.invec_$ ){ src.invec_$ }; | 
|---|
|  | 235 | ( this.item_$ ){ src.item_$ }; | 
|---|
|  | 236 | insert_first( src.invec_$->live_iters_$, this ); | 
|---|
|  | 237 | } | 
|---|
|  | 238 |  | 
|---|
|  | 239 | T ?`val( vector_permit(T) & src ){ | 
|---|
|  | 240 | return *src.item_$; | 
|---|
|  | 241 | } | 
|---|
|  | 242 |  | 
|---|
|  | 243 | // vec internals | 
|---|
|  | 244 |  | 
|---|
|  | 245 | void grow( vector( T ) & this ) { | 
|---|
|  | 246 | size_t newCapacity = 2 * (this.buffer_end_$ - this.buffer_first_$); | 
|---|
|  | 247 | T * newItems = aalloc( newCapacity ); | 
|---|
|  | 248 | size_t elemCount = this`length; | 
|---|
|  | 249 | for ( ptrdiff_t pos = 0; pos < elemCount; pos += 1 ) { | 
|---|
|  | 250 | newItems[pos] = findElemMem_$(this, pos); | 
|---|
|  | 251 | } | 
|---|
|  | 252 |  | 
|---|
|  | 253 | while ( vector_permit(T) & liveIter = this.live_iters_$`elems; liveIter`moveNext ) { | 
|---|
|  | 254 | liveIter.item_$ += (newItems - this.buffer_first_$); | 
|---|
|  | 255 | } | 
|---|
|  | 256 |  | 
|---|
|  | 257 | free( this.buffer_first_$ ); | 
|---|
|  | 258 | this.buffer_first_$ = newItems; | 
|---|
|  | 259 | this.buffer_end_$ = newItems + newCapacity; | 
|---|
|  | 260 | this.elems_first_$ = this.buffer_first_$; | 
|---|
|  | 261 | this.elems_end_$ = this.buffer_first_$ + elemCount; | 
|---|
|  | 262 | assert (this.elems_end_$ < this.buffer_end_$); | 
|---|
|  | 263 | } | 
|---|
|  | 264 |  | 
|---|
|  | 265 | // vec api | 
|---|
|  | 266 |  | 
|---|
|  | 267 | vector_transit(T) push_last( vector( T ) & col, T val ) { | 
|---|
|  | 268 | assert (col.exit_refcount_$ == 0); | 
|---|
|  | 269 | if (col`length >= col`capacity) { | 
|---|
|  | 270 | assert (col`length == col`capacity); | 
|---|
|  | 271 | grow(col); | 
|---|
|  | 272 | } | 
|---|
|  | 273 | // call autogen constructor deleted at end of hfa | 
|---|
|  | 274 | vector_transit(T) ret = { & col, col`length }; | 
|---|
|  | 275 | *col.elems_end_$ = val; | 
|---|
|  | 276 | if (col.elems_first_$ == 0p) col.elems_first_$ = col.elems_end_$; | 
|---|
|  | 277 | col.elems_end_$ += 1; | 
|---|
|  | 278 | if (col.elems_end_$ >= col.buffer_end_$) col.elems_end_$ = col.buffer_first_$; | 
|---|
|  | 279 | return ret; | 
|---|
|  | 280 | } | 
|---|
|  | 281 |  | 
|---|
|  | 282 | vector_exit(T) ?`origin( vector( T ) & vec ) { | 
|---|
|  | 283 |  | 
|---|
|  | 284 | // private memberwise constructor, deleted from global namespace at end | 
|---|
|  | 285 | // autogen constructor would not do the raii | 
|---|
|  | 286 | void ?{}( vector_exit(T) & this, vector(T) * invec_$, T * item_$ ) { | 
|---|
|  | 287 | ( this.invec_$ ){ invec_$ }; | 
|---|
|  | 288 | ( this.item_$ ){ item_$ }; | 
|---|
|  | 289 | this.invec_$->exit_refcount_$ ++; | 
|---|
|  | 290 | } | 
|---|
|  | 291 |  | 
|---|
|  | 292 | vector_exit(T) ret = { &vec, 0p }; | 
|---|
|  | 293 | return ret; | 
|---|
|  | 294 | } | 
|---|
|  | 295 |  | 
|---|
|  | 296 | bool inRange_$( T * query, T * from, T * to) { | 
|---|
|  | 297 | if (from == to) return false; | 
|---|
|  | 298 | if (from < to) return from <= query && query < to; | 
|---|
|  | 299 | return query < to || from <= query; | 
|---|
|  | 300 | } | 
|---|
|  | 301 |  | 
|---|
|  | 302 | void insert_before( vector( T ) & col, ptrdiff_t idx, T val ) { | 
|---|
|  | 303 | assert (col.exit_refcount_$ == 0); | 
|---|
|  | 304 | if (col`length >= col`capacity) { | 
|---|
|  | 305 | assert (col`length == col`capacity); | 
|---|
|  | 306 | grow(col); | 
|---|
|  | 307 | } | 
|---|
|  | 308 |  | 
|---|
|  | 309 | T & insertTargetR = findElemMem_$( col, idx ); | 
|---|
|  | 310 | T * insertTarget = & insertTargetR; // doesn't work in one line; must be a bug | 
|---|
|  | 311 |  | 
|---|
|  | 312 | // bubble toward back | 
|---|
|  | 313 | if ( col.elems_end_$ < insertTarget ) { | 
|---|
|  | 314 | // two phases of bubbling, to wrap around | 
|---|
|  | 315 | for (T * tgt = col.elems_end_$; tgt > col.buffer_first_$; tgt--) { | 
|---|
|  | 316 | *tgt = *(tgt-1); | 
|---|
|  | 317 | } | 
|---|
|  | 318 | *col.buffer_first_$ = *(col.buffer_end_$ - 1); | 
|---|
|  | 319 | for (T * tgt = col.buffer_end_$ - 1; tgt > insertTarget; tgt--) { | 
|---|
|  | 320 | *tgt = *(tgt-1); | 
|---|
|  | 321 | } | 
|---|
|  | 322 | } else { | 
|---|
|  | 323 | for (T * tgt = col.elems_end_$; tgt > insertTarget; tgt--) { | 
|---|
|  | 324 | *tgt = *(tgt-1); | 
|---|
|  | 325 | } | 
|---|
|  | 326 | } | 
|---|
|  | 327 |  | 
|---|
|  | 328 | col.elems_end_$ += 1; | 
|---|
|  | 329 | if (col.elems_end_$ == col.buffer_end_$) col.elems_end_$ = col.buffer_first_$; | 
|---|
|  | 330 |  | 
|---|
|  | 331 | *insertTarget = val; | 
|---|
|  | 332 |  | 
|---|
|  | 333 | while ( vector_permit(T) & liveIter = col.live_iters_$`elems; liveIter`moveNext ) { | 
|---|
|  | 334 | if ( inRange_$(liveIter.item_$, insertTarget, col.elems_end_$) ) { | 
|---|
|  | 335 | liveIter.item_$ += 1; | 
|---|
|  | 336 | if (liveIter.item_$ >= col.buffer_end_$) liveIter.item_$ = col.buffer_first_$; | 
|---|
|  | 337 | } | 
|---|
|  | 338 | } | 
|---|
|  | 339 | } | 
|---|
|  | 340 |  | 
|---|
|  | 341 | size_t ?`capacity( vector(T) & v ) { | 
|---|
|  | 342 | return v.buffer_end_$ - v.buffer_first_$; | 
|---|
|  | 343 | } | 
|---|
|  | 344 |  | 
|---|
|  | 345 | size_t ?`length( vector(T) & v ) { | 
|---|
|  | 346 | if (v.elems_first_$ == 0p) return 0; | 
|---|
|  | 347 | if (v.elems_first_$ < v.elems_end_$ ) return v.elems_end_$ - v.elems_first_$; | 
|---|
|  | 348 | return v.buffer_end_$ - v.elems_first_$ + v.elems_end_$ - v.buffer_first_$; | 
|---|
|  | 349 | } | 
|---|
|  | 350 |  | 
|---|
|  | 351 |  | 
|---|
|  | 352 | } // forall T | 
|---|
|  | 353 |  | 
|---|
|  | 354 | forall( T ) { | 
|---|
|  | 355 | void ?{}( vector_exit(T) &, vector(T) *, T * ) = void; | 
|---|
|  | 356 | void ?{}( vector_transit(T) & this, vector( T ) * col, ptrdiff_t idx ) = void; | 
|---|
|  | 357 | } | 
|---|