/* 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" #if CGS_3D_ASYMLUT #include "TEnc3DAsymLUT.h" #endif #include "TEncAnalyze.h" #include "TEncRateCtrl.h" #include //! \ingroup TLibEncoder //! \{ class TEncTop; // ==================================================================================================================== // Class definition // ==================================================================================================================== class TEncGOP { private: // 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; #if ALLOW_RECOVERY_POINT_AS_RAP Int m_iLastRecoveryPicPOC; #endif #if Q0074_COLOUR_REMAPPING_SEI string m_colourRemapSEIFile; Int m_colourRemapSEIId; Bool m_colourRemapSEICancelFlag; Bool m_colourRemapSEIPersistenceFlag; Bool m_colourRemapSEIVideoSignalInfoPresentFlag; Bool m_colourRemapSEIFullRangeFlag; Int m_colourRemapSEIPrimaries; Int m_colourRemapSEITransferFunction; Int m_colourRemapSEIMatrixCoefficients; Int m_colourRemapSEIInputBitDepth; Int m_colourRemapSEIBitDepth; Int m_colourRemapSEIPreLutNumValMinus1[3]; Int* m_colourRemapSEIPreLutCodedValue[3]; Int* m_colourRemapSEIPreLutTargetValue[3]; Bool m_colourRemapSEIMatrixPresentFlag; Int m_colourRemapSEILog2MatrixDenom; Int m_colourRemapSEICoeffs[3][3]; Int m_colourRemapSEIPostLutNumValMinus1[3]; Int* m_colourRemapSEIPostLutCodedValue[3]; Int* m_colourRemapSEIPostLutTargetValue[3]; #endif // 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; //--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; UInt m_cpbRemovalDelay; UInt m_tl0Idx; UInt m_rapIdx; Bool m_activeParameterSetSEIPresentInAU; Bool m_bufferingPeriodSEIPresentInAU; Bool m_pictureTimingSEIPresentInAU; Bool m_nestedBufferingPeriodSEIPresentInAU; Bool m_nestedPictureTimingSEIPresentInAU; #if SVC_EXTENSION Int m_pocCraWithoutReset; Int m_associatedIrapPocBeforeReset; UInt m_layerId; TEncTop** m_ppcTEncTop; TEncSearch* m_pcPredSearch; ///< encoder search class #if CGS_3D_ASYMLUT TEnc3DAsymLUT m_Enc3DAsymLUTPicUpdate; TEnc3DAsymLUT m_Enc3DAsymLUTPPS; TComPicYuv* m_pColorMappedPic; Int m_iTap; const Int (*m_phase_filter)[13]; const Int (*m_phase_filter_luma)[13]; const Int (*m_phase_filter_chroma)[13]; Int m_iM, m_iN; static const Int m_phase_filter_0_t0[4][13]; static const Int m_phase_filter_0_t1[4][13]; static const Int m_phase_filter_0_t1_chroma[4][13]; static const Int m_phase_filter_1[8][13]; Int **m_temp; #endif Int m_lastPocPeriodId; Bool m_noRaslOutputFlag; Bool m_prevPicHasEos; #endif public: TEncGOP(); virtual ~TEncGOP(); #if SVC_EXTENSION Void create ( UInt layerId ); #else Void create (); #endif Void destroy (); Void init ( TEncTop* pcTEncTop ); #if SVC_EXTENSION Void compressGOP ( Int iPicIdInGOP, Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRec, std::list& accessUnitsInGOP, Bool isField, Bool isTff, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE ); #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); Int getGOPSize() { return m_iGopSize; } TComList* getListPic() { return m_pcListPic; } #if !SVC_EXTENSION Void printOutSummary ( UInt uiNumAllPicCoded, Bool isField, const Bool printMSEBasedSNR, const Bool printSequenceMSE ); #endif Void preLoopFilterPicAll ( TComPic* pcPic, UInt64& ruiDist ); TEncSlice* getSliceEncoder() { return m_pcSliceEncoder; } NalUnitType getNalUnitType( Int pocCurr, Int lastIdr, Bool isField ); Void arrangeLongtermPicturesInRPS(TComSlice *, TComList& ); #if SVC_EXTENSION Void determinePocResetIdc( Int const pocCurr, TComSlice *const slice); Int getIntraRefreshInterval() { return m_pcCfg->getIntraPeriod(); } Int getIntraRefreshType() { return m_pcCfg->getDecodingRefreshType(); } Int getLastPocPeriodId() { return m_lastPocPeriodId; } Void setLastPocPeriodId(Int x) { m_lastPocPeriodId = x; } Void updatePocValuesOfPics( Int const pocCurr, TComSlice *const slice); #endif protected: TEncRateCtrl* getRateCtrl() { return m_pcRateCtrl; } protected: Void xInitGOP ( Int iPOCLast, Int iNumPicRcvd, TComList& rcListPic, TComList& rcListPicYuvRecOut, Bool isField ); 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, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE ); Void xCalculateInterlacedAddPSNR( TComPic* pcPicOrgFirstField, TComPic* pcPicOrgSecondField, TComPicYuv* pcPicRecFirstField, TComPicYuv* pcPicRecSecondField, const AccessUnit& accessUnit, Double dEncTime, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE ); UInt64 xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1); Double xCalculateRVM(); SEIActiveParameterSets* xCreateSEIActiveParameterSets (const TComSPS *sps); SEIFramePacking* xCreateSEIFramePacking(); SEISegmentedRectFramePacking* xCreateSEISegmentedRectFramePacking(); SEIDisplayOrientation* xCreateSEIDisplayOrientation(); SEIToneMappingInfo* xCreateSEIToneMappingInfo(); SEITempMotionConstrainedTileSets* xCreateSEITempMotionConstrainedTileSets (const TComPPS *pps); SEIKneeFunctionInfo* xCreateSEIKneeFunctionInfo(); SEIChromaSamplingFilterHint* xCreateSEIChromaSamplingFilterHint(Bool bChromaLocInfoPresent, Int iHorFilterIndex, Int iVerFilterIdc); Void xCreateLeadingSEIMessages (/*SEIMessages seiMessages,*/ AccessUnit &accessUnit, const TComSPS *sps, const TComPPS *pps); Int xGetFirstSeiLocation (AccessUnit &accessUnit); Void xResetNonNestedSEIPresentFlags() { m_activeParameterSetSEIPresentInAU = false; m_bufferingPeriodSEIPresentInAU = false; m_pictureTimingSEIPresentInAU = false; } Void xResetNestedSEIPresentFlags() { m_nestedBufferingPeriodSEIPresentInAU = false; m_nestedPictureTimingSEIPresentInAU = false; } Void applyDeblockingFilterMetric( TComPic* pcPic, UInt uiNumSlices ); #if Q0074_COLOUR_REMAPPING_SEI Void setCRISEIFile( Char* pch ) { m_colourRemapSEIFile = pch; } Void freeColourCRI(); Int readingCRIparameters(); Void xCheckParameter(); SEIColourRemappingInfo* xCreateSEIColourRemappingInfo(); #endif #if SVC_EXTENSION #if LAYERS_NOT_PRESENT_SEI SEILayersNotPresent* xCreateSEILayersNotPresent (); #endif #if N0383_IL_CONSTRAINED_TILE_SETS_SEI Void xBuildTileSetsMap(TComPicSym* picSym); SEIInterLayerConstrainedTileSets* xCreateSEIInterLayerConstrainedTileSets(); #endif #if O0164_MULTI_LAYER_HRD SEIScalableNesting* xCreateBspNestingSEI(TComSlice *pcSlice, Int olsIdx, Int partitioningSchemeIdx, Int bspIdx); #endif #if P0123_ALPHA_CHANNEL_SEI SEIAlphaChannelInfo* xCreateSEIAlphaChannelInfo(); #endif #if Q0096_OVERLAY_SEI SEIOverlayInfo* xCreateSEIOverlayInfo(); #endif #if CGS_3D_ASYMLUT Void xDetermin3DAsymLUT( TComSlice * pSlice , TComPic * pCurPic , UInt refLayerIdc , TEncCfg * pCfg , Bool bSignalPPS ); Void downScalePic( TComPicYuv* pcYuvSrc, TComPicYuv* pcYuvDest); Void downScaleComponent2x2( const Pel* pSrc, Pel* pDest, const Int iSrcStride, const Int iDestStride, const Int iSrcWidth, const Int iSrcHeight, const Int inputBitDepth, const Int outputBitDepth ); inline Short xClip( Short x , Int bitdepth ); Void initDs(Int iWidth, Int iHeight, Int iType); Void filterImg( Pel *src, Int iSrcStride, Pel *dst, Int iDstStride, Int height1, Int width1, Int shift, Int plane); Int get_mem2DintWithPad(Int ***array2D, Int dim0, Int dim1, Int iPadY, Int iPadX); Void free_mem2DintWithPad(Int **array2D, Int iPadY, Int iPadX); #endif Void xCheckLayerReset(TComSlice *slice); Void xSetNoRaslOutputFlag(TComSlice *slice); Void xSetLayerInitializedFlag(TComSlice *slice); #endif //SVC_EXTENSION };// END CLASS DEFINITION TEncGOP //! \} #endif // __TENCGOP__