/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* JBoss, Home of Professional Open Source
* Copyright 2011 Red Hat Inc. and/or its affiliates and other contributors
* as indicated by the @authors tag. All rights reserved.
* See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This copyrighted material is made available to anyone wishing to use,
* modify, copy, or redistribute it subject to the terms and conditions
* of the GNU Lesser General Public License, v. 2.1.
* This program is distributed in the hope that it will be useful, but WITHOUT A
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
* PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
* You should have received a copy of the GNU Lesser General Public License,
* v.2.1 along with this distribution; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*/
package org.hibernate.search.spatial.impl;
import java.io.IOException;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.search.DocIdSet;
import org.apache.lucene.search.FieldCache;
import org.apache.lucene.search.Filter;
import org.apache.lucene.search.FilteredDocIdSet;
import org.apache.lucene.search.MatchAllDocsQuery;
import org.apache.lucene.search.QueryWrapperFilter;
/**
* Lucene Filter for filtering documents which have been indexed with Hibernate Search spatial Field bridge
* Use double lat,long field in the index from a Coordinates field declaration
*
* @author Nicolas Helleringer <nicolas.helleringer@novacodex.net>
* @see org.hibernate.search.spatial.SpatialFieldBridgeByQuadTree
* @see org.hibernate.search.spatial.SpatialFieldBridgeByRange
* @see org.hibernate.search.spatial.Coordinates
*/
public final class DistanceFilter extends Filter {
private Filter previousFilter;
private Point center;
private double radius;
private String coordinatesField;
private String latitudeField;
private String longitudeField;
/**
* Construct a Distance Filter to match document distant at most of radius from center Point
*
* @param previousFilter previous Filter in the chain. As Distance is costly by retrieving the lat and long field
* it is better to use it last
* @param center center of the search perimeter
* @param radius radius of the search perimeter
* @param coordinatesField name of the field implementing Coordinates
* @see org.hibernate.search.spatial.Coordinates
*/
public DistanceFilter(Filter previousFilter, Point center, double radius, String coordinatesField) {
if ( previousFilter != null ) {
this.previousFilter = previousFilter;
}
else {
this.previousFilter = new QueryWrapperFilter( new MatchAllDocsQuery() );
}
this.center = center;
this.radius = radius;
this.coordinatesField = coordinatesField;
}
/**
* Construct a Distance Filter to match document distant at most of radius from center Point
*
* @param previousFilter previous Filter in the chain. As Distance is costly by retrieving the lat and long field
* it is better to use it last
* @param center center of the search perimeter
* @param radius radius of the search perimeter
* @param latitudeField name of the field hosting latitude
* @param longitudeField name of the field hosting longitude
* @see org.hibernate.search.spatial.Coordinates
*/
public DistanceFilter(Filter previousFilter, Point center, double radius, String latitudeField, String longitudeField) {
if ( previousFilter != null ) {
this.previousFilter = previousFilter;
}
else {
this.previousFilter = new QueryWrapperFilter( new MatchAllDocsQuery() );
}
this.center = center;
this.radius = radius;
this.coordinatesField = null;
this.latitudeField = latitudeField;
this.longitudeField = longitudeField;
}
/**
* Returns Doc Ids by retrieving their lat,long and checking if within distance(radius) of the center of the search
*
* @param reader reader to the index
*/
@Override
public DocIdSet getDocIdSet(IndexReader reader) throws IOException {
final double[] latitudeValues = FieldCache.DEFAULT.getDoubles( reader, getLatitudeField() );
final double[] longitudeValues = FieldCache.DEFAULT.getDoubles( reader, getLongitudeField() );
DocIdSet docs = previousFilter.getDocIdSet( reader );
if ( ( docs == null ) || ( docs.iterator() == null ) ) {
return null;
}
return new FilteredDocIdSet( docs ) {
@Override
protected boolean match(int documentIndex) {
if ( center.getDistanceTo( latitudeValues[documentIndex], longitudeValues[documentIndex] ) <= radius ) {
return true;
}
else {
return false;
}
}
};
}
private String getLatitudeField() {
if ( latitudeField != null ) {
return latitudeField;
}
else {
return SpatialHelper.formatLatitude( coordinatesField );
}
}
private String getLongitudeField() {
if ( longitudeField != null ) {
return longitudeField;
}
else {
return SpatialHelper.formatLongitude( coordinatesField );
}
}
@Override
public String toString() {
final StringBuilder sb = new StringBuilder();
sb.append( "DistanceFilter" );
sb.append( "{previousFilter=" ).append( previousFilter );
sb.append( ", center=" ).append( center );
sb.append( ", radius=" ).append( radius );
if ( coordinatesField != null) {
sb.append( ", coordinatesField='" ).append( coordinatesField ).append( '\'' );
}
else {
sb.append( ", latitudeField=" ).append( latitudeField );
sb.append( ", longitudeField=" ).append( longitudeField ).append( '\'' );
}
sb.append( '}' );
return sb.toString();
}
}