Add the piston solenoid object to shooter and add punch methods/constants

[3501/stronghold-2016] / src / org / usfirst / frc / team3501 / robot / subsystems / Shooter.java

diff --git a/src/org/usfirst/frc/team3501/robot/subsystems/Shooter.java b/src/org/usfirst/frc/team3501/robot/subsystems/Shooter.java
index 675b4b8cbf52353aa2a70418ee73c5b3ae3f150e..bda816f402fa2bae9c0ec35d09601275a9cbc7a9 100755 (executable)
--- a/src/org/usfirst/frc/team3501/robot/subsystems/Shooter.java
+++ b/src/org/usfirst/frc/team3501/robot/subsystems/Shooter.java
@@ -1,17 +1,19 @@
 package org.usfirst.frc.team3501.robot.subsystems;
 
 import org.usfirst.frc.team3501.robot.Constants;
-import org.usfirst.frc.team3501.robot.Constants.Shooter.State;
 
 import edu.wpi.first.wpilibj.CANTalon;
+import edu.wpi.first.wpilibj.DoubleSolenoid;
 import edu.wpi.first.wpilibj.command.Subsystem;
 
 public class Shooter extends Subsystem {
-  // TODO: check all files for control m characters
   private CANTalon shooter;
+  private DoubleSolenoid punch;
 
   public Shooter() {
     shooter = new CANTalon(Constants.Shooter.PORT);
+    punch = new DoubleSolenoid(Constants.Shooter.FORWARD_PORT,
+        Constants.Shooter.REVERSE_PORT);
   }
 
   public double getCurrentSetPoint() {
@@ -19,9 +21,9 @@ public class Shooter extends Subsystem {
   }
 
   public void setSpeed(double speed) {
-    if (speed >= 1.0)
+    if (speed > 1.0)
       shooter.set(1.0);
-    else if (speed <= -1.0)
+    else if (speed < -1.0)
       shooter.set(-1.0);
     else
       shooter.set(speed);
@@ -31,22 +33,27 @@ public class Shooter extends Subsystem {
     this.setSpeed(0.0);
   }
 
-  public State getState() {
-    return (this.getCurrentSetPoint() == 0) ? State.RUNNING : State.STOPPED;
-  }
-
   // Use negative # for decrement. Positive for increment.
   public void changeSpeed(double change) {
-    if (getCurrentSetPoint() + change >= 1.0)
+    double newSpeed = getCurrentSetPoint() + change;
+    if (newSpeed > 1.0)
       shooter.set(1.0);
-    else if (getCurrentSetPoint() + change <= -1.0)
+    else if (newSpeed < -1.0)
       shooter.set(-1.0);
     else {
-      double newSpeed = getCurrentSetPoint() + change;
       setSpeed(newSpeed);
     }
   }
 
+  // Punch Commands
+  public void punch() {
+    punch.set(Constants.Shooter.punch);
+  }
+
+  public void resetPunch() {
+    punch.set(Constants.Shooter.retract);
+  }
+
   @Override
   protected void initDefaultCommand() {
   }