inv_txfm.h 3.77 KB
Newer Older
1
/*
Yaowu Xu's avatar
Yaowu Xu committed
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
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.
10
11
 */

Yaowu Xu's avatar
Yaowu Xu committed
12
13
#ifndef AOM_DSP_INV_TXFM_H_
#define AOM_DSP_INV_TXFM_H_
14
15
16

#include <assert.h>

Yaowu Xu's avatar
Yaowu Xu committed
17
#include "./aom_config.h"
18
#include "aom_dsp/txfm_common.h"
19
#include "av1/common/odintrin.h"
20
#include "aom_ports/mem.h"
21
22
23
24
25

#ifdef __cplusplus
extern "C" {
#endif

26
static INLINE tran_high_t dct_const_round_shift(tran_high_t input) {
Sebastien Alaiwan's avatar
Sebastien Alaiwan committed
27
  return ROUND_POWER_OF_TWO(input, DCT_CONST_BITS);
28
29
30
}

static INLINE tran_high_t check_range(tran_high_t input, int bd) {
31
32
#if CONFIG_DAALA_TX
  // Daala TX coeffs cover a different range from AV1 TX
33
34
  // all depths: 18 bit integer
  const int32_t int_max = (1 << (TX_COEFF_DEPTH + 6)) - 1;
35
  const int32_t int_min = -int_max - 1;
36
  (void)bd;
37
38
#else
  // AV1 TX case
39
40
41
  // - 8 bit: signed 16 bit integer
  // - 10 bit: signed 18 bit integer
  // - 12 bit: signed 20 bit integer
42
43
44
45
  const int32_t int_max = (1 << (7 + bd)) - 1;
  const int32_t int_min = -int_max - 1;
#endif
#if CONFIG_COEFFICIENT_RANGE_CHECKING
46
47
48
  assert(int_min <= input);
  assert(input <= int_max);
#endif  // CONFIG_COEFFICIENT_RANGE_CHECKING
49
  return clamp64(input, int_min, int_max);
50
51
}

52
53
#define WRAPLOW(x) ((int32_t)check_range(x, 8))
#define HIGHBD_WRAPLOW(x, bd) ((int32_t)check_range((x), bd))
54

55
56
#if CONFIG_MRC_TX
// These each perform dct but add coefficients based on a mask
Sarah Parker's avatar
Sarah Parker committed
57
58
59
void aom_imrc32x32_1_add_c(const tran_low_t *input, uint8_t *dest, int stride,
                           uint8_t *mask);

60
void aom_imrc32x32_1024_add_c(const tran_low_t *input, uint8_t *dest,
Sarah Parker's avatar
Sarah Parker committed
61
                              int stride, uint8_t *mask);
62
63

void aom_imrc32x32_135_add_c(const tran_low_t *input, uint8_t *dest, int stride,
Sarah Parker's avatar
Sarah Parker committed
64
                             uint8_t *mask);
65
66

void aom_imrc32x32_34_add_c(const tran_low_t *input, uint8_t *dest, int stride,
Sarah Parker's avatar
Sarah Parker committed
67
                            uint8_t *mask);
68
69
#endif  // CONFIG_MRC_TX

Luca Barbato's avatar
Luca Barbato committed
70
71
72
73
void aom_idct4_c(const tran_low_t *input, tran_low_t *output);
void aom_idct8_c(const tran_low_t *input, tran_low_t *output);
void aom_idct16_c(const tran_low_t *input, tran_low_t *output);
void aom_idct32_c(const tran_low_t *input, tran_low_t *output);
74
#if CONFIG_TX64X64 && CONFIG_DAALA_TX64
75
76
void aom_idct64_c(const tran_low_t *input, tran_low_t *output);
#endif
Luca Barbato's avatar
Luca Barbato committed
77
78
79
void aom_iadst4_c(const tran_low_t *input, tran_low_t *output);
void aom_iadst8_c(const tran_low_t *input, tran_low_t *output);
void aom_iadst16_c(const tran_low_t *input, tran_low_t *output);
80

Yaowu Xu's avatar
Yaowu Xu committed
81
82
83
84
void aom_highbd_idct4_c(const tran_low_t *input, tran_low_t *output, int bd);
void aom_highbd_idct8_c(const tran_low_t *input, tran_low_t *output, int bd);
void aom_highbd_idct16_c(const tran_low_t *input, tran_low_t *output, int bd);
void aom_highbd_idct32_c(const tran_low_t *input, tran_low_t *output, int bd);
85

Yaowu Xu's avatar
Yaowu Xu committed
86
87
88
void aom_highbd_iadst4_c(const tran_low_t *input, tran_low_t *output, int bd);
void aom_highbd_iadst8_c(const tran_low_t *input, tran_low_t *output, int bd);
void aom_highbd_iadst16_c(const tran_low_t *input, tran_low_t *output, int bd);
89
90
91

static INLINE uint16_t highbd_clip_pixel_add(uint16_t dest, tran_high_t trans,
                                             int bd) {
92
  trans = HIGHBD_WRAPLOW(trans, bd);
Yaowu Xu's avatar
Yaowu Xu committed
93
  return clip_pixel_highbd(dest + (int)trans, bd);
94
95
96
}

static INLINE uint8_t clip_pixel_add(uint8_t dest, tran_high_t trans) {
97
  trans = WRAPLOW(trans);
Yaowu Xu's avatar
Yaowu Xu committed
98
  return clip_pixel(dest + (int)trans);
99
100
101
102
103
}
#ifdef __cplusplus
}  // extern "C"
#endif

Yaowu Xu's avatar
Yaowu Xu committed
104
#endif  // AOM_DSP_INV_TXFM_H_