chromium/third_party/ffmpeg/libavutil/tx.c

/*
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "avassert.h"
#include "intmath.h"
#include "cpu.h"
#include "mem.h"
#include "qsort.h"
#include "bprint.h"

#include "tx_priv.h"

#define TYPE_IS(type, x) …

/* Calculates the modular multiplicative inverse */
static av_always_inline int mulinv(int n, int m)
{ … }

int ff_tx_gen_pfa_input_map(AVTXContext *s, FFTXCodeletOptions *opts,
                            int d1, int d2)
{ … }

/* Guaranteed to work for any n, m where gcd(n, m) == 1 */
int ff_tx_gen_compound_mapping(AVTXContext *s, FFTXCodeletOptions *opts,
                               int inv, int n, int m)
{ … }

static inline int split_radix_permutation(int i, int len, int inv)
{ … }

int ff_tx_gen_ptwo_revtab(AVTXContext *s, FFTXCodeletOptions *opts)
{ … }

int ff_tx_gen_inplace_map(AVTXContext *s, int len)
{ … }

static void parity_revtab_generator(int *revtab, int n, int inv, int offset,
                                    int is_dual, int dual_high, int len,
                                    int basis, int dual_stride, int inv_lookup)
{ … }

int ff_tx_gen_split_radix_parity_revtab(AVTXContext *s, int len, int inv,
                                        FFTXCodeletOptions *opts,
                                        int basis, int dual_stride)
{ … }

static void reset_ctx(AVTXContext *s, int free_sub)
{ … }

void ff_tx_clear_ctx(AVTXContext *s)
{ … }

av_cold void av_tx_uninit(AVTXContext **ctx)
{ … }

static av_cold int ff_tx_null_init(AVTXContext *s, const FFTXCodelet *cd,
                                   uint64_t flags, FFTXCodeletOptions *opts,
                                   int len, int inv, const void *scale)
{ … }

/* Null transform when the length is 1 */
static void ff_tx_null(AVTXContext *s, void *_out, void *_in, ptrdiff_t stride)
{ … }

static const FFTXCodelet ff_tx_null_def = …;

static const FFTXCodelet * const ff_tx_null_list[] = …;

/* Array of all compiled codelet lists. Order is irrelevant. */
static const FFTXCodelet * const * const codelet_list[] = …;
static const int codelet_list_num = …;

static const int cpu_slow_mask = …;

static const int cpu_slow_penalties[][2] = …;

static int get_codelet_prio(const FFTXCodelet *cd, int cpu_flags, int len)
{ … }

FFTXLenDecomp;

static int cmp_decomp(FFTXLenDecomp *a, FFTXLenDecomp *b)
{ … }

int ff_tx_decompose_length(int dst[TX_MAX_DECOMPOSITIONS], enum AVTXType type,
                           int len, int inv)
{ … }

int ff_tx_gen_default_map(AVTXContext *s, FFTXCodeletOptions *opts)
{ … }

#if !CONFIG_SMALL
static void print_flags(AVBPrint *bp, uint64_t f)
{ … }

static void print_type(AVBPrint *bp, enum AVTXType type)
{ … }

static void print_cd_info(const FFTXCodelet *cd, int prio, int len, int print_prio,
                          int log_level)
{ … }

static void print_tx_structure(AVTXContext *s, int depth)
{ … }
#endif /* CONFIG_SMALL */

TXCodeletMatch;

static int cmp_matches(TXCodeletMatch *a, TXCodeletMatch *b)
{ … }

/* We want all factors to completely cover the length */
static inline int check_cd_factors(const FFTXCodelet *cd, int len)
{ … }

av_cold int ff_tx_init_subtx(AVTXContext *s, enum AVTXType type,
                             uint64_t flags, FFTXCodeletOptions *opts,
                             int len, int inv, const void *scale)
{ … }

av_cold int av_tx_init(AVTXContext **ctx, av_tx_fn *tx, enum AVTXType type,
                       int inv, int len, const void *scale, uint64_t flags)
{ … }