Context navigation

TEncGOP.cpp @ 494

Visit:

Last change on this file since 494 was 408, checked in by lg, 12 years ago
D0135->D0092->D0091
Property svn:eol-style set to `native`
File size: 85.3 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 TEncGOP.cpp
35	\brief GOP encoder class
36	*/
37
38	#include <list>
39	#include <algorithm>
40
41	#include "TEncTop.h"
42	#include "TEncGOP.h"
43	#include "TEncAnalyze.h"
44	#include "libmd5/MD5.h"
45	#include "TLibCommon/SEI.h"
46	#include "TLibCommon/NAL.h"
47	#include "NALwrite.h"
48	#include "../../App/TAppEncoder/TAppEncTop.h"
49
50	#include <time.h>
51	#include <math.h>
52
53	using namespace std;
54
55	//! \ingroup TLibEncoder
56	//! \{
57
58	// ====================================================================================================================
59	// Constructor / destructor / initialization / destroy
60	// ====================================================================================================================
61
62	TEncGOP::TEncGOP()
63	{
64	m_iLastIDR = 0;
65	m_iGopSize = 0;
66	m_iNumPicCoded = 0; //Niko
67	m_bFirst = true;
68
69	m_pcCfg = NULL;
70	m_pcSliceEncoder = NULL;
71	m_pcListPic = NULL;
72
73	m_pcEntropyCoder = NULL;
74	m_pcCavlcCoder = NULL;
75	m_pcSbacCoder = NULL;
76	m_pcBinCABAC = NULL;
77	#if DEPTH_MAP_GENERATION
78	m_pcDepthMapGenerator = NULL;
79	#endif
80	#if H3D_IVRP
81	m_pcResidualGenerator = NULL;
82	#endif
83
84	m_bSeqFirst = true;
85
86	m_bRefreshPending = 0;
87	m_pocCRA = 0;
88
89	return;
90	}
91
92	TEncGOP::~TEncGOP()
93	{
94	}
95
96	/** Create list to contain pointers to LCU start addresses of slice.
97	* \param iWidth, iHeight are picture width, height. iMaxCUWidth, iMaxCUHeight are LCU width, height.
98	*/
99	Void TEncGOP::create( Int iWidth, Int iHeight, UInt iMaxCUWidth, UInt iMaxCUHeight )
100	{
101	UInt uiWidthInCU = ( iWidth %iMaxCUWidth ) ? iWidth /iMaxCUWidth + 1 : iWidth /iMaxCUWidth;
102	UInt uiHeightInCU = ( iHeight%iMaxCUHeight ) ? iHeight/iMaxCUHeight + 1 : iHeight/iMaxCUHeight;
103	UInt uiNumCUsInFrame = uiWidthInCU * uiHeightInCU;
104	m_uiStoredStartCUAddrForEncodingSlice = new UInt [uiNumCUsInFrame*(1<<(g_uiMaxCUDepth<<1))+1];
105	m_uiStoredStartCUAddrForEncodingEntropySlice = new UInt [uiNumCUsInFrame*(1<<(g_uiMaxCUDepth<<1))+1];
106	m_bLongtermTestPictureHasBeenCoded = 0;
107	m_bLongtermTestPictureHasBeenCoded2 = 0;
108	}
109
110	Void TEncGOP::destroy()
111	{
112	delete [] m_uiStoredStartCUAddrForEncodingSlice; m_uiStoredStartCUAddrForEncodingSlice = NULL;
113	delete [] m_uiStoredStartCUAddrForEncodingEntropySlice; m_uiStoredStartCUAddrForEncodingEntropySlice = NULL;
114	}
115
116	Void TEncGOP::init ( TEncTop* pcTEncTop )
117	{
118	m_pcEncTop = pcTEncTop;
119	m_pcCfg = pcTEncTop;
120	m_pcSliceEncoder = pcTEncTop->getSliceEncoder();
121	m_pcListPic = pcTEncTop->getListPic();
122
123	m_pcEntropyCoder = pcTEncTop->getEntropyCoder();
124	m_pcCavlcCoder = pcTEncTop->getCavlcCoder();
125	m_pcSbacCoder = pcTEncTop->getSbacCoder();
126	m_pcBinCABAC = pcTEncTop->getBinCABAC();
127	m_pcLoopFilter = pcTEncTop->getLoopFilter();
128	m_pcBitCounter = pcTEncTop->getBitCounter();
129
130	#if DEPTH_MAP_GENERATION
131	m_pcDepthMapGenerator = pcTEncTop->getDepthMapGenerator();
132	#endif
133	#if H3D_IVRP
134	m_pcResidualGenerator = pcTEncTop->getResidualGenerator();
135	#endif
136
137	// Adaptive Loop filter
138	m_pcAdaptiveLoopFilter = pcTEncTop->getAdaptiveLoopFilter();
139	//--Adaptive Loop filter
140	m_pcSAO = pcTEncTop->getSAO();
141	m_pcRdCost = pcTEncTop->getRdCost();
142	}
143
144	// ====================================================================================================================
145	// Public member functions
146	// ====================================================================================================================
147
148	Void TEncGOP::initGOP( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic>& rcListPic, TComList<TComPicYuv>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsInGOP)
149	{
150	xInitGOP( iPOCLast, iNumPicRcvd, rcListPic, rcListPicYuvRecOut );
151	m_iNumPicCoded = 0;
152	}
153
154	Void TEncGOP::compressPicInGOP( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic>& rcListPic, TComList<TComPicYuv>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsInGOP, Int iGOPid)
155	{
156	TComPic* pcPic;
157	TComPicYuv* pcPicYuvRecOut;
158	TComSlice* pcSlice;
159	TComOutputBitstream *pcBitstreamRedirect;
160	pcBitstreamRedirect = new TComOutputBitstream;
161	AccessUnit::iterator itLocationToPushSliceHeaderNALU; // used to store location where NALU containing slice header is to be inserted
162	UInt uiOneBitstreamPerSliceLength = 0;
163	TEncSbac* pcSbacCoders = NULL;
164	TComOutputBitstream* pcSubstreamsOut = NULL;
165
166	{
167	UInt uiColDir = 1;
168	//-- For time output for each slice
169	long iBeforeTime = clock();
170
171	//select uiColDir
172	Int iCloseLeft=1, iCloseRight=-1;
173	for(Int i = 0; i<m_pcCfg->getGOPEntry(iGOPid).m_numRefPics; i++)
174	{
175	Int iRef = m_pcCfg->getGOPEntry(iGOPid).m_referencePics[i];
176	if(iRef>0&&(iRef<iCloseRight\|\|iCloseRight==-1))
177	{
178	iCloseRight=iRef;
179	}
180	else if(iRef<0&&(iRef>iCloseLeft\|\|iCloseLeft==1))
181	{
182	iCloseLeft=iRef;
183	}
184	}
185	if(iCloseRight>-1)
186	{
187	iCloseRight=iCloseRight+m_pcCfg->getGOPEntry(iGOPid).m_POC-1;
188	}
189	if(iCloseLeft<1)
190	{
191	iCloseLeft=iCloseLeft+m_pcCfg->getGOPEntry(iGOPid).m_POC-1;
192	while(iCloseLeft<0)
193	{
194	iCloseLeft+=m_iGopSize;
195	}
196	}
197	Int iLeftQP=0, iRightQP=0;
198	for(Int i=0; i<m_iGopSize; i++)
199	{
200	if(m_pcCfg->getGOPEntry(i).m_POC==(iCloseLeft%m_iGopSize)+1)
201	{
202	iLeftQP= m_pcCfg->getGOPEntry(i).m_QPOffset;
203	}
204	if (m_pcCfg->getGOPEntry(i).m_POC==(iCloseRight%m_iGopSize)+1)
205	{
206	iRightQP=m_pcCfg->getGOPEntry(i).m_QPOffset;
207	}
208	}
209	if(iCloseRight>-1&&iRightQP<iLeftQP)
210	{
211	uiColDir=0;
212	}
213
214	/////////////////////////////////////////////////////////////////////////////////////////////////// Initial to start encoding
215	UInt uiPOCCurr = iPOCLast -iNumPicRcvd+ m_pcCfg->getGOPEntry(iGOPid).m_POC;
216	Int iTimeOffset = m_pcCfg->getGOPEntry(iGOPid).m_POC;
217	if(iPOCLast == 0)
218	{
219	uiPOCCurr=0;
220	iTimeOffset = 1;
221	}
222	if(uiPOCCurr>=m_pcCfg->getFrameToBeEncoded())
223	{
224	return;
225	}
226	if( getNalUnitTypeBaseViewMvc( uiPOCCurr ) == NAL_UNIT_CODED_SLICE_IDR )
227	{
228	m_iLastIDR = uiPOCCurr;
229	}
230
231	/* start a new access unit: create an entry in the list of output
232	* access units */
233	accessUnitsInGOP.push_back(AccessUnit());
234	AccessUnit& accessUnit = accessUnitsInGOP.back();
235	xGetBuffer( rcListPic, rcListPicYuvRecOut, iNumPicRcvd, iTimeOffset, pcPic, pcPicYuvRecOut, uiPOCCurr );
236
237	// Slice data initialization
238	pcPic->clearSliceBuffer();
239	assert(pcPic->getNumAllocatedSlice() == 1);
240	m_pcSliceEncoder->setSliceIdx(0);
241	pcPic->setCurrSliceIdx(0);
242
243	std::vector<TComAPS>& vAPS = m_pcEncTop->getAPS();
244	#if VIDYO_VPS_INTEGRATION\|QC_MVHEVC_B0046
245	#if MTK_DEPTH_MERGE_TEXTURE_CANDIDATE_C0137
246	m_pcSliceEncoder->initEncSlice ( pcPic, iPOCLast, uiPOCCurr, iNumPicRcvd, iGOPid, pcSlice, m_pcEncTop->getEncTop()->getVPS(), m_pcEncTop->getSPS(), m_pcEncTop->getPPS(), m_pcEncTop->getIsDepth() );
247	#else
248	m_pcSliceEncoder->initEncSlice ( pcPic, iPOCLast, uiPOCCurr, iNumPicRcvd, iGOPid, pcSlice, m_pcEncTop->getEncTop()->getVPS(), m_pcEncTop->getSPS(), m_pcEncTop->getPPS() );
249	#endif
250	#else
251	m_pcSliceEncoder->initEncSlice ( pcPic, iPOCLast, uiPOCCurr, iNumPicRcvd, iGOPid, pcSlice, m_pcEncTop->getSPS(), m_pcEncTop->getPPS() );
252	#endif
253	pcSlice->setLastIDR(m_iLastIDR);
254	pcSlice->setSliceIdx(0);
255	pcSlice->setViewId( m_pcEncTop->getViewId() );
256	pcSlice->setIsDepth( m_pcEncTop->getIsDepth() );
257	#if INTER_VIEW_VECTOR_SCALING_C0115
258	pcSlice->setIVScalingFlag( m_pcEncTop->getUseIVS() );
259	#endif
260
261	m_pcEncTop->getSPS()->setDisInter4x4(m_pcEncTop->getDisInter4x4());
262	pcSlice->setScalingList ( m_pcEncTop->getScalingList() );
263	if(m_pcEncTop->getUseScalingListId() == SCALING_LIST_OFF)
264	{
265	m_pcEncTop->getTrQuant()->setFlatScalingList();
266	m_pcEncTop->getTrQuant()->setUseScalingList(false);
267	}
268	else if(m_pcEncTop->getUseScalingListId() == SCALING_LIST_DEFAULT)
269	{
270	pcSlice->setDefaultScalingList ();
271	pcSlice->getScalingList()->setScalingListPresentFlag(true);
272	m_pcEncTop->getTrQuant()->setScalingList(pcSlice->getScalingList());
273	m_pcEncTop->getTrQuant()->setUseScalingList(true);
274	}
275	else if(m_pcEncTop->getUseScalingListId() == SCALING_LIST_FILE_READ)
276	{
277	if(pcSlice->getScalingList()->xParseScalingList(m_pcCfg->getScalingListFile()))
278	{
279	pcSlice->setDefaultScalingList ();
280	}
281	pcSlice->getScalingList()->checkDcOfMatrix();
282	pcSlice->getScalingList()->setScalingListPresentFlag(pcSlice->checkDefaultScalingList());
283	m_pcEncTop->getTrQuant()->setScalingList(pcSlice->getScalingList());
284	m_pcEncTop->getTrQuant()->setUseScalingList(true);
285	}
286	else
287	{
288	printf("error : ScalingList == %d no support\n",m_pcEncTop->getUseScalingListId());
289	assert(0);
290	}
291
292	#if HHI_INTERVIEW_SKIP
293	if ( m_pcEncTop->getInterViewSkip() )
294	{
295	m_pcEncTop->getEncTop()->getUsedPelsMap( pcPic->getViewId(), pcPic->getPOC(), pcPic->getUsedPelsMap() );
296	}
297	#endif
298	// Slice info. refinement
299	if( pcSlice->getSliceType() == B_SLICE )
300	{
301	#if QC_REM_IDV_B0046
302	if( m_pcCfg->getGOPEntry(pcSlice->getSPS()->getViewId() && ((getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDR) \|\| (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_CRA))? MAX_GOP : iGOPid ).m_sliceType == 'P' ) { pcSlice->setSliceType( P_SLICE ); }
303	#else
304	if( m_pcCfg->getGOPEntry( (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDV) ? MAX_GOP : iGOPid ).m_sliceType == 'P' ) { pcSlice->setSliceType( P_SLICE ); }
305	#endif
306	}
307
308	// Set the nal unit type
309	pcSlice->setNalUnitType( getNalUnitType(uiPOCCurr) );
310	pcSlice->setNalUnitTypeBaseViewMvc( getNalUnitTypeBaseViewMvc(uiPOCCurr) );
311
312	// Do decoding refresh marking if any
313	pcSlice->decodingRefreshMarking(m_pocCRA, m_bRefreshPending, rcListPic);
314
315	if ( !pcSlice->getPPS()->getEnableTMVPFlag() && pcPic->getTLayer() == 0 )
316	{
317	pcSlice->decodingMarkingForNoTMVP( rcListPic, pcSlice->getPOC() );
318	}
319
320	m_pcEncTop->selectReferencePictureSet(pcSlice, uiPOCCurr, iGOPid,rcListPic);
321	pcSlice->getRPS()->setNumberOfLongtermPictures(0);
322
323	if(pcSlice->checkThatAllRefPicsAreAvailable(rcListPic, pcSlice->getRPS(), false) != 0)
324	{
325	pcSlice->createExplicitReferencePictureSetFromReference(rcListPic, pcSlice->getRPS());
326	}
327	pcSlice->applyReferencePictureSet(rcListPic, pcSlice->getRPS());
328
329	#if H0566_TLA_SET_FOR_SWITCHING_POINTS
330	if(pcSlice->getTLayer() > 0)
331	{
332	if(pcSlice->isTemporalLayerSwitchingPoint(rcListPic, pcSlice->getRPS()))
333	{
334	pcSlice->setNalUnitType(NAL_UNIT_CODED_SLICE_TLA);
335	}
336	}
337	#endif
338
339	#if !QC_REM_IDV_B0046
340	pcSlice->setNumRefIdx( REF_PIC_LIST_0, min( m_pcCfg->getGOPEntry( (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDV) ? MAX_GOP : iGOPid ).m_numRefPicsActive, (pcSlice->getRPS()->getNumberOfPictures() + pcSlice->getSPS()->getNumberOfUsableInterViewRefs()) ) );
341	pcSlice->setNumRefIdx( REF_PIC_LIST_1, min( m_pcCfg->getGOPEntry( (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDV) ? MAX_GOP : iGOPid ).m_numRefPicsActive, (pcSlice->getRPS()->getNumberOfPictures() + pcSlice->getSPS()->getNumberOfUsableInterViewRefs()) ) );
342	#else
343
344	Bool bNalRAP = ((getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_CRA) \|\| (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDR)) && (pcSlice->getSPS()->getViewId()) ? 1: 0;
345	pcSlice->setNumRefIdx( REF_PIC_LIST_0, min( m_pcCfg->getGOPEntry( bNalRAP ? MAX_GOP : iGOPid ).m_numRefPicsActive, (pcSlice->getRPS()->getNumberOfPictures() + pcSlice->getSPS()->getNumberOfUsableInterViewRefs()) ) );
346	pcSlice->setNumRefIdx( REF_PIC_LIST_1, min( m_pcCfg->getGOPEntry( bNalRAP ? MAX_GOP : iGOPid ).m_numRefPicsActive, (pcSlice->getRPS()->getNumberOfPictures() + pcSlice->getSPS()->getNumberOfUsableInterViewRefs()) ) );
347	#endif
348	TComRefPicListModification* refPicListModification = pcSlice->getRefPicListModification();
349	refPicListModification->setRefPicListModificationFlagL0( false );
350	refPicListModification->setRefPicListModificationFlagL1( false );
351	xSetRefPicListModificationsMvc( pcSlice, uiPOCCurr, iGOPid );
352
353	#if ADAPTIVE_QP_SELECTION
354	pcSlice->setTrQuant( m_pcEncTop->getTrQuant() );
355	#endif
356	// Set reference list
357	TAppEncTop* tAppEncTop = m_pcEncTop->getEncTop();
358	assert( tAppEncTop != NULL );
359
360
361	#if FLEX_CODING_ORDER_M23723
362	TComPic * pcTexturePic;
363	if(m_pcEncTop->getIsDepth() == 1)
364	{
365	TComPicYuv * recText;
366	recText = tAppEncTop->getPicYuvFromView(m_pcEncTop->getViewId(), pcSlice->getPOC(), false ,true);
367	if(recText == NULL)
368	{
369	pcSlice->setTexturePic(NULL);
370	}
371	else
372	{
373	pcTexturePic = m_pcEncTop->getIsDepth() ? tAppEncTop->getPicFromView( m_pcEncTop->getViewId(), pcSlice->getPOC(), false ) : NULL;
374	pcSlice->setTexturePic( pcTexturePic );
375	}
376	}
377	else
378	{
379	pcTexturePic = m_pcEncTop->getIsDepth() ? tAppEncTop->getPicFromView( m_pcEncTop->getViewId(), pcSlice->getPOC(), false ) : NULL;
380	assert( !m_pcEncTop->getIsDepth() \|\| pcTexturePic != NULL );
381	pcSlice->setTexturePic( pcTexturePic );
382	}
383
384	#else
385	TComPic * const pcTexturePic = m_pcEncTop->getIsDepth() ? tAppEncTop->getPicFromView( m_pcEncTop->getViewId(), pcSlice->getPOC(), false ) : NULL;
386	assert( !m_pcEncTop->getIsDepth() \|\| pcTexturePic != NULL );
387	pcSlice->setTexturePic( pcTexturePic );
388
389	#endif
390	std::vector<TComPic*> apcInterViewRefPics = tAppEncTop->getInterViewRefPics( m_pcEncTop->getViewId(), pcSlice->getPOC(), m_pcEncTop->getIsDepth(), pcSlice->getSPS() );
391	pcSlice->setRefPicListMvc( rcListPic, apcInterViewRefPics );
392
393	// Slice info. refinement
394	if( pcSlice->getSliceType() == B_SLICE )
395	{
396	#if !QC_REM_IDV_B0046
397	if( m_pcCfg->getGOPEntry( (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDV) ? MAX_GOP : iGOPid ).m_sliceType == 'P' ) { pcSlice->setSliceType( P_SLICE ); }
398	#else
399	Bool bRAP = ((getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_CRA) \|\| (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDR)) && (pcSlice->getSPS()->getViewId()) ? 1: 0;
400	if( m_pcCfg->getGOPEntry( bRAP ? MAX_GOP : iGOPid ).m_sliceType == 'P' ) { pcSlice->setSliceType( P_SLICE ); }
401	#endif
402	}
403
404	if (pcSlice->getSliceType() != B_SLICE \|\| !pcSlice->getSPS()->getUseLComb())
405	{
406	pcSlice->setNumRefIdx(REF_PIC_LIST_C, 0);
407	pcSlice->setRefPicListCombinationFlag(false);
408	pcSlice->setRefPicListModificationFlagLC(false);
409	}
410	else
411	{
412	pcSlice->setRefPicListCombinationFlag(pcSlice->getSPS()->getUseLComb());
413	pcSlice->setRefPicListModificationFlagLC(pcSlice->getSPS()->getLCMod());
414	pcSlice->setNumRefIdx(REF_PIC_LIST_C, pcSlice->getNumRefIdx(REF_PIC_LIST_0));
415	}
416
417	if (pcSlice->getSliceType() == B_SLICE)
418	{
419	pcSlice->setColDir(uiColDir);
420	Bool bLowDelay = true;
421	Int iCurrPOC = pcSlice->getPOC();
422	Int iRefIdx = 0;
423
424	for (iRefIdx = 0; iRefIdx < pcSlice->getNumRefIdx(REF_PIC_LIST_0) && bLowDelay; iRefIdx++)
425	{
426	if ( pcSlice->getRefPic(REF_PIC_LIST_0, iRefIdx)->getPOC() > iCurrPOC )
427	{
428	bLowDelay = false;
429	}
430	}
431	for (iRefIdx = 0; iRefIdx < pcSlice->getNumRefIdx(REF_PIC_LIST_1) && bLowDelay; iRefIdx++)
432	{
433	if ( pcSlice->getRefPic(REF_PIC_LIST_1, iRefIdx)->getPOC() > iCurrPOC )
434	{
435	bLowDelay = false;
436	}
437	}
438
439	pcSlice->setCheckLDC(bLowDelay);
440	}
441
442	uiColDir = 1-uiColDir;
443
444	//-------------------------------------------------------------
445	pcSlice->setRefPOCnViewListsMvc();
446
447	pcSlice->setNoBackPredFlag( false );
448	if ( pcSlice->getSliceType() == B_SLICE && !pcSlice->getRefPicListCombinationFlag())
449	{
450	if ( pcSlice->getNumRefIdx(RefPicList( 0 ) ) == pcSlice->getNumRefIdx(RefPicList( 1 ) ) )
451	{
452	pcSlice->setNoBackPredFlag( true );
453	int i;
454	for ( i=0; i < pcSlice->getNumRefIdx(RefPicList( 1 ) ); i++ )
455	{
456	if ( pcSlice->getRefPOC(RefPicList(1), i) != pcSlice->getRefPOC(RefPicList(0), i) )
457	{
458	pcSlice->setNoBackPredFlag( false );
459	break;
460	}
461	}
462	}
463	}
464
465	if(pcSlice->getNoBackPredFlag())
466	{
467	pcSlice->setNumRefIdx(REF_PIC_LIST_C, 0);
468	}
469	pcSlice->generateCombinedList();
470
471	#if HHI_VSO
472	Bool bUseVSO = m_pcEncTop->getUseVSO();
473	m_pcRdCost->setUseVSO( bUseVSO );
474	#if SAIT_VSO_EST_A0033
475	m_pcRdCost->setUseEstimatedVSD( m_pcEncTop->getUseEstimatedVSD() );
476	#endif
477
478	if ( bUseVSO )
479	{
480	Int iVSOMode = m_pcEncTop->getVSOMode();
481	m_pcRdCost->setVSOMode( iVSOMode );
482
483	#if HHI_VSO_DIST_INT
484	m_pcRdCost->setAllowNegDist( m_pcEncTop->getAllowNegDist() );
485	#endif
486
487
488	#if SAIT_VSO_EST_A0033
489	#ifdef FLEX_CODING_ORDER_M23723
490	{
491	Bool flagRec;
492	flagRec = ((m_pcEncTop->getEncTop()->getPicYuvFromView( pcSlice->getViewId(), pcSlice->getPOC(), false, true) == NULL) ? false: true);
493	m_pcRdCost->setVideoRecPicYuv( m_pcEncTop->getEncTop()->getPicYuvFromView( pcSlice->getViewId(), pcSlice->getPOC(), false, flagRec ) );
494	m_pcRdCost->setDepthPicYuv ( m_pcEncTop->getEncTop()->getPicYuvFromView( pcSlice->getViewId(), pcSlice->getPOC(), true, false ) );
495	}
496	#else
497	m_pcRdCost->setVideoRecPicYuv( m_pcEncTop->getEncTop()->getPicYuvFromView( pcSlice->getViewId(), pcSlice->getPOC(), false, true ) );
498	m_pcRdCost->setDepthPicYuv ( m_pcEncTop->getEncTop()->getPicYuvFromView( pcSlice->getViewId(), pcSlice->getPOC(), true, false ) );
499	#endif
500	#endif
501	#if LGE_WVSO_A0119
502	Bool bUseWVSO = m_pcEncTop->getUseWVSO();
503	m_pcRdCost->setUseWVSO( bUseWVSO );
504	#endif
505
506	}
507	#endif
508	/////////////////////////////////////////////////////////////////////////////////////////////////// Compress a slice
509	// Slice compression
510	if (m_pcCfg->getUseASR())
511	{
512	m_pcSliceEncoder->setSearchRange(pcSlice);
513	}
514
515	Bool bGPBcheck=false;
516	if ( pcSlice->getSliceType() == B_SLICE)
517	{
518	if ( pcSlice->getNumRefIdx(RefPicList( 0 ) ) == pcSlice->getNumRefIdx(RefPicList( 1 ) ) )
519	{
520	bGPBcheck=true;
521	int i;
522	for ( i=0; i < pcSlice->getNumRefIdx(RefPicList( 1 ) ); i++ )
523	{
524	if ( pcSlice->getRefPOC(RefPicList(1), i) != pcSlice->getRefPOC(RefPicList(0), i) )
525	{
526	bGPBcheck=false;
527	break;
528	}
529	}
530	}
531	}
532	if(bGPBcheck)
533	{
534	pcSlice->setMvdL1ZeroFlag(true);
535	}
536	else
537	{
538	pcSlice->setMvdL1ZeroFlag(false);
539	}
540	pcPic->getSlice(pcSlice->getSliceIdx())->setMvdL1ZeroFlag(pcSlice->getMvdL1ZeroFlag());
541
542	UInt uiNumSlices = 1;
543
544	UInt uiInternalAddress = pcPic->getNumPartInCU()-4;
545	UInt uiExternalAddress = pcPic->getPicSym()->getNumberOfCUsInFrame()-1;
546	UInt uiPosX = ( uiExternalAddress % pcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
547	UInt uiPosY = ( uiExternalAddress / pcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
548	UInt uiWidth = pcSlice->getSPS()->getPicWidthInLumaSamples();
549	UInt uiHeight = pcSlice->getSPS()->getPicHeightInLumaSamples();
550	while(uiPosX>=uiWidth\|\|uiPosY>=uiHeight)
551	{
552	uiInternalAddress--;
553	uiPosX = ( uiExternalAddress % pcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
554	uiPosY = ( uiExternalAddress / pcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
555	}
556	uiInternalAddress++;
557	if(uiInternalAddress==pcPic->getNumPartInCU())
558	{
559	uiInternalAddress = 0;
560	uiExternalAddress++;
561	}
562	UInt uiRealEndAddress = uiExternalAddress*pcPic->getNumPartInCU()+uiInternalAddress;
563
564	UInt uiCummulativeTileWidth;
565	UInt uiCummulativeTileHeight;
566	Int p, j;
567	UInt uiEncCUAddr;
568
569
570	if( pcSlice->getPPS()->getColumnRowInfoPresent() == 1 ) //derive the tile parameters from PPS
571	{
572	//set NumColumnsMinus1 and NumRowsMinus1
573	pcPic->getPicSym()->setNumColumnsMinus1( pcSlice->getPPS()->getNumColumnsMinus1() );
574	pcPic->getPicSym()->setNumRowsMinus1( pcSlice->getPPS()->getNumRowsMinus1() );
575
576	//create the TComTileArray
577	pcPic->getPicSym()->xCreateTComTileArray();
578
579	if( pcSlice->getPPS()->getUniformSpacingIdr() == 1 )
580	{
581	//set the width for each tile
582	for(j=0; j < pcPic->getPicSym()->getNumRowsMinus1()+1; j++)
583	{
584	for(p=0; p < pcPic->getPicSym()->getNumColumnsMinus1()+1; p++)
585	{
586	pcPic->getPicSym()->getTComTile( j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p )->
587	setTileWidth( (p+1)*pcPic->getPicSym()->getFrameWidthInCU()/(pcPic->getPicSym()->getNumColumnsMinus1()+1)
588	- (p*pcPic->getPicSym()->getFrameWidthInCU())/(pcPic->getPicSym()->getNumColumnsMinus1()+1) );
589	}
590	}
591
592	//set the height for each tile
593	for(j=0; j < pcPic->getPicSym()->getNumColumnsMinus1()+1; j++)
594	{
595	for(p=0; p < pcPic->getPicSym()->getNumRowsMinus1()+1; p++)
596	{
597	pcPic->getPicSym()->getTComTile( p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j )->
598	setTileHeight( (p+1)*pcPic->getPicSym()->getFrameHeightInCU()/(pcPic->getPicSym()->getNumRowsMinus1()+1)
599	- (p*pcPic->getPicSym()->getFrameHeightInCU())/(pcPic->getPicSym()->getNumRowsMinus1()+1) );
600	}
601	}
602	}
603	else
604	{
605	//set the width for each tile
606	for(j=0; j < pcPic->getPicSym()->getNumRowsMinus1()+1; j++)
607	{
608	uiCummulativeTileWidth = 0;
609	for(p=0; p < pcPic->getPicSym()->getNumColumnsMinus1(); p++)
610	{
611	pcPic->getPicSym()->getTComTile( j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p )->setTileWidth( pcSlice->getPPS()->getColumnWidth(p) );
612	uiCummulativeTileWidth += pcSlice->getPPS()->getColumnWidth(p);
613	}
614	pcPic->getPicSym()->getTComTile(j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p)->setTileWidth( pcPic->getPicSym()->getFrameWidthInCU()-uiCummulativeTileWidth );
615	}
616
617	//set the height for each tile
618	for(j=0; j < pcPic->getPicSym()->getNumColumnsMinus1()+1; j++)
619	{
620	uiCummulativeTileHeight = 0;
621	for(p=0; p < pcPic->getPicSym()->getNumRowsMinus1(); p++)
622	{
623	pcPic->getPicSym()->getTComTile( p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j )->setTileHeight( pcSlice->getPPS()->getRowHeight(p) );
624	uiCummulativeTileHeight += pcSlice->getPPS()->getRowHeight(p);
625	}
626	pcPic->getPicSym()->getTComTile(p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j)->setTileHeight( pcPic->getPicSym()->getFrameHeightInCU()-uiCummulativeTileHeight );
627	}
628	}
629	}
630	else //derive the tile parameters from SPS
631	{
632	//set NumColumnsMins1 and NumRowsMinus1
633	pcPic->getPicSym()->setNumColumnsMinus1( pcSlice->getSPS()->getNumColumnsMinus1() );
634	pcPic->getPicSym()->setNumRowsMinus1( pcSlice->getSPS()->getNumRowsMinus1() );
635
636	//create the TComTileArray
637	pcPic->getPicSym()->xCreateTComTileArray();
638
639	if( pcSlice->getSPS()->getUniformSpacingIdr() == 1 )
640	{
641	//set the width for each tile
642	for(j=0; j < pcPic->getPicSym()->getNumRowsMinus1()+1; j++)
643	{
644	for(p=0; p < pcPic->getPicSym()->getNumColumnsMinus1()+1; p++)
645	{
646	pcPic->getPicSym()->getTComTile( j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p )->
647	setTileWidth( (p+1)*pcPic->getPicSym()->getFrameWidthInCU()/(pcPic->getPicSym()->getNumColumnsMinus1()+1)
648	- (p*pcPic->getPicSym()->getFrameWidthInCU())/(pcPic->getPicSym()->getNumColumnsMinus1()+1) );
649	}
650	}
651
652	//set the height for each tile
653	for(j=0; j < pcPic->getPicSym()->getNumColumnsMinus1()+1; j++)
654	{
655	for(p=0; p < pcPic->getPicSym()->getNumRowsMinus1()+1; p++)
656	{
657	pcPic->getPicSym()->getTComTile( p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j )->
658	setTileHeight( (p+1)*pcPic->getPicSym()->getFrameHeightInCU()/(pcPic->getPicSym()->getNumRowsMinus1()+1)
659	- (p*pcPic->getPicSym()->getFrameHeightInCU())/(pcPic->getPicSym()->getNumRowsMinus1()+1) );
660	}
661	}
662	}
663
664	else
665	{
666	//set the width for each tile
667	for(j=0; j < pcPic->getPicSym()->getNumRowsMinus1()+1; j++)
668	{
669	uiCummulativeTileWidth = 0;
670	for(p=0; p < pcPic->getPicSym()->getNumColumnsMinus1(); p++)
671	{
672	pcPic->getPicSym()->getTComTile( j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p )->setTileWidth( pcSlice->getSPS()->getColumnWidth(p) );
673	uiCummulativeTileWidth += pcSlice->getSPS()->getColumnWidth(p);
674	}
675	pcPic->getPicSym()->getTComTile(j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p)->setTileWidth( pcPic->getPicSym()->getFrameWidthInCU()-uiCummulativeTileWidth );
676	}
677
678	//set the height for each tile
679	for(j=0; j < pcPic->getPicSym()->getNumColumnsMinus1()+1; j++)
680	{
681	uiCummulativeTileHeight = 0;
682	for(p=0; p < pcPic->getPicSym()->getNumRowsMinus1(); p++)
683	{
684	pcPic->getPicSym()->getTComTile( p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j )->setTileHeight( pcSlice->getSPS()->getRowHeight(p) );
685	uiCummulativeTileHeight += pcSlice->getSPS()->getRowHeight(p);
686	}
687	pcPic->getPicSym()->getTComTile(p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j)->setTileHeight( pcPic->getPicSym()->getFrameHeightInCU()-uiCummulativeTileHeight );
688	}
689	}
690	}
691
692	//initialize each tile of the current picture
693	pcPic->getPicSym()->xInitTiles();
694
695	// Allocate some coders, now we know how many tiles there are.
696	Int iNumSubstreams = pcSlice->getPPS()->getNumSubstreams();
697
698	//generate the Coding Order Map and Inverse Coding Order Map
699	for(p=0, uiEncCUAddr=0; p<pcPic->getPicSym()->getNumberOfCUsInFrame(); p++, uiEncCUAddr = pcPic->getPicSym()->xCalculateNxtCUAddr(uiEncCUAddr))
700	{
701	pcPic->getPicSym()->setCUOrderMap(p, uiEncCUAddr);
702	pcPic->getPicSym()->setInverseCUOrderMap(uiEncCUAddr, p);
703	}
704	pcPic->getPicSym()->setCUOrderMap(pcPic->getPicSym()->getNumberOfCUsInFrame(), pcPic->getPicSym()->getNumberOfCUsInFrame());
705	pcPic->getPicSym()->setInverseCUOrderMap(pcPic->getPicSym()->getNumberOfCUsInFrame(), pcPic->getPicSym()->getNumberOfCUsInFrame());
706	if (pcSlice->getPPS()->getEntropyCodingMode())
707	{
708	// Allocate some coders, now we know how many tiles there are.
709	m_pcEncTop->createWPPCoders(iNumSubstreams);
710	pcSbacCoders = m_pcEncTop->getSbacCoders();
711	pcSubstreamsOut = new TComOutputBitstream[iNumSubstreams];
712	}
713
714	UInt uiStartCUAddrSliceIdx = 0; // used to index "m_uiStoredStartCUAddrForEncodingSlice" containing locations of slice boundaries
715	UInt uiStartCUAddrSlice = 0; // used to keep track of current slice's starting CU addr.
716	pcSlice->setSliceCurStartCUAddr( uiStartCUAddrSlice ); // Setting "start CU addr" for current slice
717	memset(m_uiStoredStartCUAddrForEncodingSlice, 0, sizeof(UInt) * (pcPic->getPicSym()->getNumberOfCUsInFrame()*pcPic->getNumPartInCU()+1));
718
719	UInt uiStartCUAddrEntropySliceIdx = 0; // used to index "m_uiStoredStartCUAddrForEntropyEncodingSlice" containing locations of slice boundaries
720	UInt uiStartCUAddrEntropySlice = 0; // used to keep track of current Entropy slice's starting CU addr.
721	pcSlice->setEntropySliceCurStartCUAddr( uiStartCUAddrEntropySlice ); // Setting "start CU addr" for current Entropy slice
722
723	memset(m_uiStoredStartCUAddrForEncodingEntropySlice, 0, sizeof(UInt) * (pcPic->getPicSym()->getNumberOfCUsInFrame()*pcPic->getNumPartInCU()+1));
724	UInt uiNextCUAddr = 0;
725	m_uiStoredStartCUAddrForEncodingSlice[uiStartCUAddrSliceIdx++] = uiNextCUAddr;
726	m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx++] = uiNextCUAddr;
727
728	#if FCO_DVP_REFINE_C0132_C0170
729	pcPic->setDepthCoded(false);
730
731	if(pcSlice->getViewId() != 0)
732	{
733	if(pcSlice->getSPS()->isDepth() == 0 )
734	{
735	TComPic * recDepthMapBuffer;
736	recDepthMapBuffer = m_pcEncTop->getEncTop()->getPicFromView( pcSlice->getViewId(), pcSlice->getPOC(), true );
737	pcSlice->getPic()->setRecDepthMap(recDepthMapBuffer);
738	if(recDepthMapBuffer->getReconMark())
739	{
740	pcPic->setDepthCoded(true);
741	}
742	}
743	}
744	#endif
745
746	#if DEPTH_MAP_GENERATION
747	// init view component and predict virtual depth map
748	m_pcDepthMapGenerator->initViewComponent( pcPic );
749	#if !H3D_NBDV
750	m_pcDepthMapGenerator->predictDepthMap ( pcPic );
751	#endif
752	#endif
753	#if H3D_IVMP
754	m_pcDepthMapGenerator->covertOrgDepthMap( pcPic );
755	#endif
756	#if H3D_IVRP
757	m_pcResidualGenerator->initViewComponent( pcPic );
758	#endif
759
760	#if H3D_NBDV
761	if(pcSlice->getViewId() && pcSlice->getSPS()->getMultiviewMvPredMode())
762	{
763	Int iColPoc = pcSlice->getRefPOC(RefPicList(pcSlice->getColDir()), pcSlice->getColRefIdx());
764	pcPic->setRapbCheck(pcPic->getDisCandRefPictures(iColPoc));
765	}
766	#endif
767	while(uiNextCUAddr<uiRealEndAddress) // determine slice boundaries
768	{
769	pcSlice->setNextSlice ( false );
770	pcSlice->setNextEntropySlice( false );
771	assert(pcPic->getNumAllocatedSlice() == uiStartCUAddrSliceIdx);
772	m_pcSliceEncoder->precompressSlice( pcPic );
773	m_pcSliceEncoder->compressSlice ( pcPic );
774
775	Bool bNoBinBitConstraintViolated = (!pcSlice->isNextSlice() && !pcSlice->isNextEntropySlice());
776	if (pcSlice->isNextSlice() \|\| (bNoBinBitConstraintViolated && m_pcCfg->getSliceMode()==AD_HOC_SLICES_FIXED_NUMBER_OF_LCU_IN_SLICE))
777	{
778	uiStartCUAddrSlice = pcSlice->getSliceCurEndCUAddr();
779	// Reconstruction slice
780	m_uiStoredStartCUAddrForEncodingSlice[uiStartCUAddrSliceIdx++] = uiStartCUAddrSlice;
781	// Entropy slice
782	if (uiStartCUAddrEntropySliceIdx>0 && m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx-1] != uiStartCUAddrSlice)
783	{
784	m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx++] = uiStartCUAddrSlice;
785	}
786
787	if (uiStartCUAddrSlice < uiRealEndAddress)
788	{
789	pcPic->allocateNewSlice();
790	pcPic->setCurrSliceIdx ( uiStartCUAddrSliceIdx-1 );
791	m_pcSliceEncoder->setSliceIdx ( uiStartCUAddrSliceIdx-1 );
792	pcSlice = pcPic->getSlice ( uiStartCUAddrSliceIdx-1 );
793	pcSlice->copySliceInfo ( pcPic->getSlice(0) );
794	pcSlice->setSliceIdx ( uiStartCUAddrSliceIdx-1 );
795	pcSlice->setSliceCurStartCUAddr ( uiStartCUAddrSlice );
796	pcSlice->setEntropySliceCurStartCUAddr ( uiStartCUAddrSlice );
797	pcSlice->setSliceBits(0);
798	uiNumSlices ++;
799	}
800	}
801	else if (pcSlice->isNextEntropySlice() \|\| (bNoBinBitConstraintViolated && m_pcCfg->getEntropySliceMode()==SHARP_FIXED_NUMBER_OF_LCU_IN_ENTROPY_SLICE))
802	{
803	uiStartCUAddrEntropySlice = pcSlice->getEntropySliceCurEndCUAddr();
804	m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx++] = uiStartCUAddrEntropySlice;
805	pcSlice->setEntropySliceCurStartCUAddr( uiStartCUAddrEntropySlice );
806	}
807	else
808	{
809	uiStartCUAddrSlice = pcSlice->getSliceCurEndCUAddr();
810	uiStartCUAddrEntropySlice = pcSlice->getEntropySliceCurEndCUAddr();
811	}
812
813	uiNextCUAddr = (uiStartCUAddrSlice > uiStartCUAddrEntropySlice) ? uiStartCUAddrSlice : uiStartCUAddrEntropySlice;
814	}
815	m_uiStoredStartCUAddrForEncodingSlice[uiStartCUAddrSliceIdx++] = pcSlice->getSliceCurEndCUAddr();
816	m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx++] = pcSlice->getSliceCurEndCUAddr();
817
818	pcSlice = pcPic->getSlice(0);
819
820	#if H3D_IVRP
821	// set residual picture
822	m_pcResidualGenerator->setRecResidualPic( pcPic );
823	#endif
824	#if DEPTH_MAP_GENERATION
825	#if !H3D_NBDV
826	// update virtual depth map
827	m_pcDepthMapGenerator->updateDepthMap( pcPic );
828	#endif
829	#endif
830
831	//-- Loop filter
832	Bool bLFCrossTileBoundary = (pcSlice->getPPS()->getTileBehaviorControlPresentFlag() == 1)?
833	(pcSlice->getPPS()->getLFCrossTileBoundaryFlag()):(pcSlice->getPPS()->getSPS()->getLFCrossTileBoundaryFlag());
834	m_pcLoopFilter->setCfg(pcSlice->getPPS()->getDeblockingFilterControlPresent(), pcSlice->getLoopFilterDisable(), pcSlice->getLoopFilterBetaOffset(), pcSlice->getLoopFilterTcOffset(), bLFCrossTileBoundary);
835	m_pcLoopFilter->loopFilterPic( pcPic );
836
837	pcSlice = pcPic->getSlice(0);
838	if(pcSlice->getSPS()->getUseSAO() \|\| pcSlice->getSPS()->getUseALF())
839	{
840	Int sliceGranularity = pcSlice->getPPS()->getSliceGranularity();
841	pcPic->createNonDBFilterInfo(m_uiStoredStartCUAddrForEncodingSlice, uiNumSlices, sliceGranularity, pcSlice->getSPS()->getLFCrossSliceBoundaryFlag(),pcPic->getPicSym()->getNumTiles() ,bLFCrossTileBoundary);
842	}
843
844
845	pcSlice = pcPic->getSlice(0);
846
847	if(pcSlice->getSPS()->getUseSAO())
848	{
849	m_pcSAO->createPicSaoInfo(pcPic, uiNumSlices);
850	}
851
852	AlfParamSet* alfSliceParams = NULL;
853	std::vector<AlfCUCtrlInfo>* alfCUCtrlParam = NULL;
854	pcSlice = pcPic->getSlice(0);
855
856	if(pcSlice->getSPS()->getUseALF())
857	{
858	m_pcAdaptiveLoopFilter->createPicAlfInfo(pcPic, uiNumSlices, pcSlice->getSliceQp());
859	m_pcAdaptiveLoopFilter->initALFEnc(m_pcCfg->getALFParamInSlice(), m_pcCfg->getALFPicBasedEncode(), uiNumSlices, alfSliceParams, alfCUCtrlParam);
860	}
861
862	/////////////////////////////////////////////////////////////////////////////////////////////////// File writing
863	// Set entropy coder
864	m_pcEntropyCoder->setEntropyCoder ( m_pcCavlcCoder, pcSlice );
865
866	/* write various header sets. */
867	if ( m_bSeqFirst )
868	{
869	#if QC_MVHEVC_B0046
870	if(!m_pcEncTop->getLayerId())
871	{
872	#endif
873	#if VIDYO_VPS_INTEGRATION\|QC_MVHEVC_B0046
874	{
875	OutputNALUnit nalu(NAL_UNIT_VPS, true, m_pcEncTop->getLayerId());
876	m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
877	m_pcEntropyCoder->encodeVPS(m_pcEncTop->getEncTop()->getVPS());
878	writeRBSPTrailingBits(nalu.m_Bitstream);
879	accessUnit.push_back(new NALUnitEBSP(nalu));
880	}
881	#endif
882	#if VIDYO_VPS_INTEGRATION\|QC_MVHEVC_B0046
883	OutputNALUnit nalu(NAL_UNIT_SPS, true, m_pcEncTop->getLayerId());
884	#else
885	OutputNALUnit nalu(NAL_UNIT_SPS, true, m_pcEncTop->getViewId(), m_pcEncTop->getIsDepth());
886	#endif
887	m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
888	pcSlice->getSPS()->setNumSubstreams( pcSlice->getPPS()->getNumSubstreams() );
889	#if HHI_MPI \|\| H3D_QTL
890	m_pcEntropyCoder->encodeSPS(pcSlice->getSPS(), m_pcEncTop->getIsDepth());
891	#else
892	m_pcEntropyCoder->encodeSPS(pcSlice->getSPS());
893	#endif
894	writeRBSPTrailingBits(nalu.m_Bitstream);
895	accessUnit.push_back(new NALUnitEBSP(nalu));
896
897	#if VIDYO_VPS_INTEGRATION\|QC_MVHEVC_B0046
898	#if QC_MVHEVC_B0046
899	nalu = NALUnit(NAL_UNIT_PPS, true, m_pcEncTop->getLayerId());
900	#else
901	nalu = NALUnit(NAL_UNIT_PPS, true, m_pcEncTop->getLayerId());
902	#endif
903	#else
904	nalu = NALUnit(NAL_UNIT_PPS, true, m_pcEncTop->getViewId(), m_pcEncTop->getIsDepth());
905	#endif
906	m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
907	m_pcEntropyCoder->encodePPS(pcSlice->getPPS());
908	writeRBSPTrailingBits(nalu.m_Bitstream);
909	accessUnit.push_back(new NALUnitEBSP(nalu));
910	#if QC_MVHEVC_B0046
911	}
912	#endif
913	m_bSeqFirst = false;
914	}
915
916	/* use the main bitstream buffer for storing the marshalled picture */
917	m_pcEntropyCoder->setBitstream(NULL);
918
919	uiStartCUAddrSliceIdx = 0;
920	uiStartCUAddrSlice = 0;
921
922	uiStartCUAddrEntropySliceIdx = 0;
923	uiStartCUAddrEntropySlice = 0;
924	uiNextCUAddr = 0;
925	pcSlice = pcPic->getSlice(uiStartCUAddrSliceIdx);
926
927	Int processingState = (pcSlice->getSPS()->getUseALF() \|\| pcSlice->getSPS()->getUseSAO() \|\| pcSlice->getSPS()->getScalingListFlag() \|\| pcSlice->getSPS()->getUseDF())?(EXECUTE_INLOOPFILTER):(ENCODE_SLICE);
928
929	static Int iCurrAPSIdx = 0;
930	Int iCodedAPSIdx = 0;
931	TComSlice* pcSliceForAPS = NULL;
932
933	bool skippedSlice=false;
934	while (uiNextCUAddr < uiRealEndAddress) // Iterate over all slices
935	{
936	switch(processingState)
937	{
938	case ENCODE_SLICE:
939	{
940	pcSlice->setNextSlice ( false );
941	pcSlice->setNextEntropySlice( false );
942	if (uiNextCUAddr == m_uiStoredStartCUAddrForEncodingSlice[uiStartCUAddrSliceIdx])
943	{
944	pcSlice = pcPic->getSlice(uiStartCUAddrSliceIdx);
945	#if COLLOCATED_REF_IDX
946	if(uiStartCUAddrSliceIdx > 0 && pcSlice->getSliceType()!= I_SLICE)
947	{
948	pcSlice->checkColRefIdx(uiStartCUAddrSliceIdx, pcPic);
949	}
950	#endif
951	pcPic->setCurrSliceIdx(uiStartCUAddrSliceIdx);
952	m_pcSliceEncoder->setSliceIdx(uiStartCUAddrSliceIdx);
953	assert(uiStartCUAddrSliceIdx == pcSlice->getSliceIdx());
954	// Reconstruction slice
955	pcSlice->setSliceCurStartCUAddr( uiNextCUAddr ); // to be used in encodeSlice() + context restriction
956	pcSlice->setSliceCurEndCUAddr ( m_uiStoredStartCUAddrForEncodingSlice[uiStartCUAddrSliceIdx+1 ] );
957	// Entropy slice
958	pcSlice->setEntropySliceCurStartCUAddr( uiNextCUAddr ); // to be used in encodeSlice() + context restriction
959	pcSlice->setEntropySliceCurEndCUAddr ( m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx+1 ] );
960
961	pcSlice->setNextSlice ( true );
962
963	uiStartCUAddrSliceIdx++;
964	uiStartCUAddrEntropySliceIdx++;
965	}
966	else if (uiNextCUAddr == m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx])
967	{
968	// Entropy slice
969	pcSlice->setEntropySliceCurStartCUAddr( uiNextCUAddr ); // to be used in encodeSlice() + context restriction
970	pcSlice->setEntropySliceCurEndCUAddr ( m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx+1 ] );
971
972	pcSlice->setNextEntropySlice( true );
973
974	uiStartCUAddrEntropySliceIdx++;
975	}
976
977	pcSlice->setRPS(pcPic->getSlice(0)->getRPS());
978	pcSlice->setRPSidx(pcPic->getSlice(0)->getRPSidx());
979	UInt uiDummyStartCUAddr;
980	UInt uiDummyBoundingCUAddr;
981	m_pcSliceEncoder->xDetermineStartAndBoundingCUAddr(uiDummyStartCUAddr,uiDummyBoundingCUAddr,pcPic,true);
982
983	uiInternalAddress = pcPic->getPicSym()->getPicSCUAddr(pcSlice->getEntropySliceCurEndCUAddr()-1) % pcPic->getNumPartInCU();
984	uiExternalAddress = pcPic->getPicSym()->getPicSCUAddr(pcSlice->getEntropySliceCurEndCUAddr()-1) / pcPic->getNumPartInCU();
985	uiPosX = ( uiExternalAddress % pcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
986	uiPosY = ( uiExternalAddress / pcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
987	uiWidth = pcSlice->getSPS()->getPicWidthInLumaSamples();
988	uiHeight = pcSlice->getSPS()->getPicHeightInLumaSamples();
989	while(uiPosX>=uiWidth\|\|uiPosY>=uiHeight)
990	{
991	uiInternalAddress--;
992	uiPosX = ( uiExternalAddress % pcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
993	uiPosY = ( uiExternalAddress / pcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
994	}
995	uiInternalAddress++;
996	if(uiInternalAddress==pcPic->getNumPartInCU())
997	{
998	uiInternalAddress = 0;
999	uiExternalAddress = pcPic->getPicSym()->getCUOrderMap(pcPic->getPicSym()->getInverseCUOrderMap(uiExternalAddress)+1);
1000	}
1001	UInt uiEndAddress = pcPic->getPicSym()->getPicSCUEncOrder(uiExternalAddress*pcPic->getNumPartInCU()+uiInternalAddress);
1002	if(uiEndAddress<=pcSlice->getEntropySliceCurStartCUAddr()) {
1003	UInt uiBoundingAddrSlice, uiBoundingAddrEntropySlice;
1004	uiBoundingAddrSlice = m_uiStoredStartCUAddrForEncodingSlice[uiStartCUAddrSliceIdx];
1005	uiBoundingAddrEntropySlice = m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx];
1006	uiNextCUAddr = min(uiBoundingAddrSlice, uiBoundingAddrEntropySlice);
1007	if(pcSlice->isNextSlice())
1008	{
1009	skippedSlice=true;
1010	}
1011	continue;
1012	}
1013	if(skippedSlice)
1014	{
1015	pcSlice->setNextSlice ( true );
1016	pcSlice->setNextEntropySlice( false );
1017	}
1018	skippedSlice=false;
1019	if (pcSlice->getPPS()->getEntropyCodingMode())
1020	{
1021	pcSlice->allocSubstreamSizes( iNumSubstreams );
1022	for ( UInt ui = 0 ; ui < iNumSubstreams; ui++ )
1023	pcSubstreamsOut[ui].clear();
1024	}
1025
1026	m_pcEntropyCoder->setEntropyCoder ( m_pcCavlcCoder, pcSlice );
1027	m_pcEntropyCoder->resetEntropy ();
1028	/* start slice NALunit */
1029	OutputNALUnit nalu( pcSlice->getNalUnitType(), pcSlice->isReferenced(),
1030	#if !VIDYO_VPS_INTEGRATION &!QC_MVHEVC_B0046
1031	m_pcEncTop->getViewId(), m_pcEncTop->getIsDepth(), pcSlice->getTLayer() );
1032	#else
1033	m_pcEncTop->getLayerId(), pcSlice->getTLayer() );
1034	#endif
1035
1036	Bool bEntropySlice = (!pcSlice->isNextSlice());
1037	if (!bEntropySlice)
1038	{
1039	uiOneBitstreamPerSliceLength = 0; // start of a new slice
1040	}
1041
1042	// used while writing slice header
1043	Int iTransmitLWHeader = (m_pcCfg->getTileMarkerFlag()==0) ? 0 : 1;
1044	pcSlice->setTileMarkerFlag ( iTransmitLWHeader );
1045	m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1046	#if !CABAC_INIT_FLAG
1047	pcSlice->setCABACinitIDC(pcSlice->getSliceType());
1048	#endif
1049
1050	m_pcEntropyCoder->encodeSliceHeader(pcSlice);
1051
1052	if(pcSlice->isNextSlice())
1053	{
1054	if (pcSlice->getSPS()->getUseALF())
1055	{
1056	if(pcSlice->getAlfEnabledFlag())
1057	{
1058
1059	if( pcSlice->getSPS()->getUseALFCoefInSlice())
1060	{
1061	Int iNumSUinLCU = 1<< (g_uiMaxCUDepth << 1);
1062	Int firstLCUAddr = pcSlice->getSliceCurStartCUAddr() / iNumSUinLCU;
1063	Bool isAcrossSlice = pcSlice->getSPS()->getLFCrossSliceBoundaryFlag();
1064	m_pcEntropyCoder->encodeAlfParam( &(alfSliceParams[pcSlice->getSliceIdx()]), false, firstLCUAddr, isAcrossSlice);
1065	}
1066
1067	if( !pcSlice->getSPS()->getUseALFCoefInSlice())
1068	{
1069	AlfCUCtrlInfo& cAlfCUCtrlParam = (*alfCUCtrlParam)[pcSlice->getSliceIdx()];
1070	if(cAlfCUCtrlParam.cu_control_flag)
1071	{
1072	m_pcEntropyCoder->setAlfCtrl( true );
1073	m_pcEntropyCoder->setMaxAlfCtrlDepth(cAlfCUCtrlParam.alf_max_depth);
1074	m_pcCavlcCoder->setAlfCtrl(true);
1075	m_pcCavlcCoder->setMaxAlfCtrlDepth(cAlfCUCtrlParam.alf_max_depth);
1076	}
1077	else
1078	{
1079	m_pcEntropyCoder->setAlfCtrl(false);
1080	}
1081	m_pcEntropyCoder->encodeAlfCtrlParam(cAlfCUCtrlParam, m_pcAdaptiveLoopFilter->getNumCUsInPic());
1082
1083	}
1084	}
1085	}
1086	}
1087	m_pcEntropyCoder->encodeTileMarkerFlag(pcSlice);
1088
1089	// is it needed?
1090	{
1091	if (!bEntropySlice)
1092	{
1093	pcBitstreamRedirect->writeAlignOne();
1094	}
1095	else
1096	{
1097	// We've not completed our slice header info yet, do the alignment later.
1098	}
1099	m_pcSbacCoder->init( (TEncBinIf*)m_pcBinCABAC );
1100	m_pcEntropyCoder->setEntropyCoder ( m_pcSbacCoder, pcSlice );
1101	m_pcEntropyCoder->resetEntropy ();
1102	for ( UInt ui = 0 ; ui < pcSlice->getPPS()->getNumSubstreams() ; ui++ )
1103	{
1104	m_pcEntropyCoder->setEntropyCoder ( &pcSbacCoders[ui], pcSlice );
1105	m_pcEntropyCoder->resetEntropy ();
1106	}
1107	}
1108
1109	if(pcSlice->isNextSlice())
1110	{
1111	// set entropy coder for writing
1112	m_pcSbacCoder->init( (TEncBinIf*)m_pcBinCABAC );
1113	{
1114	for ( UInt ui = 0 ; ui < pcSlice->getPPS()->getNumSubstreams() ; ui++ )
1115	{
1116	m_pcEntropyCoder->setEntropyCoder ( &pcSbacCoders[ui], pcSlice );
1117	m_pcEntropyCoder->resetEntropy ();
1118	}
1119	pcSbacCoders[0].load(m_pcSbacCoder);
1120	m_pcEntropyCoder->setEntropyCoder ( &pcSbacCoders[0], pcSlice ); //ALF is written in substream #0 with CABAC coder #0 (see ALF param encoding below)
1121	}
1122	m_pcEntropyCoder->resetEntropy ();
1123	// File writing
1124	if (!bEntropySlice)
1125	{
1126	m_pcEntropyCoder->setBitstream(pcBitstreamRedirect);
1127	}
1128	else
1129	{
1130	m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1131	}
1132	// for now, override the TILES_DECODER setting in order to write substreams.
1133	m_pcEntropyCoder->setBitstream ( &pcSubstreamsOut[0] );
1134
1135	}
1136	pcSlice->setFinalized(true);
1137
1138	m_pcSbacCoder->load( &pcSbacCoders[0] );
1139
1140	pcSlice->setTileOffstForMultES( uiOneBitstreamPerSliceLength );
1141	if (!bEntropySlice)
1142	{
1143	pcSlice->setTileLocationCount ( 0 );
1144	m_pcSliceEncoder->encodeSlice(pcPic, pcBitstreamRedirect, pcSubstreamsOut); // redirect is only used for CAVLC tile position info.
1145	}
1146	else
1147	{
1148	m_pcSliceEncoder->encodeSlice(pcPic, &nalu.m_Bitstream, pcSubstreamsOut); // nalu.m_Bitstream is only used for CAVLC tile position info.
1149	}
1150
1151	{
1152	// Construct the final bitstream by flushing and concatenating substreams.
1153	// The final bitstream is either nalu.m_Bitstream or pcBitstreamRedirect;
1154	UInt* puiSubstreamSizes = pcSlice->getSubstreamSizes();
1155	UInt uiTotalCodedSize = 0; // for padding calcs.
1156	UInt uiNumSubstreamsPerTile = iNumSubstreams;
1157	if (iNumSubstreams > 1)
1158	{
1159	uiNumSubstreamsPerTile /= pcPic->getPicSym()->getNumTiles();
1160	}
1161	for ( UInt ui = 0 ; ui < iNumSubstreams; ui++ )
1162	{
1163	// Flush all substreams -- this includes empty ones.
1164	// Terminating bit and flush.
1165	m_pcEntropyCoder->setEntropyCoder ( &pcSbacCoders[ui], pcSlice );
1166	m_pcEntropyCoder->setBitstream ( &pcSubstreamsOut[ui] );
1167	m_pcEntropyCoder->encodeTerminatingBit( 1 );
1168	m_pcEntropyCoder->encodeSliceFinish();
1169	pcSubstreamsOut[ui].write( 1, 1 ); // stop bit.
1170	pcSubstreamsOut[ui].writeAlignZero();
1171	// Byte alignment is necessary between tiles when tiles are independent.
1172	uiTotalCodedSize += pcSubstreamsOut[ui].getNumberOfWrittenBits();
1173
1174	{
1175	Bool bNextSubstreamInNewTile = ((ui+1) < iNumSubstreams)
1176	&& ((ui+1)%uiNumSubstreamsPerTile == 0);
1177	if (bNextSubstreamInNewTile)
1178	{
1179	// byte align.
1180	while (uiTotalCodedSize&0x7)
1181	{
1182	pcSubstreamsOut[ui].write(0, 1);
1183	uiTotalCodedSize++;
1184	}
1185	}
1186	Bool bRecordOffsetNext = m_pcCfg->getTileLocationInSliceHeaderFlag()
1187	&& bNextSubstreamInNewTile;
1188	if (bRecordOffsetNext)
1189	pcSlice->setTileLocation(ui/uiNumSubstreamsPerTile, pcSlice->getTileOffstForMultES()+(uiTotalCodedSize>>3));
1190	}
1191	if (ui+1 < pcSlice->getPPS()->getNumSubstreams())
1192	puiSubstreamSizes[ui] = pcSubstreamsOut[ui].getNumberOfWrittenBits();
1193	}
1194	// Complete the slice header info.
1195	m_pcEntropyCoder->setEntropyCoder ( m_pcCavlcCoder, pcSlice );
1196	m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1197	if (m_pcCfg->getTileLocationInSliceHeaderFlag()==0)
1198	{
1199	pcSlice->setTileLocationCount( 0 );
1200	}
1201	m_pcEntropyCoder->encodeTilesWPPEntryPoint( pcSlice );
1202	// Substreams...
1203	TComOutputBitstream *pcOut = pcBitstreamRedirect;
1204	// xWriteTileLocation will perform byte-alignment...
1205	{
1206	if (bEntropySlice)
1207	{
1208	// In these cases, padding is necessary here.
1209	pcOut = &nalu.m_Bitstream;
1210	pcOut->writeAlignOne();
1211	}
1212	}
1213	UInt uiAccumulatedLength = 0;
1214	for ( UInt ui = 0 ; ui < pcSlice->getPPS()->getNumSubstreams(); ui++ )
1215	{
1216	pcOut->addSubstream(&pcSubstreamsOut[ui]);
1217
1218	// Update tile marker location information
1219	for (Int uiMrkIdx = 0; uiMrkIdx < pcSubstreamsOut[ui].getTileMarkerLocationCount(); uiMrkIdx++)
1220	{
1221	UInt uiBottom = pcOut->getTileMarkerLocationCount();
1222	pcOut->setTileMarkerLocation ( uiBottom, uiAccumulatedLength + pcSubstreamsOut[ui].getTileMarkerLocation( uiMrkIdx ) );
1223	pcOut->setTileMarkerLocationCount ( uiBottom + 1 );
1224	}
1225	uiAccumulatedLength = (pcOut->getNumberOfWrittenBits() >> 3);
1226	}
1227	}
1228
1229	UInt uiBoundingAddrSlice, uiBoundingAddrEntropySlice;
1230	uiBoundingAddrSlice = m_uiStoredStartCUAddrForEncodingSlice[uiStartCUAddrSliceIdx];
1231	uiBoundingAddrEntropySlice = m_uiStoredStartCUAddrForEncodingEntropySlice[uiStartCUAddrEntropySliceIdx];
1232	uiNextCUAddr = min(uiBoundingAddrSlice, uiBoundingAddrEntropySlice);
1233	// If current NALU is the first NALU of slice (containing slice header) and more NALUs exist (due to multiple entropy slices) then buffer it.
1234	// If current NALU is the last NALU of slice and a NALU was buffered, then (a) Write current NALU (b) Update an write buffered NALU at approproate location in NALU list.
1235	Bool bNALUAlignedWrittenToList = false; // used to ensure current NALU is not written more than once to the NALU list.
1236	xWriteTileLocationToSliceHeader(nalu, pcBitstreamRedirect, pcSlice);
1237	writeRBSPTrailingBits(nalu.m_Bitstream);
1238	accessUnit.push_back(new NALUnitEBSP(nalu));
1239	bNALUAlignedWrittenToList = true;
1240	uiOneBitstreamPerSliceLength += nalu.m_Bitstream.getNumberOfWrittenBits(); // length of bitstream after byte-alignment
1241
1242	if (!bNALUAlignedWrittenToList)
1243	{
1244	{
1245	nalu.m_Bitstream.writeAlignZero();
1246	}
1247	accessUnit.push_back(new NALUnitEBSP(nalu));
1248	uiOneBitstreamPerSliceLength += nalu.m_Bitstream.getNumberOfWrittenBits() + 24; // length of bitstream after byte-alignment + 3 byte startcode 0x000001
1249	}
1250
1251
1252	processingState = ENCODE_SLICE;
1253	}
1254	break;
1255	case EXECUTE_INLOOPFILTER:
1256	{
1257	TComAPS cAPS;
1258	allocAPS(&cAPS, pcSlice->getSPS());
1259	#if !LGE_SAO_MIGRATION_D0091
1260	cAPS.setSaoInterleavingFlag(m_pcCfg->getSaoInterleavingFlag());
1261	#endif
1262	// set entropy coder for RD
1263	m_pcEntropyCoder->setEntropyCoder ( m_pcCavlcCoder, pcSlice );
1264
1265	if ( pcSlice->getSPS()->getUseSAO() )
1266	{
1267	m_pcEntropyCoder->resetEntropy();
1268	m_pcEntropyCoder->setBitstream( m_pcBitCounter );
1269	#if LGE_SAO_MIGRATION_D0091
1270	m_pcSAO->startSaoEnc(pcPic, m_pcEntropyCoder, m_pcEncTop->getRDSbacCoder(), m_pcEncTop->getRDGoOnSbacCoder());
1271	#else
1272	m_pcSAO->startSaoEnc(pcPic, m_pcEntropyCoder, m_pcEncTop->getRDSbacCoder(), NULL);
1273	#endif
1274	SAOParam& cSaoParam = *(cAPS.getSaoParam());
1275
1276	#if SAO_CHROMA_LAMBDA
1277	#if LGE_SAO_MIGRATION_D0091
1278	#if SAO_ENCODING_CHOICE
1279	m_pcSAO->SAOProcess(&cSaoParam, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma(), pcPic->getSlice(0)->getDepth());
1280	#else
1281	m_pcSAO->SAOProcess(&cSaoParam, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma());
1282	#endif
1283	#else
1284	m_pcSAO->SAOProcess(&cSaoParam, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma());
1285	#endif
1286	#else
1287	#if ALF_CHROMA_LAMBDA
1288	m_pcSAO->SAOProcess(&cSaoParam, pcPic->getSlice(0)->getLambdaLuma());
1289	#else
1290	m_pcSAO->SAOProcess(&cSaoParam, pcPic->getSlice(0)->getLambda());
1291	#endif
1292	#endif
1293	m_pcSAO->endSaoEnc();
1294
1295	m_pcAdaptiveLoopFilter->PCMLFDisableProcess(pcPic);
1296	}
1297
1298	// adaptive loop filter
1299	if ( pcSlice->getSPS()->getUseALF())
1300	{
1301	m_pcEntropyCoder->resetEntropy ();
1302	m_pcEntropyCoder->setBitstream ( m_pcBitCounter );
1303	m_pcAdaptiveLoopFilter->startALFEnc(pcPic, m_pcEntropyCoder );
1304	AlfParamSet* pAlfEncParam = (pcSlice->getSPS()->getUseALFCoefInSlice())?( alfSliceParams ):( cAPS.getAlfParam());
1305	#if ALF_CHROMA_LAMBDA
1306	#if HHI_INTERVIEW_SKIP
1307	m_pcAdaptiveLoopFilter->ALFProcess(pAlfEncParam, alfCUCtrlParam, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma(), m_pcEncTop->getInterViewSkip() );
1308	#else
1309	m_pcAdaptiveLoopFilter->ALFProcess(pAlfEncParam, alfCUCtrlParam, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma() );
1310	#endif
1311	#else
1312	#if SAO_CHROMA_LAMBDA
1313	#if HHI_INTERVIEW_SKIP
1314	m_pcAdaptiveLoopFilter->ALFProcess(pAlfEncParam, alfCUCtrlParam, pcPic->getSlice(0)->getLambdaLuma(), m_pcEncTop->getInterViewSkip());
1315	#else
1316	m_pcAdaptiveLoopFilter->ALFProcess(pAlfEncParam, alfCUCtrlParam, pcPic->getSlice(0)->getLambdaLuma());
1317	#endif
1318	#else
1319	#if HHI_INTERVIEW_SKIP
1320	m_pcAdaptiveLoopFilter->ALFProcess(pAlfEncParam, alfCUCtrlParam, pcPic->getSlice(0)->getLambda(), m_pcEncTop->getInterViewSkip() );
1321	#else
1322	m_pcAdaptiveLoopFilter->ALFProcess(pAlfEncParam, alfCUCtrlParam, pcPic->getSlice(0)->getLambda());
1323	#endif
1324	#endif
1325	#endif
1326
1327	m_pcAdaptiveLoopFilter->endALFEnc();
1328
1329	m_pcAdaptiveLoopFilter->PCMLFDisableProcess(pcPic);
1330	}
1331	iCodedAPSIdx = iCurrAPSIdx;
1332	pcSliceForAPS = pcSlice;
1333
1334	assignNewAPS(cAPS, iCodedAPSIdx, vAPS, pcSliceForAPS);
1335	iCurrAPSIdx = (iCurrAPSIdx +1)%MAX_NUM_SUPPORTED_APS;
1336	processingState = ENCODE_APS;
1337
1338	//set APS link to the slices
1339	for(Int s=0; s< uiNumSlices; s++)
1340	{
1341	if (pcSlice->getSPS()->getUseALF())
1342	{
1343	pcPic->getSlice(s)->setAlfEnabledFlag( (pcSlice->getSPS()->getUseALFCoefInSlice())?(alfSliceParams[s].isEnabled[ALF_Y]):(cAPS.getAlfEnabled()) );
1344	}
1345	if (pcSlice->getSPS()->getUseSAO())
1346	{
1347	pcPic->getSlice(s)->setSaoEnabledFlag((cAPS.getSaoParam()->bSaoFlag[0]==1)?true:false);
1348	}
1349	pcPic->getSlice(s)->setAPS(&(vAPS[iCodedAPSIdx]));
1350	pcPic->getSlice(s)->setAPSId(iCodedAPSIdx);
1351	}
1352	}
1353	break;
1354	case ENCODE_APS:
1355	{
1356	#if VIDYO_VPS_INTEGRATION \| QC_MVHEVC_B0046
1357	OutputNALUnit nalu(NAL_UNIT_APS, true, m_pcEncTop->getLayerId());
1358	#else
1359	OutputNALUnit nalu(NAL_UNIT_APS, true, m_pcEncTop->getViewId(), m_pcEncTop->getIsDepth());
1360	#endif
1361	encodeAPS(&(vAPS[iCodedAPSIdx]), nalu.m_Bitstream, pcSliceForAPS);
1362	accessUnit.push_back(new NALUnitEBSP(nalu));
1363
1364	processingState = ENCODE_SLICE;
1365	}
1366	break;
1367	default:
1368	{
1369	printf("Not a supported encoding state\n");
1370	assert(0);
1371	exit(-1);
1372	}
1373	}
1374	} // end iteration over slices
1375
1376
1377	if(pcSlice->getSPS()->getUseSAO() \|\| pcSlice->getSPS()->getUseALF())
1378	{
1379	if(pcSlice->getSPS()->getUseSAO())
1380	{
1381	m_pcSAO->destroyPicSaoInfo();
1382	}
1383
1384	if(pcSlice->getSPS()->getUseALF())
1385	{
1386	m_pcAdaptiveLoopFilter->uninitALFEnc(alfSliceParams, alfCUCtrlParam);
1387	m_pcAdaptiveLoopFilter->destroyPicAlfInfo();
1388	}
1389
1390	pcPic->destroyNonDBFilterInfo();
1391	}
1392
1393	#if HHI_INTERVIEW_SKIP
1394	if (pcPic->getUsedPelsMap())
1395	pcPic->removeUsedPelsMapBuffer() ;
1396	#endif
1397	#if H3D_IVMP
1398	pcPic->removeOrgDepthMapBuffer();
1399	#endif
1400
1401	// pcPic->compressMotion();
1402	m_pocLastCoded = pcPic->getPOC();
1403
1404	//-- For time output for each slice
1405	Double dEncTime = (double)(clock()-iBeforeTime) / CLOCKS_PER_SEC;
1406
1407	const char* digestStr = NULL;
1408	if (m_pcCfg->getPictureDigestEnabled())
1409	{
1410	/* calculate MD5sum for entire reconstructed picture */
1411	SEIpictureDigest sei_recon_picture_digest;
1412	sei_recon_picture_digest.method = SEIpictureDigest::MD5;
1413	calcMD5(*pcPic->getPicYuvRec(), sei_recon_picture_digest.digest);
1414	digestStr = digestToString(sei_recon_picture_digest.digest);
1415
1416	#if VIDYO_VPS_INTEGRATION \| QC_MVHEVC_B0046
1417	OutputNALUnit nalu(NAL_UNIT_SEI, false, m_pcEncTop->getLayerId());
1418	#else
1419	OutputNALUnit nalu(NAL_UNIT_SEI, false, m_pcEncTop->getViewId(), m_pcEncTop->getIsDepth());
1420	#endif
1421
1422	/* write the SEI messages */
1423	m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1424	m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1425	m_pcEntropyCoder->encodeSEI(sei_recon_picture_digest);
1426	writeRBSPTrailingBits(nalu.m_Bitstream);
1427
1428	/* insert the SEI message NALUnit before any Slice NALUnits */
1429	AccessUnit::iterator it = find_if(accessUnit.begin(), accessUnit.end(), mem_fun(&NALUnit::isSlice));
1430	accessUnit.insert(it, new NALUnitEBSP(nalu));
1431	}
1432
1433	xCalculateAddPSNR( pcPic, pcPic->getPicYuvRec(), accessUnit, dEncTime );
1434	if (digestStr)
1435	printf(" [MD5:%s]", digestStr);
1436
1437	#if FIXED_ROUNDING_FRAME_MEMORY
1438	/* TODO: this should happen after copyToPic(pcPicYuvRecOut) */
1439	pcPic->getPicYuvRec()->xFixedRoundingPic();
1440	#endif
1441	pcPic->getPicYuvRec()->copyToPic(pcPicYuvRecOut);
1442
1443	pcPic->setReconMark ( true );
1444
1445	pcPic->setUsedForTMVP ( true );
1446
1447	m_bFirst = false;
1448	m_iNumPicCoded++;
1449
1450	/* logging: insert a newline at end of picture period */
1451	printf("\n");
1452	fflush(stdout);
1453	}
1454
1455	delete[] pcSubstreamsOut;
1456	delete pcBitstreamRedirect;
1457
1458	}
1459
1460	/** Memory allocation for APS
1461	* \param [out] pAPS APS pointer
1462	* \param [in] pSPS SPS pointer
1463	*/
1464	Void TEncGOP::allocAPS (TComAPS* pAPS, TComSPS* pSPS)
1465	{
1466	if(pSPS->getUseSAO())
1467	{
1468	pAPS->createSaoParam();
1469	m_pcSAO->allocSaoParam(pAPS->getSaoParam());
1470	}
1471	if(pSPS->getUseALF())
1472	{
1473	pAPS->createAlfParam();
1474	//alf Enabled flag in APS is false after pAPS->createAlfParam();
1475	if(!pSPS->getUseALFCoefInSlice())
1476	{
1477	pAPS->getAlfParam()->create(m_pcAdaptiveLoopFilter->getNumLCUInPicWidth(), m_pcAdaptiveLoopFilter->getNumLCUInPicHeight(), m_pcAdaptiveLoopFilter->getNumCUsInPic());
1478	pAPS->getAlfParam()->createALFParam();
1479	}
1480	}
1481	}
1482
1483	/** Memory deallocation for APS
1484	* \param [out] pAPS APS pointer
1485	* \param [in] pSPS SPS pointer
1486	*/
1487	Void TEncGOP::freeAPS (TComAPS* pAPS, TComSPS* pSPS)
1488	{
1489	if(pSPS->getUseSAO())
1490	{
1491	if(pAPS->getSaoParam() != NULL)
1492	{
1493	m_pcSAO->freeSaoParam(pAPS->getSaoParam());
1494	pAPS->destroySaoParam();
1495
1496	}
1497	}
1498	if(pSPS->getUseALF())
1499	{
1500	if(pAPS->getAlfParam() != NULL)
1501	{
1502	if(!pSPS->getUseALFCoefInSlice())
1503	{
1504	pAPS->getAlfParam()->releaseALFParam();
1505	}
1506	pAPS->destroyAlfParam();
1507	}
1508	}
1509	}
1510
1511	/** Assign APS object into APS container according to APS ID
1512	* \param [in] cAPS APS object
1513	* \param [in] apsID APS ID
1514	* \param [in,out] vAPS APS container
1515	* \param [in] pcSlice pointer to slice
1516	*/
1517	Void TEncGOP::assignNewAPS(TComAPS& cAPS, Int apsID, std::vector<TComAPS>& vAPS, TComSlice* pcSlice)
1518	{
1519
1520	cAPS.setAPSID(apsID);
1521	if(pcSlice->getPOC() == 0)
1522	{
1523	cAPS.setScalingListEnabled(pcSlice->getSPS()->getScalingListFlag());
1524	}
1525	else
1526	{
1527	cAPS.setScalingListEnabled(false);
1528	}
1529
1530	cAPS.setSaoEnabled(pcSlice->getSPS()->getUseSAO() ? (cAPS.getSaoParam()->bSaoFlag[0] ):(false));
1531	cAPS.setAlfEnabled(pcSlice->getSPS()->getUseALF() ? (cAPS.getAlfParam()->isEnabled[0]):(false));
1532	cAPS.setLoopFilterOffsetInAPS(m_pcCfg->getLoopFilterOffsetInAPS());
1533	cAPS.setLoopFilterDisable(m_pcCfg->getLoopFilterDisable());
1534	cAPS.setLoopFilterBetaOffset(m_pcCfg->getLoopFilterBetaOffset());
1535	cAPS.setLoopFilterTcOffset(m_pcCfg->getLoopFilterTcOffset());
1536
1537	//assign new APS into APS container
1538	Int apsBufSize= (Int)vAPS.size();
1539
1540	if(apsID >= apsBufSize)
1541	{
1542	vAPS.resize(apsID +1);
1543	}
1544
1545	freeAPS(&(vAPS[apsID]), pcSlice->getSPS());
1546	vAPS[apsID] = cAPS;
1547	}
1548
1549
1550	/** encode APS syntax elements
1551	* \param [in] pcAPS APS pointer
1552	* \param [in, out] APSbs bitstream
1553	* \param [in] pointer to slice (just used for entropy coder initialization)
1554	*/
1555	Void TEncGOP::encodeAPS(TComAPS* pcAPS, TComOutputBitstream& APSbs, TComSlice* pcSlice)
1556	{
1557	m_pcEntropyCoder->setEntropyCoder ( m_pcCavlcCoder, pcSlice);
1558	m_pcEntropyCoder->resetEntropy ();
1559	m_pcEntropyCoder->setBitstream(&APSbs);
1560
1561	m_pcEntropyCoder->encodeAPSInitInfo(pcAPS);
1562	if(pcAPS->getScalingListEnabled())
1563	{
1564	m_pcEntropyCoder->encodeScalingList( pcSlice->getScalingList() );
1565	}
1566	if(pcAPS->getLoopFilterOffsetInAPS())
1567	{
1568	m_pcEntropyCoder->encodeDFParams(pcAPS);
1569	}
1570	#if !LGE_SAO_MIGRATION_D0091
1571	m_pcEntropyCoder->encodeSaoParam(pcAPS);
1572	#endif
1573	m_pcEntropyCoder->encodeAPSAlfFlag( pcAPS->getAlfEnabled()?1:0);
1574	if(pcAPS->getAlfEnabled())
1575	{
1576	m_pcEntropyCoder->encodeAlfParam(pcAPS->getAlfParam());
1577	}
1578
1579	m_pcEntropyCoder->encodeApsExtensionFlag();
1580	//neither SAO and ALF is enabled
1581	writeRBSPTrailingBits(APSbs);
1582	}
1583
1584	Void TEncGOP::preLoopFilterPicAll( TComPic* pcPic, UInt64& ruiDist, UInt64& ruiBits )
1585	{
1586	TComSlice* pcSlice = pcPic->getSlice(pcPic->getCurrSliceIdx());
1587	Bool bCalcDist = false;
1588	m_pcLoopFilter->setCfg(pcSlice->getPPS()->getDeblockingFilterControlPresent(), pcSlice->getLoopFilterDisable(), m_pcCfg->getLoopFilterBetaOffset(), m_pcCfg->getLoopFilterTcOffset(), m_pcCfg->getLFCrossTileBoundaryFlag());
1589	m_pcLoopFilter->loopFilterPic( pcPic );
1590
1591	m_pcEntropyCoder->setEntropyCoder ( m_pcEncTop->getRDGoOnSbacCoder(), pcSlice );
1592	m_pcEntropyCoder->resetEntropy ();
1593	m_pcEntropyCoder->setBitstream ( m_pcBitCounter );
1594	pcSlice = pcPic->getSlice(0);
1595	if(pcSlice->getSPS()->getUseSAO() \|\| pcSlice->getSPS()->getUseALF())
1596	{
1597	pcPic->createNonDBFilterInfo();
1598	}
1599
1600	// Adaptive Loop filter
1601	if( pcSlice->getSPS()->getUseALF() )
1602	{
1603	m_pcAdaptiveLoopFilter->createPicAlfInfo(pcPic);
1604
1605	AlfParamSet* alfParamSet;
1606	std::vector<AlfCUCtrlInfo>* alfCUCtrlParam = NULL;
1607	alfParamSet= new AlfParamSet;
1608	alfParamSet->create( m_pcAdaptiveLoopFilter->getNumLCUInPicWidth(), m_pcAdaptiveLoopFilter->getNumLCUInPicHeight(), m_pcAdaptiveLoopFilter->getNumCUsInPic());
1609	alfParamSet->createALFParam();
1610	m_pcAdaptiveLoopFilter->initALFEnc(false, true, 1, alfParamSet, alfCUCtrlParam);
1611	m_pcAdaptiveLoopFilter->startALFEnc(pcPic, m_pcEntropyCoder);
1612
1613
1614
1615	#if ALF_CHROMA_LAMBDA
1616	#if HHI_INTERVIEW_SKIP
1617	m_pcAdaptiveLoopFilter->ALFProcess(alfParamSet, NULL, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma(), m_pcEncTop->getInterViewSkip() );
1618	#else
1619	m_pcAdaptiveLoopFilter->ALFProcess(alfParamSet, NULL, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma() );
1620	#endif
1621	#else
1622	#if SAO_CHROMA_LAMBDA
1623	m_pcAdaptiveLoopFilter->ALFProcess(alfParamSet, NULL, pcPic->getSlice(0)->getLambdaLuma(), m_pcEncTop->getInterViewSkip());
1624	#if HHI_INTERVIEW_SKIP
1625	#else
1626	m_pcAdaptiveLoopFilter->ALFProcess(alfParamSet, NULL, pcPic->getSlice(0)->getLambdaLuma());
1627	#endif
1628	#else
1629	#if HHI_INTERVIEW_SKIP
1630	m_pcAdaptiveLoopFilter->ALFProcess(alfParamSet, NULL, pcPic->getSlice(0)->getLambda(), m_pcEncTop->getInterViewSkip());
1631	#else
1632	m_pcAdaptiveLoopFilter->ALFProcess(alfParamSet, NULL, pcPic->getSlice(0)->getLambda());
1633	#endif
1634	#endif
1635	#endif
1636
1637	m_pcAdaptiveLoopFilter->endALFEnc();
1638
1639	alfParamSet->releaseALFParam();
1640	delete alfParamSet;
1641	delete alfCUCtrlParam;
1642	m_pcAdaptiveLoopFilter->PCMLFDisableProcess(pcPic);
1643	m_pcAdaptiveLoopFilter->destroyPicAlfInfo();
1644	}
1645	if( pcSlice->getSPS()->getUseSAO() \|\| pcSlice->getSPS()->getUseALF())
1646	{
1647	pcPic->destroyNonDBFilterInfo();
1648	}
1649
1650	m_pcEntropyCoder->resetEntropy ();
1651	ruiBits += m_pcEntropyCoder->getNumberOfWrittenBits();
1652
1653	if (!bCalcDist)
1654	ruiDist = xFindDistortionFrame(pcPic->getPicYuvOrg(), pcPic->getPicYuvRec());
1655	}
1656
1657	// ====================================================================================================================
1658	// Protected member functions
1659	// ====================================================================================================================
1660
1661	Void TEncGOP::xInitGOP( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic>& rcListPic, TComList<TComPicYuv>& rcListPicYuvRecOut )
1662	{
1663	assert( iNumPicRcvd > 0 );
1664	// Exception for the first frame
1665	if ( iPOCLast == 0 )
1666	{
1667	m_iGopSize = 1;
1668	}
1669	else
1670	m_iGopSize = m_pcCfg->getGOPSize();
1671
1672	assert (m_iGopSize > 0);
1673
1674	return;
1675	}
1676
1677	Void TEncGOP::xGetBuffer( TComList<TComPic*>& rcListPic,
1678	TComList<TComPicYuv*>& rcListPicYuvRecOut,
1679	Int iNumPicRcvd,
1680	Int iTimeOffset,
1681	TComPic*& rpcPic,
1682	TComPicYuv*& rpcPicYuvRecOut,
1683	UInt uiPOCCurr )
1684	{
1685	Int i;
1686	// Rec. output
1687	TComList<TComPicYuv*>::iterator iterPicYuvRec = rcListPicYuvRecOut.end();
1688	for ( i = 0; i < iNumPicRcvd - iTimeOffset + 1; i++ )
1689	{
1690	iterPicYuvRec--;
1691	}
1692
1693	rpcPicYuvRecOut = *(iterPicYuvRec);
1694
1695	// Current pic.
1696	TComList<TComPic*>::iterator iterPic = rcListPic.begin();
1697	while (iterPic != rcListPic.end())
1698	{
1699	rpcPic = *(iterPic);
1700	rpcPic->setCurrSliceIdx(0);
1701	if (rpcPic->getPOC() == (Int)uiPOCCurr)
1702	{
1703	break;
1704	}
1705	iterPic++;
1706	}
1707
1708	assert (rpcPic->getPOC() == (Int)uiPOCCurr);
1709
1710	return;
1711	}
1712
1713	UInt64 TEncGOP::xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1)
1714	{
1715	Int x, y;
1716	Pel* pSrc0 = pcPic0 ->getLumaAddr();
1717	Pel* pSrc1 = pcPic1 ->getLumaAddr();
1718	#if IBDI_DISTORTION
1719	Int iShift = g_uiBitIncrement;
1720	Int iOffset = 1<<(g_uiBitIncrement-1);
1721	#else
1722	UInt uiShift = g_uiBitIncrement<<1;
1723	#endif
1724	Int iTemp;
1725
1726	Int iStride = pcPic0->getStride();
1727	Int iWidth = pcPic0->getWidth();
1728	Int iHeight = pcPic0->getHeight();
1729
1730	UInt64 uiTotalDiff = 0;
1731
1732	for( y = 0; y < iHeight; y++ )
1733	{
1734	for( x = 0; x < iWidth; x++ )
1735	{
1736	#if IBDI_DISTORTION
1737	iTemp = ((pSrc0[x]+iOffset)>>iShift) - ((pSrc1[x]+iOffset)>>iShift); uiTotalDiff += iTemp * iTemp;
1738	#else
1739	iTemp = pSrc0[x] - pSrc1[x]; uiTotalDiff += (iTemp*iTemp) >> uiShift;
1740	#endif
1741	}
1742	pSrc0 += iStride;
1743	pSrc1 += iStride;
1744	}
1745
1746	iHeight >>= 1;
1747	iWidth >>= 1;
1748	iStride >>= 1;
1749
1750	pSrc0 = pcPic0->getCbAddr();
1751	pSrc1 = pcPic1->getCbAddr();
1752
1753	for( y = 0; y < iHeight; y++ )
1754	{
1755	for( x = 0; x < iWidth; x++ )
1756	{
1757	#if IBDI_DISTORTION
1758	iTemp = ((pSrc0[x]+iOffset)>>iShift) - ((pSrc1[x]+iOffset)>>iShift); uiTotalDiff += iTemp * iTemp;
1759	#else
1760	iTemp = pSrc0[x] - pSrc1[x]; uiTotalDiff += (iTemp*iTemp) >> uiShift;
1761	#endif
1762	}
1763	pSrc0 += iStride;
1764	pSrc1 += iStride;
1765	}
1766
1767	pSrc0 = pcPic0->getCrAddr();
1768	pSrc1 = pcPic1->getCrAddr();
1769
1770	for( y = 0; y < iHeight; y++ )
1771	{
1772	for( x = 0; x < iWidth; x++ )
1773	{
1774	#if IBDI_DISTORTION
1775	iTemp = ((pSrc0[x]+iOffset)>>iShift) - ((pSrc1[x]+iOffset)>>iShift); uiTotalDiff += iTemp * iTemp;
1776	#else
1777	iTemp = pSrc0[x] - pSrc1[x]; uiTotalDiff += (iTemp*iTemp) >> uiShift;
1778	#endif
1779	}
1780	pSrc0 += iStride;
1781	pSrc1 += iStride;
1782	}
1783
1784	return uiTotalDiff;
1785	}
1786
1787	#if VERBOSE_RATE
1788	static const char* nalUnitTypeToString(NalUnitType type)
1789	{
1790	switch (type)
1791	{
1792	case NAL_UNIT_CODED_SLICE: return "SLICE";
1793	#if !QC_REM_IDV_B0046
1794	case NAL_UNIT_CODED_SLICE_IDV: return "IDV";
1795	#endif
1796	case NAL_UNIT_CODED_SLICE_CRA: return "CRA";
1797	case NAL_UNIT_CODED_SLICE_TLA: return "TLA";
1798	case NAL_UNIT_CODED_SLICE_IDR: return "IDR";
1799	case NAL_UNIT_SEI: return "SEI";
1800	case NAL_UNIT_SPS: return "SPS";
1801	case NAL_UNIT_PPS: return "PPS";
1802	case NAL_UNIT_FILLER_DATA: return "FILLER";
1803	default: return "UNK";
1804	}
1805	}
1806	#endif
1807
1808	Void TEncGOP::xCalculateAddPSNR( TComPic* pcPic, TComPicYuv* pcPicD, const AccessUnit& accessUnit, Double dEncTime )
1809	{
1810	Int x, y;
1811	UInt64 uiSSDY = 0;
1812	UInt64 uiSSDU = 0;
1813	UInt64 uiSSDV = 0;
1814
1815	Double dYPSNR = 0.0;
1816	Double dUPSNR = 0.0;
1817	Double dVPSNR = 0.0;
1818
1819	//===== calculate PSNR =====
1820	Pel* pOrg = pcPic ->getPicYuvOrg()->getLumaAddr();
1821	Pel* pRec = pcPicD->getLumaAddr();
1822	Int iStride = pcPicD->getStride();
1823
1824	Int iWidth;
1825	Int iHeight;
1826
1827	iWidth = pcPicD->getWidth () - m_pcEncTop->getPad(0);
1828	iHeight = pcPicD->getHeight() - m_pcEncTop->getPad(1);
1829
1830	Int iSize = iWidth*iHeight;
1831
1832	for( y = 0; y < iHeight; y++ )
1833	{
1834	for( x = 0; x < iWidth; x++ )
1835	{
1836	Int iDiff = (Int)( pOrg[x] - pRec[x] );
1837	uiSSDY += iDiff * iDiff;
1838	}
1839	pOrg += iStride;
1840	pRec += iStride;
1841	}
1842
1843	#if HHI_VSO
1844	#if HHI_VSO_SYNTH_DIST_OUT
1845	if ( m_pcRdCost->getUseRenModel() )
1846	{
1847	unsigned int maxval = 255 * (1<<(g_uiBitDepth + g_uiBitIncrement -8));
1848	Double fRefValueY = (double) maxval * maxval * iSize;
1849	Double fRefValueC = fRefValueY / 4.0;
1850	TRenModel* pcRenModel = m_pcEncTop->getEncTop()->getRenModel();
1851	Int64 iDistVSOY, iDistVSOU, iDistVSOV;
1852	pcRenModel->getTotalSSE( iDistVSOY, iDistVSOU, iDistVSOV );
1853	dYPSNR = ( iDistVSOY ? 10.0 * log10( fRefValueY / (Double) iDistVSOY ) : 99.99 );
1854	dUPSNR = ( iDistVSOU ? 10.0 * log10( fRefValueC / (Double) iDistVSOU ) : 99.99 );
1855	dVPSNR = ( iDistVSOV ? 10.0 * log10( fRefValueC / (Double) iDistVSOV ) : 99.99 );
1856	}
1857	else
1858	#endif
1859	#endif
1860	{
1861	iHeight >>= 1;
1862	iWidth >>= 1;
1863	iStride >>= 1;
1864	pOrg = pcPic ->getPicYuvOrg()->getCbAddr();
1865	pRec = pcPicD->getCbAddr();
1866
1867	for( y = 0; y < iHeight; y++ )
1868	{
1869	for( x = 0; x < iWidth; x++ )
1870	{
1871	Int iDiff = (Int)( pOrg[x] - pRec[x] );
1872	uiSSDU += iDiff * iDiff;
1873	}
1874	pOrg += iStride;
1875	pRec += iStride;
1876	}
1877
1878	pOrg = pcPic ->getPicYuvOrg()->getCrAddr();
1879	pRec = pcPicD->getCrAddr();
1880
1881	for( y = 0; y < iHeight; y++ )
1882	{
1883	for( x = 0; x < iWidth; x++ )
1884	{
1885	Int iDiff = (Int)( pOrg[x] - pRec[x] );
1886	uiSSDV += iDiff * iDiff;
1887	}
1888	pOrg += iStride;
1889	pRec += iStride;
1890	}
1891
1892	unsigned int maxval = 255 * (1<<(g_uiBitDepth + g_uiBitIncrement -8));
1893	Double fRefValueY = (double) maxval * maxval * iSize;
1894	Double fRefValueC = fRefValueY / 4.0;
1895	dYPSNR = ( uiSSDY ? 10.0 * log10( fRefValueY / (Double)uiSSDY ) : 99.99 );
1896	dUPSNR = ( uiSSDU ? 10.0 * log10( fRefValueC / (Double)uiSSDU ) : 99.99 );
1897	dVPSNR = ( uiSSDV ? 10.0 * log10( fRefValueC / (Double)uiSSDV ) : 99.99 );
1898	}
1899	/* calculate the size of the access unit, excluding:
1900	* - any AnnexB contributions (start_code_prefix, zero_byte, etc.,)
1901	* - SEI NAL units
1902	*/
1903	unsigned numRBSPBytes = 0;
1904	for (AccessUnit::const_iterator it = accessUnit.begin(); it != accessUnit.end(); it++)
1905	{
1906	unsigned numRBSPBytes_nal = unsigned((*it)->m_nalUnitData.str().size());
1907	#if VERBOSE_RATE
1908	printf("*** %6s numBytesInNALunit: %u\n", nalUnitTypeToString((*it)->m_nalUnitType), numRBSPBytes_nal);
1909	#endif
1910	if ((*it)->m_nalUnitType != NAL_UNIT_SEI)
1911	numRBSPBytes += numRBSPBytes_nal;
1912	}
1913
1914	unsigned uibits = numRBSPBytes * 8;
1915	m_vRVM_RP.push_back( uibits );
1916
1917	//===== add PSNR =====
1918	m_pcEncTop->getAnalyzeAll()->addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
1919	TComSlice* pcSlice = pcPic->getSlice(0);
1920	if (pcSlice->isIntra())
1921	{
1922	m_pcEncTop->getAnalyzeI()->addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
1923	}
1924	if (pcSlice->isInterP())
1925	{
1926	m_pcEncTop->getAnalyzeP()->addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
1927	}
1928	if (pcSlice->isInterB())
1929	{
1930	m_pcEncTop->getAnalyzeB()->addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
1931	}
1932
1933	Char c = (pcSlice->isIntra() ? 'I' : pcSlice->isInterP() ? 'P' : 'B');
1934	if (!pcSlice->isReferenced()) c += 32;
1935
1936	#if ADAPTIVE_QP_SELECTION
1937	printf("%s View %3d POC %4d TId: %1d ( %c-SLICE, nQP %d QP %d ) %10d bits",
1938	pcSlice->getIsDepth() ? "Depth " : "Texture",
1939	pcSlice->getViewId(),
1940	pcSlice->getPOC(),
1941	pcSlice->getTLayer(),
1942	c,
1943	pcSlice->getSliceQpBase(),
1944	pcSlice->getSliceQp(),
1945	uibits );
1946	#else
1947	printf("%s View %3d POC %4d TId: %1d ( %c-SLICE, QP %d ) %10d bits",
1948	pcSlice->getIsDepth() ? "Depth " : "Texture",
1949	pcSlice->getViewId(),
1950	pcSlice->getPOC()-pcSlice->getLastIDR(),
1951	pcSlice->getTLayer(),
1952	c,
1953	pcSlice->getSliceQp(),
1954	uibits );
1955	#endif
1956
1957	printf(" [Y %6.4lf dB U %6.4lf dB V %6.4lf dB]", dYPSNR, dUPSNR, dVPSNR );
1958	printf(" [ET %5.0f ]", dEncTime );
1959
1960	for (Int iRefList = 0; iRefList < 2; iRefList++)
1961	{
1962	printf(" [L%d ", iRefList);
1963	for (Int iRefIndex = 0; iRefIndex < pcSlice->getNumRefIdx(RefPicList(iRefList)); iRefIndex++)
1964	{
1965	if( pcSlice->getViewId() != pcSlice->getRefViewId( RefPicList(iRefList), iRefIndex ) )
1966	{
1967	printf( "V%d ", pcSlice->getRefViewId( RefPicList(iRefList), iRefIndex ) );
1968	}
1969	else
1970	{
1971	printf ("%d ", pcSlice->getRefPOC(RefPicList(iRefList), iRefIndex)-pcSlice->getLastIDR());
1972	}
1973	}
1974	printf("]");
1975	}
1976	if(pcSlice->getNumRefIdx(REF_PIC_LIST_C)>0 && !pcSlice->getNoBackPredFlag())
1977	{
1978	printf(" [LC ");
1979	for (Int iRefIndex = 0; iRefIndex < pcSlice->getNumRefIdx(REF_PIC_LIST_C); iRefIndex++)
1980	{
1981	if( pcSlice->getViewId() != pcSlice->getRefViewId( (RefPicList)pcSlice->getListIdFromIdxOfLC(iRefIndex), pcSlice->getRefIdxFromIdxOfLC(iRefIndex) ) )
1982	{
1983	printf( "V%d ", pcSlice->getRefViewId( (RefPicList)pcSlice->getListIdFromIdxOfLC(iRefIndex), pcSlice->getRefIdxFromIdxOfLC(iRefIndex) ) );
1984	}
1985	else
1986	{
1987	printf ("%d ", pcSlice->getRefPOC((RefPicList)pcSlice->getListIdFromIdxOfLC(iRefIndex), pcSlice->getRefIdxFromIdxOfLC(iRefIndex))-pcSlice->getLastIDR());
1988	}
1989	}
1990	printf("]");
1991	}
1992	}
1993
1994	/** Function for deciding the nal_unit_type.
1995	* \param uiPOCCurr POC of the current picture
1996	* \returns the nal_unit type of the picture
1997	* This function checks the configuration and returns the appropriate nal_unit_type for the picture.
1998	*/
1999	NalUnitType TEncGOP::getNalUnitType(UInt uiPOCCurr)
2000	{
2001	Bool bInterViewOnlySlice = ( m_pcCfg->getGOPEntry(MAX_GOP).m_POC == 0 && (m_pcCfg->getGOPEntry(MAX_GOP).m_sliceType == 'P' \|\| m_pcCfg->getGOPEntry(MAX_GOP).m_sliceType == 'B') );
2002
2003	if (uiPOCCurr == 0)
2004	{
2005	if( bInterViewOnlySlice )
2006	{
2007	#if !QC_REM_IDV_B0046
2008	return NAL_UNIT_CODED_SLICE_IDV;
2009	#else
2010	return NAL_UNIT_CODED_SLICE_IDR;
2011	#endif
2012	}
2013	else
2014	{
2015	return NAL_UNIT_CODED_SLICE_IDR;
2016	}
2017	}
2018	if (uiPOCCurr % m_pcCfg->getIntraPeriod() == 0)
2019	{
2020	if (m_pcCfg->getDecodingRefreshType() == 1)
2021	{
2022	if( bInterViewOnlySlice )
2023	{
2024	#if !QC_REM_IDV_B0046
2025	return NAL_UNIT_CODED_SLICE_IDV;
2026	#else
2027	return NAL_UNIT_CODED_SLICE_CRA;
2028	#endif
2029	}
2030	else
2031	{
2032	return NAL_UNIT_CODED_SLICE_CRA;
2033	}
2034	}
2035	else if (m_pcCfg->getDecodingRefreshType() == 2)
2036	{
2037	if( bInterViewOnlySlice )
2038	{
2039	#if !QC_REM_IDV_B0046
2040	return NAL_UNIT_CODED_SLICE_IDV;
2041	#else
2042	return NAL_UNIT_CODED_SLICE_IDR;
2043	#endif
2044	}
2045	else
2046	{
2047	return NAL_UNIT_CODED_SLICE_IDR;
2048	}
2049	}
2050	}
2051	return NAL_UNIT_CODED_SLICE;
2052	}
2053
2054	NalUnitType TEncGOP::getNalUnitTypeBaseViewMvc(UInt uiPOCCurr)
2055	{
2056	if( uiPOCCurr == 0 )
2057	{
2058	return NAL_UNIT_CODED_SLICE_IDR;
2059	}
2060	if( uiPOCCurr % m_pcCfg->getIntraPeriod() == 0 )
2061	{
2062	if( m_pcCfg->getDecodingRefreshType() == 1 )
2063	{
2064	return NAL_UNIT_CODED_SLICE_CRA;
2065	}
2066	else if( m_pcCfg->getDecodingRefreshType() == 2 )
2067	{
2068	return NAL_UNIT_CODED_SLICE_IDR;
2069	}
2070	}
2071	return NAL_UNIT_CODED_SLICE;
2072	}
2073
2074	Double TEncGOP::xCalculateRVM()
2075	{
2076	Double dRVM = 0;
2077
2078	if( m_pcCfg->getGOPSize() == 1 && m_pcCfg->getIntraPeriod() != 1 && m_pcCfg->getFrameToBeEncoded() > RVM_VCEGAM10_M * 2 )
2079	{
2080	// calculate RVM only for lowdelay configurations
2081	std::vector<Double> vRL , vB;
2082	size_t N = m_vRVM_RP.size();
2083	vRL.resize( N );
2084	vB.resize( N );
2085
2086	Int i;
2087	Double dRavg = 0 , dBavg = 0;
2088	vB[RVM_VCEGAM10_M] = 0;
2089	for( i = RVM_VCEGAM10_M + 1 ; i < N - RVM_VCEGAM10_M + 1 ; i++ )
2090	{
2091	vRL[i] = 0;
2092	for( Int j = i - RVM_VCEGAM10_M ; j <= i + RVM_VCEGAM10_M - 1 ; j++ )
2093	vRL[i] += m_vRVM_RP[j];
2094	vRL[i] /= ( 2 * RVM_VCEGAM10_M );
2095	vB[i] = vB[i-1] + m_vRVM_RP[i] - vRL[i];
2096	dRavg += m_vRVM_RP[i];
2097	dBavg += vB[i];
2098	}
2099
2100	dRavg /= ( N - 2 * RVM_VCEGAM10_M );
2101	dBavg /= ( N - 2 * RVM_VCEGAM10_M );
2102
2103	double dSigamB = 0;
2104	for( i = RVM_VCEGAM10_M + 1 ; i < N - RVM_VCEGAM10_M + 1 ; i++ )
2105	{
2106	Double tmp = vB[i] - dBavg;
2107	dSigamB += tmp * tmp;
2108	}
2109	dSigamB = sqrt( dSigamB / ( N - 2 * RVM_VCEGAM10_M ) );
2110
2111	double f = sqrt( 12.0 * ( RVM_VCEGAM10_M - 1 ) / ( RVM_VCEGAM10_M + 1 ) );
2112
2113	dRVM = dSigamB / dRavg * f;
2114	}
2115
2116	return( dRVM );
2117	}
2118
2119	/** Determine the difference between consecutive tile sizes (in bytes) and writes it to bistream rNalu [slice header]
2120	* \param rpcBitstreamRedirect contains the bitstream to be concatenated to rNalu. rpcBitstreamRedirect contains slice payload. rpcSlice contains tile location information.
2121	* \returns Updates rNalu to contain concatenated bitstream. rpcBitstreamRedirect is cleared at the end of this function call.
2122	*/
2123	Void TEncGOP::xWriteTileLocationToSliceHeader (OutputNALUnit& rNalu, TComOutputBitstream& rpcBitstreamRedirect, TComSlice& rpcSlice)
2124	{
2125	{
2126	}
2127
2128	// Byte-align
2129	rNalu.m_Bitstream.writeAlignOne();
2130
2131	// Update tile marker locations
2132	TComOutputBitstream *pcOut = &rNalu.m_Bitstream;
2133	UInt uiAccumulatedLength = pcOut->getNumberOfWrittenBits() >> 3;
2134	for (Int uiMrkIdx = 0; uiMrkIdx < rpcBitstreamRedirect->getTileMarkerLocationCount(); uiMrkIdx++)
2135	{
2136	UInt uiBottom = pcOut->getTileMarkerLocationCount();
2137	pcOut->setTileMarkerLocation ( uiBottom, uiAccumulatedLength + rpcBitstreamRedirect->getTileMarkerLocation( uiMrkIdx ) );
2138	pcOut->setTileMarkerLocationCount ( uiBottom + 1 );
2139	}
2140
2141	// Perform bitstream concatenation
2142	if (rpcBitstreamRedirect->getNumberOfWrittenBits() > 0)
2143	{
2144	UInt uiBitCount = rpcBitstreamRedirect->getNumberOfWrittenBits();
2145	if (rpcBitstreamRedirect->getByteStreamLength()>0)
2146	{
2147	UChar pucStart = reinterpret_cast<UChar>(rpcBitstreamRedirect->getByteStream());
2148	UInt uiWriteByteCount = 0;
2149	while (uiWriteByteCount < (uiBitCount >> 3) )
2150	{
2151	UInt uiBits = (*pucStart);
2152	rNalu.m_Bitstream.write(uiBits, 8);
2153	pucStart++;
2154	uiWriteByteCount++;
2155	}
2156	}
2157	UInt uiBitsHeld = (uiBitCount & 0x07);
2158	for (UInt uiIdx=0; uiIdx < uiBitsHeld; uiIdx++)
2159	{
2160	rNalu.m_Bitstream.write((rpcBitstreamRedirect->getHeldBits() & (1 << (7-uiIdx))) >> (7-uiIdx), 1);
2161	}
2162	}
2163
2164	m_pcEntropyCoder->setBitstream(&rNalu.m_Bitstream);
2165
2166	delete rpcBitstreamRedirect;
2167	rpcBitstreamRedirect = new TComOutputBitstream;
2168	}
2169
2170	Void TEncGOP::xSetRefPicListModificationsMvc( TComSlice* pcSlice, UInt uiPOCCurr, UInt iGOPid )
2171	{
2172	if( pcSlice->getSliceType() == I_SLICE \|\| !(pcSlice->getSPS()->getListsModificationPresentFlag()) \|\| pcSlice->getSPS()->getNumberOfUsableInterViewRefs() == 0 )
2173	{
2174	return;
2175	}
2176
2177	// analyze inter-view modifications
2178	#if !QC_REM_IDV_B0046
2179	GOPEntryMvc gem = m_pcCfg->getGOPEntry( (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDV) ? MAX_GOP : iGOPid );
2180	#else
2181	Bool bRAP = ((getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_IDR) \|\| (getNalUnitType(uiPOCCurr) == NAL_UNIT_CODED_SLICE_CRA)) && (pcSlice->getSPS()->getViewId()) ? 1:0;
2182	GOPEntryMvc gem = m_pcCfg->getGOPEntry( bRAP ? MAX_GOP : iGOPid );
2183	#endif
2184	Int numL0Modifications = 0;
2185	Int numL1Modifications = 0;
2186	for( Int k = 0; k < gem.m_numInterViewRefPics; k++ )
2187	{
2188	if( gem.m_interViewRefPosL0[k] > 0 ) { numL0Modifications++; }
2189	if( gem.m_interViewRefPosL1[k] > 0 ) { numL1Modifications++; }
2190	}
2191
2192	TComRefPicListModification* refPicListModification = pcSlice->getRefPicListModification();
2193	Int maxRefListSize = pcSlice->getNumPocTotalCurrMvc();
2194	Int numTemporalRefs = pcSlice->getNumPocTotalCurr();
2195
2196	// set L0 inter-view modifications
2197	if( (maxRefListSize > 1) && (numL0Modifications > 0) )
2198	{
2199	refPicListModification->setRefPicListModificationFlagL0( true );
2200	Int tempListEntryL0[16];
2201	for( Int k = 0; k < 16; k++ ) { tempListEntryL0[k] = -1; }
2202
2203	Bool hasModification = false;
2204	for( Int k = 0; k < gem.m_numInterViewRefPics; k++ )
2205	{
2206	if( gem.m_interViewRefPosL0[k] > 0 )
2207	{
2208	for( Int l = 0; l < pcSlice->getSPS()->getNumberOfUsableInterViewRefs(); l++ )
2209	{
2210	if( gem.m_interViewRefs[k] == pcSlice->getSPS()->getUsableInterViewRef( l ) && (gem.m_interViewRefPosL0[k] - 1) != (numTemporalRefs + l) )
2211	{
2212	tempListEntryL0[gem.m_interViewRefPosL0[k]-1] = numTemporalRefs + l;
2213	hasModification = true;
2214	}
2215	}
2216	}
2217	}
2218
2219	if( hasModification )
2220	{
2221	Int temporalRefIdx = 0;
2222	for( Int i = 0; i < pcSlice->getNumRefIdx( REF_PIC_LIST_0 ); i++ )
2223	{
2224	if( tempListEntryL0[i] >= 0 )
2225	{
2226	refPicListModification->setRefPicSetIdxL0( i, tempListEntryL0[i] );
2227	}
2228	else
2229	{
2230	refPicListModification->setRefPicSetIdxL0( i, temporalRefIdx );
2231	temporalRefIdx++;
2232	}
2233	}
2234	}
2235	else
2236	{
2237	refPicListModification->setRefPicListModificationFlagL0( false );
2238	}
2239	}
2240
2241	// set L1 inter-view modifications
2242	if( (maxRefListSize > 1) && (numL1Modifications > 0) )
2243	{
2244	refPicListModification->setRefPicListModificationFlagL1( true );
2245	Int tempListEntryL1[16];
2246	for( Int k = 0; k < 16; k++ ) { tempListEntryL1[k] = -1; }
2247
2248	Bool hasModification = false;
2249	for( Int k = 0; k < gem.m_numInterViewRefPics; k++ )
2250	{
2251	if( gem.m_interViewRefPosL1[k] > 0 )
2252	{
2253	for( Int l = 0; l < pcSlice->getSPS()->getNumberOfUsableInterViewRefs(); l++ )
2254	{
2255	if( gem.m_interViewRefs[k] == pcSlice->getSPS()->getUsableInterViewRef( l ) && (gem.m_interViewRefPosL1[k] - 1) != (numTemporalRefs + l) )
2256	{
2257	tempListEntryL1[gem.m_interViewRefPosL1[k]-1] = numTemporalRefs + l;
2258	hasModification = true;
2259	}
2260	}
2261	}
2262	}
2263
2264	if( hasModification )
2265	{
2266	Int temporalRefIdx = 0;
2267	for( Int i = 0; i < pcSlice->getNumRefIdx( REF_PIC_LIST_1 ); i++ )
2268	{
2269	if( tempListEntryL1[i] >= 0 )
2270	{
2271	refPicListModification->setRefPicSetIdxL1( i, tempListEntryL1[i] );
2272	}
2273	else
2274	{
2275	refPicListModification->setRefPicSetIdxL1( i, temporalRefIdx );
2276	temporalRefIdx++;
2277	}
2278	}
2279	}
2280	else
2281	{
2282	refPicListModification->setRefPicListModificationFlagL1( false );
2283	}
2284	}
2285
2286	return;
2287	}
2288	//! \}

Note: See TracBrowser for help on using the repository browser.

JCT-3V 3D-HEVC

Context navigation

source: 3DVCSoftware/branches/HTM-6.2-dev1-LG/source/Lib/TLibEncoder/TEncGOP.cpp @ 494

Download in other formats: