Implement RunFLyWheel class, add motor methods to Shooter class

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

diff --git a/src/org/usfirst/frc/team3501/robot/commands/shooter/RunFlyWheel.java b/src/org/usfirst/frc/team3501/robot/commands/shooter/RunFlyWheel.java
index f57cd5bc607865b133f1e18ec786cc796a5075cc..203f9dce79a4eb3088eb1155e9e67dfa1fbac036 100644 (file)
--- a/src/org/usfirst/frc/team3501/robot/commands/shooter/RunFlyWheel.java
+++ b/src/org/usfirst/frc/team3501/robot/commands/shooter/RunFlyWheel.java
@@ -1,5 +1,8 @@
 package org.usfirst.frc.team3501.robot.commands.shooter;
 
+import org.usfirst.frc.team3501.robot.Robot;
+
+import edu.wpi.first.wpilibj.Timer;
 import edu.wpi.first.wpilibj.command.Command;
 
 /**
@@ -9,6 +12,7 @@ import edu.wpi.first.wpilibj.command.Command;
  * @author Shaina
  */
 public class RunFlyWheel extends Command {
+  Timer timer;
   private double motorVal;
   private double time;
 
@@ -21,6 +25,9 @@ public class RunFlyWheel extends Command {
    *          in seconds, amount of time to run fly wheel motor
    */
   public RunFlyWheel(double motorVal, double time) {
+    requires(Robot.getShooter());
+
+    timer = new Timer();
     this.motorVal = motorVal;
     this.time = time;
   }
@@ -28,6 +35,8 @@ public class RunFlyWheel extends Command {
   // Called just before this Command runs the first time
   @Override
   protected void initialize() {
+    timer.start();
+    Robot.getShooter().setFlyWheelMotorVal(motorVal);
   }
 
   // Called repeatedly when this Command is scheduled to run
@@ -38,17 +47,19 @@ public class RunFlyWheel extends Command {
   // Called once after isFinished returns true
   @Override
   protected void end() {
+    Robot.getShooter().stopFlyWheel();
   }
 
   // Called when another command which requires one or more of the same
   // subsystems is scheduled to run
   @Override
   protected void interrupted() {
+    end();
   }
 
   @Override
   protected boolean isFinished() {
-    return false;
+    return timer.get() >= time;
   }
 
 }