Cycles: Make all #include statements relative to cycles source directory
[blender.git] / intern / cycles / util / util_aligned_malloc.cpp
1 /*
2  * Copyright 2011-2015 Blender Foundation
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include "util/util_aligned_malloc.h"
18 #include "util/util_guarded_allocator.h"
19
20 #include <cassert>
21
22 /* Adopted from Libmv. */
23
24 #if !defined(__APPLE__) && !defined(__FreeBSD__) && !defined(__NetBSD__)
25 /* Needed for memalign on Linux and _aligned_alloc on Windows. */
26 #  ifdef FREE_WINDOWS
27 /* Make sure _aligned_malloc is included. */
28 #    ifdef __MSVCRT_VERSION__
29 #      undef __MSVCRT_VERSION__
30 #    endif
31 #    define __MSVCRT_VERSION__ 0x0700
32 #  endif  /* FREE_WINDOWS */
33 #  include <malloc.h>
34 #else
35 /* Apple's malloc is 16-byte aligned, and does not have malloc.h, so include
36  * stdilb instead.
37  */
38 #  include <cstdlib>
39 #endif
40
41 CCL_NAMESPACE_BEGIN
42
43 void *util_aligned_malloc(size_t size, int alignment)
44 {
45 #ifdef WITH_BLENDER_GUARDEDALLOC
46         return MEM_mallocN_aligned(size, alignment, "Cycles Aligned Alloc");
47 #elif defined(_WIN32)
48         return _aligned_malloc(size, alignment);
49 #elif defined(__APPLE__)
50         /* On Mac OS X, both the heap and the stack are guaranteed 16-byte aligned so
51          * they work natively with SSE types with no further work.
52          */
53         assert(alignment == 16);
54         return malloc(size);
55 #elif defined(__FreeBSD__) || defined(__NetBSD__)
56         void *result;
57         if(posix_memalign(&result, alignment, size)) {
58                 /* Non-zero means allocation error
59                  * either no allocation or bad alignment value.
60                  */
61                 return NULL;
62         }
63         return result;
64 #else  /* This is for Linux. */
65         return memalign(alignment, size);
66 #endif
67 }
68
69 void util_aligned_free(void *ptr)
70 {
71 #if defined(WITH_BLENDER_GUARDEDALLOC)
72         if(ptr != NULL) {
73                 MEM_freeN(ptr);
74         }
75 #elif defined(_WIN32)
76         _aligned_free(ptr);
77 #else
78         free(ptr);
79 #endif
80 }
81
82 CCL_NAMESPACE_END