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 "pbd/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         if (((*memptr) = malloc (size)) == 0) {
  51                 fatal << string_compose (_("Memory allocation error: malloc (%1 * %2) failed (%3)"),
  52                                          CPU_CACHE_ALIGN, size, strerror (errno)) << endmsg;
  53                 return errno;
  54         } else {
  55                 return 0;
  56         }
  57 #endif
  58 #else
  59         if (posix_memalign (memptr, CPU_CACHE_ALIGN, size)) {
  60                 fatal << string_compose (_("Memory allocation error: posix_memalign (%1 * %2) failed (%3)"),
  61                                          CPU_CACHE_ALIGN, size, strerror (errno)) << endmsg;
  62         }
  63
  64         return 0;
  65 #endif
  66 }
  67
  68 void cache_aligned_free (void* memptr)
  69 {
  70 #ifdef PLATFORM_WINDOWS
  71         _aligned_free (memptr);
  72 #else
  73         free (memptr);
  74 #endif
  75 }
  76
  77 int  aligned_malloc (void** memptr, size_t size, size_t alignment)
  78 {
  79 #ifndef HAVE_POSIX_MEMALIGN
  80 #ifdef PLATFORM_WINDOWS
  81         if (((*memptr) = _aligned_malloc (size, alignment)) == 0) {
  82                 fatal << string_compose (_("Memory allocation error: malloc (%1 * %2) failed (%3)"),
  83                                          alignment, size, strerror (errno)) << endmsg;
  84                 return errno;
  85         } else {
  86                 return 0;
  87         }
  88 #else
  89         if (((*memptr) = malloc (size)) == 0) {
  90                 fatal << string_compose (_("Memory allocation error: malloc (%1 * %2) failed (%3)"),
  91                                          alignment, size, strerror (errno)) << endmsg;
  92                 return errno;
  93         } else {
  94                 return 0;
  95         }
  96 #endif
  97 #else
  98         if (posix_memalign (memptr, alignment, size)) {
  99                 fatal << string_compose (_("Memory allocation error: posix_memalign (%1 * %2) failed (%3)"),
 100                                          alignment, size, strerror (errno)) << endmsg;
 101         }
 102
 103         return 0;
 104 #endif
 105 }
 106
 107 void aligned_free (void* memptr)
 108 {
 109 #ifdef PLATFORM_WINDOWS
 110         _aligned_free (memptr);
 111 #else
 112         free (memptr);
 113 #endif
 114 }