TestGraph

package my.annotation;

import java.util.*;
import org.graphstream.graph.*;
import org.graphstream.graph.implementations.*;
import org.graphstream.graph.implementations.*;

public class TestGraph {

    /**
     * @param args
     * @throws Exception
     */
    public static Graph createRandomGraph(){
        Graph g = new SingleGraph("A");
        Node n;
        Integer j;
        for (int i=0;i<10;i++){
            j = new Integer(i);
            n= g.addNode(j.toString());
            n.addAttribute("ui.label", j);
        }
        Edge e;
        int c=0;
        for (int i=1;i<10;i++){
            j = new Integer(i);
            e = g.addEdge(""+c++,"0",j.toString());
        }
        return g;
    }
    public static Graph processGraph(List<String> l,Graph g){
        int c=0;
        System.out.println("Node count "+g.getNodeCount());
        for(Node n: g){
            if(!l.contains(n.getId())){
                g.removeNode(n.getId());
            }
        c++;
        }
        System.out.println("Iterations performed "+c);
        return g;
    }
    public static void main(String[] args) throws Exception {
        // TODO Auto-generated method stub

        Graph g = new SingleGraph("A");
        g = createRandomGraph();
        List<String> l = new ArrayList<String>();
        l.add("0");
        g = processGraph(l, g);
        g.display();
    }

    }