lucene/spatial/src/java/org/apache/lucene/spatial/spatial4j/geo3d/GeoWideNorthRectangle.java - lucene-solr - Git at Google

 package org.apache.lucene.spatial.spatial4j.geo3d;

 /*
  * Licensed to the Apache Software Foundation (ASF) under one or more
  * contributor license agreements.  See the NOTICE file distributed with
  * this work for additional information regarding copyright ownership.
  * The ASF licenses this file to You under the Apache License, Version 2.0
  * (the "License"); you may not use this file except in compliance with
  * the License.  You may obtain a copy of the License at
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 /**
  * Bounding box wider than PI but limited on three sides (
  * bottom lat, left lon, right lon).
  *
  * @lucene.internal
  */
 public class GeoWideNorthRectangle extends GeoBaseBBox {
   public final double bottomLat;
   public final double leftLon;
   public final double rightLon;

   public final double cosMiddleLat;

   public final GeoPoint LRHC;
   public final GeoPoint LLHC;

   public final SidedPlane bottomPlane;
   public final SidedPlane leftPlane;
   public final SidedPlane rightPlane;

   public final GeoPoint[] bottomPlanePoints;
   public final GeoPoint[] leftPlanePoints;
   public final GeoPoint[] rightPlanePoints;

   public final GeoPoint centerPoint;

   public final EitherBound eitherBound;

   public final GeoPoint[] edgePoints;

   /**
    * Accepts only values in the following ranges: lat: {@code -PI/2 -> PI/2}, lon: {@code -PI -> PI}.
    * Horizontal angle must be greater than or equal to PI.
    */
   public GeoWideNorthRectangle(final PlanetModel planetModel, final double bottomLat, final double leftLon, double rightLon) {
     super(planetModel);
     // Argument checking
     if (bottomLat > Math.PI * 0.5 || bottomLat < -Math.PI * 0.5)
       throw new IllegalArgumentException("Bottom latitude out of range");
     if (leftLon < -Math.PI || leftLon > Math.PI)
       throw new IllegalArgumentException("Left longitude out of range");
     if (rightLon < -Math.PI || rightLon > Math.PI)
       throw new IllegalArgumentException("Right longitude out of range");
     double extent = rightLon - leftLon;
     if (extent < 0.0) {
       extent += 2.0 * Math.PI;
     }
     if (extent < Math.PI)
       throw new IllegalArgumentException("Width of rectangle too small");

     this.bottomLat = bottomLat;
     this.leftLon = leftLon;
     this.rightLon = rightLon;

     final double sinBottomLat = Math.sin(bottomLat);
     final double cosBottomLat = Math.cos(bottomLat);
     final double sinLeftLon = Math.sin(leftLon);
     final double cosLeftLon = Math.cos(leftLon);
     final double sinRightLon = Math.sin(rightLon);
     final double cosRightLon = Math.cos(rightLon);

     // Now build the four points
     this.LRHC = new GeoPoint(planetModel, sinBottomLat, sinRightLon, cosBottomLat, cosRightLon);
     this.LLHC = new GeoPoint(planetModel, sinBottomLat, sinLeftLon, cosBottomLat, cosLeftLon);

     final double middleLat = (Math.PI * 0.5 + bottomLat) * 0.5;
     final double sinMiddleLat = Math.sin(middleLat);
     this.cosMiddleLat = Math.cos(middleLat);
     // Normalize
     while (leftLon > rightLon) {
       rightLon += Math.PI * 2.0;
     }
     final double middleLon = (leftLon + rightLon) * 0.5;
     final double sinMiddleLon = Math.sin(middleLon);
     final double cosMiddleLon = Math.cos(middleLon);

     this.centerPoint = new GeoPoint(planetModel, sinMiddleLat, sinMiddleLon, cosMiddleLat, cosMiddleLon);

     this.bottomPlane = new SidedPlane(centerPoint, planetModel, sinBottomLat);
     this.leftPlane = new SidedPlane(centerPoint, cosLeftLon, sinLeftLon);
     this.rightPlane = new SidedPlane(centerPoint, cosRightLon, sinRightLon);

     this.bottomPlanePoints = new GeoPoint[]{LLHC, LRHC};
     this.leftPlanePoints = new GeoPoint[]{planetModel.NORTH_POLE, LLHC};
     this.rightPlanePoints = new GeoPoint[]{planetModel.NORTH_POLE, LRHC};

     this.eitherBound = new EitherBound();
     this.edgePoints = new GeoPoint[]{planetModel.NORTH_POLE};
   }

   @Override
   public GeoBBox expand(final double angle) {
     final double newTopLat = Math.PI * 0.5;
     final double newBottomLat = bottomLat - angle;
     // Figuring out when we escalate to a special case requires some prefiguring
     double currentLonSpan = rightLon - leftLon;
     if (currentLonSpan < 0.0)
       currentLonSpan += Math.PI * 2.0;
     double newLeftLon = leftLon - angle;
     double newRightLon = rightLon + angle;
     if (currentLonSpan + 2.0 * angle >= Math.PI * 2.0) {
       newLeftLon = -Math.PI;
       newRightLon = Math.PI;
     }
     return GeoBBoxFactory.makeGeoBBox(planetModel, newTopLat, newBottomLat, newLeftLon, newRightLon);
   }

   @Override
   public boolean isWithin(final Vector point) {
     return
         bottomPlane.isWithin(point) &&
             (leftPlane.isWithin(point) ||
                 rightPlane.isWithin(point));
   }

   @Override
   public boolean isWithin(final double x, final double y, final double z) {
     return
         bottomPlane.isWithin(x, y, z) &&
             (leftPlane.isWithin(x, y, z) ||
                 rightPlane.isWithin(x, y, z));
   }

   @Override
   public double getRadius() {
     // Here we compute the distance from the middle point to one of the corners.  However, we need to be careful
     // to use the longest of three distances: the distance to a corner on the top; the distnace to a corner on the bottom, and
     // the distance to the right or left edge from the center.
     final double centerAngle = (rightLon - (rightLon + leftLon) * 0.5) * cosMiddleLat;
     final double bottomAngle = centerPoint.arcDistance(LLHC);
     return Math.max(centerAngle, bottomAngle);
   }

   /**
    * Returns the center of a circle into which the area will be inscribed.
    *
    * @return the center.
    */
   @Override
   public GeoPoint getCenter() {
     return centerPoint;
   }

   @Override
   public GeoPoint[] getEdgePoints() {
     return edgePoints;
   }

   @Override
   public boolean intersects(final Plane p, final GeoPoint[] notablePoints, final Membership... bounds) {
     // Right and left bounds are essentially independent hemispheres; crossing into the wrong part of one
     // requires crossing into the right part of the other.  So intersection can ignore the left/right bounds.
     return
         p.intersects(planetModel, bottomPlane, notablePoints, bottomPlanePoints, bounds, eitherBound) ||
             p.intersects(planetModel, leftPlane, notablePoints, leftPlanePoints, bounds, bottomPlane) ||
             p.intersects(planetModel, rightPlane, notablePoints, rightPlanePoints, bounds, bottomPlane);
   }

   /**
    * Compute longitude/latitude bounds for the shape.
    *
    * @param bounds is the optional input bounds object.  If this is null,
    *               a bounds object will be created.  Otherwise, the input object will be modified.
    * @return a Bounds object describing the shape's bounds.  If the bounds cannot
    * be computed, then return a Bounds object with noLongitudeBound,
    * noTopLatitudeBound, and noBottomLatitudeBound.
    */
   @Override
   public Bounds getBounds(Bounds bounds) {
     if (bounds == null)
       bounds = new Bounds();
     bounds.noTopLatitudeBound().addLatitudeZone(bottomLat)
         .addLongitudeSlice(leftLon, rightLon);
     return bounds;
   }

   @Override
   public int getRelationship(final GeoShape path) {
     //System.err.println(this+" comparing to "+path);
     final int insideRectangle = isShapeInsideBBox(path);
     if (insideRectangle == SOME_INSIDE) {
       //System.err.println(" some inside");
       return OVERLAPS;
     }

     final boolean insideShape = path.isWithin(planetModel.NORTH_POLE);

     if (insideRectangle == ALL_INSIDE && insideShape) {
       //System.err.println(" both inside each other");
       return OVERLAPS;
     }

     if (
         path.intersects(bottomPlane, bottomPlanePoints, eitherBound) ||
             path.intersects(leftPlane, leftPlanePoints, bottomPlane) ||
             path.intersects(rightPlane, rightPlanePoints, bottomPlane)) {
       //System.err.println(" edges intersect");
       return OVERLAPS;
     }

     if (insideRectangle == ALL_INSIDE) {
       //System.err.println(" shape inside rectangle");
       return WITHIN;
     }

     if (insideShape) {
       //System.err.println(" rectangle inside shape");
       return CONTAINS;
     }

     //System.err.println(" disjoint");
     return DISJOINT;
   }

   @Override
   public boolean equals(Object o) {
     if (!(o instanceof GeoWideNorthRectangle))
       return false;
     GeoWideNorthRectangle other = (GeoWideNorthRectangle) o;
     return super.equals(other) && other.LLHC.equals(LLHC) && other.LRHC.equals(LRHC);
   }

   @Override
   public int hashCode() {
     int result = super.hashCode();
     result = 31 * result + LLHC.hashCode();
     result = 31 * result + LRHC.hashCode();
     return result;
   }

   @Override
   public String toString() {
     return "GeoWideNorthRectangle: {planetmodel="+planetModel+", bottomlat=" + bottomLat + "(" + bottomLat * 180.0 / Math.PI + "), leftlon=" + leftLon + "(" + leftLon * 180.0 / Math.PI + "), rightlon=" + rightLon + "(" + rightLon * 180.0 / Math.PI + ")}";
   }

   protected class EitherBound implements Membership {
     public EitherBound() {
     }

     @Override
     public boolean isWithin(final Vector v) {
       return leftPlane.isWithin(v) || rightPlane.isWithin(v);
     }

     @Override
     public boolean isWithin(final double x, final double y, final double z) {
       return leftPlane.isWithin(x, y, z) || rightPlane.isWithin(x, y, z);
     }
   }
 }
	package org.apache.lucene.spatial.spatial4j.geo3d;

	/*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You under the Apache License, Version 2.0
	* (the "License"); you may not use this file except in compliance with
	* the License. You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	/**
	* Bounding box wider than PI but limited on three sides (
	* bottom lat, left lon, right lon).
	*
	* @lucene.internal
	*/
	public class GeoWideNorthRectangle extends GeoBaseBBox {
	public final double bottomLat;
	public final double leftLon;
	public final double rightLon;

	public final double cosMiddleLat;

	public final GeoPoint LRHC;
	public final GeoPoint LLHC;

	public final SidedPlane bottomPlane;
	public final SidedPlane leftPlane;
	public final SidedPlane rightPlane;

	public final GeoPoint[] bottomPlanePoints;
	public final GeoPoint[] leftPlanePoints;
	public final GeoPoint[] rightPlanePoints;

	public final GeoPoint centerPoint;

	public final EitherBound eitherBound;

	public final GeoPoint[] edgePoints;

	/**
	* Accepts only values in the following ranges: lat: {@code -PI/2 -> PI/2}, lon: {@code -PI -> PI}.
	* Horizontal angle must be greater than or equal to PI.
	*/
	public GeoWideNorthRectangle(final PlanetModel planetModel, final double bottomLat, final double leftLon, double rightLon) {
	super(planetModel);
	// Argument checking
	if (bottomLat > Math.PI * 0.5 \|\| bottomLat < -Math.PI * 0.5)
	throw new IllegalArgumentException("Bottom latitude out of range");
	if (leftLon < -Math.PI \|\| leftLon > Math.PI)
	throw new IllegalArgumentException("Left longitude out of range");
	if (rightLon < -Math.PI \|\| rightLon > Math.PI)
	throw new IllegalArgumentException("Right longitude out of range");
	double extent = rightLon - leftLon;
	if (extent < 0.0) {
	extent += 2.0 * Math.PI;
	}
	if (extent < Math.PI)
	throw new IllegalArgumentException("Width of rectangle too small");

	this.bottomLat = bottomLat;
	this.leftLon = leftLon;
	this.rightLon = rightLon;

	final double sinBottomLat = Math.sin(bottomLat);
	final double cosBottomLat = Math.cos(bottomLat);
	final double sinLeftLon = Math.sin(leftLon);
	final double cosLeftLon = Math.cos(leftLon);
	final double sinRightLon = Math.sin(rightLon);
	final double cosRightLon = Math.cos(rightLon);

	// Now build the four points
	this.LRHC = new GeoPoint(planetModel, sinBottomLat, sinRightLon, cosBottomLat, cosRightLon);
	this.LLHC = new GeoPoint(planetModel, sinBottomLat, sinLeftLon, cosBottomLat, cosLeftLon);

	final double middleLat = (Math.PI * 0.5 + bottomLat) * 0.5;
	final double sinMiddleLat = Math.sin(middleLat);
	this.cosMiddleLat = Math.cos(middleLat);
	// Normalize
	while (leftLon > rightLon) {
	rightLon += Math.PI * 2.0;
	}
	final double middleLon = (leftLon + rightLon) * 0.5;
	final double sinMiddleLon = Math.sin(middleLon);
	final double cosMiddleLon = Math.cos(middleLon);

	this.centerPoint = new GeoPoint(planetModel, sinMiddleLat, sinMiddleLon, cosMiddleLat, cosMiddleLon);

	this.bottomPlane = new SidedPlane(centerPoint, planetModel, sinBottomLat);
	this.leftPlane = new SidedPlane(centerPoint, cosLeftLon, sinLeftLon);
	this.rightPlane = new SidedPlane(centerPoint, cosRightLon, sinRightLon);

	this.bottomPlanePoints = new GeoPoint[]{LLHC, LRHC};
	this.leftPlanePoints = new GeoPoint[]{planetModel.NORTH_POLE, LLHC};
	this.rightPlanePoints = new GeoPoint[]{planetModel.NORTH_POLE, LRHC};

	this.eitherBound = new EitherBound();
	this.edgePoints = new GeoPoint[]{planetModel.NORTH_POLE};
	}

	@Override
	public GeoBBox expand(final double angle) {
	final double newTopLat = Math.PI * 0.5;
	final double newBottomLat = bottomLat - angle;
	// Figuring out when we escalate to a special case requires some prefiguring
	double currentLonSpan = rightLon - leftLon;
	if (currentLonSpan < 0.0)
	currentLonSpan += Math.PI * 2.0;
	double newLeftLon = leftLon - angle;
	double newRightLon = rightLon + angle;
	if (currentLonSpan + 2.0 * angle >= Math.PI * 2.0) {
	newLeftLon = -Math.PI;
	newRightLon = Math.PI;
	}
	return GeoBBoxFactory.makeGeoBBox(planetModel, newTopLat, newBottomLat, newLeftLon, newRightLon);
	}

	@Override
	public boolean isWithin(final Vector point) {
	return
	bottomPlane.isWithin(point) &&
	(leftPlane.isWithin(point) \|\|
	rightPlane.isWithin(point));
	}

	@Override
	public boolean isWithin(final double x, final double y, final double z) {
	return
	bottomPlane.isWithin(x, y, z) &&
	(leftPlane.isWithin(x, y, z) \|\|
	rightPlane.isWithin(x, y, z));
	}

	@Override
	public double getRadius() {
	// Here we compute the distance from the middle point to one of the corners. However, we need to be careful
	// to use the longest of three distances: the distance to a corner on the top; the distnace to a corner on the bottom, and
	// the distance to the right or left edge from the center.
	final double centerAngle = (rightLon - (rightLon + leftLon) * 0.5) * cosMiddleLat;
	final double bottomAngle = centerPoint.arcDistance(LLHC);
	return Math.max(centerAngle, bottomAngle);
	}

	/**
	* Returns the center of a circle into which the area will be inscribed.
	*
	* @return the center.
	*/
	@Override
	public GeoPoint getCenter() {
	return centerPoint;
	}

	@Override
	public GeoPoint[] getEdgePoints() {
	return edgePoints;
	}

	@Override
	public boolean intersects(final Plane p, final GeoPoint[] notablePoints, final Membership... bounds) {
	// Right and left bounds are essentially independent hemispheres; crossing into the wrong part of one
	// requires crossing into the right part of the other. So intersection can ignore the left/right bounds.
	return
	p.intersects(planetModel, bottomPlane, notablePoints, bottomPlanePoints, bounds, eitherBound) \|\|
	p.intersects(planetModel, leftPlane, notablePoints, leftPlanePoints, bounds, bottomPlane) \|\|
	p.intersects(planetModel, rightPlane, notablePoints, rightPlanePoints, bounds, bottomPlane);
	}

	/**
	* Compute longitude/latitude bounds for the shape.
	*
	* @param bounds is the optional input bounds object. If this is null,
	* a bounds object will be created. Otherwise, the input object will be modified.
	* @return a Bounds object describing the shape's bounds. If the bounds cannot
	* be computed, then return a Bounds object with noLongitudeBound,
	* noTopLatitudeBound, and noBottomLatitudeBound.
	*/
	@Override
	public Bounds getBounds(Bounds bounds) {
	if (bounds == null)
	bounds = new Bounds();
	bounds.noTopLatitudeBound().addLatitudeZone(bottomLat)
	.addLongitudeSlice(leftLon, rightLon);
	return bounds;
	}

	@Override
	public int getRelationship(final GeoShape path) {
	//System.err.println(this+" comparing to "+path);
	final int insideRectangle = isShapeInsideBBox(path);
	if (insideRectangle == SOME_INSIDE) {
	//System.err.println(" some inside");
	return OVERLAPS;
	}

	final boolean insideShape = path.isWithin(planetModel.NORTH_POLE);

	if (insideRectangle == ALL_INSIDE && insideShape) {
	//System.err.println(" both inside each other");
	return OVERLAPS;
	}

	if (
	path.intersects(bottomPlane, bottomPlanePoints, eitherBound) \|\|
	path.intersects(leftPlane, leftPlanePoints, bottomPlane) \|\|
	path.intersects(rightPlane, rightPlanePoints, bottomPlane)) {
	//System.err.println(" edges intersect");
	return OVERLAPS;
	}

	if (insideRectangle == ALL_INSIDE) {
	//System.err.println(" shape inside rectangle");
	return WITHIN;
	}

	if (insideShape) {
	//System.err.println(" rectangle inside shape");
	return CONTAINS;
	}

	//System.err.println(" disjoint");
	return DISJOINT;
	}

	@Override
	public boolean equals(Object o) {
	if (!(o instanceof GeoWideNorthRectangle))
	return false;
	GeoWideNorthRectangle other = (GeoWideNorthRectangle) o;
	return super.equals(other) && other.LLHC.equals(LLHC) && other.LRHC.equals(LRHC);
	}

	@Override
	public int hashCode() {
	int result = super.hashCode();
	result = 31 * result + LLHC.hashCode();
	result = 31 * result + LRHC.hashCode();
	return result;
	}

	@Override
	public String toString() {
	return "GeoWideNorthRectangle: {planetmodel="+planetModel+", bottomlat=" + bottomLat + "(" + bottomLat * 180.0 / Math.PI + "), leftlon=" + leftLon + "(" + leftLon * 180.0 / Math.PI + "), rightlon=" + rightLon + "(" + rightLon * 180.0 / Math.PI + ")}";
	}

	protected class EitherBound implements Membership {
	public EitherBound() {
	}

	@Override
	public boolean isWithin(final Vector v) {
	return leftPlane.isWithin(v) \|\| rightPlane.isWithin(v);
	}

	@Override
	public boolean isWithin(final double x, final double y, final double z) {
	return leftPlane.isWithin(x, y, z) \|\| rightPlane.isWithin(x, y, z);
	}
	}
	}