1/*
2 * ARM-optimized IDCT functions
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/idctdsp.h"
29#include "idctdsp_arm.h"
30
31void ff_j_rev_dct_arm(int16_t *data);
32void ff_simple_idct_arm(int16_t *data);
33
34/* XXX: local hack */
35static void (*ff_put_pixels_clamped)(const int16_t *block, uint8_t *pixels, int line_size);
36static void (*ff_add_pixels_clamped)(const int16_t *block, uint8_t *pixels, int line_size);
37
38void 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 */
43static 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
49static 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
55static 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
61static 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
67av_cold void ff_idctdsp_init_arm(IDCTDSPContext *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_idctdsp_init_armv5te(c, avctx, high_bit_depth);
94    if (have_armv6(cpu_flags))
95        ff_idctdsp_init_armv6(c, avctx, high_bit_depth);
96    if (have_neon(cpu_flags))
97        ff_idctdsp_init_neon(c, avctx, high_bit_depth);
98}
99