Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

         *

         * Copyright (C) 2006 Instituto de Desarrollo Regional and Generalitat Valenciana.

         *

         * 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, or (at your option) any later version.

         *

         * 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 for more details.

         *

         * You should have received a copy of the GNU General Public License

         * along with this program; if not, write to the Free Software

         * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

         *

         * For more information, contact:

         *

         *  Generalitat Valenciana

         *   Conselleria d'Infraestructures i Transport

         *   Av. Blasco Iba?ez, 50

         *   46010 VALENCIA

         *   SPAIN

         *

         *      +34 963862235

         *   gvsig@gva.es

         *      www.gvsig.gva.es

         *

         *    or

         *

         *   Instituto de Desarrollo Regional (Universidad de Castilla La-Mancha)

         *   Campus Universitario s/n

         *   02071 Alabacete

         *   Spain

         *

         *   +34 967 599 200

         */

package org.gvsig.remotesensing.decisiontrees.gui;

import java.awt.Color;

import java.awt.Dimension;

import javax.swing.BorderFactory;

import javax.swing.JScrollPane;

import org.gvsig.gui.beans.defaultbuttonspanel.DefaultButtonsPanel;

import org.gvsig.remotesensing.decisiontrees.DecisionTreeNode;

import org.gvsig.remotesensing.decisiontrees.gui.listener.GraphListener;

import org.jgraph.JGraph;

import org.jgraph.graph.DefaultEdge;

import org.jgraph.graph.DefaultGraphCell;

import org.jgraph.graph.DefaultGraphModel;

import org.jgraph.graph.DefaultPort;

import org.jgraph.graph.GraphConstants;

import org.jgraph.graph.GraphModel;

/**

 * Panel para la herramienta de ?rboles de decisi?n.

 * 

 * @author Diego Guerrero Sevilla (diego.guerrero@uclm.es)

 *

 */

public class DecisionTreePanel extends DefaultButtonsPanel {

        /**

         * 

         */

        private static final long                 serialVersionUID         = 2800193252856199039L;

        private JGraph                                           jGraph                                = null;

        private DecisionTreeDialog          decisionTreeDialog         = null;

        private JScrollPane                                scrollPane                        = null;

        private DecisionTreeNode                 root                                = null;

        public DecisionTreePanel(DecisionTreeDialog decisionTreeDialog) {

                super();

                this.decisionTreeDialog = decisionTreeDialog;

                initialize();

        }

        private void initialize() {

                initTree();

                reloadGraph();

                add(getScrollPane());

        }

        /**

         * Crea el ?rbol de decisi?n inicial

         *

         */

        private void initTree() {

                root = new DecisionTreeNode();

                root.addChildren();

                root.setExpression("a<b");

                root.getLeftChild().setClassID(0);

                root.getRightChild().setClassID(1);

        }

        /**

         * Reconstruye el gr?fico a partir del ?rbol (root).

         *

         */

        private void reloadGraph(){

                insertCells(root,null);

        }

        private void insertCells(DecisionTreeNode node, DefaultGraphCell parent) {

                DefaultGraphCell nodeVertex = createVertex(node, 20, 20, 40, 20, null, false);

                getJGraph().getGraphLayoutCache().insert(nodeVertex);

                if (parent!=null){

                        DefaultEdge edge = new DefaultEdge();

                        edge.setSource(parent);

                        edge.setTarget(nodeVertex);

                        GraphConstants.setLineEnd(edge.getAttributes(), GraphConstants.ARROW_CLASSIC);

                        GraphConstants.setEndFill(edge.getAttributes(), true);

                        GraphConstants.setAutoSize(edge.getAttributes(), true);

                        getJGraph().getGraphLayoutCache().insert(edge);

                }

                if (node.getLeftChild()!=null){

                        insertCells(node.getLeftChild(), nodeVertex);

                }

                if (node.getRightChild()!=null){

                        insertCells(node.getRightChild(), nodeVertex);

                }

        }

        public DefaultGraphCell createVertex(Object name, double x,

                        double y, double w, double h, Color bg, boolean raised) {

                        // Create vertex with the given name

                        DefaultGraphCell cell = new DefaultGraphCell(name);

                        // Set bounds

                //        GraphConstants.setBounds(cell.getAttributes(),

                        //                new Rectangle2D.Double(x, y, w, h));

                        // Set fill color

                        if (bg != null) {

                                GraphConstants.setGradientColor(

                                        cell.getAttributes(), bg);

                                GraphConstants.setOpaque(

                                        cell.getAttributes(), true);

                        }

                        // Set raised border

                        if (raised)

                                GraphConstants.setBorder(

                                        cell.getAttributes(), 

                                        BorderFactory.createRaisedBevelBorder());

                        else

                                // Set black border

                                GraphConstants.setBorderColor(

                                        cell.getAttributes(), Color.black);

                        // Add a Port

                        DefaultPort port = new DefaultPort();

                        cell.add(port);

                        port.setParent(cell);

                        GraphConstants.setBorder(cell.getAttributes(), BorderFactory.createEtchedBorder());

                    GraphConstants.setAutoSize(cell.getAttributes(),true);

                    GraphConstants.setResize(cell.getAttributes(),true);

                    GraphConstants.setInset(cell.getAttributes(), 10);

                        return cell;

                }

        public JScrollPane getScrollPane() {

                if (scrollPane == null){

                        scrollPane = new JScrollPane(getJGraph());

                        scrollPane.setPreferredSize(new Dimension(500,500));

                }

                return scrollPane;

        }

        public JGraph getJGraph() {

                if (jGraph == null){

                        GraphModel model = new DefaultGraphModel();

                        jGraph = new JGraph(model);

                        jGraph.setCloneable(false);

                jGraph.setInvokesStopCellEditing(true);

                jGraph.setJumpToDefaultPort(true);

                jGraph.setConnectable(false);

                jGraph.setDisconnectable(false);

                jGraph.setEditable(false);

                jGraph.setEnabled(true);

                jGraph.setSize(300, 300);

                        GraphListener listener = new GraphListener();

                        jGraph.addMouseListener(listener);

                }

                return jGraph;

        }

}