org.protelis.lang.util.Op3 Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of protelis Show documentation
Practical aggregate programming, hosted in Java
There is a newer version: 17.7.1
/*******************************************************************************
 * Copyright (C) 2014, 2015, Danilo Pianini and contributors
 * listed in the project's build.gradle 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.util;

import gnu.trove.list.TIntList;
import gnu.trove.list.array.TIntArrayList;

import java8.util.J8Arrays;
import java.util.Map;
import java8.util.Optional;
import java.util.concurrent.ConcurrentHashMap;

import org.danilopianini.lang.TriFunction;
import org.protelis.lang.datatype.Field;
import org.protelis.lang.datatype.Fields;

/**
 * Collection of functions and helper methods for ternary syntactic operators.
 */
public enum Op3 {

    /**
     * Functional mux.
     */
    MUX("mux", Op3::mux);

    private static final Map MAP = new ConcurrentHashMap<>();
    private final TriFunction fun;
    private final String opName;

    Op3(final String name, final TriFunction function) {
        fun = function;
        opName = name;
    }

    /**
     * @param a
     *            first argument
     * @param b
     *            second argument
     * @param c
     *            third argument
     * @return the function applied to the three arguments. Any of them can be a
     *         {@link Field}.
     */
    public Object run(final Object a, final Object b, final Object c) {
        final Object[] args = { a, b, c };
        Optional idx = Optional.empty();
        for (int i = 0; i < args.length; i++) {
            if (args[i] instanceof Field) {
                if (!idx.isPresent()) {
                    idx = Optional.of(new TIntArrayList(3));
                }
                idx.get().add(i);
            }
        }
        if (idx.isPresent()) {
            return Fields.apply(fun, idx.get().toArray(), a, b, c);
        }
        return fun.apply(a, b, c);
    }

    @Override
    public String toString() {
        return opName;
    }

    /**
     * @param name
     *            the operation name
     * @return the corresponding {@link Op3}
     */
    public static Op3 getOp(final String name) {
        Op3 op = MAP.get(name);
        if (op == null) {
            op = J8Arrays.stream(values()).parallel().filter(o -> o.opName.equals(name)).findFirst().get();
            MAP.put(name, op);
        }
        return op;
    }

    private static Object mux(final Object a, final Object b, final Object c) {
        if (a instanceof Boolean) {
            if ((Boolean) a) {
                return b;
            }
            return c;
        } else {
            if (a != null) {
                return b;
            }
            return c;
        }
    }
}