BLI_math_rotation: properly name the quaternion power function.
[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                 /* calculate best chunk-count to fit a power of two */
47                 unsigned int chunk_count = chunk_size;
48                 {
49                         unsigned int size = chunk_count * stride;
50                         size = power_of_2_max_u(size);
51                         size = MEM_SIZE_OPTIMAL(size);
52                         chunk_count = size / stride;
53                 }
54
55                 (*bs_p) = BLI_array_store_create(stride, chunk_count);
56         }
57         return *bs_p;
58 }
59
60 BArrayStore *BLI_array_store_at_size_get(
61         struct BArrayStore_AtSize *bs_stride,
62         const int stride)
63 {
64         BLI_assert(stride > 0 && stride <= bs_stride->stride_table_len);
65         return bs_stride->stride_table[stride - 1];
66 }
67
68 void BLI_array_store_at_size_clear(
69         struct BArrayStore_AtSize *bs_stride)
70 {
71         for (int i = 0; i < bs_stride->stride_table_len; i += 1) {
72                 if (bs_stride->stride_table[i]) {
73                         BLI_array_store_destroy(bs_stride->stride_table[i]);
74                 }
75         }
76
77         MEM_freeN(bs_stride->stride_table);
78         bs_stride->stride_table = NULL;
79         bs_stride->stride_table_len = 0;
80 }
81
82
83 void BLI_array_store_at_size_calc_memory_usage(
84         struct BArrayStore_AtSize *bs_stride,
85         size_t *r_size_expanded, size_t *r_size_compacted)
86 {
87         size_t size_compacted = 0;
88         size_t size_expanded  = 0;
89         for (int i = 0; i < bs_stride->stride_table_len; i++) {
90                 BArrayStore *bs = bs_stride->stride_table[i];
91                 if (bs) {
92                         size_compacted += BLI_array_store_calc_size_compacted_get(bs);
93                         size_expanded  += BLI_array_store_calc_size_expanded_get(bs);
94                 }
95         }
96
97         *r_size_expanded = size_expanded;
98         *r_size_compacted = size_compacted;
99 }