source: SHVCSoftware/branches/0.1.1-bugfix/source/Lib/TLibEncoder/TEncCavlc.h @ 1565

Last change on this file since 1565 was 2, checked in by seregin, 12 years ago

Initial import by Vadim Seregin <vseregin@…>
File size: 8.9 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license. 
5 *
6 * Copyright (c) 2010-2012, ITU/ISO/IEC
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are met:
11 *
12 *  * Redistributions of source code must retain the above copyright notice,
13 *    this list of conditions and the following disclaimer.
14 *  * Redistributions in binary form must reproduce the above copyright notice,
15 *    this list of conditions and the following disclaimer in the documentation
16 *    and/or other materials provided with the distribution.
17 *  * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18 *    be used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 * THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34/** \file     TEncCavlc.h
35    \brief    CAVLC encoder class (header)
36*/
37
38#ifndef __TENCCAVLC__
39#define __TENCCAVLC__
40
41#if _MSC_VER > 1000
42#pragma once
43#endif // _MSC_VER > 1000
44
45#include "TLibCommon/CommonDef.h"
46#include "TLibCommon/TComBitStream.h"
47#include "TLibCommon/TComRom.h"
48#include "TEncEntropy.h"
49#include "SyntaxElementWriter.h"
50
51//! \ingroup TLibEncoder
52//! \{
53
54class TEncTop;
55
56// ====================================================================================================================
57// Class definition
58// ====================================================================================================================
59
60/// CAVLC encoder class
61class TEncCavlc : public SyntaxElementWriter, public TEncEntropyIf
62{
63public:
64  TEncCavlc();
65  virtual ~TEncCavlc();
66 
67protected:
68  TComSlice*    m_pcSlice;
69  UInt          m_uiCoeffCost;
70#if !REMOVE_FGS
71  Int           m_iSliceGranularity;  //!< slice granularity
72#endif
73
74  Void  xWritePCMAlignZero    ();
75  Void  xWriteEpExGolomb      ( UInt uiSymbol, UInt uiCount );
76  Void  xWriteExGolombLevel    ( UInt uiSymbol );
77  Void  xWriteUnaryMaxSymbol  ( UInt uiSymbol, UInt uiMaxSymbol );
78
79#if J0234_INTER_RPS_SIMPL
80  Void codeShortTermRefPicSet              ( TComSPS* pcSPS, TComReferencePictureSet* pcRPS, Bool calledFromSliceHeader );
81#else
82  Void codeShortTermRefPicSet              ( TComSPS* pcSPS, TComReferencePictureSet* pcRPS );
83#endif
84#if LTRP_IN_SPS
85  Bool findMatchingLTRP ( TComSlice* pcSlice, UInt *ltrpsIndex, Int ltrpPOC, Bool usedFlag );
86#endif 
87 
88public:
89 
90  Void  resetEntropy          ();
91  Void  determineCabacInitIdx  () {};
92
93  Void  setBitstream          ( TComBitIf* p )  { m_pcBitIf = p;  }
94  Void  setSlice              ( TComSlice* p )  { m_pcSlice = p;  }
95  Void  resetBits             ()                { m_pcBitIf->resetBits(); }
96  Void  resetCoeffCost        ()                { m_uiCoeffCost = 0;  }
97  UInt  getNumberOfWrittenBits()                { return  m_pcBitIf->getNumberOfWrittenBits();  }
98  UInt  getCoeffCost          ()                { return  m_uiCoeffCost;  }
99  Void  codeVPS                 ( TComVPS* pcVPS );
100#if SUPPORT_FOR_VUI
101  Void  codeVUI                 ( TComVUI *pcVUI, TComSPS* pcSPS );
102#endif
103  Void  codeSPS                 ( TComSPS* pcSPS );
104  Void  codePPS                 ( TComPPS* pcPPS );
105  Void  codeSliceHeader         ( TComSlice* pcSlice );
106#if PROFILE_TIER_LEVEL_SYNTAX
107  Void  codePTL                 ( TComPTL* pcPTL, Bool profilePresentFlag, Int maxNumSubLayersMinus1);
108  Void  codeProfileTier         ( ProfileTierLevel* ptl );
109#endif
110  Void  codeTilesWPPEntryPoint( TComSlice* pSlice );
111  Void  codeTerminatingBit      ( UInt uilsLast );
112  Void  codeSliceFinish         ();
113  Void  codeFlush               () {}
114  Void  encodeStart             () {}
115 
116  Void codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList );
117  Void xGolombEncode(Int coeff, Int k);
118#if !REMOVE_ALF
119  Void codeAlfParam(ALFParam* alfParam);
120#endif
121  Void codeSAOSign       ( UInt code   ) { printf("Not supported\n"); assert (0); }
122  Void codeSaoMaxUvlc    ( UInt   code, UInt maxSymbol ){printf("Not supported\n"); assert (0);}
123#if SAO_MERGE_ONE_CTX
124  Void codeSaoMerge  ( UInt uiCode ){printf("Not supported\n"); assert (0);}
125#else
126  Void codeSaoMergeLeft  ( UInt uiCode, UInt compIdx ){printf("Not supported\n"); assert (0);}
127  Void codeSaoMergeUp    ( UInt uiCode ){printf("Not supported\n"); assert (0);}
128#endif
129  Void codeSaoTypeIdx    ( UInt uiCode ){printf("Not supported\n"); assert (0);}
130#if SAO_TYPE_CODING
131  Void codeSaoUflc       ( UInt uiLength, UInt   uiCode ){ assert(uiCode < 32); printf("Not supported\n"); assert (0);}
132#else
133  Void codeSaoUflc       ( UInt uiCode ){ assert(uiCode < 32); printf("Not supported\n"); assert (0);}
134#endif
135
136  Void codeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx );
137  Void codeSkipFlag      ( TComDataCU* pcCU, UInt uiAbsPartIdx );
138  Void codeMergeFlag     ( TComDataCU* pcCU, UInt uiAbsPartIdx );
139  Void codeMergeIndex    ( TComDataCU* pcCU, UInt uiAbsPartIdx );
140#if INTRA_BL
141  Void codeIntraBLFlag   ( TComDataCU* pcCU, UInt uiAbsPartIdx );
142#endif
143  Void codeApsExtensionFlag ();
144
145#if !REMOVE_FGS
146  /// set slice granularity
147  Void setSliceGranularity(Int iSliceGranularity)  {m_iSliceGranularity = iSliceGranularity;}
148
149  ///get slice granularity
150  Int  getSliceGranularity()                       {return m_iSliceGranularity;             }
151#endif
152 
153  Void codeAlfCtrlFlag   ( Int compIdx, UInt code ) {printf("Not supported\n"); assert(0);}
154  Void codeInterModeFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiEncMode );
155  Void codeSplitFlag     ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth );
156 
157  Void codePartSize      ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth );
158  Void codePredMode      ( TComDataCU* pcCU, UInt uiAbsPartIdx );
159 
160  Void codeIPCMInfo      ( TComDataCU* pcCU, UInt uiAbsPartIdx, Int numIPCM, Bool firstIPCMFlag);
161
162  Void codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx );
163  Void codeQtCbf         ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth );
164  Void codeQtRootCbf     ( TComDataCU* pcCU, UInt uiAbsPartIdx );
165#if TU_ZERO_CBF_RDO
166  Void codeQtCbfZero     ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth );
167  Void codeQtRootCbfZero ( TComDataCU* pcCU, UInt uiAbsPartIdx );
168#endif
169  Void codeIntraDirLumaAng( TComDataCU* pcCU, UInt absPartIdx, Bool isMultiple);
170  Void codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx );
171  Void codeInterDir      ( TComDataCU* pcCU, UInt uiAbsPartIdx );
172  Void codeRefFrmIdx     ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList );
173  Void codeMvd           ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList );
174 
175  Void codeDeltaQP       ( TComDataCU* pcCU, UInt uiAbsPartIdx );
176 
177  Void codeCoeffNxN      ( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType );
178  Void codeTransformSkipFlags ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, UInt uiDepth, TextType eTType );
179
180  Void estBit               (estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType);
181 
182  Void xCodePredWeightTable          ( TComSlice* pcSlice );
183  Void updateContextTables           ( SliceType eSliceType, Int iQp, Bool bExecuteFinish=true ) { return;   }
184  Void updateContextTables           ( SliceType eSliceType, Int iQp  )                          { return;   }
185
186#if !REMOVE_APS
187  Void  codeAPSInitInfo(TComAPS* pcAPS);  //!< code APS flags before encoding SAO and ALF parameters
188#endif
189  Void  codeFinish(Bool bEnd) { /*do nothing*/}
190  Void codeScalingList  ( TComScalingList* scalingList );
191  Void xCodeScalingList ( TComScalingList* scalingList, UInt sizeId, UInt listId);
192  Void codeDFFlag       ( UInt uiCode, const Char *pSymbolName );
193  Void codeDFSvlc       ( Int   iCode, const Char *pSymbolName );
194
195};
196
197//! \}
198
199#endif // !defined(AFX_TENCCAVLC_H__EE8A0B30_945B_4169_B290_24D3AD52296F__INCLUDED_)
200
Note: See TracBrowser for help on using the repository browser.