Statistics
| Branch: | Revision:

ffmpeg / libavcodec / mem.c @ ad78c0f3

History | View | Annotate | Download (3.88 KB)

1 d01fe86d Fabrice Bellard
/*
2
 * default memory allocator for libavcodec
3
 * Copyright (c) 2002 Fabrice Bellard.
4
 *
5
 * This library is free software; you can redistribute it and/or
6
 * modify it under the terms of the GNU Lesser General Public
7
 * License as published by the Free Software Foundation; either
8
 * version 2 of the License, or (at your option) any later version.
9
 *
10
 * This library is distributed in the hope that it will be useful,
11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13
 * Lesser General Public License for more details.
14
 *
15
 * You should have received a copy of the GNU Lesser General Public
16
 * License along with this library; if not, write to the Free Software
17 5509bffa Diego Biurrun
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18 d01fe86d Fabrice Bellard
 */
19 115329f1 Diego Biurrun
20 983e3246 Michael Niedermayer
/**
21
 * @file mem.c
22
 * default memory allocator for libavcodec.
23
 */
24 115329f1 Diego Biurrun
25 d01fe86d Fabrice Bellard
#include "avcodec.h"
26 8e1e6f31 Fabrice Bellard
27
/* here we can use OS dependant allocation functions */
28
#undef malloc
29
#undef free
30
#undef realloc
31
32 d01fe86d Fabrice Bellard
#ifdef HAVE_MALLOC_H
33
#include <malloc.h>
34
#endif
35
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
40 115329f1 Diego Biurrun
/**
41 8e1e6f31 Fabrice Bellard
 * Memory allocation of size byte with alignment suitable for all
42
 * memory accesses (including vectors if available on the
43
 * CPU). av_malloc(0) must return a non NULL pointer.
44
 */
45 18f77016 Zdenek Kabelac
void *av_malloc(unsigned int size)
46 d01fe86d Fabrice Bellard
{
47
    void *ptr;
48 88730be6 Måns Rullgård
#ifdef MEMALIGN_HACK
49 ed96aeea Dieter
    long diff;
50 88730be6 Måns Rullgård
#endif
51 0ecca7a4 Michael Niedermayer
52 7d77d5f6 Diego Biurrun
    /* let's disallow possible ambiguous cases */
53 a9493601 Herve W
    if(size > (INT_MAX-16) )
54 0ecca7a4 Michael Niedermayer
        return NULL;
55 115329f1 Diego Biurrun
56 da9b170c Michael Niedermayer
#ifdef MEMALIGN_HACK
57 a9493601 Herve W
    ptr = malloc(size+16);
58
    if(!ptr)
59
        return ptr;
60 ed96aeea Dieter
    diff= ((-(long)ptr - 1)&15) + 1;
61 da9b170c Michael Niedermayer
    ptr += diff;
62
    ((char*)ptr)[-1]= diff;
63 115329f1 Diego Biurrun
#elif defined (HAVE_MEMALIGN)
64 8f2b21a8 Michael Niedermayer
    ptr = memalign(16,size);
65 115329f1 Diego Biurrun
    /* Why 64?
66 d01fe86d Fabrice Bellard
       Indeed, we should align it:
67
         on 4 for 386
68
         on 16 for 486
69 bb270c08 Diego Biurrun
         on 32 for 586, PPro - k6-III
70
         on 64 for K7 (maybe for P3 too).
71 d01fe86d Fabrice Bellard
       Because L1 and L2 caches are aligned on those values.
72
       But I don't want to code such logic here!
73
     */
74 8f2b21a8 Michael Niedermayer
     /* Why 16?
75
        because some cpus need alignment, for example SSE2 on P4, & most RISC cpus
76
        it will just trigger an exception and the unaligned load will be done in the
77
        exception handler or it will just segfault (SSE2 on P4)
78
        Why not larger? because i didnt see a difference in benchmarks ...
79
     */
80
     /* benchmarks with p3
81 bb270c08 Diego Biurrun
        memalign(64)+1          3071,3051,3032
82
        memalign(64)+2          3051,3032,3041
83
        memalign(64)+4          2911,2896,2915
84
        memalign(64)+8          2545,2554,2550
85
        memalign(64)+16         2543,2572,2563
86
        memalign(64)+32         2546,2545,2571
87
        memalign(64)+64         2570,2533,2558
88 115329f1 Diego Biurrun

89 8f2b21a8 Michael Niedermayer
        btw, malloc seems to do 8 byte alignment by default here
90
     */
91 d01fe86d Fabrice Bellard
#else
92
    ptr = malloc(size);
93
#endif
94
    return ptr;
95
}
96
97 b7a22d84 Michael Niedermayer
/**
98 8e1e6f31 Fabrice Bellard
 * av_realloc semantics (same as glibc): if ptr is NULL and size > 0,
99
 * identical to malloc(size). If size is zero, it is identical to
100 115329f1 Diego Biurrun
 * free(ptr) and NULL is returned.
101 b7a22d84 Michael Niedermayer
 */
102 8e1e6f31 Fabrice Bellard
void *av_realloc(void *ptr, unsigned int size)
103
{
104 88730be6 Måns Rullgård
#ifdef MEMALIGN_HACK
105 d5f121e2 Michael Niedermayer
    int diff;
106 88730be6 Måns Rullgård
#endif
107
108 7d77d5f6 Diego Biurrun
    /* let's disallow possible ambiguous cases */
109 a9493601 Herve W
    if(size > (INT_MAX-16) )
110 0ecca7a4 Michael Niedermayer
        return NULL;
111
112 da9b170c Michael Niedermayer
#ifdef MEMALIGN_HACK
113 7d77d5f6 Diego Biurrun
    //FIXME this isn't aligned correctly, though it probably isn't needed
114 c07a22fb Michael Niedermayer
    if(!ptr) return av_malloc(size);
115
    diff= ((char*)ptr)[-1];
116 da9b170c Michael Niedermayer
    return realloc(ptr - diff, size + diff) + diff;
117
#else
118 8e1e6f31 Fabrice Bellard
    return realloc(ptr, size);
119 da9b170c Michael Niedermayer
#endif
120 b7a22d84 Michael Niedermayer
}
121
122 1499e0be Michael Niedermayer
/**
123
 * Free memory which has been allocated with av_malloc(z)() or av_realloc().
124 4f59b684 Diego Biurrun
 * NOTE: ptr = NULL is explicetly allowed
125 1499e0be Michael Niedermayer
 * Note2: it is recommanded that you use av_freep() instead
126
 */
127 d01fe86d Fabrice Bellard
void av_free(void *ptr)
128
{
129
    /* XXX: this test should not be needed on most libcs */
130
    if (ptr)
131 da9b170c Michael Niedermayer
#ifdef MEMALIGN_HACK
132
        free(ptr - ((char*)ptr)[-1]);
133
#else
134 d01fe86d Fabrice Bellard
        free(ptr);
135 da9b170c Michael Niedermayer
#endif
136 d01fe86d Fabrice Bellard
}