add code to stop motors in end()

[3501/stronghold-2016] / src / org / usfirst / frc / team3501 / robot / commands / driving / TurnForTime.java

diff --git a/src/org/usfirst/frc/team3501/robot/commands/driving/TurnForTime.java b/src/org/usfirst/frc/team3501/robot/commands/driving/TurnForTime.java
index a15008b56d2869cfe2305b5ec113ebde42f913e8..b1b37cdda46a014977db6638828348609acbb634 100755 (executable)
--- a/src/org/usfirst/frc/team3501/robot/commands/driving/TurnForTime.java
+++ b/src/org/usfirst/frc/team3501/robot/commands/driving/TurnForTime.java
@@ -1,39 +1,62 @@
 package org.usfirst.frc.team3501.robot.commands.driving;
 
 import org.usfirst.frc.team3501.robot.Constants.Direction;
+import org.usfirst.frc.team3501.robot.Robot;
 
 import edu.wpi.first.wpilibj.Timer;
 import edu.wpi.first.wpilibj.command.Command;
 
+/***
+ * This command turns the robot in a specified direction for a specified
+ * duration in seconds.
+ *
+ * pre-condition: robot is on a flat surface
+ *
+ * post-condition: robot has turned in the specified direction for the specified
+ * time
+ *
+ * @author Meryem, Avi, and Sarvesh
+ *
+ */
+
 public class TurnForTime extends Command {
+  private final double SPEED = 0.5;
+  private Direction direction;
+  private double seconds;
+  private Timer timer;
 
   public TurnForTime(double seconds, Direction direction) {
-    Timer timer = new Timer();
+    this.seconds = seconds;
+    this.direction = direction;
   }
 
   @Override
   protected void initialize() {
+    timer = new Timer();
+    timer.start();
+
+    if (direction == Direction.RIGHT) {
+      Robot.driveTrain.setMotorSpeeds(SPEED, -SPEED);
+    } else if (direction == Direction.LEFT) {
+      Robot.driveTrain.setMotorSpeeds(-SPEED, SPEED);
+    }
   }
 
   @Override
   protected void execute() {
-    /*
-     * if direction is right make the left motor run forward make the right
-     * motor run backward if direction is left make right motor run forward make
-     * the left motor run backward
-     */
+
   }
 
   @Override
   protected boolean isFinished() {
-    /*
-     * when time is up return true
-     */
+    if (timer.get() >= seconds)
+      return true;
     return false;
   }
 
   @Override
   protected void end() {
+    Robot.driveTrain.setMotorSpeeds(0, 0);
   }
 
   @Override