mem.c
上传用户:jylinhe
上传日期:2022-07-11
资源大小:334k
文件大小:4k
源码类别:

多媒体编程

开发平台:

Visual C++

  1. /*
  2.  * default memory allocator for libavutil
  3.  * Copyright (c) 2002 Fabrice Bellard.
  4.  *
  5.  * This file is part of FFmpeg.
  6.  *
  7.  * FFmpeg is free software; you can redistribute it and/or
  8.  * modify it under the terms of the GNU Lesser General Public
  9.  * License as published by the Free Software Foundation; either
  10.  * version 2.1 of the License, or (at your option) any later version.
  11.  *
  12.  * FFmpeg is distributed in the hope that it will be useful,
  13.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15.  * Lesser General Public License for more details.
  16.  *
  17.  * You should have received a copy of the GNU Lesser General Public
  18.  * License along with FFmpeg; if not, write to the Free Software
  19.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20.  */
  21. /**
  22.  * @file mem.c
  23.  * default memory allocator for libavutil.
  24.  */
  25. #include "dsputil.h"
  26. //#include "internal.h"
  27. //#include "integer.h"
  28. /* here we can use OS dependent allocation functions */
  29. #undef malloc
  30. #undef free
  31. #undef realloc
  32. #include <stdlib.h>
  33. #ifdef HAVE_MALLOC_H
  34. #include <malloc.h>
  35. #endif
  36. /* you can redefine av_malloc and av_free in your project to use your
  37.    memory allocator. You do not need to suppress this file because the
  38.    linker will do it automatically */
  39. void *av_malloc(unsigned int size)
  40. {
  41.     void *ptr = NULL;
  42. #ifdef CONFIG_MEMALIGN_HACK
  43.     long diff;
  44. #endif
  45.     /* let's disallow possible ambiguous cases */
  46.     if(size > (INT_MAX-16) )
  47.         return NULL;
  48. #ifdef CONFIG_MEMALIGN_HACK
  49.     ptr = malloc(size+16);
  50.     if(!ptr)
  51.         return ptr;
  52.     diff= ((-(long)ptr - 1)&15) + 1;
  53.     ptr = (char*)ptr + diff;
  54.     ((char*)ptr)[-1]= diff;
  55. #elif defined (HAVE_POSIX_MEMALIGN)
  56.     posix_memalign(&ptr,16,size);
  57. #elif defined (HAVE_MEMALIGN)
  58.     ptr = memalign(16,size);
  59.     /* Why 64?
  60.        Indeed, we should align it:
  61.          on 4 for 386
  62.          on 16 for 486
  63.          on 32 for 586, PPro - k6-III
  64.          on 64 for K7 (maybe for P3 too).
  65.        Because L1 and L2 caches are aligned on those values.
  66.        But I don't want to code such logic here!
  67.      */
  68.      /* Why 16?
  69.         Because some CPUs need alignment, for example SSE2 on P4, & most RISC CPUs
  70.         it will just trigger an exception and the unaligned load will be done in the
  71.         exception handler or it will just segfault (SSE2 on P4)
  72.         Why not larger? Because I did not see a difference in benchmarks ...
  73.      */
  74.      /* benchmarks with p3
  75.         memalign(64)+1          3071,3051,3032
  76.         memalign(64)+2          3051,3032,3041
  77.         memalign(64)+4          2911,2896,2915
  78.         memalign(64)+8          2545,2554,2550
  79.         memalign(64)+16         2543,2572,2563
  80.         memalign(64)+32         2546,2545,2571
  81.         memalign(64)+64         2570,2533,2558
  82.         btw, malloc seems to do 8 byte alignment by default here
  83.      */
  84. #else
  85.     ptr = malloc(size);
  86. #endif
  87.     return ptr;
  88. }
  89. void *av_realloc(void *ptr, unsigned int size)
  90. {
  91. #ifdef CONFIG_MEMALIGN_HACK
  92.     int diff;
  93. #endif
  94.     /* let's disallow possible ambiguous cases */
  95.     if(size > (INT_MAX-16) )
  96.         return NULL;
  97. #ifdef CONFIG_MEMALIGN_HACK
  98.     //FIXME this isn't aligned correctly, though it probably isn't needed
  99.     if(!ptr) return av_malloc(size);
  100.     diff= ((char*)ptr)[-1];
  101.     return (char*)realloc((char*)ptr - diff, size + diff) + diff;
  102. #else
  103.     return realloc(ptr, size);
  104. #endif
  105. }
  106. void av_free(void *ptr)
  107. {
  108.     /* XXX: this test should not be needed on most libcs */
  109.     if (ptr)
  110. #ifdef CONFIG_MEMALIGN_HACK
  111.         free((char*)ptr - ((char*)ptr)[-1]);
  112. #else
  113.         free(ptr);
  114. #endif
  115. }
  116. void av_freep(void *arg)
  117. {
  118.     void **ptr= (void**)arg;
  119.     av_free(*ptr);
  120.     *ptr = NULL;
  121. }
  122. void *av_mallocz(unsigned int size)
  123. {
  124.     void *ptr = av_malloc(size);
  125.     if (ptr)
  126.         memset(ptr, 0, size);
  127.     return ptr;
  128. }
  129. char *av_strdup(const char *s)
  130. {
  131.     char *ptr= NULL;
  132.     if(s){
  133.         int len = strlen(s) + 1;
  134.         ptr = av_malloc(len);
  135.         if (ptr)
  136.             memcpy(ptr, s, len);
  137.     }
  138.     return ptr;
  139. }