cannam@154
|
1 /* Copyright (C) 2002-2003 Jean-Marc Valin
|
cannam@154
|
2 Copyright (C) 2007-2009 Xiph.Org Foundation */
|
cannam@154
|
3 /**
|
cannam@154
|
4 @file stack_alloc.h
|
cannam@154
|
5 @brief Temporary memory allocation on stack
|
cannam@154
|
6 */
|
cannam@154
|
7 /*
|
cannam@154
|
8 Redistribution and use in source and binary forms, with or without
|
cannam@154
|
9 modification, are permitted provided that the following conditions
|
cannam@154
|
10 are met:
|
cannam@154
|
11
|
cannam@154
|
12 - Redistributions of source code must retain the above copyright
|
cannam@154
|
13 notice, this list of conditions and the following disclaimer.
|
cannam@154
|
14
|
cannam@154
|
15 - Redistributions in binary form must reproduce the above copyright
|
cannam@154
|
16 notice, this list of conditions and the following disclaimer in the
|
cannam@154
|
17 documentation and/or other materials provided with the distribution.
|
cannam@154
|
18
|
cannam@154
|
19 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
cannam@154
|
20 ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
cannam@154
|
21 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
cannam@154
|
22 A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
cannam@154
|
23 OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
cannam@154
|
24 EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
cannam@154
|
25 PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
cannam@154
|
26 PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
cannam@154
|
27 LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
cannam@154
|
28 NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
cannam@154
|
29 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
cannam@154
|
30 */
|
cannam@154
|
31
|
cannam@154
|
32 #ifndef STACK_ALLOC_H
|
cannam@154
|
33 #define STACK_ALLOC_H
|
cannam@154
|
34
|
cannam@154
|
35 #include "opus_types.h"
|
cannam@154
|
36 #include "opus_defines.h"
|
cannam@154
|
37
|
cannam@154
|
38 #if (!defined (VAR_ARRAYS) && !defined (USE_ALLOCA) && !defined (NONTHREADSAFE_PSEUDOSTACK))
|
cannam@154
|
39 #error "Opus requires one of VAR_ARRAYS, USE_ALLOCA, or NONTHREADSAFE_PSEUDOSTACK be defined to select the temporary allocation mode."
|
cannam@154
|
40 #endif
|
cannam@154
|
41
|
cannam@154
|
42 #ifdef USE_ALLOCA
|
cannam@154
|
43 # ifdef WIN32
|
cannam@154
|
44 # include <malloc.h>
|
cannam@154
|
45 # else
|
cannam@154
|
46 # ifdef HAVE_ALLOCA_H
|
cannam@154
|
47 # include <alloca.h>
|
cannam@154
|
48 # else
|
cannam@154
|
49 # include <stdlib.h>
|
cannam@154
|
50 # endif
|
cannam@154
|
51 # endif
|
cannam@154
|
52 #endif
|
cannam@154
|
53
|
cannam@154
|
54 /**
|
cannam@154
|
55 * @def ALIGN(stack, size)
|
cannam@154
|
56 *
|
cannam@154
|
57 * Aligns the stack to a 'size' boundary
|
cannam@154
|
58 *
|
cannam@154
|
59 * @param stack Stack
|
cannam@154
|
60 * @param size New size boundary
|
cannam@154
|
61 */
|
cannam@154
|
62
|
cannam@154
|
63 /**
|
cannam@154
|
64 * @def PUSH(stack, size, type)
|
cannam@154
|
65 *
|
cannam@154
|
66 * Allocates 'size' elements of type 'type' on the stack
|
cannam@154
|
67 *
|
cannam@154
|
68 * @param stack Stack
|
cannam@154
|
69 * @param size Number of elements
|
cannam@154
|
70 * @param type Type of element
|
cannam@154
|
71 */
|
cannam@154
|
72
|
cannam@154
|
73 /**
|
cannam@154
|
74 * @def VARDECL(var)
|
cannam@154
|
75 *
|
cannam@154
|
76 * Declare variable on stack
|
cannam@154
|
77 *
|
cannam@154
|
78 * @param var Variable to declare
|
cannam@154
|
79 */
|
cannam@154
|
80
|
cannam@154
|
81 /**
|
cannam@154
|
82 * @def ALLOC(var, size, type)
|
cannam@154
|
83 *
|
cannam@154
|
84 * Allocate 'size' elements of 'type' on stack
|
cannam@154
|
85 *
|
cannam@154
|
86 * @param var Name of variable to allocate
|
cannam@154
|
87 * @param size Number of elements
|
cannam@154
|
88 * @param type Type of element
|
cannam@154
|
89 */
|
cannam@154
|
90
|
cannam@154
|
91 #if defined(VAR_ARRAYS)
|
cannam@154
|
92
|
cannam@154
|
93 #define VARDECL(type, var)
|
cannam@154
|
94 #define ALLOC(var, size, type) type var[size]
|
cannam@154
|
95 #define SAVE_STACK
|
cannam@154
|
96 #define RESTORE_STACK
|
cannam@154
|
97 #define ALLOC_STACK
|
cannam@154
|
98 /* C99 does not allow VLAs of size zero */
|
cannam@154
|
99 #define ALLOC_NONE 1
|
cannam@154
|
100
|
cannam@154
|
101 #elif defined(USE_ALLOCA)
|
cannam@154
|
102
|
cannam@154
|
103 #define VARDECL(type, var) type *var
|
cannam@154
|
104
|
cannam@154
|
105 # ifdef WIN32
|
cannam@154
|
106 # define ALLOC(var, size, type) var = ((type*)_alloca(sizeof(type)*(size)))
|
cannam@154
|
107 # else
|
cannam@154
|
108 # define ALLOC(var, size, type) var = ((type*)alloca(sizeof(type)*(size)))
|
cannam@154
|
109 # endif
|
cannam@154
|
110
|
cannam@154
|
111 #define SAVE_STACK
|
cannam@154
|
112 #define RESTORE_STACK
|
cannam@154
|
113 #define ALLOC_STACK
|
cannam@154
|
114 #define ALLOC_NONE 0
|
cannam@154
|
115
|
cannam@154
|
116 #else
|
cannam@154
|
117
|
cannam@154
|
118 #ifdef CELT_C
|
cannam@154
|
119 char *scratch_ptr=0;
|
cannam@154
|
120 char *global_stack=0;
|
cannam@154
|
121 #else
|
cannam@154
|
122 extern char *global_stack;
|
cannam@154
|
123 extern char *scratch_ptr;
|
cannam@154
|
124 #endif /* CELT_C */
|
cannam@154
|
125
|
cannam@154
|
126 #ifdef ENABLE_VALGRIND
|
cannam@154
|
127
|
cannam@154
|
128 #include <valgrind/memcheck.h>
|
cannam@154
|
129
|
cannam@154
|
130 #ifdef CELT_C
|
cannam@154
|
131 char *global_stack_top=0;
|
cannam@154
|
132 #else
|
cannam@154
|
133 extern char *global_stack_top;
|
cannam@154
|
134 #endif /* CELT_C */
|
cannam@154
|
135
|
cannam@154
|
136 #define ALIGN(stack, size) ((stack) += ((size) - (long)(stack)) & ((size) - 1))
|
cannam@154
|
137 #define PUSH(stack, size, type) (VALGRIND_MAKE_MEM_NOACCESS(stack, global_stack_top-stack),ALIGN((stack),sizeof(type)/sizeof(char)),VALGRIND_MAKE_MEM_UNDEFINED(stack, ((size)*sizeof(type)/sizeof(char))),(stack)+=(2*(size)*sizeof(type)/sizeof(char)),(type*)((stack)-(2*(size)*sizeof(type)/sizeof(char))))
|
cannam@154
|
138 #define RESTORE_STACK ((global_stack = _saved_stack),VALGRIND_MAKE_MEM_NOACCESS(global_stack, global_stack_top-global_stack))
|
cannam@154
|
139 #define ALLOC_STACK char *_saved_stack; ((global_stack = (global_stack==0) ? ((global_stack_top=opus_alloc_scratch(GLOBAL_STACK_SIZE*2)+(GLOBAL_STACK_SIZE*2))-(GLOBAL_STACK_SIZE*2)) : global_stack),VALGRIND_MAKE_MEM_NOACCESS(global_stack, global_stack_top-global_stack)); _saved_stack = global_stack;
|
cannam@154
|
140
|
cannam@154
|
141 #else
|
cannam@154
|
142
|
cannam@154
|
143 #define ALIGN(stack, size) ((stack) += ((size) - (long)(stack)) & ((size) - 1))
|
cannam@154
|
144 #define PUSH(stack, size, type) (ALIGN((stack),sizeof(type)/sizeof(char)),(stack)+=(size)*(sizeof(type)/sizeof(char)),(type*)((stack)-(size)*(sizeof(type)/sizeof(char))))
|
cannam@154
|
145 #if 0 /* Set this to 1 to instrument pseudostack usage */
|
cannam@154
|
146 #define RESTORE_STACK (printf("%ld %s:%d\n", global_stack-scratch_ptr, __FILE__, __LINE__),global_stack = _saved_stack)
|
cannam@154
|
147 #else
|
cannam@154
|
148 #define RESTORE_STACK (global_stack = _saved_stack)
|
cannam@154
|
149 #endif
|
cannam@154
|
150 #define ALLOC_STACK char *_saved_stack; (global_stack = (global_stack==0) ? (scratch_ptr=opus_alloc_scratch(GLOBAL_STACK_SIZE)) : global_stack); _saved_stack = global_stack;
|
cannam@154
|
151
|
cannam@154
|
152 #endif /* ENABLE_VALGRIND */
|
cannam@154
|
153
|
cannam@154
|
154 #include "os_support.h"
|
cannam@154
|
155 #define VARDECL(type, var) type *var
|
cannam@154
|
156 #define ALLOC(var, size, type) var = PUSH(global_stack, size, type)
|
cannam@154
|
157 #define SAVE_STACK char *_saved_stack = global_stack;
|
cannam@154
|
158 #define ALLOC_NONE 0
|
cannam@154
|
159
|
cannam@154
|
160 #endif /* VAR_ARRAYS */
|
cannam@154
|
161
|
cannam@154
|
162
|
cannam@154
|
163 #ifdef ENABLE_VALGRIND
|
cannam@154
|
164
|
cannam@154
|
165 #include <valgrind/memcheck.h>
|
cannam@154
|
166 #define OPUS_CHECK_ARRAY(ptr, len) VALGRIND_CHECK_MEM_IS_DEFINED(ptr, len*sizeof(*ptr))
|
cannam@154
|
167 #define OPUS_CHECK_VALUE(value) VALGRIND_CHECK_VALUE_IS_DEFINED(value)
|
cannam@154
|
168 #define OPUS_CHECK_ARRAY_COND(ptr, len) VALGRIND_CHECK_MEM_IS_DEFINED(ptr, len*sizeof(*ptr))
|
cannam@154
|
169 #define OPUS_CHECK_VALUE_COND(value) VALGRIND_CHECK_VALUE_IS_DEFINED(value)
|
cannam@154
|
170 #define OPUS_PRINT_INT(value) do {fprintf(stderr, #value " = %d at %s:%d\n", value, __FILE__, __LINE__);}while(0)
|
cannam@154
|
171 #define OPUS_FPRINTF fprintf
|
cannam@154
|
172
|
cannam@154
|
173 #else
|
cannam@154
|
174
|
cannam@154
|
175 static OPUS_INLINE int _opus_false(void) {return 0;}
|
cannam@154
|
176 #define OPUS_CHECK_ARRAY(ptr, len) _opus_false()
|
cannam@154
|
177 #define OPUS_CHECK_VALUE(value) _opus_false()
|
cannam@154
|
178 #define OPUS_PRINT_INT(value) do{}while(0)
|
cannam@154
|
179 #define OPUS_FPRINTF (void)
|
cannam@154
|
180
|
cannam@154
|
181 #endif
|
cannam@154
|
182
|
cannam@154
|
183
|
cannam@154
|
184 #endif /* STACK_ALLOC_H */
|