mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-12-03 21:44:06 +00:00
87 lines
3.5 KiB
C
87 lines
3.5 KiB
C
#ifndef RDO_H_
|
|
#define RDO_H_
|
|
/*****************************************************************************
|
|
* This file is part of Kvazaar HEVC encoder.
|
|
*
|
|
* Copyright (C) 2013-2015 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 Lesser General Public License as published by the
|
|
* Free Software Foundation; either version 2.1 of the License, or (at your
|
|
* option) any later version.
|
|
*
|
|
* 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 Lesser General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with Kvazaar. If not, see <http://www.gnu.org/licenses/>.
|
|
****************************************************************************/
|
|
|
|
/**
|
|
* \ingroup Compression
|
|
* \file
|
|
* Rate-Distortion Optimization related functionality.
|
|
*/
|
|
|
|
#include "cabac.h"
|
|
#include "cu.h"
|
|
#include "encoderstate.h"
|
|
#include "global.h" // IWYU pragma: keep
|
|
#include "search_inter.h"
|
|
|
|
|
|
extern const uint32_t kvz_g_go_rice_range[5];
|
|
extern const uint32_t kvz_g_go_rice_prefix_len[5];
|
|
|
|
int kvz_init_rdcost_outfiles(const char *fn_template);
|
|
void kvz_close_rdcost_outfiles(void);
|
|
|
|
void kvz_rdoq(encoder_state_t *state, coeff_t *coef, coeff_t *dest_coeff, int32_t width,
|
|
int32_t height, int8_t type, int8_t scan_mode, int8_t block_type, int8_t tr_depth, uint16_t cbf);
|
|
|
|
|
|
int kvz_ts_rdoq(encoder_state_t* const state, coeff_t* src_coeff, coeff_t* dest_coeff, int32_t width,
|
|
int32_t height, int8_t type, int8_t scan_mode);
|
|
|
|
|
|
uint32_t kvz_get_coeff_cost(const encoder_state_t * const state,
|
|
const coeff_t *coeff,
|
|
int32_t width,
|
|
int32_t type,
|
|
int8_t scan_mode,
|
|
int8_t tr_skip);
|
|
|
|
int32_t kvz_get_ic_rate(encoder_state_t *state, uint32_t abs_level, uint16_t ctx_num_gt1, uint16_t ctx_num_gt2, uint16_t ctx_num_par,
|
|
uint16_t abs_go_rice, uint32_t reg_bins, int8_t type, int use_limited_prefix_length);
|
|
uint32_t kvz_get_coded_level(encoder_state_t * state, double* coded_cost, double* coded_cost0, double* coded_cost_sig,
|
|
int32_t level_double, uint32_t max_abs_level,
|
|
uint16_t ctx_num_sig, uint16_t ctx_num_gt1, uint16_t ctx_num_gt2, uint16_t ctx_num_par,
|
|
uint16_t abs_go_rice,
|
|
uint32_t reg_bins,
|
|
int32_t q_bits,double temp, int8_t last, int8_t type);
|
|
|
|
kvz_mvd_cost_func kvz_calc_mvd_cost_cabac;
|
|
|
|
uint32_t kvz_get_mvd_coding_cost_cabac(const encoder_state_t *state,
|
|
const cabac_data_t* cabac,
|
|
int32_t mvd_hor,
|
|
int32_t mvd_ver);
|
|
|
|
// Number of fixed point fractional bits used in the fractional bit table.
|
|
#define CTX_FRAC_BITS 15
|
|
#define CTX_FRAC_ONE_BIT (1 << CTX_FRAC_BITS)
|
|
#define CTX_FRAC_HALF_BIT (1 << (CTX_FRAC_BITS - 1))
|
|
|
|
extern const uint32_t kvz_entropy_bits[512];
|
|
#define CTX_ENTROPY_BITS(ctx, val) kvz_entropy_bits[CTX_STATE(ctx) ^ (val)]
|
|
|
|
// Floating point fractional bits, derived from kvz_entropy_bits
|
|
extern const float kvz_f_entropy_bits[512];
|
|
// ToDo: generate a new table for VVC?
|
|
#define CTX_ENTROPY_FBITS(ctx, val) kvz_f_entropy_bits[(CTX_STATE(ctx)) ^ (val)]
|
|
|
|
#endif
|