83cd28ddf11de0144137d4a6e513ff98e621a2a7
[blender.git] / source / blender / blenlib / intern / array_store_utils.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * ***** END GPL LICENSE BLOCK *****
19  */
20
21 /** \file blender/blenlib/intern/array_store_utils.c
22  *  \ingroup bli
23  *  \brief Helper functions for BLI_array_store API.
24  */
25
26 #include "MEM_guardedalloc.h"
27
28 #include "BLI_utildefines.h"
29
30 #include "BLI_array_store.h"
31 #include "BLI_array_store_utils.h"  /* own include */
32
33 #include "BLI_math_base.h"
34
35 BArrayStore *BLI_array_store_at_size_ensure(
36         struct BArrayStore_AtSize *bs_stride,
37         const int stride, const int chunk_size)
38 {
39         if (bs_stride->stride_table_len < stride) {
40                 bs_stride->stride_table_len = stride;
41                 bs_stride->stride_table = MEM_recallocN(bs_stride->stride_table, sizeof(*bs_stride->stride_table) * stride);
42         }
43         BArrayStore **bs_p = &bs_stride->stride_table[stride - 1];
44
45         if ((*bs_p) == NULL) {
46 #if 0
47                 unsigned int chunk_count = chunk_size;
48 #else
49                 /* calculate best chunk-count to fit a power of two */
50                 unsigned int chunk_count = chunk_size;
51                 {
52                         unsigned int size = chunk_count * stride;
53                         size = power_of_2_max_u(size);
54                         size = MEM_SIZE_OPTIMAL(size);
55                         chunk_count = size / stride;
56                 }
57 #endif
58
59                 (*bs_p) = BLI_array_store_create(stride, chunk_count);
60         }
61         return *bs_p;
62 }
63
64 BArrayStore *BLI_array_store_at_size_get(
65         struct BArrayStore_AtSize *bs_stride,
66         const int stride)
67 {
68         BLI_assert(stride > 0 && stride <= bs_stride->stride_table_len);
69         return bs_stride->stride_table[stride - 1];
70 }
71
72 void BLI_array_store_at_size_clear(
73         struct BArrayStore_AtSize *bs_stride)
74 {
75         for (int i = 0; i < bs_stride->stride_table_len; i += 1) {
76                 if (bs_stride->stride_table[i]) {
77                         BLI_array_store_destroy(bs_stride->stride_table[i]);
78                 }
79         }
80
81         MEM_freeN(bs_stride->stride_table);
82         bs_stride->stride_table = NULL;
83         bs_stride->stride_table_len = 0;
84 }
85
86
87 void BLI_array_store_at_size_calc_memory_usage(
88         struct BArrayStore_AtSize *bs_stride,
89         size_t *r_size_expanded, size_t *r_size_compacted)
90 {
91         size_t size_compacted = 0;
92         size_t size_expanded  = 0;
93         for (int i = 0; i < bs_stride->stride_table_len; i++) {
94                 BArrayStore *bs = bs_stride->stride_table[i];
95                 if (bs) {
96                         size_compacted += BLI_array_store_calc_size_compacted_get(bs);
97                         size_expanded  += BLI_array_store_calc_size_expanded_get(bs);
98                 }
99         }
100
101         *r_size_expanded = size_expanded;
102         *r_size_compacted = size_compacted;
103 }