MultUnit.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;
import org.wpilib.units.measure.Mult;

public class MultUnit<A extends Unit, B extends Unit> extends Unit {
  private final A m_unitA;
  private final B m_unitB;
 
  @SuppressWarnings("rawtypes")
  private static final CombinatoryUnitCache<Unit, Unit, MultUnit> cache =
      new CombinatoryUnitCache<>(MultUnit::new);

  private MultUnit(A a, B b) {
    super(
        a.isBaseUnit() && b.isBaseUnit() ? null : combine(a.getBaseUnit(), b.getBaseUnit()),
        a.toBaseUnits(1) * b.toBaseUnits(1),
        a.name() + "-" + b.name(),
        a.symbol() + "*" + b.symbol());
    m_unitA = a;
    m_unitB = b;
  }

  protected MultUnit(MultUnit<A, B> baseUnit, A a, B b) {
    super(
        baseUnit,
        a.toBaseUnits(1) * b.toBaseUnits(1),
        a.name() + "-" + b.name(),
        a.symbol() + "*" + b.symbol());
    m_unitA = a;
    m_unitB = b;
  }
 
  MultUnit(
      MultUnit<A, B> baseUnit,
      UnaryFunction toBaseConverter,
      UnaryFunction fromBaseConverter,
      String name,
      String symbol) {
    super(baseUnit, toBaseConverter, fromBaseConverter, name, symbol);
    m_unitA = getBaseUnit().unitA();
    m_unitB = getBaseUnit().unitB();
  }

  @SuppressWarnings("unchecked")
  public static <A extends Unit, B extends Unit> MultUnit<A, B> combine(A a, B b) {
    return cache.combine(a, b);
  }

  @Override
  public Measure<? extends MultUnit<A, B>> of(double magnitude) {
    return ofNative(magnitude);
  }

  @Override
  public Measure<? extends MultUnit<A, B>> ofBaseUnits(double baseUnitMagnitude) {
    return ofNativeBaseUnits(baseUnitMagnitude);
  }

  public final Mult<A, B> ofNative(double magnitude) {
    return new Mult<>(magnitude, toBaseUnits(magnitude), this);
  }

  public final Mult<A, B> ofNativeBaseUnits(double baseUnitMagnitude) {
    return new Mult<>(fromBaseUnits(baseUnitMagnitude), baseUnitMagnitude, this);
  }
 
  @Override
  @SuppressWarnings("unchecked")
  public Measure<? extends MultUnit<A, B>> zero() {
    return (Measure<? extends MultUnit<A, B>>) super.zero();
  }
 
  @Override
  @SuppressWarnings("unchecked")
  public Measure<? extends MultUnit<A, B>> one() {
    return (Measure<? extends MultUnit<A, B>>) super.one();
  }
 
  @Override
  public Unit per(TimeUnit time) {
    return VelocityUnit.combine(this, time);
  }

  public double convertFrom(double magnitude, MultUnit<A, B> otherUnit) {
    return fromBaseUnits(otherUnit.toBaseUnits(magnitude));
  }
 
  @Override
  @SuppressWarnings("unchecked")
  public MultUnit<A, B> getBaseUnit() {
    return (MultUnit<A, B>) super.getBaseUnit();
  }

  public A unitA() {
    return m_unitA;
  }

  public B unitB() {
    return m_unitB;
  }
 
  @Override
  public String toString() {
    return "(" + m_unitA.toString() + " * " + m_unitB.toString() + ")";
  }
 
  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }
    if (!super.equals(o)) {
      return false;
    }
    MultUnit<?, ?> multUnit = (MultUnit<?, ?>) o;
    return Objects.equals(m_unitA, multUnit.m_unitA) && Objects.equals(m_unitB, multUnit.m_unitB);
  }
 
  @Override
  public int hashCode() {
    return Objects.hash(super.hashCode(), m_unitA, m_unitB);
  }
}