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 
10 package org.locationtech.geomesa.filter.function
11 
12 import org.geotools.api.temporal.Instant
13 import org.geotools.filter.FunctionExpressionImpl
14 import org.geotools.filter.capability.FunctionNameImpl
15 import org.geotools.filter.capability.FunctionNameImpl._
16 import org.locationtech.geomesa.utils.bin.BinaryEncodeCallback.ByteArrayCallback
17 import org.locationtech.geomesa.utils.bin.BinaryOutputEncoder
18 import org.locationtech.jts.geom.{Geometry, Point}
19 
20 import java.time.temporal.{ChronoField, Temporal}
21 import java.util.{Base64, Date}
22 
23 class Convert2ViewerFunction extends FunctionExpressionImpl(Convert2ViewerFunction.Name) {
24 
25   override def evaluate(obj: AnyRef): String = {
26     val id    = getExpression(0).evaluate(obj).asInstanceOf[String]
27     val track = BinaryOutputEncoder.convertToTrack(id)
28     val label = BinaryOutputEncoder.convertToLabel(id)
29     val geom  = getExpression(1).evaluate(obj).asInstanceOf[Point]
30     val dtg   = Convert2ViewerFunction.dtg2Long(getExpression(2).evaluate(obj))
31     ByteArrayCallback.apply(track, geom.getY.toFloat, geom.getX.toFloat, dtg, label)
32     Base64.getEncoder.encodeToString(ByteArrayCallback.result)
33   }
34 }
35 
36 object Convert2ViewerFunction {
37 
38   val Name = new FunctionNameImpl(
39     "convert2viewer",
40     classOf[String],
41     parameter("id", classOf[String]),
42     parameter("geom", classOf[Geometry]),
43     parameter("dtg", classOf[Long])
44   )
45 
46   private def dtg2Long(dtg: Any): Long = dtg match {
47     case d: Long     => d
48     case d: Date     => d.getTime
49     case d: Temporal => d.getLong(ChronoField.INSTANT_SECONDS) + d.get(ChronoField.MILLI_OF_SECOND)
50     case d: Instant  => d.getPosition.getDate.getTime
51   }
52 }