source: SHVCSoftware/branches/SHM-dev/source/App/TAppDecoder/TAppDecTop.h @ 1103

Last change on this file since 1103 was 1100, checked in by seregin, 9 years ago

macro cleanup: RESOLUTION_BASED_DPB

  • Property svn:eol-style set to native
File size: 7.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-2014, 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#if ALIGNED_BUMPING
58struct DpbStatus;
59#endif
60/// decoder application class
61class TAppDecTop : public TAppDecCfg
62{
63private:
64  // class interface
65#if SVC_EXTENSION
66  TDecTop                         m_acTDecTop [MAX_LAYERS];                             ///< decoder class [WARNING] suppose to use MAX_NUM_LAYER_IDS but decoder crashes, pending for further investigation
67  TDecTop*                        m_apcTDecTop [MAX_NUM_LAYER_IDS];                     ///< decoder point class
68  TVideoIOYuv                     m_acTVideoIOYuvReconFile [MAX_NUM_LAYER_IDS];         ///< reconstruction YUV class
69#else
70  TDecTop                         m_cTDecTop;                     ///< decoder class
71  TVideoIOYuv                     m_cTVideoIOYuvReconFile;        ///< reconstruction YUV class
72#endif
73 
74  // for output control 
75#if SVC_EXTENSION
76#if CONFORMANCE_BITSTREAM_MODE
77  TVideoIOYuv                      m_confReconFile[63];        ///< decode YUV files
78#endif
79  Int                             m_aiPOCLastDisplay [MAX_LAYERS]; ///< last POC in display order
80#else
81  Int                             m_iPOCLastDisplay;              ///< last POC in display order
82#endif
83  std::ofstream                   m_seiMessageFileStream;         ///< Used for outputing SEI messages. 
84
85public:
86  TAppDecTop();
87  virtual ~TAppDecTop() {}
88
89  Void  create            (); ///< create internal members
90  Void  destroy           (); ///< destroy internal members
91  Void  decode            (); ///< main decoding function
92
93protected:
94  Void  xCreateDecLib     (); ///< create internal classes
95  Void  xDestroyDecLib    (); ///< destroy internal classes
96  Void  xInitDecLib       (); ///< initialize decoder class
97 
98#if SVC_EXTENSION
99  Void  xWriteOutput      ( TComList<TComPic*>* pcListPic, UInt layerId, UInt tId ); ///< write YUV to file
100  Void  xFlushOutput      ( TComList<TComPic*>* pcListPic, UInt layerId ); ///< flush all remaining decoded pictures to file
101#else
102  Void  xWriteOutput      ( TComList<TComPic*>* pcListPic , UInt tId); ///< write YUV to file
103  Void  xFlushOutput      ( TComList<TComPic*>* pcListPic ); ///< flush all remaining decoded pictures to file
104#endif
105  Bool  isNaluWithinTargetDecLayerIdSet ( InputNALUnit* nalu ); ///< check whether given Nalu is within targetDecLayerIdSet
106#if ALIGNED_BUMPING
107  Void checkOutputBeforeDecoding(Int layerIdx);
108  Void checkOutputAfterDecoding();
109  Void flushAllPictures(Bool outputPictures); 
110  Void flushAllPictures(Int layerId, Bool outputPictures);
111
112  Void xOutputAndMarkPic( TComPic *pic, const Char *reconFile, const Int layerId, Int &pocLastDisplay, DpbStatus &dpbStatus);
113#if POC_RESET_IDC_DECODER
114  Void outputAllPictures(Int layerId, Bool notOutputCurrAu);
115#endif
116  Void xFindDPBStatus( std::vector<Int> &listOfPocs
117                            , std::vector<Int> *listOfPocsInEachLayer
118                            , std::vector<Int> *listOfPocsPositionInEachLayer
119                            , DpbStatus &dpbStatus
120#if POC_RESET_IDC_DECODER
121                            , Bool notOutputCurrAu = true
122#endif
123                            );
124
125  Bool ifInvokeBumpingBeforeDecoding( const DpbStatus &dpbStatus, const DpbStatus &dpbLimit, const Int layerIdx, const Int subDpbIdx );
126  Bool ifInvokeBumpingAfterDecoding ( const DpbStatus &dpbStatus, const DpbStatus &dpbLimit );
127  Void bumpingProcess(std::vector<Int> &listOfPocs, std::vector<Int> *listOfPocsInEachLayer, std::vector<Int> *listOfPocsPositionInEachLayer, DpbStatus &dpbStatus);
128  Void emptyUnusedPicturesNotNeededForOutput();
129  Void markAllPicturesAsErased();
130  Void markAllPicturesAsErased(Int layerIdx);
131  TComVPS* findDpbParametersFromVps(std::vector<Int> const &listOfPocs, std::vector<Int> const *listOfPocsInEachLayer, std::vector<Int> const *listOfPocsPositionInEachLayer, DpbStatus &maxDpbLimit);
132#endif
133};
134
135#if ALIGNED_BUMPING
136struct DpbStatus
137{
138  // Number of AUs and pictures
139  Int m_numAUsNotDisplayed;
140  Int m_numPicsNotDisplayedInLayer[MAX_LAYERS];
141  Int m_numPicsInSubDpb[MAX_LAYERS];  // Pictures marked as used_for_reference or needed for output in the sub-DPB
142#if FIX_ALIGN_BUMPING
143  Int m_layerIdToSubDpbIdMap[MAX_VPS_LAYER_IDX_PLUS1];
144  Int m_targetDecLayerIdList[MAX_LAYERS];
145#endif
146  Bool m_maxLatencyIncrease;
147  Int m_maxLatencyPictures;
148 
149  Int m_numSubDpbs;
150  Int m_numLayers;
151
152  DpbStatus()
153  {
154    init();
155  }
156  Void init()
157  {
158    m_numAUsNotDisplayed = 0;
159    m_maxLatencyIncrease  = false;
160    m_maxLatencyPictures  = 0;
161    ::memset( m_numPicsInSubDpb, 0, sizeof(m_numPicsInSubDpb) );
162    ::memset(m_numPicsNotDisplayedInLayer, 0, sizeof(m_numPicsNotDisplayedInLayer) );
163    m_numSubDpbs = -1;
164    m_numLayers = -1;
165#if FIX_ALIGN_BUMPING
166    ::memset( m_targetDecLayerIdList, 0, sizeof(m_targetDecLayerIdList) );
167    for(Int i = 0; i < MAX_VPS_LAYER_IDX_PLUS1; i++)
168    {
169      m_layerIdToSubDpbIdMap[i] = -1;
170    }
171#endif
172  }
173};
174#endif
175//! \}
176
177#endif
178
Note: See TracBrowser for help on using the repository browser.