4debe1a4b0415a5982b4bf13fa4c4ca8a755f560
[blender.git] / source / blender / blenlib / intern / BLI_memarena.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  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  * Efficient memory allocation for lots of similar small chunks.
27  */
28
29 /** \file blender/blenlib/intern/BLI_memarena.c
30  *  \ingroup bli
31  */
32
33
34
35 #include "MEM_guardedalloc.h"
36
37 #include "BLI_memarena.h"
38 #include "BLI_linklist.h"
39
40 struct MemArena {
41         unsigned char *curbuf;
42         int bufsize, cursize;
43         const char *name;
44         
45         int use_calloc; 
46         int align;
47         
48         LinkNode *bufs;
49 };
50
51 MemArena *BLI_memarena_new(int bufsize, const char *name)
52 {
53         MemArena *ma= MEM_callocN(sizeof(*ma), "memarena");
54         ma->bufsize= bufsize;
55         ma->align = 8;
56         ma->name= name;
57         
58         return ma;
59 }
60
61 void BLI_memarena_use_calloc(MemArena *ma)
62 {
63         ma->use_calloc= 1;
64 }
65
66 void BLI_memarena_use_malloc(MemArena *ma)
67 {
68         ma->use_calloc= 0;
69 }
70
71 void BLI_memarena_use_align(struct MemArena *ma, int align)
72 {
73         /* align should be a power of two */
74         ma->align = align;
75 }
76
77 void BLI_memarena_free(MemArena *ma)
78 {
79         BLI_linklist_free(ma->bufs, (void(*)(void*)) MEM_freeN);
80         MEM_freeN(ma);
81 }
82
83         /* amt must be power of two */
84 #define PADUP(num, amt) ((num+(amt-1))&~(amt-1))
85
86 void *BLI_memarena_alloc(MemArena *ma, int size)
87 {
88         void *ptr;
89
90                 /* ensure proper alignment by rounding
91                  * size up to multiple of 8 */  
92         size= PADUP(size, ma->align);
93         
94         if (size>ma->cursize) {
95                 unsigned char *tmp;
96                 
97                 if(size > ma->bufsize - (ma->align - 1))
98                 {
99                         ma->cursize = PADUP(size+1, ma->align);
100                 }
101                 else ma->cursize = ma->bufsize;
102
103                 if(ma->use_calloc)
104                         ma->curbuf= MEM_callocN(ma->cursize, ma->name);
105                 else
106                         ma->curbuf= MEM_mallocN(ma->cursize, ma->name);
107                 
108                 BLI_linklist_prepend(&ma->bufs, ma->curbuf);
109
110                 /* align alloc'ed memory (needed if align > 8) */
111                 tmp = (unsigned char*)PADUP( (intptr_t) ma->curbuf, ma->align);
112                 ma->cursize -= (tmp - ma->curbuf);
113                 ma->curbuf = tmp;               
114         }
115         
116         ptr= ma->curbuf;
117         ma->curbuf+= size;
118         ma->cursize-= size;
119         
120         return ptr;
121 }
122