148 lines
		
	
	
		
			5.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			148 lines
		
	
	
		
			5.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*******************************************************************************************[Vec.h]
 | |
| MiniSat -- Copyright (c) 2003-2006, Niklas Een, Niklas Sorensson
 | |
| 
 | |
| Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
 | |
| associated documentation files (the "Software"), to deal in the Software without restriction,
 | |
| including without limitation the rights to use, copy, modify, merge, publish, distribute,
 | |
| sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
 | |
| furnished to do so, subject to the following conditions:
 | |
| 
 | |
| The above copyright notice and this permission notice shall be included in all copies or
 | |
| substantial portions of the Software.
 | |
| 
 | |
| THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT
 | |
| NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 | |
| NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
 | |
| DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT
 | |
| OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 | |
| **************************************************************************************************/
 | |
| 
 | |
| #ifndef BoxedVec_h
 | |
| #define BoxedVec_h
 | |
| 
 | |
| #include <cstdlib>
 | |
| #include <cassert>
 | |
| #include <new>
 | |
| 
 | |
| //=================================================================================================
 | |
| // Automatically resizable arrays
 | |
| //
 | |
| // NOTE! Don't use this vector on datatypes that cannot be re-located in memory (with realloc)
 | |
| 
 | |
| template<class T>
 | |
| class bvec {
 | |
| 
 | |
|     static inline int imin(int x, int y) {
 | |
|         int mask = (x-y) >> (sizeof(int)*8-1);
 | |
|         return (x&mask) + (y&(~mask)); }
 | |
| 
 | |
|     static inline int imax(int x, int y) {
 | |
|         int mask = (y-x) >> (sizeof(int)*8-1);
 | |
|         return (x&mask) + (y&(~mask)); }
 | |
| 
 | |
|     struct Vec_t {
 | |
|         int sz;
 | |
|         int cap;
 | |
|         T   data[0];
 | |
| 
 | |
|         static Vec_t* alloc(Vec_t* x, int size){
 | |
|             x = (Vec_t*)realloc((void*)x, sizeof(Vec_t) + sizeof(T)*size);
 | |
|             x->cap = size;
 | |
|             return x;
 | |
|         }
 | |
|         
 | |
|     };
 | |
| 
 | |
|     Vec_t* ref;
 | |
| 
 | |
|     static const int init_size = 2;
 | |
|     static int   nextSize (int current) { return (current * 3 + 1) >> 1; }
 | |
|     static int   fitSize  (int needed)  { int x; for (x = init_size; needed > x; x = nextSize(x)); return x; }
 | |
| 
 | |
|     void fill (int size) {
 | |
|         assert(ref != NULL);
 | |
|         for (T* i = ref->data; i < ref->data + size; i++)
 | |
|             new (i) T();
 | |
|     }
 | |
| 
 | |
|     void fill (int size, const T& pad) {
 | |
|         assert(ref != NULL);
 | |
|         for (T* i = ref->data; i < ref->data + size; i++)
 | |
|             new (i) T(pad);
 | |
|     }
 | |
| 
 | |
|     // Don't allow copying (error prone):
 | |
|     altvec<T>&  operator = (altvec<T>& other) { assert(0); }
 | |
|     altvec (altvec<T>& other)                  { assert(0); }
 | |
| 
 | |
| public:
 | |
|     void     clear  (bool dealloc = false) { 
 | |
|         if (ref != NULL){
 | |
|             for (int i = 0; i < ref->sz; i++) 
 | |
|                 (*ref).data[i].~T();
 | |
| 
 | |
|             if (dealloc) { 
 | |
|                 free(ref); ref = NULL; 
 | |
|             }else 
 | |
|                 ref->sz = 0;
 | |
|         } 
 | |
|     }
 | |
| 
 | |
|     // Constructors:
 | |
|     altvec(void)                   : ref (NULL) { }
 | |
|     altvec(int size)               : ref (Vec_t::alloc(NULL, fitSize(size))) { fill(size);      ref->sz = size; }
 | |
|     altvec(int size, const T& pad) : ref (Vec_t::alloc(NULL, fitSize(size))) { fill(size, pad); ref->sz = size; }
 | |
|    ~altvec(void) { clear(true); }
 | |
| 
 | |
|     // Ownership of underlying array:
 | |
|     operator T*       (void)           { return ref->data; }     // (unsafe but convenient)
 | |
|     operator const T* (void) const     { return ref->data; }
 | |
| 
 | |
|     // Size operations:
 | |
|     int      size   (void) const       { return ref != NULL ? ref->sz : 0; }
 | |
| 
 | |
|     void     pop    (void)             { assert(ref != NULL && ref->sz > 0); int last = --ref->sz; ref->data[last].~T(); }
 | |
|     void     push   (const T& elem) {
 | |
|         int size = ref != NULL ? ref->sz  : 0;
 | |
|         int cap  = ref != NULL ? ref->cap : 0;
 | |
|         if (size == cap){
 | |
|             cap = cap != 0 ? nextSize(cap) : init_size;
 | |
|             ref = Vec_t::alloc(ref, cap); 
 | |
|         }
 | |
|         //new (&ref->data[size]) T(elem); 
 | |
|         ref->data[size] = elem; 
 | |
|         ref->sz = size+1; 
 | |
|     }
 | |
| 
 | |
|     void     push   () {
 | |
|         int size = ref != NULL ? ref->sz  : 0;
 | |
|         int cap  = ref != NULL ? ref->cap : 0;
 | |
|         if (size == cap){
 | |
|             cap = cap != 0 ? nextSize(cap) : init_size;
 | |
|             ref = Vec_t::alloc(ref, cap); 
 | |
|         }
 | |
|         new (&ref->data[size]) T(); 
 | |
|         ref->sz = size+1; 
 | |
|     }
 | |
| 
 | |
|     void     shrink (int nelems)             { for (int i = 0; i < nelems; i++) pop(); }
 | |
|     void     shrink_(int nelems)             { for (int i = 0; i < nelems; i++) pop(); }
 | |
|     void     growTo (int size)               { while (this->size() < size) push(); }
 | |
|     void     growTo (int size, const T& pad) { while (this->size() < size) push(pad); }
 | |
|     void     capacity (int size)             { growTo(size); }
 | |
| 
 | |
|     const T& last  (void) const              { return ref->data[ref->sz-1]; }
 | |
|     T&       last  (void)                    { return ref->data[ref->sz-1]; }
 | |
| 
 | |
|     // Vector interface:
 | |
|     const T& operator [] (int index) const  { return ref->data[index]; }
 | |
|     T&       operator [] (int index)        { return ref->data[index]; }
 | |
| 
 | |
|     void copyTo(altvec<T>& copy) const { copy.clear(); for (int i = 0; i < size(); i++) copy.push(ref->data[i]); }
 | |
|     void moveTo(altvec<T>& dest) { dest.clear(true); dest.ref = ref; ref = NULL; }
 | |
| 
 | |
| };
 | |
| 
 | |
| 
 | |
| #endif
 |