encodemb.h 2.78 KB
Newer Older
Jingning Han's avatar
Jingning Han committed
1 2 3 4 5 6 7 8 9 10
/*
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

Yaowu Xu's avatar
Yaowu Xu committed
11 12
#ifndef AV1_ENCODER_ENCODEMB_H_
#define AV1_ENCODER_ENCODEMB_H_
Jingning Han's avatar
Jingning Han committed
13

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

#ifdef __cplusplus
extern "C" {
#endif

21 22 23 24 25
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
26 27 28 29
struct encode_b_args {
  MACROBLOCK *x;
  struct optimize_ctx *ctx;
  int8_t *skip;
30 31
  ENTROPY_CONTEXT *ta;
  ENTROPY_CONTEXT *tl;
32
  int8_t enable_optimize_b;
Jingning Han's avatar
Jingning Han committed
33
};
Angie Chiang's avatar
Angie Chiang committed
34

Yaowu Xu's avatar
Yaowu Xu committed
35 36 37 38 39 40 41
typedef enum AV1_XFORM_QUANT {
  AV1_XFORM_QUANT_FP = 0,
  AV1_XFORM_QUANT_B = 1,
  AV1_XFORM_QUANT_DC = 2,
  AV1_XFORM_QUANT_SKIP_QUANT = 3,
  AV1_XFORM_QUANT_LAST = 4
} AV1_XFORM_QUANT;
Angie Chiang's avatar
Angie Chiang committed
42

Yaowu Xu's avatar
Yaowu Xu committed
43
void av1_encode_sb(MACROBLOCK *x, BLOCK_SIZE bsize);
44
#if CONFIG_SUPERTX
Yaowu Xu's avatar
Yaowu Xu committed
45
void av1_encode_sb_supertx(MACROBLOCK *x, BLOCK_SIZE bsize);
46
#endif  // CONFIG_SUPERTX
Yaowu Xu's avatar
Yaowu Xu committed
47 48 49 50
void av1_encode_sby_pass1(MACROBLOCK *x, BLOCK_SIZE bsize);
void av1_xform_quant(MACROBLOCK *x, int plane, int block, int blk_row,
                     int blk_col, BLOCK_SIZE plane_bsize, TX_SIZE tx_size,
                     AV1_XFORM_QUANT xform_quant_idx);
51
#if CONFIG_NEW_QUANT
Yaowu Xu's avatar
Yaowu Xu committed
52 53 54 55 56 57 58 59 60 61 62 63
void av1_xform_quant_nuq(MACROBLOCK *x, int plane, int block, int blk_row,
                         int blk_col, BLOCK_SIZE plane_bsize, TX_SIZE tx_size,
                         int ctx);
void av1_xform_quant_dc_nuq(MACROBLOCK *x, int plane, int block, int blk_row,
                            int blk_col, BLOCK_SIZE plane_bsize,
                            TX_SIZE tx_size, int ctx);
void av1_xform_quant_fp_nuq(MACROBLOCK *x, int plane, int block, int blk_row,
                            int blk_col, BLOCK_SIZE plane_bsize,
                            TX_SIZE tx_size, int ctx);
void av1_xform_quant_dc_fp_nuq(MACROBLOCK *x, int plane, int block, int blk_row,
                               int blk_col, BLOCK_SIZE plane_bsize,
                               TX_SIZE tx_size, int ctx);
64
#endif
Jingning Han's avatar
Jingning Han committed
65

Yaowu Xu's avatar
Yaowu Xu committed
66 67
int av1_optimize_b(MACROBLOCK *mb, int plane, int block, TX_SIZE tx_size,
                   int ctx);
68

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

Yaowu Xu's avatar
Yaowu Xu committed
71 72
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
73

Yaowu Xu's avatar
Yaowu Xu committed
74 75
void av1_encode_intra_block_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane,
                                  int enable_optimize_b);
Jingning Han's avatar
Jingning Han committed
76 77 78 79 80

#ifdef __cplusplus
}  // extern "C"
#endif

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