vda: Check the correct pointer for buffer allocation
[ffmpeg.git] / libavcodec / ppc / huffyuvdsp_altivec.c
1 /*
2  * Copyright (c) 2002 Brian Foley
3  * Copyright (c) 2002 Dieter Shirley
4  * Copyright (c) 2003-2004 Romain Dolbeau <romain@dolbeau.org>
5  *
6  * This file is part of Libav.
7  *
8  * Libav is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * Libav is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #include "config.h"
24 #if HAVE_ALTIVEC_H
25 #include <altivec.h>
26 #endif
27
28 #include "libavutil/attributes.h"
29 #include "libavutil/cpu.h"
30 #include "libavutil/ppc/cpu.h"
31 #include "libavutil/ppc/types_altivec.h"
32 #include "libavutil/ppc/util_altivec.h"
33 #include "libavcodec/huffyuvdsp.h"
34
35 #if HAVE_ALTIVEC && HAVE_BIGENDIAN
36 static void add_bytes_altivec(uint8_t *dst, uint8_t *src, int w)
37 {
38     register int i;
39     register vector unsigned char vdst, vsrc;
40
41     /* dst and src are 16 bytes-aligned (guaranteed). */
42     for (i = 0; i + 15 < w; i += 16) {
43         vdst = vec_ld(i, (unsigned char *) dst);
44         vsrc = vec_ld(i, (unsigned char *) src);
45         vdst = vec_add(vsrc, vdst);
46         vec_st(vdst, i, (unsigned char *) dst);
47     }
48     /* If w is not a multiple of 16. */
49     for (; i < w; i++)
50         dst[i] = src[i];
51 }
52 #endif /* HAVE_ALTIVEC */
53
54 av_cold void ff_huffyuvdsp_init_ppc(HuffYUVDSPContext *c)
55 {
56 #if HAVE_ALTIVEC && HAVE_BIGENDIAN
57     if (!PPC_ALTIVEC(av_get_cpu_flags()))
58         return;
59
60     c->add_bytes = add_bytes_altivec;
61 #endif /* HAVE_ALTIVEC */
62 }