// [License]
// The Ariba-Underlay Copyright
//
// Copyright (c) 2008-2009, Institute of Telematics, Universität Karlsruhe (TH)
//
// Institute of Telematics
// Universität Karlsruhe (TH)
// Zirkel 2, 76128 Karlsruhe
// Germany
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// 1. Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
// 2. Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// THIS SOFTWARE IS PROVIDED BY THE INSTITUTE OF TELEMATICS ``AS IS'' AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE ARIBA PROJECT OR
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// The views and conclusions contained in the software and documentation
// are those of the authors and should not be interpreted as representing
// official policies, either expressed or implied, of the Institute of
// Telematics.
// [License]

#ifndef OVLVIS_H__
#define OVLVIS_H__

#include <sstream>
#include <iostream>
#include <string>
#include <map>
#include <boost/utility.hpp>
#include <boost/asio.hpp>
#include "ariba/utility/types/NodeID.h"
#include "ariba/utility/logging/Logging.h"
#include "ariba/utility/system/Timer.h"
#include "ariba/utility/misc/Helper.h"
#include "ariba/utility/misc/KeyMapping.hpp"
#include "ariba/utility/configuration/Configuration.h"

using std::string;
using std::map;
using std::pair;
using std::make_pair;
using std::cout;
using std::ostringstream;
using boost::asio::ip::tcp;
using ariba::utility::NodeID;
using ariba::utility::Configuration;
using ariba::utility::KeyMapping;
using ariba::utility::Timer;

namespace ariba {
namespace utility {

class OvlVis : private boost::noncopyable {
	use_logging_h(OvlVis);
public:
	static OvlVis& instance() { static OvlVis the_inst; return the_inst; }

	typedef enum _NETWORK_ID {
		NETWORK_ID_BASE_COMMUNICATION 	= 1,
		NETWORK_ID_BASE_OVERLAY 	= 2,
		NETWORK_ID_EONSON 		= 3,
		NETWORK_ID_MCPO 		= 4,
		NETWORK_ID_CLIO 		= 5,
		NETWORK_ID_VIDEOSTREAM 		= 6,
		NETWORK_ID_GAME 		= 7,
		NETWORK_ID_SECURITY 		= 8,
	} NETWORK_ID;

	//****************************************************************
	// Node creation, node connections, status, node deletion, ...
	//****************************************************************

	/**
	 * Create a node in the network that is initially unconnected.
	 */
	void visCreate (
		NETWORK_ID network,
		NodeID& node,
		string nodename,
		string info
		);

	/**
	 * Change the status of a node -> enable/disable a node.
	 */
	void visChangeStatus(
		NETWORK_ID network,
		NodeID& node,
		bool enable,
		string info
		);

	/**
	 * Connect two nodes using a link.
	 */
	void visConnect (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode,
		string info
		);

	/**
	 * Disconnect the link between two nodes.
	 */
	void visDisconnect (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode,
		string info
		);

	/**
	 * Indicate that the connection procedure
	 * between two nodes failed.
	 */
	void visFailedConnect (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode,
		string info
		);

	/**
	 * Delete a node from the network.
	 */
	void visShutdown (
		NETWORK_ID network,
		NodeID& node,
		string info
		);

	//****************************************************************
	// Node manipulation: change color, change icon
	//****************************************************************

	/**
	 * Change the color of the node.
	 */
	void visChangeNodeColor (
		NETWORK_ID network,
		NodeID& node,
		unsigned char r,
		unsigned char g,
		unsigned char b
		);

	typedef enum _NODE_COLORS {
		NODE_COLORS_GREY,
		NODE_COLORS_GREEN,
		NODE_COLORS_RED,
	} NODE_COLORS;

	/**
	 * Change the color of the node.
	 */
	void visChangeNodeColor (
		NETWORK_ID network,
		NodeID& node,
		NODE_COLORS color
		);

	/**
	 * The available icons for changing the
	 * icon of a node, showing an icon besides
	 * a node and showing an icon at a link.
	 */
	typedef enum _ICON_ID {
		ICON_ID_DEFAULT_NODE 	= 0,
		ICON_ID_PC 		= 1,
		ICON_ID_PC_WORLD 	= 2,
		ICON_ID_FAILURE 	= 3,
		ICON_ID_RED_CROSS 	= 4,
		ICON_ID_CHARACTER_A 	= 5,
		ICON_ID_CHARACTER_W 	= 6,
		ICON_ID_CAMERA          = 7,
	} ICON_ID;

	/**
	 * Change the icon of a node.
	 */
        void visChangeNodeIcon (
                NETWORK_ID network,
                NodeID& node,
                ICON_ID icon
                );

