source: osgVisual/src/dataIO/visual_dataIO.cpp @ 81

/* -*-c++-*- osgVisual - Copyright (C) 2009-2010 Torben Dannhauer
 *
 * This library is based on OpenSceneGraph, open source and may be redistributed and/or modified under 
 * the terms of the OpenSceneGraph Public License (OSGPL) version 0.0 or
 * (at your option) any later version.  The full license is in LICENSE file
 * included with this distribution, and on the openscenegraph.org website.
 *
 * osgVisual requires for some proprietary modules a license from the correspondig manufacturer.
 * You have to aquire licenses for all used proprietary modules.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * OpenSceneGraph Public License for more details.
*/

#include "visual_dataIO.h"

using namespace osgVisual;

visual_dataIO::visual_dataIO()
{
        OSG_NOTIFY( osg::ALWAYS ) << "visual_dataIO constructed" << std::endl;

        initialized = false;
}

visual_dataIO::~visual_dataIO()
{
        OSG_NOTIFY( osg::ALWAYS ) << "visual_dataIO destructed" << std::endl;
}

visual_dataIO* visual_dataIO::getInstance()
{
        static visual_dataIO instance; 
        return &instance; 
};

void visual_dataIO::init(osgViewer::Viewer* viewer_, osg::ArgumentParser& arguments_)
{
        OSG_NOTIFY( osg::ALWAYS ) << "visual_dataIO init()" << std::endl;

        // Init variables
        viewer = viewer_;

        // Parse operating Mode
    if (arguments_.read("-m"))
        {
                OSG_NOTIFY( osg::ALWAYS ) << "Configure osgVisual as MASTER" << std::endl;
                clusterMode = osgVisual::dataIO_cluster::MASTER;
                // Create Container:
                slotContainer = new osgVisual::dataIO_transportContainer();
        }
        else if (arguments_.read("-s"))
        {
                OSG_NOTIFY( osg::ALWAYS ) << "Configure osgVisual as SLAVE" << std::endl;
                clusterMode = osgVisual::dataIO_cluster::SLAVE;
                slotContainer = NULL;   // Slave only recieves container, therefor set this Pointer null.
        }
        else
        {
                OSG_NOTIFY( osg::ALWAYS ) << "Configure osgVisual as STANDALONE" << std::endl;
                clusterMode = osgVisual::dataIO_cluster::STANDALONE;
                // Create Container:
                slotContainer = new osgVisual::dataIO_transportContainer();
        }

        // Create Cluster.
        #ifdef USE_CLUSTER_DUMMY
                cluster = new dataIO_clusterDummy();
        #endif
        #ifdef USE_CLUSTER_ASIO_TCP_IOSTREAM
                cluster = new dataIO_clusterAsioTcpIostream();
        #endif 
        #ifdef USE_CLUSTER_UDP
                cluster = new dataIO_clusterUDP();
        #endif
        #ifdef USE_CLUSTER_ENET
                cluster = new dataIO_clusterENet();
                cluster->enableHardSync( false );       /** \todo : rebuild this structure in cluster.h and move it this way to a general implementation. */
        #endif
        if(cluster.valid())
                //cluster->init(arguments_, clusterMode, slotContainer, true, false);
                cluster->init(arguments_, viewer_, clusterMode, slotContainer, false, false);

        // Create extLink.
        #ifdef USE_EXTLINK_DUMMY
                extLink = new dataIO_extLinkDummy( dataSlots );
        #endif
        #ifdef USE_EXTLINK_VCL
                extLink = new dataIO_extLinkVCL( dataSlots );
        #endif
        extLink->init();

        

        // Install callbacks to perform DataIO activities every frame:
        //// EventCallback at the absolute beginning of the frame
        eventCallback = new dataIO_eventCallback(this);
        viewer->getCamera()->setEventCallback( eventCallback );
        //// FinalDrawCallback at the end of event and update handling, but BEFORE rendering the frame
        finalDrawCallback = new dataIO_finalDrawCallback(this);
        viewer->getCamera()->setFinalDrawCallback( finalDrawCallback );

        initialized = true;
}

