Application: gvSIG desktop

package org.gvsig.geocoding.index;

import java.util.ArrayList;

import java.util.List;

import java.util.Set;

import org.gvsig.LayerServer;

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

import org.gvsig.geocoding.DataGeocoder;

import org.gvsig.geocoding.address.Address;

import org.gvsig.geocoding.address.ComposedAddress;

import org.gvsig.geocoding.address.Literal;

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

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

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

import org.gvsig.geocoding.impl.LuceneGeocoderImpl;

import org.gvsig.geocoding.pattern.GeocodingPattern;

import org.gvsig.geocoding.pattern.GeocodingSource;

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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class LucenGeocoderImplTest extends

                org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase {

        private static final Logger log = LoggerFactory

                        .getLogger(LucenGeocoderImplTest.class);

        @Override

        protected void doSetUp() throws Exception {

                // TODO Auto-generated method stub

        }

        public void testStandardIndex() throws Exception {

                //get Pattern

                log.debug("Loading pattern ...");

                GeocodingPattern pattern = this.getPattern();

                //get Address

                log.debug("Loading address ...");

                Address address = this.getAddressLines();

                //get Pattern

                log.debug("Loading faeture store ...");

                FeatureStore store = LayerServer.getSHPStore(LayerServer

                                .getStreetsFile());

                // create geocoder

                log.debug("Builing geocoder ...");

                LuceneGeocoderImpl geocoder = new LuceneGeocoderImpl();                

                geocoder.setPattern(pattern);

                geocoder.setStore(store);

                //geocoding process

                log.debug("Run geocoding process ...");

                geocoder.indexer(store, pattern.getDataSource().getStyle().getRelationsLiteral());

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

        }

public void testComposedIndex() throws Exception {

                //get Pattern

                log.debug("Loading pattern ...");

                GeocodingPattern pattern = this.getComposedPattern();

                //get Address

                log.debug("Loading address ...");

                Address address = this.getCrossAddress();

                //get Pattern

                log.debug("Loading faeture store ...");

                FeatureStore store = LayerServer.getSHPStore(LayerServer

                                .getStreetsFile());

                // create geocoder

                log.debug("Builing geocoder ...");

                LuceneGeocoderImpl geocoder = new LuceneGeocoderImpl();                

                geocoder.setPattern(pattern);

                geocoder.setStore(store);

                //geocoding process

                log.debug("Run geocoding process ...");

                geocoder.indexer(store, pattern.getDataSource().getStyle().getRelationsLiteral());

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

        }

        /**

         * get pattern

         * 

         * @return

         */

        private GeocodingPattern getPattern() {

                GeocodingPattern pat = new DefaultGeocodingPattern();

                try {

                        pat.setPatternName("indexLucenePattern");

                        GeocodingSource source = new DefaultGeocodingSource();

                        source.setLayerName("Streets.shp");

                        source.setLayerProvider("Shape file");

                        AbstractGeocodingStyle style = new SimpleCentroid();

                        Literal relations = new DefaultLiteral();

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

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

                        relations.put("Municipio", "MUNI");

                        relations.put("Provincia", "PROV");

                        relations.put("Pais", "PAIS");

                        style.setRelationsLiteral(relations);

                        source.setStyle(style);

                        pat.setSource(source);

                } catch (Exception e) {

                        log.error("Building a pattern", e);

                }

                return pat;

        }

        /**

         * get Address Lines

         * 

         * @return

         */

        private Address getAddressLines() {

                Literal literal = new DefaultLiteral();

                literal.put("Calle", "colon");

                literal.put("TipoVia", "c");

                literal.put("Municipio", "valencia");

                literal.put("Provincia", "valencia");

                literal.put("Pais", "espanya");

                Address address = new DefaultAddress(literal);

                return address;

        }

        /**

         * get pattern

         * 

         * @return

         */

        private GeocodingPattern getComposedPattern() {

                GeocodingPattern pat = new DefaultGeocodingPattern();

                try {

                        pat.setPatternName("ComposedCentroid");

                        GeocodingSource source = new DefaultGeocodingSource();

                        AbstractGeocodingStyle style = new Composed();

                        Literal relations = new DefaultLiteral();

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

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

                        relations.put("Municipio", "MUNI");

                        relations.put("Provincia", "PROV");

                        relations.put("Pais", "PAIS");

                        style.setRelationsLiteral(relations);

                        source.setStyle(style);

                        source.setLayerName("Streets.shp");

                        source.setLayerProvider("Shape file");

                        pat.setSource(source);

                } catch (Exception e) {

                        log.error("Building a pattern", e);

                }

                return pat;

        }

        /**

         * get Address

         * 

         * @return

         */

        private Address getCrossAddress() {

                Literal literal = new DefaultLiteral();

                literal.put("Calle", "DOCTOR ROMAGOSA");

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

                literal.put("Municipio", "Valencia");

                literal.put("Provincia", "Valencia");

                literal.put("Pais", "Espanya");

                Literal literal2 = new DefaultLiteral();

                literal2.put("Calle", "DON JUAN DE AUSTRIA");

                literal2.put("TipoVia", "C");

                literal2.put("Municipio", "Valencia");

                literal2.put("Provincia", "Valencia");

                literal2.put("Pais", "Espanya");

                Address address = new DefaultComposedAddress(literal);

                List<Literal> intersect = new ArrayList<Literal>();

                intersect.add(literal2);

                ((ComposedAddress) address).setIntersectionLiterals(intersect);

                return address;

        }

        /**

         * get Address

         * 

         * @return

         */

        private Address getBetweenAddress() {

                DefaultLiteral literal = new DefaultLiteral();

                literal.put("Calle", "ROGER DE LAURIA");

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

                literal.put("Municipio", "Valencia");

                literal.put("Provincia", "Valencia");

                literal.put("Pais", "Espanya");

                Literal literal2 = new DefaultLiteral();

                literal2.put("Calle", "PASCUAL Y GENIS");

                literal2.put("TipoVia", "C");

                literal2.put("Municipio", "Valencia");

                literal2.put("Provincia", "Valencia");

                literal2.put("Pais", "Espanya");

                Literal literal3 = new DefaultLiteral();

                literal3.put("Calle", "PEREZ PUJOL");

                literal3.put("TipoVia", "C");

                literal3.put("Municipio", "Valencia");

                literal3.put("Provincia", "Valencia");

                literal3.put("Pais", "Espanya");

                Address address = new DefaultComposedAddress(literal);

                List<Literal> intersect = new ArrayList<Literal>();

                intersect.add(literal2);

                intersect.add(literal3);

                ((ComposedAddress) address).setIntersectionLiterals(intersect);

                return address;

        }

}