private void mergePrediction(int[] blk0, int[] blk1, int stride, nVideo.Codecs.H264.H264Const.PartPred p0, int off, int blkW, int blkH, int[] o)
{
for (int i = 0; i < blkH; i++, off += stride - blkW)
{
for (int j = 0; j < blkW; j++, off++)
{
o[off] = (blk0[off] + blk1[off] + 1) >> 1;
}
}
}
private void mergeAvg(int[] blk0, int[] blk1, int stride, nVideo.Codecs.H264.H264Const.PartPred p0, int off, int blkW, int blkH, int[] o)
{
if (p0 == nVideo.Codecs.H264.H264Const.PartPred.Bi)
{
mergePrediction(blk0, blk1, stride, p0, off, blkW, blkH, o);
}
else if (p0 == nVideo.Codecs.H264.H264Const.PartPred.L0)
{
copyPrediction(blk0, stride, off, blkW, blkH, o);
}
else if (p0 == nVideo.Codecs.H264.H264Const.PartPred.L1)
{
copyPrediction(blk1, stride, off, blkW, blkH, o);
}
}
private void mergeWeight(int[] blk0, int[] blk1, int stride, nVideo.Codecs.H264.H264Const.PartPred partPred, int off, int blkW, int blkH,
int logWD, int w0, int w1, int o0, int o1, int[] outs)
{
if (partPred == nVideo.Codecs.H264.H264Const.PartPred.L0)
{
weight(blk0, stride, off, blkW, blkH, logWD, w0, o0, outs);
}
else if (partPred == nVideo.Codecs.H264.H264Const.PartPred.L1)
{
weight(blk1, stride, off, blkW, blkH, logWD, w1, o1, outs);
}
else if (partPred == nVideo.Codecs.H264.H264Const.PartPred.Bi)
{
weightPrediction(blk0, blk1, stride, off, blkW, blkH, logWD, w0, w1, o0, o1, outs);
}
}
public void mergePrediction(int refIdxL0, int refIdxL1, nVideo.Codecs.H264.H264Const.PartPred predType, int comp, int[] pred0, int[] pred1,
int off, int stride, int blkW, int blkH, int[] outs, Frame[][] refs, Frame thisFrame)
{
PictureParameterSet pps = sh.pps;
if (sh.slice_type == SliceType.P)
{
if (pps.weighted_pred_flag && sh.pred_weight_table != null)
{
PredictionWeightTable w = sh.pred_weight_table;
weight(pred0, stride, off, blkW, blkH, comp == 0 ? w.luma_log2_weight_denom
: w.chroma_log2_weight_denom, comp == 0 ? w.luma_weight[0][refIdxL0]
: w.chroma_weight[0][comp - 1][refIdxL0], comp == 0 ? w.luma_offset[0][refIdxL0]
: w.chroma_offset[0][comp - 1][refIdxL0], outs);
}
else
{
copyPrediction(pred0, stride, off, blkW, blkH, outs);
}
}
else
{
if (!pps.weighted_pred_flag || sh.pps.weighted_bipred_idc == 0 ||
(sh.pps.weighted_bipred_idc == 2 && predType != nVideo.Codecs.H264.H264Const.PartPred.Bi))
{
mergeAvg(pred0, pred1, stride, predType, off, blkW, blkH, outs);
}
else if (sh.pps.weighted_bipred_idc == 1)
{
PredictionWeightTable w = sh.pred_weight_table;
int w0 = refIdxL0 == -1 ? 0 : (comp == 0 ? w.luma_weight[0][refIdxL0]
: w.chroma_weight[0][comp - 1][refIdxL0]);
int w1 = refIdxL1 == -1 ? 0 : (comp == 0 ? w.luma_weight[1][refIdxL1]
: w.chroma_weight[1][comp - 1][refIdxL1]);
int o0 = refIdxL0 == -1 ? 0 : (comp == 0 ? w.luma_offset[0][refIdxL0]
: w.chroma_offset[0][comp - 1][refIdxL0]);
int o1 = refIdxL1 == -1 ? 0 : (comp == 0 ? w.luma_offset[1][refIdxL1]
: w.chroma_offset[1][comp - 1][refIdxL1]);
mergeWeight(pred0, pred1, stride, predType, off, blkW, blkH, comp == 0 ? w.luma_log2_weight_denom
: w.chroma_log2_weight_denom, w0, w1, o0, o1, outs);
}
else
{
int tb = CABAC.clip(thisFrame.getPOC() - refs[0][refIdxL0].getPOC(), -128, 127);
int td = CABAC.clip(refs[1][refIdxL1].getPOC() - refs[0][refIdxL0].getPOC(), -128, 127);
int w0 = 32, w1 = 32;
if (td != 0 && refs[0][refIdxL0].isShortTerm() && refs[1][refIdxL1].isShortTerm())
{
int tx = (16384 + Math.Abs(td / 2)) / td;
int dsf = CABAC.clip((tb * tx + 32) >> 6, -1024, 1023) >> 2;
if (dsf >= -64 && dsf <= 128)
{
w1 = dsf;
w0 = 64 - dsf;
}
}
mergeWeight(pred0, pred1, stride, predType, off, blkW, blkH, 5, w0, w1, 0, 0, outs);
}
}
}
public static bool usesList(this nVideo.Codecs.H264.H264Const.PartPred p, int l)
{
return(p == nVideo.Codecs.H264.H264Const.PartPred.Bi ? true : (p == nVideo.Codecs.H264.H264Const.PartPred.L0 && l == 0 || p == nVideo.Codecs.H264.H264Const.PartPred.L1 && l == 1));
}