source: source/ariba/utility/visual/DddVis.cpp @ 6900

// [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]

#include "DddVis.h"

namespace ariba {
namespace utility {

const string DddVis::del = ":";
use_logging_cpp(DddVis);

DddVis::DddVis() : commandid(0) {
        srand(time(NULL));
}

DddVis::~DddVis(){
}

void DddVis::sendMessage( const string msg, NETWORK_ID nid ) {
        sendSocket( msg + "\n" );
}

long DddVis::getCommandID() {
        return ++commandid;
}

long DddVis::getTimestamp() {
        return 0; //time(NULL);
}

unsigned int DddVis::getNodeNumber(const NodeID& node){
        NodeSet::iterator i = nodeSet.find(node);

        if(i == nodeSet.end()){
                unsigned int number = node.get(MAX_KEYLENGTH-16, 16);
                nodeSet.insert(make_pair(node, number));
                return number;
        } else {
                return i->second;
        }
}

void DddVis::visCreate(
                NETWORK_ID network,
                NodeID& node,
                string nodename,
                string info
){
        //
        // create layer first if not already done
        //

        if(layerSet.find(network) == layerSet.end()){

                //
                // create layer
                //

                {
                        ostringstream out;
                        out             << CREATE_LAYER_TYPE            << del
                                        << getCommandID()                       << del
                                        << getTimestamp()                       << del
                                        << getNetworkName(network)      << del
                                        << 0                                            << del;

                        sendMessage( out.str(), network );
                        layerSet.insert(network);
                }

                //
                // set layer layout
                //

                {
                        LayoutType layout = FORCE_LAYOUT;
                        LayoutOrderStrategie order = ORDER_RANDOMLY;

                        switch(network){
                        case NETWORK_ID_BASE_COMMUNICATION:
                                layout = FORCE_LAYOUT;
                                order = ORDER_RANDOMLY;
                                break;
                        case NETWORK_ID_BASE_OVERLAY:
                                layout = CIRCULAR_LAYOUT;
                                order = ORDER_BY_ID;
                                break;
                        case NETWORK_ID_MCPO:
                                layout = FORCE_LAYOUT;
                                order = ORDER_RANDOMLY;
                                break;
                        }

                        ostringstream out;
                        out             << SET_CLUSTER_LAYOUT_TYPE      << del
                                        << getCommandID()                       << del
                                        << getTimestamp()                       << del
                                        << getNetworkName(network)      << del
                                        << 0                                            << del
                                        << layout                                       << del
                                        << order                                        << del;

                        sendMessage( out.str(), network );
                }

        } // if(layerSet.find(network) == layerSet.end())

        //
        // create node
        //

        ostringstream out;
        out     << CREATE_NODE_TYPE             << del
                        << getCommandID()                       << del
                        << getTimestamp()                       << del
                        << getNetworkName(network)      << del
                        << 0                                            << del
                        << getNodeNumber(node)          << del
                        << 0x00000000                           << del
                        << "null"                                       << del;

        sendMessage( out.str(), network );
}

void DddVis::visConnect(
                NETWORK_ID network,
                NodeID& srcnode,
                NodeID& destnode,
                string info
){
        // if we already have a link between the two nodes
        // we just ignore the call and leave the old link

        if( networkLinks.exists( network, NodePair(srcnode,destnode) )) return;

        ostringstream out;
        unsigned long edgekey = networkLinks.insert( network, NodePair(srcnode,destnode) );

        out     << CREATE_EDGE_TYPE                     << del
                        << getCommandID()                       << del
                        << getTimestamp()                       << del
                        << getNetworkName(network)      << del
                        << 0                                            << del
                        << getNodeNumber(srcnode)       << del
                        << getNodeNumber(destnode)      << del
                        << edgekey                                      << del
                        << 0x00000000                           << del
                        << "null"                                       << del;

        sendMessage( out.str(), network );
}

void DddVis::visDisconnect(
                NETWORK_ID network,
                NodeID& srcnode,
                NodeID& destnode,
                string info
){
        if( !networkLinks.exists(network, NodePair(srcnode, destnode)) ) return;

        unsigned long edgekey = networkLinks.get( network, NodePair(srcnode, destnode) );
        networkLinks.remove( network, NodePair(srcnode, destnode) );

        ostringstream out;
        out     << REMOVE_EDGE_TYPE             << del
                << getCommandID()                       << del
                << getTimestamp()                       << del
                << getNetworkName(network)      << del
                << edgekey                                      << del;

        sendMessage( out.str(), network );
}

void DddVis::visShutdown(
                NETWORK_ID network,
                NodeID& node,
                string info
){
        ostringstream out;

        out     << REMOVE_NODE_TYPE             << del
                << getCommandID()                       << del
                << getTimestamp()                       << del
                << getNetworkName(network)      << del
                << getNodeNumber(node)          << del;

        sendMessage( out.str(), network );
}

void DddVis::visChangeNodeColor (
                NETWORK_ID network,
                NodeID& node,
                unsigned char r,
                unsigned char g,
                unsigned char b
){
        NodeSet::iterator i = colorSet.find(node);
        unsigned int color = makeColor(r,g,b);

        if( i == colorSet.end() ){
                colorSet.insert(make_pair( node, color )); // color not set for node, set
        }else{
                if( i->second == color ) return; // color already set, ignore
                else i->second = color;                  // new color, set
        }

        ostringstream out;
        out     << SET_NODE_COLOR_TYPE          << del
                        << getCommandID()                       << del
                        << getTimestamp()                       << del
                        << getNetworkName(network)      << del
                        << getNodeNumber(node)          << del
                        << makeColor(r, g, b)           << del;

        sendMessage( out.str(), network );
}

int DddVis::makeColor(unsigned char r, unsigned char g, unsigned char b){
        return ((r<<16)+(g<<8)+b);
}

void DddVis::visChangeNodeColor (
                NETWORK_ID network,
                NodeID& node,
                NODE_COLORS color
){
        unsigned char r = 0;
        unsigned char g = 0;
        unsigned char b = 0;

        switch( color ) {
        case NODE_COLORS_GREY:  r = 128; g = 128; b = 128; break;
        case NODE_COLORS_GREEN: r = 0;   g = 200; b = 0;   break;
        case NODE_COLORS_RED:   r = 255; g = 0;   b = 0;   break;
        }

        visChangeNodeColor( network, node, r, g, b );
}

void DddVis::visChangeLinkColor (
                NETWORK_ID network,
                NodeID& srcnode,
                NodeID& destnode,
                unsigned char r,
                unsigned char g,
                unsigned char b
){
        ostringstream out;
        unsigned long edgekey = networkLinks.get( network, NodePair(srcnode, destnode) );

        out     << SET_EDGE_COLOR_TYPE          << del
                        << getCommandID()                       << del
                        << getTimestamp()                       << del
                        << getNetworkName(network)      << del
                        << edgekey                                      << del
                        << makeColor(r, g, b)           << del;

        sendMessage( out.str(), network );
}

void DddVis::visChangeLinkColor (
                NETWORK_ID network,
                NodeID& srcnode,
                NodeID& destnode,
                NODE_COLORS color
){
        unsigned char r = 0;
        unsigned char g = 0;
        unsigned char b = 0;

        switch( color ) {
        case NODE_COLORS_GREY:  r = 128; g = 128; b = 128; break;
        case NODE_COLORS_GREEN: r = 0;   g = 200; b = 0;   break;
        case NODE_COLORS_RED:   r = 255; g = 0;   b = 0;   break;
        }

        visChangeLinkColor( network, srcnode, destnode, r, g, b );
}

}} // namespace ariba, common