avidec: avoid infinite loop due to negative ast->sample_size
[ffmpeg.git] / libavcodec / arm / h264chroma_init_arm.c
1 /*
2  * ARM NEON optimised H.264 chroma functions
3  * Copyright (c) 2008 Mans Rullgard <mans@mansr.com>
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 #include <stdint.h>
23
24 #include "libavutil/attributes.h"
25 #include "libavutil/cpu.h"
26 #include "libavutil/arm/cpu.h"
27 #include "libavcodec/h264chroma.h"
28
29 void ff_put_h264_chroma_mc8_neon(uint8_t *, uint8_t *, int, int, int, int);
30 void ff_put_h264_chroma_mc4_neon(uint8_t *, uint8_t *, int, int, int, int);
31 void ff_put_h264_chroma_mc2_neon(uint8_t *, uint8_t *, int, int, int, int);
32
33 void ff_avg_h264_chroma_mc8_neon(uint8_t *, uint8_t *, int, int, int, int);
34 void ff_avg_h264_chroma_mc4_neon(uint8_t *, uint8_t *, int, int, int, int);
35 void ff_avg_h264_chroma_mc2_neon(uint8_t *, uint8_t *, int, int, int, int);
36
37 av_cold void ff_h264chroma_init_arm(H264ChromaContext *c, int bit_depth)
38 {
39     const int high_bit_depth = bit_depth > 8;
40     int cpu_flags = av_get_cpu_flags();
41
42     if (have_neon(cpu_flags) && !high_bit_depth) {
43         c->put_h264_chroma_pixels_tab[0] = ff_put_h264_chroma_mc8_neon;
44         c->put_h264_chroma_pixels_tab[1] = ff_put_h264_chroma_mc4_neon;
45         c->put_h264_chroma_pixels_tab[2] = ff_put_h264_chroma_mc2_neon;
46
47         c->avg_h264_chroma_pixels_tab[0] = ff_avg_h264_chroma_mc8_neon;
48         c->avg_h264_chroma_pixels_tab[1] = ff_avg_h264_chroma_mc4_neon;
49         c->avg_h264_chroma_pixels_tab[2] = ff_avg_h264_chroma_mc2_neon;
50     }
51 }