1 files changed, 119 insertions, 0 deletions
diff --git a/container-search/src/main/java/com/yahoo/prelude/query/GeoLocationItem.java b/container-search/src/main/java/com/yahoo/prelude/query/GeoLocationItem.java
new file mode 100644
index 00000000000..8202c8fb279
--- /dev/null
+++ b/container-search/src/main/java/com/yahoo/prelude/query/GeoLocationItem.java
@@ -0,0 +1,119 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+package com.yahoo.prelude.query;
+
+import com.google.common.annotations.Beta;
+import com.yahoo.prelude.Location;
+import java.nio.ByteBuffer;
+
+/**
+ * This represents a geo-location in the query tree.
+ * Used for closeness(fieldname) and distance(fieldname) rank features.
+ * @author arnej
+ */
+@Beta
+public class GeoLocationItem extends TermItem {
+
+    private Location location;
+
+    /**
+     * Construct from a Location, which must be geo circle with an attribute set.
+     **/
+    public GeoLocationItem(Location location) {
+        this(location, location.getAttribute());
+        if (! location.hasAttribute()) {
+            throw new IllegalArgumentException("missing attribute on location: "+location);
+        }
+    }
+
+    /**
+     * Construct from a Location and a field name.
+     * The Location must be a geo circle.
+     * If the Location has an attribute set, it must match the field name.
+     **/
+    public GeoLocationItem(Location location, String fieldName) {
+        super(fieldName, false);
+        if (location.hasAttribute() && ! location.getAttribute().equals(fieldName)) {
+            throw new IllegalArgumentException("inconsistent attribute on location: "+location.getAttribute()+" versus fieldName: "+fieldName);
+        }
+        if (! location.isGeoCircle()) {
+            throw new IllegalArgumentException("GeoLocationItem only supports Geo Circles, got: "+location);
+        }
+        if (location.hasBoundingBox()) {
+            throw new IllegalArgumentException("GeoLocationItem does not support bounding box yet, got: "+location);
+        }
+        this.location = new Location(location.toString());
+        this.location.setAttribute(null); // keep this in (superclass) indexName only
+        setNormalizable(false);
+    }
+
+    public Location getLocation() {
+        return location;
+    }
+
+    @Override
+    public String getRawWord() {
+        return stringValue();
+    }
+
+    @Override
+    public ItemType getItemType() {
+        return ItemType.GEO_LOCATION_TERM;
+    }
+
+    @Override
+    public String getName() {
+        return "GEO_LOCATION";
+    }
+
+    @Override
+    public String stringValue() {
+        return location.toString();
+    }
+
+    @Override
+    public void setValue(String value) {
+        throw new UnsupportedOperationException("Cannot setValue("+value+") on "+getName());
+    }
+
+    @Override
+    public int hashCode() {
+        return java.util.Objects.hash(super.hashCode(), location);
+    }
+
+    @Override
+    public boolean equals(Object object) {
+        if ( ! super.equals(object)) return false;
+        GeoLocationItem other = (GeoLocationItem) object; // Ensured by superclass
+        if ( ! location.equals(other.location)) return false;
+        return true;
+    }
+
+    @Override
+    public String getIndexedString() {
+        return location.toString();
+    }
+
+    @Override
+    protected void encodeThis(ByteBuffer buffer) {
+        super.encodeThis(buffer); // takes care of index bytes
+        // TODO: use a better format for encoding the location on the wire.
+        putString(location.backendString(), buffer);
+    }
+
+    @Override
+    public int getNumWords() {
+        return 1;
+    }
+
+    @Override
+    public boolean isStemmed() {
+        return true;
+    }
+
+    @Override
+    public boolean isWords() {
+        return false;
+    }
+
+}