package com.datastax.dse.driver.internal.core.graph;
import com.datastax.dse.driver.api.core.data.geometry.Geometry;
import com.datastax.dse.driver.api.core.data.geometry.LineString;
import com.datastax.dse.driver.api.core.data.geometry.Point;
import com.datastax.dse.driver.api.core.data.geometry.Polygon;
import com.datastax.dse.driver.api.core.graph.predicates.Geo;
import com.datastax.dse.driver.internal.core.data.geometry.Distance;
import com.datastax.oss.driver.shaded.guava.common.base.Preconditions;
public enum GeoPredicate implements DsePredicate {
inside {
@Override
public boolean test(Object value, Object condition) {
preEvaluate(condition);
if (value == null) {
return false;
}
Preconditions.checkArgument(value instanceof Geometry);
Distance distance = (Distance) condition;
if (value instanceof Point) {
return haversineDistanceInDegrees(distance.getCenter(), (Point) value)
<= distance.getRadius();
} else if (value instanceof Polygon) {
for (Point point : ((Polygon) value).getExteriorRing()) {
if (haversineDistanceInDegrees(distance.getCenter(), point) > distance.getRadius()) {
return false;
}
}
} else if (value instanceof LineString) {
for (Point point : ((LineString) value).getPoints()) {
if (haversineDistanceInDegrees(distance.getCenter(), point) > distance.getRadius()) {
return false;
}
}
} else {
throw new UnsupportedOperationException(
String.format("Value type '%s' unsupported", value.getClass().getName()));
}
return true;
}
@Override
public String toString() {
return "inside";
}
},
insideCartesian {
@Override
public boolean test(Object value, Object condition) {
preEvaluate(condition);
if (value == null) {
return false;
}
Preconditions.checkArgument(value instanceof Geometry);
return ((Geometry) condition).contains((Geometry) value);
}
@Override
public String toString() {
return "insideCartesian";
}
};
@Override
public boolean isValidCondition(Object condition) {
return condition != null;
}
static double haversineDistanceInDegrees(Point p1, Point p2) {
double dLat = Math.toRadians(p2.Y() - p1.Y());
double dLon = Math.toRadians(p2.X() - p1.X());
double lat1 = Math.toRadians(p1.Y());
double lat2 = Math.toRadians(p2.Y());
double a =
Math.pow(Math.sin(dLat / 2), 2)
+ Math.pow(Math.sin(dLon / 2), 2) * Math.cos(lat1) * Math.cos(lat2);
double c = 2 * Math.asin(Math.sqrt(a));
return Math.toDegrees(c);
}
}