mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-11-24 02:24:07 +00:00
Move transform unit coding to its own function.
This commit is contained in:
Ari Koivula
parent
a38a5fd647
commit
35ba873abc
167
src/encoder.c
167
src/encoder.c
|
|
@ -1673,89 +1673,15 @@ void encode_transform_tree(encoder_control *encoder, int32_t x_pu, int32_t y_pu,
|
|||
// end Residual Coding
|
||||
}
|
||||
|
||||
/**
|
||||
* \param encoder
|
||||
* \param x_pu Prediction units' x coordinate.
|
||||
* \param y_pu Prediction units' y coordinate.
|
||||
* \param depth Depth from LCU.
|
||||
* \param tr_depth Depth from last CU.
|
||||
* \param parent_coeff_u What was signaled at previous level for cbf_cb.
|
||||
* \param parent_coeff_v What was signlaed at previous level for cbf_cr.
|
||||
*/
|
||||
void encode_transform_coeff(encoder_control *encoder, int32_t x_pu,int32_t y_pu,
|
||||
int8_t depth, int8_t tr_depth, uint8_t parent_coeff_u, uint8_t parent_coeff_v)
|
||||
void encode_transform_unit(encoder_control *encoder, int x_pu, int y_pu, int depth, int tr_depth)
|
||||
{
|
||||
int32_t x_cu = x_pu / 2;
|
||||
int32_t y_cu = y_pu / 2;
|
||||
int width = LCU_WIDTH >> depth;
|
||||
int width_c = (depth == MAX_PU_DEPTH ? width : width >> 1);
|
||||
|
||||
int x_cu = x_pu / 2;
|
||||
int y_cu = y_pu / 2;
|
||||
cu_info *cur_cu = &encoder->in.cur_pic->cu_array[MAX_DEPTH][x_cu + y_cu * (encoder->in.width_in_lcu << MAX_DEPTH)];
|
||||
int8_t width = LCU_WIDTH >> depth;
|
||||
int8_t width_c = (depth == MAX_PU_DEPTH ? width : width >> 1);
|
||||
|
||||
// NxN signifies implicit transform split at the first transform level.
|
||||
// There is a similar implicit split for inter, but it is only used when
|
||||
// transform hierarchy is not in use.
|
||||
int intra_split_flag = (cur_cu->type == CU_INTRA && cur_cu->part_size == SIZE_NxN);
|
||||
|
||||
// The implicit split by intra NxN is not counted towards max_tr_depth.
|
||||
int max_tr_depth = (cur_cu->type == CU_INTRA ? TR_DEPTH_INTRA + intra_split_flag : TR_DEPTH_INTER);
|
||||
|
||||
int8_t split = cur_cu->tr_depth > depth;
|
||||
|
||||
int8_t cb_flag_u = !split ? cur_cu->coeff_u : cur_cu->coeff_top_u[depth];
|
||||
int8_t cb_flag_v = !split ? cur_cu->coeff_v : cur_cu->coeff_top_v[depth];
|
||||
|
||||
// The split_transform_flag is not signaled when:
|
||||
// - transform size is greater than 32 (depth == 0)
|
||||
// - transform size is 4 (depth == MAX_PU_DEPTH)
|
||||
// - transform depth is max
|
||||
// - cu is intra NxN and it's the first split
|
||||
if (depth > 0 &&
|
||||
depth < MAX_PU_DEPTH &&
|
||||
tr_depth < max_tr_depth &&
|
||||
!(intra_split_flag && tr_depth == 0))
|
||||
{
|
||||
cabac.ctx = &g_trans_subdiv_model[5 - ((g_convert_to_bit[LCU_WIDTH] + 2) - depth)];
|
||||
CABAC_BIN(&cabac, split, "split_transform_flag");
|
||||
}
|
||||
|
||||
// Chroma cb flags are not signaled when one of the following:
|
||||
// - transform size is 4 (2x2 chroma transform doesn't exist)
|
||||
// - they have already been signaled to 0 previously
|
||||
// When they are not present they are inferred to be 0, except for size 4
|
||||
// when the flags from previous level are used.
|
||||
if (depth < MAX_PU_DEPTH) {
|
||||
cabac.ctx = &g_qt_cbf_model_chroma[tr_depth];
|
||||
if (tr_depth == 0 || parent_coeff_u) {
|
||||
CABAC_BIN(&cabac, cb_flag_u, "cbf_cb");
|
||||
}
|
||||
if (tr_depth == 0 || parent_coeff_v) {
|
||||
CABAC_BIN(&cabac, cb_flag_v, "cbf_cr");
|
||||
}
|
||||
}
|
||||
|
||||
if (split) {
|
||||
uint8_t pu_offset = 1 << (MAX_PU_DEPTH - (depth + 1));
|
||||
encode_transform_coeff(encoder, x_pu, y_pu, depth + 1, tr_depth + 1, cb_flag_u, cb_flag_v);
|
||||
encode_transform_coeff(encoder, x_pu + pu_offset, y_pu, depth + 1, tr_depth + 1, cb_flag_u, cb_flag_v);
|
||||
encode_transform_coeff(encoder, x_pu, y_pu + pu_offset, depth + 1, tr_depth + 1, cb_flag_u, cb_flag_v);
|
||||
encode_transform_coeff(encoder, x_pu + pu_offset, y_pu + pu_offset, depth + 1, tr_depth + 1, cb_flag_u, cb_flag_v);
|
||||
return;
|
||||
}
|
||||
|
||||
// Luma coded block flag is signaled when one of the following:
|
||||
// - prediction mode is intra
|
||||
// - transform depth > 0
|
||||
// - we have chroma coefficients at this level
|
||||
// When it is not present, it is inferred to be 1.
|
||||
if(cur_cu->type == CU_INTRA || tr_depth > 0 || cur_cu->coeff_u || cur_cu->coeff_v) {
|
||||
cabac.ctx = &g_qt_cbf_model_luma[!tr_depth];
|
||||
CABAC_BIN(&cabac, cur_cu->coeff_y, "cbf_luma");
|
||||
}
|
||||
|
||||
// End of transform tree.
|
||||
// Beginning of transform unit.
|
||||
|
||||
{
|
||||
coefficient coeff_y[LCU_WIDTH*LCU_WIDTH+1];
|
||||
coefficient coeff_u[LCU_WIDTH*LCU_WIDTH>>2];
|
||||
coefficient coeff_v[LCU_WIDTH*LCU_WIDTH>>2];
|
||||
|
|
@ -1864,6 +1790,87 @@ void encode_transform_coeff(encoder_control *encoder, int32_t x_pu,int32_t y_pu,
|
|||
encode_coeff_nxn(encoder, coeff_v, width_c, 2, scan_idx);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* \param encoder
|
||||
* \param x_pu Prediction units' x coordinate.
|
||||
* \param y_pu Prediction units' y coordinate.
|
||||
* \param depth Depth from LCU.
|
||||
* \param tr_depth Depth from last CU.
|
||||
* \param parent_coeff_u What was signaled at previous level for cbf_cb.
|
||||
* \param parent_coeff_v What was signlaed at previous level for cbf_cr.
|
||||
*/
|
||||
void encode_transform_coeff(encoder_control *encoder, int32_t x_pu,int32_t y_pu,
|
||||
int8_t depth, int8_t tr_depth, uint8_t parent_coeff_u, uint8_t parent_coeff_v)
|
||||
{
|
||||
int32_t x_cu = x_pu / 2;
|
||||
int32_t y_cu = y_pu / 2;
|
||||
cu_info *cur_cu = &encoder->in.cur_pic->cu_array[MAX_DEPTH][x_cu + y_cu * (encoder->in.width_in_lcu << MAX_DEPTH)];
|
||||
|
||||
// NxN signifies implicit transform split at the first transform level.
|
||||
// There is a similar implicit split for inter, but it is only used when
|
||||
// transform hierarchy is not in use.
|
||||
int intra_split_flag = (cur_cu->type == CU_INTRA && cur_cu->part_size == SIZE_NxN);
|
||||
|
||||
// The implicit split by intra NxN is not counted towards max_tr_depth.
|
||||
int max_tr_depth = (cur_cu->type == CU_INTRA ? TR_DEPTH_INTRA + intra_split_flag : TR_DEPTH_INTER);
|
||||
|
||||
int8_t split = cur_cu->tr_depth > depth;
|
||||
|
||||
int8_t cb_flag_u = !split ? cur_cu->coeff_u : cur_cu->coeff_top_u[depth];
|
||||
int8_t cb_flag_v = !split ? cur_cu->coeff_v : cur_cu->coeff_top_v[depth];
|
||||
|
||||
// The split_transform_flag is not signaled when:
|
||||
// - transform size is greater than 32 (depth == 0)
|
||||
// - transform size is 4 (depth == MAX_PU_DEPTH)
|
||||
// - transform depth is max
|
||||
// - cu is intra NxN and it's the first split
|
||||
if (depth > 0 &&
|
||||
depth < MAX_PU_DEPTH &&
|
||||
tr_depth < max_tr_depth &&
|
||||
!(intra_split_flag && tr_depth == 0))
|
||||
{
|
||||
cabac.ctx = &g_trans_subdiv_model[5 - ((g_convert_to_bit[LCU_WIDTH] + 2) - depth)];
|
||||
CABAC_BIN(&cabac, split, "split_transform_flag");
|
||||
}
|
||||
|
||||
// Chroma cb flags are not signaled when one of the following:
|
||||
// - transform size is 4 (2x2 chroma transform doesn't exist)
|
||||
// - they have already been signaled to 0 previously
|
||||
// When they are not present they are inferred to be 0, except for size 4
|
||||
// when the flags from previous level are used.
|
||||
if (depth < MAX_PU_DEPTH) {
|
||||
cabac.ctx = &g_qt_cbf_model_chroma[tr_depth];
|
||||
if (tr_depth == 0 || parent_coeff_u) {
|
||||
CABAC_BIN(&cabac, cb_flag_u, "cbf_cb");
|
||||
}
|
||||
if (tr_depth == 0 || parent_coeff_v) {
|
||||
CABAC_BIN(&cabac, cb_flag_v, "cbf_cr");
|
||||
}
|
||||
}
|
||||
|
||||
if (split) {
|
||||
uint8_t pu_offset = 1 << (MAX_PU_DEPTH - (depth + 1));
|
||||
encode_transform_coeff(encoder, x_pu, y_pu, depth + 1, tr_depth + 1, cb_flag_u, cb_flag_v);
|
||||
encode_transform_coeff(encoder, x_pu + pu_offset, y_pu, depth + 1, tr_depth + 1, cb_flag_u, cb_flag_v);
|
||||
encode_transform_coeff(encoder, x_pu, y_pu + pu_offset, depth + 1, tr_depth + 1, cb_flag_u, cb_flag_v);
|
||||
encode_transform_coeff(encoder, x_pu + pu_offset, y_pu + pu_offset, depth + 1, tr_depth + 1, cb_flag_u, cb_flag_v);
|
||||
return;
|
||||
}
|
||||
|
||||
// Luma coded block flag is signaled when one of the following:
|
||||
// - prediction mode is intra
|
||||
// - transform depth > 0
|
||||
// - we have chroma coefficients at this level
|
||||
// When it is not present, it is inferred to be 1.
|
||||
if(cur_cu->type == CU_INTRA || tr_depth > 0 || cur_cu->coeff_u || cur_cu->coeff_v) {
|
||||
cabac.ctx = &g_qt_cbf_model_luma[!tr_depth];
|
||||
CABAC_BIN(&cabac, cur_cu->coeff_y, "cbf_luma");
|
||||
}
|
||||
|
||||
if (cur_cu->coeff_y | cur_cu->coeff_u | cur_cu->coeff_v) {
|
||||
encode_transform_unit(encoder, x_pu, y_pu, depth, tr_depth);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue