forked from mirror/libbpg
libbpg-0.9.3
This commit is contained in:
commit
b21307932d
266 changed files with 108670 additions and 0 deletions
143
jctvc/TLibCommon/TComPrediction.h
Normal file
143
jctvc/TLibCommon/TComPrediction.h
Normal file
|
@ -0,0 +1,143 @@
|
|||
/* 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 TComPrediction.h
|
||||
\brief prediction class (header)
|
||||
*/
|
||||
|
||||
#ifndef __TCOMPREDICTION__
|
||||
#define __TCOMPREDICTION__
|
||||
|
||||
|
||||
// Include files
|
||||
#include "TComPic.h"
|
||||
#include "TComMotionInfo.h"
|
||||
#include "TComPattern.h"
|
||||
#include "TComTrQuant.h"
|
||||
#include "TComInterpolationFilter.h"
|
||||
#include "TComWeightPrediction.h"
|
||||
|
||||
class TComTU; // forward declaration
|
||||
|
||||
//! \ingroup TLibCommon
|
||||
//! \{
|
||||
|
||||
// ====================================================================================================================
|
||||
// Class definition
|
||||
// ====================================================================================================================
|
||||
|
||||
/// prediction class
|
||||
typedef enum PRED_BUF_E
|
||||
{
|
||||
PRED_BUF_UNFILTERED=0,
|
||||
PRED_BUF_FILTERED=1,
|
||||
NUM_PRED_BUF=2
|
||||
} PRED_BUF;
|
||||
|
||||
static const UInt MAX_INTRA_FILTER_DEPTHS=5;
|
||||
|
||||
class TComPrediction : public TComWeightPrediction
|
||||
{
|
||||
private:
|
||||
static const UChar m_aucIntraFilter[MAX_NUM_CHANNEL_TYPE][MAX_INTRA_FILTER_DEPTHS];
|
||||
|
||||
protected:
|
||||
Pel* m_piYuvExt[MAX_NUM_COMPONENT][NUM_PRED_BUF];
|
||||
Int m_iYuvExtSize;
|
||||
|
||||
TComYuv m_acYuvPred[NUM_REF_PIC_LIST_01];
|
||||
TComYuv m_cYuvPredTemp;
|
||||
TComYuv m_filteredBlock[LUMA_INTERPOLATION_FILTER_SUB_SAMPLE_POSITIONS][LUMA_INTERPOLATION_FILTER_SUB_SAMPLE_POSITIONS];
|
||||
TComYuv m_filteredBlockTmp[LUMA_INTERPOLATION_FILTER_SUB_SAMPLE_POSITIONS];
|
||||
|
||||
TComInterpolationFilter m_if;
|
||||
|
||||
Pel* m_pLumaRecBuffer; ///< array for downsampled reconstructed luma sample
|
||||
Int m_iLumaRecStride; ///< stride of #m_pLumaRecBuffer array
|
||||
|
||||
Void xPredIntraAng ( Int bitDepth, const Pel* pSrc, Int srcStride, Pel* pDst, Int dstStride, UInt width, UInt height, ChannelType channelType, ChromaFormat format, UInt dirMode, Bool blkAboveAvailable, Bool blkLeftAvailable, const Bool bEnableEdgeFilters );
|
||||
Void xPredIntraPlanar ( const Pel* pSrc, Int srcStride, Pel* rpDst, Int dstStride, UInt width, UInt height, ChannelType channelType, ChromaFormat format );
|
||||
|
||||
// motion compensation functions
|
||||
Void xPredInterUni ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, RefPicList eRefPicList, TComYuv* pcYuvPred, Bool bi=false );
|
||||
Void xPredInterBi ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv* pcYuvPred );
|
||||
Void xPredInterBlk(const ComponentID compID, TComDataCU *cu, TComPicYuv *refPic, UInt partAddr, TComMv *mv, Int width, Int height, TComYuv *dstPic, Bool bi );
|
||||
Void xWeightedAverage ( TComYuv* pcYuvSrc0, TComYuv* pcYuvSrc1, Int iRefIdx0, Int iRefIdx1, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv* pcYuvDst );
|
||||
|
||||
Void xGetLLSPrediction ( const Pel* pSrc0, Int iSrcStride, Pel* pDst0, Int iDstStride, UInt uiWidth, UInt uiHeight, UInt uiExt0, const ChromaFormat chFmt DEBUG_STRING_FN_DECLARE(sDebug) );
|
||||
|
||||
Void xDCPredFiltering( const Pel* pSrc, Int iSrcStride, Pel* pDst, Int iDstStride, Int iWidth, Int iHeight, ChannelType channelType );
|
||||
Bool xCheckIdenticalMotion ( TComDataCU* pcCU, UInt PartAddr);
|
||||
Void destroy();
|
||||
|
||||
public:
|
||||
TComPrediction();
|
||||
virtual ~TComPrediction();
|
||||
|
||||
Void initTempBuff(ChromaFormat chromaFormatIDC);
|
||||
|
||||
ChromaFormat getChromaFormat() const { return m_cYuvPredTemp.getChromaFormat(); }
|
||||
|
||||
// inter
|
||||
Void motionCompensation ( TComDataCU* pcCU, TComYuv* pcYuvPred, RefPicList eRefPicList = REF_PIC_LIST_X, Int iPartIdx = -1 );
|
||||
|
||||
// motion vector prediction
|
||||
Void getMvPredAMVP ( TComDataCU* pcCU, UInt uiPartIdx, UInt uiPartAddr, RefPicList eRefPicList, TComMv& rcMvPred );
|
||||
|
||||
// Angular Intra
|
||||
Void predIntraAng ( const ComponentID compID, UInt uiDirMode, Pel *piOrg /* Will be null for decoding */, UInt uiOrgStride, Pel* piPred, UInt uiStride, TComTU &rTu, Bool bAbove, Bool bLeft, const Bool bUseFilteredPredSamples, const Bool bUseLosslessDPCM = false );
|
||||
|
||||
Pel predIntraGetPredValDC ( const Pel* pSrc, Int iSrcStride, UInt iWidth, UInt iHeight, ChannelType channelType, ChromaFormat format, Bool bAbove, Bool bLeft );
|
||||
|
||||
Pel* getPredictorPtr ( const ComponentID compID, const Bool bUseFilteredPredictions )
|
||||
{
|
||||
return m_piYuvExt[compID][bUseFilteredPredictions?PRED_BUF_FILTERED:PRED_BUF_UNFILTERED];
|
||||
}
|
||||
|
||||
// This function is actually still in TComPattern.cpp
|
||||
/// set parameters from CU data for accessing ADI data
|
||||
Void initAdiPatternChType ( TComTU &rTu,
|
||||
Bool& bAbove,
|
||||
Bool& bLeft,
|
||||
const ComponentID compID, const Bool bFilterRefSamples
|
||||
DEBUG_STRING_FN_DECLARE(sDebug)
|
||||
);
|
||||
|
||||
static Bool filteringIntraReferenceSamples(const ComponentID compID, UInt uiDirMode, UInt uiTuChWidth, UInt uiTuChHeight, const ChromaFormat chFmt, const Bool intraReferenceSmoothingDisabled);
|
||||
|
||||
static Bool UseDPCMForFirstPassIntraEstimation(TComTU &rTu, const UInt uiDirMode);
|
||||
};
|
||||
|
||||
//! \}
|
||||
|
||||
#endif // __TCOMPREDICTION__
|
Loading…
Add table
Add a link
Reference in a new issue