TEncSlice.h

Go to the documentation of this file.

/* 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-2017, 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.
 */

#ifndef __TENCSLICE__
#define __TENCSLICE__

// Include files
#include "TLibCommon/CommonDef.h"
#include "TLibCommon/TComList.h"
#include "TLibCommon/TComPic.h"
#include "TLibCommon/TComPicYuv.h"
#include "TEncCu.h"
#include "WeightPredAnalysis.h"
#include "TEncRateCtrl.h"


class TEncTop;
class TEncGOP;

// ====================================================================================================================
// Class definition
// ====================================================================================================================

class TEncSlice
  : public WeightPredAnalysis
{
private:
  // encoder configuration
  TEncCfg*                m_pcCfg;                              

  // pictures
  TComList<TComPic*>*     m_pcListPic;                          
  TComPicYuv              m_picYuvPred;                         
  TComPicYuv              m_picYuvResi;                         

  // processing units
  TEncGOP*                m_pcGOPEncoder;                       
  TEncCu*                 m_pcCuEncoder;                        

  // encoder search
  TEncSearch*             m_pcPredSearch;                       

  // coding tools
  TEncEntropy*            m_pcEntropyCoder;                     
  TEncSbac*               m_pcSbacCoder;                        
  TEncBinCABAC*           m_pcBinCABAC;                         
  TComTrQuant*            m_pcTrQuant;                          

  // RD optimization
  TComRdCost*             m_pcRdCost;                           
  TEncSbac***             m_pppcRDSbacCoder;                    
  TEncSbac*               m_pcRDGoOnSbacCoder;                  
  UInt64                  m_uiPicTotalBits;                     
  UInt64                  m_uiPicDist;                          
  Double                  m_dPicRdCost;                         
  std::vector<Double>     m_vdRdPicLambda;                      
  std::vector<Double>     m_vdRdPicQp;                          
  std::vector<Int>        m_viRdPicQp;                          
  TEncRateCtrl*           m_pcRateCtrl;                         
  UInt                    m_uiSliceIdx;
  TEncSbac                m_lastSliceSegmentEndContextState;    
  TEncSbac                m_entropyCodingSyncContextState;      
  SliceType               m_encCABACTableIdx;
  Int                     m_gopID;

  Double   calculateLambda( const TComSlice* pSlice, const Int GOPid, const Int depth, const Double refQP, const Double dQP, Int &iQP );
  Void     setUpLambda(TComSlice* slice, const Double dLambda, Int iQP);
  Void     calculateBoundingCtuTsAddrForSlice(UInt &startCtuTSAddrSlice, UInt &boundingCtuTSAddrSlice, Bool &haveReachedTileBoundary, TComPic* pcPic, const Int sliceMode, const Int sliceArgument);

public:
  TEncSlice();
  virtual ~TEncSlice();

  Void    create              ( Int iWidth, Int iHeight, ChromaFormat chromaFormat, UInt iMaxCUWidth, UInt iMaxCUHeight, UChar uhTotalDepth );
  Void    destroy             ();
  Void    init                ( TEncTop* pcEncTop );

  Void    initEncSlice        ( TComPic*  pcPic, const Int pocLast, const Int pocCurr,
                                const Int iGOPid,   TComSlice*& rpcSlice, const Bool isField );
  Void    resetQP             ( TComPic* pic, Int sliceQP, Double lambda );
  Void    setGopID( Int iGopID )      { m_gopID = iGopID; }
  Int     getGopID() const            { return m_gopID;   }
  Void    updateLambda(TComSlice* pSlice, Double dQP);

  // compress and encode slice
  Void    precompressSlice    ( TComPic* pcPic                                     );      
  Void    compressSlice       ( TComPic* pcPic, const Bool bCompressEntireSlice, const Bool bFastDeltaQP );      
  Void    calCostSliceI       ( TComPic* pcPic );
  Void    encodeSlice         ( TComPic* pcPic, TComOutputBitstream* pcSubstreams, UInt &numBinsCoded );

  // misc. functions
  Void    setSearchRange      ( TComSlice* pcSlice  );                                  

  TEncCu*        getCUEncoder() { return m_pcCuEncoder; }                        
  Void    xDetermineStartAndBoundingCtuTsAddr  ( UInt& startCtuTsAddr, UInt& boundingCtuTsAddr, TComPic* pcPic );
  UInt    getSliceIdx()         { return m_uiSliceIdx;                    }
  Void    setSliceIdx(UInt i)   { m_uiSliceIdx = i;                       }

  SliceType getEncCABACTableIdx() const           { return m_encCABACTableIdx;        }

private:
  Double  xGetQPValueAccordingToLambda ( Double lambda );
};


#endif // __TENCSLICE__