Changeset 350 for experimental/distortionNG/extViewer.cpp

Timestamp:

Apr 21, 2012, 4:53:52 PM (12 years ago)

Author:

Torben Dannhauer

Message:

 

File:

• experimental/distortionNG/extViewer.cpp (modified) (9 diffs)

experimental/distortionNG/extViewer.cpp

@@ -57 +57 @@
 }
 
-//static osg::Texture2D* loadTexture( const std::string& fileName )
-//{
-//    std::string foundFileName = osgDB::findDataFile(fileName);
-//    if (foundFileName.length() != 0 )
-//    {
-//        // load distortion map texture file
-//        osg::Image* image = osgDB::readImageFile(foundFileName);
-//        if (image)
-//        {
-//                      osg::Texture2D* texture = new osg::Texture2D;
-//            texture->setImage(image);
-//                      texture->setFilter(osg::Texture::MIN_FILTER,osg::Texture::LINEAR);
-//                      texture->setFilter(osg::Texture::MAG_FILTER,osg::Texture::LINEAR);
-//                      texture->setWrap(osg::Texture::WRAP_S,osg::Texture::CLAMP_TO_EDGE);
-//                      texture->setWrap(osg::Texture::WRAP_T,osg::Texture::CLAMP_TO_EDGE);
-//            return texture;
-//        }
-//    }
-//
-//    OSG_NOTIFY(osg::WARN) << "File \"" << fileName << "\" not found." << std::endl;
-//
-//    return NULL;
-//}
-
-//static osg::TextureRectangle* loadTextureRect( const std::string& fileName )
-//{
-//    std::string foundFileName = osgDB::findDataFile(fileName);
-//    if (foundFileName.length() != 0 )
-//    {
-//        // load distortion map texture file
-//        osg::Image* image = osgDB::readImageFile(foundFileName);
-//        if (image)
-//        {
-//                      osg::TextureRectangle* texture = new osg::TextureRectangle;
-//            texture->setImage(image);
-//            texture->setFilter(osg::Texture2D::MIN_FILTER, osg::Texture2D::LINEAR);
-//            texture->setFilter(osg::Texture2D::MAG_FILTER, osg::Texture2D::LINEAR);
-//            texture->setWrap(osg::Texture2D::WRAP_S, osg::Texture2D::CLAMP_TO_EDGE);
-//            texture->setWrap(osg::Texture2D::WRAP_T, osg::Texture2D::CLAMP_TO_EDGE);
-//            return texture;
-//        }
-//    }
-//
-//    OSG_NOTIFY(osg::WARN) << "File \"" << fileName << "\" not found." << std::endl;
-//
-//    return NULL;
-//}
-
 static osg::Geometry* createMesh(const osg::Vec3& origin, const osg::Vec3& widthVector, const osg::Vec3& heightVector, unsigned int columns, unsigned int rows, osg::Image* intensityMap, bool applyIntensityMapAsColours, const osg::Matrix& projectorMatrix)
 {
@@ -199 +151 @@
 }
 
+
 void extViewer::setUpViewForManualDistortion(unsigned int screenNum, osg::Image* intensityMap, const osg::Matrixd& projectorMatrix)
 {
-        OSG_INFO<<"View::setUpViewForManualDistortion(sn="<<screenNum<<", im="<<intensityMap<<")"<<std::endl;
+        OSG_INFO<<"View::setUpViewForManualDistortion(sn="<<screenNum<<")"<<std::endl;
+
+        // Create DistortionSet
+        _distortionSet = new osgViewer::DistortionSet();
+        _distortionSet->setIntensityMap( intensityMap );
+        _distortionSet->setDistortionMeshRows( 20 );
+        _distortionSet->setDistortionMeshColumns( 20 );
+
+
+
 
     osg::GraphicsContext::WindowingSystemInterface* wsi = osg::GraphicsContext::getWindowingSystemInterface();
@@ -233 +195 @@
     traits->sharedContext = 0;
 
-        bool applyIntensityMapAsColours = false;
-
         osg::ref_ptr<osg::GraphicsContext> gc = osg::GraphicsContext::createGraphicsContext(traits.get());
     if (!gc)
@@ -302 +262 @@
         // StateSet to contain the Texture StateAttribute.
             osg::StateSet* stateset = geode->getOrCreateStateSet();
-        stateset->setTextureAttributeAndModes(0, sceneTexture,osg::StateAttribute::ON);
+        stateset->setTextureAttributeAndModes(_distortionSet->getTexUnitScene(), sceneTexture,osg::StateAttribute::ON);
         stateset->setMode(GL_LIGHTING,osg::StateAttribute::OFF);
 
@@ -309 +269 @@
         stateset->setTextureAttributeAndModes(0, texmat, osg::StateAttribute::ON);
 
-                // If the intensityMap is used but not applyIntensityMapAsColours: Apply intensityMap as intensityMapTexture on unit 1
-                if (!applyIntensityMapAsColours && intensityMap)
-                        setUpIntensityMapBlending(stateset, intensityMap, screenNum, 0, 1);     
-
-                osg::Geometry* distortionMesh = createMesh(osg::Vec3(0.0f,0.0f,0.0f), osg::Vec3(width,0.0f,0.0f), osg::Vec3(0.0f,height,0.0f), 20, 20, intensityMap, applyIntensityMapAsColours, projectorMatrix);
+                setUpIntensityMapBlending(_distortionSet, stateset, screenNum);
+
+                osg::Geometry* distortionMesh = createMesh(osg::Vec3(0.0f,0.0f,0.0f), osg::Vec3(width,0.0f,0.0f), osg::Vec3(0.0f,height,0.0f), 20, 20, _distortionSet->getIntensityMap(), false, projectorMatrix);
                 geode->addDrawable(distortionMesh);
 
@@ -347 +305 @@
                 camera->setCullingMode( mode & (~osg::CullSettings::SMALL_FEATURE_CULLING) );
 
-                // Create DistortionSet
-                _distortionSet = new osgViewer::DistortionSet();
-                _distortionSet->setIntensityMap( intensityMap );
-
-
                 // Add the distortionHandler
                 osgViewer::DistortionManipulator* distManip = new osgViewer::DistortionManipulator();
@@ -364 +317 @@
 }
 
-void extViewer::setUpIntensityMapBlending(osg::StateSet* stateset, osg::Image* intensityMap, unsigned int screenNum, int rttSceneTextureUnit, int intensityMapTextureUnit)
-{
-        if(stateset == NULL || intensityMap == NULL)
+void extViewer::setUpIntensityMapBlending(osgViewer::DistortionSet* distSet, osg::StateSet* stateset, unsigned int screenNum)
+{
+        osg::Image* intensityMap = _distortionSet->getIntensityMap();
+        if(intensityMap == NULL)        return;
+
+        if(stateset == NULL)
         {
                 OSG_NOTICE<<"Error, no intensityMap or stateset for intensityMapBlending specified."<<std::endl;
@@ -400 +356 @@
         intensityMapTexture->setWrap(osg::Texture::WRAP_T,osg::Texture::CLAMP_TO_EDGE);
 
-    stateset->setTextureAttributeAndModes(intensityMapTextureUnit, intensityMapTexture, osg::StateAttribute::ON);
+    stateset->setTextureAttributeAndModes(_distortionSet->getTexUnitIntensityMap(), intensityMapTexture, osg::StateAttribute::ON);
 
         // create shaders for blending
@@ -411 +367 @@
         {
                 distortProgram->addShader( fShader );
-                stateset->addUniform( new osg::Uniform("sceneTexture", rttSceneTextureUnit) );
-                stateset->addUniform( new osg::Uniform("intensityMapTexture", intensityMapTextureUnit) );
+                //stateset->addUniform( new osg::Uniform("sceneTexture", _distortionSet->getTexUnitScene()) );
+                //stateset->addUniform( new osg::Uniform("intensityMapTexture", _distortionSet->getTexUnitIntensityMap()) );
+                stateset->addUniform( new osg::Uniform("sceneTexture", 0 ) );
+                stateset->addUniform( new osg::Uniform("intensityMapTexture", 1 ) );
                 stateset->setAttributeAndModes(distortProgram, osg::StateAttribute::ON);
         }
 }
+
+void extViewer::setUpIntensityMapBlending(std::string intensityMap)
+{
+        // Try to load intensityMap
+        osg::Image* intMap = osgDB::readImageFile(intensityMap);
+        if (!intMap)
+        {
+                osg::notify(osg::WARN) << "Quitting, couldn't find intensity map: " << intensityMap << std::endl;
+                return;
+        } 
+
+        // Create DistortionSet
+        _distortionSet = new osgViewer::DistortionSet();
+        _distortionSet->setIntensityMap( intMap );
+        _distortionSet->setDistortionMeshRows( 1 );
+        _distortionSet->setDistortionMeshColumns( 1 );
+        _distortionSet->setTexUnitScene( 0 );
+        _distortionSet->setTexUnitIntensityMap( 1 );
+
+        // Todo: call setupDistortion/Blendung to perform the blending
+}