1 /*
2 * (C) Copyright 2005- ECMWF.
3 *
4 * This software is licensed under the terms of the Apache Licence Version 2.0
5 * which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
6 *
7 * In applying this licence, ECMWF does not waive the privileges and immunities granted to it by
8 * virtue of its status as an intergovernmental organisation nor does it submit to any jurisdiction.
9 */
10
11 /***************************************************************************
12 *
13 * Enrico Fucile
14 *
15 ***************************************************************************/
16
17 #include "grib_api_internal.h"
18
19 /* For debugging purposes */
grib_iarray_print(const char * title,const grib_iarray * iarray)20 void grib_iarray_print(const char* title, const grib_iarray* iarray)
21 {
22 size_t i;
23 Assert(iarray);
24 printf("%s: iarray.n=%lu \t", title, (unsigned long)iarray->n);
25 for (i = 0; i < iarray->n; i++) {
26 printf("iarray[%lu]=%ld\t", (unsigned long)i, iarray->v[i]);
27 }
28 printf("\n");
29 }
30
grib_iarray_new_from_array(grib_context * c,long * a,size_t size)31 grib_iarray* grib_iarray_new_from_array(grib_context* c, long* a, size_t size)
32 {
33 size_t i;
34 grib_iarray* v;
35
36 if (!c)
37 c = grib_context_get_default();
38
39 v = grib_iarray_new(c, size, 100);
40 for (i = 0; i < size; i++)
41 v->v[i] = a[i];
42 v->n = size;
43 v->number_of_pop_front = 0;
44 v->context = c;
45 return v;
46 }
47
grib_iarray_new(grib_context * c,size_t size,size_t incsize)48 grib_iarray* grib_iarray_new(grib_context* c, size_t size, size_t incsize)
49 {
50 grib_iarray* v = NULL;
51
52 if (!c)
53 c = grib_context_get_default();
54
55 v = (grib_iarray*)grib_context_malloc(c, sizeof(grib_iarray));
56 if (!v) {
57 grib_context_log(c, GRIB_LOG_ERROR,
58 "grib_iarray_new unable to allocate %ld bytes\n", sizeof(grib_iarray));
59 return NULL;
60 }
61 v->context = c;
62 v->size = size;
63 v->n = 0;
64 v->incsize = incsize;
65 v->v = (long*)grib_context_malloc(c, sizeof(long) * size);
66 v->number_of_pop_front = 0;
67 if (!v->v) {
68 grib_context_log(c, GRIB_LOG_ERROR,
69 "grib_iarray_new unable to allocate %ld bytes\n", sizeof(long) * size);
70 return NULL;
71 }
72 return v;
73 }
74
grib_iarray_pop(grib_iarray * a)75 long grib_iarray_pop(grib_iarray* a)
76 {
77 a->n -= 1;
78 return a->v[a->n];
79 }
80
grib_iarray_pop_front(grib_iarray * a)81 long grib_iarray_pop_front(grib_iarray* a)
82 {
83 long v = a->v[0];
84 /* size_t i=0; */
85 if (a->n == 0)
86 Assert(0);
87 a->n--;
88 a->v++;
89 a->number_of_pop_front++;
90 /* for (i=0;i<a->n;i++) a->v[i]=a->v[i+1]; */
91
92 return v;
93 }
94
grib_iarray_resize_to(grib_iarray * v,size_t newsize)95 static grib_iarray* grib_iarray_resize_to(grib_iarray* v, size_t newsize)
96 {
97 long* newv;
98 size_t i;
99 grib_context* c = v->context;
100
101 if (newsize < v->size)
102 return v;
103
104 if (!c)
105 c = grib_context_get_default();
106
107 newv = (long*)grib_context_malloc_clear(c, newsize * sizeof(long));
108 if (!newv) {
109 grib_context_log(c, GRIB_LOG_ERROR,
110 "grib_iarray_resize unable to allocate %ld bytes\n", sizeof(long) * newsize);
111 return NULL;
112 }
113
114 for (i = 0; i < v->n; i++)
115 newv[i] = v->v[i];
116
117 v->v -= v->number_of_pop_front;
118 grib_context_free(c, v->v);
119
120 v->v = newv;
121 v->size = newsize;
122 v->number_of_pop_front = 0;
123
124 return v;
125 }
126
grib_iarray_resize(grib_iarray * v)127 static grib_iarray* grib_iarray_resize(grib_iarray* v)
128 {
129 const size_t newsize = v->incsize + v->size;
130 return grib_iarray_resize_to(v, newsize);
131 }
132
grib_iarray_push(grib_iarray * v,long val)133 grib_iarray* grib_iarray_push(grib_iarray* v, long val)
134 {
135 size_t start_size = 100;
136 size_t start_incsize = 100;
137
138 if (!v)
139 v = grib_iarray_new(0, start_size, start_incsize);
140
141 if (v->n >= v->size - v->number_of_pop_front)
142 v = grib_iarray_resize(v);
143
144 v->v[v->n] = val;
145 v->n++;
146 return v;
147 }
148
grib_iarray_push_front(grib_iarray * v,long val)149 grib_iarray* grib_iarray_push_front(grib_iarray* v, long val)
150 {
151 size_t start_size = 100;
152 size_t start_incsize = 100;
153 int i;
154 if (!v)
155 v = grib_iarray_new(0, start_size, start_incsize);
156
157 if (v->number_of_pop_front) {
158 v->v--;
159 v->number_of_pop_front--;
160 }
161 else {
162 if (v->n >= v->size)
163 v = grib_iarray_resize(v);
164 for (i = v->n; i > 0; i--)
165 v[i] = v[i - 1];
166 }
167 v->v[0] = val;
168 v->n++;
169
170 return v;
171 }
172
grib_iarray_push_array(grib_iarray * v,long * val,size_t size)173 grib_iarray* grib_iarray_push_array(grib_iarray* v, long* val, size_t size)
174 {
175 size_t start_size = size;
176 size_t start_incsize = 100;
177 long* vp = 0;
178 long* valp = val;
179 if (!v)
180 v = grib_iarray_new(0, start_size, start_incsize);
181
182 v = grib_iarray_resize_to(v, size + v->n);
183 vp = v->v + v->n + v->number_of_pop_front;
184 v->n += size;
185 while (size) {
186 *(vp++) = *(valp++);
187 size--;
188 }
189 return v;
190 }
191
grib_iarray_delete(grib_iarray * v)192 void grib_iarray_delete(grib_iarray* v)
193 {
194 grib_context* c;
195
196 if (!v)
197 return;
198 c = v->context;
199
200 grib_iarray_delete_array(v);
201
202 grib_context_free(c, v);
203 }
204
grib_iarray_delete_array(grib_iarray * v)205 void grib_iarray_delete_array(grib_iarray* v)
206 {
207 grib_context* c;
208
209 if (!v)
210 return;
211 c = v->context;
212
213 if (v->v) {
214 long* vv = v->v - v->number_of_pop_front;
215 grib_context_free(c, vv);
216 }
217 }
218
grib_iarray_get_array(grib_iarray * v)219 long* grib_iarray_get_array(grib_iarray* v)
220 {
221 long* vv;
222 size_t i;
223 grib_context* c = v->context;
224 DebugAssert(c);
225
226 vv = (long*)grib_context_malloc_clear(c, sizeof(long) * v->n);
227 for (i = 0; i < v->n; i++)
228 vv[i] = v->v[i];
229
230 return vv;
231 }
232
grib_iarray_used_size(grib_iarray * v)233 size_t grib_iarray_used_size(grib_iarray* v)
234 {
235 return v == NULL ? 0 : v->n;
236 }
237