Context Navigation

DistortionManipulator.cpp @ 427

Last change on this file since 427 was 408, checked in by Torben Dannhauer, 12 years ago

File size: 24.6 KB

Line
1	/* osgVisual test. distortionNG, experimental.
2	*
3	* Permission is hereby granted, free of charge, to any person obtaining a copy
4	* of this software and associated documentation files (the "Software"), to deal
5	* in the Software without restriction, including without limitation the rights
6	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7	* copies of the Software, and to permit persons to whom the Software is
8	* furnished to do so, subject to the following conditions:
9	*
10	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
11	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
12	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
13	* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
14	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
15	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
16	* THE SOFTWARE.
17	*/
18
19	#include "DistortionManipulator.h"
20	#include "extViewer.h"
21
22	#include <osgViewer/Viewer>
23	#include <osg/Point>
24	#include <osg/PolygonMode>
25	#include <osg/TextureRectangle>
26
27	#include <osgDB/Registry>
28	#include <osgDB/ReaderWriter>
29	#include <osgDB/WriteFile>
30
31	#include "DistortionSetupStrategy.h"
32
33	using namespace osg;
34	using namespace osgViewer;
35
36
37	DistortionManipulator::DistortionManipulator(extViewer* viewer, DistortionSet* ds)
38	: _highlightColor( osg::Vec4(1.0f, 1.0f, 0.0f, 1.0f) ), _distortionSet( ds ), _viewer(viewer)
39	{
40	activeSetupMode = DISABLED;
41	activeDistortionMode = MESH;
42	activeManualSetupMode = DISTORTION;
43	activeVisualizationMode = NONE;
44
45	_delegatedDistortionSetupStrategy = NULL;
46
47	_highlighter = NULL;
48
49	createVertexHighlighter();
50	createHUD();
51
52	_camera = 0;
53	//_distortionMesh(distortionMesh)
54
55	// Create Shader to visualize intensitymap during blending setup.
56	osg::Shader* sh = osg::Shader::readShaderFile( osg::Shader::FRAGMENT, "shaderIntensityMapVis.frag" );
57	sh->setName("shaderIntensityMapVis");
58	ds->setShaderIntensityMapVis( sh );
59
60	//LF Get Screen Resolution for Cursor Position Scaling
61	osg::GraphicsContext::WindowingSystemInterface* wsi = osg::GraphicsContext::getWindowingSystemInterface();
62	if (!wsi) { osg::notify(osg::NOTICE)<<"Error, no WindowSystemInterface available, cannot create windows."<<std::endl; }
63	wsi->getScreenResolution(osg::GraphicsContext::ScreenIdentifier(0), _screenPixWidth, _screenPixHeight);
64
65
66	}
67
68	DistortionManipulator::~DistortionManipulator()
69	{
70	}
71
72	void DistortionManipulator::getUsage(osg::ApplicationUsage& usage) const
73	{
74	usage.addKeyboardMouseBinding("Keypad 0","Show/Hide distortion HUD.");
75	usage.addKeyboardMouseBinding("Keypad 1","Reset distortion.");
76	usage.addKeyboardMouseBinding("Keypad 2","Reset intensity blending.");
77	usage.addKeyboardMouseBinding("Keypad 4","Toggles Setup Mode between DISABLED, MANUAL & DELEGATED.");
78	usage.addKeyboardMouseBinding("Keypad 5","MANUAL Mode: Toggle between blending & distortion setup.");
79	usage.addKeyboardMouseBinding("Keypad 6","MANUAL Mode: Toggle if distortion drags affect mesh or rtt texture coordinates."); // Defaults to Mesh
80	usage.addKeyboardMouseBinding("Keypad 7","Show distortion mesh / intensity map / none.");
81	usage.addKeyboardMouseBinding("Keypad 8","Save distortion set."); // via plugin
82	}
83
84	bool DistortionManipulator::handle(const osgGA::GUIEventAdapter& ea, osgGA::GUIActionAdapter& aa, osg::Object* obj, osg::NodeVisitor* nv)
85	{
86	switch(ea.getEventType())
87	{
88	case(osgGA::GUIEventAdapter::MOVE): break;
89	case(osgGA::GUIEventAdapter::DRAG):
90	{
91	if ( activeSetupMode == MANUAL)
92	{
93	OSG_ALWAYS<<std::endl<<"Drag:"<<std::endl<<"ea.getGraphicsContext()="<<ea.getGraphicsContext()<<std::endl;
94	OSG_ALWAYS<<"ea.getXnormalized()="<<ea.getXnormalized()<<std::endl;
95	OSG_ALWAYS<<"ea.getYnormalized()="<<ea.getYnormalized()<<std::endl;
96	OSG_ALWAYS<<"ea.getX()="<<ea.getX()<<std::endl;
97	OSG_ALWAYS<<"ea.getXin()="<<ea.getXmin()<<std::endl;
98	OSG_ALWAYS<<"ea.getXmax()="<<ea.getXmax()<<std::endl;
99	OSG_ALWAYS<<"ea.getY()="<<ea.getY()<<std::endl;
100	OSG_ALWAYS<<"ea.getYin()="<<ea.getYmin()<<std::endl;
101	OSG_ALWAYS<<"ea.getYmax()="<<ea.getYmax()<<std::endl;
102
103	//LF get X,Y normalized to lowerleftcorner (0,0) / upperrightcorner (1,1)
104	OSG_ALWAYS<<"getXnormalizedLF()="<<(ea.getXnormalized()+1.0)/2.0<<std::endl;
105	OSG_ALWAYS<<"getYnormalizedLF()="<<(ea.getYnormalized()+1.0)/2.0<<std::endl;
106
107	//LF get X,Y in pixels (lowerleftcorner (0,0) / upperrightcorner (screenwidth,screenheigth)
108	OSG_ALWAYS<<std::endl<<"Screen Resolution: "<<_screenPixWidth<<" x "<<_screenPixHeight<<std::endl;
109	OSG_ALWAYS<<"getXPixelspaceLF()="<<(int)((float)_screenPixWidth*(ea.getXnormalized()+1.0)/2.0)<<std::endl;
110	OSG_ALWAYS<<"getYPixelspaceLF()="<<(int)((float)_screenPixHeight*(ea.getYnormalized()+1.0)/2.0)<<std::endl;
111
112	if(activeDistortionMode == MESH)
113	{
114	OSG_ALWAYS<<"MESH!"<<std::endl;
115	}
116	else if (activeDistortionMode == TEXCOORDINATES)
117	{
118	OSG_ALWAYS<<"TEXCOORDINATES!"<<std::endl;
119	}
120
121	return true; // true means event handled: not forwarded to the camera manipulator;
122	}
123	break;
124
125	}
126	case(osgGA::GUIEventAdapter::PUSH):
127	{
128	OSG_ALWAYS<<"mouse click!"<<std::endl;
129	if ( activeSetupMode == MANUAL /&& ea.getModKeyMask()&osgGA::GUIEventAdapter::MODKEY_CTRL/ && ea.getButton() == osgGA::GUIEventAdapter::LEFT_MOUSE_BUTTON )
130	{
131	osgViewer::View* viewer = dynamic_cast<osgViewer::View*>(&aa);
132	if ( viewer )
133	{
134	osg::notify(osg::ALWAYS)<<std::endl<<"Intersection:"<<std::endl<<"ea.getX()="<<ea.getX()<<std::endl;
135	osg::notify(osg::ALWAYS)<<"ea.getY()="<<ea.getY()<<std::endl;
136	osg::ref_ptr<osgUtil::LineSegmentIntersector> intersector = new osgUtil::LineSegmentIntersector(osgUtil::Intersector::PROJECTION, ea.getX(), ea.getY());
137	osgUtil::IntersectionVisitor iv( intersector.get() );
138	_distortionSet->getDistortionCamera()->accept( iv );
139
140	if ( intersector->containsIntersections() )
141	{
142	//osgUtil::LineSegmentIntersector::Intersection& result = *(intersector->getIntersections().begin());
143	osgUtil::LineSegmentIntersector::Intersection *result;
144	result=&intersector->getFirstIntersection();
145	computeSelectedVertex( *result );
146	//computeSelectedVertex( result );
147	}
148	}
149	}
150	break;
151	}
152	case(osgGA::GUIEventAdapter::RELEASE): break;
153	case(osgGA::GUIEventAdapter::KEYDOWN):
154	{
155	if (ea.getKey()==osgGA::GUIEventAdapter::KEY_KP_Insert) // KP 0: Show/Hide HUD
156	{
157	bool oldValue = _distortionSet->getDistortionInternals()->getValue(DistortionSet::HUD);
158	_distortionSet->getDistortionInternals()->setValue(DistortionSet::HUD, !oldValue);
159	return(true);
160	}
161	if (ea.getKey()==osgGA::GUIEventAdapter::KEY_KP_End) // KP 1: reset distortion
162	{
163	resetDistortion();
164	return(true);
165	}
166	if (ea.getKey()==osgGA::GUIEventAdapter::KEY_KP_Down) // KP 2: reset intensity map
167	{
168	resetIntensityMap();
169	return(true);
170	}
171	if (ea.getKey()==osgGA::GUIEventAdapter::KEY_KP_Left) // KP 4: Toggles Setup Mode between DISABLED, MANUAL & DELEGATED.
172	{
173	switch(activeSetupMode)
174	{
175	case DISABLED :
176	{
177	activeSetupMode = MANUAL;
178	OSG_NOTICE<<"SetupMode MANUAL activated"<<std::endl;
179	_distortionSet->getDistortionInternals()->setValue(DistortionSet::HIGHLIGHTER, true);
180	break;
181	}
182	case MANUAL :
183	{
184	activeSetupMode = DELEGATED;
185	_distortionSet->getDistortionInternals()->setValue(DistortionSet::HIGHLIGHTER, false);
186	OSG_NOTICE<<"SetupMode DELEGATED activated"<<std::endl;
187	break;
188	}
189	case DELEGATED :
190	{
191	activeSetupMode = DISABLED;
192	_distortionSet->getDistortionInternals()->setValue(DistortionSet::HIGHLIGHTER, false);
193	OSG_NOTICE<<"SetupMode DISABLED activated"<<std::endl;
194	break;
195	}
196	}
197	updateHUD();
198	return(true);
199	}
200	if (ea.getKey()==osgGA::GUIEventAdapter::KEY_KP_Begin) // KP 5: MANUAL Mode: Toggle between blending & distortion setup.
201	{
202
203	activeManualSetupMode = (activeManualSetupMode==DISTORTION?BLENDING:DISTORTION);
204	updateHUD();
205	return(true);
206	}
207	if (ea.getKey()==osgGA::GUIEventAdapter::KEY_KP_Right) // KP 6: MANUAL Mode: Toggle if distortion drags affect mesh or rtt texture coordinates -> defaults to Mesh
208	{
209	activeDistortionMode = (activeDistortionMode==MESH?TEXCOORDINATES:MESH);
210	updateHUD();
211	return(true);
212	}
213	if (ea.getKey()==osgGA::GUIEventAdapter::KEY_KP_Home) // KP 7: Show distortion mesh / intensity map / none.
214	{
215	switch(activeVisualizationMode)
216	{
217	case DISTORTION_MESH :
218	{
219	activeVisualizationMode = INTENSITY_MAP;
220	showDistortionMesh(false);
221	showIntensityMap(true);
222	break;
223	}
224	case INTENSITY_MAP :
225	{
226	activeVisualizationMode = NONE;
227	showDistortionMesh(false);
228	showIntensityMap(false);
229	break;
230	}
231	case NONE :
232	{
233	activeVisualizationMode = DISTORTION_MESH;
234	showDistortionMesh(true);
235	showIntensityMap(false);
236	break;
237	}
238	}
239
240	updateHUD();
241	return(true);
242	}
243	if (ea.getKey()==osgGA::GUIEventAdapter::KEY_KP_Up) // KP 8: Save distortion set via plugin
244	{
245	OSG_ALWAYS<<"KEY_KP_8 : todo: Save DistortionContainer"<<std::endl;
246
247	osgDB::writeObjectFile( *_distortionSet, "distortionset.dist" );
248	return(true);
249	}
250
251	return false; // Event ignored
252	}
253	case(osgGA::GUIEventAdapter::KEYUP): break;
254	case(osgGA::GUIEventAdapter::FRAME):
255	{
256	if ( activeSetupMode == DELEGATED && _delegatedDistortionSetupStrategy.valid())
257	{
258	//OSG_ALWAYS<<"Calling delegated class.."<<std::endl;
259	_delegatedDistortionSetupStrategy->delegateDistortionSetup(_distortionSet);
260	if( _distortionSet->isDirty() && _viewer.valid())
261	{
262	updateDistortionSetup();
263	}
264	//OSG_ALWAYS<<"..done"<<std::endl;
265	}
266	break;
267	}
268	case (osgGA::GUIEventAdapter::RESIZE): // todo: adapt distortion mesh to new screen size
269	{
270	OSG_ALWAYS<<"RESIZE!"<<std::endl;
271	break;
272	}
273
274	default:
275	return false;
276	}
277	return false;
278	}
279
280	void DistortionManipulator::resetIntensityMap()
281	{
282	if(!_distortionSet.valid())
283	return;
284
285	osg::ref_ptr<osg::Image> image = _distortionSet->getIntensityMap();
286
287	OSG_ALWAYS<<"Reseting IntensityMap Blending."<<std::endl;
288
289	//isDataContiguous is unknown to MSVS2010 and GCC
290	/*if (!image->isDataContiguous())
291	{
292	OSG_WARN<<"Warning: DistortionManipulator does not support working with non contiguous imagery as blendmaps!"<<std::endl;
293	return;
294	}*/
295
296	// Fill intensity map with 255
297	unsigned char* dataPtr = image->data();
298	for(unsigned int i=0;i<image->getTotalSizeInBytes();i++)
299	{
300	*dataPtr++ = 255;
301	}
302	image->dirty();
303	}
304
305	void DistortionManipulator::resetDistortion()
306	{
307	if(!_distortionSet.valid())
308	return;
309
310
311	_distortionSet->setDistortionMeshDimensions(2, 2, true); // Reset dimensions and coordinates/texcoords & rebuild mesh data
312	_distortionSet->dirtyMesh();
313
314	// Restore ProjectionMatrix
315	// Set up basic projection Matrix
316	double fovy = 30., aspectRatio = 1., zNear = 1., zFar = 1000.;
317	_distortionSet->getProjectionOffset().makePerspective( fovy, aspectRatio, zNear, zFar );
318
319	double newAspectRatio = double(_distortionSet->getIntensityMap()->s()) / double(_distortionSet->getIntensityMap()->t());
320	double aspectRatioChange = newAspectRatio / aspectRatio;
321	if (aspectRatioChange != 1.0)
322	{
323	_distortionSet->getProjectionOffset() *= osg::Matrix::scale(1.0/aspectRatioChange,1.0,1.0);
324	}
325
326	_distortionSet->dirtyMatrix();
327	}
328
329	void DistortionManipulator::showDistortionMesh(bool show)
330	{
331	osg::StateSet* stateset = _distortionSet->getDistortionInternals()->getChild(DistortionSet::MESH)->getOrCreateStateSet();
332
333	// Append a PolygonMode stateset if required
334	osg::PolygonMode* polyModeObj = dynamic_cast<osg::PolygonMode*>(stateset->getAttribute(osg::StateAttribute::POLYGONMODE));
335	if (!polyModeObj)
336	{
337	polyModeObj = new osg::PolygonMode;
338	stateset->setAttribute(polyModeObj, osg::StateAttribute::PROTECTED\|osg::StateAttribute::ON);
339	}
340
341	// Set Polygonmode
342	if(show)
343	polyModeObj->setMode(osg::PolygonMode::FRONT_AND_BACK,osg::PolygonMode::LINE);
344	else
345	polyModeObj->setMode(osg::PolygonMode::FRONT_AND_BACK,osg::PolygonMode::FILL);
346	}
347
348	void DistortionManipulator::showIntensityMap(bool show)
349	{
350	if(_distortionSet.valid())
351	{
352	osg::StateSet* stateset = _distortionSet->getDistortionInternals()->getChild(DistortionSet::MESH)->getOrCreateStateSet();
353	osg::Program* program = static_cast<osg::Program*>(stateset->getAttribute(::osg::StateAttribute::PROGRAM));
354
355	if(show)
356	{
357	program->removeShader( _distortionSet->getShaderIntensityMap() );
358	program->addShader( _distortionSet->getShaderIntensityMapVis() );
359	}
360	else
361	{
362	program->removeShader( _distortionSet->getShaderIntensityMapVis() );
363	program->addShader( _distortionSet->getShaderIntensityMap() );
364	}
365	}
366	}
367
368	void DistortionManipulator::createVertexHighlighter()
369	{
370	osg::ref_ptr<osg::Vec4Array> colors = new osg::Vec4Array(1);
371	(*colors)[0] = _highlightColor;
372
373	_highlighter = new osg::Geometry;
374	_highlighter->setDataVariance( osg::Object::DYNAMIC );
375	_highlighter->setUseDisplayList( false );
376	_highlighter->setUseVertexBufferObjects( true );
377	_highlighter->setVertexArray( new osg::Vec3Array(1) ); // The highlighter vertex is updated by computeSelectedVertex(..)
378	_highlighter->setColorArray( colors.get() );
379	_highlighter->setColorBinding( osg::Geometry::BIND_OVERALL );
380	_highlighter->addPrimitiveSet( new osg::DrawArrays(GL_POINTS, 0, 1) );
381
382
383	osg::ref_ptr<osg::Geode> geode = new osg::Geode;
384	geode->addDrawable( _highlighter.get() );
385	geode->getOrCreateStateSet()->setAttributeAndModes( new osg::Point(8.0f) );
386	geode->getOrCreateStateSet()->setMode( GL_LIGHTING, osg::StateAttribute::OFF );
387	geode->setCullingActive(false); // disable the culling for the selector, otherwise the selector is culled away on the edge.
388
389	_distortionSet->getDistortionInternals()->addChild(geode, false); // = child #1 :definition: child #0 = mesh, #1 = highlighter, #2 HUD
390	}
391
392	void DistortionManipulator::createHUD()
393	{
394	osg::ref_ptr<osg::Geode> geode = new osg::Geode;
395
396	osg::StateSet* stateSet = geode->getOrCreateStateSet();
397	stateSet->setRenderBinDetails( 99, "RenderBin");
398	// disable depth test to ensure that it is always drawn.
399	stateSet->setMode(GL_DEPTH_TEST,osg::StateAttribute::OFF);
400	stateSet->setMode(GL_LIGHTING,osg::StateAttribute::OFF);
401
402	// Modify StateSet to protect it against state changes by the stateset Manipulator
403	osg::PolygonMode* polyModeObj = new osg::PolygonMode;
404	stateSet->setAttribute(polyModeObj, osg::StateAttribute::PROTECTED\|osg::StateAttribute::ON);
405
406	// Add Text:
407	osg::Vec3 position(20.0f,120.0f,0.0f);
408	osg::Vec3 delta(0.0f,-25.0f,0.0f);
409	std::string timesFont("fonts/arial.ttf");
410	{
411	osg::ref_ptr<osgText::Text> textHeader = new osgText::Text;
412	textHeader->setText("Distortion Setup:");
413	textHeader->setFont(timesFont);
414	textHeader->setPosition(position);
415	textHeader->setCharacterSize(21);
416	textHeader->setDataVariance(osg::Object::STATIC);
417	geode->addDrawable( textHeader );
418
419	position += delta;
420
421	hudSetupMode = new osgText::Text;
422	hudSetupMode->setFont(timesFont);
423	hudSetupMode->setPosition(position);
424	hudSetupMode->setCharacterSize(21);
425	hudSetupMode->setDataVariance(osg::Object::DYNAMIC);
426	geode->addDrawable( hudSetupMode );
427
428	position += delta;
429
430	hudDistortionMode = new osgText::Text;
431	hudDistortionMode->setFont(timesFont);
432	hudDistortionMode->setPosition(position);
433	hudDistortionMode->setCharacterSize(21);
434	hudDistortionMode->setDataVariance(osg::Object::DYNAMIC);
435	geode->addDrawable( hudDistortionMode );
436
437	position += delta;
438
439	hudManualSetupMode = new osgText::Text;
440	hudManualSetupMode->setFont(timesFont);
441	hudManualSetupMode->setPosition(position);
442	hudManualSetupMode->setCharacterSize(21);
443	hudManualSetupMode->setDataVariance(osg::Object::DYNAMIC);
444	geode->addDrawable( hudManualSetupMode );
445
446	position += delta;
447
448	hudVisualizationMode = new osgText::Text;
449	hudVisualizationMode->setFont(timesFont);
450	hudVisualizationMode->setPosition(position);
451	hudVisualizationMode->setCharacterSize(21);
452	hudVisualizationMode->setDataVariance(osg::Object::DYNAMIC);
453	geode->addDrawable( hudVisualizationMode );
454	}
455
456	updateHUD();
457	_distortionSet->getDistortionInternals()->addChild(geode, false); // = child #2. The definition is: child #0 = mesh, #1 = highlighter, #2 HUD
458	}
459
460	void DistortionManipulator::updateHUD()
461	{
462	switch(activeSetupMode)
463	{
464	case DISABLED : hudSetupMode->setText("Setup Mode : DISABLED"); break;
465	case MANUAL : hudSetupMode->setText("Setup Mode : MANUAL"); break;
466	case DELEGATED : hudSetupMode->setText("Setup Mode : DELEGATED"); break;
467	default: hudSetupMode->setText("");
468	};
469	switch(activeDistortionMode)
470	{
471	case MESH : hudDistortionMode->setText("Distortion Mode : MESH"); break;
472	case TEXCOORDINATES : hudDistortionMode->setText("Distortion Mode : TEXCOORDINATES"); break;
473	default: hudDistortionMode->setText("");
474	};
475	switch(activeManualSetupMode)
476	{
477	case DISTORTION : hudManualSetupMode->setText("Manual Setup Mode : DISTORTION"); break;
478	case BLENDING : hudManualSetupMode->setText("Manual Setup Mode : BLENDING"); break;
479	default: hudManualSetupMode->setText("");
480	};
481	switch(activeVisualizationMode)
482	{
483	case DISTORTION_MESH : hudVisualizationMode->setText("Visualization Mode : DISTORTION_MESH"); break;
484	case INTENSITY_MAP : hudVisualizationMode->setText("Visualization Mode : INTENSITY_MAP"); break;
485	case NONE : hudVisualizationMode->setText("Visualization Mode : NONE"); break;
486	default: hudVisualizationMode->setText("");
487	};
488	}
489
490	void DistortionManipulator::computeSelectedVertex( osgUtil::LineSegmentIntersector::Intersection& result )
491	{
492	osg::Geometry* geom = dynamic_cast<osg::Geometry*>( result.drawable.get() );
493	if ( !geom \|\| geom==_highlighter )
494	return;
495
496	osg::Vec3Array* vertices = dynamic_cast<osg::Vec3Array*>( geom->getVertexArray() );
497	osg::Vec3Array* selVertices = dynamic_cast<osg::Vec3Array*>( _highlighter->getVertexArray() );
498	if ( !vertices \|\| !selVertices )
499	return;
500
501	OSG_NOTIFY(osg::ALWAYS)<<"size of vertices="<<vertices->size()<<std::endl;
502	OSG_NOTIFY(osg::ALWAYS)<<"size of selVertices="<<selVertices->size()<<std::endl;
503
504	osg::Vec3 point = result.getWorldIntersectPoint();
505	osg::Matrix matrix = osg::computeLocalToWorld( result.nodePath ); // To compute the intersection vertices in world coordinates not in model coordinates
506	OSG_NOTIFY(osg::ALWAYS) << "Intersection-indices: Size=" << result.indexList.size() << std::endl;
507	const std::vector<unsigned int>& selIndices = result.indexList;
508	{
509	double maxRatio = 0.0;
510	int closestVertexIndex = 0;
511	for ( unsigned int i=0; i<3 && i<result.ratioList.size(); i++ ) //iterate through rations and search for maxRation=nearestVertex
512	{
513	if(result.ratioList[i] > maxRatio)
514	{
515	maxRatio = result.ratioList[i];
516	closestVertexIndex = result.indexList[i];
517	}
518	OSG_NOTIFY(osg::ALWAYS)<<"maxRatio="<<maxRatio<<std::endl;
519	OSG_NOTIFY(osg::ALWAYS)<<"closestVertexIndex="<<closestVertexIndex<<std::endl;
520	}
521	OSG_NOTIFY(osg::ALWAYS)<<"nearest vertex: X="<<(vertices)[closestVertexIndex].x()<<" Y="<<(vertices)[closestVertexIndex].y()<<" Z="<<(*vertices)[closestVertexIndex].z()<<std::endl;
522	osg::Vec3 vertex = (vertices)[closestVertexIndex] matrix;
523
524	selVertices->front() = vertex;
525	OSG_NOTIFY(osg::ALWAYS)<<"selected vertice: X="<<vertex.x()<<" Y="<<vertex.y()<<" Z="<<vertex.z()<<std::endl;
526	}
527	selVertices->dirty();
528	_highlighter->dirtyBound();
529	}
530
531	void DistortionManipulator::updateDistortionSetup()
532	{
533	updateDistortionViewProjectionMatrix();
534	updateDistortionMesh();
535	_distortionSet->clearDirtyStatus();
536	}
537
538	void DistortionManipulator::updateDistortionViewProjectionMatrix()
539	{
540	if(!_distortionSet.valid())
541	{
542	osg::notify(osg::WARN) << "You cannot update the distortion without distortion set up." << std::endl;
543	return;
544	}
545
546	if(_viewer.valid())
547	{
548
549	#if 0
550	/*******************************************/
551	/* ViewMatrix-Testcase "Garching" */
552	/*******************************************/
553
554	osg::Matrixd viewMatrix, transView, rotView;
555
556	rotView.makeRotate(
557	osg::DegreesToRadians(-130.0), osg::Vec3(0,1,0), // heading
558	osg::DegreesToRadians(0.0), osg::Vec3(-1,0,0), // pitch
559	osg::DegreesToRadians(-80.0), osg::Vec3(0,0,1)); // roll
560
561	transView.makeTranslate(-4169457.750.99975, -860009.060.99975, -4737192.00*0.99975);
562	viewMatrix = transView * rotView;
563
564	if(_viewer->getCameraManipulator()==NULL)
565	{
566	_viewer->getCamera()->setViewMatrix(viewMatrix);
567	}
568	else
569	{
570	_viewer->getCameraManipulator()->setByMatrix(osg::Matrixd::inverse(viewMatrix));
571	}
572
573	/**************************************/
574	/* End of Testcase "Garching" */
575	/**************************************/
576	#endif
577
578	_viewer->getCamera()->setProjectionMatrix(osg::Matrix::identity());
579
580	// Get slave struct to update the offset matrixes.
581	//Slave Camera Settings
582	osgViewer::View::Slave* sceneSlave = _viewer->findSlaveForCamera( _distortionSet->getSceneCamera() );
583	sceneSlave->_projectionOffset = _distortionSet->getProjectionOffset();
584	sceneSlave->_viewOffset = _distortionSet->getViewOffset();
585
586	#if 0
587	// Print viewMatrixAsLookAt for Debug Purposes
588	osg::Vec3 Eye,Focus,Vertical;
589	_viewer->getCamera()->getViewMatrixAsLookAt(Eye,Focus,Vertical);
590	OSG_ALWAYS<<"Camera Eyepoint: X:"<<Eye.x()<<" Y:"<<Eye.y()<<" Z:"<<Eye.z()<<std::endl;
591	printf("Camera Eyepoint: %.2lf, %.2lf, %.2lf\n",Eye.x(),Eye.y(),Eye.z());
592	printf("Camera Direction: %.2lf, %.2lf, %.2lf\n",Focus.x()-Eye.x(),Focus.y()-Eye.y(),Focus.z()-Eye.z());
593	printf("Vertical after: %.2lf, %.2lf, %.2lf\n",Vertical.x()-Eye.x(),Vertical.y()-Eye.y(),Vertical.z()-Eye.z());
594	#endif
595
596	}
597	else
598	OSG_NOTIFY(osg::ALWAYS)<<"Invalid viewer!"<<std::endl;
599	}
600
601	void DistortionManipulator::updateDistortionMesh()
602	{
603	// Create mesh vertex array & textCoord array
604	osg::Vec3Array* vertices = new osg::Vec3Array;
605	osg::Vec2Array* texCoords = new osg::Vec2Array;
606	osg::Vec4Array* colors = new osg::Vec4Array;
607
608	for(unsigned int i=0;i<_distortionSet->getDistortionMesh()->size();i++)
609	{
610	osg::Vec4 tmpSrc = _distortionSet->getDistortionMesh()->at(i);
611
612	// Scale vector with the screen resolution
613	osg::Vec3 tmpVec = osg::Vec3( tmpSrc.x()*_screenPixWidth,
614	tmpSrc.y()*_screenPixHeight,
615	0.0 );
616	vertices->push_back( tmpVec );
617	//OSG_ALWAYS<<"Transfering mesh vertex : X:"<<tmpVec.x()<<" Y:"<<tmpVec.y()<<" Z:"<<tmpVec.z()<<std::endl;
618
619	// Note texCoords
620	osg::Vec2 tmpTexCoords = osg::Vec2( tmpSrc.z()*_screenPixWidth,
621	tmpSrc.w()*_screenPixHeight );
622
623	texCoords->push_back(tmpTexCoords);
624	//OSG_ALWAYS<<"Transfering texcoord vertex : X:"<<tmpTexCoords.x()<<" Y:"<<tmpTexCoords.y()<<std::endl;
625
626	colors->push_back(osg::Vec4(1.0f,1.0f,1.0f,1.0f));
627	}
628
629	// --------- Apply Arrays -----------------
630	osg::Geometry* geom = _distortionSet->getDistortionInternals()->getChild(osgViewer::DistortionSet::MESH)->asGeode()->getDrawable(0)->asGeometry();
631
632	// If arraysize differs: Replace primite sets
633	bool regeneratePrimitiveSets = true;
634	osg::Vec3Array* prevArray = dynamic_cast<osg::Vec3Array*>(geom->getVertexArray());
635	if(prevArray)
636	regeneratePrimitiveSets = prevArray->size()!=vertices->size()?true:false;
637
638	// Set new Arrays
639	geom->setVertexArray(vertices);
640	geom->setTexCoordArray(0, texCoords); // todo: 0 ist hardcoded und müsste durch getTexUnitScene() ersetzt werden
641	geom->setColorArray(colors);
642
643	// If required: regenerate primitiveset:
644	OSG_ALWAYS<<"PrimitiveSet regeneration required!"<<std::endl;
645	geom->removePrimitiveSet(0, geom->getNumPrimitiveSets());
646	//// Create new primitives : Quads grid
647	unsigned int rows = _distortionSet->getDistortionMeshRows();
648	unsigned int columns = _distortionSet->getDistortionMeshColumns();
649	for ( unsigned int row=0; row<rows-1; row++ ) // each strip consists of two affected vertex rows, so we need only row-1 strips.
650	{
651	osg::ref_ptr<osg::DrawElementsUInt> de = new osg::DrawElementsUInt(osg::PrimitiveSet::QUAD_STRIP, columns2); // columns2 = number of involved vertices for this strip.
652	for ( unsigned int col=0; col<columns; col++ )
653	{
654	(de)[col2 + 0] = row*columns + col;
655	(de)[col2 + 1] = (row+1)*columns + col;
656	}
657	geom->addPrimitiveSet( de.get() );
658	}
659	}

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

Context Navigation

source: experimental/distortionNG/DistortionManipulator.cpp @ 427

Download in other formats: