| gvSIG desktop 1

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

 *

 * Copyright (C) 2005 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

 */

/* CVS MESSAGES:

*

* $Id: AbstractIntervalLegend.java 10293 2007-02-13 16:19:19Z jaume $

* $Log$

* Revision 1.1.2.1  2007-02-13 16:19:19  jaume

* graduated symbol legends (start commiting)

*

*

*/

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

import java.util.ArrayList;

import java.util.Comparator;

import java.util.TreeMap;

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

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

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

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.ISymbol;

import com.iver.cit.gvsig.fmap.core.symbols.SimpleFillSymbol;

import com.iver.cit.gvsig.fmap.core.symbols.SimpleLineSymbol;

import com.iver.cit.gvsig.fmap.core.symbols.SimpleMarkerSymbol;

public abstract class AbstractIntervalLegend implements IntervalLegend{

        public static final int EQUAL_INTERVALS = 0;

        public static final int NATURAL_INTERVALS = 1;

        public static final int QUANTILE_INTERVALS = 2;

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

                public int compare(Object o1, Object o2) {

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

                                if (o1 instanceof NullIntervalValue &&

                                                o2 instanceof NullIntervalValue) {

                                        return 0;

                                }

                                if (o2 instanceof NullIntervalValue) {

                                        return 1;

                                }

                                if (o1 instanceof NullIntervalValue) {

                                        return -1;

                                }

                                FInterval i2 = (FInterval) o2;

                                FInterval i1 = (FInterval) o1;

                                if (i1.getMin() > i2.getMin()) {

                                        return 1;

                                }

                                if (i1.getMin() < i2.getMin()) {

                                        return -1;

                                }

                                if (i1.getMax() < i2.getMax()) {

                                        return -1;

                                }

                                if (i1.getMax() > i2.getMax()) {

                                        return 1;

                                }

                        }

