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.convert2.transforms
10 
11 import org.locationtech.geomesa.convert.EvaluationContext
12 import org.locationtech.geomesa.convert.EvaluationContext.ContextDependent
13 
14 sealed trait Predicate extends ContextDependent[Predicate] {
15   def apply(args: Array[AnyRef]): Boolean
16 }
17 
18 object Predicate {
19 
20   def apply(e: String): Predicate = PredicateParser.parse(e)
21 
22   case class BinaryEquals(left: Expression, right: Expression) extends Predicate {
23     override def apply(args: Array[AnyRef]): Boolean = left(args) == right(args)
24     override def withContext(ec: EvaluationContext): Predicate =
25       BinaryEquals(left.withContext(ec), right.withContext(ec))
26   }
27 
28   case class BinaryNotEquals(left: Expression, right: Expression) extends Predicate {
29     override def apply(args: Array[AnyRef]): Boolean = left(args) != right(args)
30     override def withContext(ec: EvaluationContext): Predicate =
31       BinaryNotEquals(left.withContext(ec), right.withContext(ec))
32   }
33 
34   case class BinaryLessThan(left: Expression, right: Expression) extends Predicate {
35     override def apply(args: Array[AnyRef]): Boolean =
36       left(args).asInstanceOf[Comparable[AnyRef]].compareTo(right(args)) < 0
37     override def withContext(ec: EvaluationContext): Predicate =
38       BinaryLessThan(left.withContext(ec), right.withContext(ec))
39   }
40 
41   case class BinaryLessThanOrEquals(left: Expression, right: Expression) extends Predicate {
42     override def apply(args: Array[AnyRef]): Boolean =
43       left(args).asInstanceOf[Comparable[AnyRef]].compareTo(right(args)) <= 0
44     override def withContext(ec: EvaluationContext): Predicate =
45       BinaryLessThanOrEquals(left.withContext(ec), right.withContext(ec))
46   }
47 
48   case class BinaryGreaterThan(left: Expression, right: Expression) extends Predicate {
49     override def apply(args: Array[AnyRef]): Boolean =
50       left(args).asInstanceOf[Comparable[AnyRef]].compareTo(right(args)) > 0
51     override def withContext(ec: EvaluationContext): Predicate =
52       BinaryGreaterThan(left.withContext(ec), right.withContext(ec))
53   }
54 
55   case class BinaryGreaterThanOrEquals(left: Expression, right: Expression) extends Predicate {
56     override def apply(args: Array[AnyRef]): Boolean =
57       left(args).asInstanceOf[Comparable[AnyRef]].compareTo(right(args)) >= 0
58     override def withContext(ec: EvaluationContext): Predicate =
59       BinaryGreaterThanOrEquals(left.withContext(ec), right.withContext(ec))
60   }
61 
62   case class And(clause: Predicate, clauses: Seq[Predicate]) extends Predicate {
63     override def apply(args: Array[AnyRef]): Boolean =
64       clause(args) && clauses.forall(_.apply(args))
65     override def withContext(ec: EvaluationContext): Predicate =
66       And(clause.withContext(ec), clauses.map(_.withContext(ec)))
67   }
68 
69   case class Or(clause: Predicate, clauses: Seq[Predicate])  extends Predicate {
70     override def apply(args: Array[AnyRef]): Boolean =
71       clause(args) || clauses.exists(_.apply(args))
72     override def withContext(ec: EvaluationContext): Predicate =
73       Or(clause.withContext(ec), clauses.map(_.withContext(ec)))
74   }
75 
76   case class Not(p: Predicate) extends Predicate {
77     override def apply(args: Array[AnyRef]): Boolean = !p(args)
78     override def withContext(ec: EvaluationContext): Predicate = Not(p.withContext(ec))
79   }
80 }