allwpilib/wpilibcExamples/src/main/cpp/examples/ArmSimulation/include/Robot.hpp

// 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.

#pragma once

#include "subsystems/Arm.hpp"
#include "wpi/driverstation/Joystick.hpp"
#include "wpi/framework/TimedRobot.hpp"

/**
 * This is a sample program to demonstrate the use of arm simulation.
 */
class Robot : public wpi::TimedRobot {
 public:
  Robot() {}
  void SimulationPeriodic() override;
  void TeleopInit() override;
  void TeleopPeriodic() override;
  void DisabledInit() override;

 private:
  wpi::Joystick m_joystick{kJoystickPort};
  Arm m_arm;
};