encodemb.h 3.69 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_ENCODER_ENCODEMB_H_
#define AV1_ENCODER_ENCODEMB_H_
Jingning Han's avatar
Jingning Han committed
14

Yaowu Xu's avatar
Yaowu Xu committed
15
#include "./aom_config.h"
16
#include "av1/common/onyxc_int.h"
17
#include "av1/encoder/block.h"
Jingning Han's avatar
Jingning Han committed
18 19 20 21 22

#ifdef __cplusplus
extern "C" {
#endif

23 24 25 26 27
struct optimize_ctx {
  ENTROPY_CONTEXT ta[MAX_MB_PLANE][2 * MAX_MIB_SIZE];
  ENTROPY_CONTEXT tl[MAX_MB_PLANE][2 * MAX_MIB_SIZE];
};

Jingning Han's avatar
Jingning Han committed
28
struct encode_b_args {
Angie Chiang's avatar
Angie Chiang committed
29
  AV1_COMMON *cm;
Jingning Han's avatar
Jingning Han committed
30 31 32
  MACROBLOCK *x;
  struct optimize_ctx *ctx;
  int8_t *skip;
33 34
  ENTROPY_CONTEXT *ta;
  ENTROPY_CONTEXT *tl;
35
  int8_t enable_optimize_b;
Jingning Han's avatar
Jingning Han committed
36
};
Angie Chiang's avatar
Angie Chiang committed
37

Yaowu Xu's avatar
Yaowu Xu committed
38 39 40 41
typedef enum AV1_XFORM_QUANT {
  AV1_XFORM_QUANT_FP = 0,
  AV1_XFORM_QUANT_B = 1,
  AV1_XFORM_QUANT_DC = 2,
42 43
  AV1_XFORM_QUANT_SKIP_QUANT,
  AV1_XFORM_QUANT_TYPES,
Yaowu Xu's avatar
Yaowu Xu committed
44
} AV1_XFORM_QUANT;
Angie Chiang's avatar
Angie Chiang committed
45

46 47
void av1_encode_sb(AV1_COMMON *cm, MACROBLOCK *x, BLOCK_SIZE bsize, int mi_row,
                   int mi_col);
48
#if CONFIG_SUPERTX
Angie Chiang's avatar
Angie Chiang committed
49
void av1_encode_sb_supertx(AV1_COMMON *cm, MACROBLOCK *x, BLOCK_SIZE bsize);
50
#endif  // CONFIG_SUPERTX
Angie Chiang's avatar
Angie Chiang committed
51 52 53
void av1_encode_sby_pass1(AV1_COMMON *cm, MACROBLOCK *x, BLOCK_SIZE bsize);
void av1_xform_quant(const AV1_COMMON *cm, MACROBLOCK *x, int plane, int block,
                     int blk_row, int blk_col, BLOCK_SIZE plane_bsize,
54
                     TX_SIZE tx_size, int ctx, AV1_XFORM_QUANT xform_quant_idx);
Jingning Han's avatar
Jingning Han committed
55

Angie Chiang's avatar
Angie Chiang committed
56 57
int av1_optimize_b(const AV1_COMMON *cm, MACROBLOCK *mb, int plane, int block,
                   TX_SIZE tx_size, int ctx);
58

59 60 61
void av1_subtract_txb(MACROBLOCK *x, int plane, BLOCK_SIZE plane_bsize,
                      int blk_col, int blk_row, TX_SIZE tx_size);

Yaowu Xu's avatar
Yaowu Xu committed
62
void av1_subtract_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
Jingning Han's avatar
Jingning Han committed
63

64
void av1_set_txb_context(MACROBLOCK *x, int plane, int block, TX_SIZE tx_size,
65 66
                         ENTROPY_CONTEXT *a, ENTROPY_CONTEXT *l);

Yaowu Xu's avatar
Yaowu Xu committed
67 68
void av1_encode_block_intra(int plane, int block, int blk_row, int blk_col,
                            BLOCK_SIZE plane_bsize, TX_SIZE tx_size, void *arg);
Jingning Han's avatar
Jingning Han committed
69

Angie Chiang's avatar
Angie Chiang committed
70 71
void av1_encode_intra_block_plane(AV1_COMMON *cm, MACROBLOCK *x,
                                  BLOCK_SIZE bsize, int plane,
72 73
                                  int enable_optimize_b, int mi_row,
                                  int mi_col);
Jingning Han's avatar
Jingning Han committed
74

75
#if CONFIG_PVQ
Thomas Daede's avatar
Thomas Daede committed
76 77 78 79 80 81
PVQ_SKIP_TYPE av1_pvq_encode_helper(MACROBLOCK *x, tran_low_t *const coeff,
                                    tran_low_t *ref_coeff,
                                    tran_low_t *const dqcoeff, uint16_t *eob,
                                    const int16_t *quant, int plane,
                                    int tx_size, TX_TYPE tx_type, int *rate,
                                    int speed, PVQ_INFO *pvq_info);
82

83 84 85
void av1_store_pvq_enc_info(PVQ_INFO *pvq_info, int *qg, int *theta, int *k,
                            od_coeff *y, int nb_bands, const int *off,
                            int *size, int skip_rest, int skip_dir, int bs);
86 87
#endif

Luc Trudeau's avatar
Luc Trudeau committed
88
#if CONFIG_CFL
89 90
void av1_predict_intra_block_encoder_facade(MACROBLOCK *x,
                                            FRAME_CONTEXT *ec_ctx, int plane,
Luc Trudeau's avatar
Luc Trudeau committed
91
                                            int block_idx, int blk_col,
92 93
                                            int blk_row, TX_SIZE tx_size,
                                            BLOCK_SIZE plane_bsize);
Luc Trudeau's avatar
Luc Trudeau committed
94 95
#endif

Jingning Han's avatar
Jingning Han committed
96 97 98 99
#ifdef __cplusplus
}  // extern "C"
#endif

Yaowu Xu's avatar
Yaowu Xu committed
100
#endif  // AV1_ENCODER_ENCODEMB_H_