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

Go to download
package hydra.lib.flows;

import hydra.Flows;
import hydra.compute.Flow;
import hydra.compute.FlowState;
import hydra.core.Name;
import hydra.core.Term;
import hydra.core.Type;
import hydra.dsl.Types;
import hydra.dsl.prims.Optionals;
import hydra.graph.Graph;
import hydra.tools.PrimitiveFunction;

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

import static hydra.dsl.Terms.app;
import static hydra.dsl.Terms.flowState;
import static hydra.dsl.Terms.lambda;
import static hydra.dsl.Terms.project;
import static hydra.dsl.Terms.unwrap;
import static hydra.dsl.Terms.variable;


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

    @Override
    public Type type() {
        return Types.lambda("s", "x", "y",
                Types.function(Types.function("x", "y"), Types.flow("s", "x"), Types.flow("s", "y")));
    }

    @Override
    protected Function>, Flow, Term>> implementation() {
        return args -> {
            Term mapping = args.get(0);
            Term input = args.get(1);
            Term output = lambda("s", "t",
                    app(lambda("q", flowState(
                                    app(Optionals.map(), mapping, app(project(FlowState.NAME, "value"), variable("q"))),
                                    app(project(FlowState.NAME, "state"), variable("q")),
                                    app(project(FlowState.NAME, "trace"), variable("q")))),
                            (app(unwrap(Flow.NAME), input, variable("s"), variable("t")))));
            return Flows.pure(output);
        };
    }

    public static  Function, Flow> apply(Function mapping) {
        return input -> apply(mapping, input);
    }

    public static  Flow apply(Function mapping, Flow input) {
        return Flows.map(mapping, input);
    }
}