libs/pbd/malign.cc

   1 /*
   2     Copyright (C) 2012 Paul Davis
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include "libpbd-config.h"
  21
  22 #include <cstring>
  23 #include <cerrno>
  24
  25 #include "pbd/malign.h"
  26 #include "pbd/error.h"
  27
  28 #include "i18n.h"
  29
  30 using namespace PBD;
  31
  32 #if ( defined(__x86_64__) || defined(_M_X64) )
  33 static const int CPU_CACHE_ALIGN = 64;
  34 #else
  35 static const int CPU_CACHE_ALIGN = 16; /* arguably 32 on most arches, but it matters less */
  36 #endif
  37
  38 int cache_aligned_malloc (void** memptr, size_t size)
  39 {
  40 #ifndef HAVE_POSIX_MEMALIGN
  41 #ifdef PLATFORM_WINDOWS
  42         if (((*memptr) = _aligned_malloc (size, CPU_CACHE_ALIGN)) == 0) {
  43                 fatal << string_compose (_("Memory allocation error: malloc (%1 * %2) failed (%3)"),
  44                                          CPU_CACHE_ALIGN, size, strerror (errno)) << endmsg;
  45                 return errno;
  46         } else {
  47                 return 0;
  48         }
  49 #else
  50         std::string << string_compose (_("Memory allocation error: malloc (%1 * %2)"),
  51                                          CPU_CACHE_ALIGN, size) << endmsg;
  52         if (((*memptr) = malloc (size)) == 0) {
  53                 fatal << string_compose (_("Memory allocation error: malloc (%1 * %2) failed (%3)"),
  54                                          CPU_CACHE_ALIGN, size, strerror (errno)) << endmsg;
  55                 return errno;
  56         } else {
  57                 return 0;
  58         }
  59 #endif
  60 #else
  61         if (posix_memalign (memptr, CPU_CACHE_ALIGN, size)) {
  62                 fatal << string_compose (_("Memory allocation error: posix_memalign (%1 * %2) failed (%3)"),
  63                                          CPU_CACHE_ALIGN, size, strerror (errno)) << endmsg;
  64         }
  65
  66         return 0;
  67 #endif
  68 }
  69
  70 void cache_aligned_free (void* memptr)
  71 {
  72 #ifdef PLATFORM_WINDOWS
  73         _aligned_free (memptr);
  74 #else
  75         free (memptr);
  76 #endif
  77 }
  78
  79 int  aligned_malloc (void** memptr, size_t size, size_t alignment)
  80 {
  81 #ifndef HAVE_POSIX_MEMALIGN
  82 #ifdef PLATFORM_WINDOWS
  83         if (((*memptr) = _aligned_malloc (size, alignment)) == 0) {
  84                 fatal << string_compose (_("Memory allocation error: malloc (%1 * %2) failed (%3)"),
  85                                          alignment, size, strerror (errno)) << endmsg;
  86                 return errno;
  87         } else {
  88                 return 0;
  89         }
  90 #else
  91         if (((*memptr) = malloc (size)) == 0) {
  92                 fatal << string_compose (_("Memory allocation error: malloc (%1 * %2) failed (%3)"),
  93                                          alignment, size, strerror (errno)) << endmsg;
  94                 return errno;
  95         } else {
  96                 return 0;
  97         }
  98 #endif
  99 #else
 100         if (posix_memalign (memptr, alignment, size)) {
 101                 fatal << string_compose (_("Memory allocation error: posix_memalign (%1 * %2) failed (%3)"),
 102                                          alignment, size, strerror (errno)) << endmsg;
 103         }
 104
 105         return 0;
 106 #endif
 107 }
 108
 109 void aligned_free (void* memptr)
 110 {
 111 #ifdef PLATFORM_WINDOWS
 112         _aligned_free (memptr);
 113 #else
 114         free (memptr);
 115 #endif
 116 }