/* 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-2013, 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 "TEncAnalyze.h" #include "TEncRateCtrl.h" #if KWU_RC_MADPRED_E0227 #include "../App/TAppEncoder/TAppEncTop.h" #endif #include //! \ingroup TLibEncoder //! \{ class TEncTop; // ==================================================================================================================== // Class definition // ==================================================================================================================== /// GOP encoder class class TEncGOP { private: // Data Bool m_bLongtermTestPictureHasBeenCoded; Bool m_bLongtermTestPictureHasBeenCoded2; UInt m_numLongTermRefPicSPS; UInt m_ltRefPicPocLsbSps[33]; Bool m_ltRefPicUsedByCurrPicFlag[33]; Int m_iLastIDR; Int m_iGopSize; Int m_iNumPicCoded; Bool m_bFirst; // 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 H_MV TComPicLists* m_ivPicLists; std::vector m_refPicSetInterLayer0; std::vector m_refPicSetInterLayer1; Int m_pocLastCoded; Int m_layerId; Int m_viewId; #if H_3D Int m_viewIndex; Bool m_isDepth; #endif #endif //--Adaptive Loop filter TEncSampleAdaptiveOffset* m_pcSAO; TComBitCounter* m_pcBitCounter; TEncRateCtrl* m_pcRateCtrl; // indicate sequence first Bool m_bSeqFirst; // clean decoding refresh Bool m_bRefreshPending; Int m_pocCRA; std::vector m_storedStartCUAddrForEncodingSlice; std::vector m_storedStartCUAddrForEncodingSliceSegment; std::vector m_vRVM_RP; UInt m_lastBPSEI; UInt m_totalCoded; UInt m_cpbRemovalDelay; UInt m_tl0Idx; UInt m_rapIdx; Bool m_activeParameterSetSEIPresentInAU; Bool m_bufferingPeriodSEIPresentInAU; Bool m_pictureTimingSEIPresentInAU; Bool m_nestedBufferingPeriodSEIPresentInAU; Bool m_nestedPictureTimingSEIPresentInAU; public: TEncGOP(); virtual ~TEncGOP(); Void create (); Void destroy (); Void init ( TEncTop* pcTEncTop ); #if H_MV Void initGOP ( Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRecOut, std::list& accessUnitsInGOP); Void compressPicInGOP ( Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRecOut, std::list& accessUnitsInGOP, Int iGOPid, Bool isField, Bool isTff ); #else Void compressGOP( Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRec, std::list& accessUnitsInGOP, Bool isField, Bool isTff ); #endif Void xAttachSliceDataToNalUnit (OutputNALUnit& rNalu, TComOutputBitstream*& rpcBitstreamRedirect); #if H_MV Int getPocLastCoded () { return m_pocLastCoded; } Int getLayerId () { return m_layerId; } Int getViewId () { return m_viewId; } #if H_3D Int getViewIndex () { return m_viewIndex; } Bool getIsDepth () { return m_isDepth; } #endif #endif Int getGOPSize() { return m_iGopSize; } TComList* getListPic() { return m_pcListPic; } #if !H_MV Void printOutSummary ( UInt uiNumAllPicCoded , bool isField); #endif #if H_3D_VSO Void preLoopFilterPicAll ( TComPic* pcPic, Dist64& ruiDist, UInt64& ruiBits ); #else Void preLoopFilterPicAll ( TComPic* pcPic, UInt64& ruiDist, UInt64& ruiBits ); #endif #if KWU_RC_MADPRED_E0227 TEncTop* getEncTop() { return m_pcEncTop; } #endif TEncSlice* getSliceEncoder() { return m_pcSliceEncoder; } NalUnitType getNalUnitType( Int pocCurr, Int lastIdr ); Void arrangeLongtermPicturesInRPS(TComSlice *, TComList& ); protected: TEncRateCtrl* getRateCtrl() { return m_pcRateCtrl; } protected: Void xInitGOP( Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRecOut, bool isField ); Void xInitGOP ( Int iPOC, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRecOut ); Void xGetBuffer ( TComList& rcListPic, TComList& rcListPicYuvRecOut, Int iNumPicRcvd, Int iTimeOffset, TComPic*& rpcPic, TComPicYuv*& rpcPicYuvRecOut, Int pocCurr, bool isField ); Void xCalculateAddPSNR ( TComPic* pcPic, TComPicYuv* pcPicD, const AccessUnit&, Double dEncTime ); Void xCalculateInterlacedAddPSNR( TComPic* pcPicOrgTop, TComPic* pcPicOrgBottom, TComPicYuv* pcPicRecTop, TComPicYuv* pcPicRecBottom, const AccessUnit& accessUnit, Double dEncTime ); #if H_3D_VSO Dist64 xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1); #else UInt64 xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1); #endif Double xCalculateRVM(); SEIActiveParameterSets* xCreateSEIActiveParameterSets (TComSPS *sps); SEIFramePacking* xCreateSEIFramePacking(); SEIDisplayOrientation* xCreateSEIDisplayOrientation(); SEIToneMappingInfo* xCreateSEIToneMappingInfo(); Void xCreateLeadingSEIMessages (/*SEIMessages seiMessages,*/ AccessUnit &accessUnit, TComSPS *sps); Int xGetFirstSeiLocation (AccessUnit &accessUnit); Void xResetNonNestedSEIPresentFlags() { m_activeParameterSetSEIPresentInAU = false; m_bufferingPeriodSEIPresentInAU = false; m_pictureTimingSEIPresentInAU = false; } Void xResetNestedSEIPresentFlags() { m_nestedBufferingPeriodSEIPresentInAU = false; m_nestedPictureTimingSEIPresentInAU = false; } #if H_MV Void xSetRefPicListModificationsMv( std::vector tempPicLists[2], TComSlice* pcSlice, UInt iGOPid ); #endif Void dblMetric( TComPic* pcPic, UInt uiNumSlices ); };// END CLASS DEFINITION TEncGOP // ==================================================================================================================== // Enumeration // ==================================================================================================================== enum PROCESSING_STATE { EXECUTE_INLOOPFILTER, ENCODE_SLICE }; enum SCALING_LIST_PARAMETER { SCALING_LIST_OFF, SCALING_LIST_DEFAULT, SCALING_LIST_FILE_READ }; //! \} #endif // __TENCGOP__