summaryrefslogtreecommitdiff
path: root/openEMS/tools/aligned_allocator.h
diff options
context:
space:
mode:
Diffstat (limited to 'openEMS/tools/aligned_allocator.h')
-rw-r--r--openEMS/tools/aligned_allocator.h173
1 files changed, 173 insertions, 0 deletions
diff --git a/openEMS/tools/aligned_allocator.h b/openEMS/tools/aligned_allocator.h
new file mode 100644
index 0000000..efabe7d
--- /dev/null
+++ b/openEMS/tools/aligned_allocator.h
@@ -0,0 +1,173 @@
+/*
+* Copyright (C) 2010 Sebastian Held (sebastian.held@gmx.de)
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+// based on http://blogs.msdn.com/b/vcblog/archive/2008/08/28/the-aligned_allocator.aspx
+// from Stephan T. Lavavej
+
+
+// The following headers are required for all allocators.
+#include <stddef.h> // Required for size_t and ptrdiff_t and NULL
+#include <new> // Required for placement new and std::bad_alloc
+#include <stdexcept> // Required for std::length_error
+
+
+#ifdef WIN32
+#define __MSVCRT_VERSION__ 0x0700
+#include <malloc.h>
+#define MEMALIGN( array, alignment, size ) !(*array = _aligned_malloc( size, alignment ))
+#define FREE( array ) _aligned_free( array )
+#else
+#define MEMALIGN( array, alignment, size ) posix_memalign( array, alignment, size )
+#define FREE( array ) free( array )
+#endif
+
+
+template <typename T> class aligned_allocator
+{
+public:
+ // The following will be the same for virtually all allocators.
+ typedef T * pointer;
+ typedef const T * const_pointer;
+ typedef T& reference;
+ typedef const T& const_reference;
+ typedef T value_type;
+ typedef size_t size_type;
+ typedef ptrdiff_t difference_type;
+
+ T * address(T& r) const
+ {
+ return &r;
+ }
+
+ const T * address(const T& s) const
+ {
+ return &s;
+ }
+
+ size_t max_size() const
+ {
+ // The following has been carefully written to be independent of
+ // the definition of size_t and to avoid signed/unsigned warnings.
+ return (static_cast<size_t>(0) - static_cast<size_t>(1)) / sizeof(T);
+ }
+
+ // The following must be the same for all allocators.
+ template <typename U> struct rebind
+ {
+ typedef aligned_allocator<U> other;
+ };
+
+ bool operator!=(const aligned_allocator& other) const
+ {
+ return !(*this == other);
+ }
+
+ void construct(T * const p, const T& t) const
+ {
+ void * const pv = static_cast<void *>(p);
+ new (pv) T(t);
+ }
+
+ void destroy(T * const p) const; // Defined below.
+
+ // Returns true if and only if storage allocated from *this
+ // can be deallocated from other, and vice versa.
+ // Always returns true for stateless allocators.
+ bool operator==(const aligned_allocator& other) const
+ {
+ return true;
+ }
+
+ // Default constructor, copy constructor, rebinding constructor, and destructor.
+ // Empty for stateless allocators.
+ aligned_allocator() { }
+ aligned_allocator(const aligned_allocator&) { }
+ template <typename U> aligned_allocator(const aligned_allocator<U>&) { }
+ ~aligned_allocator() { }
+
+ // The following will be different for each allocator.
+ T * allocate(const size_t n) const
+ {
+// std::cout << "Allocating " << n << (n == 1 ? " object" : "objects") << " of size " << sizeof(T) << "." << std::endl;
+ // The return value of allocate(0) is unspecified.
+ // aligned_allocator returns NULL in order to avoid depending
+ // on malloc(0)'s implementation-defined behavior
+ // (the implementation can define malloc(0) to return NULL,
+ // in which case the bad_alloc check below would fire).
+ // All allocators can return NULL in this case.
+ if (n == 0)
+ {
+ return NULL;
+ }
+
+ // All allocators should contain an integer overflow check.
+ // The Standardization Committee recommends that std::length_error
+ // be thrown in the case of integer overflow.
+ if (n > max_size())
+ {
+ throw std::length_error("aligned_allocator<T>::allocate() - Integer overflow.");
+ }
+
+ // Allocators should throw std::bad_alloc in the case of memory allocation failure.
+ void * pv;
+ if (MEMALIGN( &pv, 16, n * sizeof(T)))
+ throw std::bad_alloc();
+
+ return static_cast<T *>(pv);
+ }
+
+ void deallocate(T * const p, const size_t n) const
+ {
+// std::cout << "Deallocating " << n << (n == 1 ? " object" : "objects") << " of size " << sizeof(T) << "." << std::endl;
+ // aligned_allocator wraps free().
+ UNUSED(n);
+ FREE(p);
+ }
+
+ // The following will be the same for all allocators that ignore hints.
+ template <typename U> T * allocate(const size_t n, const U * /* const hint */) const
+ {
+ return allocate(n);
+ }
+
+ // Allocators are not required to be assignable, so
+ // all allocators should have a private unimplemented
+ // assignment operator. Note that this will trigger the
+ // off-by-default (enabled under /Wall) warning C4626
+ // "assignment operator could not be generated because a
+ // base class assignment operator is inaccessible" within
+ // the STL headers, but that warning is useless.
+
+private:
+ aligned_allocator& operator=(const aligned_allocator&);
+};
+
+// A compiler bug causes it to believe that p->~T() doesn't reference p.
+#ifdef _MSC_VER
+#pragma warning(push)
+#pragma warning(disable: 4100) // unreferenced formal parameter
+#endif
+
+// The definition of destroy() must be the same for all allocators.
+template <typename T> void aligned_allocator<T>::destroy(T * const p) const
+{
+ p->~T();
+}
+
+#ifdef _MSC_VER
+#pragma warning(pop)
+#endif