UnaryFunction.java

// Copyright (c) FIRST and other WPILib contributors.
// Open Source Software; you can modify and/or share it under the terms of
// the WPILib BSD license file in the root directory of this project.
 
package org.wpilib.units;
 
import java.util.Objects;

@FunctionalInterface
public interface UnaryFunction {
  UnaryFunction IDENTITY = x -> x;

  double apply(double input);

  default UnaryFunction pipeTo(UnaryFunction next) {
    Objects.requireNonNull(next, "The next operation in the chain must be provided");
 
    return x -> next.apply(this.apply(x));
  }

  default UnaryFunction mult(UnaryFunction multiplier) {
    Objects.requireNonNull(multiplier, "A multiplier function must be provided");
 
    return x -> this.apply(x) * multiplier.apply(x);
  }

  default UnaryFunction mult(double multiplier) {
    return x -> this.apply(x) * multiplier;
  }

  default UnaryFunction div(UnaryFunction divisor) {
    Objects.requireNonNull(divisor, "A divisor function must be provided");
 
    return x -> {
      double numerator = this.apply(x);
 
      // fast-track to avoid another function call
      // avoids returning NaN if divisor is also zero
      if (numerator == 0) {
        return 0;
      }
 
      double div = divisor.apply(x);
      return numerator / div; // NOTE: returns +Infinity or -Infinity if div is zero
    };
  }

  default UnaryFunction div(double divisor) {
    return x -> this.apply(x) / divisor;
  }

  default UnaryFunction exp(UnaryFunction exponent) {
    Objects.requireNonNull(exponent, "An exponent function must be provided");
 
    return x -> Math.pow(this.apply(x), exponent.apply(x));
  }

  default UnaryFunction exp(double exponent) {
    return x -> Math.pow(this.apply(x), exponent);
  }
}