Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 * 

 * 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., 51 Franklin Street, Fifth Floor, Boston, 

 * MA  02110-1301, USA.

 * 

 */

/*

 * AUTHORS (In addition to CIT):

 * 2008 PRODEVELOP                Main development

 */

package org.gvsig.geocoding;

import java.util.Iterator;

import java.util.Set;

import org.gvsig.LayerServer;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.exception.InitializeException;

import org.gvsig.fmap.dal.exception.ProviderNotRegisteredException;

import org.gvsig.fmap.dal.exception.ValidateDataParametersException;

import org.gvsig.fmap.dal.feature.FeatureQuery;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.geom.primitive.Point;

import org.gvsig.geocoding.address.Address;

import org.gvsig.geocoding.address.Literal;

import org.gvsig.geocoding.address.impl.DefaultAddress;

import org.gvsig.geocoding.address.impl.DefaultLiteral;

import org.gvsig.geocoding.impl.DataGeocoderImpl;

import org.gvsig.geocoding.impl.DefaultGeocodingLibrary;

import org.gvsig.geocoding.pattern.GeocodingPattern;

import org.gvsig.geocoding.pattern.GeocodingSettings;

import org.gvsig.geocoding.pattern.GeocodingSource;

import org.gvsig.geocoding.pattern.impl.DefaultGeocodingPattern;

import org.gvsig.geocoding.pattern.impl.DefaultGeocodingSettings;

import org.gvsig.geocoding.pattern.impl.DefaultGeocodingSource;

import org.gvsig.geocoding.result.GeocodingResult;

import org.gvsig.geocoding.styles.AbstractGeocodingStyle;

import org.gvsig.geocoding.styles.impl.SimpleCentroid;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.evaluator.Evaluator;

import org.gvsig.tools.evaluator.sqljep.SQLJEPEvaluator;

import org.gvsig.tools.evaluator.sqljep.SQLJEPLibrary;

import org.gvsig.tools.library.Library;

import org.jfree.util.Log;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Test

 * 

 * @author <a href="mailto:jsanz@prodevelop.es"> Jorge Gaspar Sanz Salinas</a>

 * @author <a href="mailto:vsanjaime@prodevelop.es"> Vicent Sanjaime Calvet</a>

 */

public class GeocoderTest extends

                org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase {

        protected Logger logger = LoggerFactory.getLogger(getClass());;

        @Override

        protected void doSetUp() throws Exception {

                Library geocoLib = new DefaultGeocodingLibrary();

                geocoLib.initialize();

                geocoLib.postInitialize();

                Library sql = new SQLJEPLibrary();

                sql.initialize();

                sql.postInitialize();

                DataManager manager = DALLocator.getDataManager();

                manager.registerDefaultEvaluator(SQLJEPEvaluator.class);

        }

        protected void tearDown() throws Exception {

                super.tearDown();

        }

        public void testGeocode() throws Exception {

                /* Set the pattern */

                GeocodingPattern pat = getAPattern();

                Geocoder geocoder = GeocodingLocator.getInstance().getGeocoder();

                ((DataGeocoderImpl) geocoder).setPattern(pat);

                ((DataGeocoderImpl) geocoder).setStore(LayerServer

                                .getSHPStore(LayerServer.getStreetsFile()));

                /* Define a address search */

                Literal literal = new DefaultLiteral();

                literal.put("via", "C");

                literal.put("calle", "LIRIA");

                Address address = new DefaultAddress(literal);

                Set<GeocodingResult> results = geocoder.geocode(address);

                assertEquals(554, results.size());

                Iterator<GeocodingResult> it = results.iterator();

                while (it.hasNext()) {

                        GeocodingResult res = it.next();

                        Point pto = (Point) res.getGeometry();

                        logger.debug("PTO: " + pto.getX() + "," + pto.getY());

                }

        }

        public void testFilter() throws Exception {

                String expression = "STREET_NAM = 'LIRIA'";

                FeatureStore store = (FeatureStore) LayerServer.getSHPStore(LayerServer

                                .getStreetsFile());

                FeatureQuery query = store.createFeatureQuery();

                DataManager manager = DALLocator.getDataManager();

                Evaluator filter = manager.createExpresion(expression);

                assertNotNull(filter);

                query.setFilter(filter);

                FeatureSet set = store.getFeatureSet(query);

                assertNotNull(set);                

                logger.info("SELECTED FEATURES:"+set.getSize());

                assertTrue(set.getSize()==2);

        }

        private GeocodingPattern getAPattern() throws InitializeException,

                        ProviderNotRegisteredException, ValidateDataParametersException {

                GeocodingPattern pat = new DefaultGeocodingPattern();

                pat.setPatternName("Test pattern");

                GeocodingSource source = new DefaultGeocodingSource();

                AbstractGeocodingStyle style = new SimpleCentroid();

                Literal relations = new DefaultLiteral();

                relations.put("via", "STREET_TYP");

                relations.put("calle", "STREET_NAM");

                style.setRelationsLiteral(relations);

                source.setStyle(style);

                source.setLayerName("Streets.shp");

                source.setLayerProvider("Shape file");

                pat.setSource(source);

                GeocodingSettings settings = new DefaultGeocodingSettings();

                pat.setSettings(settings);

                return pat;

        }

}