
hydra.lib.sets.Union Maven / Gradle / Ivy
package hydra.lib.sets;
import hydra.Flows;
import hydra.compute.Flow;
import hydra.core.Name;
import hydra.core.Term;
import hydra.core.TypeScheme;
import hydra.dsl.Expect;
import hydra.dsl.Terms;
import hydra.graph.Graph;
import hydra.tools.PrimitiveFunction;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.function.Function;
import static hydra.dsl.Types.function;
import static hydra.dsl.Types.scheme;
import static hydra.dsl.Types.set;
public class Union extends PrimitiveFunction {
public Name name() {
return new Name("hydra/lib/sets.union");
}
@Override
public TypeScheme type() {
return scheme("x", function(set("x"), set("x"), set("x")));
}
@Override
protected Function, Flow> implementation() {
return args -> Flows.map2(
Expect.set(Flows::pure, args.get(0)),
Expect.set(Flows::pure, args.get(1)),
(s1, s2) -> Terms.set(apply(s1, s2)));
}
public static Function, Set> apply(Set s1) {
return (s2) -> apply(s1, s2);
}
/**
* Apply the function to both arguments.
*/
public static Set apply(Set s1, Set s2) {
Set newSet = new HashSet<>(s1);
newSet.addAll(s2);
return newSet;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy