modify RunFlyWheel and RunFlyWheelContinuous commands

[3501/2017steamworks] / src / org / usfirst / frc / team3501 / robot / commands / shooter / RunFlyWheel.java

package org.usfirst.frc.team3501.robot.commands.shooter;

import org.usfirst.frc.team3501.robot.Robot;
import org.usfirst.frc.team3501.robot.subsystems.Shooter;
import org.usfirst.frc.team3501.robot.utils.PIDController;

import edu.wpi.first.wpilibj.command.Command;

/**
 * This command runs the fly wheel at a given speed for a given time. The fly
 * wheel is intended to shoot balls fed by the intake wheel.
 *
 * @author Shaina & Chris
 */
public class RunFlyWheel extends Command {
  private Shooter shooter = Robot.getShooter();
  private double maxTimeOut;

  private PIDController wheelController;
  private double wheelP;
  private double wheelI;
  private double wheelD;
  private double target;

  public RunFlyWheel(double maxTimeOut) {
    requires(shooter);

    this.wheelP = this.shooter.wheelP;
    this.wheelI = this.shooter.wheelI;
    this.wheelD = this.shooter.wheelD;
    this.wheelController = new PIDController(this.wheelP, this.wheelI,
        this.wheelD);
    this.wheelController.setDoneRange(0.5);
    this.wheelController.setMaxOutput(1.0);
    this.wheelController.setMinDoneCycles(3);
    this.target = this.shooter.CURRENT_SHOOTING_SPEED;
  }

  // Called just before this Command runs the first time
  protected void initialize() {
    this.wheelController.setSetPoint(this.target);
  }

  // Called repeatedly when this Command is scheduled to run
  protected void execute() {
    double shooterSpeed = this.wheelController
        .calcPID(this.shooter.getShooterSpeed());

    this.shooter.setFlyWheelMotorVal(shooterSpeed);
  }

  // Make this return true when this Command no longer needs to run execute()
  protected boolean isFinished() {
    return timeSinceInitialized() >= maxTimeOut;
  }

  // Called once after isFinished returns true
  protected void end() {
    this.shooter.stopFlyWheel();
  }

  // Called when another command which requires one or more of the same
  // subsystems is scheduled to run
  protected void interrupted() {
    end();
  }
}