source: 3DVCSoftware/trunk/source/App/TAppEncoder/TAppEncTop.cpp @ 21

Last change on this file since 21 was 21, checked in by hschwarz, 12 years ago

updated with HHI branch (0.2-HHI)

  • Property svn:eol-style set to native
File size: 50.7 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-2011, 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 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
35
36/** \file     TAppEncTop.cpp
37    \brief    Encoder application class
38*/
39
40#include <list>
41#include <stdio.h>
42#include <fcntl.h>
43#include <assert.h>
44
45#include "TAppEncTop.h"
46
47// ====================================================================================================================
48// Constructor / destructor / initialization / destroy
49// ====================================================================================================================
50
51TAppEncTop::TAppEncTop()
52{
53#if HHI_VSO
54  m_iLastFramePutInERViewBuffer = -1;
55#endif
56}
57
58TAppEncTop::~TAppEncTop()
59{
60}
61
62
63Void TAppEncTop::xInitLibCfg()
64{
65  for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
66  {
67    m_iFrameRcvdVector.push_back(0) ;
68    m_acTEncTopList.push_back(new TEncTop);
69
70    m_acTVideoIOYuvInputFileList.push_back(new TVideoIOYuv);
71
72    m_acTVideoIOYuvReconFileList.push_back(new TVideoIOYuv);
73    m_cListPicYuvRecList.push_back(new TComList<TComPicYuv*>) ;
74    m_aiNextPocToDump.push_back( 0 );
75    m_cListPicYuvRecMap.push_back( std::map<PicOrderCnt,TComPicYuv*>() );
76    m_acTEncTopList[iViewIdx]->setFrameRate                    ( m_iFrameRate );
77    m_acTEncTopList[iViewIdx]->setFrameSkip                    ( m_FrameSkip );
78    m_acTEncTopList[iViewIdx]->setSourceWidth                  ( m_iSourceWidth );
79    m_acTEncTopList[iViewIdx]->setSourceHeight                 ( m_iSourceHeight );
80    m_acTEncTopList[iViewIdx]->setFrameToBeEncoded             ( m_iFrameToBeEncoded );
81
82  //====== Coding Structure ========
83#if DCM_DECODING_REFRESH
84    m_acTEncTopList[iViewIdx]->setDecodingRefreshType          ( m_iDecodingRefreshType );
85#endif
86    m_acTEncTopList[iViewIdx]->setCPSSize                      ( m_uiCodedPictureStoreSize );
87    m_acTEncTopList[iViewIdx]->setGOPSize                      ( m_iGOPSize );
88    m_acTEncTopList[iViewIdx]->setRateGOPSize                  ( m_iRateGOPSize );
89
90    m_acTEncTopList[iViewIdx]->setSeqStructure                              ( m_cInputFormatString );
91
92    m_acTEncTopList[iViewIdx]->setQP                           ( m_aiQP[0] );
93
94    m_acTEncTopList[iViewIdx]->setTemporalLayerQPOffset        ( m_aiTLayerQPOffset );
95    m_acTEncTopList[iViewIdx]->setPad                          ( m_aiPad );
96
97    //===== Slice ========
98
99    //====== Entropy Coding ========
100    m_acTEncTopList[iViewIdx]->setSymbolMode                   ( m_iSymbolMode );
101
102    //====== Loop/Deblock Filter ========
103    m_acTEncTopList[iViewIdx]->setLoopFilterDisable            ( m_abLoopFilterDisable[0]   );
104    m_acTEncTopList[iViewIdx]->setLoopFilterAlphaC0Offset      ( m_iLoopFilterAlphaC0Offset );
105    m_acTEncTopList[iViewIdx]->setLoopFilterBetaOffset         ( m_iLoopFilterBetaOffset    );
106
107    //====== Motion search ========
108    m_acTEncTopList[iViewIdx]->setFastSearch                   ( m_iFastSearch  );
109    m_acTEncTopList[iViewIdx]->setSearchRange                  ( m_iSearchRange );
110    m_acTEncTopList[iViewIdx]->setBipredSearchRange            ( m_bipredSearchRange );
111    m_acTEncTopList[iViewIdx]->setMaxDeltaQP                   ( m_iMaxDeltaQP  );
112
113#if HHI_VSO
114    //====== VSO =========
115    m_acTEncTopList[iViewIdx]->setForceLambdaScaleVSO          ( false );
116    m_acTEncTopList[iViewIdx]->setLambdaScaleVSO               ( 1     );
117    m_acTEncTopList[iViewIdx]->setVSOMode                      ( 0     );
118#endif
119
120    //====== Tool list ========
121    m_acTEncTopList[iViewIdx]->setUseSBACRD                    ( m_bUseSBACRD   );
122    m_acTEncTopList[iViewIdx]->setDeltaQpRD                    ( m_uiDeltaQpRD  );
123    m_acTEncTopList[iViewIdx]->setUseASR                       ( m_bUseASR      );
124    m_acTEncTopList[iViewIdx]->setUseHADME                     ( m_bUseHADME    );
125    m_acTEncTopList[iViewIdx]->setUseALF                       ( m_abUseALF[0]  );
126#if MQT_ALF_NPASS
127    m_acTEncTopList[iViewIdx]->setALFEncodePassReduction       ( m_iALFEncodePassReduction );
128#endif
129#if DCM_COMB_LIST
130    m_acTEncTopList[iViewIdx]->setUseLComb                     ( m_bUseLComb    );
131    m_acTEncTopList[iViewIdx]->setLCMod                        ( m_bLCMod         );
132#endif
133    m_acTEncTopList[iViewIdx]->setdQPs                         ( m_aidQP        );
134    m_acTEncTopList[iViewIdx]->setUseRDOQ                      ( m_abUseRDOQ[0] );
135    m_acTEncTopList[iViewIdx]->setUseLDC                       ( m_bUseLDC      );
136    m_acTEncTopList[iViewIdx]->setUsePAD                       ( m_bUsePAD      );
137    m_acTEncTopList[iViewIdx]->setQuadtreeTULog2MaxSize        ( m_uiQuadtreeTULog2MaxSize );
138    m_acTEncTopList[iViewIdx]->setQuadtreeTULog2MinSize        ( m_uiQuadtreeTULog2MinSize );
139    m_acTEncTopList[iViewIdx]->setQuadtreeTUMaxDepthInter      ( m_uiQuadtreeTUMaxDepthInter );
140    m_acTEncTopList[iViewIdx]->setQuadtreeTUMaxDepthIntra      ( m_uiQuadtreeTUMaxDepthIntra );
141    m_acTEncTopList[iViewIdx]->setUseFastEnc                   ( m_bUseFastEnc  );
142
143#if HHI_VSO
144    m_acTEncTopList[iViewIdx]->setUseVSO                       ( false ); //GT: might be enabled later for VSO Mode 4
145#endif
146
147    m_acTEncTopList[iViewIdx]->setViewId                       ( (UInt)iViewIdx );
148    m_acTEncTopList[iViewIdx]->setViewOrderIdx                 ( m_cCameraData.getViewOrderIndex()[ iViewIdx ] );
149    m_acTEncTopList[iViewIdx]->setIsDepth                      ( false );
150    m_acTEncTopList[iViewIdx]->setCamParPrecision              ( m_cCameraData.getCamParsCodedPrecision  () );
151    m_acTEncTopList[iViewIdx]->setCamParInSliceHeader          ( m_cCameraData.getVaryingCameraParameters() );
152    m_acTEncTopList[iViewIdx]->setCodedScale                   ( m_cCameraData.getCodedScale             () );
153    m_acTEncTopList[iViewIdx]->setCodedOffset                  ( m_cCameraData.getCodedOffset            () );
154#if DEPTH_MAP_GENERATION
155    m_acTEncTopList[iViewIdx]->setPredDepthMapGeneration       ( m_uiPredDepthMapGeneration );
156    m_acTEncTopList[iViewIdx]->setPdmPrecision                 ( (UInt)m_cCameraData.getPdmPrecision     () );
157    m_acTEncTopList[iViewIdx]->setPdmScaleNomDelta             (       m_cCameraData.getPdmScaleNomDelta () );
158    m_acTEncTopList[iViewIdx]->setPdmOffset                    (       m_cCameraData.getPdmOffset        () );
159#endif
160#if HHI_INTER_VIEW_MOTION_PRED
161    m_acTEncTopList[iViewIdx]->setMultiviewMvPredMode          ( m_uiMultiviewMvPredMode );
162    m_acTEncTopList[iViewIdx]->setMultiviewMvRegMode           ( iViewIdx ? m_uiMultiviewMvRegMode       : 0   );
163    m_acTEncTopList[iViewIdx]->setMultiviewMvRegLambdaScale    ( iViewIdx ? m_dMultiviewMvRegLambdaScale : 0.0 );
164#endif
165#if HHI_INTER_VIEW_RESIDUAL_PRED
166    m_acTEncTopList[iViewIdx]->setMultiviewResPredMode         ( m_uiMultiviewResPredMode );
167#endif
168
169
170#if HHI_INTERVIEW_SKIP
171    m_acTEncTopList[iViewIdx]->setInterViewSkip            ( iViewIdx ? m_uiInterViewSkip : 0 );
172#if HHI_INTERVIEW_SKIP_LAMBDA_SCALE
173    m_acTEncTopList[iViewIdx]->setInterViewSkipLambdaScale ( iViewIdx ? (Int)m_dInterViewSkipLambdaScale : 1 );
174#endif
175#endif
176    m_acTEncTopList[iViewIdx]->setUseMRG                       ( m_bUseMRG      ); // SOPH:
177
178#if LM_CHROMA
179    m_acTEncTopList[iViewIdx]->setUseLMChroma                  ( m_bUseLMChroma );
180#endif
181
182#if HHI_RMP_SWITCH
183    m_acTEncTopList[iViewIdx]->setUseRMP                     ( m_bUseRMP );
184#endif
185#ifdef ROUNDING_CONTROL_BIPRED
186    m_acTEncTopList[iViewIdx]->setUseRoundingControlBipred(m_useRoundingControlBipred);
187#endif
188#if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX
189    m_acTEncTopList[iViewIdx]->setUseDMM( false );
190#endif
191#if CONSTRAINED_INTRA_PRED
192    m_acTEncTopList[iViewIdx]->setUseConstrainedIntraPred      ( m_bUseConstrainedIntraPred );
193#endif
194#ifdef WEIGHT_PRED
195    //====== Weighted Prediction ========
196    m_acTEncTopList[iViewIdx]->setUseWP                         ( m_bUseWeightPred      );
197    m_acTEncTopList[iViewIdx]->setWPBiPredIdc                   ( m_uiBiPredIdc         );
198#endif
199    //====== Slice ========
200    m_acTEncTopList[iViewIdx]->setSliceMode               ( m_iSliceMode                );
201    m_acTEncTopList[iViewIdx]->setSliceArgument           ( m_iSliceArgument            );
202
203    //====== Entropy Slice ========
204    m_acTEncTopList[iViewIdx]->setEntropySliceMode        ( m_iEntropySliceMode         );
205    m_acTEncTopList[iViewIdx]->setEntropySliceArgument    ( m_iEntropySliceArgument     );
206#if MTK_NONCROSS_INLOOP_FILTER
207    if(m_iSliceMode == 0 )
208    {
209      m_bLFCrossSliceBoundaryFlag = true;
210    }
211    m_acTEncTopList[iViewIdx]->setLFCrossSliceBoundaryFlag( m_bLFCrossSliceBoundaryFlag );
212#endif
213#if MTK_SAO
214    m_acTEncTopList[iViewIdx]->setUseSAO               ( m_abUseSAO[0]     );
215#endif
216#if HHI_MPI
217    m_acTEncTopList[iViewIdx]->setUseMVI( false );
218#endif
219
220    m_acTEncTopList[iViewIdx]->setPictureDigestEnabled(m_pictureDigestEnabled);
221    m_acTEncTopList[iViewIdx]->setQpChangeFrame( m_iQpChangeFrame );
222    m_acTEncTopList[iViewIdx]->setQpChangeOffsetVideo( m_iQpChangeOffsetVideo );
223    m_acTEncTopList[iViewIdx]->setQpChangeOffsetDepth( m_iQpChangeOffsetDepth );
224  }
225  if( m_bUsingDepthMaps )
226  {
227
228#if HHI_VSO
229    for (Int iViewIdx=0; iViewIdx<m_iNumberOfExternalRefs; iViewIdx++)
230    {
231        m_acTVideoIOYuvERFileList.push_back(new TVideoIOYuv);
232    }
233#endif
234
235    for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
236    {
237      m_iDepthFrameRcvdVector.push_back(0) ;
238      m_acTEncDepthTopList.push_back(new TEncTop);
239
240      m_acTVideoIOYuvDepthInputFileList.push_back(new TVideoIOYuv);
241
242      m_acTVideoIOYuvDepthReconFileList.push_back(new TVideoIOYuv);
243      m_cListPicYuvDepthRecList.push_back(new TComList<TComPicYuv*>) ;
244      m_aiNextDepthPocToDump.push_back( 0 );
245      m_cListPicYuvDepthRecMap.push_back( std::map<PicOrderCnt,TComPicYuv*>() );
246      m_acTEncDepthTopList[iViewIdx]->setFrameRate                    ( m_iFrameRate );
247      m_acTEncDepthTopList[iViewIdx]->setFrameSkip                    ( m_FrameSkip );
248      m_acTEncDepthTopList[iViewIdx]->setSourceWidth                  ( m_iSourceWidth );
249      m_acTEncDepthTopList[iViewIdx]->setSourceHeight                 ( m_iSourceHeight );
250      m_acTEncDepthTopList[iViewIdx]->setFrameToBeEncoded             ( m_iFrameToBeEncoded );
251
252      //====== Coding Structure ========
253#if DCM_DECODING_REFRESH
254      m_acTEncDepthTopList[iViewIdx]->setDecodingRefreshType          ( m_iDecodingRefreshType );
255#endif
256      m_acTEncDepthTopList[iViewIdx]->setCPSSize                      ( m_uiCodedPictureStoreSize );
257      m_acTEncDepthTopList[iViewIdx]->setGOPSize                      ( m_iGOPSize );
258      m_acTEncDepthTopList[iViewIdx]->setRateGOPSize                  ( m_iRateGOPSize );
259
260      m_acTEncDepthTopList[iViewIdx]->setSeqStructure                              ( m_cInputFormatString );
261
262      m_acTEncDepthTopList[iViewIdx]->setQP                           ( m_aiQP[1] );
263
264      m_acTEncDepthTopList[iViewIdx]->setTemporalLayerQPOffset        ( m_aiTLayerQPOffset );
265      m_acTEncDepthTopList[iViewIdx]->setPad                          ( m_aiPad );
266
267      //===== Slice ========
268
269      //====== Entropy Coding ========
270      m_acTEncDepthTopList[iViewIdx]->setSymbolMode                   ( m_iSymbolMode );
271
272      //====== Loop/Deblock Filter ========
273      m_acTEncDepthTopList[iViewIdx]->setLoopFilterDisable            ( m_abLoopFilterDisable[1]   );
274      m_acTEncDepthTopList[iViewIdx]->setLoopFilterAlphaC0Offset      ( m_iLoopFilterAlphaC0Offset );
275      m_acTEncDepthTopList[iViewIdx]->setLoopFilterBetaOffset         ( m_iLoopFilterBetaOffset    );
276
277      //====== Motion search ========
278      m_acTEncDepthTopList[iViewIdx]->setFastSearch                   ( m_iFastSearch  );
279      m_acTEncDepthTopList[iViewIdx]->setSearchRange                  ( m_iSearchRange );
280      m_acTEncDepthTopList[iViewIdx]->setBipredSearchRange            ( m_bipredSearchRange );
281      m_acTEncDepthTopList[iViewIdx]->setMaxDeltaQP                   ( m_iMaxDeltaQP  );
282
283      //====== Tool list ========
284      m_acTEncDepthTopList[iViewIdx]->setUseSBACRD                    ( m_bUseSBACRD   );
285      m_acTEncDepthTopList[iViewIdx]->setDeltaQpRD                    ( m_uiDeltaQpRD  );
286      m_acTEncDepthTopList[iViewIdx]->setUseASR                       ( m_bUseASR      );
287      m_acTEncDepthTopList[iViewIdx]->setUseHADME                     ( m_bUseHADME    );
288      m_acTEncDepthTopList[iViewIdx]->setUseALF                       ( m_abUseALF[1]  );
289#if MQT_ALF_NPASS
290      m_acTEncDepthTopList[iViewIdx]->setALFEncodePassReduction       ( m_iALFEncodePassReduction );
291#endif
292#if DCM_COMB_LIST
293      m_acTEncDepthTopList[iViewIdx]->setUseLComb                     ( m_bUseLComb    );
294      m_acTEncDepthTopList[iViewIdx]->setLCMod                        ( m_bLCMod         );
295#endif
296      m_acTEncDepthTopList[iViewIdx]->setdQPs                         ( m_aidQP        );
297      m_acTEncDepthTopList[iViewIdx]->setUseRDOQ                      ( m_abUseRDOQ[1] );
298      m_acTEncDepthTopList[iViewIdx]->setUseLDC                       ( m_bUseLDC      );
299      m_acTEncDepthTopList[iViewIdx]->setUsePAD                       ( m_bUsePAD      );
300      m_acTEncDepthTopList[iViewIdx]->setQuadtreeTULog2MaxSize        ( m_uiQuadtreeTULog2MaxSize );
301      m_acTEncDepthTopList[iViewIdx]->setQuadtreeTULog2MinSize        ( m_uiQuadtreeTULog2MinSize );
302      m_acTEncDepthTopList[iViewIdx]->setQuadtreeTUMaxDepthInter      ( m_uiQuadtreeTUMaxDepthInter );
303      m_acTEncDepthTopList[iViewIdx]->setQuadtreeTUMaxDepthIntra      ( m_uiQuadtreeTUMaxDepthIntra );
304      m_acTEncDepthTopList[iViewIdx]->setUseFastEnc                   ( m_bUseFastEnc  );
305
306#if HHI_VSO
307      m_acTEncDepthTopList[iViewIdx]->setUseVSO                       ( m_bUseVSO      ); //GT: might be enabled/disabled later for VSO Mode 4
308      m_acTEncDepthTopList[iViewIdx]->setForceLambdaScaleVSO          ( m_bForceLambdaScaleVSO );
309      m_acTEncDepthTopList[iViewIdx]->setLambdaScaleVSO               ( m_dLambdaScaleVSO );
310#if HHI_VSO_DIST_INT
311      m_acTEncDepthTopList[iViewIdx]->setAllowNegDist                 ( m_bAllowNegDist );
312#endif
313      m_acTEncDepthTopList[iViewIdx]->setVSOMode                      ( m_uiVSOMode );
314#endif
315
316      m_acTEncDepthTopList[iViewIdx]->setViewId                       ( (UInt)iViewIdx );
317      m_acTEncDepthTopList[iViewIdx]->setViewOrderIdx                 ( m_cCameraData.getViewOrderIndex()[ iViewIdx ] );
318      m_acTEncDepthTopList[iViewIdx]->setIsDepth                      ( true );
319      m_acTEncDepthTopList[iViewIdx]->setCamParPrecision              ( 0 );
320      m_acTEncDepthTopList[iViewIdx]->setCamParInSliceHeader          ( false );
321      m_acTEncDepthTopList[iViewIdx]->setCodedScale                   ( 0 );
322      m_acTEncDepthTopList[iViewIdx]->setCodedOffset                  ( 0 );
323#if DEPTH_MAP_GENERATION
324      m_acTEncDepthTopList[iViewIdx]->setPredDepthMapGeneration       ( 0 );
325#endif
326#if HHI_INTER_VIEW_MOTION_PRED
327      m_acTEncDepthTopList[iViewIdx]->setMultiviewMvPredMode          ( 0 );
328      m_acTEncDepthTopList[iViewIdx]->setMultiviewMvRegMode           ( 0 );
329      m_acTEncDepthTopList[iViewIdx]->setMultiviewMvRegLambdaScale    ( 0.0 );
330#endif
331#if HHI_INTER_VIEW_RESIDUAL_PRED
332      m_acTEncDepthTopList[iViewIdx]->setMultiviewResPredMode         ( 0 );
333#endif
334
335#if HHI_INTERVIEW_SKIP
336      m_acTEncDepthTopList[iViewIdx]->setInterViewSkip            ( 0 );
337#if HHI_INTERVIEW_SKIP_LAMBDA_SCALE
338      m_acTEncDepthTopList[iViewIdx]->setInterViewSkipLambdaScale ( 1 );
339#endif
340#endif
341      m_acTEncDepthTopList[iViewIdx]->setUseMRG                       ( m_bUseMRG      ); // SOPH:
342
343#if LM_CHROMA
344      m_acTEncDepthTopList[iViewIdx]->setUseLMChroma                  ( m_bUseLMChroma );
345#endif
346
347#if HHI_RMP_SWITCH
348      m_acTEncDepthTopList[iViewIdx]->setUseRMP                     ( m_bUseRMP );
349#endif
350#ifdef ROUNDING_CONTROL_BIPRED
351      m_acTEncDepthTopList[iViewIdx]->setUseRoundingControlBipred(m_useRoundingControlBipred);
352#endif
353#if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX
354      m_acTEncDepthTopList[iViewIdx]->setUseDMM( m_bUseDMM );
355#endif
356#if CONSTRAINED_INTRA_PRED
357      m_acTEncDepthTopList[iViewIdx]->setUseConstrainedIntraPred      ( m_bUseConstrainedIntraPred );
358#endif
359#ifdef WEIGHT_PRED
360    //====== Weighted Prediction ========
361      m_acTEncDepthTopList[iViewIdx]->setUseWP                         ( m_bUseWeightPred      );
362      m_acTEncDepthTopList[iViewIdx]->setWPBiPredIdc                   ( m_uiBiPredIdc         );
363#endif
364      //====== Slice ========
365      m_acTEncDepthTopList[iViewIdx]->setSliceMode               ( m_iSliceMode                );
366      m_acTEncDepthTopList[iViewIdx]->setSliceArgument           ( m_iSliceArgument            );
367
368      //====== Entropy Slice ========
369      m_acTEncDepthTopList[iViewIdx]->setEntropySliceMode        ( m_iEntropySliceMode         );
370      m_acTEncDepthTopList[iViewIdx]->setEntropySliceArgument    ( m_iEntropySliceArgument     );
371#if MTK_NONCROSS_INLOOP_FILTER
372      if(m_iSliceMode == 0 )
373      {
374        m_bLFCrossSliceBoundaryFlag = true;
375      }
376      m_acTEncDepthTopList[iViewIdx]->setLFCrossSliceBoundaryFlag( m_bLFCrossSliceBoundaryFlag );
377#endif
378#if MTK_SAO
379      m_acTEncDepthTopList[iViewIdx]->setUseSAO               ( m_abUseSAO[1]     );
380#endif
381#if HHI_MPI
382      m_acTEncDepthTopList[iViewIdx]->setUseMVI( m_bUseMVI );
383#endif
384
385      m_acTEncDepthTopList[iViewIdx]->setPictureDigestEnabled(m_pictureDigestEnabled);
386
387      m_acTEncDepthTopList[iViewIdx]->setQpChangeFrame( m_iQpChangeFrame );
388      m_acTEncDepthTopList[iViewIdx]->setQpChangeOffsetVideo( m_iQpChangeOffsetVideo );
389      m_acTEncDepthTopList[iViewIdx]->setQpChangeOffsetDepth( m_iQpChangeOffsetDepth );
390    }
391  }
392#if HHI_INTER_VIEW_MOTION_PRED
393  else if( m_uiMultiviewMvRegMode )
394  {
395    for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
396    {
397      m_acTVideoIOYuvDepthInputFileList.push_back( new TVideoIOYuv );
398    }
399  }
400#endif
401
402#if HHI_VSO
403  if ( m_bUseVSO )
404  {
405    if ( m_uiVSOMode == 4 )
406    {
407      m_cRendererModel.create( m_cRenModStrParser.getNumOfBaseViews(), m_cRenModStrParser.getNumOfModels(), m_iSourceWidth, m_iSourceHeight, LOG2_DISP_PREC_LUT, 0 );
408
409      for ( Int iViewNum = 0; iViewNum < m_iNumberOfViews; iViewNum++ )
410      {
411        for (Int iContent = 0; iContent < 2; iContent++ )
412        {
413          Int iNumOfModels   = m_cRenModStrParser.getNumOfModelsForView(iViewNum, iContent);
414          Bool bUseVSO = (iNumOfModels != 0);
415
416          TEncTop* pcEncTop = ( iContent == 0 ) ? m_acTEncTopList[iViewNum] : m_acTEncDepthTopList[iViewNum];
417          pcEncTop->setUseVSO( bUseVSO );
418          pcEncTop->getRdCost()->setRenModel( bUseVSO ? &m_cRendererModel : NULL );
419
420          for (Int iCurModel = 0; iCurModel < iNumOfModels; iCurModel++ )
421          {
422            Int iModelNum; Int iLeftViewNum; Int iRightViewNum; Int iDump; Int iOrgRefNum; Int iBlendMode;
423
424            m_cRenModStrParser.getSingleModelData  ( iViewNum, iContent, iCurModel, iModelNum, iBlendMode, iLeftViewNum, iRightViewNum, iOrgRefNum, iDump ) ;
425            m_cRendererModel  .createSingleModel   ( iViewNum, iContent, iModelNum, iLeftViewNum, iRightViewNum, (iOrgRefNum != -1), iBlendMode );
426          }
427        }
428      }
429    }
430    else
431    {
432      m_cRendererTop.init(m_iSourceWidth, m_iSourceHeight,true,0,0,true, 0,0,0,0,0,0,0,1,0,0 );  //GT: simplest configuration
433    }
434  }
435#endif
436
437#if HHI_INTERVIEW_SKIP
438  m_cUsedPelsRenderer.init(m_iSourceWidth, m_iSourceHeight, true, 0, LOG2_DISP_PREC_LUT, true, 0, 0, 0, 0, 0, 6, 4, 1, 0, 6 );
439#endif
440}
441
442Void TAppEncTop::xCreateLib()
443{
444  // Video I/O
445  m_cTVideoIOBitsFile.openBits( m_pchBitstreamFile, true  );  // write mode
446
447  for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
448  {
449    m_acTVideoIOYuvInputFileList[iViewIdx]->open( m_pchInputFileList[iViewIdx],     false, m_uiInputBitDepth, m_uiInternalBitDepth );  // read  mode
450    m_acTVideoIOYuvInputFileList[iViewIdx]->skipFrames(m_FrameSkip, m_iSourceWidth, m_iSourceHeight);
451    m_acTVideoIOYuvReconFileList[iViewIdx]->open( m_pchReconFileList[iViewIdx],     true, m_uiOutputBitDepth, m_uiInternalBitDepth);  // write mode
452
453    // Neo Decoder
454    m_acTEncTopList[iViewIdx]->create();
455
456    if( m_bUsingDepthMaps )
457    {
458      m_acTVideoIOYuvDepthInputFileList[iViewIdx]->open( m_pchDepthInputFileList[iViewIdx],     false, m_uiInputBitDepth, m_uiInternalBitDepth );  // read  mode
459      m_acTVideoIOYuvDepthInputFileList[iViewIdx]->skipFrames(m_FrameSkip, m_iSourceWidth, m_iSourceHeight);
460
461      m_acTVideoIOYuvDepthReconFileList[iViewIdx]->open( m_pchDepthReconFileList[iViewIdx],     true, m_uiOutputBitDepth, m_uiInternalBitDepth);  // write mode
462
463      // Neo Decoder
464      m_acTEncDepthTopList[iViewIdx]->create();
465    }
466#if HHI_INTER_VIEW_MOTION_PRED
467    else if( m_uiMultiviewMvRegMode )
468    {
469      m_acTVideoIOYuvDepthInputFileList[iViewIdx]->open( m_pchDepthInputFileList[iViewIdx],     false, m_uiInputBitDepth, m_uiInternalBitDepth );  // read  mode
470      m_acTVideoIOYuvDepthInputFileList[iViewIdx]->skipFrames(m_FrameSkip, m_iSourceWidth, m_iSourceHeight);
471    }
472#endif
473  }
474
475#if HHI_VSO
476  for(Int iViewIdx=0; iViewIdx < m_iNumberOfExternalRefs; iViewIdx++)
477  {
478    m_acTVideoIOYuvERFileList[iViewIdx]->open( m_pchERRefFileList[iViewIdx], false, m_uiInputBitDepth, m_uiInternalBitDepth ); // read mode
479  }
480#endif
481}
482
483Void TAppEncTop::xDestroyLib()
484{
485
486  m_cTVideoIOBitsFile.closeBits();
487
488#if HHI_VSO
489  for ( Int iViewIdx = 0; iViewIdx < m_iNumberOfExternalRefs; iViewIdx++ )
490  {
491    m_acTVideoIOYuvERFileList[iViewIdx]->close();
492    delete m_acTVideoIOYuvERFileList[iViewIdx];
493    m_acTVideoIOYuvERFileList[iViewIdx] = 0;
494  };
495#endif
496
497  for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
498  {
499    m_acTVideoIOYuvInputFileList[iViewIdx]->close();
500    m_acTVideoIOYuvReconFileList[iViewIdx]->close();
501
502    delete m_acTVideoIOYuvInputFileList[iViewIdx] ; m_acTVideoIOYuvInputFileList[iViewIdx] = NULL;
503    delete m_acTVideoIOYuvReconFileList[iViewIdx] ; m_acTVideoIOYuvReconFileList[iViewIdx] = NULL;
504
505    delete m_cListPicYuvRecList[iViewIdx] ; m_cListPicYuvRecList[iViewIdx] = NULL;
506
507    m_acTEncTopList[iViewIdx]->destroy();
508    delete m_acTEncTopList[iViewIdx] ; m_acTEncTopList[iViewIdx] = NULL;
509
510    if( iViewIdx < Int( m_acTVideoIOYuvDepthInputFileList.size() ) && m_acTVideoIOYuvDepthInputFileList[iViewIdx] )
511    {
512      m_acTVideoIOYuvDepthInputFileList[iViewIdx]->close( );
513      delete m_acTVideoIOYuvDepthInputFileList[iViewIdx] ;
514    }
515    if( iViewIdx < Int( m_acTVideoIOYuvDepthReconFileList.size() ) && m_acTVideoIOYuvDepthReconFileList[iViewIdx] )
516    {
517      m_acTVideoIOYuvDepthReconFileList[iViewIdx]->close () ;
518      delete m_acTVideoIOYuvDepthReconFileList[iViewIdx];
519    }
520    if( iViewIdx < Int( m_acTEncDepthTopList.size() ) && m_acTEncDepthTopList[iViewIdx] )
521    {
522      m_acTEncDepthTopList[iViewIdx]->destroy();
523      delete m_acTEncDepthTopList[iViewIdx];
524    }
525    if( iViewIdx < Int( m_cListPicYuvDepthRecList.size() ) && m_cListPicYuvDepthRecList[iViewIdx] )
526    {
527      delete m_cListPicYuvDepthRecList[iViewIdx ];
528    }
529  }
530}
531
532Void TAppEncTop::xInitLib()
533{
534  for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
535  {
536    m_acTEncTopList[iViewIdx]->init( this );
537  }
538  for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
539  {
540    m_acTEncTopList[iViewIdx]->setTEncTopList( &m_acTEncTopList  );
541  }
542  if ( m_bUsingDepthMaps )
543  {
544    for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
545    {
546      m_acTEncDepthTopList[iViewIdx]->init( this );
547    }
548    for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
549    {
550      m_acTEncDepthTopList[iViewIdx]->setTEncTopList( &m_acTEncDepthTopList  );
551    }
552  }
553}
554
555// ====================================================================================================================
556// Public member functions
557// ====================================================================================================================
558
559/**
560 - create internal class
561 - initialize internal variable
562 - until the end of input YUV file, call encoding function in TEncTop class
563 - delete allocated buffers
564 - destroy internal class
565 .
566 */
567//GT PRE LOAD ENC BUFFER
568Void TAppEncTop::encode()
569{
570  TComPicYuv*       pcPicYuvOrg      = new TComPicYuv;
571  TComPicYuv*       pcDepthPicYuvOrg = new TComPicYuv;
572  TComPicYuv*       pcPdmDepthOrg    = new TComPicYuv;
573  TComPicYuv*       pcPicYuvRec      = NULL;
574  //TComPicYuv*       pcDepthPicYuvRec = NULL;
575
576  // initialize internal class & member variables
577  xInitLibCfg();
578  xCreateLib();
579  xInitLib();
580
581  // main encoder loop
582
583  //GT: setup and init Bools for Eos and Continue Reading
584  Bool  bAllEos = false;
585  Bool  bAllContinueReadingPics = false;
586  std::vector<Bool>  bEos ;
587  std::vector<Bool>  bContinueReadingPics ;
588
589  Bool  bAllDepthEos = false;
590  Bool  bAllContinueReadingDepthPics = false;
591  std::vector<Bool>  bDepthEos ;
592  std::vector<Bool>  bContinueReadingDepthPics ;
593
594  for(Int iViewIdx=0; iViewIdx < m_iNumberOfViews; iViewIdx++ )
595  {
596    bEos.push_back( false ) ;
597    bContinueReadingPics.push_back( true );
598    if( m_bUsingDepthMaps)
599    {
600      bDepthEos.push_back( false ) ;
601      bContinueReadingDepthPics.push_back( true ) ;
602    }
603  }
604  // allocate original YUV buffer
605  pcPicYuvOrg->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
606  if( m_bUsingDepthMaps)
607  {
608    pcDepthPicYuvOrg->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
609  }
610#if HHI_INTER_VIEW_MOTION_PRED
611  if( m_uiMultiviewMvRegMode )
612  {
613    pcPdmDepthOrg->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
614  }
615#endif
616
617  TComBitstream*    pcBitstream = new TComBitstream;
618  pcBitstream->create( (m_iSourceWidth * m_iSourceHeight * 3) >> 1 ) ; //GT: is size reasonable ??
619
620  while ( !(bAllEos&& bAllContinueReadingPics) )
621  {
622    bAllContinueReadingPics = false;
623    bAllContinueReadingDepthPics = false;
624
625    //GT: Read all Buffers
626    for(Int iViewIdx=0; iViewIdx < m_iNumberOfViews; iViewIdx++ ) //GT; store frames first
627    {
628      if (!bEos[iViewIdx] && bContinueReadingPics[iViewIdx] ) //GT: read frames to buffer
629      {
630        // get buffers
631        xGetBuffer( pcPicYuvRec, iViewIdx, m_cListPicYuvRecList ); // ringbuffer of size gopsize -> m_cListPicYuvRec, m_cListBitstream
632        // read input YUV file
633        m_acTVideoIOYuvInputFileList[iViewIdx]->read( pcPicYuvOrg, m_aiPad  ) ;
634        bEos[iViewIdx] = ( m_acTVideoIOYuvInputFileList[iViewIdx]->isEof() == 1 ?   true : false  );             //GT: End of File
635        bEos[iViewIdx] = ( m_iFrameRcvdVector[iViewIdx] == (m_iFrameToBeEncoded - 1) ?    true : bEos[iViewIdx]   );   //GT: FramesToBeEncoded Reached
636        bAllEos = bAllEos|bEos[iViewIdx] ;
637
638#if HHI_INTER_VIEW_MOTION_PRED
639        if( m_uiMultiviewMvRegMode && iViewIdx )
640        {
641          m_acTVideoIOYuvDepthInputFileList[iViewIdx]->read( pcPdmDepthOrg, m_aiPad, m_bUsingDepthMaps );
642        }
643#endif
644
645        // increase number of received frames
646        m_iFrameRcvdVector[iViewIdx]++ ;
647      }
648
649#if HHI_INTER_VIEW_MOTION_PRED
650      m_acTEncTopList[iViewIdx]->receivePic( bEos[iViewIdx],  pcPicYuvOrg, m_cListPicYuvRecList[iViewIdx]->back(), ( m_uiMultiviewMvRegMode && iViewIdx ? pcPdmDepthOrg : 0 ) );
651#else
652      m_acTEncTopList[iViewIdx]->receivePic( bEos[iViewIdx],  pcPicYuvOrg, m_cListPicYuvRecList[iViewIdx]->back(), 0 );
653#endif
654
655      if( m_bUsingDepthMaps )
656      {
657        if (!bDepthEos[iViewIdx] && bContinueReadingDepthPics[iViewIdx] )
658        {
659          // get buffers
660          xGetBuffer( pcPicYuvRec, iViewIdx, m_cListPicYuvDepthRecList ); // ringbuffer of size gopsize -> m_cListPicYuvRec, m_cListBitstream
661          // read input YUV file
662          m_acTVideoIOYuvDepthInputFileList[iViewIdx]->read( pcDepthPicYuvOrg, m_aiPad  ) ;
663          bDepthEos[iViewIdx] = ( m_acTVideoIOYuvDepthInputFileList[iViewIdx]->isEof() == 1 ?   true : false  );
664          bDepthEos[iViewIdx] = ( m_iDepthFrameRcvdVector[iViewIdx] == (m_iFrameToBeEncoded - 1) ?    true : bDepthEos[iViewIdx]   );
665          bAllDepthEos = bAllDepthEos|bDepthEos[iViewIdx] ;
666          // increase number of received frames
667          m_iDepthFrameRcvdVector[iViewIdx]++ ;
668        }
669        m_acTEncDepthTopList[iViewIdx]->receivePic( bDepthEos[iViewIdx],  pcDepthPicYuvOrg, m_cListPicYuvDepthRecList[iViewIdx]->back() );
670      }
671    }
672
673    //===== store current POC =====
674    Bool  bCurrPocCoded = m_acTEncTopList[ 0 ]->currentPocWillBeCoded();
675    Int   iCurrPoc      = m_acTEncTopList[ 0 ]->getNextFrameId();
676
677    //===== update camera parameters =====
678    if( bCurrPocCoded )
679    {
680      m_cCameraData.update( (UInt)iCurrPoc );
681    }
682
683#if HHI_VSO   
684    if ( m_bUseVSO && ( m_uiVSOMode != 4) )
685    {
686      //GT: Read external reference pics or render references
687      xStoreVSORefPicsInBuffer();       //GT;
688    }
689#endif
690
691    //GT: Encode
692    for(Int iViewIdx=0; iViewIdx < m_iNumberOfViews; iViewIdx++ )     // Start encoding
693    {
694      bool bThisViewContinueReadingPics = bContinueReadingPics[iViewIdx];
695      m_acTEncTopList[iViewIdx]->encode( bEos[iViewIdx], m_cListPicYuvRecMap[iViewIdx], pcBitstream, bThisViewContinueReadingPics );
696      bContinueReadingPics[iViewIdx]=bThisViewContinueReadingPics;
697      bAllContinueReadingPics = bAllContinueReadingPics||bContinueReadingPics[iViewIdx];
698
699      if(pcBitstream->getNumberOfWrittenBits()!=0)
700      {
701        m_cTVideoIOBitsFile.writeBits( pcBitstream );
702      }
703      pcBitstream->resetBits(); //GT: also done later in ....
704      pcBitstream->rewindStreamPacket( );
705      // write bistream to file if necessary
706      xWriteOutput( iViewIdx ); //GT: Write Reconfiles (when gop is complete?)
707
708      if( m_bUsingDepthMaps )
709      {
710        bool bThisViewContinueReadingDepthPics = bContinueReadingDepthPics[iViewIdx];
711        m_acTEncDepthTopList[iViewIdx]->encode( bDepthEos[iViewIdx], m_cListPicYuvDepthRecMap[iViewIdx], pcBitstream, bThisViewContinueReadingDepthPics );
712        bContinueReadingDepthPics[iViewIdx]=bThisViewContinueReadingDepthPics;
713
714        bAllContinueReadingDepthPics = bAllContinueReadingDepthPics||bContinueReadingDepthPics[iViewIdx];
715        if(pcBitstream->getNumberOfWrittenBits()!=0)
716        {
717          m_cTVideoIOBitsFile.writeBits( pcBitstream );
718        }
719        pcBitstream->resetBits();
720        pcBitstream->rewindStreamPacket( );
721        // write bistream to file if necessary
722        xWriteOutput( iViewIdx, true );
723      }
724    }
725
726    // delete extra picture buffers
727    if( bCurrPocCoded )
728    {
729      for( Int iViewIdx = 0; iViewIdx < m_iNumberOfViews; iViewIdx++ )
730      {
731        if( iViewIdx < (Int)m_acTEncTopList.size() && m_acTEncTopList[iViewIdx] )
732        {
733          m_acTEncTopList[iViewIdx]->deleteExtraPicBuffers( iCurrPoc );
734        }
735        if( iViewIdx < (Int)m_acTEncDepthTopList.size() && m_acTEncDepthTopList[iViewIdx] )
736        {
737          m_acTEncDepthTopList[iViewIdx]->deleteExtraPicBuffers( iCurrPoc );
738        }
739      }
740    }
741
742#if AMVP_BUFFERCOMPRESS
743    // compress motion for entire access
744    if( bCurrPocCoded )
745    {
746      for( Int iViewIdx = 0; iViewIdx < m_iNumberOfViews; iViewIdx++ )
747      {
748        if( iViewIdx < (Int)m_acTEncTopList.size() && m_acTEncTopList[iViewIdx] )
749        {
750          m_acTEncTopList[iViewIdx]->compressMotion( iCurrPoc );
751        }
752        if( iViewIdx < (Int)m_acTEncDepthTopList.size() && m_acTEncDepthTopList[iViewIdx] )
753        {
754          m_acTEncDepthTopList[iViewIdx]->compressMotion( iCurrPoc );
755        }
756      }
757    }
758#endif
759  }
760
761  // write output
762  for(Int iViewIdx=0; iViewIdx < m_iNumberOfViews; iViewIdx++ )
763  {
764    m_acTEncTopList[iViewIdx]->printOutSummary(m_acTEncTopList[iViewIdx]->getNumAllPicCoded());
765    if ( m_bUsingDepthMaps )
766    {
767      m_acTEncDepthTopList[iViewIdx]->printOutSummary(m_acTEncDepthTopList[iViewIdx]->getNumAllPicCoded());
768    }
769  }
770
771  // delete original YUV buffer
772  pcPicYuvOrg->destroy();
773  delete pcPicYuvOrg;
774  pcPicYuvOrg = NULL;
775
776  // valgrind
777  if( m_bUsingDepthMaps)
778  {
779    pcDepthPicYuvOrg->destroy();
780  }
781  delete pcDepthPicYuvOrg ;
782  pcDepthPicYuvOrg = NULL ;
783
784  pcBitstream->destroy();
785  delete pcBitstream;
786  pcBitstream = NULL ;
787
788
789  // delete used buffers in encoder class
790  for(Int iViewIdx =0; iViewIdx < m_iNumberOfViews; iViewIdx++)
791  {
792    m_acTEncTopList[iViewIdx]->deletePicBuffer() ;
793  }
794
795  if( m_bUsingDepthMaps)
796  {
797    for(Int iViewIdx =0; iViewIdx < m_iNumberOfViews; iViewIdx++)
798    {
799      m_acTEncDepthTopList[iViewIdx]->deletePicBuffer() ;
800    }
801  }
802
803  if ( pcPdmDepthOrg )
804  {
805    pcPdmDepthOrg->destroy();
806    delete pcPdmDepthOrg;
807    pcPdmDepthOrg = NULL;
808  }
809
810  // delete buffers & classes
811  xDeleteBuffer();
812  xDestroyLib();
813}
814
815// ====================================================================================================================
816// Protected member functions
817// ====================================================================================================================
818
819/**
820 - application has picture buffer list with size of GOP
821 - picture buffer list acts as ring buffer
822 - end of the list has the latest picture
823 .
824 */
825Void TAppEncTop::xGetBuffer( TComPicYuv*& rpcPicYuvRec, Int iViewIdx, std::vector< TComList<TComPicYuv*>*>& racBuffer )
826{
827  if ( m_uiCodedPictureStoreSize   == 0 )
828    {
829      if (racBuffer[iViewIdx]->size() == 0)
830      {
831        rpcPicYuvRec = new TComPicYuv;
832        rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
833
834        racBuffer[iViewIdx]->pushBack( rpcPicYuvRec );
835      }
836      rpcPicYuvRec = racBuffer[iViewIdx]->popFront(); //GT why? only one in list ?? A: only to get rpcPicYuvRec
837      racBuffer[iViewIdx]->pushBack( rpcPicYuvRec );
838      return;
839    }
840
841    // org. buffer
842  if ( racBuffer[iViewIdx]->size() == (UInt)m_uiCodedPictureStoreSize )
843    {
844      rpcPicYuvRec = racBuffer[iViewIdx]->popFront();
845    }
846    else
847    {
848      rpcPicYuvRec = new TComPicYuv;
849      rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
850    }
851    racBuffer[iViewIdx]->pushBack( rpcPicYuvRec );
852}
853
854Void TAppEncTop::xDeleteBuffer( )
855{
856  TComList<TComBitstream*>::iterator iterBitstream = m_cListBitstream.begin();
857
858  Int iSize = Int( m_cListBitstream.size() );
859
860  for ( Int i = 0; i < iSize; i++ )
861  {
862    TComBitstream* pcBitstream = *(iterBitstream++);
863
864    pcBitstream->destroy();
865
866    delete pcBitstream; pcBitstream = NULL;
867  }
868
869  for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
870  {
871    TComList<TComPicYuv*>::iterator iterPicYuvRec  = m_cListPicYuvRecList[iViewIdx]->begin();
872
873    iSize = Int( m_cListPicYuvRecList[iViewIdx]->size() );
874
875    for ( Int i = 0; i < iSize; i++ )
876    {
877        TComPicYuv*  pcPicYuvRec  = *(iterPicYuvRec++);
878        pcPicYuvRec->destroy();
879        delete pcPicYuvRec; pcPicYuvRec = NULL;
880    }
881  }
882  if( m_bUsingDepthMaps)
883  {
884    for(Int iViewIdx=0; iViewIdx<m_iNumberOfViews; iViewIdx++)
885    {
886      TComList<TComPicYuv*>::iterator iterPicYuvRec  = m_cListPicYuvDepthRecList[iViewIdx]->begin();
887
888      iSize = Int( m_cListPicYuvDepthRecList[iViewIdx]->size() );
889
890      for ( Int i = 0; i < iSize; i++ )
891      {
892          TComPicYuv*  pcPicYuvRec  = *(iterPicYuvRec++);
893          pcPicYuvRec->destroy();
894          delete pcPicYuvRec; pcPicYuvRec = NULL;
895      }
896    }
897  }
898
899  // Delete ERFiles
900
901#if HHI_VSO
902  std::map< Int,vector<TComPicYuv*> >::iterator iterMapPicExtRefView = m_cMapPicExtRefView.begin();
903  while ( iterMapPicExtRefView != m_cMapPicExtRefView.end() )
904  {
905    for ( UInt uiViewNumber = 0; uiViewNumber < iterMapPicExtRefView->second.size(); uiViewNumber++ )
906    {
907      if ( iterMapPicExtRefView->second[uiViewNumber] )
908      {
909        iterMapPicExtRefView->second[uiViewNumber]->destroy();
910        delete iterMapPicExtRefView->second[uiViewNumber];
911      }
912    }
913    iterMapPicExtRefView++;
914  }
915#endif
916}
917
918/** \param iNumEncoded  number of encoded frames
919 */
920Void TAppEncTop::xWriteOutput( Int iViewIdx, Bool isDepth )
921{
922  std::map<PicOrderCnt, TComPicYuv*> &rcMap = ( isDepth ? m_cListPicYuvDepthRecMap          : m_cListPicYuvRecMap          )[iViewIdx];
923  PicOrderCnt  &riNextPocToDump             = ( isDepth ? m_aiNextDepthPocToDump            : m_aiNextPocToDump            )[iViewIdx];
924  TVideoIOYuv* &rpcTVideoIOYuvReconFile     = ( isDepth ? m_acTVideoIOYuvDepthReconFileList : m_acTVideoIOYuvReconFileList )[iViewIdx];
925  std::map<PicOrderCnt, TComPicYuv*>::iterator i;
926
927  while( ! rcMap.empty() && ( i = rcMap.begin() )->first == riNextPocToDump )
928  {
929    riNextPocToDump++;
930    rpcTVideoIOYuvReconFile->write( i->second, m_aiPad );
931    rcMap.erase( i );
932  }
933}
934
935// GT FIX
936std::vector<TComPic*> TAppEncTop::getSpatialRefPics( Int iViewIdx, Int iPoc, Bool bIsDepthCoder )
937{
938  std::vector<TComPic*> apcRefPics( iViewIdx, (TComPic*)NULL );
939  for( int iRefViewIdx = 0; iRefViewIdx < iViewIdx; iRefViewIdx++ )
940  {
941// GT FIX
942    TComPic* pcRefPic = xGetPicFromView(iRefViewIdx, iPoc, bIsDepthCoder);
943
944    assert( pcRefPic != NULL );
945    apcRefPics[iRefViewIdx] = pcRefPic;
946  }
947  return apcRefPics;
948}
949
950
951TComPic* TAppEncTop::xGetPicFromView( Int iViewIdx, Int iPoc, bool bDepth )
952{
953  assert( ( iViewIdx >= 0) && ( iViewIdx < m_iNumberOfViews ) );
954
955  TComPic* pcPic = 0;
956  TComList<TComPic*>* apcListPic;
957
958  apcListPic = (bDepth ?  m_acTEncDepthTopList[iViewIdx] : m_acTEncTopList[iViewIdx])->getListPic() ;
959
960  for(TComList<TComPic*>::iterator it=apcListPic->begin(); it!=apcListPic->end(); it++)
961  {
962    if( (*it)->getPOC() == iPoc )
963    {
964      pcPic = *it ;
965      break ;
966    }
967  }
968
969  return pcPic;
970};
971
972TComPicYuv* TAppEncTop::xGetPicYuvFromView( Int iViewIdx, Int iPoc, Bool bDepth, Bool bRecon )
973{
974  TComPic*    pcPic = xGetPicFromView( iViewIdx, iPoc, bDepth);
975  TComPicYuv* pcPicYuv = NULL;
976
977  if (pcPic != NULL)
978  {
979    if( bRecon )
980    {
981      if ( pcPic->getReconMark() )
982      {
983        pcPicYuv = pcPic->getPicYuvRec();
984      }
985    }
986    else
987    {
988      pcPicYuv = pcPic->getPicYuvOrg();
989    }
990  };
991
992  return pcPicYuv;
993};
994
995#if HHI_VSO
996Void TAppEncTop::xSetBasePicYuv( Int iViewIdx, Int iPoc, TComMVDRefData* pcRefInfo, InterViewReference eView, bool bDepth )
997{
998
999  if ( ( iViewIdx < 0) || ( iViewIdx >= m_iNumberOfViews ) )
1000    return;
1001
1002  if ( !m_bUsingDepthMaps && bDepth )
1003    return;
1004
1005  TComPic* pcPic = xGetPicFromView( iViewIdx, iPoc, bDepth);
1006
1007  if (pcPic == 0)
1008    return;
1009
1010  pcRefInfo->setPicYuvBaseView(eView, bDepth, pcPic->getPicYuvOrg(),  pcPic->getReconMark() ? pcPic->getPicYuvRec() : NULL );
1011
1012};
1013
1014Void TAppEncTop::setMVDPic( Int iViewIdx, Int iPoc, TComMVDRefData* pcMVDRefData )
1015{
1016  AOF( iViewIdx >= 0);
1017  AOF( iViewIdx <  m_iNumberOfViews  );
1018
1019  xSetBasePicYuv(iViewIdx - 1, iPoc, pcMVDRefData, PREVVIEW, false);
1020  xSetBasePicYuv(iViewIdx    , iPoc, pcMVDRefData, CURRVIEW, false );
1021  xSetBasePicYuv(iViewIdx + 1, iPoc, pcMVDRefData, NEXTVIEW, false );
1022
1023  if ( m_bUsingDepthMaps )
1024  {
1025    xSetBasePicYuv(iViewIdx - 1, iPoc, pcMVDRefData, PREVVIEW, true );
1026    xSetBasePicYuv(iViewIdx    , iPoc, pcMVDRefData, CURRVIEW, true );
1027    xSetBasePicYuv(iViewIdx + 1, iPoc, pcMVDRefData, NEXTVIEW, true );
1028  }
1029
1030
1031  xSetBaseLUT       (iViewIdx, iViewIdx-1  , pcMVDRefData, PREVVIEW );
1032  xSetBaseLUT       (iViewIdx, iViewIdx+1  , pcMVDRefData, NEXTVIEW );
1033
1034
1035  if ( m_bUseVSO && m_uiVSOMode != 4)
1036  {
1037    xSetERPicYuvs               (iViewIdx, iPoc, pcMVDRefData);
1038    pcMVDRefData->setShiftLUTsERView(m_cCameraData.getSynthViewShiftLUTD()[iViewIdx],  m_cCameraData.getSynthViewShiftLUTI()[iViewIdx] );
1039    pcMVDRefData->setRefViewInd     (m_aaiBaseViewRefInd[iViewIdx], m_aaiERViewRefInd[iViewIdx], m_aaiERViewRefLutInd[iViewIdx]);
1040  }
1041};
1042
1043
1044Void TAppEncTop::xSetBaseLUT( Int iViewIdxSource, Int iViewIdxTarget, TComMVDRefData* pcRefInfo, InterViewReference eView )
1045{
1046  if ( ( iViewIdxSource < 0) || ( iViewIdxSource >= m_iNumberOfViews )||( iViewIdxTarget < 0) || ( iViewIdxTarget >= m_iNumberOfViews ) )
1047    return;
1048  assert( abs( iViewIdxTarget - iViewIdxSource ) <= 1 );  //GT; Not supported yet
1049  pcRefInfo->setShiftLUTsBaseView(eView, m_cCameraData.getBaseViewShiftLUTD()[iViewIdxSource][iViewIdxTarget],m_cCameraData.getBaseViewShiftLUTI()[iViewIdxSource][iViewIdxTarget] );
1050};
1051
1052Void TAppEncTop::xSetERPicYuvs( Int iViewIdx, Int iPoc, TComMVDRefData* pcReferenceInfo )
1053{
1054  std::vector<TComPicYuv*> apcExtRefViews;
1055
1056  std::map< Int, vector<TComPicYuv*> >::iterator cMapIt;
1057  cMapIt = m_cMapPicExtRefView.find(iPoc);
1058
1059  assert(  cMapIt != m_cMapPicExtRefView.end() );
1060
1061  pcReferenceInfo->setPicYuvERViews( cMapIt->second );
1062}
1063
1064Void TAppEncTop::xStoreVSORefPicsInBuffer()
1065{
1066  // X-Check if all Encoders have received the same number of pics
1067  Int iNumRcvd = m_iFrameRcvdVector[0];
1068  for ( UInt uiViewNumber = 0; uiViewNumber < m_iNumberOfViews; uiViewNumber ++ )
1069  {
1070    assert( ( m_iFrameRcvdVector[uiViewNumber] == iNumRcvd ) && ( m_iDepthFrameRcvdVector[uiViewNumber] == iNumRcvd ) ); //GT; if assert here, the encoder instances are not synchronized any more, re-think this function and the ERView Buffer!!
1071  };
1072
1073  Int iCurPoc = iNumRcvd - 1;
1074
1075  if ( iCurPoc <= m_iLastFramePutInERViewBuffer )
1076  {
1077    return;
1078  }
1079
1080  std::vector<TComPicYuv*> apcExtRefViewVec;
1081
1082  Int iNumberOfReferenceViews = 0;
1083  if (m_iNumberOfExternalRefs != 0)
1084  {
1085    m_aaiERViewRefLutInd = m_aaiERViewRefInd;
1086    // Insert Rendered Views form File
1087
1088    iNumberOfReferenceViews = m_iNumberOfExternalRefs;
1089
1090    for ( UInt uiViewNumber = 0; uiViewNumber < iNumberOfReferenceViews; uiViewNumber++ )
1091    {
1092      TComPicYuv* pcPicYuvERView = new TComPicYuv;
1093      pcPicYuvERView->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
1094
1095      m_acTVideoIOYuvERFileList[uiViewNumber]->read( pcPicYuvERView, m_aiPad  ) ;
1096
1097      apcExtRefViewVec.push_back( pcPicYuvERView );
1098    }
1099  }
1100  else
1101  { //Render Views
1102
1103    for ( UInt uiViewNumber = 0; uiViewNumber < m_iNumberOfViews; uiViewNumber++ )
1104    {
1105      m_aaiERViewRefInd   [uiViewNumber].clear();
1106      m_aaiERViewRefLutInd[uiViewNumber].clear();
1107    }
1108
1109    iNumberOfReferenceViews = 0;
1110    for ( UInt iSynthViewIdx = 0; iSynthViewIdx < m_cCameraData.getSynthViewNumbers().size(); iSynthViewIdx++ )
1111    {
1112      // Get Left and right view
1113      Int  iLeftViewIdx  = -1;
1114      Int  iRightViewIdx = -1;
1115      Bool bIsBaseView;
1116
1117      Int iRelDistToLeft;
1118      m_cCameraData.getLeftRightBaseView( iSynthViewIdx, iLeftViewIdx, iRightViewIdx, iRelDistToLeft, bIsBaseView );
1119
1120      if  ((iLeftViewIdx == -1) || (iRightViewIdx == -1))
1121      {
1122        std::cerr << "Left or right View not given." << endl;
1123        exit(EXIT_FAILURE);
1124      }
1125
1126      m_cRendererTop.setShiftLUTs(
1127        m_cCameraData.getSynthViewShiftLUTD()[iLeftViewIdx] [iSynthViewIdx],
1128        m_cCameraData.getSynthViewShiftLUTI()[iLeftViewIdx] [iSynthViewIdx],
1129        m_cCameraData.getBaseViewShiftLUTI ()[iLeftViewIdx] [iRightViewIdx],
1130        m_cCameraData.getSynthViewShiftLUTD()[iRightViewIdx][iSynthViewIdx],
1131        m_cCameraData.getSynthViewShiftLUTI()[iRightViewIdx][iSynthViewIdx],
1132        m_cCameraData.getBaseViewShiftLUTI ()[iRightViewIdx][iLeftViewIdx],
1133        iRelDistToLeft
1134      );
1135      if ( bIsBaseView ) continue;
1136
1137      // Render from left
1138      TComPicYuv* pcPicYuvERView = new TComPicYuv;
1139      pcPicYuvERView->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
1140      m_cRendererTop.extrapolateView( xGetPicFromView( iLeftViewIdx, iCurPoc, false )->getPicYuvOrg(), xGetPicFromView( iLeftViewIdx, iCurPoc, true )->getPicYuvOrg(), pcPicYuvERView, true );
1141
1142      apcExtRefViewVec.push_back( pcPicYuvERView );
1143
1144      m_aaiERViewRefInd   [ iLeftViewIdx].push_back( iNumberOfReferenceViews );
1145      m_aaiERViewRefLutInd[ iLeftViewIdx].push_back( iSynthViewIdx );
1146      iNumberOfReferenceViews++;
1147
1148      //Render from right
1149      pcPicYuvERView = new TComPicYuv;
1150      pcPicYuvERView->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
1151
1152      m_cRendererTop.extrapolateView( xGetPicFromView( iRightViewIdx, iCurPoc, false )->getPicYuvOrg(), xGetPicFromView( iRightViewIdx, iCurPoc, true )->getPicYuvOrg(), pcPicYuvERView, false );
1153      apcExtRefViewVec.push_back( pcPicYuvERView );
1154
1155      m_aaiERViewRefInd   [ iRightViewIdx].push_back( iNumberOfReferenceViews );
1156      m_aaiERViewRefLutInd[ iRightViewIdx].push_back( iSynthViewIdx );
1157      iNumberOfReferenceViews++;
1158    }
1159  }
1160
1161  m_iLastFramePutInERViewBuffer++;
1162  m_cMapPicExtRefView.insert( std::make_pair( m_iLastFramePutInERViewBuffer, apcExtRefViewVec ) );
1163
1164  if ( m_cMapPicExtRefView.size() >  (UInt)m_iGOPSize + 1 )
1165  {
1166    for ( UInt uiViewNumber = 0; uiViewNumber < (UInt) m_cMapPicExtRefView.begin()->second.size(); uiViewNumber++ )
1167    {
1168      if ( m_cMapPicExtRefView.begin()->second[uiViewNumber] )
1169      {
1170        m_cMapPicExtRefView.begin()->second[uiViewNumber]->destroy();
1171        delete m_cMapPicExtRefView.begin()->second[uiViewNumber];
1172      }
1173    }
1174    m_cMapPicExtRefView.erase ( m_cMapPicExtRefView.begin() );
1175  }
1176}
1177#endif
1178
1179#if HHI_INTERVIEW_SKIP
1180Void TAppEncTop::getUsedPelsMap( Int iViewIdx, Int iPoc, TComPicYuv* pcPicYuvUsedSplsMap )
1181{
1182  AOT( iViewIdx <= 0);
1183  AOT( iViewIdx >= m_iNumberOfViews );
1184  AOF( m_uiInterViewSkip != 0 );
1185  AOF( m_cCameraData.getCurFrameId() == iPoc );
1186
1187  Int iViewSIdx      = m_cCameraData.getBaseId2SortedId()[iViewIdx];
1188  Int iFirstViewSIdx = m_cCameraData.getBaseId2SortedId()[0];
1189
1190  AOT( iViewSIdx == iFirstViewSIdx );
1191
1192  Bool bFirstIsLeft = (iFirstViewSIdx < iViewSIdx);
1193
1194    m_cUsedPelsRenderer.setShiftLUTs(
1195      m_cCameraData.getBaseViewShiftLUTD()[0][iViewIdx],
1196      m_cCameraData.getBaseViewShiftLUTI()[0][iViewIdx],
1197      m_cCameraData.getBaseViewShiftLUTI()[0][iViewIdx],
1198      m_cCameraData.getBaseViewShiftLUTD()[0][iViewIdx],
1199      m_cCameraData.getBaseViewShiftLUTI()[0][iViewIdx],
1200      m_cCameraData.getBaseViewShiftLUTI()[0][iViewIdx],
1201      -1
1202      );
1203
1204
1205  TComPicYuv* pcPicYuvDepth = xGetPicYuvFromView(0, iPoc, true, true );
1206  AOF( pcPicYuvDepth);
1207
1208  m_cUsedPelsRenderer.getUsedSamplesMap( pcPicYuvDepth, pcPicYuvUsedSplsMap, bFirstIsLeft );
1209
1210}
1211#endif
1212
1213#if HHI_VSO
1214Void TAppEncTop::setupRenModel( Int iPoc, Int iEncViewIdx, Int iEncContent )
1215{
1216  Int iEncViewSIdx = m_cCameraData.getBaseId2SortedId()[ iEncViewIdx ];
1217
1218  // setup base views
1219  Int iNumOfBV = m_cRenModStrParser.getNumOfBaseViewsForView( iEncViewSIdx, iEncContent );
1220
1221  for (Int iCurView = 0; iCurView < iNumOfBV; iCurView++ )
1222  {
1223    Int iBaseViewSIdx;
1224    Int iVideoDistMode;
1225    Int iDepthDistMode;
1226
1227    m_cRenModStrParser.getBaseViewData( iEncViewSIdx, iEncContent, iCurView, iBaseViewSIdx, iVideoDistMode, iDepthDistMode );
1228
1229    AOT( iVideoDistMode < 0 || iVideoDistMode > 2 );
1230
1231    Int iBaseViewIdx = m_cCameraData.getBaseSortedId2Id()[ iBaseViewSIdx ];
1232
1233    TComPicYuv* pcPicYuvVideoRec  = xGetPicYuvFromView( iBaseViewIdx, iPoc, false, true  );
1234    TComPicYuv* pcPicYuvDepthRec  = xGetPicYuvFromView( iBaseViewIdx, iPoc, true , true  );
1235    TComPicYuv* pcPicYuvVideoOrg  = xGetPicYuvFromView( iBaseViewIdx, iPoc, false, false );
1236    TComPicYuv* pcPicYuvDepthOrg  = xGetPicYuvFromView( iBaseViewIdx, iPoc, true , false );
1237
1238    TComPicYuv* pcPicYuvVideoRef  = ( iVideoDistMode == 2 ) ? pcPicYuvVideoOrg  : NULL;
1239    TComPicYuv* pcPicYuvDepthRef  = ( iDepthDistMode == 2 ) ? pcPicYuvDepthOrg  : NULL;
1240
1241    TComPicYuv* pcPicYuvVideoTest = ( iVideoDistMode == 0 ) ? pcPicYuvVideoOrg  : pcPicYuvVideoRec;
1242    TComPicYuv* pcPicYuvDepthTest = ( iDepthDistMode == 0 ) ? pcPicYuvDepthOrg  : pcPicYuvDepthRec;
1243
1244    AOT( (iVideoDistMode == 2) != (pcPicYuvVideoRef != NULL) );
1245    AOT( (iDepthDistMode == 2) != (pcPicYuvDepthRef != NULL) );
1246    AOT( pcPicYuvDepthTest == NULL );
1247    AOT( pcPicYuvVideoTest == NULL );
1248
1249    m_cRendererModel.setBaseView( iBaseViewSIdx, pcPicYuvVideoTest, pcPicYuvDepthTest, pcPicYuvVideoRef, pcPicYuvDepthRef );
1250  }
1251
1252  m_cRendererModel.setErrorMode( iEncViewSIdx, iEncContent, 0 );
1253  // setup virtual views
1254  Int iNumOfSV  = m_cRenModStrParser.getNumOfModelsForView( iEncViewSIdx, iEncContent );
1255  for (Int iCurView = 0; iCurView < iNumOfSV; iCurView++ )
1256  {
1257    Int iOrgRefBaseViewSIdx;
1258    Int iLeftBaseViewSIdx;
1259    Int iRightBaseViewSIdx;
1260    Int iSynthViewRelNum;
1261    Int iModelNum;
1262    Int iBlendMode;
1263    m_cRenModStrParser.getSingleModelData(iEncViewSIdx, iEncContent, iCurView, iModelNum, iBlendMode,iLeftBaseViewSIdx, iRightBaseViewSIdx, iOrgRefBaseViewSIdx, iSynthViewRelNum );
1264
1265    Int iLeftBaseViewIdx    = -1;
1266    Int iRightBaseViewIdx   = -1;
1267
1268    TComPicYuv* pcPicYuvOrgRef  = NULL;
1269    Int**      ppiShiftLUTLeft  = NULL;
1270    Int**      ppiShiftLUTRight = NULL;
1271    Int**      ppiBaseShiftLUTLeft  = NULL;
1272    Int**      ppiBaseShiftLUTRight = NULL;
1273
1274
1275    Int        iDistToLeft      = -1;
1276
1277    Int iSynthViewIdx = m_cCameraData.synthRelNum2Idx( iSynthViewRelNum );
1278
1279    if ( iLeftBaseViewSIdx != -1 )
1280    {
1281      iLeftBaseViewIdx   = m_cCameraData.getBaseSortedId2Id()   [ iLeftBaseViewSIdx ];
1282      ppiShiftLUTLeft    = m_cCameraData.getSynthViewShiftLUTI()[ iLeftBaseViewIdx  ][ iSynthViewIdx  ];
1283    }
1284
1285    if ( iRightBaseViewSIdx != -1 )
1286    {
1287      iRightBaseViewIdx  = m_cCameraData.getBaseSortedId2Id()   [iRightBaseViewSIdx ];
1288      ppiShiftLUTRight   = m_cCameraData.getSynthViewShiftLUTI()[ iRightBaseViewIdx ][ iSynthViewIdx ];
1289    }
1290
1291    if ( iRightBaseViewSIdx != -1 && iLeftBaseViewSIdx != -1 )
1292    {
1293      iDistToLeft    = m_cCameraData.getRelDistLeft(  iSynthViewIdx , iLeftBaseViewIdx, iRightBaseViewIdx);
1294      ppiBaseShiftLUTLeft  = m_cCameraData.getBaseViewShiftLUTI() [ iLeftBaseViewIdx  ][ iRightBaseViewIdx ];
1295      ppiBaseShiftLUTRight = m_cCameraData.getBaseViewShiftLUTI() [ iRightBaseViewIdx ][ iLeftBaseViewIdx  ];
1296
1297    }
1298
1299    if ( iOrgRefBaseViewSIdx != -1 )
1300    {
1301      pcPicYuvOrgRef = xGetPicYuvFromView( m_cCameraData.getBaseSortedId2Id()[ iOrgRefBaseViewSIdx ] , iPoc, false, false );
1302      AOF ( pcPicYuvOrgRef );
1303    }
1304
1305    m_cRendererModel.setSingleModel( iModelNum, ppiShiftLUTLeft, ppiBaseShiftLUTLeft, ppiShiftLUTRight, ppiBaseShiftLUTRight, iDistToLeft, pcPicYuvOrgRef );
1306  }
1307}
1308#endif
1309
Note: See TracBrowser for help on using the repository browser.