1 /***********************************************************************
2  * Copyright (c) 2013-2024 Commonwealth Computer Research, Inc.
3  * All rights reserved. This program and the accompanying materials
4  * are made available under the terms of the Apache License, Version 2.0
5  * which accompanies this distribution and is available at
6  * http://www.opensource.org/licenses/apache2.0.php.
7  ***********************************************************************/
8 
9 package org.locationtech.geomesa.utils.index
10 
11 import org.locationtech.jts.geom.{Envelope, Geometry}
12 import org.locationtech.jts.index.{SpatialIndex => JSTSpatialIndex}
13 
14 import scala.collection.JavaConverters._
15 
16 class WrapperIndex[T, Index <: JSTSpatialIndex](val indexBuider : () => Index) extends SpatialIndex[T] with Serializable {
17 
18   protected var index = indexBuider()
19 
20   override def insert(geom: Geometry, key: String, value: T): Unit = index.insert(geom.getEnvelopeInternal, (key, value))
21 
22   override def remove(geom: Geometry, key: String): T = {
23     val envelope = geom.getEnvelopeInternal
24     index.query(envelope).asScala.asInstanceOf[Seq[(String, T)]].find(_._1 == key) match {
25       case None => null.asInstanceOf[T]
26       case Some(kv) => index.remove(envelope, kv); kv._2
27     }
28   }
29 
30   override def get(geom: Geometry, key: String): T = {
31     val intersect = index.query(geom.getEnvelopeInternal).asScala.asInstanceOf[Seq[(String, T)]]
32     intersect.find(_._1 == key).map(_._2).getOrElse(null.asInstanceOf[T])
33   }
34 
35   override def query(xmin: Double, ymin: Double, xmax: Double, ymax: Double): Iterator[T] =
36     index.query(new Envelope(xmin, xmax, ymin, ymax)).iterator.asScala.asInstanceOf[Iterator[(String, T)]].map(_._2)
37 
38   override def query(): Iterator[T] = {
39     query(-180,-90,180,90)
40   }
41 
42   override def size(): Int = query().size
43 
44   override def clear(): Unit = index = indexBuider()
45 }