                        return 0;

                }

        });

        protected ArrayList keys = new ArrayList();

        protected int index = 0;

        protected String fieldName;

        private int fieldId;

        private ISymbol defaultSymbol;

        private DataSource dataSource;

        private int shapeType;

        protected int intervalType = NATURAL_INTERVALS;

        protected boolean useDefaultSymbol = false;

        protected String labelFieldName;

        protected String labelFieldHeight;

        protected String labelFieldRotation;

        public void addSymbol(Object key, ISymbol symbol) {

                //TODO guardar los intervalos.

                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.m_Descrip +

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

                 } else {

                 */

                keys.add(key);

                //}

        }

        /**

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

         */

        public ISymbol getSymbol(int recordIndex) throws DriverException {

                try {

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

                        IInterval interval = getInterval(val);

                        ISymbol theSymbol = getSymbolByInterval(interval);

                        if (theSymbol != null) {

                                return theSymbol;

                        } else if (useDefaultSymbol) {

                                return getDefaultSymbol();

                        }

                        return null;

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

                        throw new DriverException(e);

                }

        }

        /**

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

         * OJO!! Cuando usamos un feature iterator de base de datos

         * el ?nico campo que vendr? rellenado es el de fieldID.

         * Los dem?s vendr?n a nulos para ahorra tiempo de creaci?n.

         *

         * @param feat IFeature.

         *

         * @return S?mbolo.

         */

        public ISymbol getSymbolByFeature(IFeature feat) {

                Value val = feat.getAttribute(fieldId);

                IInterval interval = getInterval(val);

                ISymbol theSymbol = getSymbolByInterval(interval);

                if (theSymbol != null) {

                        return theSymbol;

                } else {

                        return getDefaultSymbol();

                }

        }

        /**

         * Devuelve el s?mbolo a partir del intervalo.

         *

         * @param key intervalo.

         *

         * @return s?mbolo.

         */

        public ISymbol getSymbolByInterval(IInterval key) {

                if (key == null) {

                        return null;

                }

                if (symbols.containsKey(key)) {

                        return (ISymbol) symbols.get(key);

                }

                return null;

        }

        public String[] getDescriptions() {

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

                ISymbol[] auxSym = getSymbols();

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

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

                return descriptions;

        }

        public Object[] getValues() {

                return symbols.keySet().toArray();

        }

        public void setIntervalSymbol(IInterval interval, ISymbol symbol) {

                /*symbols.put(interval, symbol);

                 values.put(new Integer(index), interval);

                 index++;

                 */

        }

        public void changeInterval(int index, IInterval newInterval) {

                /*Object value = values.remove(new Integer(index));

                 Object symbol = symbols.remove(value);

                 values.put(new Integer(index), newInterval);

                 symbols.put(newInterval, symbol);

                 */

        }

        public void clear() {

                index = 0;

                keys.clear();

                symbols.clear();

        }

        public ISymbol[] getSymbols() {

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

        }

        public String getFieldName() {

                return fieldName;

        }

        public void setDefaultSymbol(ISymbol s) {

                defaultSymbol = s;

        }

        public void setFieldName(String str) {

                fieldName = str;

        }

        public void setLabelField(String fieldName) {

                labelFieldName = fieldName;

        }

        public String getLabelField() {

                return labelFieldName;

        }

        public ISymbol getDefaultSymbol() {

                NullIntervalValue niv=new NullIntervalValue();

                if (symbols.containsKey(niv))

                        return (ISymbol)symbols.get(niv);

                return defaultSymbol;

        }

        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);

                }

        }

        /**

         * Devuelve el intervalo a partir del valor.

         *

         * @param v valor.

         *

         * @return intervalo.

         */

        public IInterval getInterval(Value v) {

                /*if (v instanceof NullValue){

                 System.out.println("Si");

                 }*/

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

                        if (((IInterval) keys.get(i)).isInInterval(v)) {

                                return (IInterval) keys.get(i);

                        }

                }

                return null;

        }

        public int getShapeType() {

                return shapeType;

        }

        public void setShapeType(int shapeType) {

                if (this.shapeType != shapeType) {

                        switch (shapeType) {

                        case FShape.POINT:

                                defaultSymbol = new SimpleMarkerSymbol();

                                break;

                        case FShape.LINE:

                                defaultSymbol = new SimpleLineSymbol();

                                break;

                        case FShape.POLYGON:

                                defaultSymbol = new SimpleFillSymbol();

                                break;

                        default:

                                throw new Error("Unknown symbol type");

                        //                  defaultSymbol = new FSymbol(shapeType);

                        }

                        this.shapeType = shapeType;

                }

        }

        /**

         * Inserta el tipo de clasificaci?n de los intervalos.

         *

         * @param tipoClasificacion Tipo de clasificaci?n.

         */

        public void setIntervalType(int tipoClasificacion) {

                intervalType = tipoClasificacion;

        }

        /**

         * Devuelve el tipo de clasificaci?n de los intervalos.

         *

         * @return Tipo de clasificaci?n.

         */

        public int getIntervalType() {

                return intervalType;

        }

        /**

         * Inserta si se representan el resto de valores o no.

         *

         * @param b True si se tienen que representar 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;

        }

        /**

         * Elimina un s?mbolo a partir de su clave.

         *

         * @param obj clave.

         */

        public void delSymbol(Object obj) {

                keys.remove(obj);

                symbols.remove(obj);

        }

        public String[] getUsedFields() {

                ArrayList usedFields = new ArrayList();

                if (getFieldName() != null)

                        usedFields.add(getFieldName());

                if (getLabelField() != null)

                        usedFields.add(getLabelField());

                if (getLabelHeightField() != null)

                        usedFields.add(getLabelHeightField());

                if (getLabelRotationField() != null)

                        usedFields.add(getLabelRotationField());

                return (String[]) usedFields.toArray(new String[0]);

        }

        public String getLabelHeightField() {

            return labelFieldHeight;

        }

        /**

         * Inserta el alto del campo.

         *

         * @param str alto.

         */

        public void setLabelHeightField(String str) {

            labelFieldHeight = str;

        }

        public String getLabelRotationField() {

            return labelFieldRotation;

        }

        /**

         * Inserta la rotaci?n del campo.

         *

         * @param str Rotaci?n.

         */

        public void setLabelRotationField(String str) {

            labelFieldRotation = str;

        }

}