Merge commit '502512ea0f179165782777265f441d4061ff6012'
[ffmpeg.git] / libavcodec / arm / dsputil_init_arm.c
1 /*
2  * ARM optimized DSP utils
3  * Copyright (c) 2001 Lionel Ulmer
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/avcodec.h"
28 #include "libavcodec/dsputil.h"
29 #include "dsputil_arm.h"
30
31 void ff_j_rev_dct_arm(int16_t *data);
32 void ff_simple_idct_arm(int16_t *data);
33
34 /* XXX: local hack */
35 static void (*ff_put_pixels_clamped)(const int16_t *block, uint8_t *pixels, int line_size);
36 static void (*ff_add_pixels_clamped)(const int16_t *block, uint8_t *pixels, int line_size);
37
38 void ff_add_pixels_clamped_arm(const int16_t *block, uint8_t *dest,
39                                int line_size);
40
41 /* XXX: those functions should be suppressed ASAP when all IDCTs are
42  * converted */
43 static void j_rev_dct_arm_put(uint8_t *dest, int line_size, int16_t *block)
44 {
45     ff_j_rev_dct_arm(block);
46     ff_put_pixels_clamped(block, dest, line_size);
47 }
48
49 static void j_rev_dct_arm_add(uint8_t *dest, int line_size, int16_t *block)
50 {
51     ff_j_rev_dct_arm(block);
52     ff_add_pixels_clamped(block, dest, line_size);
53 }
54
55 static void simple_idct_arm_put(uint8_t *dest, int line_size, int16_t *block)
56 {
57     ff_simple_idct_arm(block);
58     ff_put_pixels_clamped(block, dest, line_size);
59 }
60
61 static void simple_idct_arm_add(uint8_t *dest, int line_size, int16_t *block)
62 {
63     ff_simple_idct_arm(block);
64     ff_add_pixels_clamped(block, dest, line_size);
65 }
66
67 av_cold void ff_dsputil_init_arm(DSPContext *c, AVCodecContext *avctx,
68                                  unsigned high_bit_depth)
69 {
70     int cpu_flags = av_get_cpu_flags();
71
72     ff_put_pixels_clamped = c->put_pixels_clamped;
73     ff_add_pixels_clamped = c->add_pixels_clamped;
74
75     if (!avctx->lowres && !high_bit_depth) {
76         if (avctx->idct_algo == FF_IDCT_AUTO ||
77             avctx->idct_algo == FF_IDCT_ARM) {
78             c->idct_put              = j_rev_dct_arm_put;
79             c->idct_add              = j_rev_dct_arm_add;
80             c->idct                  = ff_j_rev_dct_arm;
81             c->idct_permutation_type = FF_LIBMPEG2_IDCT_PERM;
82         } else if (avctx->idct_algo == FF_IDCT_SIMPLEARM) {
83             c->idct_put              = simple_idct_arm_put;
84             c->idct_add              = simple_idct_arm_add;
85             c->idct                  = ff_simple_idct_arm;
86             c->idct_permutation_type = FF_NO_IDCT_PERM;
87         }
88     }
89
90     c->add_pixels_clamped = ff_add_pixels_clamped_arm;
91
92     if (have_armv5te(cpu_flags))
93         ff_dsputil_init_armv5te(c, avctx, high_bit_depth);
94     if (have_armv6(cpu_flags))
95         ff_dsputil_init_armv6(c, avctx, high_bit_depth);
96     if (have_neon(cpu_flags))
97         ff_dsputil_init_neon(c, avctx, high_bit_depth);
98 }