src/lib/tvgArray.h

   1 /*
   2  * Copyright (c) 2020 - 2023 the ThorVG project. All rights reserved.
   3
   4  * Permission is hereby granted, free of charge, to any person obtaining a copy
   5  * of this software and associated documentation files (the "Software"), to deal
   6  * in the Software without restriction, including without limitation the rights
   7  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   8  * copies of the Software, and to permit persons to whom the Software is
   9  * furnished to do so, subject to the following conditions:
  10
  11  * The above copyright notice and this permission notice shall be included in all
  12  * copies or substantial portions of the Software.
  13
  14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  20  * SOFTWARE.
  21  */
  22
  23 #ifndef _TVG_ARRAY_H_
  24 #define _TVG_ARRAY_H_
  25
  26 #include <memory.h>
  27
  28 namespace tvg
  29 {
  30
  31 template<class T>
  32 struct Array
  33 {
  34     T* data = nullptr;
  35     uint32_t count = 0;
  36     uint32_t reserved = 0;
  37
  38     void push(T element)
  39     {
  40         if (count + 1 > reserved) {
  41             reserved = (count + 1) * 2;
  42             auto p  = data;
  43             data = static_cast<T*>(realloc(data, sizeof(T) * reserved));
  44             if (!data) {
  45                 data = p;
  46                 return;
  47             }
  48         }
  49         data[count++] = element;
  50     }
  51
  52     bool reserve(uint32_t size)
  53     {
  54         if (size > reserved) {
  55             reserved = size;
  56             auto p = data;
  57             data = static_cast<T*>(realloc(data, sizeof(T) * reserved));
  58             if (!data) {
  59                 data = p;
  60                 return false;
  61             }
  62         }
  63         return true;
  64     }
  65
  66     bool grow(uint32_t size)
  67     {
  68         return reserve(count + size);
  69     }
  70
  71     T* ptr()
  72     {
  73         return data + count;
  74     }
  75
  76     void pop()
  77     {
  78         if (count > 0) --count;
  79     }
  80
  81     void reset()
  82     {
  83         if (data) {
  84             free(data);
  85             data = nullptr;
  86         }
  87         count = reserved = 0;
  88     }
  89
  90     void clear()
  91     {
  92         count = 0;
  93     }
  94
  95     void operator=(const Array& rhs)
  96     {
  97         reserve(rhs.count);
  98         if (rhs.count > 0) memcpy(data, rhs.data, sizeof(T) * reserved);
  99         count = rhs.count;
 100     }
 101
 102     ~Array()
 103     {
 104         if (data) free(data);
 105     }
 106 };
 107
 108 }
 109
 110 #endif //_TVG_ARRAY_H_