/* The copyright in this software is being made available under the BSD * License, included below. This software may be subject to other third party * and contributor rights, including patent rights, and no such rights are * granted under this license. * * Copyright (c) 2010-2015, ITU/ISO/IEC * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * Neither the name of the ITU/ISO/IEC nor the names of its contributors may * be used to endorse or promote products derived from this software without * specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF * THE POSSIBILITY OF SUCH DAMAGE. */ /** \file TEncGOP.h \brief GOP encoder class (header) */ #ifndef __TENCGOP__ #define __TENCGOP__ #include #include #include "TLibCommon/TComList.h" #include "TLibCommon/TComPic.h" #include "TLibCommon/TComBitCounter.h" #include "TLibCommon/TComLoopFilter.h" #include "TLibCommon/AccessUnit.h" #include "TEncSampleAdaptiveOffset.h" #include "TEncSlice.h" #include "TEncEntropy.h" #include "TEncCavlc.h" #include "TEncSbac.h" #include "SEIwrite.h" #include "SEIEncoder.h" #include "TEncAnalyze.h" #include "TEncRateCtrl.h" #if KWU_RC_MADPRED_E0227 #include "../App/TAppEncoder/TAppEncTop.h" #endif #include //! \ingroup TLibEncoder //! \{ class TEncTop; // ==================================================================================================================== // Class definition // ==================================================================================================================== class TEncGOP { class DUData { public: DUData() :accumBitsDU(0) ,accumNalsDU(0) {}; Int accumBitsDU; Int accumNalsDU; }; private: TEncAnalyze m_gcAnalyzeAll; TEncAnalyze m_gcAnalyzeI; TEncAnalyze m_gcAnalyzeP; TEncAnalyze m_gcAnalyzeB; TEncAnalyze m_gcAnalyzeAll_in; // Data Bool m_bLongtermTestPictureHasBeenCoded; Bool m_bLongtermTestPictureHasBeenCoded2; UInt m_numLongTermRefPicSPS; UInt m_ltRefPicPocLsbSps[MAX_NUM_LONG_TERM_REF_PICS]; Bool m_ltRefPicUsedByCurrPicFlag[MAX_NUM_LONG_TERM_REF_PICS]; Int m_iLastIDR; Int m_iGopSize; Int m_iNumPicCoded; Bool m_bFirst; Int m_iLastRecoveryPicPOC; // Access channel TEncTop* m_pcEncTop; TEncCfg* m_pcCfg; TEncSlice* m_pcSliceEncoder; TComList* m_pcListPic; TEncEntropy* m_pcEntropyCoder; TEncCavlc* m_pcCavlcCoder; TEncSbac* m_pcSbacCoder; TEncBinCABAC* m_pcBinCABAC; TComLoopFilter* m_pcLoopFilter; SEIWriter m_seiWriter; #if NH_MV TComPicLists* m_ivPicLists; std::vector m_refPicSetInterLayer0; std::vector m_refPicSetInterLayer1; Int m_pocLastCoded; Int m_layerId; Int m_viewId; #if NH_3D Int m_viewIndex; Bool m_isDepth; #endif #endif #if NH_3D_IC Int* m_aICEnableCandidate; Int* m_aICEnableNum; #endif //--Adaptive Loop filter TEncSampleAdaptiveOffset* m_pcSAO; TEncRateCtrl* m_pcRateCtrl; // indicate sequence first Bool m_bSeqFirst; // clean decoding refresh Bool m_bRefreshPending; Int m_pocCRA; NalUnitType m_associatedIRAPType; Int m_associatedIRAPPOC; std::vector m_vRVM_RP; UInt m_lastBPSEI; UInt m_totalCoded; Bool m_bufferingPeriodSEIPresentInAU; SEIEncoder m_seiEncoder; public: TEncGOP(); virtual ~TEncGOP(); Void create (); Void destroy (); Void init ( TEncTop* pcTEncTop ); #if NH_MV Void initGOP ( Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRecOut, std::list& accessUnitsInGOP); Void compressPicInGOP ( Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRec, std::list& accessUnitsInGOP, Bool isField, Bool isTff, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE, Int iGOPid ); #else Void compressGOP ( Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRec, std::list& accessUnitsInGOP, Bool isField, Bool isTff, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE ); #endif Void xAttachSliceDataToNalUnit (OutputNALUnit& rNalu, TComOutputBitstream* pcBitstreamRedirect); #if NH_MV Int getPocLastCoded () { return m_pocLastCoded; } Int getLayerId () { return m_layerId; } Int getViewId () { return m_viewId; } #if NH_3D Int getViewIndex () { return m_viewIndex; } Bool getIsDepth () { return m_isDepth; } #endif #endif Int getGOPSize() { return m_iGopSize; } TComList* getListPic() { return m_pcListPic; } Void printOutSummary ( UInt uiNumAllPicCoded, Bool isField, const Bool printMSEBasedSNR, const Bool printSequenceMSE, const BitDepths &bitDepths ); #if NH_3D_VSO Void preLoopFilterPicAll ( TComPic* pcPic, Dist64& ruiDist); #else Void preLoopFilterPicAll ( TComPic* pcPic, UInt64& ruiDist ); #endif #if KWU_RC_MADPRED_E0227 TEncTop* getEncTop() { return m_pcEncTop; } #endif TEncSlice* getSliceEncoder() { return m_pcSliceEncoder; } NalUnitType getNalUnitType( Int pocCurr, Int lastIdr, Bool isField ); Void arrangeLongtermPicturesInRPS(TComSlice *, TComList& ); protected: TEncRateCtrl* getRateCtrl() { return m_pcRateCtrl; } protected: Void xInitGOP ( Int iPOCLast, Int iNumPicRcvd, Bool isField ); Void xGetBuffer ( TComList& rcListPic, TComList& rcListPicYuvRecOut, Int iNumPicRcvd, Int iTimeOffset, TComPic*& rpcPic, TComPicYuv*& rpcPicYuvRecOut, Int pocCurr, Bool isField ); Void xCalculateAddPSNRs ( const Bool isField, const Bool isFieldTopFieldFirst, const Int iGOPid, TComPic* pcPic, const AccessUnit&accessUnit, TComList &rcListPic, Double dEncTime, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE ); Void xCalculateAddPSNR ( TComPic* pcPic, TComPicYuv* pcPicD, const AccessUnit&, Double dEncTime, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE ); Void xCalculateInterlacedAddPSNR( TComPic* pcPicOrgFirstField, TComPic* pcPicOrgSecondField, TComPicYuv* pcPicRecFirstField, TComPicYuv* pcPicRecSecondField, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE ); #if NH_3D_VSO Dist64 xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1, const BitDepths &bitDepths); #else UInt64 xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1, const BitDepths &bitDepths); #endif Double xCalculateRVM(); Void xCreateIRAPLeadingSEIMessages (SEIMessages& seiMessages, const TComSPS *sps, const TComPPS *pps); Void xCreatePerPictureSEIMessages (Int picInGOP, SEIMessages& seiMessages, SEIMessages& nestedSeiMessages, TComSlice *slice); Void xCreatePictureTimingSEI (Int IRAPGOPid, SEIMessages& seiMessages, SEIMessages& nestedSeiMessages, SEIMessages& duInfoSeiMessages, TComSlice *slice, Bool isField, std::deque &duData); Void xUpdateDuData(AccessUnit &testAU, std::deque &duData); Void xUpdateTimingSEI(SEIPictureTiming *pictureTimingSEI, std::deque &duData, const TComSPS *sps); Void xUpdateDuInfoSEI(SEIMessages &duInfoSeiMessages, SEIPictureTiming *pictureTimingSEI); Void xCreateScalableNestingSEI (SEIMessages& seiMessages, SEIMessages& nestedSeiMessages); Void xWriteSEI (NalUnitType naluType, SEIMessages& seiMessages, AccessUnit &accessUnit, AccessUnit::iterator &auPos, Int temporalId, const TComSPS *sps); Void xWriteSEISeparately (NalUnitType naluType, SEIMessages& seiMessages, AccessUnit &accessUnit, AccessUnit::iterator &auPos, Int temporalId, const TComSPS *sps); Void xClearSEIs(SEIMessages& seiMessages, Bool deleteMessages); Void xWriteLeadingSEIOrdered (SEIMessages& seiMessages, SEIMessages& duInfoSeiMessages, AccessUnit &accessUnit, Int temporalId, const TComSPS *sps, Bool testWrite); Void xWriteLeadingSEIMessages (SEIMessages& seiMessages, SEIMessages& duInfoSeiMessages, AccessUnit &accessUnit, Int temporalId, const TComSPS *sps, std::deque &duData); Void xWriteTrailingSEIMessages (SEIMessages& seiMessages, AccessUnit &accessUnit, Int temporalId, const TComSPS *sps); Void xWriteDuSEIMessages (SEIMessages& duInfoSeiMessages, AccessUnit &accessUnit, Int temporalId, const TComSPS *sps, std::deque &duData); Int xWriteVPS (AccessUnit &accessUnit, const TComVPS *vps); Int xWriteSPS (AccessUnit &accessUnit, const TComSPS *sps); Int xWritePPS (AccessUnit &accessUnit, const TComPPS *pps); Int xWriteParameterSets (AccessUnit &accessUnit, TComSlice *slice); #if NH_MV Void xSetRefPicListModificationsMv( std::vector tempPicLists[2], TComSlice* pcSlice, UInt iGOPid ); #endif Void applyDeblockingFilterMetric( TComPic* pcPic, UInt uiNumSlices ); };// END CLASS DEFINITION TEncGOP //! \} #endif // __TENCGOP__