GeometryWrapper.java

package mil.nga.giat.geowave.core.geotime.store.dimension;

import mil.nga.giat.geowave.core.index.sfc.data.NumericData;
import mil.nga.giat.geowave.core.index.sfc.data.NumericRange;
import mil.nga.giat.geowave.core.store.dimension.NumericDimensionField;
import mil.nga.giat.geowave.core.store.index.CommonIndexValue;

import com.google.common.math.DoubleMath;
import com.vividsolutions.jts.geom.Coordinate;
import com.vividsolutions.jts.geom.Envelope;

/**
 * This class wraps JTS geometry with visibility so that it can be used within
 * GeoWave as a CommonIndexValue
 * 
 */
public class GeometryWrapper implements
		CommonIndexValue
{
	private byte[] visibility;
	private final com.vividsolutions.jts.geom.Geometry geometry;
	private static final double DOUBLE_TOLERANCE = 1E-12d;

	public GeometryWrapper(
			final com.vividsolutions.jts.geom.Geometry geometry ) {
		this.geometry = geometry;
	}

	public GeometryWrapper(
			final com.vividsolutions.jts.geom.Geometry geometry,
			final byte[] visibility ) {
		this.visibility = visibility;
		this.geometry = geometry;
	}

	@Override
	public void setVisibility(
			final byte[] visibility ) {
		this.visibility = visibility;
	}

	@Override
	public byte[] getVisibility() {
		return visibility;
	}

	public com.vividsolutions.jts.geom.Geometry getGeometry() {
		return geometry;
	}

	/**
	 * Expects Longitude before Latitude
	 */
	@Override
	public boolean overlaps(
			final NumericDimensionField[] fields,
			final NumericData[] rangeData ) {

		final int latPosition = fields[0] instanceof LatitudeField ? 0 : 1;
		final int longPosition = fields[0] instanceof LatitudeField ? 1 : 0;
		if (fields.length == 1) {
			final Envelope env = geometry.getEnvelopeInternal();
			final NumericRange r = latPosition == 0 ? new NumericRange(
					env.getMinY(),
					env.getMaxY()) : new NumericRange(
					env.getMinX(),
					env.getMaxX());
			return ((rangeData[0].getMin() < r.getMax()) || DoubleMath.fuzzyEquals(
					rangeData[0].getMin(),
					r.getMax(),
					DOUBLE_TOLERANCE)) && ((rangeData[0].getMax() > r.getMin()) || DoubleMath.fuzzyEquals(
					rangeData[0].getMax(),
					r.getMin(),
					DOUBLE_TOLERANCE));
		}
		return geometry.getFactory().createPolygon(
				new Coordinate[] {
					new Coordinate(
							rangeData[longPosition].getMin() - DOUBLE_TOLERANCE,
							rangeData[latPosition].getMin() - DOUBLE_TOLERANCE),
					new Coordinate(
							rangeData[longPosition].getMin() - DOUBLE_TOLERANCE,
							rangeData[latPosition].getMax() + DOUBLE_TOLERANCE),
					new Coordinate(
							rangeData[longPosition].getMax() + DOUBLE_TOLERANCE,
							rangeData[latPosition].getMax() + DOUBLE_TOLERANCE),
					new Coordinate(
							rangeData[longPosition].getMax() + DOUBLE_TOLERANCE,
							rangeData[latPosition].getMin() - DOUBLE_TOLERANCE),
					new Coordinate(
							rangeData[longPosition].getMin() - DOUBLE_TOLERANCE,
							rangeData[latPosition].getMin() - DOUBLE_TOLERANCE)
				}).intersects(
				geometry);
	}
}