mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-11-24 10:34:05 +00:00
1efa2708b2
Moves SAO reconstruction into encoder_state_worker_encode_lcu instead of doing it in a separate step for the whole CTU row. Reconstruction of the rightmost 10 pixels and bottommost 10 pixels of a CTU is delayed until the neighboring CTU has been deblocked. Doing SAO for the whole CTU row at a time caused unnecessary inter-CTU dependencies when using WPP and OWF. The first CTU of a row would need to wait until SAO was done for the row below in the previous frame. Moving SAO reconstruction to immediately after deblocking each CTU fixes this problem.
89 lines
3.1 KiB
C
89 lines
3.1 KiB
C
#ifndef SAO_H_
|
|
#define SAO_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
|
|
* Sample Adaptive Offset filter.
|
|
*/
|
|
|
|
#include "checkpoint.h"
|
|
#include "cu.h"
|
|
#include "encoder.h"
|
|
#include "encoderstate.h"
|
|
#include "global.h" // IWYU pragma: keep
|
|
#include "kvazaar.h"
|
|
#include "videoframe.h"
|
|
|
|
|
|
typedef enum { SAO_TYPE_NONE = 0, SAO_TYPE_BAND, SAO_TYPE_EDGE } sao_type;
|
|
typedef enum { SAO_EO0 = 0, SAO_EO1, SAO_EO2, SAO_EO3, SAO_NUM_EO } sao_eo_class;
|
|
typedef enum { SAO_EO_CAT0 = 0, SAO_EO_CAT1, SAO_EO_CAT2, SAO_EO_CAT3, SAO_EO_CAT4, NUM_SAO_EDGE_CATEGORIES } sao_eo_cat;
|
|
|
|
|
|
typedef struct sao_info_t {
|
|
sao_type type;
|
|
sao_eo_class eo_class;
|
|
int ddistortion;
|
|
int merge_left_flag;
|
|
int merge_up_flag;
|
|
int band_position[2];
|
|
int offsets[NUM_SAO_EDGE_CATEGORIES * 2];
|
|
} sao_info_t;
|
|
|
|
|
|
// Offsets of a and b in relation to c.
|
|
// dir_offset[dir][a or b]
|
|
// | | a | a | a |
|
|
// | a c b | c | c | c |
|
|
// | | b | b | b |
|
|
static const vector2d_t g_sao_edge_offsets[SAO_NUM_EO][2] = {
|
|
{ { -1, 0 }, { 1, 0 } },
|
|
{ { 0, -1 }, { 0, 1 } },
|
|
{ { -1, -1 }, { 1, 1 } },
|
|
{ { 1, -1 }, { -1, 1 } }
|
|
};
|
|
|
|
|
|
#define CHECKPOINT_SAO_INFO(prefix_str, sao) CHECKPOINT(prefix_str " type=%d eo_class=%d ddistortion=%d " \
|
|
"merge_left_flag=%d merge_up_flag=%d band_position=%d " \
|
|
"offsets[0]=%d offsets[1]=%d offsets[2]=%d offsets[3]=%d offsets[4]=%d", \
|
|
(sao).type, (sao).eo_class, (sao).ddistortion, \
|
|
(sao).merge_left_flag, (sao).merge_up_flag, (sao).band_position[0], \
|
|
(sao).offsets[0], (sao).offsets[1], (sao).offsets[2], (sao).offsets[3], (sao).offsets[4])
|
|
|
|
|
|
void kvz_sao_reconstruct(const encoder_state_t *state,
|
|
const kvz_pixel *buffer,
|
|
int stride,
|
|
int frame_x,
|
|
int frame_y,
|
|
int width,
|
|
int height,
|
|
const sao_info_t *sao,
|
|
color_t color);
|
|
|
|
void kvz_sao_search_lcu(const encoder_state_t* const state, int lcu_x, int lcu_y);
|
|
void kvz_calc_sao_offset_array(const encoder_control_t * const encoder, const sao_info_t *sao, int *offset, color_t color_i);
|
|
|
|
#endif
|