hydra.lib.sets.Insert Maven / Gradle / Ivy

Go to download
package hydra.lib.sets;

import hydra.Flows;
import hydra.compute.Flow;
import hydra.core.Name;
import hydra.core.Term;
import hydra.core.Type;
import hydra.dsl.Expect;
import hydra.dsl.Terms;
import hydra.graph.Graph;
import hydra.tools.PrimitiveFunction;

import java.util.List;
import java.util.function.Function;
import java.util.HashSet;
import java.util.Set;

import static hydra.dsl.Types.function;
import static hydra.dsl.Types.lambda;
import static hydra.dsl.Types.set;


public class Insert extends PrimitiveFunction {
    public Name name() {
        return new Name("hydra/lib/sets.insert");
    }

    @Override
    public Type type() {
        return lambda("x", function("x", set("x"), set("x")));
    }

    @Override
    protected Function>, Flow, Term>> implementation() {
        return args -> Flows.map(Expect.set(Flows::pure, args.get(1)), arg -> Terms.set(apply(args.get(0), arg)));
    }

    public static  Function, Set> apply(X elem) {
        return (arg) -> apply(elem, arg);
    }

    /**
     * Apply the function to both arguments.
     */
    public static  Set apply(X elem, Set arg) {
        Set newSet = new HashSet<>(arg);
        newSet.add(elem);
        return newSet;
    }
}