	/**
	 * Show the label of the node.
	 */
	void visShowNodeLabel (
		NETWORK_ID network,
		NodeID& node,
		string label
		);

	/**
	 * Delete the label of the node.
	 */
	void visDeleteNodeLabel (
		NETWORK_ID network,
		NodeID& node
		);

	/**
	 * Show a bubble at the node.
	 */
        void visShowNodeBubble (
                NETWORK_ID network,
                NodeID& node,
                string label
                );

	/**
	 * Delete a bubble at the node.
	 */
	void visDeleteNodeBubble (
                NETWORK_ID network,
		NodeID& node
                );

	/**
	 * Show an icon besides the node.
	 */
        void visShowShiftedNodeIcon (
                NETWORK_ID network,
                NodeID& node,
                ICON_ID iconID,
		unsigned int timeout = 0
                );

	/**
	 * Delete an icon besides the node
	 */
        void visDeleteShiftedNodeIcon (
                NETWORK_ID network,
                NodeID& node
                );

	//****************************************************************
	// Link manipulation: change width, color, show bubbles, icons, ...
	//****************************************************************

	/**
	 * Change the link width
	 */
	void visChangeLinkWidth (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode,
		unsigned int width
		);

	/**
	 * Change the link color
	 */
	void visChangeLinkColor (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode,
		unsigned char r,
		unsigned char g,
		unsigned char b
		);

	/**
	 * Change the link color
	 */
	void visChangeLinkColor (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode,
		NODE_COLORS color
		);

	/**
	 * Show a link label
	 */
	void visShowLinkLabel (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode,
		string label
		);

	/**
	 * Delete a link label
	 */
	void visDeleteLinkLabel (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode
		);

	/**
	 * Show an icon at the link
	 */
        void visShowOnLinkIcon (
                NETWORK_ID network,
                NodeID& srcnode,
                NodeID& destnode,
                ICON_ID iconID
                );

	/**
	 * Delete an icon at the link
	 */
        void visDeleteOnLinkIcon (
                NETWORK_ID network,
                NodeID& srcnode,
		NodeID& destnode
                );

	/**
	 * Show a bubble besides the link
	 */
	void visShowLinkBubble (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode,
		string label
		);

	/**
	 * Delete a bubble besides the link
	 */
	void visDeleteLinkBubble (
		NETWORK_ID network,
		NodeID& srcnode,
		NodeID& destnode
		);

	//****************************************************************
	// Send message between two nodes
	//****************************************************************

	/**
	 * Animate the message sending between two nodes
	 */
	void visSendMessage (
		NETWORK_ID network,
		NodeID& startnode,
		NodeID& endnode
		);

//*******************************************************
//*
	void visCLIOInitMeasurement (
		NETWORK_ID network,
		unsigned long edgekey,
		NodeID& srcnode,
		NodeID& destnode,
		string info
		);

	void visCLIOEndMeasurement (
		NETWORK_ID network,
		unsigned long edgekey,
		NodeID& srcnode,
		NodeID& destnode,
		string info,
		string value,
		string unit
		);
//*
//*******************************************************

protected:
	OvlVis();
	~OvlVis();

private:
	void sendMessage( const string msg, NETWORK_ID nid );
	void sendSocket( const string msg );

	/**
	 * boost asio stuff for connecting to ovlvis
	 */
	boost::asio::io_service io_service;
	tcp::socket socket;
	volatile bool socketOpened;

	typedef pair<NodeID, NodeID> NodePair;
	typedef KeyMapping<NodePair> NetworkLinks;
	typedef KeyMapping<NodePair> LinkBubbles;
	typedef KeyMapping<NodeID>   NodeBubbles;
	typedef KeyMapping<NodeID>   ShiftedNodeIcons;
	typedef KeyMapping<NodePair> OnLinkIcons;

	NetworkLinks 		networkLinks;
	LinkBubbles 		linkBubbles;
	NodeBubbles 		nodeBubbles;
	ShiftedNodeIcons 	shiftedNodeIcons;
	OnLinkIcons 		onLinkIcons;

	class TimedoutIcon : public Timer {
	private:
		NETWORK_ID network;
		NodeID node;
		unsigned int timeout;
	public:
		TimedoutIcon(NETWORK_ID _network, NodeID _node, unsigned int _timeout) :
			network(_network), node(_node), timeout(_timeout) {
		}

		virtual ~TimedoutIcon(){
			Timer::stop();
		}

		void startIcon(){
			Timer::setInterval( timeout, true );
			Timer::start();
		}

	protected:
		virtual void eventFunction(){
			OvlVis::instance().visDeleteShiftedNodeIcon( network, node );
			delete this;
		}
	};

};

}} // namespace ariba, common

#endif // OVLVIS_H__