libbpg/jctvc/TLibEncoder/TEncTop.h

184 lines
9.3 KiB
C
Raw Normal View History

2015-01-16 12:46:18 +00:00
/* 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-2014, 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 TEncTop.h
\brief encoder class (header)
*/
#ifndef __TENCTOP__
#define __TENCTOP__
// Include files
#include "TLibCommon/TComList.h"
#include "TLibCommon/TComPrediction.h"
#include "TLibCommon/TComTrQuant.h"
#include "TLibCommon/TComLoopFilter.h"
#include "TLibCommon/AccessUnit.h"
#include "TLibVideoIO/TVideoIOYuv.h"
#include "TEncCfg.h"
#include "TEncGOP.h"
#include "TEncSlice.h"
#include "TEncEntropy.h"
#include "TEncCavlc.h"
#include "TEncSbac.h"
#include "TEncSearch.h"
#include "TEncSampleAdaptiveOffset.h"
#include "TEncPreanalyzer.h"
#include "TEncRateCtrl.h"
//! \ingroup TLibEncoder
//! \{
// ====================================================================================================================
// Class definition
// ====================================================================================================================
/// encoder class
class TEncTop : public TEncCfg
{
private:
// picture
Int m_iPOCLast; ///< time index (POC)
Int m_iNumPicRcvd; ///< number of received pictures
UInt m_uiNumAllPicCoded; ///< number of coded pictures
TComList<TComPic*> m_cListPic; ///< dynamic list of pictures
// encoder search
TEncSearch m_cSearch; ///< encoder search class
//TEncEntropy* m_pcEntropyCoder; ///< entropy encoder
TEncCavlc* m_pcCavlcCoder; ///< CAVLC encoder
// coding tool
TComTrQuant m_cTrQuant; ///< transform & quantization class
TComLoopFilter m_cLoopFilter; ///< deblocking filter class
TEncSampleAdaptiveOffset m_cEncSAO; ///< sample adaptive offset class
TEncEntropy m_cEntropyCoder; ///< entropy encoder
TEncCavlc m_cCavlcCoder; ///< CAVLC encoder
TEncSbac m_cSbacCoder; ///< SBAC encoder
TEncBinCABAC m_cBinCoderCABAC; ///< bin coder CABAC
// processing unit
TEncGOP m_cGOPEncoder; ///< GOP encoder
TEncSlice m_cSliceEncoder; ///< slice encoder
TEncCu m_cCuEncoder; ///< CU encoder
// SPS
TComSPS m_cSPS; ///< SPS
TComPPS m_cPPS; ///< PPS
// RD cost computation
TComRdCost m_cRdCost; ///< RD cost computation class
TEncSbac*** m_pppcRDSbacCoder; ///< temporal storage for RD computation
TEncSbac m_cRDGoOnSbacCoder; ///< going on SBAC model for RD stage
#if FAST_BIT_EST
TEncBinCABACCounter*** m_pppcBinCoderCABAC; ///< temporal CABAC state storage for RD computation
TEncBinCABACCounter m_cRDGoOnBinCoderCABAC; ///< going on bin coder CABAC for RD stage
#else
TEncBinCABAC*** m_pppcBinCoderCABAC; ///< temporal CABAC state storage for RD computation
TEncBinCABAC m_cRDGoOnBinCoderCABAC; ///< going on bin coder CABAC for RD stage
#endif
// quality control
TEncPreanalyzer m_cPreanalyzer; ///< image characteristics analyzer for TM5-step3-like adaptive QP
TComScalingList m_scalingList; ///< quantization matrix information
TEncRateCtrl m_cRateCtrl; ///< Rate control class
protected:
Void xGetNewPicBuffer ( TComPic*& rpcPic ); ///< get picture buffer which will be processed
Void xInitSPS (); ///< initialize SPS from encoder options
Void xInitPPS (); ///< initialize PPS from encoder options
Void xInitPPSforTiles ();
Void xInitRPS (Bool isFieldCoding); ///< initialize PPS from encoder options
public:
TEncTop();
virtual ~TEncTop();
Void create ();
Void destroy ();
Void init (Bool isFieldCoding);
Void deletePicBuffer ();
// -------------------------------------------------------------------------------------------------------------------
// member access functions
// -------------------------------------------------------------------------------------------------------------------
TComList<TComPic*>* getListPic () { return &m_cListPic; }
TEncSearch* getPredSearch () { return &m_cSearch; }
TComTrQuant* getTrQuant () { return &m_cTrQuant; }
TComLoopFilter* getLoopFilter () { return &m_cLoopFilter; }
TEncSampleAdaptiveOffset* getSAO () { return &m_cEncSAO; }
TEncGOP* getGOPEncoder () { return &m_cGOPEncoder; }
TEncSlice* getSliceEncoder () { return &m_cSliceEncoder; }
TEncCu* getCuEncoder () { return &m_cCuEncoder; }
TEncEntropy* getEntropyCoder () { return &m_cEntropyCoder; }
TEncCavlc* getCavlcCoder () { return &m_cCavlcCoder; }
TEncSbac* getSbacCoder () { return &m_cSbacCoder; }
TEncBinCABAC* getBinCABAC () { return &m_cBinCoderCABAC; }
TComRdCost* getRdCost () { return &m_cRdCost; }
TEncSbac*** getRDSbacCoder () { return m_pppcRDSbacCoder; }
TEncSbac* getRDGoOnSbacCoder () { return &m_cRDGoOnSbacCoder; }
TEncRateCtrl* getRateCtrl () { return &m_cRateCtrl; }
TComSPS* getSPS () { return &m_cSPS; }
TComPPS* getPPS () { return &m_cPPS; }
Void selectReferencePictureSet(TComSlice* slice, Int POCCurr, Int GOPid );
Int getReferencePictureSetIdxForSOP(TComSlice* slice, Int POCCurr, Int GOPid );
TComScalingList* getScalingList () { return &m_scalingList; }
// -------------------------------------------------------------------------------------------------------------------
// encoder function
// -------------------------------------------------------------------------------------------------------------------
/// encode several number of pictures until end-of-sequence
Void encode( Bool bEos,
TComPicYuv* pcPicYuvOrg,
TComPicYuv* pcPicYuvTrueOrg, const InputColourSpaceConversion snrCSC, // used for SNR calculations. Picture in original colour space.
TComList<TComPicYuv*>& rcListPicYuvRecOut,
std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded );
/// encode several number of pictures until end-of-sequence
Void encode( Bool bEos, TComPicYuv* pcPicYuvOrg,
TComPicYuv* pcPicYuvTrueOrg, const InputColourSpaceConversion snrCSC, // used for SNR calculations. Picture in original colour space.
TComList<TComPicYuv*>& rcListPicYuvRecOut,
std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded, Bool isTff);
Void printSummary(Bool isField) { m_cGOPEncoder.printOutSummary (m_uiNumAllPicCoded, isField, m_printMSEBasedSequencePSNR, m_printSequenceMSE); }
};
//! \}
#endif // __TENCTOP__