hydra.lib.lists.Map Maven / Gradle / Ivy

Go to download
package hydra.lib.lists;

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.function.Function;
import java.util.List;
import java.util.stream.Collectors;

import static hydra.Flows.bind;
import static hydra.Flows.pure;
import static hydra.dsl.Types.function;
import static hydra.dsl.Types.lambda;
import static hydra.dsl.Types.list;

public class Map extends PrimitiveFunction {
    public Name name() {
        return new Name("hydra/lib/lists.map");
    }

    @Override
    public Type type() {
        return lambda("a", lambda("b",
                function(function("a", "b"), list("a"), list("b"))));
    }

    @Override
    protected Function>, Flow, Term>> implementation() {
        return args -> bind(Expect.list(instance -> pure(Terms.apply(args.get(0), instance)), args.get(1)),
            l -> pure(Terms.list(l)));
    }

    public static  Function, List> apply(Function mapping) {
        return (arg) -> apply(mapping, arg);
    }

    public static  List apply(Function mapping, List arg) {
        return arg.stream().map(mapping).collect(Collectors.toList());
    }
}