| gvSIG desktop 1

package com.iver.cit.gvsig.fmap.drivers.kml;

import java.awt.Color;

import java.awt.Font;

import java.io.BufferedInputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.io.InputStream;

import java.util.Random;

import java.util.zip.ZipEntry;

import java.util.zip.ZipFile;

import java.util.zip.ZipInputStream;

import org.apache.log4j.Logger;

import org.gvsig.exceptions.BaseException;

import org.gvsig.remoteClient.gml.engine.IFeaturesIterator;

import org.gvsig.remoteClient.gml.exceptions.GMLException;

import org.gvsig.remoteClient.gml.factories.IGeometriesFactory;

import org.gvsig.remoteClient.kml.KmlDriver;

import org.gvsig.remoteClient.kml.KmlTags;

import org.gvsig.remoteClient.kml.exceptions.KmlException;

import com.hardcode.gdbms.driver.exceptions.InitializeDriverException;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.hardcode.gdbms.engine.data.driver.ObjectDriver;

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

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

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

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

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

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

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

import com.iver.cit.gvsig.fmap.drivers.ConcreteMemoryDriver;

import com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver;

import com.iver.cit.gvsig.fmap.drivers.WithDefaultLegend;

import com.iver.cit.gvsig.fmap.drivers.gml.FMAPGeometryFactory;

import com.iver.cit.gvsig.fmap.drivers.gml.FMAPGeometryFactory.FeatureWithAttributes;

import com.iver.cit.gvsig.fmap.rendering.Legend;

import com.iver.cit.gvsig.fmap.rendering.LegendFactory;

import com.iver.cit.gvsig.fmap.rendering.VectorialUniqueValueLegend;

/* 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

 */

/* CVS MESSAGES:

 *

 * $Id: KmlFMapDriver.java 10627 2007-03-06 17:10:21Z caballero $

 * $Log$

 * Revision 1.2  2007-03-06 17:09:00  caballero

 * Exceptions

 *

 * Revision 1.1  2007/02/12 13:50:08  jorpiell

 * A?adido el driver de KML

 *

 *

 */

/**

 * @author Jorge Piera Llodr? (piera_jor@gva.es)

 */

public class KmlFMapDriver extends ConcreteMemoryDriver implements VectorialFileDriver,WithDefaultLegend{

        private static Logger logger = Logger.getLogger(KmlDriver.class.getName());

        private File m_Fich;

        private VectorialUniqueValueLegend defaultLegend = null;

        private FSymbol defaultSymbol = null;

        private IGeometriesFactory factory = null;

        private KmlDriver driver = null;

        private IFeaturesIterator iterator = null;

        private static final float heightText = 10;

        private Random r = null;

        //it contains the position for the folder name.

        private int FOLDER_NAME_POSITION = 0;

        /*

         *  (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#open(java.io.File)

         */

        public void open(File f) {

                m_Fich = f;

                factory = new FMAPGeometryFactory();

                r = new Random();

        }

        /*

         *  (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#close()

         */

        public void close() {

                // TODO Auto-generated method stub

        }

        /*

         *  (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#initialize()

         */

        public void initialize() throws InitializeDriverException {

                try {

                        driver = new KmlDriver(createInputStream(),factory);

                        iterator =         driver.getFeaturesIterator();

                } catch (BaseException e) {

                        logger.error("Error openning the kml file",e);

                        throw new InitializeDriverException(getName(),e);

                }

                int index = 0;

                IntValue clave;

                boolean setModel = false;

                FeatureWithAttributes feature;

                try {

                        while (iterator.hasNext()) {

                                try{

                                        feature = (FeatureWithAttributes)iterator.next();

                                        //There are features without geometry

                                        if (feature.getFeature().getGeometry() != null){

                                                if (!setModel){

                                                        getTableModel().setColumnIdentifiers(feature.getAttributeName());

                                                        setModel = true;

                                                }

                                                clave = ValueFactory.createValue(index);

                                                IGeometry geom = feature.getFeature().getGeometry();

                                                if (geom != null){

                                                        FSymbol theSymbol = null;

                                                        theSymbol = getDefaultSymbol();

                                                        theSymbol.setDescription(clave.toString());

                                                        addGeometry(geom, feature.getAttributeValue());

                                                }

                                        }

                                        index++;

                                }catch(GMLException e){

                                        logger.error(e.getMessage(),e);

                                        e.printStackTrace();

                                }

                        }

                }catch (BaseException e) {

                        logger.error("Error retrieving the geometries",e);

                        e.printStackTrace();

                }

                try {

                        createLegend();

                } catch (ReadDriverException e) {

                        throw new InitializeDriverException(getName(),e);

                }

        }

