source: 3DVCSoftware/branches/HTM-15.2-dev/source/App/TAppDecoder/TAppDecTop.h @ 1360

Last change on this file since 1360 was 1360, checked in by tech, 9 years ago

Update to HM-16.7.

  • Property svn:eol-style set to native
File size: 10.2 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-2015, 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     TAppDecTop.h
35    \brief    Decoder application class (header)
36*/
37
38#ifndef __TAPPDECTOP__
39#define __TAPPDECTOP__
40
41#if _MSC_VER > 1000
42#pragma once
43#endif // _MSC_VER > 1000
44
45#include "TLibVideoIO/TVideoIOYuv.h"
46#include "TLibCommon/TComList.h"
47#include "TLibCommon/TComPicYuv.h"
48#include "TLibDecoder/TDecTop.h"
49#include "TAppDecCfg.h"
50
51//! \ingroup TAppDecoder
52//! \{
53
54// ====================================================================================================================
55// Class definition
56// ====================================================================================================================
57
58/// decoder application class
59class TAppDecTop : public TAppDecCfg
60{
61private:
62  // class interface 
63#if NH_MV
64  TDecTop*                        m_tDecTop             [ MAX_NUM_LAYERS ];    ///< decoder classes
65  TVideoIOYuv*                    m_tVideoIOYuvReconFile[ MAX_NUM_LAYERS ];    ///< reconstruction YUV class
66  Int                             m_layerIdToDecIdx     [ MAX_NUM_LAYER_IDS ]; ///< mapping from layer id to decoder index
67  Int                             m_numDecoders;                               ///< number of decoder instances
68  TComPicLists                    m_dpb;                                       ///< picture buffers of decoder instances
69
70  TComPic*                        m_curPic;                                    ///< currently decoded picture
71  TComAu                          m_curAu;                                     ///< currently decoded Au
72 
73  // Random access related
74  Bool                            m_handleCraAsBlaFlag; 
75  Bool                            m_handleCraAsBlaFlagSetByExtMeans; 
76  Bool                            m_noClrasOutputFlag; 
77  Bool                            m_noClrasOutputFlagSetByExtMeans; 
78  Bool                            m_noRaslOutputFlagAssocIrap           [ MAX_NUM_LAYER_IDS ]; 
79
80  // Layer wise startup
81  Bool                            m_firstPicInLayerDecodedFlag          [ MAX_NUM_LAYER_IDS ];
82  Bool                            m_layerInitilizedFlag                 [ MAX_NUM_LAYER_IDS ]; 
83  Bool                            m_layerResetFlag; 
84
85  // DPB related variables
86  Int                             m_maxNumReorderPics; 
87  Int                             m_maxLatencyIncreasePlus1; 
88  Int                             m_maxLatencyValue; 
89  Int                             m_maxDecPicBufferingMinus1            [ MAX_NUM_LAYER_IDS ];
90
91  // Poc resetting
92  Int                             m_lastPresentPocResetIdc              [ MAX_NUM_LAYER_IDS ];
93  Bool                            m_firstPicInPocResettingPeriodReceived[ MAX_NUM_LAYER_IDS ];
94  Bool                            m_pocDecrementedInDpbFlag             [ MAX_NUM_LAYER_IDS ]; 
95  Bool                            m_newPicIsFstPicOfAllLayOfPocResetPer;
96  Bool                            m_newPicIsPocResettingPic;
97
98  // General decoding state
99  Bool                            m_newVpsActivatedbyCurAu; 
100  Bool                            m_newVpsActivatedbyCurPic; 
101  Bool                            m_eosInLayer                          [ MAX_NUM_LAYER_IDS ];
102  Bool                            m_initilizedFromVPS;
103  Bool                            m_firstSliceInBitstream;
104  UInt64                          m_decodingOrder                       [ MAX_NUM_LAYER_IDS ];
105  UInt64                          m_totalNumofPicsReceived;
106  Bool                            m_cvsStartFound; 
107  Int                             m_smallestLayerId;
108
109  // Decoding processes for current  picture
110  DecodingProcess                 m_decProcPocAndRps;
111  DecodingProcess                 m_decProcCvsg; 
112 
113  // Active parameter sets
114  const TComPPS*                  m_pps;                                ///< active PPS
115  const TComSPS*                  m_sps;                                ///< active SPS
116  const TComVPS*                  m_vps;                                ///< active VPS
117
118  Bool                            m_reconOpen           [ MAX_NUM_LAYERS ]; ///< reconstruction file opened
119#if NH_3D
120  FILE*                           m_pScaleOffsetFile;
121  CamParsCollector                m_cCamParsCollector;
122#endif
123#else
124  TDecTop                         m_cTDecTop;                     ///< decoder class
125  TVideoIOYuv                     m_cTVideoIOYuvReconFile;        ///< reconstruction YUV class
126  // for output control
127  Int                             m_iPOCLastDisplay;              ///< last POC in display order
128#endif
129  std::ofstream                   m_seiMessageFileStream;         ///< Used for outputing SEI messages. 
130
131  SEIColourRemappingInfo*         m_pcSeiColourRemappingInfoPrevious;
132
133public:
134  TAppDecTop();
135  virtual ~TAppDecTop() {}
136
137  Void  create            (); ///< create internal members
138  Void  destroy           (); ///< destroy internal members
139  Void  decode            (); ///< main decoding function
140#if NH_MV
141  UInt  getNumberOfChecksumErrorsDetected( ) const;
142  UInt  getNumberOfChecksumErrorsDetected( Int decIdx ) const { return m_tDecTop[decIdx]->getNumberOfChecksumErrorsDetected(); }
143#else
144  UInt  getNumberOfChecksumErrorsDetected() const { return m_cTDecTop.getNumberOfChecksumErrorsDetected(); }
145#endif
146
147protected:
148
149  Void  xCreateDecLib     (); ///< create internal classes
150  Void  xDestroyDecLib    (); ///< destroy internal classes
151  Void  xInitDecLib       (); ///< initialize decoder class
152
153#if !NH_MV
154  Void  xWriteOutput      ( TComList<TComPic*>* pcListPic , UInt tId); ///< write YUV to file
155  Void  xFlushOutput      ( TComList<TComPic*>* pcListPic ); ///< flush all remaining decoded pictures to file
156  Bool  isNaluWithinTargetDecLayerIdSet    ( InputNALUnit* nalu ); ///< check whether given Nalu is within targetDecLayerIdSet
157
158#else
159  // Process NAL units
160  Bool xExtractAndRewrite                  ( InputNALUnit* nalu );
161  Void xProcessVclNalu                     ( InputNALUnit nalu );
162  Bool xIsSkipVclNalu                      ( InputNALUnit& nalu, Bool isFirstSliceOfPic );
163  Void xProcessNonVclNalu                  ( InputNALUnit nalu );
164  Void xTerminateDecoding                  ( );
165
166  // Process slice
167  Void xDecodeFirstSliceOfPicture          ( InputNALUnit nalu, Bool sliceIsFirstOfNewAu ); 
168  Void xDecodeFollowSliceOfPicture         ( InputNALUnit nalu );
169
170  // Process picture 
171  Void xFinalizePreviousPictures           ( Bool sliceIsFirstOfNewAU );
172  Void xFinalizePic                        ( Bool curPicIsLastInAu );
173  Void xFinalizeAU                         ( );
174  Void xPicDecoding                        ( DecProcPart curPart, Bool picPosInAuIndication ); 
175
176  // Clause 8
177  Void x812CvsgDecodingProcess             ( Int decIdx );
178  Void x813decProcForCodPicWithLIdZero     ( DecProcPart curPart );
179
180  // Annex C (DPB)
181  Void xC522OutputAndRemOfPicsFromDpb      ( );
182  Void xC523PicDecMarkAddBumpAndStor       ( );
183  Void xC524Bumping                        ( );
184
185  // Annex F.8
186  Void xF811GeneralDecProc                 ( InputNALUnit nalu );
187  Void xF812CvsgDecodingProcess            ( Int decIdx );
188  Void xF813ComDecProcForACodedPic         ( DecProcPart curPart, Bool picPosInAuIndication );
189  Void xF814decProcForCodPicWithLIdZero    ( DecProcPart curPart );
190  Void xF816decProcEndDecOfCodPicLIdGrtZero( );
191
192  // Annex F.13 (DPB)
193  Void xF13521InitDpb                      ( );
194  Void xF13522OutputAndRemOfPicsFromDpb    ( Bool beforePocDerivation );
195  Void xF13523PicDecMarkAddBumpAndStor     ( Bool curPicIsLastInAu   );
196  Void xF13524Bumping                      ( TComList<TComAu*> aus );
197
198  // Helpers
199  TDecTop* xGetDecoder                     ( InputNALUnit& nalu ); 
200  Int   xGetDecoderIdx                     ( Int layerId, Bool createFlag = false );
201  Int   xPreDecodePoc                      ( InputNALUnit& nalu );
202  Bool  xDetectNewAu                       ( InputNALUnit& nalu );
203  Void  xDetectNewPocResettingPeriod       ( InputNALUnit& nalu );
204  Bool  xIsNaluInTargetDecLayerIdSet       ( InputNALUnit* nalu ); ///< check whether given Nalu is within targetDecLayerIdSet
205  Bool  xAllRefLayersInitilized            ( Int curLayerId );
206  Void  xInitFileIO                        ( ); 
207  Void  xOpenReconFile                     ( TComPic* curPic );
208  Void  xFlushOutput                       ( );
209  Void  xCropAndOutput                     ( TComPic* curPic ); 
210#endif
211
212private:
213  Void applyColourRemapping(const TComPicYuv& pic, SEIColourRemappingInfo& pCriSEI, const TComSPS &activeSPS);
214  Void xOutputColourRemapPic(TComPic* pcPic);
215};
216
217//! \}
218
219#endif
Note: See TracBrowser for help on using the repository browser.