[Part 1] Add geo support

pinot-common/src/main/java/org/apache/pinot/common/function/TransformFunctionType.java

@@ -57,8 +57,27 @@ public enum TransformFunctionType {
   MAPVALUE("mapValue"),
 
   // Special type for annotation based scalar functions
-  SCALAR("scalar");
+  SCALAR("scalar"),
+  // geo constructors
+  ST_GEOG_FROM_TEXT("ST_GeogFromText"),
+  ST_GEOM_FROM_TEXT("ST_GeomFromText"),
+  ST_GEOG_FROM_WKB("ST_GeogFromWKB"),
+  ST_GEOM_FROM_WKB("ST_GeomFromWKB"),
+  ST_POINT("ST_Point"),
+  ST_POLYGON("ST_Polygon"),
 
+  // geo measurements
+  ST_AREA("ST_Area"),
+  ST_DISTANCE("ST_Distance"),
+  ST_GEOMETRY_TYPE("ST_GEOMETRY_TYPE"),
+
+  // geo outputs
+  ST_AS_BINARY("ST_AsBinary"),
+  ST_AS_TEXT("ST_AsText"),
+
+  // geo relationship
+  ST_CONTAINS("ST_Contains"),
+  ST_EQUALS("ST_Equals");
   private final String _name;
 
   TransformFunctionType(String name) {

pinot-core/pom.xml

@@ -163,7 +163,14 @@
       <groupId>com.jayway.jsonpath</groupId>
       <artifactId>json-path</artifactId>
     </dependency>
-
+    <dependency>
+      <groupId>org.locationtech.jts</groupId>
+      <artifactId>jts-core</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>com.esotericsoftware</groupId>
+      <artifactId>kryo</artifactId>
+    </dependency>
     <!-- test -->
     <dependency>
       <groupId>org.apache.pinot</groupId>
@@ -227,5 +234,17 @@
       <artifactId>lucene-analyzers-common</artifactId>
       <version>${lucene.version}</version>
     </dependency>
+    <dependency>
+      <groupId>org.openjdk.jmh</groupId>
+      <artifactId>jmh-core</artifactId>
+      <version>${jmh.version}</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.openjdk.jmh</groupId>
+      <artifactId>jmh-generator-annprocess</artifactId>
+      <version>${jmh.version}</version>
+      <scope>test</scope>
+    </dependency>
   </dependencies>
 </project>

pinot-core/src/main/java/org/apache/pinot/core/geospatial/GeometryType.java

@@ -0,0 +1,49 @@
+/**
+ * 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.
+ */
+package org.apache.pinot.core.geospatial;
+
+/**
+ * The geometry type.
+ */
+public enum GeometryType {
+
+  POINT(false, "ST_Point"),
+  MULTI_POINT(true, "ST_MultiPoint"),
+  LINE_STRING(false, "ST_LineString"),
+  MULTI_LINE_STRING(true, "ST_MultiLineString"),
+  POLYGON(false, "ST_Polygon"),
+  MULTI_POLYGON(true, "ST_MultiPolygon"),
+  GEOMETRY_COLLECTION(true, "ST_GeomCollection");
+
+  private final boolean _multitype;
+  private final String _name;
+
+  GeometryType(boolean multitype, String name) {
+    _multitype = multitype;
+    _name = name;
+  }
+
+  public boolean isMultitype() {
+    return _multitype;
+  }
+
+  public String getName() {
+    return _name;
+  }
+}

pinot-core/src/main/java/org/apache/pinot/core/geospatial/GeometryUtils.java

@@ -0,0 +1,54 @@
+/**
+ * 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.
+ */
+package org.apache.pinot.core.geospatial;
+
+import com.google.common.base.Joiner;
+import org.locationtech.jts.geom.Geometry;
+import org.locationtech.jts.geom.GeometryFactory;
+import org.locationtech.jts.geom.PrecisionModel;
+
+
+/**
+ * Utility methods for the geometry.
+ */
+public class GeometryUtils {
+  /**
+   * Coordinate system of lat/lng per https://epsg.io/4326
+   */
+  public static int GEOGRAPHY_SRID = 4326;
+  public static byte GEOGRAPHY_SET_MASK = (byte) 0x80;
+  public static byte GEOGRAPHY_GET_MASK = (byte) 0x7f;
+  public static final GeometryFactory GEOMETRY_FACTORY = new GeometryFactory();
+  public static final GeometryFactory GEOGRAPHY_FACTORY = new GeometryFactory(new PrecisionModel(), GEOGRAPHY_SRID);
+  public static final double EARTH_RADIUS_KM = 6371.01;
+  public static final double EARTH_RADIUS_M = EARTH_RADIUS_KM * 1000.0;
+  public static final Joiner OR_JOINER = Joiner.on(" or ");
+
+  private GeometryUtils() {
+  }
+
+  /**
+   * Checks if the given geo-spatial object is a geography object.
+   * @param geometry the given object to check
+   * @return <code>true</code> if the given geo-spatial object is a geography object, <code>false</code> otherwise
+   */
+  public static boolean isGeography(Geometry geometry) {
+    return geometry.getSRID() == GEOGRAPHY_SRID;
+  }
+}