Cycles: Code cleanup, spaces around keywords
[blender-staging.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_aligned_malloc.h"
18 #include "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 #endif
48 #ifdef _WIN32
49         return _aligned_malloc(size, alignment);
50 #elif defined(__APPLE__)
51         /* On Mac OS X, both the heap and the stack are guaranteed 16-byte aligned so
52          * they work natively with SSE types with no further work.
53          */
54         assert(alignment == 16);
55         return malloc(size);
56 #elif defined(__FreeBSD__) || defined(__NetBSD__)
57         void *result;
58         if(posix_memalign(&result, alignment, size)) {
59                 /* Non-zero means allocation error
60                  * either no allocation or bad alignment value.
61                  */
62                 return NULL;
63         }
64         return result;
65 #else  /* This is for Linux. */
66         return memalign(alignment, size);
67 #endif
68 }
69
70 void util_aligned_free(void *ptr)
71 {
72 #if defined(WITH_BLENDER_GUARDEDALLOC)
73         if(ptr != NULL) {
74                 MEM_freeN(ptr);
75         }
76 #elif defined(_WIN32)
77         _aligned_free(ptr);
78 #else
79         free(ptr);
80 #endif
81 }
82
83 CCL_NAMESPACE_END