VecBuilder.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.math.linalg;
 
import java.util.Objects;
import org.ejml.simple.SimpleMatrix;
import org.wpilib.math.numbers.N1;
import org.wpilib.math.numbers.N10;
import org.wpilib.math.numbers.N2;
import org.wpilib.math.numbers.N3;
import org.wpilib.math.numbers.N4;
import org.wpilib.math.numbers.N5;
import org.wpilib.math.numbers.N6;
import org.wpilib.math.numbers.N7;
import org.wpilib.math.numbers.N8;
import org.wpilib.math.numbers.N9;
import org.wpilib.math.util.Nat;
import org.wpilib.math.util.Num;

public final class VecBuilder {
  private VecBuilder() {
    throw new UnsupportedOperationException("this is a utility class!");
  }
 
  private static <N extends Num> Vector<N> fillVec(Nat<N> rows, double... data) {
    if (Objects.requireNonNull(data).length != rows.getNum()) {
      throw new IllegalArgumentException(
          "Invalid vector data provided. Wanted "
              + rows.getNum()
              + " vector, but got "
              + data.length
              + " elements");
    }
    return new Vector<>(new SimpleMatrix(data));
  }

  public static Vector<N1> fill(double n1) {
    return fillVec(Nat.N1(), n1);
  }

  public static Vector<N2> fill(double n1, double n2) {
    return fillVec(Nat.N2(), n1, n2);
  }

  public static Vector<N3> fill(double n1, double n2, double n3) {
    return fillVec(Nat.N3(), n1, n2, n3);
  }

  public static Vector<N4> fill(double n1, double n2, double n3, double n4) {
    return fillVec(Nat.N4(), n1, n2, n3, n4);
  }

  public static Vector<N5> fill(double n1, double n2, double n3, double n4, double n5) {
    return fillVec(Nat.N5(), n1, n2, n3, n4, n5);
  }

  public static Vector<N6> fill(double n1, double n2, double n3, double n4, double n5, double n6) {
    return fillVec(Nat.N6(), n1, n2, n3, n4, n5, n6);
  }

  public static Vector<N7> fill(
      double n1, double n2, double n3, double n4, double n5, double n6, double n7) {
    return fillVec(Nat.N7(), n1, n2, n3, n4, n5, n6, n7);
  }

  public static Vector<N8> fill(
      double n1, double n2, double n3, double n4, double n5, double n6, double n7, double n8) {
    return fillVec(Nat.N8(), n1, n2, n3, n4, n5, n6, n7, n8);
  }

  public static Vector<N9> fill(
      double n1,
      double n2,
      double n3,
      double n4,
      double n5,
      double n6,
      double n7,
      double n8,
      double n9) {
    return fillVec(Nat.N9(), n1, n2, n3, n4, n5, n6, n7, n8, n9);
  }

  public static Vector<N10> fill(
      double n1,
      double n2,
      double n3,
      double n4,
      double n5,
      double n6,
      double n7,
      double n8,
      double n9,
      double n10) {
    return fillVec(Nat.N10(), n1, n2, n3, n4, n5, n6, n7, n8, n9, n10);
  }
}