void visual_dataIO::shutdown()
{
        if(initialized)
        {
                OSG_NOTIFY( osg::ALWAYS ) << "Shutdown visual_dataIO..." << std::endl;

                viewer->getCamera()->removeEventCallback( eventCallback );
                eventCallback = NULL;
                viewer->getCamera()->setFinalDrawCallback( NULL );
                finalDrawCallback = NULL;
                
                viewer = NULL;
                

                if(cluster.valid())
                        cluster->shutdown();
                if(extLink.valid())
                extLink->shutdown();
        }
}

void visual_dataIO::dataIO_eventCallback::operator()(osg::Node* node, osg::NodeVisitor* nv)
{
        // perform all actions for the eventDrawCallback.
        OSG_NOTIFY( osg::INFO ) << "---- Executing EventCallback.." <<  std::endl;

        switch( dataIO->clusterMode )
        {
                case osgVisual::dataIO_cluster::MASTER : 
                        {
                                dataIO->extLink->readTO_OBJvalues();
                                dataIO->cluster->sendTO_OBJvaluesToSlaves();
                        }
                        break;
                case osgVisual::dataIO_cluster::SLAVE : 
                        {
                                dataIO->cluster->readTO_OBJvaluesFromMaster();
                        }
                        break;
                case osgVisual::dataIO_cluster::STANDALONE : 
                        {
                                dataIO->extLink->readTO_OBJvalues();
                        }
                        break;
                default:
                        OSG_NOTIFY( osg::FATAL ) << "ERROR: Unkown clustermode!" <<  std::endl;
                        break;
        };
        traverse(node, nv);
}

void visual_dataIO::dataIO_finalDrawCallback::operator() (const osg::Camera& camera) const
{
        // perform all actions for the initialDrawCallback.
        OSG_NOTIFY( osg::INFO ) << "---- Executing InitialDrawCallback.." << std::endl;

        switch( dataIO->clusterMode )
        {
                case osgVisual::dataIO_cluster::MASTER : 
                        {
                                dataIO->extLink->writebackFROM_OBJvalues();
                                dataIO->cluster->waitForAllReadyToSwap();
                                dataIO->cluster->sendSwapCommand();
                        }
                        break;
                case osgVisual::dataIO_cluster::SLAVE : 
                        {
                                dataIO->cluster->reportAsReadyToSwap();
                                dataIO->cluster->waitForSwap();
                        }
                        break;
                case osgVisual::dataIO_cluster::STANDALONE : 
                        {
                                dataIO->extLink->writebackFROM_OBJvalues();
                        }
                        break;
                default:
                        OSG_NOTIFY( osg::FATAL ) << "ERROR: visual_dataIO::dataIO_finalDrawCallback::operator() - Unkown clustermode!" <<  std::endl;
                        break;
        };
}

void* visual_dataIO::getSlotPointer(std::string variableName_, osgVisual::dataIO_slot::dataDirection direction_, osgVisual::dataIO_slot::varType variableTyp_ )
{
        // iterate through slotlist. If found, return pointer, else add slot to list and return pointer
        for (unsigned int i=0; i<dataSlots.size(); i++)
        {
                // Check if this variable name&-type already exists
                if( dataSlots[i].variableName == variableName_ && dataSlots[i].direction == direction_  && dataSlots[i].variableType ==  variableTyp_)
                {
                        //OSG_NOTIFY( osg::INFO ) << "visual_dataIO::getSlotPointer() - Slot found at position " << i << std::endl;
                        // Return pointer to the value
                        if (variableTyp_ == osgVisual::dataIO_slot::STRING )
                                return &(dataSlots[i].sValue);
                        else
                                return &(dataSlots[i].value);
                }
        }

        // Slot does not exist -> add it to slot list
        //OSG_NOTIFY( osg::INFO ) << "visual_dataIO::getSlotPointer() - Slot not found, will add as new slot " << std::endl;
        dataIO_slot newSlot;
        newSlot.variableName = variableName_;
        newSlot.variableType = variableTyp_;
        newSlot.value = 0;
        newSlot.sValue = "";
        dataSlots.push_back( newSlot );
        if (variableTyp_ == osgVisual::dataIO_slot::STRING )
                return &(dataSlots.back().sValue);
        else
        {
                return &(dataSlots.back().value);
        }
}

