Fix merge conflicts

[3501/2017steamworks] / src / org / usfirst / frc / team3501 / robot / subsystems / DriveTrain.java

diff --git a/src/org/usfirst/frc/team3501/robot/subsystems/DriveTrain.java b/src/org/usfirst/frc/team3501/robot/subsystems/DriveTrain.java
index 5422e973078ba2d99b0d9556732de89a0d8d4b31..10bc4450fa9c2f41421541d9f43e808ac1d27922 100644 (file)
--- a/src/org/usfirst/frc/team3501/robot/subsystems/DriveTrain.java
+++ b/src/org/usfirst/frc/team3501/robot/subsystems/DriveTrain.java
@@ -29,6 +29,9 @@ public class DriveTrain extends Subsystem {
   public static final double TIME_TO_CLIMB_FOR = 0;
   public static final double CLIMBER_SPEED = 0;
 
+  public static final boolean DRIVE_BRAKE_MODE = true;
+  public static final boolean DRIVE_COAST_MODE = false;
+
   private static DriveTrain driveTrain;
 
   private final CANTalon frontLeft, frontRight, rearLeft, rearRight;
@@ -167,31 +170,45 @@ public class DriveTrain extends Subsystem {
     return rightGearPiston.get();
   }
 
+  public DoubleSolenoid getLeftPiston() {
+    return this.leftGearPiston;
+  }
+
+  public DoubleSolenoid getRightPiston() {
+    return this.rightGearPiston;
+  }
+
   /*
    * Changes the ball shift gear assembly to high
    */
-  public void setHighGear() {
-    changeGear(Constants.DriveTrain.HIGH_GEAR);
+  public void setHighGear(DoubleSolenoid p) {
+    changeGear(Constants.DriveTrain.HIGH_GEAR, p);
   }
 
   /*
    * Changes the ball shift gear assembly to low
    */
-  public void setLowGear() {
-    changeGear(Constants.DriveTrain.LOW_GEAR);
+  public void setLowGear(DoubleSolenoid p) {
+    changeGear(Constants.DriveTrain.LOW_GEAR, p);
   }
 
   /*
    * Changes the gear to a DoubleSolenoid.Value
    */
-  private void changeGear(DoubleSolenoid.Value gear) {
-    System.out.println(gear);
-    leftGearPiston.set(gear);
-    rightGearPiston.set(gear);
+  private void changeGear(DoubleSolenoid.Value gear, DoubleSolenoid piston) {
+    piston.set(gear);
   }
 
   @Override
   protected void initDefaultCommand() {
     setDefaultCommand(new JoystickDrive());
   }
+
+  public void setCANTalonsBrakeMode(boolean mode) {
+    frontLeft.enableBrakeMode(mode);
+    rearLeft.enableBrakeMode(mode);
+
+    frontRight.enableBrakeMode(mode);
+    rearRight.enableBrakeMode(mode);
+  }
 }