mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-12-03 21:44:06 +00:00
179 lines
6 KiB
C
179 lines
6 KiB
C
#pragma once
|
|
/*****************************************************************************
|
|
* This file is part of uvg266 VVC encoder.
|
|
*
|
|
* Copyright (c) 2021, Tampere University, ITU/ISO/IEC, project contributors
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without modification,
|
|
* are permitted provided that the following conditions are met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright notice, this
|
|
* list of conditions and the following disclaimer.
|
|
*
|
|
* * Redistributions in binary form must reproduce the above copyright notice, this
|
|
* list of conditions and the following disclaimer in the documentation and/or
|
|
* other materials provided with the distribution.
|
|
*
|
|
* * Neither the name of the Tampere University or ITU/ISO/IEC nor the names of its
|
|
* contributors may be used to endorse or promote products derived from
|
|
* this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
|
|
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
* INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION HOWEVER CAUSED AND ON
|
|
* ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
* INCLUDING NEGLIGENCE OR OTHERWISE ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
****************************************************************************/
|
|
|
|
/**
|
|
* \ingroup Reconstruction
|
|
* \file
|
|
* Intra prediction.
|
|
*/
|
|
|
|
#include "cu.h"
|
|
#include "encoderstate.h"
|
|
#include "global.h" // IWYU pragma: keep
|
|
#include "uvg266.h"
|
|
|
|
// Maximum possible reference line length for intra blocks
|
|
#define INTRA_REF_LENGTH (2 * 128 + 3 + 33 * MAX_REF_LINE_IDX)
|
|
|
|
typedef struct {
|
|
uvg_pixel left[INTRA_REF_LENGTH];
|
|
uvg_pixel top[INTRA_REF_LENGTH];
|
|
} uvg_intra_ref;
|
|
typedef struct
|
|
{
|
|
uvg_intra_ref ref;
|
|
uvg_intra_ref filtered_ref;
|
|
bool filtered_initialized;
|
|
} uvg_intra_references;
|
|
|
|
typedef struct
|
|
{
|
|
int16_t a;
|
|
int16_t shift;
|
|
int16_t b;
|
|
} cclm_parameters_t;
|
|
|
|
typedef struct {
|
|
cu_info_t pred_cu;
|
|
cclm_parameters_t cclm_parameters[2];
|
|
double cost;
|
|
double bits;
|
|
double coeff_bits;
|
|
double distortion;
|
|
double lfnst_costs[3];
|
|
uint8_t best_isp_cbfs;
|
|
} intra_search_data_t ;
|
|
|
|
|
|
#define UVG_NUM_INTRA_MODES 67
|
|
|
|
/**
|
|
* \brief Function for deriving intra luma predictions
|
|
* \param x x-coordinate of the PU in pixels
|
|
* \param y y-coordinate of the PU in pixels
|
|
* \param preds output buffer for 3 predictions
|
|
* \param cur_pu PU to check
|
|
* \param left_pu PU to the left of cur_pu
|
|
* \param above_pu PU above cur_pu
|
|
* \returns 1 if predictions are found, otherwise 0
|
|
*/
|
|
int8_t uvg_intra_get_dir_luma_predictor(
|
|
const uint32_t x,
|
|
const uint32_t y,
|
|
int8_t *preds,
|
|
const cu_info_t *const cur_pu,
|
|
const cu_info_t *const left_pu,
|
|
const cu_info_t *const above_pu);
|
|
|
|
/**
|
|
* \brief Build intra prediction reference buffers.
|
|
* \param log2_width Log2 of width, range 2..5.
|
|
* \param color What color pixels to use.
|
|
* \param luma_px Luma coordinates of the prediction block.
|
|
* \param pic_px Picture dimensions in luma pixels.
|
|
* \param lcu LCU struct.
|
|
* \param refs Pointer to top and left references.
|
|
* \param entropy_sync Indicate that top right is not available if WPP is enabled.
|
|
* \param extra_refs Additional left edge reference lines for use with MRL.
|
|
* \param multi_ref_idx Multi reference line index for the prediction block.
|
|
*/
|
|
void uvg_intra_build_reference(
|
|
const cu_loc_t* const pu_loc,
|
|
const cu_loc_t* const cu_loc,
|
|
const color_t color,
|
|
const vector2d_t *const luma_px,
|
|
const vector2d_t *const pic_px,
|
|
const lcu_t *const lcu,
|
|
uvg_intra_references *const refs,
|
|
bool entropy_sync,
|
|
uvg_pixel *extra_refs,
|
|
uint8_t multi_ref_idx,
|
|
const uint8_t isp_mode);
|
|
|
|
/**
|
|
* \brief Generate intra predictions.
|
|
* \param refs Reference pixels used for the prediction.
|
|
* \param log2_width Width of the predicted block.
|
|
* \param mode Intra mode used for the prediction.
|
|
* \param color Color of the prediction.
|
|
* \param dst Buffer for the predicted pixels.
|
|
* \param filter_boundary Whether to filter the boundary on modes 10 and 26.
|
|
*/
|
|
void uvg_intra_predict(
|
|
const encoder_state_t* const state,
|
|
uvg_intra_references* const refs,
|
|
const cu_loc_t* const cu_loc,
|
|
const color_t color,
|
|
uvg_pixel* dst,
|
|
const intra_search_data_t* data,
|
|
const lcu_t* lcu,
|
|
enum uvg_tree_type tree_type
|
|
);
|
|
|
|
void uvg_intra_recon_cu(
|
|
encoder_state_t* const state,
|
|
int x,
|
|
int y,
|
|
int depth,
|
|
intra_search_data_t* search_data,
|
|
cu_info_t *cur_cu,
|
|
lcu_t *lcu,
|
|
enum uvg_tree_type tree_type,
|
|
bool recon_luma,
|
|
bool recon_chroma);
|
|
|
|
const cu_info_t* uvg_get_co_located_luma_cu(
|
|
int x,
|
|
int y,
|
|
int width,
|
|
int height,
|
|
const lcu_t* const lcu,
|
|
const cu_array_t* const cu_array,
|
|
enum uvg_tree_type tree_type);
|
|
|
|
int uvg_get_mip_flag_context(int x, int y, int width, int height, const lcu_t* lcu, cu_array_t* const cu_a);
|
|
|
|
// ISP related defines
|
|
#define NUM_ISP_MODES 3
|
|
#define ISP_MODE_NO_ISP 0
|
|
#define ISP_MODE_HOR 1
|
|
#define ISP_MODE_VER 2
|
|
#define SPLIT_TYPE_HOR 1
|
|
#define SPLIT_TYPE_VER 2
|
|
|
|
int uvg_get_isp_split_dim(const int width, const int height, const int split_type);
|
|
int uvg_get_isp_split_num(const int width, const int height, const int split_type);
|
|
void uvg_get_isp_split_loc(cu_loc_t *loc, const int x, const int y, const int block_w, const int block_h, const int split_idx, const int split_type);
|
|
bool uvg_can_use_isp(const int width, const int height, const int max_tr_size);
|
|
bool uvg_can_use_isp_with_lfnst(const int width, const int height, const int isp_mode, const enum uvg_tree_type tree_type);
|