idct.h 5.14 KB
Newer Older
Jingning Han's avatar
Jingning Han committed
1
/*
Yaowu Xu's avatar
Yaowu Xu committed
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
Jingning Han's avatar
Jingning Han committed
3
 *
Yaowu Xu's avatar
Yaowu Xu committed
4 5 6 7 8 9
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
Jingning Han's avatar
Jingning Han committed
10 11
 */

Yaowu Xu's avatar
Yaowu Xu committed
12 13
#ifndef AV1_COMMON_IDCT_H_
#define AV1_COMMON_IDCT_H_
Jingning Han's avatar
Jingning Han committed
14 15 16

#include <assert.h>

Yaowu Xu's avatar
Yaowu Xu committed
17
#include "./aom_config.h"
18 19 20 21 22 23
#include "av1/common/blockd.h"
#include "av1/common/common.h"
#include "av1/common/enums.h"
#include "aom_dsp/inv_txfm.h"
#include "aom_dsp/txfm_common.h"
#include "aom_ports/mem.h"
Jingning Han's avatar
Jingning Han committed
24 25 26 27 28

#ifdef __cplusplus
extern "C" {
#endif

29 30 31 32 33 34 35 36 37 38
// TODO(kslu) Combine FWD_TXFM_PARAM and INV_TXFM_PARAM into a common struct.
// and move the common stuff in idct.h to av1_txfm.h or txfm_common.h
typedef struct fwd_txfm_param {
  TX_TYPE tx_type;
  TX_SIZE tx_size;
  int lossless;
  int bd;
} FWD_TXFM_PARAM;

typedef struct inv_txfm_param {
39 40 41
#if CONFIG_ADAPT_SCAN
  const int16_t *eob_threshold;
#endif
Angie Chiang's avatar
Angie Chiang committed
42 43 44 45 46 47 48
  TX_TYPE tx_type;
  TX_SIZE tx_size;
  int eob;
  int lossless;
  int bd;
} INV_TXFM_PARAM;

clang-format's avatar
clang-format committed
49
typedef void (*transform_1d)(const tran_low_t *, tran_low_t *);
Jingning Han's avatar
Jingning Han committed
50 51 52 53 54

typedef struct {
  transform_1d cols, rows;  // vertical and horizontal
} transform_2d;

55
#if CONFIG_HIGHBITDEPTH
56
typedef void (*highbd_transform_1d)(const tran_low_t *, tran_low_t *, int bd);
Jingning Han's avatar
Jingning Han committed
57 58 59 60

typedef struct {
  highbd_transform_1d cols, rows;  // vertical and horizontal
} highbd_transform_2d;
61
#endif  // CONFIG_HIGHBITDEPTH
Jingning Han's avatar
Jingning Han committed
62

63
#define MAX_TX_SCALE 1
64
int av1_get_tx_scale(const TX_SIZE tx_size);
65

Yaowu Xu's avatar
Yaowu Xu committed
66 67 68 69
void av1_iwht4x4_add(const tran_low_t *input, uint8_t *dest, int stride,
                     int eob);
void av1_idct4x4_add(const tran_low_t *input, uint8_t *dest, int stride,
                     int eob);
Jingning Han's avatar
Jingning Han committed
70

hui su's avatar
hui su committed
71 72
void av1_inv_txfm_add(const tran_low_t *input, uint8_t *dest, int stride,
                      INV_TXFM_PARAM *inv_txfm_param);
Urvang Joshi's avatar
Urvang Joshi committed
73 74 75 76
void av1_inverse_transform_block(const MACROBLOCKD *xd,
                                 const tran_low_t *dqcoeff, TX_TYPE tx_type,
                                 TX_SIZE tx_size, uint8_t *dst, int stride,
                                 int eob);
77 78
void av1_inverse_transform_block_facade(MACROBLOCKD *xd, int plane, int block,
                                        int blk_row, int blk_col, int eob);
79

Yaowu Xu's avatar
Yaowu Xu committed
80 81 82
void av1_highbd_iwht4x4_add(const tran_low_t *input, uint8_t *dest, int stride,
                            int eob, int bd);
void av1_highbd_inv_txfm_add_4x4(const tran_low_t *input, uint8_t *dest,
83
                                 int stride, const INV_TXFM_PARAM *param);
Yaowu Xu's avatar
Yaowu Xu committed
84
void av1_highbd_inv_txfm_add_4x8(const tran_low_t *input, uint8_t *dest,
85
                                 int stride, const INV_TXFM_PARAM *param);
Yaowu Xu's avatar
Yaowu Xu committed
86
void av1_highbd_inv_txfm_add_8x4(const tran_low_t *input, uint8_t *dest,
87
                                 int stride, const INV_TXFM_PARAM *param);
hui su's avatar
hui su committed
88 89
void av1_highbd_inv_txfm_add(const tran_low_t *input, uint8_t *dest, int stride,
                             INV_TXFM_PARAM *inv_txfm_param);
90

hui su's avatar
hui su committed
91 92 93 94 95 96 97 98 99 100 101 102 103 104
#if CONFIG_DPCM_INTRA
void av1_dpcm_inv_txfm_add_4_c(const tran_low_t *input, int stride,
                               TX_TYPE_1D tx_type, uint8_t *dest);
void av1_dpcm_inv_txfm_add_8_c(const tran_low_t *input, int stride,
                               TX_TYPE_1D tx_type, uint8_t *dest);
void av1_dpcm_inv_txfm_add_16_c(const tran_low_t *input, int stride,
                                TX_TYPE_1D tx_type, uint8_t *dest);
void av1_dpcm_inv_txfm_add_32_c(const tran_low_t *input, int stride,
                                TX_TYPE_1D tx_type, uint8_t *dest);
typedef void (*dpcm_inv_txfm_add_func)(const tran_low_t *input, int stride,
                                       TX_TYPE_1D tx_type, uint8_t *dest);
dpcm_inv_txfm_add_func av1_get_dpcm_inv_txfm_add_func(int tx_length);
#if CONFIG_HIGHBITDEPTH
void av1_hbd_dpcm_inv_txfm_add_4_c(const tran_low_t *input, int stride,
105 106
                                   TX_TYPE_1D tx_type, int bd, uint16_t *dest,
                                   int dir);
hui su's avatar
hui su committed
107
void av1_hbd_dpcm_inv_txfm_add_8_c(const tran_low_t *input, int stride,
108 109
                                   TX_TYPE_1D tx_type, int bd, uint16_t *dest,
                                   int dir);
hui su's avatar
hui su committed
110
void av1_hbd_dpcm_inv_txfm_add_16_c(const tran_low_t *input, int stride,
111 112
                                    TX_TYPE_1D tx_type, int bd, uint16_t *dest,
                                    int dir);
hui su's avatar
hui su committed
113
void av1_hbd_dpcm_inv_txfm_add_32_c(const tran_low_t *input, int stride,
114 115
                                    TX_TYPE_1D tx_type, int bd, uint16_t *dest,
                                    int dir);
hui su's avatar
hui su committed
116 117
typedef void (*hbd_dpcm_inv_txfm_add_func)(const tran_low_t *input, int stride,
                                           TX_TYPE_1D tx_type, int bd,
118
                                           uint16_t *dest, int dir);
hui su's avatar
hui su committed
119 120 121
hbd_dpcm_inv_txfm_add_func av1_get_hbd_dpcm_inv_txfm_add_func(int tx_length);
#endif  // CONFIG_HIGHBITDEPTH
#endif  // CONFIG_DPCM_INTRA
Jingning Han's avatar
Jingning Han committed
122 123 124 125
#ifdef __cplusplus
}  // extern "C"
#endif

Yaowu Xu's avatar
Yaowu Xu committed
126
#endif  // AV1_COMMON_IDCT_H_