hydra.lib.literals.BigfloatToBigint Maven / Gradle / Ivy

Go to download
package hydra.lib.literals;

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.dsl.Types;
import hydra.graph.Graph;
import hydra.tools.PrimitiveFunction;

import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.List;
import java.util.function.Function;

import static hydra.dsl.Types.function;

public class BigfloatToBigint extends PrimitiveFunction {
    public Name name() {
        return new Name("hydra/lib/literals.bigfloatToBigint");
    }

    @Override
    public Type type() {
        return function(Types.bigfloat(), Types.bigint());
    }

    @Override
    protected Function>, Flow, Term>> implementation() {
        return args -> Flows.map(Expect.bigfloat(args.get(0)), s -> Terms.bigint(apply(s)));
    }

    public static BigInteger apply(Double value) {
        return BigDecimal.valueOf(value).toBigInteger();
    }
}