/* -*-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;
	clusterMode = osgVisual::dataIO_cluster::STANDALONE;
	// Create Transport-Container:
	slotContainer = new osgVisual::dataIO_transportContainer();
}

visual_dataIO::~visual_dataIO()
{
	// Delete all slots:
	for(unsigned int i=0;i<dataSlots.size();i++)
	{
		delete dataSlots[i];
	}
	dataSlots.clear();
	
	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_, std::string configFileName)
{
	OSG_NOTIFY( osg::ALWAYS ) << "visual_dataIO init()" << std::endl;

	// Init variables
	viewer = viewer_;

	// Process XML configuration
	if(!processXMLConfiguration())
		OSG_FATAL << "ERROR: visual_dataIO::init() - Failed to initialize dataIO!";

	// Parse operating Mode
    if (arguments_.read("-m"))
	{
		OSG_NOTIFY( osg::ALWAYS ) << "Configure osgVisual as MASTER" << std::endl;
		clusterMode = osgVisual::dataIO_cluster::MASTER;
	}
	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 (created instance will be deleted because it is an auto pointer).
	}
	else
	{
		OSG_NOTIFY( osg::ALWAYS ) << "Configure osgVisual as STANDALONE" << std::endl;
		clusterMode = osgVisual::dataIO_cluster::STANDALONE;
	}

	// Create Cluster.
	#ifdef USE_CLUSTER_ASIO_TCP_IOSTREAM
		cluster = new dataIO_clusterAsioTcpIostream();
	#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
	#ifdef USE_CLUSTER_DUMMY
		cluster = new dataIO_clusterDummy();
	#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;
}

bool visual_dataIO::processXMLConfiguration()
{

	return 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(dataIO->calcViewMatrix());
			}
			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
			return dataSlots[i];
		}
	}

	// 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 = new dataIO_slot();
	newSlot->variableName = variableName_;
	newSlot->variableType = variableTyp_;
	newSlot->value = 0;
	newSlot->sValue = "";
	dataSlots.push_back( newSlot );
	return dataSlots.back();
}

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 "";
}

osgVisual::dataIO_slot* 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;
			return dataSlots[i];
		}
		
	}

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

	return NULL;
}

osgVisual::dataIO_slot* 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;
			return dataSlots[i];
		}	
	}

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

	return NULL;
}

osg::Matrixd visual_dataIO::calcViewMatrix()
{
	return viewer->getCameraManipulator()->getInverseMatrix();
}