|
1 /* Copyright (C) 2002 Jean-Marc Valin */ |
|
2 /** |
|
3 @file stack_alloc.h |
|
4 @brief Temporary memory allocation on stack |
|
5 */ |
|
6 /* |
|
7 Redistribution and use in source and binary forms, with or without |
|
8 modification, are permitted provided that the following conditions |
|
9 are met: |
|
10 |
|
11 - Redistributions of source code must retain the above copyright |
|
12 notice, this list of conditions and the following disclaimer. |
|
13 |
|
14 - Redistributions in binary form must reproduce the above copyright |
|
15 notice, this list of conditions and the following disclaimer in the |
|
16 documentation and/or other materials provided with the distribution. |
|
17 |
|
18 - Neither the name of the Xiph.org Foundation nor the names of its |
|
19 contributors may be used to endorse or promote products derived from |
|
20 this software without specific prior written permission. |
|
21 |
|
22 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
23 ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
24 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
25 A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR |
|
26 CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
|
27 EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
|
28 PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
|
29 PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
|
30 LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
|
31 NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
|
32 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
33 */ |
|
34 |
|
35 #ifndef STACK_ALLOC_H |
|
36 #define STACK_ALLOC_H |
|
37 |
|
38 #ifdef WIN32 |
|
39 # include <malloc.h> |
|
40 # ifndef alloca |
|
41 # define alloca(_x) _alloca(_x); |
|
42 # endif |
|
43 #else |
|
44 #ifdef HAVE_ALLOCA_H |
|
45 # include <alloca.h> |
|
46 # else |
|
47 # include <stdlib.h> |
|
48 # endif |
|
49 #endif |
|
50 |
|
51 /** |
|
52 * @def ALIGN(stack, size) |
|
53 * |
|
54 * Aligns the stack to a 'size' boundary |
|
55 * |
|
56 * @param stack Stack |
|
57 * @param size New size boundary |
|
58 */ |
|
59 |
|
60 /** |
|
61 * @def PUSH(stack, size, type) |
|
62 * |
|
63 * Allocates 'size' elements of type 'type' on the stack |
|
64 * |
|
65 * @param stack Stack |
|
66 * @param size Number of elements |
|
67 * @param type Type of element |
|
68 */ |
|
69 |
|
70 /** |
|
71 * @def VARDECL(var) |
|
72 * |
|
73 * Declare variable on stack |
|
74 * |
|
75 * @param var Variable to declare |
|
76 */ |
|
77 |
|
78 /** |
|
79 * @def ALLOC(var, size, type) |
|
80 * |
|
81 * Allocate 'size' elements of 'type' on stack |
|
82 * |
|
83 * @param var Name of variable to allocate |
|
84 * @param size Number of elements |
|
85 * @param type Type of element |
|
86 */ |
|
87 |
|
88 #ifdef ENABLE_VALGRIND |
|
89 |
|
90 #include <valgrind/memcheck.h> |
|
91 |
|
92 #define ALIGN(stack, size) ((stack) += ((size) - (long)(stack)) & ((size) - 1)) |
|
93 |
|
94 #define PUSH(stack, size, type) (VALGRIND_MAKE_NOACCESS(stack, 1000),ALIGN((stack),sizeof(type)),VALGRIND_MAKE_WRITABLE(stack, ((size)*sizeof(type))),(stack)+=((size)*sizeof(type)),(type*)((stack)-((size)*sizeof(type)))) |
|
95 |
|
96 #else |
|
97 |
|
98 #define ALIGN(stack, size) ((stack) += ((size) - (long)(stack)) & ((size) - 1)) |
|
99 |
|
100 #define PUSH(stack, size, type) (ALIGN((stack),sizeof(type)),(stack)+=((size)*sizeof(type)),(type*)((stack)-((size)*sizeof(type)))) |
|
101 |
|
102 #endif |
|
103 |
|
104 #if defined(VAR_ARRAYS) |
|
105 #define VARDECL(var) |
|
106 #define ALLOC(var, size, type) type var[size] |
|
107 #elif defined(USE_ALLOCA) |
|
108 #define VARDECL(var) var |
|
109 #define ALLOC(var, size, type) var = alloca(sizeof(type)*(size)) |
|
110 #else |
|
111 #define VARDECL(var) var |
|
112 #define ALLOC(var, size, type) var = PUSH(stack, size, type) |
|
113 #endif |
|
114 |
|
115 |
|
116 #endif |