/* Copyright © 2007 by Christian Fuchsberger and Lukas Forer info@pedvizapi.org.
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License.
 *
 * This program 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 GNU General Public License <http://www.pedvizapi.org/gpl.txt>
 * for more details.
 */


import javax.swing.JFrame;

import pedviz.algorithms.Sugiyama;
import pedviz.graph.*;
import pedviz.loader.CsvGraphLoader;
import pedviz.view.GraphView2D;
import pedviz.view.rules.*;
import pedviz.view.*;

import java.awt.*;

class MyFantasticRule extends Rule {

    public void applyRule(NodeView nodeview) {

	// gets a reference to node
	Node node = nodeview.getNode();

	// now you have access to the traits
	Object value = node.getUserData("TRAIT");
	if (value != null) {
	    Integer trait = Integer.parseInt(value.toString());
	    if (trait > 107)
		nodeview.setColor(Color.red);
	    else
		nodeview.setColor(Color.blue);

	    // add a wonderful hint
	    nodeview.setHintText("trait: " + trait);
	}

    }

}

public class Example4 {
    public static void main(String[] args) {

	// Step 1
	Graph graph = new Graph();
	CsvGraphLoader loader = new CsvGraphLoader("tutorial_data.csv", ",");
	loader.setSettings("PID", "MOM", "DAD");
	loader.load(graph);

	// Step 2
	Sugiyama s = new Sugiyama(graph);
	s.run();

	// Creates a frame
	JFrame frame = new JFrame("Tutorials - Example 4");
	frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
	frame.setSize(800, 600);

	// Step 3
	GraphView2D view = new GraphView2D(s.getLayoutedGraph());
	frame.add(view.getComponent());

	// Step 4 - add the new rule
	view.addRule(new MyFantasticRule());

	frame.setVisible(true);
    }
}