#ifndef ENCODER_H_ #define ENCODER_H_ /***************************************************************************** * This file is part of Kvazaar HEVC encoder. * * Copyright (C) 2013-2014 Tampere University of Technology and others (see * COPYING file). * * Kvazaar is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License version 2 as published * by the Free Software Foundation. * * Kvazaar 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 General Public License for more details. * * You should have received a copy of the GNU General Public License * along with Kvazaar. If not, see . ****************************************************************************/ /* * \file * \brief The highest level of the encoder. */ #include "global.h" #include "picture.h" #include "bitstream.h" #include "cabac.h" #include "config.h" typedef struct { int32_t scaling_list_dc [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM]; int32_t* scaling_list_coeff[SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM]; int32_t* quant_coeff[4][6][6]; int32_t *de_quant_coeff [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM][SCALING_LIST_REM_NUM]; double* error_scale[4][6][6]; } scaling_list; /* TODO: add ME data */ typedef struct { void (*IME)(); void (*FME)(); int range; } encoder_me; enum { FORMAT_400 = 0, FORMAT_420, FORMAT_422, FORMAT_444 }; /* Input info struct */ typedef struct { FILE *file; int32_t width; /*!< \brief input picture width */ int32_t height; /*!< \brief input picture height */ int32_t real_width; /*!< \brief real input picture width */ int32_t real_height; /*!< \brief real input picture width */ int32_t height_in_lcu; /*!< \brief input picture width in LCU*/ int32_t width_in_lcu; /*!< \brief input picture height in LCU */ picture *cur_pic; int8_t video_format; int8_t bitdepth; /*!< \brief input bit depth (8,10) */ } encoder_input; /* Encoder control options, the main struct */ typedef struct { int32_t frame; int32_t poc; /*!< \brief picture order count */ config *cfg; encoder_input in; encoder_me me; bitstream *stream; FILE *output; picture_list *ref; int8_t ref_list; int8_t ref_idx_num[2]; int8_t QP; // \brief Quantization parameter int8_t bitdepth; /* Filtering */ int8_t deblock_enable; // \brief Flag to enable deblocking filter int8_t sao_enable; // \brief Flag to enable sample adaptive offset filter int8_t rdoq_enable; // \brief Whether RDOQ is enabled or not. int8_t rdo; // \brief RDO level int8_t trskip_enable; // \brief Flag to enable transform skipping (4x4 intra) int8_t beta_offset_div2; // \brief (deblocking) beta offset (div 2), range -6...6 int8_t tc_offset_div2; // \brief (deblocking)tc offset (div 2), range -6...6 /* VUI */ struct { int16_t sar_width; int16_t sar_height; int8_t overscan; int8_t videoformat; int8_t fullrange; int8_t colorprim; int8_t transfer; int8_t colormatrix; int8_t chroma_loc; } vui; int8_t aud_enable; //scaling list uint8_t scaling_list_enable; scaling_list scaling_list; } encoder_control; void init_tables(void); void init_lambda(const encoder_control * const encoder); void free_tables(void); encoder_control *init_encoder_control(config *cfg); void init_encoder_input(encoder_input *input, FILE* inputfile, int32_t width, int32_t height); void encode_one_frame(encoder_control *encoder); int read_one_frame(FILE *file, const encoder_control * const encoder); void encode_seq_parameter_set(const encoder_control * const encoder); void encode_pic_parameter_set(const encoder_control * const encoder); void encode_vid_parameter_set(const encoder_control * const encoder); void encode_slice_header(const encoder_control * const encoder); void encode_access_unit_delimiter(const encoder_control * const encoder); void encode_prefix_sei_version(const encoder_control * const encoder); void encode_coding_tree(const encoder_control * const encoder, cabac_data *cabac, uint16_t x_ctb, uint16_t y_ctb, uint8_t depth); void encode_last_significant_xy(cabac_data *cabac, uint8_t lastpos_x, uint8_t lastpos_y, uint8_t width, uint8_t height, uint8_t type, uint8_t scan); void encode_coeff_nxn(const encoder_control * const encoder, cabac_data *cabac, int16_t *coeff, uint8_t width, uint8_t type, int8_t scan_mode, int8_t tr_skip); void encode_transform_tree(const encoder_control * const encoder, cabac_data* cabac, int32_t x, int32_t y, uint8_t depth, lcu_t* lcu ); void encode_transform_coeff(const encoder_control * const encoder, cabac_data *cabac, int32_t x_cu, int32_t y_cu, int8_t depth, int8_t tr_depth, uint8_t parent_coeff_u, uint8_t parent_coeff_v); void encode_block_residual(const encoder_control * const encoder, uint16_t x_ctb, uint16_t y_ctb, uint8_t depth); extern double g_lambda_cost[55]; extern double g_cur_lambda_cost; extern int8_t g_convert_to_bit[LCU_WIDTH + 1]; extern int8_t g_bitdepth; static const uint8_t g_group_idx[32] = { 0, 1, 2, 3, 4, 4, 5, 5, 6, 6, 6, 6, 7, 7, 7, 7, 8, 8, 8, 8, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 9, 9 }; static const uint8_t g_min_in_group[10] = { 0, 1, 2, 3, 4, 6, 8, 12, 16, 24 }; /** * List of mappings for coefficients within a transform block. * First index: scan pattern 0 = diagonal, 1 = horizontal, 2 = vertical * Second index: (log2 - 1) size of transform block. 2x2 .. 32x32 */ extern uint32_t* g_sig_last_scan[3][5]; /*** * List of indices for 4x4 coefficient groups within 8x8 transform block. * First index: 0 = diagonal, 1 = vertical, 2 horizontal scan pattern. * Second index: (log2 - 2) size of transform block. 4x4 .. 32x32 */ static const uint32_t g_sig_last_scan_8x8[3][4] = { {0, 2, 1, 3}, {0, 1, 2, 3}, {0, 2, 1, 3} }; /*** * List of indices for 4x4 coefficient groups within 16x16 transform block. */ static const uint32_t g_sig_last_scan_16x16[16] = { 0, 4, 1, 8, 5, 2, 12, 9, 6, 3, 13, 10, 7, 14, 11, 15 }; /*** * List of indices for 4x4 coefficient groups within 32x32 transform block. */ static const uint32_t g_sig_last_scan_32x32[64] = { 0, 8, 1, 16, 9, 2, 24, 17, 10, 3, 32, 25, 18, 11, 4, 40, 33, 26, 19, 12, 5, 48, 41, 34, 27, 20, 13, 6, 56, 49, 42, 35, 28, 21, 14, 7, 57, 50, 43, 36, 29, 22, 15, 58, 51, 44, 37, 30, 23, 59, 52, 45, 38, 31, 60, 53, 46, 39, 61, 54, 47, 62, 55, 63 }; /** * List of pointers to coefficient group mappings. * First index: (log2 - 2) of transform block size * Second index: scan pattern 0 = diagonal, 1 = horizontal, 2 = vertical */ static const uint32_t *const g_sig_last_scan_cg[4][3] = { { g_sig_last_scan_8x8[0], g_sig_last_scan_8x8[1], g_sig_last_scan_8x8[2] }, // 4x4, only first element is used { g_sig_last_scan_8x8[0], g_sig_last_scan_8x8[1], g_sig_last_scan_8x8[2] }, { g_sig_last_scan_16x16, 0, 0 }, { g_sig_last_scan_32x32, 0, 0 } }; //4 8 16 32 64 128 //0 1 2 3 4 5 static const uint8_t g_to_bits[129] = { 0, 0,0,0,0, 0,0,0,1, 0,0,0,0,0,0,0,2, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,3, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,4, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,5 }; #define TOBITS(len) g_to_bits[len] #define C1FLAG_NUMBER 8 // maximum number of largerThan1 flag coded in one chunk #define C2FLAG_NUMBER 1 // maximum number of largerThan2 flag coded in one chunk enum COEFF_SCAN_TYPE { SCAN_DIAG = 0, // up-right diagonal scan SCAN_HOR, // horizontal first scan SCAN_VER // vertical first scan }; #endif