Context navigation

source: 3DVCSoftware/branches/HTM-3.0-Qualcomm/source/App/TAppRenderer/TAppRendererTop.cpp

Visit:

Last change on this file was 56, checked in by hschwarz, 14 years ago
updated trunk (move to HM6.1)
Property svn:eol-style set to `native`
File size: 34.4 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	#include <list>
37	#include <stdio.h>
38	#include <fcntl.h>
39	#include <assert.h>
40	#include <math.h>
41
42	#include "TAppRendererTop.h"
43
44	// ====================================================================================================================
45	// Constructor / destructor / initialization / destroy
46	// ====================================================================================================================
47
48	TAppRendererTop::TAppRendererTop()
49	{
50
51	}
52
53	TAppRendererTop::~TAppRendererTop()
54	{
55
56	}
57
58
59	Void TAppRendererTop::xCreateLib()
60	{
61	Int iInteralBitDepth = g_uiBitDepth + g_uiBitIncrement;
62	Int iFileBitDepth = 8;
63	m_pcRenTop = new TRenTop();
64
65	for(Int iViewIdx=0; iViewIdx<m_iNumberOfInputViews; iViewIdx++)
66	{
67	TVideoIOYuv* pcVideoInput = new TVideoIOYuv;
68	TVideoIOYuv* pcDepthInput = new TVideoIOYuv;
69
70	pcVideoInput->open( m_pchVideoInputFileList[iViewIdx], false, iFileBitDepth, iInteralBitDepth ); // read mode
71	pcDepthInput->open( m_pchDepthInputFileList[iViewIdx], false, iFileBitDepth, iInteralBitDepth ); // read mode
72
73	m_apcTVideoIOYuvVideoInput.push_back( pcVideoInput );
74	m_apcTVideoIOYuvDepthInput.push_back( pcDepthInput );
75	}
76
77	for(Int iViewIdx=0; iViewIdx<m_iNumberOfOutputViews; iViewIdx++)
78	{
79	TVideoIOYuv* pcSynthOutput = new TVideoIOYuv;
80	pcSynthOutput->open( m_pchSynthOutputFileList[iViewIdx], true, iFileBitDepth, iInteralBitDepth ); // write mode
81	m_apcTVideoIOYuvSynthOutput.push_back( pcSynthOutput );
82	}
83	}
84
85
86	Void TAppRendererTop::xDestroyLib()
87	{
88	delete m_pcRenTop;
89
90	for ( Int iViewIdx = 0; iViewIdx < m_iNumberOfInputViews; iViewIdx++ )
91	{
92	m_apcTVideoIOYuvVideoInput[iViewIdx]->close();
93	m_apcTVideoIOYuvDepthInput[iViewIdx]->close();
94
95	delete m_apcTVideoIOYuvDepthInput[iViewIdx];
96	delete m_apcTVideoIOYuvVideoInput[iViewIdx];
97	};
98
99	for ( Int iViewIdx = 0; iViewIdx < m_iNumberOfOutputViews; iViewIdx++ )
100	{
101	m_apcTVideoIOYuvSynthOutput[iViewIdx]->close();
102	delete m_apcTVideoIOYuvSynthOutput[iViewIdx];
103	};
104	}
105
106	Void TAppRendererTop::xInitLib()
107	{
108	m_pcRenTop->init(
109	m_iSourceWidth,
110	m_iSourceHeight,
111	(m_iRenderDirection != 0),
112	m_iLog2SamplingFactor,
113	m_iLog2SamplingFactor+m_iShiftPrecision,
114	m_bUVUp,
115	m_iPreProcMode,
116	m_iPreFilterSize,
117	m_iBlendMode,
118	m_iBlendZThresPerc,
119	m_bBlendUseDistWeight,
120	m_iBlendHoleMargin,
121	m_iInterpolationMode,
122	m_iHoleFillingMode,
123	m_iPostProcMode,
124	m_iUsedPelMapMarExt
125	);
126	}
127
128	// ====================================================================================================================
129	// Public member functions
130	// ====================================================================================================================
131
132
133
134	Void TAppRendererTop::render()
135	{
136	xCreateLib();
137	xInitLib();
138
139	// Create Buffers Input Views;
140	std::vector<TComPicYuv*> apcPicYuvBaseVideo;
141	std::vector<TComPicYuv*> apcPicYuvBaseDepth;
142
143	// TemporalImprovement Filter
144	std::vector<TComPicYuv*> apcPicYuvLastBaseVideo;
145	std::vector<TComPicYuv*> apcPicYuvLastBaseDepth;
146
147	Int aiPad[2] = { 0, 0 };
148
149	for ( UInt uiBaseView = 0; uiBaseView < m_iNumberOfInputViews; uiBaseView++ )
150	{
151	TComPicYuv* pcNewVideoPic = new TComPicYuv;
152	TComPicYuv* pcNewDepthPic = new TComPicYuv;
153
154	pcNewVideoPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
155	apcPicYuvBaseVideo.push_back(pcNewVideoPic);
156
157	pcNewDepthPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
158	apcPicYuvBaseDepth.push_back(pcNewDepthPic);
159
160	//Temporal improvement Filter
161	if ( m_bTempDepthFilter )
162	{
163	pcNewVideoPic = new TComPicYuv;
164	pcNewDepthPic = new TComPicYuv;
165
166	pcNewVideoPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
167	apcPicYuvLastBaseVideo.push_back(pcNewVideoPic);
168
169	pcNewDepthPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
170	apcPicYuvLastBaseDepth.push_back(pcNewDepthPic);
171	}
172	}
173
174	// Create Buffer for synthesized View
175	TComPicYuv* pcPicYuvSynthOut = new TComPicYuv;
176	pcPicYuvSynthOut->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
177
178	Bool bAnyEOS = false;
179
180	Int iNumOfRenderedFrames = 0;
181	Int iFrame = 0;
182
183	while ( ( ( iNumOfRenderedFrames < m_iFramesToBeRendered ) \|\| ( m_iFramesToBeRendered == 0 ) ) && !bAnyEOS )
184	{
185
186	// read in depth and video
187	for(Int iBaseViewIdx=0; iBaseViewIdx < m_iNumberOfInputViews; iBaseViewIdx++ )
188	{
189	m_apcTVideoIOYuvVideoInput[iBaseViewIdx]->read( apcPicYuvBaseVideo[iBaseViewIdx], aiPad ) ;
190
191	apcPicYuvBaseVideo[iBaseViewIdx]->extendPicBorder();
192
193	bAnyEOS \|= m_apcTVideoIOYuvVideoInput[iBaseViewIdx]->isEof();
194
195	m_apcTVideoIOYuvDepthInput[iBaseViewIdx]->read( apcPicYuvBaseDepth[iBaseViewIdx], aiPad ) ;
196	apcPicYuvBaseDepth[iBaseViewIdx]->extendPicBorder();
197	bAnyEOS \|= m_apcTVideoIOYuvDepthInput[iBaseViewIdx]->isEof();
198
199	if ( m_bTempDepthFilter && (iFrame >= m_iFrameSkip) )
200	{
201	m_pcRenTop->temporalFilterVSRS( apcPicYuvBaseVideo[iBaseViewIdx], apcPicYuvBaseDepth[iBaseViewIdx], apcPicYuvLastBaseVideo[iBaseViewIdx], apcPicYuvLastBaseDepth[iBaseViewIdx], ( iFrame == m_iFrameSkip) );
202	}
203	}
204
205	if ( iFrame < m_iFrameSkip ) // Skip Frames
206	{
207	std::cout << "Skipping Frame " << iFrame << std::endl;
208
209	iFrame++;
210	continue;
211	}
212
213	m_cCameraData.update( (UInt)iFrame - m_iFrameSkip );
214
215	for(Int iSynthViewIdx=0; iSynthViewIdx < m_iNumberOfOutputViews; iSynthViewIdx++ )
216	{
217	Int iLeftBaseViewIdx = -1;
218	Int iRightBaseViewIdx = -1;
219
220	Bool bIsBaseView = false;
221
222	Int iRelDistToLeft;
223	Bool bHasLRView = m_cCameraData.getLeftRightBaseView( iSynthViewIdx, iLeftBaseViewIdx, iRightBaseViewIdx, iRelDistToLeft, bIsBaseView );
224	Bool bHasLView = ( iLeftBaseViewIdx != -1 );
225	Bool bHasRView = ( iRightBaseViewIdx != -1 );
226	Bool bRender = true;
227
228	Int iBlendMode = m_iBlendMode;
229	Int iSimEnhBaseView = 0;
230
231	switch( m_iRenderDirection )
232	{
233	/// INTERPOLATION
234	case 0:
235	AOF( bHasLRView \|\| bIsBaseView );
236
237	if ( !bHasLRView && bIsBaseView && m_iBlendMode == 0 )
238	{
239	bRender = false;
240	}
241	else
242	{
243	if ( bIsBaseView )
244	{
245	AOF( iLeftBaseViewIdx == iRightBaseViewIdx );
246	Int iSortedBaseViewIdx = m_cCameraData.getBaseId2SortedId() [iLeftBaseViewIdx];
247
248	if ( m_iBlendMode == 1 )
249	{
250	if ( iSortedBaseViewIdx - 1 >= 0 )
251	{
252	iLeftBaseViewIdx = m_cCameraData.getBaseSortedId2Id()[ iSortedBaseViewIdx - 1];
253	bRender = true;
254	}
255	else
256	{
257	bRender = false;
258	}
259	}
260	else if ( m_iBlendMode == 2 )
261	{
262	if ( iSortedBaseViewIdx + 1 < m_iNumberOfInputViews )
263	{
264	iRightBaseViewIdx = m_cCameraData.getBaseSortedId2Id()[ iSortedBaseViewIdx + 1];
265	bRender = true;
266	}
267	else
268	{
269	bRender = false;
270	}
271	}
272	}
273
274	if ( m_iBlendMode == 3 )
275	{
276	if ( bIsBaseView && (iLeftBaseViewIdx == 0) )
277	{
278	bRender = false;
279	}
280	else
281	{
282	Int iDistLeft = abs( m_cCameraData.getBaseId2SortedId()[0] - m_cCameraData.getBaseId2SortedId() [iLeftBaseViewIdx ] );
283	Int iDistRight = abs( m_cCameraData.getBaseId2SortedId()[0] - m_cCameraData.getBaseId2SortedId() [iRightBaseViewIdx] );
284
285	Int iFillViewIdx = iDistLeft > iDistRight ? iLeftBaseViewIdx : iRightBaseViewIdx;
286
287	if( m_cCameraData.getBaseId2SortedId()[0] < m_cCameraData.getBaseId2SortedId() [iFillViewIdx] )
288	{
289	iBlendMode = 1;
290	iLeftBaseViewIdx = 0;
291	iRightBaseViewIdx = iFillViewIdx;
292	}
293	else
294	{
295	iBlendMode = 2;
296	iLeftBaseViewIdx = iFillViewIdx;
297	iRightBaseViewIdx = 0;
298	}
299
300	}
301	}
302	else
303	{
304	iBlendMode = m_iBlendMode;
305	}
306	}
307
308	if ( m_bSimEnhance )
309	{
310	if ( m_iNumberOfInputViews == 3 && m_cCameraData.getRelSynthViewNumbers()[ iSynthViewIdx ] < VIEW_NUM_PREC )
311	{
312	iSimEnhBaseView = 2; // Take middle view
313	}
314	else
315	{
316	iSimEnhBaseView = 1; // Take left view
317	}
318	}
319
320	if ( bRender )
321	{
322	std::cout << "Rendering Frame " << iFrame
323	<< " of View " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx ] / VIEW_NUM_PREC
324	<< " Left BaseView: " << (Double) m_cCameraData.getBaseViewNumbers() [iLeftBaseViewIdx ] / VIEW_NUM_PREC
325	<< " Right BaseView: " << (Double) m_cCameraData.getBaseViewNumbers() [iRightBaseViewIdx] / VIEW_NUM_PREC
326	<< " BlendMode: " << iBlendMode
327	<< std::endl;
328
329	m_pcRenTop->setShiftLUTs(
330	m_cCameraData.getSynthViewShiftLUTD()[iLeftBaseViewIdx ][iSynthViewIdx],
331	m_cCameraData.getSynthViewShiftLUTI()[iLeftBaseViewIdx ][iSynthViewIdx],
332	m_cCameraData.getBaseViewShiftLUTI ()[iLeftBaseViewIdx ][iRightBaseViewIdx],
333	m_cCameraData.getSynthViewShiftLUTD()[iRightBaseViewIdx][iSynthViewIdx],
334	m_cCameraData.getSynthViewShiftLUTI()[iRightBaseViewIdx][iSynthViewIdx],
335	m_cCameraData.getBaseViewShiftLUTI ()[iRightBaseViewIdx][iLeftBaseViewIdx ],
336
337	iRelDistToLeft
338	);
339
340	m_pcRenTop->interpolateView(
341	apcPicYuvBaseVideo[iLeftBaseViewIdx ],
342	apcPicYuvBaseDepth[iLeftBaseViewIdx ],
343	apcPicYuvBaseVideo[iRightBaseViewIdx],
344	apcPicYuvBaseDepth[iRightBaseViewIdx],
345	pcPicYuvSynthOut,
346	iBlendMode,
347	iSimEnhBaseView
348	);
349	}
350	else
351	{
352	AOT(iLeftBaseViewIdx != iRightBaseViewIdx );
353	apcPicYuvBaseVideo[iLeftBaseViewIdx]->copyToPic( pcPicYuvSynthOut );
354	std::cout << "Copied Frame " << iFrame
355	<< " of View " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC
356	<< " (BaseView) " << std::endl;
357	}
358
359	break;
360	/// EXTRAPOLATION FROM LEFT
361	case 1:
362	if ( !bHasLView ) // View to render is BaseView
363	{
364	bRender = false;
365	}
366
367	if ( bIsBaseView )
368	{
369	AOF( iLeftBaseViewIdx == iRightBaseViewIdx );
370	Int iSortedBaseViewIdx = m_cCameraData.getBaseId2SortedId() [iLeftBaseViewIdx];
371	if ( iSortedBaseViewIdx - 1 >= 0 )
372	{
373	iLeftBaseViewIdx = m_cCameraData.getBaseSortedId2Id()[ iSortedBaseViewIdx - 1];
374	}
375	else
376	{
377	std::cout << "Copied Frame " << iFrame << " of BaseView " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
378	apcPicYuvBaseVideo[iLeftBaseViewIdx]->copyToPic( pcPicYuvSynthOut ); // Copy Original
379	bRender = false;
380	}
381	}
382
383
384	if (bRender)
385	{
386	std::cout << "Rendering Frame " << iFrame << " of View " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
387	m_pcRenTop->setShiftLUTs( m_cCameraData.getSynthViewShiftLUTD()[iLeftBaseViewIdx ][iSynthViewIdx],
388	m_cCameraData.getSynthViewShiftLUTI()[iLeftBaseViewIdx ][iSynthViewIdx], NULL, NULL, NULL, NULL, -1 );
389	m_pcRenTop->extrapolateView( apcPicYuvBaseVideo[iLeftBaseViewIdx ], apcPicYuvBaseDepth[iLeftBaseViewIdx ], pcPicYuvSynthOut, true );
390	}
391	break;
392	/// EXTRAPOLATION FROM RIGHT
393	case 2: // extrapolation from right
394	if ( !bHasRView ) // View to render is BaseView
395	{
396	bRender = false;
397	}
398
399	if ( bIsBaseView )
400	{
401
402	AOF( iLeftBaseViewIdx == iRightBaseViewIdx );
403	Int iSortedBaseViewIdx = m_cCameraData.getBaseId2SortedId() [iLeftBaseViewIdx];
404	if ( iSortedBaseViewIdx + 1 < m_iNumberOfInputViews )
405	{
406	iRightBaseViewIdx = m_cCameraData.getBaseSortedId2Id()[ iSortedBaseViewIdx + 1];
407	}
408	else
409	{
410	std::cout << "Copied Frame " << iFrame << " of BaseView " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
411	apcPicYuvBaseVideo[iLeftBaseViewIdx]->copyToPic( pcPicYuvSynthOut ); // Copy Original
412	bRender = false;
413	}
414	}
415
416	if ( bRender )
417	{
418	std::cout << "Rendering Frame " << iFrame << " of View " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
419	m_pcRenTop->setShiftLUTs( NULL, NULL,NULL, m_cCameraData.getSynthViewShiftLUTD()[iRightBaseViewIdx ][iSynthViewIdx],
420	m_cCameraData.getSynthViewShiftLUTI()[iRightBaseViewIdx ][iSynthViewIdx],NULL, iRelDistToLeft);
421	m_pcRenTop->extrapolateView( apcPicYuvBaseVideo[iRightBaseViewIdx ], apcPicYuvBaseDepth[iRightBaseViewIdx ], pcPicYuvSynthOut, false);
422	}
423	break;
424	}
425
426	// Write Output
427
428	#if PIC_CROPPING
429	m_apcTVideoIOYuvSynthOutput[m_bSweep ? 0 : iSynthViewIdx]->write( pcPicYuvSynthOut, 0, 0, 0, 0 );
430	#else
431	m_apcTVideoIOYuvSynthOutput[m_bSweep ? 0 : iSynthViewIdx]->write( pcPicYuvSynthOut, aiPad );
432	#endif
433	}
434	iFrame++;
435	iNumOfRenderedFrames++;
436	}
437
438	// Delete Buffers
439	for ( UInt uiBaseView = 0; uiBaseView < m_iNumberOfInputViews; uiBaseView++ )
440	{
441	apcPicYuvBaseVideo[uiBaseView]->destroy();
442	delete apcPicYuvBaseVideo[uiBaseView];
443
444	apcPicYuvBaseDepth[uiBaseView]->destroy();
445	delete apcPicYuvBaseDepth[uiBaseView];
446
447	// Temporal Filter
448	if ( m_bTempDepthFilter )
449	{
450	apcPicYuvLastBaseVideo[uiBaseView]->destroy();
451	delete apcPicYuvLastBaseVideo[uiBaseView];
452
453	apcPicYuvLastBaseDepth[uiBaseView]->destroy();
454	delete apcPicYuvLastBaseDepth[uiBaseView];
455	}
456	}
457
458	pcPicYuvSynthOut->destroy();
459	delete pcPicYuvSynthOut;
460
461	xDestroyLib();
462
463	}
464
465	Void TAppRendererTop::go()
466	{
467	switch ( m_iRenderMode )
468	{
469	case 0:
470	render();
471	break;
472	case 1:
473	renderModel();
474	break;
475	case 10:
476	renderUsedPelsMap( );
477	break;
478
479	default:
480	AOT(true);
481	}
482	}
483
484	Void TAppRendererTop::renderModel()
485	{
486	if ( m_bUseSetupString )
487	{
488	xRenderModelFromString();
489	}
490	else
491	{
492	xRenderModelFromNums();
493	}
494	}
495
496	Void TAppRendererTop::xRenderModelFromString()
497	{
498
499	xCreateLib();
500	xInitLib();
501
502	// Create Buffers Input Views;
503	std::vector<TComPicYuv*> apcPicYuvBaseVideo;
504	std::vector<TComPicYuv*> apcPicYuvBaseDepth;
505
506
507	for ( UInt uiBaseView = 0; uiBaseView < m_iNumberOfInputViews; uiBaseView++ )
508	{
509	TComPicYuv* pcNewVideoPic = new TComPicYuv;
510	TComPicYuv* pcNewDepthPic = new TComPicYuv;
511
512	pcNewVideoPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
513	apcPicYuvBaseVideo.push_back(pcNewVideoPic);
514
515	pcNewDepthPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
516	apcPicYuvBaseDepth.push_back(pcNewDepthPic);
517	}
518
519	Int aiPad[2] = { 0, 0 };
520
521	// Init Model
522	TRenModel cCurModel;
523
524	AOT( m_iLog2SamplingFactor != 0 );
525	cCurModel.create( m_cRenModStrParser.getNumOfBaseViews(), m_cRenModStrParser.getNumOfModels(), m_iSourceWidth, m_iSourceHeight, m_iShiftPrecision, m_iBlendHoleMargin );
526
527	for ( Int iViewIdx = 0; iViewIdx < m_iNumberOfInputViews; iViewIdx++ )
528	{
529	Int iNumOfModels = m_cRenModStrParser.getNumOfModelsForView(iViewIdx, 1);
530
531	for (Int iCurModel = 0; iCurModel < iNumOfModels; iCurModel++ )
532	{
533	Int iModelNum; Int iLeftViewNum; Int iRightViewNum; Int iDump; Int iOrgRefNum; Int iBlendMode;
534	m_cRenModStrParser.getSingleModelData ( iViewIdx, 1, iCurModel, iModelNum, iBlendMode, iLeftViewNum, iRightViewNum, iOrgRefNum, iDump ) ;
535	cCurModel .createSingleModel ( iViewIdx, 1, iModelNum, iLeftViewNum, iRightViewNum, false, iBlendMode );
536
537	}
538	}
539
540	// Create Buffer for synthesized View
541	TComPicYuv* pcPicYuvSynthOut = new TComPicYuv;
542	pcPicYuvSynthOut->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
543
544	Bool bAnyEOS = false;
545
546	Int iNumOfRenderedFrames = 0;
547	Int iFrame = 0;
548
549	while ( ( ( iNumOfRenderedFrames < m_iFramesToBeRendered ) \|\| ( m_iFramesToBeRendered == 0 ) ) && !bAnyEOS )
550	{
551	// read in depth and video
552	for(Int iBaseViewIdx=0; iBaseViewIdx < m_iNumberOfInputViews; iBaseViewIdx++ )
553	{
554	m_apcTVideoIOYuvVideoInput[iBaseViewIdx]->read( apcPicYuvBaseVideo[iBaseViewIdx], aiPad ) ;
555	bAnyEOS \|= m_apcTVideoIOYuvVideoInput[iBaseViewIdx]->isEof();
556
557	m_apcTVideoIOYuvDepthInput[iBaseViewIdx]->read( apcPicYuvBaseDepth[iBaseViewIdx], aiPad ) ;
558	bAnyEOS \|= m_apcTVideoIOYuvDepthInput[iBaseViewIdx]->isEof();
559	}
560
561	if ( iFrame < m_iFrameSkip )
562	{
563	continue;
564	}
565
566
567	for(Int iBaseViewIdx=0; iBaseViewIdx < m_iNumberOfInputViews; iBaseViewIdx++ )
568	{
569	TComPicYuv* pcPicYuvVideo = apcPicYuvBaseVideo[iBaseViewIdx];
570	TComPicYuv* pcPicYuvDepth = apcPicYuvBaseDepth[iBaseViewIdx];
571	Int iBaseViewSIdx = m_cCameraData.getBaseId2SortedId()[iBaseViewIdx ];
572	cCurModel.setBaseView( iBaseViewSIdx, pcPicYuvVideo, pcPicYuvDepth, NULL, NULL );
573	}
574
575	for(Int iBaseViewIdx=0; iBaseViewIdx < m_iNumberOfInputViews; iBaseViewIdx++ )
576	{
577	m_cCameraData.update( (UInt)iFrame );
578
579	// setup virtual views
580	Int iBaseViewSIdx = m_cCameraData.getBaseId2SortedId()[iBaseViewIdx];
581
582	cCurModel.setErrorMode( iBaseViewSIdx, 1, 0 );
583	Int iNumOfSV = m_cRenModStrParser.getNumOfModelsForView( iBaseViewSIdx, 1);
584	for (Int iCurView = 0; iCurView < iNumOfSV; iCurView++ )
585	{
586	Int iOrgRefBaseViewSIdx;
587	Int iLeftBaseViewSIdx;
588	Int iRightBaseViewSIdx;
589	Int iSynthViewRelNum;
590	Int iModelNum;
591	Int iBlendMode;
592
593	m_cRenModStrParser.getSingleModelData(iBaseViewSIdx, 1, iCurView, iModelNum, iBlendMode, iLeftBaseViewSIdx, iRightBaseViewSIdx, iOrgRefBaseViewSIdx, iSynthViewRelNum );
594
595	Int iLeftBaseViewIdx = -1;
596	Int iRightBaseViewIdx = -1;
597
598	TComPicYuv* pcPicYuvOrgRef = NULL;
599	Int** ppiShiftLUTLeft = NULL;
600	Int** ppiShiftLUTRight = NULL;
601	Int** ppiBaseShiftLUTLeft = NULL;
602	Int** ppiBaseShiftLUTRight = NULL;
603
604
605	Int iDistToLeft = -1;
606
607	Int iSynthViewIdx = m_cCameraData.synthRelNum2Idx( iSynthViewRelNum );
608
609	if ( iLeftBaseViewSIdx != -1 )
610	{
611	iLeftBaseViewIdx = m_cCameraData.getBaseSortedId2Id() [ iLeftBaseViewSIdx ];
612	ppiShiftLUTLeft = m_cCameraData.getSynthViewShiftLUTI()[ iLeftBaseViewIdx ][ iSynthViewIdx ];
613	}
614
615	if ( iRightBaseViewSIdx != -1 )
616	{
617	iRightBaseViewIdx = m_cCameraData.getBaseSortedId2Id() [iRightBaseViewSIdx ];
618	ppiShiftLUTRight = m_cCameraData.getSynthViewShiftLUTI()[ iRightBaseViewIdx ][ iSynthViewIdx ];
619	}
620
621	if ( iRightBaseViewSIdx != -1 && iLeftBaseViewSIdx != -1 )
622	{
623
624	ppiBaseShiftLUTLeft = m_cCameraData.getBaseViewShiftLUTI() [ iLeftBaseViewIdx ][ iRightBaseViewIdx ];
625	ppiBaseShiftLUTRight = m_cCameraData.getBaseViewShiftLUTI() [ iRightBaseViewIdx ][ iLeftBaseViewIdx ];
626	iDistToLeft = m_cCameraData.getRelDistLeft( iSynthViewIdx , iLeftBaseViewIdx, iRightBaseViewIdx);
627	}
628
629	std::cout << "Rendering Frame " << iFrame << " of View " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
630
631	cCurModel.setSingleModel( iModelNum, ppiShiftLUTLeft, ppiBaseShiftLUTLeft, ppiShiftLUTRight, ppiBaseShiftLUTRight, iDistToLeft, pcPicYuvOrgRef );
632
633	Int iViewPos;
634	if (iLeftBaseViewSIdx != -1 && iRightBaseViewSIdx != -1)
635	{
636	iViewPos = VIEWPOS_MERGED;
637	}
638	else if ( iLeftBaseViewSIdx != -1 )
639	{
640	iViewPos = VIEWPOS_LEFT;
641	}
642	else if ( iRightBaseViewSIdx != -1 )
643	{
644	iViewPos = VIEWPOS_RIGHT;
645	}
646	else
647	{
648	AOT(true);
649	}
650
651	cCurModel.getSynthVideo ( iModelNum, iViewPos, pcPicYuvSynthOut );
652
653	// Write Output
654	#if PIC_CROPPING
655	m_apcTVideoIOYuvSynthOutput[m_bSweep ? 0 : iModelNum]->write( pcPicYuvSynthOut, 0 ,0 ,0, 0 );
656	#else
657	m_apcTVideoIOYuvSynthOutput[m_bSweep ? 0 : iModelNum]->write( pcPicYuvSynthOut, aiPad );
658	#endif
659	}
660	}
661	iFrame++;
662	iNumOfRenderedFrames++;
663	}
664
665	// Delete Buffers
666	for ( UInt uiBaseView = 0; uiBaseView < m_iNumberOfInputViews; uiBaseView++ )
667	{
668	apcPicYuvBaseVideo[uiBaseView]->destroy();
669	delete apcPicYuvBaseVideo[uiBaseView];
670
671	apcPicYuvBaseDepth[uiBaseView]->destroy();
672	delete apcPicYuvBaseDepth[uiBaseView];
673	}
674	pcPicYuvSynthOut->destroy();
675	delete pcPicYuvSynthOut;
676
677	xDestroyLib();
678	}
679
680	Void TAppRendererTop::xRenderModelFromNums()
681	{
682	xCreateLib();
683	xInitLib();
684
685	// Create Buffers Input Views;
686	std::vector<TComPicYuv*> apcPicYuvBaseVideo;
687	std::vector<TComPicYuv*> apcPicYuvBaseDepth;
688
689
690	Int aiPad[2] = { 0, 0 };
691
692	// Init Model
693	TRenModel cCurModel;
694
695	AOT( m_iLog2SamplingFactor != 0 );
696	cCurModel.create( m_iNumberOfInputViews, m_iNumberOfOutputViews, m_iSourceWidth, m_iSourceHeight, m_iShiftPrecision, m_iBlendHoleMargin );
697
698	for ( UInt uiBaseView = 0; uiBaseView < m_iNumberOfInputViews; uiBaseView++ )
699	{
700	TComPicYuv* pcNewVideoPic = new TComPicYuv;
701	TComPicYuv* pcNewDepthPic = new TComPicYuv;
702
703	pcNewVideoPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
704	apcPicYuvBaseVideo.push_back(pcNewVideoPic);
705
706	pcNewDepthPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
707	apcPicYuvBaseDepth.push_back(pcNewDepthPic);
708	}
709
710	for(Int iSynthViewIdx=0; iSynthViewIdx < m_iNumberOfOutputViews; iSynthViewIdx++ )
711	{
712	Int iLeftBaseViewIdx = -1;
713	Int iRightBaseViewIdx = -1;
714	Bool bIsBaseView = false;
715
716	Int iRelDistToLeft;
717	m_cCameraData.getLeftRightBaseView( iSynthViewIdx, iLeftBaseViewIdx, iRightBaseViewIdx, iRelDistToLeft, bIsBaseView );
718
719	if (m_iRenderDirection == 1 )
720	{
721	iRightBaseViewIdx = -1;
722	AOT( iLeftBaseViewIdx == -1);
723	}
724
725	if (m_iRenderDirection == 2 )
726	{
727	iLeftBaseViewIdx = -1;
728	AOT( iRightBaseViewIdx == -1);
729	}
730
731	Int iLeftBaseViewSIdx = -1;
732	Int iRightBaseViewSIdx = -1;
733
734	if (iLeftBaseViewIdx != -1 )
735	{
736	iLeftBaseViewSIdx = m_cCameraData.getBaseId2SortedId()[iLeftBaseViewIdx];
737	}
738
739	if (iRightBaseViewIdx != -1 )
740	{
741	iRightBaseViewSIdx = m_cCameraData.getBaseId2SortedId()[iRightBaseViewIdx];
742	}
743	cCurModel.createSingleModel(-1, -1, iSynthViewIdx, iLeftBaseViewSIdx, iRightBaseViewSIdx, false, m_iBlendMode );
744	}
745
746	// Create Buffer for synthesized View
747	TComPicYuv* pcPicYuvSynthOut = new TComPicYuv;
748	pcPicYuvSynthOut->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
749
750	Bool bAnyEOS = false;
751
752	Int iNumOfRenderedFrames = 0;
753	Int iFrame = 0;
754
755	while ( ( ( iNumOfRenderedFrames < m_iFramesToBeRendered ) \|\| ( m_iFramesToBeRendered == 0 ) ) && !bAnyEOS )
756	{
757	// read in depth and video
758	for(Int iBaseViewIdx=0; iBaseViewIdx < m_iNumberOfInputViews; iBaseViewIdx++ )
759	{
760	m_apcTVideoIOYuvVideoInput[iBaseViewIdx]->read( apcPicYuvBaseVideo[iBaseViewIdx], aiPad ) ;
761	bAnyEOS \|= m_apcTVideoIOYuvVideoInput[iBaseViewIdx]->isEof();
762
763	m_apcTVideoIOYuvDepthInput[iBaseViewIdx]->read( apcPicYuvBaseDepth[iBaseViewIdx], aiPad ) ;
764	bAnyEOS \|= m_apcTVideoIOYuvDepthInput[iBaseViewIdx]->isEof();
765
766	if ( iFrame >= m_iFrameSkip )
767	{
768	Int iBaseViewSIdx = m_cCameraData.getBaseId2SortedId()[iBaseViewIdx];
769	cCurModel.setBaseView( iBaseViewSIdx, apcPicYuvBaseVideo[iBaseViewIdx], apcPicYuvBaseDepth[iBaseViewIdx], NULL, NULL );
770	}
771	}
772
773	if ( iFrame < m_iFrameSkip ) // Skip Frames
774	{
775	iFrame++;
776	continue;
777	}
778
779	m_cCameraData.update( (UInt)iFrame );
780
781	for(Int iSynthViewIdx=0; iSynthViewIdx < m_iNumberOfOutputViews; iSynthViewIdx++ )
782	{
783
784	Int iLeftBaseViewIdx = -1;
785	Int iRightBaseViewIdx = -1;
786
787	Bool bIsBaseView = false;
788
789	Int iRelDistToLeft;
790	Bool bHasLRView = m_cCameraData.getLeftRightBaseView( iSynthViewIdx, iLeftBaseViewIdx, iRightBaseViewIdx, iRelDistToLeft, bIsBaseView );
791	Bool bHasLView = ( iLeftBaseViewIdx != -1 );
792	Bool bHasRView = ( iRightBaseViewIdx != -1 );
793
794	switch( m_iRenderDirection )
795	{
796	/// INTERPOLATION
797	case 0:
798	assert( bHasLRView \|\| bIsBaseView );
799
800	if ( !bHasLRView && bIsBaseView ) // View to render is BaseView
801	{
802	std::cout << "Copied Frame " << iFrame << " of BaseView " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
803	apcPicYuvBaseVideo[iLeftBaseViewIdx]->copyToPic( pcPicYuvSynthOut ); // Copy Original
804	}
805	else // Render
806	{
807	std::cout << "Rendering Frame " << iFrame << " of View " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
808	cCurModel.setSingleModel( iSynthViewIdx,
809	m_cCameraData.getSynthViewShiftLUTI()[iLeftBaseViewIdx ][iSynthViewIdx] ,
810	m_cCameraData.getBaseViewShiftLUTI ()[iLeftBaseViewIdx ][iRightBaseViewIdx],
811	m_cCameraData.getSynthViewShiftLUTI()[iRightBaseViewIdx][iSynthViewIdx] ,
812	m_cCameraData.getBaseViewShiftLUTI ()[iRightBaseViewIdx][iLeftBaseViewIdx] ,
813	iRelDistToLeft,
814	NULL );
815	cCurModel.getSynthVideo ( iSynthViewIdx, VIEWPOS_MERGED, pcPicYuvSynthOut );
816	}
817	break;
818	/// EXTRAPOLATION FROM LEFT
819	case 1:
820
821	if ( !bHasLView ) // View to render is BaseView
822	{
823	std::cout << "Copied Frame " << iFrame << " of BaseView " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
824	apcPicYuvBaseVideo[iLeftBaseViewIdx]->copyToPic( pcPicYuvSynthOut ); // Copy Original
825	}
826	else // Render
827	{
828	std::cout << "Rendering Frame " << iFrame << " of View " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
829	cCurModel.setSingleModel( iSynthViewIdx, m_cCameraData.getSynthViewShiftLUTI()[iLeftBaseViewIdx ][iSynthViewIdx], NULL, NULL, NULL, -1, NULL);
830	cCurModel.getSynthVideo ( iSynthViewIdx, VIEWPOS_LEFT, pcPicYuvSynthOut );
831	}
832	break;
833	/// EXTRAPOLATION FROM RIGHT
834	case 2: // extrapolation from right
835	if ( !bHasRView ) // View to render is BaseView
836	{
837	std::cout << "Copied Frame " << iFrame << " of BaseView " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
838	apcPicYuvBaseVideo[iRightBaseViewIdx]->copyToPic( pcPicYuvSynthOut ); // Copy Original
839	}
840	else // Render
841	{
842	std::cout << "Rendering Frame " << iFrame << " of View " << (Double) m_cCameraData.getSynthViewNumbers()[iSynthViewIdx] / VIEW_NUM_PREC << std::endl;
843	cCurModel.setSingleModel( iSynthViewIdx, NULL , NULL, m_cCameraData.getSynthViewShiftLUTI()[iRightBaseViewIdx ][iSynthViewIdx], NULL, -1, NULL);
844	cCurModel.getSynthVideo ( iSynthViewIdx, VIEWPOS_RIGHT, pcPicYuvSynthOut );
845	}
846	break;
847	}
848
849	// Write Output
850	#if PIC_CROPPING
851	m_apcTVideoIOYuvSynthOutput[m_bSweep ? 0 : iSynthViewIdx]->write( pcPicYuvSynthOut, 0, 0, 0, 0 );
852	#else
853	m_apcTVideoIOYuvSynthOutput[m_bSweep ? 0 : iSynthViewIdx]->write( pcPicYuvSynthOut, aiPad );
854	#endif
855	}
856	iFrame++;
857	iNumOfRenderedFrames++;
858	}
859
860	// Delete Buffers
861	for ( UInt uiBaseView = 0; uiBaseView < m_iNumberOfInputViews; uiBaseView++ )
862	{
863	apcPicYuvBaseVideo[uiBaseView]->destroy();
864	delete apcPicYuvBaseVideo[uiBaseView];
865
866	apcPicYuvBaseDepth[uiBaseView]->destroy();
867	delete apcPicYuvBaseDepth[uiBaseView];
868	}
869	pcPicYuvSynthOut->destroy();
870	delete pcPicYuvSynthOut;
871
872	xDestroyLib();
873
874	}
875
876	Void TAppRendererTop::renderUsedPelsMap( )
877	{
878	xCreateLib();
879	xInitLib();
880
881	// Create Buffers Input Views;
882	std::vector<TComPicYuv*> apcPicYuvBaseVideo;
883	std::vector<TComPicYuv*> apcPicYuvBaseDepth;
884
885	// TemporalImprovement Filter
886	std::vector<TComPicYuv*> apcPicYuvLastBaseVideo;
887	std::vector<TComPicYuv*> apcPicYuvLastBaseDepth;
888
889	Int aiPad[2] = { 0, 0 };
890
891	for ( UInt uiBaseView = 0; uiBaseView < m_iNumberOfInputViews; uiBaseView++ )
892	{
893	TComPicYuv* pcNewVideoPic = new TComPicYuv;
894	TComPicYuv* pcNewDepthPic = new TComPicYuv;
895
896	pcNewVideoPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
897	apcPicYuvBaseVideo.push_back(pcNewVideoPic);
898
899	pcNewDepthPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
900	apcPicYuvBaseDepth.push_back(pcNewDepthPic);
901
902	//Temporal improvement Filter
903	if ( m_bTempDepthFilter )
904	{
905	pcNewVideoPic = new TComPicYuv;
906	pcNewDepthPic = new TComPicYuv;
907
908	pcNewVideoPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
909	apcPicYuvLastBaseVideo.push_back(pcNewVideoPic);
910
911	pcNewDepthPic->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
912	apcPicYuvLastBaseDepth.push_back(pcNewDepthPic);
913	}
914	}
915
916	// Create Buffer for synthesized View
917	TComPicYuv* pcPicYuvSynthOut = new TComPicYuv;
918	pcPicYuvSynthOut->create( m_iSourceWidth, m_iSourceHeight, 1, 1, 1 );
919
920	Bool bAnyEOS = false;
921
922	Int iNumOfRenderedFrames = 0;
923	Int iFrame = 0;
924
925	while ( ( ( iNumOfRenderedFrames < m_iFramesToBeRendered ) \|\| ( m_iFramesToBeRendered == 0 ) ) && !bAnyEOS )
926	{
927	// set shift LUT
928
929	// read in depth and video
930	for(Int iBaseViewIdx=0; iBaseViewIdx < m_iNumberOfInputViews; iBaseViewIdx++ )
931	{
932	m_apcTVideoIOYuvVideoInput[iBaseViewIdx]->read( apcPicYuvBaseVideo[iBaseViewIdx], aiPad ) ;
933	apcPicYuvBaseVideo[iBaseViewIdx]->extendPicBorder();
934	bAnyEOS \|= m_apcTVideoIOYuvVideoInput[iBaseViewIdx]->isEof();
935
936	m_apcTVideoIOYuvDepthInput[iBaseViewIdx]->read( apcPicYuvBaseDepth[iBaseViewIdx], aiPad ) ;
937	apcPicYuvBaseDepth[iBaseViewIdx]->extendPicBorder();
938	bAnyEOS \|= m_apcTVideoIOYuvDepthInput[iBaseViewIdx]->isEof();
939
940	if ( m_bTempDepthFilter && (iFrame >= m_iFrameSkip) )
941	{
942	m_pcRenTop->temporalFilterVSRS( apcPicYuvBaseVideo[iBaseViewIdx], apcPicYuvBaseDepth[iBaseViewIdx], apcPicYuvLastBaseVideo[iBaseViewIdx], apcPicYuvLastBaseDepth[iBaseViewIdx], ( iFrame == m_iFrameSkip) );
943	}
944	}
945
946	if ( iFrame < m_iFrameSkip ) // Skip Frames
947	{
948	std::cout << "Skipping Frame " << iFrame << std::endl;
949
950	iFrame++;
951	continue;
952	}
953
954	m_cCameraData.update( (UInt)iFrame );
955
956	for(Int iViewIdx=1; iViewIdx < m_iNumberOfInputViews; iViewIdx++ )
957	{
958	std::cout << "Rendering UsedPelsMap for Frame " << iFrame << " of View " << (Double) m_cCameraData.getBaseViewNumbers()[iViewIdx] << std::endl;
959
960	Int iViewSIdx = m_cCameraData.getBaseId2SortedId()[iViewIdx];
961	Int iFirstViewSIdx = m_cCameraData.getBaseId2SortedId()[0];
962
963	AOT( iViewSIdx == iFirstViewSIdx );
964
965	Bool bFirstIsLeft = (iFirstViewSIdx < iViewSIdx);
966
967	m_pcRenTop->setShiftLUTs(
968	m_cCameraData.getBaseViewShiftLUTD()[0][iViewIdx],
969	m_cCameraData.getBaseViewShiftLUTI()[0][iViewIdx],
970	m_cCameraData.getBaseViewShiftLUTI()[0][iViewIdx],
971	m_cCameraData.getBaseViewShiftLUTD()[0][iViewIdx],
972	m_cCameraData.getBaseViewShiftLUTI()[0][iViewIdx],
973	m_cCameraData.getBaseViewShiftLUTI()[0][iViewIdx],
974	-1
975	);
976
977	m_pcRenTop->getUsedSamplesMap( apcPicYuvBaseDepth[0], pcPicYuvSynthOut, bFirstIsLeft );
978
979	// Write Output
980	#if PIC_CROPPING
981	m_apcTVideoIOYuvSynthOutput[iViewIdx-1]->write( pcPicYuvSynthOut, 0, 0, 0 );
982	#else
983	m_apcTVideoIOYuvSynthOutput[iViewIdx-1]->write( pcPicYuvSynthOut, aiPad );
984	#endif
985
986	}
987	iFrame++;
988	iNumOfRenderedFrames++;
989	}
990
991	// Delete Buffers
992	for ( UInt uiBaseView = 0; uiBaseView < m_iNumberOfInputViews; uiBaseView++ )
993	{
994	apcPicYuvBaseVideo[uiBaseView]->destroy();
995	delete apcPicYuvBaseVideo[uiBaseView];
996
997	apcPicYuvBaseDepth[uiBaseView]->destroy();
998	delete apcPicYuvBaseDepth[uiBaseView];
999
1000	// Temporal Filter
1001	if ( m_bTempDepthFilter )
1002	{
1003	apcPicYuvLastBaseVideo[uiBaseView]->destroy();
1004	delete apcPicYuvLastBaseVideo[uiBaseView];
1005
1006	apcPicYuvLastBaseDepth[uiBaseView]->destroy();
1007	delete apcPicYuvLastBaseDepth[uiBaseView];
1008	}
1009	}
1010	pcPicYuvSynthOut->destroy();
1011	delete pcPicYuvSynthOut;
1012
1013	xDestroyLib();
1014
1015	}

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

Download in other formats: