This source file includes following definitions.
- markTransformBlockBoundary
- markPredictionBlockBoundary
- derive_edgeFlags_CTBRow
- derive_edgeFlags
- derive_boundaryStrength
- derive_boundaryStrength_CTB
- edge_filtering_luma_internal
- edge_filtering_luma
- edge_filtering_luma_CTB
- edge_filtering_chroma_internal
- edge_filtering_chroma
- edge_filtering_chroma_CTB
- name
- work
- add_deblocking_tasks
- apply_deblocking_filter
#include "deblock.h"
#include "util.h"
#include "transform.h"
#include "de265.h"
#include <assert.h>
void markTransformBlockBoundary(de265_image* img, int x0,int y0,
int log2TrafoSize,int trafoDepth,
int filterLeftCbEdge, int filterTopCbEdge)
{
logtrace(LogDeblock,"markTransformBlockBoundary(%d,%d, %d,%d, %d,%d)\n",x0,y0,
log2TrafoSize,trafoDepth, filterLeftCbEdge,filterTopCbEdge);
int split_transform = img->get_split_transform_flag(x0,y0,trafoDepth);
if (split_transform) {
int x1 = x0 + ((1<<log2TrafoSize)>>1);
int y1 = y0 + ((1<<log2TrafoSize)>>1);
markTransformBlockBoundary(img,x0,y0,log2TrafoSize-1,trafoDepth+1, filterLeftCbEdge, filterTopCbEdge);
markTransformBlockBoundary(img,x1,y0,log2TrafoSize-1,trafoDepth+1, DEBLOCK_FLAG_VERTI, filterTopCbEdge);
markTransformBlockBoundary(img,x0,y1,log2TrafoSize-1,trafoDepth+1, filterLeftCbEdge, DEBLOCK_FLAG_HORIZ);
markTransformBlockBoundary(img,x1,y1,log2TrafoSize-1,trafoDepth+1, DEBLOCK_FLAG_VERTI, DEBLOCK_FLAG_HORIZ);
}
else {
for (int k=0;k<(1<<log2TrafoSize);k+=4) {
img->set_deblk_flags(x0,y0+k, filterLeftCbEdge);
}
for (int k=0;k<(1<<log2TrafoSize);k+=4) {
img->set_deblk_flags(x0+k,y0, filterTopCbEdge);
}
}
}
void markPredictionBlockBoundary(de265_image* img, int x0,int y0,
int log2CbSize,
int filterLeftCbEdge, int filterTopCbEdge)
{
logtrace(LogDeblock,"markPredictionBlockBoundary(%d,%d, %d, %d,%d)\n",x0,y0,
log2CbSize, filterLeftCbEdge,filterTopCbEdge);
enum PartMode partMode = img->get_PartMode(x0,y0);
int cbSize = 1<<log2CbSize;
int cbSize2 = 1<<(log2CbSize-1);
int cbSize4 = 1<<(log2CbSize-2);
switch (partMode) {
case PART_NxN:
for (int k=0;k<cbSize;k++) {
img->set_deblk_flags(x0+cbSize2,y0+k, DEBLOCK_PB_EDGE_VERTI);
img->set_deblk_flags(x0+k,y0+cbSize2, DEBLOCK_PB_EDGE_HORIZ);
}
break;
case PART_Nx2N:
for (int k=0;k<cbSize;k++) {
img->set_deblk_flags(x0+cbSize2,y0+k, DEBLOCK_PB_EDGE_VERTI);
}
break;
case PART_2NxN:
for (int k=0;k<cbSize;k++) {
img->set_deblk_flags(x0+k,y0+cbSize2, DEBLOCK_PB_EDGE_HORIZ);
}
break;
case PART_nLx2N:
for (int k=0;k<cbSize;k++) {
img->set_deblk_flags(x0+cbSize4,y0+k, DEBLOCK_PB_EDGE_VERTI);
}
break;
case PART_nRx2N:
for (int k=0;k<cbSize;k++) {
img->set_deblk_flags(x0+cbSize2+cbSize4,y0+k, DEBLOCK_PB_EDGE_VERTI);
}
break;
case PART_2NxnU:
for (int k=0;k<cbSize;k++) {
img->set_deblk_flags(x0+k,y0+cbSize4, DEBLOCK_PB_EDGE_HORIZ);
}
break;
case PART_2NxnD:
for (int k=0;k<cbSize;k++) {
img->set_deblk_flags(x0+k,y0+cbSize2+cbSize4, DEBLOCK_PB_EDGE_HORIZ);
}
break;
case PART_2Nx2N:
break;
}
}
bool derive_edgeFlags_CTBRow(de265_image* img, int ctby)
{
const seq_parameter_set& sps = img->get_sps();
const pic_parameter_set& pps = img->get_pps();
const int minCbSize = sps.MinCbSizeY;
bool deblocking_enabled=false;
int ctb_mask = (1<<sps.Log2CtbSizeY)-1;
int picWidthInCtbs = sps.PicWidthInCtbsY;
int ctbshift = sps.Log2CtbSizeY;
int cb_y_start = ( ctby << sps.Log2CtbSizeY) >> sps.Log2MinCbSizeY;
int cb_y_end = ((ctby+1) << sps.Log2CtbSizeY) >> sps.Log2MinCbSizeY;
cb_y_end = std::min(cb_y_end, sps.PicHeightInMinCbsY);
for (int cb_y=cb_y_start;cb_y<cb_y_end;cb_y++)
for (int cb_x=0;cb_x<img->get_sps().PicWidthInMinCbsY;cb_x++)
{
int log2CbSize = img->get_log2CbSize_cbUnits(cb_x,cb_y);
if (log2CbSize==0) {
continue;
}
int x0 = cb_x * minCbSize;
int y0 = cb_y * minCbSize;
int x0ctb = x0 >> ctbshift;
int y0ctb = y0 >> ctbshift;
if (img->is_SliceHeader_available(x0,y0)==false) {
return false;
}
slice_segment_header* shdr = img->get_SliceHeader(x0,y0);
uint8_t filterLeftCbEdge = DEBLOCK_FLAG_VERTI;
uint8_t filterTopCbEdge = DEBLOCK_FLAG_HORIZ;
if (x0 == 0) filterLeftCbEdge = 0;
if (y0 == 0) filterTopCbEdge = 0;
if (x0 && ((x0 & ctb_mask) == 0)) {
if (shdr->slice_loop_filter_across_slices_enabled_flag == 0 &&
img->is_SliceHeader_available(x0-1,y0) &&
shdr->SliceAddrRS != img->get_SliceHeader(x0-1,y0)->SliceAddrRS)
{
filterLeftCbEdge = 0;
}
else if (pps.loop_filter_across_tiles_enabled_flag == 0 &&
pps.TileIdRS[ x0ctb +y0ctb*picWidthInCtbs] !=
pps.TileIdRS[((x0-1)>>ctbshift)+y0ctb*picWidthInCtbs]) {
filterLeftCbEdge = 0;
}
}
if (y0 && ((y0 & ctb_mask) == 0)) {
if (shdr->slice_loop_filter_across_slices_enabled_flag == 0 &&
img->is_SliceHeader_available(x0,y0-1) &&
shdr->SliceAddrRS != img->get_SliceHeader(x0,y0-1)->SliceAddrRS)
{
filterTopCbEdge = 0;
}
else if (pps.loop_filter_across_tiles_enabled_flag == 0 &&
pps.TileIdRS[x0ctb+ y0ctb *picWidthInCtbs] !=
pps.TileIdRS[x0ctb+((y0-1)>>ctbshift)*picWidthInCtbs]) {
filterTopCbEdge = 0;
}
}
if (shdr->slice_deblocking_filter_disabled_flag==0) {
deblocking_enabled=true;
markTransformBlockBoundary(img, x0,y0, log2CbSize,0,
filterLeftCbEdge, filterTopCbEdge);
markPredictionBlockBoundary(img, x0,y0, log2CbSize,
filterLeftCbEdge, filterTopCbEdge);
}
}
return deblocking_enabled;
}
bool derive_edgeFlags(de265_image* img)
{
bool deblocking_enabled=false;
for (int y=0;y<img->get_sps().PicHeightInCtbsY;y++) {
deblocking_enabled |= derive_edgeFlags_CTBRow(img,y);
}
return deblocking_enabled;
}
void derive_boundaryStrength(de265_image* img, bool vertical, int yStart,int yEnd,
int xStart,int xEnd)
{
int xIncr = vertical ? 2 : 1;
int yIncr = vertical ? 1 : 2;
int xOffs = vertical ? 1 : 0;
int yOffs = vertical ? 0 : 1;
int edgeMask = vertical ?
(DEBLOCK_FLAG_VERTI | DEBLOCK_PB_EDGE_VERTI) :
(DEBLOCK_FLAG_HORIZ | DEBLOCK_PB_EDGE_HORIZ);
int transformEdgeMask = vertical ? DEBLOCK_FLAG_VERTI : DEBLOCK_FLAG_HORIZ;
xEnd = libde265_min(xEnd,img->get_deblk_width());
yEnd = libde265_min(yEnd,img->get_deblk_height());
int TUShift = img->get_sps().Log2MinTrafoSize;
int TUStride= img->get_sps().PicWidthInTbsY;
for (int y=yStart;y<yEnd;y+=yIncr)
for (int x=xStart;x<xEnd;x+=xIncr) {
int xDi = x<<2;
int yDi = y<<2;
logtrace(LogDeblock,"%d %d %s = %s\n",xDi,yDi, vertical?"Vertical":"Horizontal",
(img->get_deblk_flags(xDi,yDi) & edgeMask) ? "edge" : "...");
uint8_t edgeFlags = img->get_deblk_flags(xDi,yDi);
if (edgeFlags & edgeMask) {
bool p_is_intra_pred = (img->get_pred_mode(xDi-xOffs, yDi-yOffs) == MODE_INTRA);
bool q_is_intra_pred = (img->get_pred_mode(xDi, yDi ) == MODE_INTRA);
int bS;
if (p_is_intra_pred || q_is_intra_pred) {
bS = 2;
}
else {
int xDiOpp = xDi-xOffs;
int yDiOpp = yDi-yOffs;
if ((edgeFlags & transformEdgeMask) &&
(img->get_nonzero_coefficient(xDi ,yDi) ||
img->get_nonzero_coefficient(xDiOpp,yDiOpp))) {
bS = 1;
}
else {
bS = 0;
const PBMotion& mviP = img->get_mv_info(xDiOpp,yDiOpp);
const PBMotion& mviQ = img->get_mv_info(xDi ,yDi);
slice_segment_header* shdrP = img->get_SliceHeader(xDiOpp,yDiOpp);
slice_segment_header* shdrQ = img->get_SliceHeader(xDi ,yDi);
int refPicP0 = mviP.predFlag[0] ? shdrP->RefPicList[0][ mviP.refIdx[0] ] : -1;
int refPicP1 = mviP.predFlag[1] ? shdrP->RefPicList[1][ mviP.refIdx[1] ] : -1;
int refPicQ0 = mviQ.predFlag[0] ? shdrQ->RefPicList[0][ mviQ.refIdx[0] ] : -1;
int refPicQ1 = mviQ.predFlag[1] ? shdrQ->RefPicList[1][ mviQ.refIdx[1] ] : -1;
bool samePics = ((refPicP0==refPicQ0 && refPicP1==refPicQ1) ||
(refPicP0==refPicQ1 && refPicP1==refPicQ0));
if (!samePics) {
bS = 1;
}
else {
MotionVector mvP0 = mviP.mv[0]; if (!mviP.predFlag[0]) { mvP0.x=mvP0.y=0; }
MotionVector mvP1 = mviP.mv[1]; if (!mviP.predFlag[1]) { mvP1.x=mvP1.y=0; }
MotionVector mvQ0 = mviQ.mv[0]; if (!mviQ.predFlag[0]) { mvQ0.x=mvQ0.y=0; }
MotionVector mvQ1 = mviQ.mv[1]; if (!mviQ.predFlag[1]) { mvQ1.x=mvQ1.y=0; }
int numMV_P = mviP.predFlag[0] + mviP.predFlag[1];
int numMV_Q = mviQ.predFlag[0] + mviQ.predFlag[1];
if (numMV_P!=numMV_Q) {
img->decctx->add_warning(DE265_WARNING_NUMMVP_NOT_EQUAL_TO_NUMMVQ, false);
img->integrity = INTEGRITY_DECODING_ERRORS;
}
if (refPicP0 != refPicP1) {
if (refPicP0 == refPicQ0) {
if (abs_value(mvP0.x-mvQ0.x) >= 4 ||
abs_value(mvP0.y-mvQ0.y) >= 4 ||
abs_value(mvP1.x-mvQ1.x) >= 4 ||
abs_value(mvP1.y-mvQ1.y) >= 4) {
bS = 1;
}
}
else {
if (abs_value(mvP0.x-mvQ1.x) >= 4 ||
abs_value(mvP0.y-mvQ1.y) >= 4 ||
abs_value(mvP1.x-mvQ0.x) >= 4 ||
abs_value(mvP1.y-mvQ0.y) >= 4) {
bS = 1;
}
}
}
else {
assert(refPicQ0==refPicQ1);
if ((abs_value(mvP0.x-mvQ0.x) >= 4 ||
abs_value(mvP0.y-mvQ0.y) >= 4 ||
abs_value(mvP1.x-mvQ1.x) >= 4 ||
abs_value(mvP1.y-mvQ1.y) >= 4)
&&
(abs_value(mvP0.x-mvQ1.x) >= 4 ||
abs_value(mvP0.y-mvQ1.y) >= 4 ||
abs_value(mvP1.x-mvQ0.x) >= 4 ||
abs_value(mvP1.y-mvQ0.y) >= 4)) {
bS = 1;
}
}
}
}
}
img->set_deblk_bS(xDi,yDi, bS);
}
else {
img->set_deblk_bS(xDi,yDi, 0);
}
}
}
void derive_boundaryStrength_CTB(de265_image* img, bool vertical, int xCtb,int yCtb)
{
int ctbSize = img->get_sps().CtbSizeY;
int deblkSize = ctbSize/4;
derive_boundaryStrength(img,vertical,
yCtb*deblkSize, (yCtb+1)*deblkSize,
xCtb*deblkSize, (xCtb+1)*deblkSize);
}
static uint8_t table_8_23_beta[52] = {
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 6, 7, 8,
9,10,11,12,13,14,15,16,17,18,20,22,24,26,28,30,32,34,36,
38,40,42,44,46,48,50,52,54,56,58,60,62,64
};
static uint8_t table_8_23_tc[54] = {
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4,
5, 5, 6, 6, 7, 8, 9,10,11,13,14,16,18,20,22,24
};
template <class pixel_t>
void edge_filtering_luma_internal(de265_image* img, bool vertical,
int yStart,int yEnd, int xStart,int xEnd)
{
const seq_parameter_set& sps = img->get_sps();
int xIncr = vertical ? 2 : 1;
int yIncr = vertical ? 1 : 2;
const int stride = img->get_image_stride(0);
int bitDepth_Y = sps.BitDepth_Y;
xEnd = libde265_min(xEnd,img->get_deblk_width());
yEnd = libde265_min(yEnd,img->get_deblk_height());
for (int y=yStart;y<yEnd;y+=yIncr)
for (int x=xStart;x<xEnd;x+=xIncr) {
int xDi = x<<2;
int yDi = y<<2;
int bS = img->get_deblk_bS(xDi,yDi);
logtrace(LogDeblock,"deblock POC=%d %c --- x:%d y:%d bS:%d---\n",
img->PicOrderCntVal,vertical ? 'V':'H',xDi,yDi,bS);
#if 0
{
uint8_t* ptr = img->y + stride*yDi + xDi;
for (int dy=-4;dy<4;dy++) {
for (int dx=-4;dx<4;dx++) {
printf("%02x ", ptr[dy*stride + dx]);
if (dx==-1) printf("| ");
}
printf("\n");
if (dy==-1) printf("-------------------------\n");
}
}
#endif
#if 0
if (!vertical)
{
uint8_t* ptr = img->y + stride*yDi + xDi;
for (int dy=-4;dy<4;dy++) {
for (int dx=0;dx<4;dx++) {
printf("%02x ", ptr[dy*stride + dx]);
if (dx==-1) printf("| ");
}
printf("\n");
if (dy==-1) printf("-------------------------\n");
}
}
#endif
if (bS>0) {
pixel_t* ptr = img->get_image_plane_at_pos_NEW<pixel_t>(0, xDi,yDi);
pixel_t q[4][4], p[4][4];
for (int k=0;k<4;k++)
for (int i=0;i<4;i++)
{
if (vertical) {
q[k][i] = ptr[ i +k*stride];
p[k][i] = ptr[-i-1+k*stride];
}
else {
q[k][i] = ptr[k + i *stride];
p[k][i] = ptr[k -(i+1)*stride];
}
}
#if 0
for (int k=0;k<4;k++)
{
for (int i=0;i<4;i++)
{
printf("%02x ", p[k][3-i]);
}
printf("| ");
for (int i=0;i<4;i++)
{
printf("%02x ", q[k][i]);
}
printf("\n");
}
#endif
int QP_Q = img->get_QPY(xDi,yDi);
int QP_P = (vertical ?
img->get_QPY(xDi-1,yDi) :
img->get_QPY(xDi,yDi-1) );
int qP_L = (QP_Q+QP_P+1)>>1;
logtrace(LogDeblock,"QP: %d & %d -> %d\n",QP_Q,QP_P,qP_L);
int sliceIndexQ00 = img->get_SliceHeaderIndex(xDi,yDi);
int beta_offset = img->slices[sliceIndexQ00]->slice_beta_offset;
int tc_offset = img->slices[sliceIndexQ00]->slice_tc_offset;
int Q_beta = Clip3(0,51, qP_L + beta_offset);
int betaPrime = table_8_23_beta[Q_beta];
int beta = betaPrime * (1<<(bitDepth_Y - 8));
int Q_tc = Clip3(0,53, qP_L + 2*(bS-1) + tc_offset);
int tcPrime = table_8_23_tc[Q_tc];
int tc = tcPrime * (1<<(bitDepth_Y - 8));
logtrace(LogDeblock,"beta: %d (%d) tc: %d (%d)\n",beta,beta_offset, tc,tc_offset);
int dE=0, dEp=0, dEq=0;
if (vertical || !vertical) {
int dp0 = abs_value(p[0][2] - 2*p[0][1] + p[0][0]);
int dp3 = abs_value(p[3][2] - 2*p[3][1] + p[3][0]);
int dq0 = abs_value(q[0][2] - 2*q[0][1] + q[0][0]);
int dq3 = abs_value(q[3][2] - 2*q[3][1] + q[3][0]);
int dpq0 = dp0 + dq0;
int dpq3 = dp3 + dq3;
int dp = dp0 + dp3;
int dq = dq0 + dq3;
int d = dpq0+ dpq3;
if (d<beta) {
bool dSam0 = (2*dpq0 < (beta>>2) &&
abs_value(p[0][3]-p[0][0])+abs_value(q[0][0]-q[0][3]) < (beta>>3) &&
abs_value(p[0][0]-q[0][0]) < ((5*tc+1)>>1));
bool dSam3 = (2*dpq3 < (beta>>2) &&
abs_value(p[3][3]-p[3][0])+abs_value(q[3][0]-q[3][3]) < (beta>>3) &&
abs_value(p[3][0]-q[3][0]) < ((5*tc+1)>>1));
if (dSam0 && dSam3) {
dE=2;
}
else {
dE=1;
}
if (dp < ((beta + (beta>>1))>>3)) { dEp=1; }
if (dq < ((beta + (beta>>1))>>3)) { dEq=1; }
logtrace(LogDeblock,"dE:%d dEp:%d dEq:%d\n",dE,dEp,dEq);
}
}
else {
assert(0);
}
if (dE != 0) {
bool filterP = true;
bool filterQ = true;
if (vertical) {
if (sps.pcm_loop_filter_disable_flag && img->get_pcm_flag(xDi-1,yDi)) filterP=false;
if (img->get_cu_transquant_bypass(xDi-1,yDi)) filterP=false;
if (sps.pcm_loop_filter_disable_flag && img->get_pcm_flag(xDi,yDi)) filterQ=false;
if (img->get_cu_transquant_bypass(xDi,yDi)) filterQ=false;
}
else {
if (sps.pcm_loop_filter_disable_flag && img->get_pcm_flag(xDi,yDi-1)) filterP=false;
if (img->get_cu_transquant_bypass(xDi,yDi-1)) filterP=false;
if (sps.pcm_loop_filter_disable_flag && img->get_pcm_flag(xDi,yDi)) filterQ=false;
if (img->get_cu_transquant_bypass(xDi,yDi)) filterQ=false;
}
for (int k=0;k<4;k++) {
logtrace(LogDeblock,"line:%d\n",k);
const pixel_t p0 = p[k][0];
const pixel_t p1 = p[k][1];
const pixel_t p2 = p[k][2];
const pixel_t p3 = p[k][3];
const pixel_t q0 = q[k][0];
const pixel_t q1 = q[k][1];
const pixel_t q2 = q[k][2];
const pixel_t q3 = q[k][3];
if (dE==2) {
pixel_t pnew[3],qnew[3];
pnew[0] = Clip3(p0-2*tc,p0+2*tc, (p2 + 2*p1 + 2*p0 + 2*q0 + q1 +4)>>3);
pnew[1] = Clip3(p1-2*tc,p1+2*tc, (p2 + p1 + p0 + q0+2)>>2);
pnew[2] = Clip3(p2-2*tc,p2+2*tc, (2*p3 + 3*p2 + p1 + p0 + q0 + 4)>>3);
qnew[0] = Clip3(q0-2*tc,q0+2*tc, (p1+2*p0+2*q0+2*q1+q2+4)>>3);
qnew[1] = Clip3(q1-2*tc,q1+2*tc, (p0+q0+q1+q2+2)>>2);
qnew[2] = Clip3(q2-2*tc,q2+2*tc, (p0+q0+q1+3*q2+2*q3+4)>>3);
logtrace(LogDeblock,"strong filtering\n");
if (vertical) {
for (int i=0;i<3;i++) {
if (filterP) { ptr[-i-1+k*stride] = pnew[i]; }
if (filterQ) { ptr[ i + k*stride] = qnew[i]; }
}
}
else {
for (int i=0;i<3;i++) {
if (filterP) { ptr[ k -(i+1)*stride] = pnew[i]; }
if (filterQ) { ptr[ k + i *stride] = qnew[i]; }
}
}
}
else {
int delta = (9*(q0-p0) - 3*(q1-p1) + 8)>>4;
logtrace(LogDeblock,"delta=%d, tc=%d\n",delta,tc);
if (abs_value(delta) < tc*10) {
delta = Clip3(-tc,tc,delta);
logtrace(LogDeblock," deblk + %d;%d [%02x->%02x] - %d;%d [%02x->%02x] delta:%d\n",
vertical ? xDi-1 : xDi+k,
vertical ? yDi+k : yDi-1, p0,Clip_BitDepth(p0+delta, bitDepth_Y),
vertical ? xDi : xDi+k,
vertical ? yDi+k : yDi, q0,Clip_BitDepth(q0-delta, bitDepth_Y),
delta);
if (vertical) {
if (filterP) { ptr[-0-1+k*stride] = Clip_BitDepth(p0+delta, bitDepth_Y); }
if (filterQ) { ptr[ 0 +k*stride] = Clip_BitDepth(q0-delta, bitDepth_Y); }
}
else {
if (filterP) { ptr[ k -1*stride] = Clip_BitDepth(p0+delta, bitDepth_Y); }
if (filterQ) { ptr[ k +0*stride] = Clip_BitDepth(q0-delta, bitDepth_Y); }
}
if (dEp==1 && filterP) {
int delta_p = Clip3(-(tc>>1), tc>>1, (((p2+p0+1)>>1)-p1+delta)>>1);
logtrace(LogDeblock," deblk dEp %d;%d delta:%d\n",
vertical ? xDi-2 : xDi+k,
vertical ? yDi+k : yDi-2,
delta_p);
if (vertical) { ptr[-1-1+k*stride] = Clip_BitDepth(p1+delta_p, bitDepth_Y); }
else { ptr[ k -2*stride] = Clip_BitDepth(p1+delta_p, bitDepth_Y); }
}
if (dEq==1 && filterQ) {
int delta_q = Clip3(-(tc>>1), tc>>1, (((q2+q0+1)>>1)-q1-delta)>>1);
logtrace(LogDeblock," delkb dEq %d;%d delta:%d\n",
vertical ? xDi+1 : xDi+k,
vertical ? yDi+k : yDi+1,
delta_q);
if (vertical) { ptr[ 1 +k*stride] = Clip_BitDepth(q1+delta_q, bitDepth_Y); }
else { ptr[ k +1*stride] = Clip_BitDepth(q1+delta_q, bitDepth_Y); }
}
}
}
}
}
}
}
}
void edge_filtering_luma(de265_image* img, bool vertical,
int yStart,int yEnd, int xStart,int xEnd)
{
if (img->high_bit_depth(0)) {
edge_filtering_luma_internal<uint16_t>(img,vertical,yStart,yEnd,xStart,xEnd);
}
else {
edge_filtering_luma_internal<uint8_t>(img,vertical,yStart,yEnd,xStart,xEnd);
}
}
void edge_filtering_luma_CTB(de265_image* img, bool vertical, int xCtb,int yCtb)
{
int ctbSize = img->get_sps().CtbSizeY;
int deblkSize = ctbSize/4;
edge_filtering_luma(img,vertical,
yCtb*deblkSize, (yCtb+1)*deblkSize,
xCtb*deblkSize, (xCtb+1)*deblkSize);
}
template <class pixel_t>
void edge_filtering_chroma_internal(de265_image* img, bool vertical,
int yStart,int yEnd,
int xStart,int xEnd)
{
const seq_parameter_set& sps = img->get_sps();
const int SubWidthC = sps.SubWidthC;
const int SubHeightC = sps.SubHeightC;
int xIncr = vertical ? 2 : 1;
int yIncr = vertical ? 1 : 2;
xIncr *= SubWidthC;
yIncr *= SubHeightC;
const int stride = img->get_image_stride(1);
xEnd = libde265_min(xEnd,img->get_deblk_width());
yEnd = libde265_min(yEnd,img->get_deblk_height());
int bitDepth_C = sps.BitDepth_C;
for (int y=yStart;y<yEnd;y+=yIncr)
for (int x=xStart;x<xEnd;x+=xIncr) {
int xDi = x << (3-SubWidthC);
int yDi = y << (3-SubHeightC);
int bS = img->get_deblk_bS(xDi*SubWidthC,yDi*SubHeightC);
if (bS>1) {
for (int cplane=0;cplane<2;cplane++) {
int cQpPicOffset = (cplane==0 ?
img->get_pps().pic_cb_qp_offset :
img->get_pps().pic_cr_qp_offset);
pixel_t* ptr = img->get_image_plane_at_pos_NEW<pixel_t>(cplane+1, xDi,yDi);
pixel_t p[2][4];
pixel_t q[2][4];
logtrace(LogDeblock,"-%s- %d %d\n",cplane==0 ? "Cb" : "Cr",xDi,yDi);
for (int i=0;i<2;i++)
for (int k=0;k<4;k++)
{
if (vertical) {
q[i][k] = ptr[ i +k*stride];
p[i][k] = ptr[-i-1+k*stride];
}
else {
q[i][k] = ptr[k + i *stride];
p[i][k] = ptr[k -(i+1)*stride];
}
}
#if 0
for (int k=0;k<4;k++)
{
for (int i=0;i<2;i++)
{
printf("%02x ", p[1-i][k]);
}
printf("| ");
for (int i=0;i<2;i++)
{
printf("%02x ", q[i][k]);
}
printf("\n");
}
#endif
int QP_Q = img->get_QPY(SubWidthC*xDi,SubHeightC*yDi);
int QP_P = (vertical ?
img->get_QPY(SubWidthC*xDi-1,SubHeightC*yDi) :
img->get_QPY(SubWidthC*xDi,SubHeightC*yDi-1));
int qP_i = ((QP_Q+QP_P+1)>>1) + cQpPicOffset;
int QP_C;
if (sps.ChromaArrayType == CHROMA_420) {
QP_C = table8_22(qP_i);
} else {
QP_C = libde265_min(qP_i, 51);
}
logtrace(LogDeblock,"%d %d: ((%d+%d+1)>>1) + %d = qP_i=%d (QP_C=%d)\n",
SubWidthC*xDi,SubHeightC*yDi, QP_Q,QP_P,cQpPicOffset,qP_i,QP_C);
int sliceIndexQ00 = img->get_SliceHeaderIndex(SubWidthC*xDi,SubHeightC*yDi);
int tc_offset = img->slices[sliceIndexQ00]->slice_tc_offset;
int Q = Clip3(0,53, QP_C + 2*(bS-1) + tc_offset);
int tcPrime = table_8_23_tc[Q];
int tc = tcPrime * (1<<(sps.BitDepth_C - 8));
logtrace(LogDeblock,"tc_offset=%d Q=%d tc'=%d tc=%d\n",tc_offset,Q,tcPrime,tc);
if (vertical) {
bool filterP = true;
if (sps.pcm_loop_filter_disable_flag && img->get_pcm_flag(SubWidthC*xDi-1,SubHeightC*yDi)) filterP=false;
if (img->get_cu_transquant_bypass(SubWidthC*xDi-1,SubHeightC*yDi)) filterP=false;
bool filterQ = true;
if (sps.pcm_loop_filter_disable_flag && img->get_pcm_flag(SubWidthC*xDi,SubHeightC*yDi)) filterQ=false;
if (img->get_cu_transquant_bypass(SubWidthC*xDi,SubHeightC*yDi)) filterQ=false;
for (int k=0;k<4;k++) {
int delta = Clip3(-tc,tc, ((((q[0][k]-p[0][k])<<2)+p[1][k]-q[1][k]+4)>>3));
logtrace(LogDeblock,"delta=%d\n",delta);
if (filterP) { ptr[-1+k*stride] = Clip_BitDepth(p[0][k]+delta, bitDepth_C); }
if (filterQ) { ptr[ 0+k*stride] = Clip_BitDepth(q[0][k]-delta, bitDepth_C); }
}
}
else {
bool filterP = true;
if (sps.pcm_loop_filter_disable_flag && img->get_pcm_flag(SubWidthC*xDi,SubHeightC*yDi-1)) filterP=false;
if (img->get_cu_transquant_bypass(SubWidthC*xDi,SubHeightC*yDi-1)) filterP=false;
bool filterQ = true;
if (sps.pcm_loop_filter_disable_flag && img->get_pcm_flag(SubWidthC*xDi,SubHeightC*yDi)) filterQ=false;
if (img->get_cu_transquant_bypass(SubWidthC*xDi,SubHeightC*yDi)) filterQ=false;
for (int k=0;k<4;k++) {
int delta = Clip3(-tc,tc, ((((q[0][k]-p[0][k])<<2)+p[1][k]-q[1][k]+4)>>3));
if (filterP) { ptr[ k-1*stride] = Clip_BitDepth(p[0][k]+delta, bitDepth_C); }
if (filterQ) { ptr[ k+0*stride] = Clip_BitDepth(q[0][k]-delta, bitDepth_C); }
}
}
}
}
}
}
void edge_filtering_chroma(de265_image* img, bool vertical, int yStart,int yEnd,
int xStart,int xEnd)
{
if (img->high_bit_depth(1)) {
edge_filtering_chroma_internal<uint16_t>(img,vertical,yStart,yEnd,xStart,xEnd);
}
else {
edge_filtering_chroma_internal<uint8_t>(img,vertical,yStart,yEnd,xStart,xEnd);
}
}
void edge_filtering_chroma_CTB(de265_image* img, bool vertical, int xCtb,int yCtb)
{
int ctbSize = img->get_sps().CtbSizeY;
int deblkSize = ctbSize/4;
edge_filtering_chroma(img,vertical,
yCtb*deblkSize, (yCtb+1)*deblkSize,
xCtb*deblkSize, (xCtb+1)*deblkSize);
}
class thread_task_deblock_CTBRow : public thread_task
{
public:
struct de265_image* img;
int ctb_y;
bool vertical;
virtual void work();
virtual std::string name() const {
char buf[100];
sprintf(buf,"deblock-%d",ctb_y);
return buf;
}
};
void thread_task_deblock_CTBRow::work()
{
state = Running;
img->thread_run(this);
int xStart=0;
int xEnd = img->get_deblk_width();
int ctbSize = img->get_sps().CtbSizeY;
int deblkSize = ctbSize/4;
int first = ctb_y * deblkSize;
int last = (ctb_y+1) * deblkSize;
if (last > img->get_deblk_height()) {
last = img->get_deblk_height();
}
int finalProgress = CTB_PROGRESS_DEBLK_V;
if (!vertical) finalProgress = CTB_PROGRESS_DEBLK_H;
int rightCtb = img->get_sps().PicWidthInCtbsY-1;
if (vertical) {
int CtbRow = std::min(ctb_y+1 , img->get_sps().PicHeightInCtbsY-1);
img->wait_for_progress(this, rightCtb,CtbRow, CTB_PROGRESS_PREFILTER);
}
else {
if (ctb_y>0) {
img->wait_for_progress(this, rightCtb,ctb_y-1, CTB_PROGRESS_DEBLK_V);
}
img->wait_for_progress(this, rightCtb,ctb_y, CTB_PROGRESS_DEBLK_V);
if (ctb_y+1<img->get_sps().PicHeightInCtbsY) {
img->wait_for_progress(this, rightCtb,ctb_y+1, CTB_PROGRESS_DEBLK_V);
}
}
bool deblocking_enabled;
if (vertical) {
deblocking_enabled = derive_edgeFlags_CTBRow(img, ctb_y);
int x=0; img->set_CtbDeblockFlag(x,ctb_y, deblocking_enabled);
}
else {
int x=0; deblocking_enabled=img->get_CtbDeblockFlag(x,ctb_y);
}
if (deblocking_enabled) {
derive_boundaryStrength(img, vertical, first,last, xStart,xEnd);
edge_filtering_luma(img, vertical, first,last, xStart,xEnd);
if (img->get_sps().ChromaArrayType != CHROMA_MONO) {
edge_filtering_chroma(img, vertical, first,last, xStart,xEnd);
}
}
for (int x=0;x<=rightCtb;x++) {
const int CtbWidth = img->get_sps().PicWidthInCtbsY;
img->ctb_progress[x+ctb_y*CtbWidth].set_progress(finalProgress);
}
state = Finished;
img->thread_finishes(this);
}
void add_deblocking_tasks(image_unit* imgunit)
{
de265_image* img = imgunit->img;
decoder_context* ctx = img->decctx;
int nRows = img->get_sps().PicHeightInCtbsY;
int n=0;
img->thread_start(nRows*2);
for (int pass=0;pass<2;pass++)
{
for (int y=0;y<img->get_sps().PicHeightInCtbsY;y++)
{
thread_task_deblock_CTBRow* task = new thread_task_deblock_CTBRow;
task->img = img;
task->ctb_y = y;
task->vertical = (pass==0);
imgunit->tasks.push_back(task);
add_task(&ctx->thread_pool_, task);
n++;
}
}
}
void apply_deblocking_filter(de265_image* img)
{
decoder_context* ctx = img->decctx;
char enabled_deblocking = derive_edgeFlags(img);
if (enabled_deblocking)
{
logtrace(LogDeblock,"VERTICAL\n");
derive_boundaryStrength(img, true ,0,img->get_deblk_height(),0,img->get_deblk_width());
edge_filtering_luma (img, true ,0,img->get_deblk_height(),0,img->get_deblk_width());
if (img->get_sps().ChromaArrayType != CHROMA_MONO) {
edge_filtering_chroma (img, true ,0,img->get_deblk_height(),0,img->get_deblk_width());
}
#if 0
char buf[1000];
sprintf(buf,"lf-after-V-%05d.yuv", ctx->img->PicOrderCntVal);
write_picture_to_file(ctx->img, buf);
#endif
logtrace(LogDeblock,"HORIZONTAL\n");
derive_boundaryStrength(img, false ,0,img->get_deblk_height(),0,img->get_deblk_width());
edge_filtering_luma (img, false ,0,img->get_deblk_height(),0,img->get_deblk_width());
if (img->get_sps().ChromaArrayType != CHROMA_MONO) {
edge_filtering_chroma (img, false ,0,img->get_deblk_height(),0,img->get_deblk_width());
}
#if 0
sprintf(buf,"lf-after-H-%05d.yuv", ctx->img->PicOrderCntVal);
write_picture_to_file(ctx->img, buf);
#endif
}
}