forked from mirror/libbpg
163 lines
8.7 KiB
C
163 lines
8.7 KiB
C
|
/* 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 TComPicSym.h
|
||
|
|
\brief picture symbol class (header)
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef __TCOMPICSYM__
|
||
|
|
#define __TCOMPICSYM__
|
||
|
|
|
||
|
|
|
||
|
|
// Include files
|
||
|
|
#include "CommonDef.h"
|
||
|
|
#include "TComSlice.h"
|
||
|
|
#include "TComDataCU.h"
|
||
|
|
class TComSampleAdaptiveOffset;
|
||
|
|
class TComPPS;
|
||
|
|
|
||
|
|
//! \ingroup TLibCommon
|
||
|
|
//! \{
|
||
|
|
|
||
|
|
// ====================================================================================================================
|
||
|
|
// Class definition
|
||
|
|
// ====================================================================================================================
|
||
|
|
|
||
|
|
class TComTile
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
UInt m_tileWidthInCtus;
|
||
|
|
UInt m_tileHeightInCtus;
|
||
|
|
UInt m_rightEdgePosInCtus;
|
||
|
|
UInt m_bottomEdgePosInCtus;
|
||
|
|
UInt m_firstCtuRsAddr;
|
||
|
|
|
||
|
|
public:
|
||
|
|
TComTile();
|
||
|
|
virtual ~TComTile();
|
||
|
|
|
||
|
|
Void setTileWidthInCtus ( UInt i ) { m_tileWidthInCtus = i; }
|
||
|
|
UInt getTileWidthInCtus () const { return m_tileWidthInCtus; }
|
||
|
|
Void setTileHeightInCtus ( UInt i ) { m_tileHeightInCtus = i; }
|
||
|
|
UInt getTileHeightInCtus () const { return m_tileHeightInCtus; }
|
||
|
|
Void setRightEdgePosInCtus ( UInt i ) { m_rightEdgePosInCtus = i; }
|
||
|
|
UInt getRightEdgePosInCtus () const { return m_rightEdgePosInCtus; }
|
||
|
|
Void setBottomEdgePosInCtus ( UInt i ) { m_bottomEdgePosInCtus = i; }
|
||
|
|
UInt getBottomEdgePosInCtus () const { return m_bottomEdgePosInCtus; }
|
||
|
|
Void setFirstCtuRsAddr ( UInt i ) { m_firstCtuRsAddr = i; }
|
||
|
|
UInt getFirstCtuRsAddr () const { return m_firstCtuRsAddr; }
|
||
|
|
};
|
||
|
|
|
||
|
|
/// picture symbol class
|
||
|
|
class TComPicSym
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
UInt m_frameWidthInCtus;
|
||
|
|
UInt m_frameHeightInCtus;
|
||
|
|
|
||
|
|
UInt m_uiMaxCUWidth;
|
||
|
|
UInt m_uiMaxCUHeight;
|
||
|
|
UInt m_uiMinCUWidth;
|
||
|
|
UInt m_uiMinCUHeight;
|
||
|
|
|
||
|
|
UChar m_uhTotalDepth; ///< max. depth
|
||
|
|
UInt m_numPartitionsInCtu;
|
||
|
|
UInt m_numPartInCtuWidth;
|
||
|
|
UInt m_numPartInCtuHeight;
|
||
|
|
UInt m_numCtusInFrame;
|
||
|
|
|
||
|
|
TComSlice** m_apcTComSlice;
|
||
|
|
UInt m_uiNumAllocatedSlice;
|
||
|
|
TComDataCU** m_pictureCtuArray; ///< array of CU data.
|
||
|
|
|
||
|
|
Int m_numTileColumnsMinus1;
|
||
|
|
Int m_numTileRowsMinus1;
|
||
|
|
std::vector<TComTile> m_tileParameters;
|
||
|
|
UInt* m_ctuTsToRsAddrMap; ///< for a given TS (Tile-Scan; coding order) address, returns the RS (Raster-Scan) address. cf CtbAddrTsToRs in specification.
|
||
|
|
UInt* m_puiTileIdxMap; ///< the map of the tile index relative to CTU raster scan address
|
||
|
|
UInt* m_ctuRsToTsAddrMap; ///< for a given RS (Raster-Scan) address, returns the TS (Tile-Scan; coding order) address. cf CtbAddrRsToTs in specification.
|
||
|
|
|
||
|
|
SAOBlkParam *m_saoBlkParams;
|
||
|
|
|
||
|
|
public:
|
||
|
|
Void create ( ChromaFormat chromaFormatIDC, Int iPicWidth, Int iPicHeight, UInt uiMaxWidth, UInt uiMaxHeight, UInt uiMaxDepth );
|
||
|
|
Void destroy ();
|
||
|
|
|
||
|
|
TComPicSym ();
|
||
|
|
TComSlice* getSlice(UInt i) { return m_apcTComSlice[i]; }
|
||
|
|
const TComSlice* getSlice(UInt i) const { return m_apcTComSlice[i]; }
|
||
|
|
UInt getFrameWidthInCtus() const { return m_frameWidthInCtus; }
|
||
|
|
UInt getFrameHeightInCtus() const { return m_frameHeightInCtus; }
|
||
|
|
UInt getMinCUWidth() const { return m_uiMinCUWidth; }
|
||
|
|
UInt getMinCUHeight() const { return m_uiMinCUHeight; }
|
||
|
|
UInt getNumberOfCtusInFrame() const { return m_numCtusInFrame; }
|
||
|
|
TComDataCU* getCtu( UInt ctuRsAddr ) { return m_pictureCtuArray[ctuRsAddr]; }
|
||
|
|
const TComDataCU* getCtu( UInt ctuRsAddr ) const { return m_pictureCtuArray[ctuRsAddr]; }
|
||
|
|
|
||
|
|
Void setSlice(TComSlice* p, UInt i) { m_apcTComSlice[i] = p; }
|
||
|
|
UInt getNumAllocatedSlice() const { return m_uiNumAllocatedSlice; }
|
||
|
|
Void allocateNewSlice();
|
||
|
|
Void clearSliceBuffer();
|
||
|
|
UInt getNumPartitionsInCtu() const { return m_numPartitionsInCtu; }
|
||
|
|
UInt getNumPartInCtuWidth() const { return m_numPartInCtuWidth; }
|
||
|
|
UInt getNumPartInCtuHeight() const { return m_numPartInCtuHeight; }
|
||
|
|
Void setNumTileColumnsMinus1( Int i ) { m_numTileColumnsMinus1 = i; }
|
||
|
|
Int getNumTileColumnsMinus1() const { return m_numTileColumnsMinus1; }
|
||
|
|
Void setNumTileRowsMinus1( Int i ) { m_numTileRowsMinus1 = i; }
|
||
|
|
Int getNumTileRowsMinus1() const { return m_numTileRowsMinus1; }
|
||
|
|
Int getNumTiles() const { return (m_numTileRowsMinus1+1)*(m_numTileColumnsMinus1+1); }
|
||
|
|
TComTile* getTComTile ( UInt tileIdx ) { return &(m_tileParameters[tileIdx]); }
|
||
|
|
const TComTile* getTComTile ( UInt tileIdx ) const { return &(m_tileParameters[tileIdx]); }
|
||
|
|
Void setCtuTsToRsAddrMap( Int ctuTsAddr, Int ctuRsAddr ) { *(m_ctuTsToRsAddrMap + ctuTsAddr) = ctuRsAddr; }
|
||
|
|
UInt getCtuTsToRsAddrMap( Int ctuTsAddr ) const { return *(m_ctuTsToRsAddrMap + (ctuTsAddr>=m_numCtusInFrame ? m_numCtusInFrame : ctuTsAddr)); }
|
||
|
|
UInt getTileIdxMap( Int ctuRsAddr ) const { return *(m_puiTileIdxMap + ctuRsAddr); }
|
||
|
|
Void setCtuRsToTsAddrMap( Int ctuRsAddr, Int ctuTsOrder ) { *(m_ctuRsToTsAddrMap + ctuRsAddr) = ctuTsOrder; }
|
||
|
|
UInt getCtuRsToTsAddrMap( Int ctuRsAddr ) const { return *(m_ctuRsToTsAddrMap + (ctuRsAddr>=m_numCtusInFrame ? m_numCtusInFrame : ctuRsAddr)); }
|
||
|
|
Void initTiles(TComPPS *pps);
|
||
|
|
|
||
|
|
Void initCtuTsRsAddrMaps();
|
||
|
|
SAOBlkParam* getSAOBlkParam() { return m_saoBlkParams;}
|
||
|
|
const SAOBlkParam* getSAOBlkParam() const { return m_saoBlkParams;}
|
||
|
|
Void deriveLoopFilterBoundaryAvailibility(Int ctuRsAddr,
|
||
|
|
Bool& isLeftAvail, Bool& isRightAvail, Bool& isAboveAvail, Bool& isBelowAvail,
|
||
|
|
Bool& isAboveLeftAvail, Bool& isAboveRightAvail, Bool& isBelowLeftAvail, Bool& isBelowRightAvail);
|
||
|
|
protected:
|
||
|
|
UInt xCalculateNextCtuRSAddr( UInt uiCurrCtuRSAddr );
|
||
|
|
|
||
|
|
};// END CLASS DEFINITION TComPicSym
|
||
|
|
|
||
|
|
//! \}
|
||
|
|
|
||
|
|
#endif // __TCOMPICSYM__
|
||
|
|
|