Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.protelis.lang.interpreter.util.Op2 Maven / Gradle / Ivy
/*
* Copyright (C) 2021, Danilo Pianini and contributors listed in the project's build.gradle.kts or pom.xml file.
*
* This file is part of Protelis, and is distributed under the terms of the GNU General Public License,
* with a linking exception, as described in the file LICENSE.txt in this project's top directory.
*/
package org.protelis.lang.interpreter.util;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_AND;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_DIFFERS;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_DIVIDE;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_EQUALS;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_GREATER;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_GREATER_EQUAL;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_MAX;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_MIN;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_MINUS;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_MODULUS;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_OR;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_PLUS;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_POWER;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_SMALLER;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_SMALLER_EQUAL;
import static org.protelis.lang.interpreter.util.Bytecode.BINARY_TIMES;
import static org.protelis.lang.interpreter.util.OpUtils.unsupported;
import java.io.Serializable;
import java.util.Arrays;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.BiFunction;
import java.util.function.BinaryOperator;
import java.util.stream.IntStream;
import org.apache.commons.math3.util.FastMath;
import org.protelis.lang.datatype.DatatypeFactory;
import org.protelis.lang.datatype.Field;
import org.protelis.lang.datatype.Fields;
import org.protelis.lang.datatype.Tuple;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
/**
* Infix operator that takes two inputs, such as addition, division, or
* "greater than" comparison.
*/
public enum Op2 implements WithBytecode {
/** Logical AND operation. */
AND(BINARY_AND, "&&", Op2::and),
/** Inequality comparison operation. */
DIFFERS(BINARY_DIFFERS, "!=", (a, b) -> !Op2.areEquals(a, b)),
/** Arithmetic division operation. */
DIVIDE(BINARY_DIVIDE, "/", Op2::divide),
/** Equality comparison operation. */
EQUALS(BINARY_EQUALS, "==", Op2::areEquals),
/** Greater-than comparison operation. */
GREATER(BINARY_GREATER, ">", Op2::greater),
/** Greater-than-or-equal comparison operation. */
GREATER_EQUAL(BINARY_GREATER_EQUAL, ">=", Op2::greaterEquals),
/** Maximum of two numbers or other Comparable objects. */
MAX(BINARY_MAX, "max", Op2::max),
/** Minimum of two numbers or other Comparable objects. */
MIN(BINARY_MIN, "min", Op2::min),
/** Arithmetic subtraction operation. */
MINUS(BINARY_MINUS, "-", Op2::minus),
/** Modulus operation. */
MODULUS(BINARY_MODULUS, "%", Op2::modulus),
/** Logical OR operation. */
OR(BINARY_OR, "||", Op2::or),
/** Arithmetic addition operation. */
PLUS(BINARY_PLUS, "+", Op2::plus),
/** Exponent operation. */
POWER(BINARY_POWER, "^", Op2::pow),
/** Less-than comparison operation. */
SMALLER(BINARY_SMALLER, "<", Op2::smaller),
/** Less-than-or-equal comparison operation. */
SMALLER_EQUAL(BINARY_SMALLER_EQUAL, "<=", Op2::smallerEquals),
/** Arithmetic multiplication operation. */
TIMES(BINARY_TIMES, "*", Op2::times);
private static final String UNCHECKED = "unchecked";
private static final int[] BOTH = { 0, 1 };
private static final int[] LEFT = { 0 };
private static final int[] RIGHT = { 1 };
private static final int[] NONE = {};
private static final Map MAP = new ConcurrentHashMap<>();
private final BinaryOperation fun;
private final String opName;
private final Bytecode bytecode;
Op2(final Bytecode bytecode, final String name, final BinaryOperation function) {
this.bytecode = bytecode;
fun = function;
opName = name;
}
/**
* @return The function implementing this operator
*/
public BinaryOperator getFunction() {
return fun;
}
/**
* Compute the value of applying the two-argument operator to inputs a and
* b.
*
* @param a
* First input
* @param b
* Second input
* @return result of applying the operator to a and b
*/
public Object run(final Object a, final Object b) {
final boolean afield = a instanceof Field;
final boolean bfield = b instanceof Field;
final int[] fields = afield && bfield ? BOTH : afield ? LEFT : bfield ? RIGHT : NONE;
if (fields.length > 0) {
return Fields.apply(fun, fields, a, b);
}
return fun.apply(a, b);
}
@Override
public String toString() {
return opName;
}
/**
* Translates a name into an operator.
*
* @param name
* operator name
* @return an {@link Op2}
*/
public static Op2 getOp(final String name) {
Op2 op = MAP.get(name);
if (op == null) {
op = Arrays.stream(values()).filter(o -> o.opName.equals(name)).findFirst().get();
MAP.put(name, op);
}
return op;
}
private static Object and(final Object a, final Object b) {
return logical("&&", a, b, (v1, v2) -> v1 && v2);
}
private static Object divide(final Object a, final Object b) {
return arithmetic("/", a, b, (v1, v2) -> v1 / v2);
}
@SuppressFBWarnings("FE_FLOATING_POINT_EQUALITY")
private static boolean areEquals(final Object a, final Object b) {
if (a == null && b == null) {
return true;
}
if (a == null || b == null) {
return false;
}
if (a instanceof Number && b instanceof Number) {
return ((Number) a).doubleValue() == ((Number) b).doubleValue();
}
return a.equals(b);
}
private static boolean greater(final Object a, final Object b) {
return comparison(">", a, b, (v1, v2) -> v1 > v2);
}
private static boolean greaterEquals(final Object a, final Object b) {
return comparison(">=", a, b, (v1, v2) -> v1 >= v2);
}
@SuppressWarnings({ "rawtypes", UNCHECKED })
private static boolean comparison(final String op, final T a, final T b,
final BiFunction f) {
if (a instanceof Number && b instanceof Number) {
return f.apply(((Number) a).doubleValue(), ((Number) b).doubleValue());
}
try {
if (a instanceof Comparable && b instanceof Comparable) {
return f.apply((double) ((Comparable) a).compareTo(b), 0d);
}
} catch (RuntimeException e) { // NOPMD: comparison of different types
}
/*
* Fall back to lexicographic comparison
*/
return f.apply((double) a.toString().compareTo(b.toString()), 0d);
}
private static boolean logical(final String op, final T a, final T b,
final BiFunction f) {
if (a instanceof Boolean && b instanceof Boolean) {
return f.apply((Boolean) a, (Boolean) b);
}
return unsupported(op, a, b);
}
private static T selection(final String op, final T a, final T b, final BinaryOperator selector) {
final boolean aNum = a instanceof Number;
final boolean bNum = b instanceof Number;
if (aNum && bNum) {
final double ad = ((Number) a).doubleValue();
final double bd = ((Number) b).doubleValue();
if (ad > bd) {
return selector.apply(a, b);
}
return selector.apply(b, a);
}
if (a instanceof Comparable && b instanceof Comparable) {
try {
@SuppressWarnings({ "rawtypes", UNCHECKED })
final int v = ((Comparable) a).compareTo(b);
if (v > 0) {
return selector.apply(a, b);
}
return selector.apply(b, a);
} catch (RuntimeException e) { // NOPMD: RuntimeException caught willingly
/*
* Comparison of different types, fallback to lexicographic
* comparison
*/
}
}
/*
* Fall back to lexicographic comparison
*/
final int v = a.toString().compareTo(b.toString());
if (v > 0) {
return selector.apply(a, b);
}
return selector.apply(b, a);
}
private static Object min(final Object a, final Object b) {
return selection("min", a, b, (v1, v2) -> v2);
}
private static Object max(final Object a, final Object b) {
return selection("max", a, b, (v1, v2) -> v1);
}
private static Object minus(final Object a, final Object b) {
return arithmetic("-", a, b, (v1, v2) -> v1 - v2);
}
private static Object modulus(final Object a, final Object b) {
return arithmetic("%", a, b, (v1, v2) -> v1 % v2);
}
private static Object or(final Object a, final Object b) {
return logical("||", a, b, (v1, v2) -> v1 || v2);
}
private static Object plus(final Object a, final Object b) {
if (a instanceof CharSequence || b instanceof CharSequence) {
return a.toString() + b.toString();
}
try {
return arithmetic("+", a, b, (v1, v2) -> v1 + v2);
} catch (UnsupportedOperationException e) {
/*
* Sum falls back to string sum.
*/
return a.toString() + b.toString();
}
}
private static Object pow(final Object a, final Object b) {
return arithmetic("^", a, b, (v1, v2) -> FastMath.pow(v1, v2));
}
@SuppressWarnings(UNCHECKED)
private static O arithmetic(final String op, final I a, final I b, final BiFunction f) {
if (a instanceof Double && b instanceof Double) {
return f.apply((Double) a, (Double) b);
}
final boolean aNum = a instanceof Number;
final boolean bNum = b instanceof Number;
if (aNum && bNum) {
return f.apply(((Number) a).doubleValue(), ((Number) b).doubleValue());
}
final boolean aTup = a instanceof Tuple;
final boolean bTup = b instanceof Tuple;
if (aNum && bTup || aTup && bNum) {
return (O) tupleArithmetic(op, aNum, aNum ? a : b, (Tuple) (aTup ? a : b), f);
}
if (a instanceof Tuple && b instanceof Tuple) {
final Tuple ta = (Tuple) a;
final Tuple tb = (Tuple) b;
if (ta.size() == tb.size()) { // NOPMD: ints are compare with ==
return (O) DatatypeFactory.createTuple(IntStream.range(0, ta.size())
.mapToObj(i -> arithmetic(op, (I) ta.get(i), (I) tb.get(i), f))
.toArray());
}
}
return unsupported(op, a, b);
}
@SuppressWarnings(UNCHECKED)
private static Tuple tupleArithmetic(
final String op,
final boolean numFirst,
final I num,
final Tuple t,
final BiFunction f
) {
return DatatypeFactory.createTuple(IntStream.range(0, t.size())
.mapToObj(i ->
numFirst ? arithmetic(op, num, (I) t.get(i), f) : arithmetic(op, (I) t.get(i), num, f)
)
.toArray());
}
private static boolean smaller(final Object a, final Object b) {
return comparison("<", a, b, (v1, v2) -> v1 < v2);
}
private static boolean smallerEquals(final Object a, final Object b) {
return comparison("<=", a, b, (v1, v2) -> v1 <= v2);
}
private static Object times(final Object a, final Object b) {
return arithmetic("*", a, b, (v1, v2) -> v1 * v2);
}
@Override
public Bytecode getBytecode() {
return bytecode;
}
private interface BinaryOperation extends BinaryOperator, Serializable { }
}
