1package io.protobase.f7.formulas.logic;
 2
 3import io.protobase.f7.formulas.AbstractFormula;
 4import io.protobase.f7.formulas.FormulaName;
 5import io.protobase.f7.models.GridColumnRowKey;
 6import io.protobase.f7.utils.Converters;
 7
 8import java.util.function.BiFunction;
 9import java.util.function.Function;
10
11public class IF extends AbstractFormula {
12  public static FormulaName NAME = FormulaName.IF;
13  public static IF SELF = new IF();
14
15  public IF() {
16    super();
17  }
18
19  public IF(Function<Object, Object> lookup, BiFunction<GridColumnRowKey, Object, Object> collateralLookup) {
20    super(lookup, collateralLookup);
21  }
22
23  @Override
24  public Object internal(GridColumnRowKey origin, Object... values) {
25    checkLength(values.length, 3, NAME);
26    Boolean logicalExpression = Converters.toBoolean(Converters.first(collateralLookup.apply(origin, values[0])));
27    if (logicalExpression) {
28      return collateralLookup.apply(origin, values[1]);
29    }
30    return collateralLookup.apply(origin, values[2]);
31  }
32}