allwpilib/wpilibj/src/main/java/org/wpilib/simulation/GenericHIDSim.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.simulation;

import java.util.EnumSet;
import org.wpilib.driverstation.GenericHID;
import org.wpilib.driverstation.POVDirection;

/** Class to control a simulated generic joystick. */
public class GenericHIDSim {
  /** GenericHID port. */
  protected final int m_port;

  /**
   * Constructs from a GenericHID object.
   *
   * @param joystick joystick to simulate
   */
  public GenericHIDSim(GenericHID joystick) {
    m_port = joystick.getPort();
  }

  /**
   * Constructs from a joystick port number.
   *
   * @param port port number
   */
  public GenericHIDSim(int port) {
    m_port = port;
  }

  /** Updates joystick data so that new values are visible to the user program. */
  public void notifyNewData() {
    DriverStationSim.notifyNewData();
  }

  /**
   * Set the value of a given button.
   *
   * @param button the button to set
   * @param value the new value
   */
  public void setRawButton(int button, boolean value) {
    DriverStationSim.setJoystickButton(m_port, button, value);
  }

  /**
   * Set the value of a given axis.
   *
   * @param axis the axis to set
   * @param value the new value
   */
  public void setRawAxis(int axis, double value) {
    DriverStationSim.setJoystickAxis(m_port, axis, value);
  }

  /**
   * Set the value of a given POV.
   *
   * @param pov the POV to set
   * @param value the new value
   */
  public void setPOV(int pov, POVDirection value) {
    DriverStationSim.setJoystickPOV(m_port, pov, value);
  }

  /**
   * Set the value of the default POV (port 0).
   *
   * @param value the new value
   */
  public void setPOV(POVDirection value) {
    setPOV(0, value);
  }

  /**
   * Set the maximum axis index for this device.
   *
   * @param maximumIndex the new maximum axis index
   */
  public void setAxesMaximumIndex(int maximumIndex) {
    DriverStationSim.setJoystickAxesMaximumIndex(m_port, maximumIndex);
  }

  /**
   * Set the axis count of this device.
   *
   * @param count the new axis count
   */
  public void setAxesAvailable(int count) {
    DriverStationSim.setJoystickAxesAvailable(m_port, count);
  }

  /**
   * Set the maximum POV index for this device.
   *
   * @param maximumIndex the new maximum POV index
   */
  public void setPOVsMaximumIndex(int maximumIndex) {
    DriverStationSim.setJoystickPOVsMaximumIndex(m_port, maximumIndex);
  }

  /**
   * Set the POV count of this device.
   *
   * @param count the new POV count
   */
  public void setPOVsAvailable(int count) {
    DriverStationSim.setJoystickPOVsAvailable(m_port, count);
  }

  /**
   * Set the maximum button index for this device.
   *
   * @param maximumIndex the new maximum button index
   */
  public void setButtonsMaximumIndex(int maximumIndex) {
    DriverStationSim.setJoystickButtonsMaximumIndex(m_port, maximumIndex);
  }

  /**
   * Set the button count of this device.
   *
   * @param count the new button count
   */
  public void setButtonsAvailable(long count) {
    DriverStationSim.setJoystickButtonsAvailable(m_port, count);
  }

  /**
   * Set the type of this device.
   *
   * @param type the new device type
   */
  public void setGamepadType(GenericHID.HIDType type) {
    DriverStationSim.setJoystickGamepadType(m_port, type.value);
  }

  /**
   * Set the supported outputs of this device.
   *
   * @param supportedOutputs the new supported outputs
   */
  public void setSupportedOutputs(EnumSet<GenericHID.SupportedOutput> supportedOutputs) {
    int supportedOutputsInt = 0;
    for (GenericHID.SupportedOutput output : supportedOutputs) {
      supportedOutputsInt |= output.getValue();
    }
    DriverStationSim.setJoystickSupportedOutputs(m_port, supportedOutputsInt);
  }

  /**
   * Set the name of this device.
   *
   * @param name the new device name
   */
  public void setName(String name) {
    DriverStationSim.setJoystickName(m_port, name);
  }

  /**
   * Get the led color set.
   *
   * @return the led color set
   */
  public int getLeds() {
    return DriverStationSim.getJoystickLeds(m_port);
  }

  /**
   * Get the joystick rumble.
   *
   * @param type the rumble to read
   * @return the rumble value
   */
  public double getRumble(GenericHID.RumbleType type) {
    int intType =
        switch (type) {
          case LEFT_RUMBLE -> 0;
          case RIGHT_RUMBLE -> 1;
          case LEFT_TRIGGER_RUMBLE -> 2;
          case RIGHT_TRIGGER_RUMBLE -> 3;
        };
    int value = DriverStationSim.getJoystickRumble(m_port, intType);
    return value / 65535.0;
  }
}