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;

import org.nfunk.jep.JEP;

/**

 * Clase que representa un n?do en un ?rbol de decisi?n

 * 

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

 *

 */

public class DecisionTreeNode {

        public String toString() {

                if(expression!=null)

                        return expression;

                else 

                        return String.valueOf(classID);

        }

        private JEP                                 parser                         = null;

        private String                                 expression                 = null;

        private DecisionTreeNode         leftChild                 = null;

        private DecisionTreeNode         rightChild                 = null;

        private int                                 classID                 = -1;

        public DecisionTreeNode() {

                parser = new JEP();

                parser.setAllowUndeclared(true);

                parser.addStandardFunctions();

        }

        private DecisionTreeNode(JEP parser) {

                this.parser = parser;

        }

        public boolean evaluate(){

                parser.parseExpression(expression);

                return(parser.getValue()!=0.0);

        }

        public int execute (){

                if (expression!=null)

                        if (evaluate()){

                                if (rightChild!=null) 

                                        return rightChild.execute();

                        }

                        else if (rightChild!=null)

                                return leftChild.execute();

                return classID;

        }

        public int validate(){

                parser.parseExpression(expression);

                return 0;

        }

        public JEP getParser() {

                return parser;

        }

        public String getExpression() {

                return expression;

        }

        public void setExpression(String expression) {

                this.expression = expression;

                parser.parseExpression(expression);

                classID = -1;

        }

        public int getClassID() {

                return classID;

        }

        /**

         * Establece el identificador de clase.

         * 

         * @param classID

         */

        public void setClassID(int classID) {

                this.classID = classID;

        }

        public void addChildren() {

                leftChild = new DecisionTreeNode(parser);

                rightChild = new DecisionTreeNode(parser);

                setExpression("");

        }

        public DecisionTreeNode getLeftChild() {

                return leftChild;

        }

        public DecisionTreeNode getRightChild() {

                return rightChild;

        }

        public void setVarValue (String varName, Object value){

                parser.setVarValue(varName, value);

        }

        /**

         * 

         * @return N?mero de nodos del ?rbol que radica en este nodo.

         */

        public int size(){

                int leftSize = 0;

                int rightSize = 0;

                if (leftChild != null)

                        leftSize = leftChild.size();

                if (rightChild != null)

                        rightSize = rightChild.size();

                return 1+leftSize+rightSize;

        }

}