mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-11-24 02:24:07 +00:00
8b2483ca1c
Replaces function kvz_intra_recon_lcu_luma and kvz_intra_recon_lcu_chroma in intra.c with function kvz_intra_recon_cu. The new function can handle reconstruction for both luma and chroma. Removes some duplicated code.
131 lines
3.9 KiB
C
131 lines
3.9 KiB
C
#ifndef INTRA_H_
|
|
#define INTRA_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 Reconstruction
|
|
* \file
|
|
* Intra prediction.
|
|
*/
|
|
|
|
#include "cu.h"
|
|
#include "encoderstate.h"
|
|
#include "global.h" // IWYU pragma: keep
|
|
#include "kvazaar.h"
|
|
|
|
|
|
typedef struct {
|
|
kvz_pixel left[2 * 32 + 1];
|
|
kvz_pixel top[2 * 32 + 1];
|
|
} kvz_intra_ref;
|
|
typedef struct
|
|
{
|
|
kvz_intra_ref ref;
|
|
kvz_intra_ref filtered_ref;
|
|
bool filtered_initialized;
|
|
} kvz_intra_references;
|
|
|
|
|
|
/**
|
|
* \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 kvz_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);
|
|
|
|
#if KVZ_SEL_ENCRYPTION
|
|
/**
|
|
* \brief Function for deriving intra luma predictions with encryption
|
|
* \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 kvz_intra_get_dir_luma_predictor_encry(
|
|
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);
|
|
#endif
|
|
|
|
/**
|
|
* \brief Generage angular predictions.
|
|
* \param width Width in pixels, range 4..32.
|
|
* \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 out_left_ref Left reference pixels, index 0 is the top-left.
|
|
* \param out_top_ref Top reference pixels, index 0 is the top-left.
|
|
*/
|
|
void kvz_intra_build_reference(
|
|
const int_fast8_t log2_width,
|
|
const color_t color,
|
|
const vector2d_t *const luma_px,
|
|
const vector2d_t *const pic_px,
|
|
const lcu_t *const lcu,
|
|
kvz_intra_references *const refs);
|
|
|
|
/**
|
|
* \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 kvz_intra_predict(
|
|
kvz_intra_references *refs,
|
|
int_fast8_t log2_width,
|
|
int_fast8_t mode,
|
|
color_t color,
|
|
kvz_pixel *dst,
|
|
bool filter_boundary);
|
|
|
|
void kvz_intra_recon_cu(
|
|
encoder_state_t *const state,
|
|
int x,
|
|
int y,
|
|
int depth,
|
|
int8_t mode_luma,
|
|
int8_t mode_chroma,
|
|
cu_info_t *cur_cu,
|
|
lcu_t *lcu);
|
|
|
|
#endif
|