Context Navigation

source: osgVisual/include/distortion/visual_distortion.h @ 155

Last change on this file since 155 was 151, checked in by Torben Dannhauer, 14 years ago

File size: 13.7 KB

Rev	Line
[32]	1	#pragma once
	2	/* --c++-- osgVisual - Copyright (C) 2009-2010 Torben Dannhauer
	3	*
	4	* This library is based on OpenSceneGraph, open source and may be redistributed and/or modified under
	5	* the terms of the OpenSceneGraph Public License (OSGPL) version 0.0 or
	6	* (at your option) any later version. The full license is in LICENSE file
	7	* included with this distribution, and on the openscenegraph.org website.
	8	*
	9	* osgVisual requires for some proprietary modules a license from the correspondig manufacturer.
	10	* You have to aquire licenses for all used proprietary modules.
	11	*
	12	* This library is distributed in the hope that it will be useful,
	13	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	14	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	15	* OpenSceneGraph Public License for more details.
	16	*/
	17
	18	#include <osg/NodeCallback>
	19	#include <osg/Node>
	20	#include <osg/Geode>
	21	#include <osg/Geometry>
	22	#include <osg/Texture2D>
	23	#include <osg/TextureRectangle>
	24	#include <osg/Notify>
	25	#include <osg/Referenced>
	26
	27	#include <osgViewer/Viewer>
	28
	29	#include <osgDB/ReadFile>
	30	#include <osgDB/FileUtils>
	31
	32	#include <osgGA/GUIEventHandler>
	33	#include <osgGA/GUIActionAdapter>
	34
[148]	35	#include <visual_util.h>
	36
[32]	37	#include <string>
	38	#include <iostream>
	39	#include <fstream>
	40
[148]	41	// XML Parser
	42	#include <stdio.h>
	43	#include <libxml/parser.h>
	44	#include <libxml/tree.h>
