Metacat

/**

 *  '$RCSfile$'

 *  Copyright: 2000 Regents of the University of California and the

 *             National Center for Ecological Analysis and Synthesis

 *

 * Author: Matthew Perry 

 * '$Date: 2006-09-13 16:10:55 -0700 (Wed, 13 Sep 2006) $'

 * '$Revision: 3044 $'

 *

 * 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

 */

package edu.ucsb.nceas.metacat.spatial;

import org.geotools.feature.FeatureType;

import org.geotools.data.shapefile.ShapefileDataStore;

import org.geotools.data.FeatureStore;

import org.geotools.data.FeatureSource;

import org.geotools.data.DefaultTransaction;

import org.geotools.data.Transaction;

import org.geotools.feature.Feature;

import org.geotools.filter.Filter;

import org.geotools.filter.AbstractFilter;

import org.geotools.filter.Expression;

import org.geotools.filter.CompareFilter;

import org.geotools.filter.GeometryFilter;

import org.geotools.filter.FilterFactory;

import org.geotools.filter.FilterFactoryFinder;

import org.geotools.filter.IllegalFilterException;

import org.geotools.feature.FeatureCollection;

import org.geotools.feature.FeatureCollections;

import com.vividsolutions.jts.geom.Envelope;

import java.io.File;

import java.net.URI;

import java.net.URL;

import java.net.MalformedURLException;

import java.sql.ResultSet;

import java.sql.PreparedStatement;

import java.util.Vector;

import java.util.Iterator; 

import java.io.IOException;

import org.apache.log4j.Logger;

import edu.ucsb.nceas.metacat.MetaCatUtil;

/** 

 * Class to query the persistent spatial cache

 * and returns docids matching spatial constraints

 */

public class SpatialQuery {

  private static Logger log = Logger.getLogger(SpatialQuery.class.getName());

  /** 

   * empty constructor to initialize spatial query

   */

  public SpatialQuery() { }

  /**

   * Querys all features in the spatial cache 

   * and filters based on bouding coordinates.

   * Returns Vector of docids.

   *

   * @param w West bounding coordinate 

   * @param s South bounding coordinate 

   * @param e East bounding coordinate 

   * @param n North bounding coordinate 

   *

   */

  public Vector filterByBbox( Float w, Float s, Float e, Float n ) {

      Vector docids = new Vector();

      SpatialFeatureSchema featureSchema = new SpatialFeatureSchema();

      // MPTDO, perform geotools spatial query

      // test

      docids.add("test1.1");

      docids.add("test2.1");

      ShapefileDataStore store = null;

      FeatureSource features = null;

      FeatureCollection collection = null;

      FilterFactory filterFactory = FilterFactoryFinder.createFilterFactory();

      try {

          store = new ShapefileDataStore( (new File( featureSchema.polygonShpUri )).toURL() );

          features = store.getFeatureSource(store.getTypeNames()[0]);

          //count features

          //col = features.getFeatures();

          //System.out.println("# cities (no filter)= "+col.size());

          Envelope envelope = new Envelope( w, e, s, n ); 

          Expression bbox = filterFactory.createBBoxExpression( envelope );

          FeatureType featureType = store.getSchema( store.getTypeNames()[0] );

          Expression geometry = filterFactory.createAttributeExpression( featureType.getDefaultGeometry().getName()  );

          GeometryFilter bboxFilter = filterFactory.createGeometryFilter(AbstractFilter.GEOMETRY_BBOX);

          bboxFilter.addLeftGeometry( geometry );

          bboxFilter.addRightGeometry( bbox );

          //count filtered features

          collection = features.getFeatures(bboxFilter);

          //System.out.println("# cities (after filter)= "+col.size());

          Iterator iterator = collection.iterator();

          try {

              for( Iterator i=collection.iterator(); i.hasNext(); ) {

                  Feature feature = (Feature) i.next();

                  docids.add( (String) feature.getAttribute(1) );

              }

          } finally {

              collection.close( iterator );

          }

        } catch (MalformedURLException ex) {

            ex.printStackTrace();

        } catch (IOException ex) {

            ex.printStackTrace();

        } catch (IllegalFilterException ex) {

            ex.printStackTrace();

        }

      return docids;

  }

}