        /**

         * Creates the legend. The unique value is the field

         * that is specified in the KmlTags.FOLDER_NAME constant

         * @throws IOException

         */

        private void createLegend() throws ReadDriverException{

                ObjectDriver rs = this;

                StringValue clave;

                FSymbol theSymbol = null;

                for (long j = 0; j < rs.getRowCount(); j++) {

                        clave = (StringValue) rs.getFieldValue(j, FOLDER_NAME_POSITION);

                        if (((VectorialUniqueValueLegend)getDefaultLegend()).getSymbolByValue(clave) == null) {

                                theSymbol = new FSymbol(getShapeType());

                                theSymbol.setDescription(clave.toString());

                                Color c = getRandomColor();

                                theSymbol.setColor(c);

                                theSymbol.setStyle(FConstant.SYMBOL_STYLE_FILL_SOLID);

                                theSymbol.setSize(3);

                                theSymbol.setSizeInPixels(true);

                                defaultLegend.addSymbol(clave, theSymbol);

                        }

                }

        }

        /*

         *  (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#accept(java.io.File)

         */

        public boolean accept(File f) {

                if ((f.getName().toUpperCase().endsWith("KML")) ||

                                (f.getName().toUpperCase().endsWith("KMZ"))){

                        return true;

                }

                return false;

        }

        /*

         *  (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#getFile()

         */

        public File getFile() {

                return m_Fich;

        }

        /*

         *  (non-Javadoc)

         * @see com.hardcode.driverManager.Driver#getName()

         */

        public String getName() {

                return "gvSIG KML Memory Driver";

        }

        /*

         *  (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getShapeType()

         */

        public int getShapeType() {

                return FShape.MULTI;

        }

        /**

         * It returns a random color

         * @return

         */

        private Color getRandomColor(){

                return new Color(r.nextInt(255),r.nextInt(255),r.nextInt(255));

        }

        /**

         * Creates the default symbol

         * @return

         */

        private FSymbol getDefaultSymbol(){

                if (defaultSymbol == null){

                        defaultSymbol = new FSymbol(getShapeType());

                        defaultSymbol.setShapeVisible(false);

                        defaultSymbol.setFontSizeInPixels(false);

                        defaultSymbol.setFont(new Font("SansSerif",Font.PLAIN, 9));

                        defaultSymbol.setFontColor(Color.BLACK);

                        defaultSymbol.setFontSize(heightText);

                        defaultSymbol.setStyle(FConstant.SYMBOL_STYLE_FILL_SOLID);

                        defaultSymbol.setSize(3);

                        defaultSymbol.setSizeInPixels(true);

                }

                return defaultSymbol;

        }

        /*

         *  (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.WithDefaultLegend#getDefaultLegend()

         */

        public Legend getDefaultLegend() {

                if (defaultLegend == null){

                        defaultLegend = LegendFactory.createVectorialUniqueValueLegend(getShapeType());

                        defaultLegend.setFieldName(KmlTags.FOLDER_NAME);

                        defaultLegend.setLabelField(null);

                        defaultLegend.setLabelHeightField(null);

                        defaultLegend.setLabelRotationField(null);

                        defaultLegend.setDefaultSymbol(getDefaultSymbol());

                }

                return defaultLegend;

        }

        /**

         * It creates an InputStream. The Kml file can have the

         * KML or the KMZ extension

         * @return

         * @throws KmlException

         */

        private InputStream createInputStream() throws KmlException{

                try {

                        if (getFile().getName().toUpperCase().endsWith("KML")){

                                return new FileInputStream(getFile());

                        }else if(getFile().getName().toUpperCase().endsWith("KMZ")){

                                FileInputStream fis = new FileInputStream(getFile());

                                ZipInputStream zis = new ZipInputStream(new BufferedInputStream(fis));

                                ZipEntry entry = null;

                                while((entry = zis.getNextEntry()) != null){

                                        if (!entry.isDirectory()){

                                                ZipFile fz = new ZipFile(getFile());

                                                return fz.getInputStream(entry);

                                        }

                                }

                        }else{

                                throw new KmlException(getFile());

                        }

                } catch (FileNotFoundException e) {

                        throw new KmlException(getFile(),e);

                } catch (IOException e) {

                        throw new KmlException(getFile(),e);

                }

                return null;

        }

}