com.almende.eve.algorithms.test.TestGraph.java Source code

Introduction

Here is the source code for com.almende.eve.algorithms.test.TestGraph.java

Source

/*
 * Copyright: Almende B.V. (2014), Rotterdam, The Netherlands
 * License: The Apache Software License, Version 2.0
 */
package com.almende.eve.algorithms.test;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Logger;

import junit.framework.TestCase;

import org.junit.Test;

import com.almende.eve.agent.AgentConfig;
import com.almende.eve.algorithms.Edge;
import com.almende.eve.algorithms.test.agents.NodeAgent;
import com.almende.util.jackson.JOM;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;

/**
 * @author ludo
 *         The Class TestNetwork.
 */
public class TestGraph extends TestCase {
    private static final Logger LOG = Logger.getLogger(TestGraph.class.getName());

    /**
     * Write visGraph.
     *
     * @param agents
     *            the agents
     * @return the string
     */
    private String writeVisGraph(List<NodeAgent> agents) {
        final ObjectNode result = JOM.createObjectNode();
        final ArrayNode nodes = JOM.createArrayNode();
        final ArrayNode edges = JOM.createArrayNode();

        for (NodeAgent agent : agents) {
            final ObjectNode node = JOM.createObjectNode();
            node.put("id", agent.getId());
            node.put("label", agent.getId());
            nodes.add(node);
            for (Edge edge : agent.getGraph().getEdges()) {
                final ObjectNode edgeNode = JOM.createObjectNode();
                edgeNode.put("from", agent.getId());
                edgeNode.put("to", edge.getAddress().toASCIIString().replace("local:", ""));
                edges.add(edgeNode);
            }
        }

        result.set("nodes", nodes);
        result.set("edges", edges);
        return result.toString();
    }

    /**
     * Test keys.
     *
     * @throws IOException
     *             Signals that an I/O exception has occurred.
     */
    @Test
    public void testSFNGraph() throws IOException {
        int nofEdges = 3;
        int startDegree = 2;
        int startSteps = 7;
        int nofNodes = 500;

        // Generate X agent scalefree network:
        // add M connections per step
        // start at random node, do the algo
        AgentConfig config = new AgentConfig();
        config.setClassName(NodeAgent.class.getName());

        List<NodeAgent> agents = new ArrayList<NodeAgent>(nofNodes + 10);
        // start by having X0=max(10,M) agents in a ring lattice manner with
        // degree S . each with a Random factor
        final int max = Math.max(10, nofEdges);
        for (int i = 0; i < max; i++) {
            NodeAgent agent = new NodeAgent("" + i, config);
            agents.add(agent);
        }
        for (int i = 0; i < max; i++) {
            NodeAgent agent = agents.get(i);
            for (int j = 0; j < startDegree; j++) {
                NodeAgent other = agents.get((i + j + 1) % max);
                if (other.equals(agent))
                    continue;
                agent.getGraph().addEdge(new Edge(other.getUrls().get(0), "SFN", null));
            }
        }

        NodeAgent start = agents.get(0);
        for (int i = max; i < nofNodes; i++) {
            NodeAgent agent = new NodeAgent("" + i, config);
            agents.add(agent);
            agent.getGraph().addNode2SFN(start.getUrls().get(0), "SFN", nofEdges, startSteps);
        }

        LOG.warning(writeVisGraph(agents));
    }
}