func (this *TDecGop) DecompressSlice(pcBitstream *TLibCommon.TComInputBitstream, rpcPic *TLibCommon.TComPic, pTraceFile io.Writer) { pcSlice := rpcPic.GetSlice(rpcPic.GetCurrSliceIdx()) // Table of extracted substreams. // These must be deallocated AND their internal fifos, too. //TComInputBitstream **ppcSubstreams = NULL; //-- For time output for each slice iBeforeTime := time.Now() uiStartCUAddr := pcSlice.GetSliceSegmentCurStartCUAddr() uiSliceStartCuAddr := pcSlice.GetSliceCurStartCUAddr() if uiSliceStartCuAddr == uiStartCUAddr { l := len(this.m_sliceStartCUAddress) this.m_sliceStartCUAddress[l] = int(uiSliceStartCuAddr) //this.m_sliceStartCUAddress.PushBack(uiSliceStartCuAddr); } this.m_pcSbacDecoder.Init(this.m_pcBinCABAC) //(TDecBinIf*) this.m_pcEntropyDecoder.SetEntropyDecoder(this.m_pcSbacDecoder) this.m_pcEntropyDecoder.SetTraceFile(pTraceFile) var uiNumSubstreams uint if pcSlice.GetPPS().GetEntropyCodingSyncEnabledFlag() { uiNumSubstreams = uint(pcSlice.GetNumEntryPointOffsets() + 1) } else { uiNumSubstreams = uint(pcSlice.GetPPS().GetNumSubstreams()) } // init each couple {EntropyDecoder, Substream} puiSubstreamSizes := pcSlice.GetSubstreamSizes() ppcSubstreams := make([]*TLibCommon.TComInputBitstream, uiNumSubstreams) this.m_pcSbacDecoders = make([]*TDecSbac, uiNumSubstreams) this.m_pcBinCABACs = make([]*TDecBinCabac, uiNumSubstreams) for ui := uint(0); ui < uiNumSubstreams; ui++ { this.m_pcSbacDecoders[ui] = NewTDecSbac() this.m_pcBinCABACs[ui] = NewTDecBinCabac() this.m_pcSbacDecoders[ui].Init(this.m_pcBinCABACs[ui]) if ui+1 < uiNumSubstreams { ppcSubstreams[ui] = pcBitstream.ExtractSubstream(puiSubstreamSizes[ui]) } else { ppcSubstreams[ui] = pcBitstream.ExtractSubstream(pcBitstream.GetNumBitsLeft()) } } for ui := uint(0); ui+1 < uiNumSubstreams; ui++ { this.m_pcEntropyDecoder.SetEntropyDecoder(this.m_pcSbacDecoders[uiNumSubstreams-1-ui]) this.m_pcEntropyDecoder.SetTraceFile(pTraceFile) this.m_pcEntropyDecoder.SetBitstream(ppcSubstreams[uiNumSubstreams-1-ui]) this.m_pcEntropyDecoder.ResetEntropy(pcSlice) } this.m_pcEntropyDecoder.SetEntropyDecoder(this.m_pcSbacDecoder) this.m_pcEntropyDecoder.SetTraceFile(pTraceFile) this.m_pcEntropyDecoder.SetBitstream(ppcSubstreams[0]) this.m_pcEntropyDecoder.ResetEntropy(pcSlice) if uiSliceStartCuAddr == uiStartCUAddr { l := len(this.m_LFCrossSliceBoundaryFlag) this.m_LFCrossSliceBoundaryFlag[l] = pcSlice.GetLFCrossSliceBoundaryFlag() //this.m_LFCrossSliceBoundaryFlag.PushBack( pcSlice.GetLFCrossSliceBoundaryFlag()); } this.m_pcSbacDecoders[0].Load(this.m_pcSbacDecoder) this.m_pcSliceDecoder.DecompressSlice(ppcSubstreams, rpcPic, this.m_pcSbacDecoder, this.m_pcSbacDecoders) this.m_pcEntropyDecoder.SetBitstream(ppcSubstreams[uiNumSubstreams-1]) // deallocate all created substreams, including internal buffers. /*for ui := uint(0); ui < uiNumSubstreams; ui++ { ppcSubstreams[ui]->deleteFifo(); delete ppcSubstreams[ui]; } delete[] ppcSubstreams; delete[] m_pcSbacDecoders; delete[] m_pcBinCABACs; */ this.m_pcSbacDecoders = nil this.m_pcBinCABACs = nil lAfterTime := time.Now() this.m_dDecTime += lAfterTime.Sub(iBeforeTime) }
func (this *InputNALUnit) convertPayloadToRBSP(nalUnitBuf *list.List, pcBitstream *TLibCommon.TComInputBitstream, isVclNalUnit bool) *list.List { zeroCount := 0 it_write := list.New() oldBuf := list.New() pos := uint(0) pcBitstream.ClearEmulationPreventionByteLocation() for e := nalUnitBuf.Front(); e != nil; e, pos = e.Next(), pos+1 { //assert(zeroCount < 2 || *it_read >= 0x03); it_read := e.Value.(byte) oldBuf.PushBack(it_read) if zeroCount == 2 && it_read == 0x03 { pcBitstream.PushEmulationPreventionByteLocation(pos) pos++ zeroCount = 0 e = e.Next() if e == nil { break } else { it_read = e.Value.(byte) oldBuf.PushBack(it_read) } } if it_read == 0x00 { zeroCount++ } else { zeroCount = 0 } it_write.PushBack(it_read) } //assert(zeroCount == 0); if isVclNalUnit { // Remove cabac_zero_word from payload if present n := 0 e := it_write.Back() it_read := e.Value.(byte) for it_read == 0x00 { it_write.Remove(e) n++ e = it_write.Back() if e != nil { it_read = e.Value.(byte) } else { break } } if n > 0 { fmt.Printf("\nDetected %d instances of cabac_zero_word", n/2) } } nalUnitBuf.Init() // = .resize(it_write - nalUnitBuf.begin()); for e := it_write.Front(); e != nil; e = e.Next() { it_read := e.Value.(byte) nalUnitBuf.PushBack(it_read) } return oldBuf }