git-svn-id: https://svn.apache.org/repos/asf/lucene/solr/trunk@895660 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Grant Ingersoll 2010-01-04 15:27:06 +00:00
parent f6ac4ff2bb
commit 5838738379
1 changed files with 28 additions and 28 deletions

View File

@ -22,23 +22,21 @@ import org.apache.solr.common.SolrException;
/** /**
* Useful distance utiltities. * Useful distance utiltities.
* solr-internal: subject to change w/o notification. * solr-internal: subject to change w/o notification.
* */
**/
public class DistanceUtils { public class DistanceUtils {
public static final double DEGREES_TO_RADIANS = Math.PI / 180.0; public static final double DEGREES_TO_RADIANS = Math.PI / 180.0;
public static final double RADIANS_TO_DEGREES = 180.0 / Math.PI; public static final double RADIANS_TO_DEGREES = 180.0 / Math.PI;
/** /**
* @see org.apache.solr.search.function.distance.HaversineFunction * @param x1 The x coordinate of the first point
* * @param y1 The y coordinate of the first point
* @param x1 The x coordinate of the first point * @param x2 The x coordinate of the second point
* @param y1 The y coordinate of the first point * @param y2 The y coordinate of the second point
* @param x2 The x coordinate of the second point
* @param y2 The y coordinate of the second point
* @param radius The radius of the sphere * @param radius The radius of the sphere
* @return The distance between the two points, as determined by the Haversine formula. * @return The distance between the two points, as determined by the Haversine formula.
* @see org.apache.solr.search.function.distance.HaversineFunction
*/ */
public static double haversine(double x1, double y1, double x2, double y2, double radius){ public static double haversine(double x1, double y1, double x2, double y2, double radius) {
double result = 0; double result = 0;
//make sure they aren't all the same, as then we can just return 0 //make sure they aren't all the same, as then we can just return 0
if ((x1 != x2) || (y1 != y2)) { if ((x1 != x2) || (y1 != y2)) {
@ -56,38 +54,37 @@ public class DistanceUtils {
/** /**
* Given a string containing <i>dimension</i> values encoded in it, separated by commas, return a String array of length <i>dimension</i> * Given a string containing <i>dimension</i> values encoded in it, separated by commas, return a String array of length <i>dimension</i>
* containing the values. * containing the values.
* @param out A preallocated array. Must be size dimension. If it is not it will be resized.
* @param externalVal The value to parse
* @param dimension The expected number of values for the point
* @return An array of the values that make up the point (aka vector)
* *
* @param out A preallocated array. Must be size dimension. If it is not it will be resized.
* @param externalVal The value to parse
* @param dimension The expected number of values for the point
* @return An array of the values that make up the point (aka vector)
* @throws {@link SolrException} if the dimension specified does not match the number of values in the externalValue. * @throws {@link SolrException} if the dimension specified does not match the number of values in the externalValue.
*/ */
public static String[] parsePoint(String[] out, String externalVal, int dimension) { public static String[] parsePoint(String[] out, String externalVal, int dimension) {
//TODO: Should we support sparse vectors? //TODO: Should we support sparse vectors?
if (out==null || out.length != dimension) out=new String[dimension]; if (out == null || out.length != dimension) out = new String[dimension];
int idx = externalVal.indexOf(','); int idx = externalVal.indexOf(',');
int end = idx; int end = idx;
int start = 0; int start = 0;
int i = 0; int i = 0;
if (idx == -1 && dimension == 1 && externalVal.length() > 0){//we have a single point, dimension better be 1 if (idx == -1 && dimension == 1 && externalVal.length() > 0) {//we have a single point, dimension better be 1
out[0] = externalVal.trim(); out[0] = externalVal.trim();
i = 1; i = 1;
} } else if (idx > 0) {//if it is zero, that is an error
else if (idx > 0) {//if it is zero, that is an error
//Parse out a comma separated list of point values, as in: 73.5,89.2,7773.4 //Parse out a comma separated list of point values, as in: 73.5,89.2,7773.4
for (; i < dimension; i++){ for (; i < dimension; i++) {
while (start<end && externalVal.charAt(start)==' ') start++; while (start < end && externalVal.charAt(start) == ' ') start++;
while (end>start && externalVal.charAt(end-1)==' ') end--; while (end > start && externalVal.charAt(end - 1) == ' ') end--;
out[i] = externalVal.substring(start, end); out[i] = externalVal.substring(start, end);
start = idx+1; start = idx + 1;
end = externalVal.indexOf(',', start); end = externalVal.indexOf(',', start);
if (end == -1){ if (end == -1) {
end = externalVal.length(); end = externalVal.length();
} }
} }
} }
if (i != dimension){ if (i != dimension) {
throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, "incompatible dimension (" + dimension + throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, "incompatible dimension (" + dimension +
") and values (" + externalVal + "). Only " + i + " values specified"); ") and values (" + externalVal + "). Only " + i + " values specified");
} }
@ -96,13 +93,16 @@ public class DistanceUtils {
/** /**
* extract (by calling {@link #parsePoint(String[], String, int)} and validate the latitude and longitude contained * extract (by calling {@link #parsePoint(String[], String, int)} and validate the latitude and longitude contained
* in the String by making sure the latitude is between 90 & -90 and longitude is between -180 and 180 * in the String by making sure the latitude is between 90 & -90 and longitude is between -180 and 180.
* @param latLon A preallocated array to hold the result * <p/>
* The latitude is assumed to be the first part of the string and the longitude the second part.
*
* @param latLon A preallocated array to hold the result
* @param latLonStr The string to parse * @param latLonStr The string to parse
* @return The lat long * @return The lat long
*/ */
public static final double[] parseLatitudeLongitude(double [] latLon, String latLonStr) { public static final double[] parseLatitudeLongitude(double[] latLon, String latLonStr) {
if (latLon == null){ if (latLon == null) {
latLon = new double[2]; latLon = new double[2];
} }
String[] toks = DistanceUtils.parsePoint(null, latLonStr, 2); String[] toks = DistanceUtils.parsePoint(null, latLonStr, 2);