double visual_dataIO::getSlotDataAsDouble(std::string variableName_, osgVisual::dataIO_slot::dataDirection direction_ )
{
        // iterate through slotlist. If found, return value
        for (unsigned int i=0; i<dataSlots.size(); i++)
        {
                // Check if this variable name&-type already exists
                if( dataSlots[i].variableName == variableName_ && dataSlots[i].direction == direction_  && dataSlots[i].variableType == osgVisual::dataIO_slot::DOUBLE )
                {
                        //OSG_NOTIFY( osg::INFO ) << "visual_dataIO::getSlotDataAsDouble() - Slot found at position " << i << std::endl;
                        return dataSlots[i].value;
                }
        }
        return 0;
}

std::string visual_dataIO::getSlotDataAsString(std::string variableName_, osgVisual::dataIO_slot::dataDirection direction_ )
{
        // iterate through slotlist. If found, return value
        for (unsigned int i=0; i<dataSlots.size(); i++)
        {
                // Check if this variable name&-type already exists
                if( dataSlots[i].variableName == variableName_ && dataSlots[i].direction == direction_  && dataSlots[i].variableType == osgVisual::dataIO_slot::STRING )
                {
                        //OSG_NOTIFY( osg::INFO ) << "visual_dataIO::getSlotDataAsDouble() - Slot found at position " << i << std::endl;
                        return dataSlots[i].sValue;
                }
        }
        return "";
}

void visual_dataIO::setSlotData(std::string variableName_, osgVisual::dataIO_slot::dataDirection direction_, std::string sValue_ )
{
        bool slotFound = false;
        // iterate through slotlist. If found, return pointer, else add slot to list
        for (unsigned int i=0; i<dataSlots.size(); i++)
        {
                // Check if this variable name&-type already exists
                if( dataSlots[i].variableName == variableName_ && dataSlots[i].direction == direction_ && dataSlots[i].variableType ==  osgVisual::dataIO_slot::STRING)
                {
                        // Update value
                        dataSlots[i].sValue = sValue_;
                        slotFound = true;
                }
                
        }

        if (!slotFound)
        {
                // Slot does not exist -> add it to slot list
                dataIO_slot newSlot;
                newSlot.variableName = variableName_;
                newSlot.direction = direction_;
                newSlot.variableType = osgVisual::dataIO_slot::STRING;
                newSlot.value = 0;
                newSlot.sValue = sValue_;
                dataSlots.push_back( newSlot );
        }
}

void visual_dataIO::setSlotData(std::string variableName_, osgVisual::dataIO_slot::dataDirection direction_, double value_ )
{
        // iterate through slotlist. If found, return pointer, else add slot to list
        bool slotFound = false;
        for (unsigned int i=0; i<dataSlots.size(); i++)
        {
                // Check if this variableName & -type already exists
                if( dataSlots[i].variableName == variableName_ && dataSlots[i].direction == direction_ && dataSlots[i].variableType ==  osgVisual::dataIO_slot::DOUBLE)
                {
                        // Update value
                        //OSG_NOTIFY( osg::ALWAYS ) << "setSlotData: " << variableName_ << " - value: " << value_ << std::endl;
                        dataSlots[i].value = value_;
                        slotFound = true;
                }       
        }

        if (!slotFound)
        {
                // Slot does not exist -> add it to slot list
                dataIO_slot newSlot;
                newSlot.variableName = variableName_;
                newSlot.direction = direction_;
                newSlot.variableType = osgVisual::dataIO_slot::DOUBLE;
                newSlot.value = value_;
                newSlot.sValue = "";
                dataSlots.push_back( newSlot );
        }
}