| gvSIG desktop 1

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

 *

 * Copyright (C) 2004 IVER T.I. 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 Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

package com.iver.cit.gvsig.fmap.rendering;

import com.hardcode.gdbms.engine.data.DataSource;

import com.hardcode.gdbms.engine.instruction.FieldNotFoundException;

import com.hardcode.gdbms.engine.instruction.IncompatibleTypesException;

import com.hardcode.gdbms.engine.instruction.SemanticException;

import com.hardcode.gdbms.engine.values.BooleanValue;

import com.hardcode.gdbms.engine.values.NullValue;

import com.hardcode.gdbms.engine.values.StringValue;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.iver.cit.gvsig.fmap.DriverException;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.IFeature;

import com.iver.cit.gvsig.fmap.core.v02.FConstant;

import com.iver.cit.gvsig.fmap.core.v02.FSymbol;

import com.iver.cit.gvsig.fmap.layers.XMLException;

import com.iver.utiles.XMLEntity;

import java.sql.Types;

import java.text.ParseException;

import java.util.ArrayList;

import java.util.Comparator;

import java.util.TreeMap;

/**

 * Leyenda vectorial por valores ?nicos.

 *

 * @author Vicente Caballero Navarro

 */

public class VectorialUniqueValueLegend implements UniqueValueLegend,

    VectorialLegend {

    private TreeMap symbols = new TreeMap(new Comparator() {

                public int compare(Object o1, Object o2) {

                    if ((o1 != null) && (o2 != null)) {

                        Value v2 = (Value) o2;

                        Value v1 = (Value) o1;

                        BooleanValue boolVal;

                        //                                                TODO estas dos comprobaciones son por evitar un bug en el gdbms, cuando se solucione se puede eliminar.

                        if (v1 instanceof NullValue && v2 instanceof NullValue) {

                            return 0;

                        }

                        if (v1 instanceof NullValue) {

                            return -1;

                        }

                        if (v2 instanceof NullValue) {

                            return 1;

                        }

                        try {

                            boolVal = (BooleanValue) (v1.greater(v2));

                            if (boolVal.getValue()) {

                                return 1;

                            }

                            boolVal = (BooleanValue) (v1.less(v2));

                            if (boolVal.getValue()) {

                                return -1;

                            }

                        } catch (IncompatibleTypesException e) {

                            // TODO Auto-generated catch block

                            //e.printStackTrace();

                        }

                        try {

                            if (((BooleanValue) v1.equals(v2)).getValue()) {

                                return 0;

                            }

                        } catch (IncompatibleTypesException e) {

                            // TODO Auto-generated catch block

                            //e.printStackTrace();

                        }

                        if (v1 instanceof StringValue) {

                            return -1;

                        }

                        if (v2 instanceof StringValue) {

                            return 1;

                        }

                    }

                    return 0;

                }

            }); // Para poder ordenar

    private ArrayList keys = new ArrayList(); // En lugar de un HashSet, para tener acceso por ?ndice

    private String fieldName;

    private int fieldId = -1;

    private String labelFieldName;

    private String labelFieldHeight;

    private String labelFieldRotation;

    private FSymbol defaultSymbol;

    private DataSource dataSource;

    private int shapeType;

    private String valueType = NullValue.class.getName();

    private boolean useDefaultSymbol = false;

    // private boolean bWithHeightText = false;

    /**

     * Crea un nuevo VectorialUniqueValueLegend.

     */

    public VectorialUniqueValueLegend() {

        //        defaultSymbol = LegendFactory.DEFAULT_POLYGON_SYMBOL;

    }

    /**

     * Crea un nuevo VectorialUniqueValueLegend.

     *

     * @param shapeType Tipo de shape.

     */

    public VectorialUniqueValueLegend(int shapeType) {

        setShapeType(shapeType);

    }

    /**

     * Inserta el tipo de shape.

     *

     * @param shapeType Tipo de shape.

     */

    public void setShapeType(int shapeType) {

        if (this.shapeType != shapeType) {

            switch (shapeType) {

                case FShape.POINT:

                    defaultSymbol = new FSymbol(FConstant.SYMBOL_TYPE_POINT);

                    break;

                case FShape.LINE:

                    defaultSymbol = new FSymbol(FConstant.SYMBOL_TYPE_LINE);

                    break;

                case FShape.POLYGON:

                    defaultSymbol = new FSymbol(FConstant.SYMBOL_TYPE_FILL);

                    break;

                default:

                    defaultSymbol = new FSymbol(shapeType);

            }

            this.shapeType = shapeType;

        }

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.UniqueValueLegend#setValueSymbolByID(int,

     *      FSymbol)

     */

    public void setValueSymbolByID(int id, FSymbol symbol) {

        symbols.put(keys.get(id), symbol);

    }

    /**

     * Devuelve un s?mbolo a partir del ID. Mira en el m_ArrayKeys  el elemento

     * ID, y con esa clave recupera el FSymbol por valor

     *

     * @param id ID.

     * @param symbol DOCUMENT ME!

     */

    /*

       public FSymbol getSymbolByID(int ID) {

           return (FSymbol) symbols.get(keys.get(ID));

       }

     */

    /**

     * Se usa en la tabla que muestra una leyenda.

     *

     * @param id

     * @param symbol

     */

    public void setValueSymbol(int id, FSymbol symbol) {

        symbols.put(keys.get(id), symbol);

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.UniqueValueLegend#getValues()

     */

    public Object[] getValues() {

        return symbols.keySet().toArray(new Object[0]);

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.UniqueValueLegend#addSymbol(java.lang.Object,

     *      FSymbol)

     */

    public void addSymbol(Object key, FSymbol symbol) {

        Object resul;

        resul = symbols.put(key, symbol);

        if (resul != null) {

            System.err.println("Error: la clave " + key +

                " ya exist?a. Resul = " + resul);

            System.err.println("symbol nuevo:" + symbol.getDescription() +

                " Sviejo= " + ((FSymbol) resul).getDescription());

        } else {

            keys.add(key);

            if (!key.getClass().equals(NullValue.class)) {

                valueType = key.getClass().getName();

            }

        }

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.ClassifiedLegend#clear()

     */

    public void clear() {

        keys.clear();

        symbols.clear();

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.ClassifiedLegend#getDescriptions()

     */

    public String[] getDescriptions() {

        String[] descriptions = new String[symbols.size()];

        FSymbol[] auxSym = getSymbols();

        for (int i = 0; i < descriptions.length; i++)

            descriptions[i] = auxSym[i].getDescription();

        return descriptions;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.ClassifiedLegend#getSymbols()

     */

    public FSymbol[] getSymbols() {

        return (FSymbol[]) symbols.values().toArray(new FSymbol[0]);

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.ClassifiedLegend#getFieldName()

     */

    public String getFieldName() {

        return fieldName;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.VectorialLegend#setDefaultSymbol(com.iver.cit.gvsig.fmap.rendering.styling.FStyle2D)

     */

    public void setDefaultSymbol(FSymbol s) {

        defaultSymbol = s;

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.rendering.VectorialLegend#getLabelField()

     */

    public String getLabelField() {

        return labelFieldName;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.Legend#setLabelField(int)

     */

    public void setLabelField(String fieldName) {

        labelFieldName = fieldName;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.ClassifiedLegend#setField()

     */

    public void setFieldName(String str) {

        fieldName = str;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.VectorialLegend#getSymbol(int)

     */

    public FSymbol getSymbol(int recordIndex) throws DriverException {

        try {

            Value val = dataSource.getFieldValue(recordIndex, fieldId);

            FSymbol theSymbol = getSymbolByValue(val);

            //if (theSymbol != null) {

            return theSymbol;

            //} else {

            //        return getDefaultSymbol();

            //}

        } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {

            throw new DriverException(e);

        }

    }

    /**

     * Devuelve un s?mbolo a partir de una IFeature.

     *

     * @param feat IFeature

     *

     * @return S?mbolo.

     */

    public FSymbol getSymbolByFeature(IFeature feat) {

        Value val = feat.getAttribute(fieldId);

        FSymbol theSymbol = getSymbolByValue(val);

        //if (theSymbol != null) {

        return theSymbol;

        //} else {

        //    return getDefaultSymbol();

        //}

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.Legend#getDefaultSymbol()

     */

    public FSymbol getDefaultSymbol() {

        return defaultSymbol;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.Legend#getXMLEntity()

     */

    public XMLEntity getXMLEntity() {

        XMLEntity xml = new XMLEntity();

        xml.putProperty("className", this.getClass().getName());

        xml.putProperty("fieldName", fieldName);

        xml.putProperty("labelfield", labelFieldName);

        xml.putProperty("labelFieldHeight", labelFieldHeight);

        xml.putProperty("labelFieldRotation", labelFieldRotation);

        xml.putProperty("useDefaultSymbol", useDefaultSymbol);

        xml.addChild(getDefaultSymbol().getXMLEntity());

        // xml.putProperty("isBWithHeightText", isBWithHeightText());

        xml.putProperty("numKeys", keys.size());

        if (keys.size() > 0) {

            xml.putProperty("tipoValueKeys", valueType);

            String[] sk = new String[keys.size()];

            String[] sv = new String[keys.size()];

            FSymbol[] fsymbols = getSymbols();

            Object[] values = (Object[]) getValues();

            for (int i = 0; i < keys.size(); i++) {

                sk[i] = ((Value) keys.get(i)).toString();

                sv[i] = ((Value) values[i]).toString();

                xml.addChild(fsymbols[i].getXMLEntity());

                ///System.out.println("get-----------"+sk[i]+"--"+fsymbols[i].getDescription()+"---"+fsymbols[i].getColor());

            }

            xml.putProperty("keys", sk);

            xml.putProperty("values", sv);

        }

        return xml;

    }

    /**

     * Inserta el XMLEntity.

     *

     * @param xml XMLEntity.

     */

    public void setXMLEntity03(XMLEntity xml) {

        clear();

        setFieldName(xml.getStringProperty("fieldName"));

        setLabelField(xml.getStringProperty("labelfield"));

        int useDefaultSymbol = xml.getIntProperty("useDefaultSymbol");

        if (useDefaultSymbol == 1) {

            setDefaultSymbol(FSymbol.createFromXML03(xml.getChild(0)));

        } else {

            setDefaultSymbol(null);

        }

        int numKeys = xml.getIntProperty("numKeys");

        if (numKeys > 0) {

            String className = xml.getStringProperty("tipoValueKeys");

            String[] sk = xml.getStringArrayProperty("keys");

            String[] sv = xml.getStringArrayProperty("values");

            Value auxValue;

            Value auxValue2;

            for (int i = 0; i < numKeys; i++) {

                try {

                    auxValue = ValueFactory.createValue(sk[i], className);

                    auxValue2 = ValueFactory.createValue(sv[i], className);

                    FSymbol sym = FSymbol.createFromXML03(xml.getChild(i +

                                useDefaultSymbol));

                    ///addSymbol(auxValue, sym);

                    symbols.put(auxValue2, sym);

                    keys.add(auxValue);

                    ///System.out.println("---set------"+auxValue.toString());

                    /// System.out.println("set-----------"+sk[i]+"--"+sym.getDescription()+"---"+sym.getColor());

                } catch (SemanticException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                }

            }

        }

    }

    /**

     * Inserta el XMLEntity.

     *

     * @param xml XMLEntity.

     */

    public void setXMLEntity(XMLEntity xml) {

        clear();

        setFieldName(xml.getStringProperty("fieldName"));

        setLabelField(xml.getStringProperty("labelfield"));

        if (xml.contains("labelFieldHeight")) {

            setLabelHeightField(xml.getStringProperty("labelFieldHeight"));

        }

        if (xml.contains("labelFieldRotation")) {

            setLabelRotationField(xml.getStringProperty("labelFieldRotation"));

        }

        useDefaultSymbol = xml.getBooleanProperty("useDefaultSymbol");

        setDefaultSymbol(FSymbol.createFromXML(xml.getChild(0)));

        // FJP: Esto no es necesario ya. Para comprobar si tenemos un campo de altura, miramos

        // si getLabelHeightField devuelve null o no.

        /* if (xml.contains("isBWithHeightText")) {

            setBWithHeightText(xml.getBooleanProperty("isBWithHeightText"));

        } */

        //addSymbol(new NullUniqueValue(),getDefaultSymbol());

        int numKeys = xml.getIntProperty("numKeys");

        if (numKeys > 0) {

            String className = xml.getStringProperty("tipoValueKeys");

            String[] sk = xml.getStringArrayProperty("keys");

            String[] sv = xml.getStringArrayProperty("values");

            Value auxValue = null;

            Value auxValue2 = null;

            for (int i = 0; i < numKeys; i++) {

                //try {

                boolean isDefault = false;

                if (getValue(sk[i]) == null) {

                    isDefault = true;

                }

                if (className.equals(

                            "com.hardcode.gdbms.engine.values.NullUniqueValue") ||

                        isDefault) {

                    auxValue = new NullUniqueValue();

                    auxValue2 = ValueFactory.createNullValue();

                } else {

                    auxValue = getValue(sk[i]); //ValueFactory.createValue(sk[i], className);

                    auxValue2 = getValue(sv[i]); // ValueFactory.createValue(sv[i], className);

                }

                FSymbol sym = FSymbol.createFromXML(xml.getChild(i + 1));

                ///addSymbol(auxValue, sym);

                symbols.put(auxValue2, sym);

                keys.add(auxValue);

                ///System.out.println("---set------"+auxValue.toString());

                /// System.out.println("set-----------"+sk[i]+"--"+sym.getDescription()+"---"+sym.getColor());

                //} catch (SemanticException e) {

                //        e.printStackTrace();

                //}

            }

        }

    }

    /**

     * Devuelve el valor a partir de su valor en un string.

     *

     * @param s String con el valor.

     *

     * @return Value.

     */

    private Value getValue(String s) {

        Value val = new NullUniqueValue();

        if (s.equals("Resto de Valores"))return val;

        try {

            try {

                val = ValueFactory.createValueByType(s, Types.INTEGER);

                return val;

            } catch (NumberFormatException e) {

            }

            try {

                val = ValueFactory.createValueByType(s, Types.BIGINT);

                return val;

            } catch (NumberFormatException e) {

            }

            try {

                val = ValueFactory.createValueByType(s, Types.FLOAT);

                return val;

            } catch (NumberFormatException e) {

            }

            try {

                val = ValueFactory.createValueByType(s, Types.DOUBLE);

                return val;

            } catch (NumberFormatException e) {

            }

            val = ValueFactory.createValueByType(s, Types.LONGVARCHAR);

        } catch (ParseException e) {

            e.printStackTrace();

        }

        return val;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.rendering.Legend#cloneLegend()

     */

    public Legend cloneLegend() throws XMLException {

        return (Legend) LegendFactory.createFromXML(getXMLEntity());

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.rendering.VectorialLegend#setDataSource(com.hardcode.gdbms.engine.data.DataSource)

     */

    public void setDataSource(DataSource ds)

        throws FieldNotFoundException, DriverException {

        try {

            dataSource = ds;

            ds.start();

            fieldId = ds.getFieldIndexByName(fieldName);

            ds.stop();

        } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {

            throw new DriverException(e);

        }

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.rendering.UniqueValueLegend#getSymbolByValue(com.hardcode.gdbms.engine.values.Value)

     */

    public FSymbol getSymbolByValue(Value key) {

        if (symbols.containsKey(key)) {

            return (FSymbol) symbols.get(key);

        } else if (useDefaultSymbol) {

            return getDefaultSymbol();

        }

        return null;

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.rendering.VectorialLegend#getShapeType()

     */

    public int getShapeType() {

        return shapeType;

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.rendering.VectorialLegend#getLabelHeightField()

     */

    public String getLabelHeightField() {

        return labelFieldHeight;

    }

    /**

     * Inserta el alto de campo.

     *

     * @param str alto.

     */

    public void setLabelHeightField(String str) {

        labelFieldHeight = str;

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.rendering.VectorialLegend#getLabelRotationField()

     */

    public String getLabelRotationField() {

        return labelFieldRotation;

    }

    /**

     * Inserta rotaci?n.

     *

     * @param str Rotaci?n.

     */

    public void setLabelRotationField(String str) {

        labelFieldRotation = str;

    }

    /**

     * Introduce si se tiene que representar el resto de valores o no.

     *

     * @param b True si se utiliza el resto de valores.

     */

    public void useDefaultSymbol(boolean b) {

        useDefaultSymbol = b;

    }

    /**

     * Devuelve si se utiliza o no el resto de valores para representarse.

     *

     * @return True si se utiliza el resto de valores.

     */

    public boolean isUseDefaultSymbol() {

        return useDefaultSymbol;

    }

    /**

     * Devuelve true si el etiquetado de la capa se ha modificado.

     *

     * @return True si se ha modificado el etiquetado de la capa.

     */

    /* public boolean isBWithHeightText() {

        return bWithHeightText;

    } */

    /**

     * Introduce si se ha modificado el etiquetado de la capa.

     *

     * @param withHeightText True si se ha modificado el etiquetado de la capa.

     */

    /* public void setBWithHeightText(boolean withHeightText) {

        bWithHeightText = withHeightText;

    } */

    /**

     * Elimina el s?mbolo que tiene como clave el valor que se pasa como par?metro.

     *

     * @param key clave.

     */

    public void delSymbol(Object key) {

        keys.remove(key);

        symbols.remove(key);

    }

}