func (this *TDecGop) DumpMotionField(rpcPic *TLibCommon.TComPic) {
pPicSym := rpcPic.GetPicSym()
if !rpcPic.GetSlice(0).IsIntra() {
fmt.Printf("L0 MV:\n")
for uiCUAddr := uint(0); uiCUAddr < pPicSym.GetFrameHeightInCU()*pPicSym.GetFrameWidthInCU(); uiCUAddr++ {
fmt.Printf("LCU %d\n", uiCUAddr)
pcCU := pPicSym.GetCU(uiCUAddr)
for uiPartIdx := uint(0); uiPartIdx < pcCU.GetTotalNumPart(); uiPartIdx++ {
cMv := pcCU.GetCUMvField(TLibCommon.REF_PIC_LIST_0).GetMv(int(uiPartIdx))
fmt.Printf("(%d,%d) ", cMv.GetHor(), cMv.GetVer())
}
fmt.Printf("\n")
}
fmt.Printf("\n")
}
if rpcPic.GetSlice(0).IsInterB() {
fmt.Printf("L1 MV:\n")
for uiCUAddr := uint(0); uiCUAddr < pPicSym.GetFrameHeightInCU()*pPicSym.GetFrameWidthInCU(); uiCUAddr++ {
fmt.Printf("LCU %d\n", uiCUAddr)
pcCU := pPicSym.GetCU(uiCUAddr)
for uiPartIdx := uint(0); uiPartIdx < pcCU.GetTotalNumPart(); uiPartIdx++ {
cMv := pcCU.GetCUMvField(TLibCommon.REF_PIC_LIST_1).GetMv(int(uiPartIdx))
fmt.Printf("(%d,%d) ", cMv.GetHor(), cMv.GetVer())
}
fmt.Printf("\n")
}
fmt.Printf("\n")
}
}
func (this *TDecSlice) DecompressSlice(ppcSubstreams []*TLibCommon.TComInputBitstream, rpcPic *TLibCommon.TComPic, pcSbacDecoder *TDecSbac, pcSbacDecoders []*TDecSbac) {
//var pcCU *TLibCommon.TComDataCU;
uiIsLast := uint(0)
var iStartCUEncOrder uint
if rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()).GetSliceCurStartCUAddr()/rpcPic.GetNumPartInCU() > rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()).GetSliceSegmentCurStartCUAddr()/rpcPic.GetNumPartInCU() {
iStartCUEncOrder = rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()).GetSliceCurStartCUAddr() / rpcPic.GetNumPartInCU()
} else {
iStartCUEncOrder = rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()).GetSliceSegmentCurStartCUAddr() / rpcPic.GetNumPartInCU()
}
iStartCUAddr := int(rpcPic.GetPicSym().GetCUOrderMap(int(iStartCUEncOrder)))
// decoder don't need prediction & residual frame buffer
rpcPic.SetPicYuvPred(nil)
rpcPic.SetPicYuvResi(nil)
//#if ENC_DEC_TRACE
// g_bJustDoIt = g_bEncDecTraceEnable;
//#endif
// DTRACE_CABAC_VL( g_nSymbolCounter++ );
// DTRACE_CABAC_T( "\tPOC: " );
// DTRACE_CABAC_V( rpcPic.GetPOC() );
// DTRACE_CABAC_T( "\n" );
//#if ENC_DEC_TRACE
// g_bJustDoIt = g_bEncDecTraceDisable;
//#endif
uiTilesAcross := rpcPic.GetPicSym().GetNumColumnsMinus1() + 1
pcSlice := rpcPic.GetSlice(rpcPic.GetCurrSliceIdx())
iNumSubstreams := pcSlice.GetPPS().GetNumSubstreams()
// delete decoders if already allocated in previous slice
/*if (m_pcBufferSbacDecoders)
{
delete [] m_pcBufferSbacDecoders;
}
if (m_pcBufferBinCABACs)
{
delete [] m_pcBufferBinCABACs;
}*/
// allocate new decoders based on tile numbaer
this.m_pcBufferSbacDecoders = make([]*TDecSbac, uiTilesAcross)
this.m_pcBufferBinCABACs = make([]*TDecBinCabac, uiTilesAcross)
for ui := 0; ui < uiTilesAcross; ui++ {
this.m_pcBufferBinCABACs[ui] = NewTDecBinCabac()
this.m_pcBufferSbacDecoders[ui] = NewTDecSbac()
this.m_pcBufferSbacDecoders[ui].Init(this.m_pcBufferBinCABACs[ui])
}
//save init. state
for ui := 0; ui < uiTilesAcross; ui++ {
this.m_pcBufferSbacDecoders[ui].Load(pcSbacDecoder)
}
// free memory if already allocated in previous call
/*if (this.m_pcBufferLowLatSbacDecoders)
{
delete [] this.m_pcBufferLowLatSbacDecoders;
}
if (this.m_pcBufferLowLatBinCABACs)
{
delete [] this.m_pcBufferLowLatBinCABACs;
}*/
this.m_pcBufferLowLatSbacDecoders = make([]*TDecSbac, uiTilesAcross)
this.m_pcBufferLowLatBinCABACs = make([]*TDecBinCabac, uiTilesAcross)
for ui := 0; ui < uiTilesAcross; ui++ {
this.m_pcBufferLowLatBinCABACs[ui] = NewTDecBinCabac()
this.m_pcBufferLowLatSbacDecoders[ui] = NewTDecSbac()
this.m_pcBufferLowLatSbacDecoders[ui].Init(this.m_pcBufferLowLatBinCABACs[ui])
}
//save init. state
for ui := 0; ui < uiTilesAcross; ui++ {
this.m_pcBufferLowLatSbacDecoders[ui].Load(pcSbacDecoder)
}
uiWidthInLCUs := rpcPic.GetPicSym().GetFrameWidthInCU()
//UInt uiHeightInLCUs = rpcPic.GetPicSym().GetFrameHeightInCU();
uiCol := uint(0)
uiLin := uint(0)
uiSubStrm := uint(0)
var uiTileCol, uiTileStartLCU, uiTileLCUX uint
iNumSubstreamsPerTile := 1 // if independent.
depSliceSegmentsEnabled := rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()).GetPPS().GetDependentSliceSegmentsEnabledFlag()
uiTileStartLCU = rpcPic.GetPicSym().GetTComTile(rpcPic.GetPicSym().GetTileIdxMap(iStartCUAddr)).GetFirstCUAddr()
if depSliceSegmentsEnabled {
if (!rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()).IsNextSlice()) && iStartCUAddr != int(rpcPic.GetPicSym().GetTComTile(rpcPic.GetPicSym().GetTileIdxMap(iStartCUAddr)).GetFirstCUAddr()) {
if pcSlice.GetPPS().GetEntropyCodingSyncEnabledFlag() {
uiTileCol = rpcPic.GetPicSym().GetTileIdxMap(iStartCUAddr) % uint(rpcPic.GetPicSym().GetNumColumnsMinus1()+1)
this.m_pcBufferSbacDecoders[uiTileCol].LoadContexts(this.CTXMem[1]) //2.LCU
if (uint(iStartCUAddr)%uiWidthInLCUs + 1) >= uiWidthInLCUs {
uiTileLCUX = uiTileStartLCU % uiWidthInLCUs
uiCol = uint(iStartCUAddr) % uiWidthInLCUs
if uiCol == uiTileLCUX {
this.CTXMem[0].LoadContexts(pcSbacDecoder)
}
}
}
pcSbacDecoder.LoadContexts(this.CTXMem[0]) //end of depSlice-1
pcSbacDecoders[uiSubStrm].LoadContexts(pcSbacDecoder)
} else {
if pcSlice.GetPPS().GetEntropyCodingSyncEnabledFlag() {
this.CTXMem[1].LoadContexts(pcSbacDecoder)
}
this.CTXMem[0].LoadContexts(pcSbacDecoder)
}
}
for iCUAddr := iStartCUAddr; uiIsLast == 0 && iCUAddr < int(rpcPic.GetNumCUsInFrame()); iCUAddr = int(rpcPic.GetPicSym().XCalculateNxtCUAddr(uint(iCUAddr))) {
pcCU := rpcPic.GetCU(uint(iCUAddr))
pcCU.InitCU(rpcPic, uint(iCUAddr))
fmt.Printf("%d ", iCUAddr)
//#ifdef ENC_DEC_TRACE
pcSbacDecoder.XTraceLCUHeader(TLibCommon.TRACE_LCU)
pcSbacDecoder.XReadAeTr(iCUAddr, "lcu_address", TLibCommon.TRACE_LCU)
pcSbacDecoder.XReadAeTr(int(rpcPic.GetPicSym().GetTileIdxMap(iCUAddr)), "tile_id", TLibCommon.TRACE_LCU)
//#endif
uiTileCol = rpcPic.GetPicSym().GetTileIdxMap(int(iCUAddr)) % uint(rpcPic.GetPicSym().GetNumColumnsMinus1()+1) // what column of tiles are we in?
uiTileStartLCU = rpcPic.GetPicSym().GetTComTile(rpcPic.GetPicSym().GetTileIdxMap(int(iCUAddr))).GetFirstCUAddr()
uiTileLCUX = uiTileStartLCU % uiWidthInLCUs
uiCol = uint(iCUAddr) % uiWidthInLCUs
// The 'line' is now relative to the 1st line in the slice, not the 1st line in the picture.
uiLin = (uint(iCUAddr) / uiWidthInLCUs) - (uint(iStartCUAddr) / uiWidthInLCUs)
// inherit from TR if necessary, select substream to use.
if (pcSlice.GetPPS().GetNumSubstreams() > 1) || (depSliceSegmentsEnabled && (uiCol == uiTileLCUX) && (pcSlice.GetPPS().GetEntropyCodingSyncEnabledFlag())) {
// independent tiles => substreams are "per tile". iNumSubstreams has already been multiplied.
iNumSubstreamsPerTile = iNumSubstreams / rpcPic.GetPicSym().GetNumTiles()
uiSubStrm = rpcPic.GetPicSym().GetTileIdxMap(iCUAddr)*uint(iNumSubstreamsPerTile) + uiLin%uint(iNumSubstreamsPerTile)
this.m_pcEntropyDecoder.SetBitstream(ppcSubstreams[uiSubStrm])
// Synchronize cabac probabilities with upper-right LCU if it's available and we're at the start of a line.
if ((pcSlice.GetPPS().GetNumSubstreams() > 1) || depSliceSegmentsEnabled) && (uiCol == uiTileLCUX) && (pcSlice.GetPPS().GetEntropyCodingSyncEnabledFlag()) {
// We'll sync if the TR is available.
pcCUUp := pcCU.GetCUAbove()
uiWidthInCU := rpcPic.GetFrameWidthInCU()
var pcCUTR *TLibCommon.TComDataCU
if pcCUUp != nil && ((uint(iCUAddr)%uiWidthInCU + 1) < uiWidthInCU) {
pcCUTR = rpcPic.GetCU(uint(iCUAddr) - uiWidthInCU + 1)
}
uiMaxParts := uint(1 << (pcSlice.GetSPS().GetMaxCUDepth() << 1))
if true && //bEnforceSliceRestriction
((pcCUTR == nil) || (pcCUTR.GetSlice() == nil) ||
((pcCUTR.GetSCUAddr() + uiMaxParts - 1) < pcSlice.GetSliceCurStartCUAddr()) ||
(rpcPic.GetPicSym().GetTileIdxMap(int(pcCUTR.GetAddr())) != rpcPic.GetPicSym().GetTileIdxMap(iCUAddr))) {
// TR not available.
} else {
// TR is available, we use it.
pcSbacDecoders[uiSubStrm].LoadContexts(this.m_pcBufferSbacDecoders[uiTileCol])
}
}
pcSbacDecoder.Load(pcSbacDecoders[uiSubStrm]) //this load is used to simplify the code (avoid to change all the call to pcSbacDecoders)
} else if pcSlice.GetPPS().GetNumSubstreams() <= 1 {
// Set variables to appropriate values to avoid later code change.
iNumSubstreamsPerTile = 1
}
if (uint(iCUAddr) == rpcPic.GetPicSym().GetTComTile(rpcPic.GetPicSym().GetTileIdxMap(iCUAddr)).GetFirstCUAddr()) && // 1st in tile.
(iCUAddr != 0) && (uint(iCUAddr) != rpcPic.GetPicSym().GetPicSCUAddr(rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()).GetSliceCurStartCUAddr())/rpcPic.GetNumPartInCU()) &&
(uint(iCUAddr) != rpcPic.GetPicSym().GetPicSCUAddr(rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()).GetSliceSegmentCurStartCUAddr())/rpcPic.GetNumPartInCU()) {
// !1st in frame && !1st in slice
if pcSlice.GetPPS().GetNumSubstreams() > 1 {
// We're crossing into another tile, tiles are independent.
// When tiles are independent, we have "substreams per tile". Each substream has already been terminated, and we no longer
// have to perform it here.
// For TILES_DECODER, there can be a header at the start of the 1st substream in a tile. These are read when the substreams
// are extracted, not here.
} else {
sliceType := pcSlice.GetSliceType()
if pcSlice.GetCabacInitFlag() {
switch sliceType {
case TLibCommon.P_SLICE: // change initialization table to B_SLICE intialization
sliceType = TLibCommon.B_SLICE
//break;
case TLibCommon.B_SLICE: // change initialization table to P_SLICE intialization
sliceType = TLibCommon.P_SLICE
//break;
//default : // should not occur
//assert(0);
}
}
this.m_pcEntropyDecoder.UpdateContextTables(sliceType, pcSlice.GetSliceQp())
}
}
//#if ENC_DEC_TRACE
// g_bJustDoIt = g_bEncDecTraceEnable;
//#endif
if pcSlice.GetSPS().GetUseSAO() && (pcSlice.GetSaoEnabledFlag() || pcSlice.GetSaoEnabledFlagChroma()) {
saoParam := rpcPic.GetPicSym().GetSaoParam()
saoParam.SaoFlag[0] = pcSlice.GetSaoEnabledFlag()
if iCUAddr == iStartCUAddr {
saoParam.SaoFlag[1] = pcSlice.GetSaoEnabledFlagChroma()
}
numCuInWidth := saoParam.NumCuInWidth
cuAddrInSlice := iCUAddr - int(rpcPic.GetPicSym().GetCUOrderMap(int(pcSlice.GetSliceCurStartCUAddr()/rpcPic.GetNumPartInCU())))
cuAddrUpInSlice := cuAddrInSlice - numCuInWidth
rx := iCUAddr % numCuInWidth
ry := iCUAddr / numCuInWidth
allowMergeLeft := true
allowMergeUp := true
if rx != 0 {
if rpcPic.GetPicSym().GetTileIdxMap(iCUAddr-1) != rpcPic.GetPicSym().GetTileIdxMap(iCUAddr) {
allowMergeLeft = false
}
}
if ry != 0 {
if rpcPic.GetPicSym().GetTileIdxMap(iCUAddr-numCuInWidth) != rpcPic.GetPicSym().GetTileIdxMap(iCUAddr) {
allowMergeUp = false
}
}
pcSbacDecoder.ParseSaoOneLcuInterleaving(rx, ry, saoParam, pcCU, cuAddrInSlice, cuAddrUpInSlice, allowMergeLeft, allowMergeUp)
} else if pcSlice.GetSPS().GetUseSAO() {
addr := pcCU.GetAddr()
saoParam := rpcPic.GetPicSym().GetSaoParam()
for cIdx := 0; cIdx < 3; cIdx++ {
saoLcuParam := &(saoParam.SaoLcuParam[cIdx][addr])
if ((cIdx == 0) && !pcSlice.GetSaoEnabledFlag()) || ((cIdx == 1 || cIdx == 2) && !pcSlice.GetSaoEnabledFlagChroma()) {
saoLcuParam.MergeUpFlag = false
saoLcuParam.MergeLeftFlag = false
saoLcuParam.SubTypeIdx = 0
saoLcuParam.TypeIdx = -1
saoLcuParam.Offset[0] = 0
saoLcuParam.Offset[1] = 0
saoLcuParam.Offset[2] = 0
saoLcuParam.Offset[3] = 0
}
}
}
this.m_pcCuDecoder.DecodeCU(pcCU, &uiIsLast)
this.m_pcCuDecoder.DecompressCU(pcCU)
//#if ENC_DEC_TRACE
// g_bJustDoIt = g_bEncDecTraceDisable;
//#endif
pcSbacDecoders[uiSubStrm].Load(pcSbacDecoder)
//Store probabilities of second LCU in line into buffer
if (uiCol == uiTileLCUX+1) && (depSliceSegmentsEnabled || (pcSlice.GetPPS().GetNumSubstreams() > 1)) && (pcSlice.GetPPS().GetEntropyCodingSyncEnabledFlag()) {
this.m_pcBufferSbacDecoders[uiTileCol].LoadContexts(pcSbacDecoders[uiSubStrm])
}
if uiIsLast != 0 && depSliceSegmentsEnabled {
if pcSlice.GetPPS().GetEntropyCodingSyncEnabledFlag() {
this.CTXMem[1].LoadContexts(this.m_pcBufferSbacDecoders[uiTileCol]) //ctx 2.LCU
}
this.CTXMem[0].LoadContexts(pcSbacDecoder) //ctx end of dep.slice
return
}
}
return
}
func (this *TDecGop) FilterPicture(rpcPic *TLibCommon.TComPic) {
pcSlice := rpcPic.GetSlice(rpcPic.GetCurrSliceIdx())
//-- For time output for each slice
iBeforeTime := time.Now()
// deblocking filter
bLFCrossTileBoundary := pcSlice.GetPPS().GetLoopFilterAcrossTilesEnabledFlag()
this.m_pcLoopFilter.SetCfg(bLFCrossTileBoundary)
this.m_pcLoopFilter.LoopFilterPic(rpcPic)
if pcSlice.GetSPS().GetUseSAO() {
l := len(this.m_sliceStartCUAddress)
this.m_sliceStartCUAddress[l] = int(rpcPic.GetNumCUsInFrame() * rpcPic.GetNumPartInCU())
///this.m_sliceStartCUAddress.PushBack(rpcPic.GetNumCUsInFrame()* rpcPic.GetNumPartInCU());
rpcPic.CreateNonDBFilterInfo(this.m_sliceStartCUAddress, 0, this.m_LFCrossSliceBoundaryFlag, rpcPic.GetPicSym().GetNumTiles(), bLFCrossTileBoundary)
}
if pcSlice.GetSPS().GetUseSAO() {
saoParam := rpcPic.GetPicSym().GetSaoParam()
saoParam.SaoFlag[0] = pcSlice.GetSaoEnabledFlag()
saoParam.SaoFlag[1] = pcSlice.GetSaoEnabledFlagChroma()
this.m_pcSAO.SetSaoLcuBasedOptimization(true)
this.m_pcSAO.CreatePicSaoInfo(rpcPic) //, len(this.m_sliceStartCUAddress)-1)
this.m_pcSAO.SAOProcess(saoParam)
this.m_pcSAO.PCMLFDisableProcess(rpcPic)
this.m_pcSAO.DestroyPicSaoInfo()
}
if pcSlice.GetSPS().GetUseSAO() {
rpcPic.DestroyNonDBFilterInfo()
}
rpcPic.CompressMotion()
//this.DumpMotionField(rpcPic);
var c string
if pcSlice.IsIntra() {
c = "I"
} else if pcSlice.IsInterP() {
if pcSlice.IsReferenced() {
c = "P"
} else {
c = "p"
}
} else {
if pcSlice.IsReferenced() {
c = "B"
} else {
c = "b"
}
}
//-- For time output for each slice
fmt.Printf("\nPOC %4d TId: %1d ( %s-SLICE, QP%3d ) ", pcSlice.GetPOC(), pcSlice.GetTLayer(), c, pcSlice.GetSliceQp())
this.m_dDecTime += time.Now().Sub(iBeforeTime)
fmt.Printf("[DT %10v] ", this.m_dDecTime)
this.m_dDecTime = 0
for iRefList := 0; iRefList < 2; iRefList++ {
fmt.Printf("[L%d ", iRefList)
for iRefIndex := 0; iRefIndex < pcSlice.GetNumRefIdx(TLibCommon.RefPicList(iRefList)); iRefIndex++ {
fmt.Printf("%d ", pcSlice.GetRefPOC(TLibCommon.RefPicList(iRefList), iRefIndex))
}
fmt.Printf("] ")
}
if this.m_decodedPictureHashSEIEnabled > 0 {
this.CalcAndPrintHashStatus(rpcPic.GetPicYuvRec(), rpcPic.GetSEIs())
}
rpcPic.SetOutputMark(true)
rpcPic.SetReconMark(true)
//this.m_sliceStartCUAddress.Init();
//this.m_LFCrossSliceBoundaryFlag.Init();
slicesize := len(this.m_sliceStartCUAddress)
for i := 0; i < slicesize; i++ {
delete(this.m_sliceStartCUAddress, i)
}
if len(this.m_sliceStartCUAddress) != 0 {
fmt.Printf("clear this.m_sliceStartCUAddress error\n")
}
lfsize := len(this.m_LFCrossSliceBoundaryFlag)
for i := 0; i < lfsize; i++ {
delete(this.m_LFCrossSliceBoundaryFlag, i)
}
if len(this.m_LFCrossSliceBoundaryFlag) != 0 {
fmt.Printf("clear this.m_LFCrossSliceBoundaryFlag error\n")
}
}