/* 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 TComPic.h \brief picture class (header) */ #ifndef __TCOMPIC__ #define __TCOMPIC__ // Include files #include "CommonDef.h" #include "TComPicSym.h" #include "TComPicYuv.h" #include "TComBitStream.h" //! \ingroup TLibCommon //! \{ // ==================================================================================================================== // Class definition // ==================================================================================================================== /// picture class (symbol + YUV buffers) class TComPic { public: typedef enum { PIC_YUV_ORG=0, PIC_YUV_REC=1, PIC_YUV_TRUE_ORG=2, NUM_PIC_YUV=3 } PIC_YUV_T; // TRUE_ORG is the input file without any pre-encoder colour space conversion (but with possible bit depth increment) TComPicYuv* getPicYuvTrueOrg() { return m_apcPicYuv[PIC_YUV_TRUE_ORG]; } private: UInt m_uiTLayer; // Temporal layer Bool m_bUsedByCurr; // Used by current picture Bool m_bIsLongTerm; // IS long term picture TComPicSym m_picSym; // Symbol TComPicYuv* m_apcPicYuv[NUM_PIC_YUV]; TComPicYuv* m_pcPicYuvPred; // Prediction TComPicYuv* m_pcPicYuvResi; // Residual Bool m_bReconstructed; Bool m_bNeededForOutput; #if NH_MV Bool m_bPicOutputFlag; // Semantics variable #endif UInt m_uiCurrSliceIdx; // Index of current slice Bool m_bCheckLTMSB; Bool m_isTop; Bool m_isField; std::vector > m_vSliceCUDataLink; SEIMessages m_SEIs; ///< Any SEI messages that have been received. If !NULL we own the object. #if NH_MV Int m_layerId; Int m_viewId; #if NH_3D Int m_viewIndex; Bool m_isDepth; Int** m_aaiCodedScale; Int** m_aaiCodedOffset; #endif #endif #if H_3D_QTLPC Bool m_bReduceBitsQTL; #endif #if H_3D_NBDV UInt m_uiRapRefIdx; RefPicList m_eRapRefList; Int m_iNumDdvCandPics; Bool m_abTIVRINCurrRL [2][2][MAX_NUM_REF]; //whether an inter-view reference picture with the same view index of the inter-view reference picture of temporal reference picture of current picture exists in current reference picture lists Int m_aiTexToDepRef [2][MAX_NUM_REF]; #endif public: TComPic(); virtual ~TComPic(); Void create( const TComSPS &sps, const TComPPS &pps, const Bool bIsVirtual /*= false*/ ); virtual Void destroy(); UInt getTLayer() const { return m_uiTLayer; } Void setTLayer( UInt uiTLayer ) { m_uiTLayer = uiTLayer; } #if NH_MV Void setLayerId ( Int layerId ) { m_layerId = layerId; } Int getLayerId () { return m_layerId; } Void setViewId ( Int viewId ) { m_viewId = viewId; } Int getViewId () { return m_viewId; } #if NH_3D Void setViewIndex ( Int viewIndex ) { m_viewIndex = viewIndex; } Int getViewIndex () { return m_viewIndex; } Void setIsDepth ( Bool isDepth ) { m_isDepth = isDepth; } Bool getIsDepth () { return m_isDepth; } Void setScaleOffset( Int** pS, Int** pO ) { m_aaiCodedScale = pS; m_aaiCodedOffset = pO; } Int** getCodedScale () { return m_aaiCodedScale; } Int** getCodedOffset() { return m_aaiCodedOffset; } #endif #endif #if H_3D_QTLPC Bool getReduceBitsFlag () { return m_bReduceBitsQTL; } Void setReduceBitsFlag ( Bool bFlag ) { m_bReduceBitsQTL = bFlag; } #endif Bool getUsedByCurr() const { return m_bUsedByCurr; } Void setUsedByCurr( Bool bUsed ) { m_bUsedByCurr = bUsed; } Bool getIsLongTerm() const { return m_bIsLongTerm; } Void setIsLongTerm( Bool lt ) { m_bIsLongTerm = lt; } Void setCheckLTMSBPresent (Bool b ) {m_bCheckLTMSB=b;} Bool getCheckLTMSBPresent () { return m_bCheckLTMSB;} TComPicSym* getPicSym() { return &m_picSym; } TComSlice* getSlice(Int i) { return m_picSym.getSlice(i); } Int getPOC() const { return m_picSym.getSlice(m_uiCurrSliceIdx)->getPOC(); } TComDataCU* getCtu( UInt ctuRsAddr ) { return m_picSym.getCtu( ctuRsAddr ); } const TComDataCU* getCtu( UInt ctuRsAddr ) const { return m_picSym.getCtu( ctuRsAddr ); } TComPicYuv* getPicYuvOrg() { return m_apcPicYuv[PIC_YUV_ORG]; } TComPicYuv* getPicYuvRec() { return m_apcPicYuv[PIC_YUV_REC]; } TComPicYuv* getPicYuvPred() { return m_pcPicYuvPred; } TComPicYuv* getPicYuvResi() { return m_pcPicYuvResi; } Void setPicYuvPred( TComPicYuv* pcPicYuv ) { m_pcPicYuvPred = pcPicYuv; } Void setPicYuvResi( TComPicYuv* pcPicYuv ) { m_pcPicYuvResi = pcPicYuv; } UInt getNumberOfCtusInFrame() const { return m_picSym.getNumberOfCtusInFrame(); } UInt getNumPartInCtuWidth() const { return m_picSym.getNumPartInCtuWidth(); } UInt getNumPartInCtuHeight() const { return m_picSym.getNumPartInCtuHeight(); } UInt getNumPartitionsInCtu() const { return m_picSym.getNumPartitionsInCtu(); } UInt getFrameWidthInCtus() const { return m_picSym.getFrameWidthInCtus(); } UInt getFrameHeightInCtus() const { return m_picSym.getFrameHeightInCtus(); } UInt getMinCUWidth() const { return m_picSym.getMinCUWidth(); } UInt getMinCUHeight() const { return m_picSym.getMinCUHeight(); } Int getStride(const ComponentID id) const { return m_apcPicYuv[PIC_YUV_REC]->getStride(id); } Int getComponentScaleX(const ComponentID id) const { return m_apcPicYuv[PIC_YUV_REC]->getComponentScaleX(id); } Int getComponentScaleY(const ComponentID id) const { return m_apcPicYuv[PIC_YUV_REC]->getComponentScaleY(id); } ChromaFormat getChromaFormat() const { return m_apcPicYuv[PIC_YUV_REC]->getChromaFormat(); } Int getNumberValidComponents() const { return m_apcPicYuv[PIC_YUV_REC]->getNumberValidComponents(); } Void setReconMark (Bool b) { m_bReconstructed = b; } Bool getReconMark () const { return m_bReconstructed; } Void setOutputMark (Bool b) { m_bNeededForOutput = b; } Bool getOutputMark () const { return m_bNeededForOutput; } #if NH_MV Void setPicOutputFlag(Bool b) { m_bPicOutputFlag = b; } Bool getPicOutputFlag() { return m_bPicOutputFlag ; } #endif #if NH_3D Void compressMotion(Int scale); #else Void compressMotion(); #endif UInt getCurrSliceIdx() const { return m_uiCurrSliceIdx; } Void setCurrSliceIdx(UInt i) { m_uiCurrSliceIdx = i; } UInt getNumAllocatedSlice() const {return m_picSym.getNumAllocatedSlice();} Void allocateNewSlice() {m_picSym.allocateNewSlice(); } Void clearSliceBuffer() {m_picSym.clearSliceBuffer(); } const Window& getConformanceWindow() const { return m_picSym.getSPS().getConformanceWindow(); } Window getDefDisplayWindow() const { return m_picSym.getSPS().getVuiParametersPresentFlag() ? m_picSym.getSPS().getVuiParameters()->getDefaultDisplayWindow() : Window(); } Bool getSAOMergeAvailability(Int currAddr, Int mergeAddr); UInt getSubstreamForCtuAddr(const UInt ctuAddr, const Bool bAddressInRaster, TComSlice *pcSlice); /* field coding parameters*/ Void setTopField(Bool b) {m_isTop = b;} Bool isTopField() {return m_isTop;} Void setField(Bool b) {m_isField = b;} Bool isField() {return m_isField;} #if NH_MV Void print( Bool legend ); #endif #if H_3D_NBDV Int getNumDdvCandPics() {return m_iNumDdvCandPics; } Int getDisCandRefPictures(Int iColPOC); Void setRapRefIdx(UInt uiRapRefIdx) {m_uiRapRefIdx = uiRapRefIdx;} Void setRapRefList(RefPicList eRefPicList) {m_eRapRefList = eRefPicList;} Void setNumDdvCandPics (Int i) {m_iNumDdvCandPics = i; } UInt getRapRefIdx() {return m_uiRapRefIdx; } RefPicList getRapRefList() {return m_eRapRefList; } Void checkTemporalIVRef(); Bool isTempIVRefValid(Int currCandPic, Int iTempRefDir, Int iTempRefIdx); Void checkTextureRef( ); Int isTextRefValid(Int iTextRefDir, Int iTextRefIdx); #endif /** transfer ownership of seis to this picture */ Void setSEIs(SEIMessages& seis) { m_SEIs = seis; } /** * return the current list of SEI messages associated with this picture. * Pointer is valid until this->destroy() is called */ SEIMessages& getSEIs() { return m_SEIs; } /** * return the current list of SEI messages associated with this picture. * Pointer is valid until this->destroy() is called */ const SEIMessages& getSEIs() const { return m_SEIs; } };// END CLASS DEFINITION TComPic #if NH_MV class TComPicLists { private: TComList*> m_lists; #if NH_3D const TComVPS* m_vps; #endif public: Void push_back( TComList* list ) { m_lists.push_back( list ); } Int size () { return (Int) m_lists.size(); } #if H_3D_ARP TComList* getPicList ( Int layerIdInNuh ); #endif TComPic* getPic ( Int layerIdInNuh, Int poc ); TComPicYuv* getPicYuv( Int layerIdInNuh, Int poc, Bool recon ); #if NH_3D Void setVPS ( const TComVPS* vps ) { m_vps = vps; }; TComPic* getPic ( Int viewIndex, Bool depthFlag, Int poc ); TComPicYuv* getPicYuv( Int viewIndex, Bool depthFlag, Int poc, Bool recon ); #endif Void print( ); }; // END CLASS DEFINITION TComPicLists #endif //! \} #endif // __TCOMPIC__