[32]	45
[148]	46
[32]	47	namespace osgVisual
	48	{
	49
	50
	51	//class CameraConfigParser;
	52
	53	/**
	54	* \brief This class distorts and blends the rendered image for projection in curved screens and multi channel setups.
	55	*
	56	* @todo Distortion algorithm seems to be erronous, small artefact lines are visible.
	57	*
	58	* @author Torben Dannhauer
	59	* @date Jul 2009
	60	*/
	61	class visual_distortion : public osg::Node
	62	{
[88]	63	#include <leakDetection.h>
[32]	64	public:
	65	/**
	66	* \brief Constructor
	67	*
	68	*/
[145]	69	visual_distortion(osgViewer::Viewer* viewer_, osg::ArgumentParser& arguments_, std::string configFileName);
[32]	70
	71	/**
	72	* \brief Destructor
	73	*
	74	*/
	75	virtual ~visual_distortion(void);
	76
	77	/**
[150]	78	* \brief This function parses the XML config file for distortion relevant parameters.
[32]	79	*
	80	* @return : True if parsing was successful.
	81	*/
[150]	82	bool processXMLConfiguration();
[32]	83
	84	/**
	85	* \brief This function initializes the distortion node and creates the distortion camera and scene graph.
	86	*
	87	* @param subgraph : Undistorted scene graph.
	88	* @param clearColor : Clear color for distortion.
[151]	89	* @return : Group node which contains the distortion and the undistorted scene graph ( distortedGraph ).
[32]	90	*/
	91	osg::Group* initialize(osg::Group* subgraph, const osg::Vec4& clearColor );
	92
	93	/**
	94	* \brief This function shuts the distortion module down.
	95	*
	96	*/
	97	void shutdown();
	98
	99	/**
	100	* \brief This function enabels or disables the distortion
	101	*/
	102	void toggleDistortion();
	103
	104	/**
	105	* \brief This functions returns a pointer to the scene camera, which renders the undistorted scene (PRE_RENDER).
	106	*
	107	* @return Pointer to the scene camera.
	108	*/
	109	osg::Camera* getSceneCamera() {return sceneCamera;};
	110
	111	/**
	112	* \brief This function returns the distortedSceneGraph if the distortion module is initialized.
	113	*
	114	* @return NULL if not Initialized, Otherwith the distorted sceneGraph.
	115	*/
	116	osg::Group* getDistortedSceneGraph() { if(initialized) return distortedGraph; else return NULL;};
	117
	118	/**
	119	* \brief The function returns the frustum values which are parsed by the nested cameraConfigParser class.
	120	*
	121	* @return : Vector of double values which represent the frustum dataset. The order is the standart glu order.
	122	*/
	123	std::vector<double> getFrustumDataset() {return parser->getFrustumDataset();};
	124
	125	/**
	126	* \brief This function returns the retation values for the rendering camera for XYZ axis in degree.
	127	*
	128	* @return : Vector of double values which represent the rotation values for XYZ axis in degree.
	129	*/
	130	std::vector<double> getRotationDataset() {return parser->getRotationDataset();};
	131
	132	/**
	133	* \brief This function returns the translation values for the rendering camera along the XYZ axis in meter.
	134	*
	135	* @return : Vector of double values which represent the translation values for XYZ axis in meter.
	136	*/
	137	std::vector<double> getTranslationDataset() {return parser->getTranslationDataset();};
	138
	139	private:
	140	/**
	141	* \brief Callback class for update callback.
	142	*
	143	* This nested class is used for updating the distortion node every frame by installing this class as updateCallback for the distortion node.
	144	* @author Torben Dannhauer
	145	* @date Jul 2009
	146	*/
	147	class distortionUpdateCallback : public osg::NodeCallback
	148	{
	149	public:
	150	/**
	151	* \brief Constructor, for setting the member variables.
	152	*
	153	* @param viewer_ : Pointer to the viewer.
	154	* @param sceneCamera_ : Pointer to the scene camera.
	155	*/
	156	distortionUpdateCallback(osgViewer::Viewer* viewer_, osg::Camera* sceneCamera_)
	157	: viewer(viewer_), sceneCamera(sceneCamera_) {};
	158
	159	/**
	160	* \brief This function is executed as callback during update traversal.
	161	*
	162	*/
	163	virtual void operator()(osg::Node* node, osg::NodeVisitor* nv);
	164	private:
	165	/**
	166	* Referenced Pointer to the applications viewer.
	167	*/
	168	osg::ref_ptr<osgViewer::Viewer> viewer;
	169
	170	/**
	171	* Referenced Pointer to the scene camera.
	172	*/
	173	osg::ref_ptr<osg::Camera> sceneCamera;
	174	};
	175
	176	/**
	177	* \brief This nested class is a keyboard event adapter which allows distortion toggle. It never blocks the key event, therefore multiple events on a single key are possible.
	178	*
	179	* @author Torben Dannhauer
	180	* @date Jul 2009
	181	*/
	182	class ToggleDistortionKBEventHandler : public osgGA::GUIEventHandler
	183	{
	184	public:
	185	/**
	186	* \brief Constructor
	187	*
	188	* @param distortion_ : Pointer to the distortion class.
	189	* @param viewer_ : Reference to the applications viewer.
	190	*/
	191	ToggleDistortionKBEventHandler(visual_distortion* distortion_): distortion(distortion_) {}
	192
	193	/**
	194	* \brief This function is executed if an keyboard event appears
	195	*
	196	* @param ea : GUI event adapter.
	197	* @param : GUI action adapter.
	198	* @return : Returns always false to pass the event to other event adapter if installed.
	199	*/
	200	virtual bool handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIActionAdapter&)
	201	{
	202	if ( ea.getEventType() == osgGA::GUIEventAdapter::KEYDOWN && ea.getKey() == 'd' )
	203	{
	204	distortion->toggleDistortion();
	205	return false;
	206	}
	207	else
	208	return false; // return false means: hand over the event to the next Eventhandler. True means: absorb the event.
	209	}
	210
	211	private:
	212	/**
	213	* Pointer to he distortion class. INside this EventHandler a Referenced Pointer ist _NOT_ used,
	214	* because a circulare reference (distortion->kBEventHandler, kbEventHandler->distortion) would block both pointer to unref.
	215	*/
	216	visual_distortion* distortion;
	217	};
	218
	219	/**
	220	* \brief This nested class reads the Projection Designer's project file and extracts the frustum, translation and rotation values
	221	*
	222	* @author Torben Dannhauer
	223	* @date Aug 2009
	224	*/
	225	class CameraConfigParser : public osg::Referenced
	226	{
	227	public:
	228	CameraConfigParser(void)
	229	{
	230	configParsed = false;
	231	frustumValues.clear();
	232	rotationValues.clear();
	233	rotationValues.resize(3,0);
	234	translationValues.clear();
	235	translationValues.resize(3,0);
	236	}
	237
	238	virtual ~CameraConfigParser(void) {};
	239
	240	bool parseConfigFile(const char * filename)
	241	{
	242	if (! osgDB::fileExists(filename) )
	243	{
	244	OSG_NOTIFY( osg::FATAL ) << "ERROR: File '" << filename << "' does not exist." << std::endl;
	245	return false;
	246	}
	247
	248	frustumValues.clear();
	249	rotationValues.clear();
	250	rotationValues.resize(3,0);
	251	translationValues.clear();
	252	translationValues.resize(3,0);
	253
	254	std::string line;
	255	std::ifstream configFile (filename);
	256
	257	if (configFile.is_open())
	258	{
	259	while (! configFile.eof() )
	260	{
	261	getline (configFile,line);
	262	//cout << line << endl;
	263
	264	// Auf Frustum untersuchen
	265	if( line.find("Frustum") != std::string::npos )
	266	{
	267	extractFrustum(line);
	268	}
	269
	270	// Auf Rotation untersuchen
	271	if( line.find("Rotate") != std::string::npos )
	272	{
	273	extractRotation(line);
	274	}
	275
	276	// Auf Translation untersuchens
	277	if( line.find("Translate") != std::string::npos )
	278	{
	279	extractTranslation(line);
	280	}
	281
	282	}
	283	configFile.close();
	284	configParsed = true;
	285	return true;
	286	}
	287	else
	288	{
	289	OSG_NOTIFY( osg::FATAL ) << "Unable to open file:" << filename << std::endl;
	290	frustumValues.clear();
	291	configParsed = false;
	292	return false;
	293	}
	294	}
	295
	296	bool extractFrustum(std::string data_)
	297	{
	298	data_.replace(0,16,""); // Delete leading spaces and "Frustum"
	299	data_.replace(data_.length()-1,1,""); // Delete trailing ";"
	300	for( int i=0; i<6; i++)
	301	{
	302	size_t pos = data_.find(" "); // Position of first value in string
	303	if( pos == -1) pos = data_.length();
	304	frustumValues.push_back( atof( data_.substr(0, pos).data() ) );
	305	data_.replace(0,pos+1,"");
	306	}
	307	return true;
	308	}
	309
	310	bool extractRotation(std::string data_)
	311	{
	312	data_.replace(0,15,""); // Delete leading spaces and "Rotate"
	313	data_.replace(data_.length()-1,1,""); // Delete trailing ";"
	314
	315	size_t valuePos = data_.find(" "); // First Value is RotationValue
	316	std::string axis = data_.substr( valuePos+1);
	317	switch(axis.find("1"))
	318	{
	319	case 0: rotationValues[0] = atof( data_.substr(0, valuePos).data() ); // X-Axis
	320	break;
	321	case 2: rotationValues[1] = atof( data_.substr(0, valuePos).data() ); // Y-Axis
	322	break;
	323	case 4: rotationValues[2] = atof( data_.substr(0, valuePos).data() ); // Z-Axis
	324	break;
	325	};
	326	return true;
	327	}
	328
	329	bool extractTranslation(std::string data_)
	330	{
	331	data_.replace(0,18,""); // Delete leading spaces and "Translate"
	332	data_.replace(data_.length()-1,1,""); // Delete trailing ";"
	333	for( unsigned int i=0;i<translationValues.size(); i++)
	334	{
	335	size_t valuePos = data_.find(" ");
	336	std::string value = data_.substr(0, valuePos);
	337	data_.replace(0,valuePos+1,"");
	338	translationValues[i] = atof( value.data() );
	339	}
	340	return true;
	341	}
	342
	343	std::vector<double> getFrustumDataset() {return(frustumValues);}
	344
	345	std::vector<double> getRotationDataset() {return(rotationValues);}
	346
	347	std::vector<double> getTranslationDataset() {return(translationValues);}
	348
	349	bool isConfigParsed() {return configParsed;};
	350	private:
	351	bool configParsed;
	352	std::vector<double> frustumValues;
	353	std::vector<double> rotationValues;
	354	std::vector<double> translationValues;
	355	};
	356
	357
	358
	359	/**
	360	* Updatecallback as member variable of this class.
	361	*/
	362	osg::ref_ptr<distortionUpdateCallback> updateCallback;
	363
	364	/**
	365	* Keyboard Eventhandler for distortion toggle.
	366	*/
	367	//osg::ref_ptr<ToggleDistortionKBEventHandler> kBEventHandler;
	368	ToggleDistortionKBEventHandler* kBEventHandler;
	369
	370	/**
	371	* \brief This functions creates the distortion environment.
	372	*
	373	* @param subgraph : Undistorted scene graph to distort.
	374	* @param clearColor : OpenGL clear color to use for the distortion.
	375	* @return : Group node which contains the distortion and the undistorted scene graph.
	376	*/
	377	osg::Group* createPreRenderSubGraph(osg::Group* subgraph, const osg::Vec4& clearColor );
	378
	379	/**
	380	* \brief This function loads a shader source file
	381	*
	382	* @param shader : Shader to load source in
	383	* @param fileName : Filename of shader source file.
	384	* @return : Returns true if successful.
	385	*/
	386	bool loadShaderSource( osg::Shader* shader, const std::string& fileName );
	387
	388	/**
	389	* \brief This function loads a shader Texture file
	390	*
	391	* @param fileName : Texture file to load.
	392	* @return : Returns the loaded Texture.
	393	*/
	394	osg::Texture* loadTexture( const std::string& fileName );
	395
	396	/**
	397	* Referenced Pointer to the loaded distortion Map
	398	*/
	399	osg::ref_ptr<osg::Texture> distortMapTexture;
	400
	401	/**
	402	* Referenced Pointer to the loaded blend map
	403	*/
	404	osg::ref_ptr<osg::Texture> blendMapTexture;
	405
	406
	407	/**
	408	* Flag which indicates the initialization status
	409	*/
	410	bool initialized;
	411
	412	/**
	413	* Flag that indicates if distortion is actually enabled. This is imported for the toggleDistortion() function.
	414	*/
	415	bool distortionEnabled;
	416
	417	/**
	418	* Reference pointer to the distorted scene graph.
	419	*/
	420	osg::ref_ptr<osg::Group> distortedGraph;
	421
	422	/**
	423	* Reference pointer to the undistorted scene graph.
	424	*/
	425	osg::ref_ptr<osg::Group> cleanGraph;
	426
	427	/**
	428	* Reference pointer to the scen camera (PRE_RENDER)
	429	*/
	430	osg::ref_ptr<osg::Camera> sceneCamera;
	431
	432	/**
	433	* Referenced pointer to the main applications viewer
	434	*/
	435	osg::ref_ptr<osgViewer::Viewer> viewer;
	436
	437	/**
	438	* Parser object for parsing the camera configuration file which controlls frustum and camera PAT
	439	*/
	440	osg::ref_ptr<CameraConfigParser> parser;
	441
	442	/**
	443	* Texture width for render to texture (RTT).
	444	*/
	445	unsigned tex_width;
	446
	447	/**
	448	* Texture height for using render to texture (RTT).
	449	*/
	450	unsigned tex_height;
	451
	452	/**
	453	* Indicates of texture rectangle should be used.
	454	*/
	455	bool useTextureRectangle;
	456
	457	/**
	458	* Indicates if shader distortion should be used.
	459	*/
	460	bool useShaderDistortion;
	461
	462	/**
	463	* Indicates if high definition rendering shold be used.
	464	*/
	465	bool useHDR;
	466
	467	/**
	468	* Filename of the distorion map.
	469	*/
	470	std::string distortMapFileName;
	471
	472	/**
	473	* Filename of the blend map.
	474	*/
	475	std::string blendMapFileName;
	476
	477	/**
[148]	478	* XML config filename
	479	*/
	480	std::string configFileName;
	481
	482	/**
[32]	483	* Target implementation which should be used to distort.
	484	*/
	485	osg::Camera::RenderTargetImplementation renderImplementation;
	486
	487	/**
	488	* Reference to the global ArgumentParser.
	489	*/
	490	osg::ArgumentParser& arguments;
	491	};
	492
	493	} // END NAMESPACE

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

Download in other formats: