From 00e1af19021ee78d44624719bff0133ea6dc33be Mon Sep 17 00:00:00 2001
From: Robert Osfield <robert@openscenegraph.com>
Date: Tue, 8 May 2007 12:18:57 +0000
Subject: [PATCH] Firt cut at full dome correction of wrap around movies and
 imagery

---
 examples/osgmovie/osgmovie.cpp | 350 +++++++++++++++++++++++++++------
 1 file changed, 288 insertions(+), 62 deletions(-)

diff --git a/examples/osgmovie/osgmovie.cpp b/examples/osgmovie/osgmovie.cpp
index 764f2325e..5a1b44e81 100644
--- a/examples/osgmovie/osgmovie.cpp
+++ b/examples/osgmovie/osgmovie.cpp
@@ -10,6 +10,7 @@
 #include <osg/Material>
 #include <osg/Texture2D>
 #include <osg/TextureRectangle>
+#include <osg/TextureCubeMap>
 #include <osg/TexMat>
 #include <osg/CullFace>
 #include <osg/ImageStream>
@@ -25,7 +26,10 @@ class MovieEventHandler : public osgGA::GUIEventHandler
 {
 public:
 
-    MovieEventHandler() {}
+    MovieEventHandler():_trackMouse(false) {}
+    
+    void setMouseTracking(bool track) { _trackMouse = track; }
+    bool getMouseTracking() const { return _trackMouse; }
     
     void set(osg::Node* node);
 
@@ -91,6 +95,7 @@ protected:
     };
 
 
+    bool            _trackMouse;
     ImageStreamList _imageStreamList;
     
 };
@@ -116,57 +121,59 @@ bool MovieEventHandler::handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIAction
         case(osgGA::GUIEventAdapter::PUSH):
         case(osgGA::GUIEventAdapter::RELEASE):
         {
-
-            osgViewer::View* view = dynamic_cast<osgViewer::View*>(&aa);
-            osgUtil::LineSegmentIntersector::Intersections intersections;
-            if (view && view->computeIntersections(ea.getX(), ea.getY(), nv->getNodePath(), intersections))
+            if (_trackMouse)
             {
-
-                // use the nearest intersection                 
-                const osgUtil::LineSegmentIntersector::Intersection& intersection = *(intersections.begin());
-                osg::Drawable* drawable = intersection.drawable.get();
-                osg::Geometry* geometry = drawable ? drawable->asGeometry() : 0;
-                osg::Vec3Array* vertices = geometry ? dynamic_cast<osg::Vec3Array*>(geometry->getVertexArray()) : 0;
-                if (vertices)
+                osgViewer::View* view = dynamic_cast<osgViewer::View*>(&aa);
+                osgUtil::LineSegmentIntersector::Intersections intersections;
+                if (view && view->computeIntersections(ea.getX(), ea.getY(), nv->getNodePath(), intersections))
                 {
-                    // get the vertex indices.
-                    const osgUtil::LineSegmentIntersector::Intersection::IndexList& indices = intersection.indexList;
-                    const osgUtil::LineSegmentIntersector::Intersection::RatioList& ratios = intersection.ratioList;
 
-                    if (indices.size()==3 && ratios.size()==3)
+                    // use the nearest intersection                 
+                    const osgUtil::LineSegmentIntersector::Intersection& intersection = *(intersections.begin());
+                    osg::Drawable* drawable = intersection.drawable.get();
+                    osg::Geometry* geometry = drawable ? drawable->asGeometry() : 0;
+                    osg::Vec3Array* vertices = geometry ? dynamic_cast<osg::Vec3Array*>(geometry->getVertexArray()) : 0;
+                    if (vertices)
                     {
-                        unsigned int i1 = indices[0];
-                        unsigned int i2 = indices[1];
-                        unsigned int i3 = indices[2];
+                        // get the vertex indices.
+                        const osgUtil::LineSegmentIntersector::Intersection::IndexList& indices = intersection.indexList;
+                        const osgUtil::LineSegmentIntersector::Intersection::RatioList& ratios = intersection.ratioList;
 
-                        float r1 = ratios[0];
-                        float r2 = ratios[1];
-                        float r3 = ratios[2];
-
-                        osg::Array* texcoords = (geometry->getNumTexCoordArrays()>0) ? geometry->getTexCoordArray(0) : 0;
-                        osg::Vec2Array* texcoords_Vec2Array = dynamic_cast<osg::Vec2Array*>(texcoords);
-                        if (texcoords_Vec2Array)
+                        if (indices.size()==3 && ratios.size()==3)
                         {
-                            // we have tex coord array so now we can compute the final tex coord at the point of intersection.                                
-                            osg::Vec2 tc1 = (*texcoords_Vec2Array)[i1];
-                            osg::Vec2 tc2 = (*texcoords_Vec2Array)[i2];
-                            osg::Vec2 tc3 = (*texcoords_Vec2Array)[i3];
-                            osg::Vec2 tc = tc1*r1 + tc2*r2 + tc3*r3;
+                            unsigned int i1 = indices[0];
+                            unsigned int i2 = indices[1];
+                            unsigned int i3 = indices[2];
 
-                            osg::notify(osg::NOTICE)<<"We hit tex coords "<<tc<<std::endl;
+                            float r1 = ratios[0];
+                            float r2 = ratios[1];
+                            float r3 = ratios[2];
 
+                            osg::Array* texcoords = (geometry->getNumTexCoordArrays()>0) ? geometry->getTexCoordArray(0) : 0;
+                            osg::Vec2Array* texcoords_Vec2Array = dynamic_cast<osg::Vec2Array*>(texcoords);
+                            if (texcoords_Vec2Array)
+                            {
+                                // we have tex coord array so now we can compute the final tex coord at the point of intersection.                                
+                                osg::Vec2 tc1 = (*texcoords_Vec2Array)[i1];
+                                osg::Vec2 tc2 = (*texcoords_Vec2Array)[i2];
+                                osg::Vec2 tc3 = (*texcoords_Vec2Array)[i3];
+                                osg::Vec2 tc = tc1*r1 + tc2*r2 + tc3*r3;
+
+                                osg::notify(osg::NOTICE)<<"We hit tex coords "<<tc<<std::endl;
+
+                            }
+                        }
+                        else
+                        {
+                            osg::notify(osg::NOTICE)<<"Intersection has insufficient indices to work with";
                         }
-                    }
-                    else
-                    {
-                        osg::notify(osg::NOTICE)<<"Intersection has insufficient indices to work with";
-                    }
 
+                    }
+                }
+                else
+                {
+                    osg::notify(osg::NOTICE)<<"No intersection"<<std::endl;
                 }
-            }
-            else
-            {
-                osg::notify(osg::NOTICE)<<"No intersection"<<std::endl;
             }
             break;
         }
@@ -276,6 +283,217 @@ osg::Geometry* myCreateTexturedQuadGeometry(const osg::Vec3& pos,float width,flo
     }
 }
 
+osg::Geometry* createDomeDistortionMesh(const osg::Vec3& origin, const osg::Vec3& widthVector, const osg::Vec3& heightVector,
+                                        osg::ArgumentParser& arguments)
+{
+    double sphere_radius = 1.0;
+    if (arguments.read("--radius", sphere_radius)) {}
+
+    double collar_radius = 0.45;
+    if (arguments.read("--collar", collar_radius)) {}
+
+    osg::Vec3d center(0.0,0.0,0.0);
+    osg::Vec3d eye(0.0,0.0,0.0);
+    
+    double distance = sqrt(sphere_radius*sphere_radius - collar_radius*collar_radius);
+    if (arguments.read("--distance", distance)) {}
+    
+    bool centerProjection = false;
+
+    osg::Vec3d projector = eye - osg::Vec3d(0.0,0.0, distance);
+    
+    
+    osg::notify(osg::NOTICE)<<"Projector position = "<<projector<<std::endl;
+    osg::notify(osg::NOTICE)<<"distance = "<<distance<<std::endl;
+
+
+    // create the quad to visualize.
+    osg::Geometry* geometry = new osg::Geometry();
+
+    geometry->setSupportsDisplayList(false);
+
+    osg::Vec3 xAxis(widthVector);
+    float width = widthVector.length();
+    xAxis /= width;
+
+    osg::Vec3 yAxis(heightVector);
+    float height = heightVector.length();
+    yAxis /= height;
+    
+    int noSteps = 50;
+
+    osg::Vec3Array* vertices = new osg::Vec3Array;
+    osg::Vec2Array* texcoords = new osg::Vec2Array;
+    osg::Vec4Array* colors = new osg::Vec4Array;
+
+    osg::Vec3 bottom = origin;
+    osg::Vec3 dx = xAxis*(width/((float)(noSteps-1)));
+    osg::Vec3 dy = yAxis*(height/((float)(noSteps-1)));
+    
+    osg::Vec3d screenCenter = origin + widthVector*0.5f + heightVector*0.5f;
+    float screenRadius = heightVector.length() * 0.5f;
+
+    osg::Vec3 cursor = bottom;
+    int i,j;
+    
+    for(i=0;i<noSteps;++i)
+    {
+        osg::Vec3 cursor = bottom+dy*(float)i;
+        for(j=0;j<noSteps;++j)
+        {
+            osg::Vec2 delta(cursor.x() - screenCenter.x(), cursor.y() - screenCenter.y());
+            double theta = atan2(delta.x(), delta.y());
+            if (theta<0.0) theta += 2*osg::PI;
+            double phi = osg::PI_2 * delta.length() / screenRadius;
+            if (phi > osg::PI_2) phi = osg::PI_2;
+
+            double f = distance * sin(phi);
+            double e = distance * cos(phi) + sqrt( sphere_radius*sphere_radius - f*f);
+            double l = e * cos(phi);
+            double h = e * sin(phi);
+            double gamma = atan2(h, l-distance);
+
+            osg::Vec2 texcoord(theta/(2.0*osg::PI), 1.0-gamma/osg::PI);
+
+            osg::notify(osg::NOTICE)<<"cursor = "<<cursor<< " theta = "<<theta<< "phi="<<phi<<" gamma = "<<gamma<<" texcoord="<<texcoord<<std::endl;
+
+            vertices->push_back(cursor);
+            colors->push_back(osg::Vec4(1.0f,1.0f,1.0f,1.0f));
+            texcoords->push_back(texcoord);
+
+            cursor += dx;
+        }
+        // osg::notify(osg::NOTICE)<<std::endl;
+    }
+    
+    // pass the created vertex array to the points geometry object.
+    geometry->setVertexArray(vertices);
+
+    geometry->setColorArray(colors);
+    geometry->setColorBinding(osg::Geometry::BIND_PER_VERTEX);
+
+    geometry->setTexCoordArray(0,texcoords);
+
+    for(i=0;i<noSteps-1;++i)
+    {
+        osg::DrawElementsUShort* elements = new osg::DrawElementsUShort(osg::PrimitiveSet::QUAD_STRIP);
+        for(j=0;j<noSteps;++j)
+        {
+            elements->push_back(j+(i+1)*noSteps);
+            elements->push_back(j+(i)*noSteps);
+        }
+        geometry->addPrimitiveSet(elements);
+    }
+    
+    return geometry;
+}
+
+
+void setDomeCorrection(osgViewer::Viewer& viewer, osg::ArgumentParser& arguments)
+{
+    // enforce single threading right now to avoid double buffering of RTT texture
+    viewer.setThreadingModel(osgViewer::Viewer::SingleThreaded);
+ 
+    osg::GraphicsContext::WindowingSystemInterface* wsi = osg::GraphicsContext::getWindowingSystemInterface();
+    if (!wsi) 
+    {
+        osg::notify(osg::NOTICE)<<"Error, no WindowSystemInterface available, cannot create windows."<<std::endl;
+        return;
+    }
+
+    unsigned int width, height;
+    wsi->getScreenResolution(osg::GraphicsContext::ScreenIdentifier(0), width, height);
+
+    while (arguments.read("--width",width)) {}
+    while (arguments.read("--height",height)) {}
+
+    osg::ref_ptr<osg::GraphicsContext::Traits> traits = new osg::GraphicsContext::Traits;
+    traits->x = 0;
+    traits->y = 0;
+    traits->width = width;
+    traits->height = height;
+    traits->windowDecoration = false;
+    traits->doubleBuffer = true;
+    traits->sharedContext = 0;
+    
+    
+
+    osg::ref_ptr<osg::GraphicsContext> gc = osg::GraphicsContext::createGraphicsContext(traits.get());
+    if (!gc)
+    {
+        osg::notify(osg::NOTICE)<<"GraphicsWindow has not been created successfully."<<std::endl;
+        return;
+    }
+
+    osg::ref_ptr<osg::Drawable> distortionCorrectionMash = createDomeDistortionMesh(osg::Vec3(0.0f,0.0f,0.0f), osg::Vec3(width,0.0f,0.0f), osg::Vec3(0.0f,height,0.0f), arguments);
+
+    osg::Texture* texture = 0;
+    for(int i=1;i<arguments.argc() && !texture;++i)
+    {
+        if (arguments.isString(i))
+        {
+            osg::Image* image = osgDB::readImageFile(arguments[i]);
+            osg::ImageStream* imagestream = dynamic_cast<osg::ImageStream*>(image);
+            if (imagestream) imagestream->play();
+
+            if (image)
+            {
+#if 0            
+                texture = new osg::TextureRectangle(image);
+#else
+                texture = new osg::Texture2D(image);
+#endif
+            }
+        }
+    }
+    
+    if (!texture)
+    {
+        return;
+    }
+
+    // distortion correction set up.
+    {
+        osg::Geode* geode = new osg::Geode();
+        geode->addDrawable(distortionCorrectionMash.get());
+
+        // new we need to add the texture to the mesh, we do so by creating a 
+        // StateSet to contain the Texture StateAttribute.
+        osg::StateSet* stateset = geode->getOrCreateStateSet();
+        stateset->setTextureAttributeAndModes(0, texture, osg::StateAttribute::ON);
+        stateset->setMode(GL_LIGHTING,osg::StateAttribute::OFF);
+#if 0        
+        osg::TexMat* texmat = new osg::TexMat;
+        texmat->setScaleByTextureRectangleSize(true);
+        stateset->setTextureAttributeAndModes(0, texmat, osg::StateAttribute::ON);
+#endif
+        osg::ref_ptr<osg::Camera> camera = new osg::Camera;
+        camera->setGraphicsContext(gc.get());
+        camera->setClearMask(GL_DEPTH_BUFFER_BIT | GL_COLOR_BUFFER_BIT );
+        camera->setClearColor( osg::Vec4(0.1,0.1,1.0,1.0) );
+        camera->setViewport(new osg::Viewport(0, 0, width, height));
+        GLenum buffer = traits->doubleBuffer ? GL_BACK : GL_FRONT;
+        camera->setDrawBuffer(buffer);
+        camera->setReadBuffer(buffer);
+        camera->setReferenceFrame(osg::Camera::ABSOLUTE_RF);
+        camera->setAllowEventFocus(false);
+        //camera->setInheritanceMask(camera->getInheritanceMask() & ~osg::CullSettings::CLEAR_COLOR & ~osg::CullSettings::COMPUTE_NEAR_FAR_MODE);
+        //camera->setComputeNearFarMode(osg::CullSettings::DO_NOT_COMPUTE_NEAR_FAR);
+        
+        camera->setProjectionMatrixAsOrtho2D(0,width,0,height);
+        camera->setViewMatrix(osg::Matrix::identity());
+
+        // add subgraph to render
+        camera->addChild(geode);
+        
+        camera->setName("DistortionCorrectionCamera");
+
+        viewer.addSlave(camera.get(), osg::Matrixd(), osg::Matrixd(), false);
+    }
+    
+    viewer.getCamera()->setNearFarRatio(0.0001f);
+}
+
 int main(int argc, char** argv)
 {
     // use an ArgumentParser object to manage the program arguments.
@@ -288,6 +506,7 @@ int main(int argc, char** argv)
     arguments.getApplicationUsage()->addCommandLineOption("-h or --help","Display this information");
     arguments.getApplicationUsage()->addCommandLineOption("--texture2D","Use Texture2D rather than TextureRectangle.");
     arguments.getApplicationUsage()->addCommandLineOption("--shader","Use shaders to post process the video.");
+    arguments.getApplicationUsage()->addCommandLineOption("--dome","Use full dome distortion correction.");
     
     bool useTextureRectangle = true;
     bool useShader = false;
@@ -350,38 +569,44 @@ int main(int argc, char** argv)
     }
 
 
-    for(int i=1;i<arguments.argc();++i)
+    if (arguments.read("--dome") || arguments.read("--puffer") )
+    {    
+        setDomeCorrection(viewer, arguments);
+    }
+    else
     {
-        if (arguments.isString(i))
+        for(int i=1;i<arguments.argc();++i)
         {
-            osg::Image* image = osgDB::readImageFile(arguments[i]);
-            osg::ImageStream* imagestream = dynamic_cast<osg::ImageStream*>(image);
-            if (imagestream) imagestream->play();
-            
-            if (image)
+            if (arguments.isString(i))
             {
-                geode->addDrawable(myCreateTexturedQuadGeometry(pos,image->s(),image->t(),image, useTextureRectangle));
-                
-                pos.z() += image->t()*1.5f;
+                osg::Image* image = osgDB::readImageFile(arguments[i]);
+                osg::ImageStream* imagestream = dynamic_cast<osg::ImageStream*>(image);
+                if (imagestream) imagestream->play();
+
+                if (image)
+                {
+                    geode->addDrawable(myCreateTexturedQuadGeometry(pos,image->s(),image->t(),image, useTextureRectangle));
+
+                    pos.z() += image->t()*1.5f;
+                }
+                else
+                {
+                    std::cout<<"Unable to read file "<<arguments[i]<<std::endl;
+                }            
             }
-            else
-            {
-                std::cout<<"Unable to read file "<<arguments[i]<<std::endl;
-            }            
+        }
+        if (!viewer.getSceneData())
+        {
+            arguments.getApplicationUsage()->write(std::cout);
+            return 1;
         }
     }
     
-    if (geode->getNumDrawables()==0)
-    {
-        // nothing loaded.
-        arguments.getApplicationUsage()->write(std::cout);
-        return 1;
-    }
 
     // pass the model to the MovieEventHandler so it can pick out ImageStream's to manipulate.
     MovieEventHandler* meh = new MovieEventHandler();
     geode->setEventCallback(meh);
-    meh->set(geode.get());
+    meh->set(viewer.getSceneData());
 
     // report any errors if they have occured when parsing the program aguments.
     if (arguments.errors())
@@ -396,6 +621,7 @@ int main(int argc, char** argv)
         return 1;
     }
 
+
     // set the scene to render
     viewer.setSceneData(geode.get());