1 package org
.usfirst
.frc
.team3501
.robot
.subsystems
;
3 import org
.usfirst
.frc
.team3501
.robot
.Constants
;
4 import org
.usfirst
.frc
.team3501
.robot
.MathLib
;
5 import org
.usfirst
.frc
.team3501
.robot
.commands
.driving
.JoystickDrive
;
6 import org
.usfirst
.frc
.team3501
.robot
.sensors
.GyroLib
;
7 import org
.usfirst
.frc
.team3501
.robot
.sensors
.Lidar
;
9 import edu
.wpi
.first
.wpilibj
.CANTalon
;
10 import edu
.wpi
.first
.wpilibj
.CounterBase
.EncodingType
;
11 import edu
.wpi
.first
.wpilibj
.DoubleSolenoid
;
12 import edu
.wpi
.first
.wpilibj
.DoubleSolenoid
.Value
;
13 import edu
.wpi
.first
.wpilibj
.Encoder
;
14 import edu
.wpi
.first
.wpilibj
.I2C
;
15 import edu
.wpi
.first
.wpilibj
.RobotDrive
;
16 import edu
.wpi
.first
.wpilibj
.command
.PIDSubsystem
;
18 public class DriveTrain
extends PIDSubsystem
{
19 // Current Drive Mode Default Drive Mode is Manual
20 private int DRIVE_MODE
= 1;
21 private static double pidOutput
= 0;
23 private Encoder leftEncoder
, rightEncoder
;
25 public static Lidar lidar
;
27 private CANTalon frontLeft
, frontRight
, rearLeft
, rearRight
;
28 private RobotDrive robotDrive
;
31 public DoubleSolenoid leftGearPiston
, rightGearPiston
;
33 // Drivetrain specific constants that relate to the inches per pulse value for
37 super(Constants
.DriveTrain
.kp
, Constants
.DriveTrain
.ki
,
38 Constants
.DriveTrain
.kd
);
40 frontLeft
= new CANTalon(Constants
.DriveTrain
.FRONT_LEFT
);
41 frontRight
= new CANTalon(Constants
.DriveTrain
.FRONT_RIGHT
);
42 rearLeft
= new CANTalon(Constants
.DriveTrain
.REAR_LEFT
);
43 rearRight
= new CANTalon(Constants
.DriveTrain
.REAR_RIGHT
);
45 robotDrive
= new RobotDrive(frontLeft
, rearLeft
, frontRight
, rearRight
);
47 lidar
= new Lidar(I2C
.Port
.kMXP
);
48 leftEncoder
= new Encoder(Constants
.DriveTrain
.ENCODER_LEFT_A
,
49 Constants
.DriveTrain
.ENCODER_LEFT_B
, false, EncodingType
.k4X
);
50 rightEncoder
= new Encoder(Constants
.DriveTrain
.ENCODER_RIGHT_A
,
51 Constants
.DriveTrain
.ENCODER_RIGHT_B
, false, EncodingType
.k4X
);
52 leftEncoder
.setDistancePerPulse(Constants
.DriveTrain
.INCHES_PER_PULSE
);
53 rightEncoder
.setDistancePerPulse(Constants
.DriveTrain
.INCHES_PER_PULSE
);
55 leftEncoder
.setDistancePerPulse(Constants
.DriveTrain
.INCHES_PER_PULSE
);
56 rightEncoder
.setDistancePerPulse(Constants
.DriveTrain
.INCHES_PER_PULSE
);
58 gyro
= new GyroLib(I2C
.Port
.kOnboard
, false);
60 DRIVE_MODE
= Constants
.DriveTrain
.ENCODER_MODE
;
65 leftGearPiston
= new DoubleSolenoid(10, Constants
.DriveTrain
.LEFT_FORWARD
,
66 Constants
.DriveTrain
.LEFT_REVERSE
);
67 rightGearPiston
= new DoubleSolenoid(10,
68 Constants
.DriveTrain
.RIGHT_FORWARD
,
69 Constants
.DriveTrain
.RIGHT_REVERSE
);
70 Constants
.DriveTrain
.inverted
= false;
74 protected void initDefaultCommand() {
75 setDefaultCommand(new JoystickDrive());
78 // Print tne PID Output
79 public void printOutput() {
80 System
.out
.println("PIDOutput: " + pidOutput
);
83 private double getAvgEncoderDistance() {
84 return (leftEncoder
.getDistance() + rightEncoder
.getDistance()) / 2;
87 // Whether or not the PID Controller thinks we have reached the target
89 public boolean reachedTarget() {
90 if (this.onTarget()) {
102 public void resetEncoders() {
104 rightEncoder
.reset();
107 public double getLidarDistance() {
108 return lidar
.pidGet();
111 public double getRightSpeed() {
112 return rightEncoder
.getRate(); // in inches per second
115 public double getLeftSpeed() {
116 return leftEncoder
.getRate(); // in inches per second
119 public double getSpeed() {
120 return (getLeftSpeed() + getRightSpeed()) / 2.0; // in inches per second
123 public double getRightDistance() {
124 return rightEncoder
.getDistance(); // in inches
127 public double getLeftDistance() {
128 return leftEncoder
.getDistance(); // in inches
131 // Get error between the setpoint of PID Controller and the current state of
133 public double getError() {
134 if (DRIVE_MODE
== Constants
.DriveTrain
.ENCODER_MODE
)
135 return Math
.abs(this.getSetpoint() - getAvgEncoderDistance());
137 return Math
.abs(this.getSetpoint() + getGyroAngle());
140 public double getGyroAngle() {
141 return gyro
.getRotationZ().getAngle();
144 public void resetGyro() {
148 public void printEncoder(int i
, int n
) {
150 System
.out
.println("Left: " + this.getLeftDistance());
151 System
.out
.println("Right: " + this.getRightDistance());
156 public void printGyroOutput() {
157 System
.out
.println("Gyro Angle" + -this.getGyroAngle());
161 * returns the PID output that is returned by the PID Controller
163 public double getOutput() {
167 // Updates the PID constants based on which control mode is being used
168 public void updatePID() {
169 if (DRIVE_MODE
== Constants
.DriveTrain
.ENCODER_MODE
)
170 this.getPIDController().setPID(Constants
.DriveTrain
.kp
,
171 Constants
.DriveTrain
.ki
, Constants
.DriveTrain
.kd
);
173 this.getPIDController().setPID(Constants
.DriveTrain
.gp
,
174 Constants
.DriveTrain
.gd
, Constants
.DriveTrain
.gi
);
177 public CANTalon
getFrontLeft() {
181 public CANTalon
getFrontRight() {
185 public CANTalon
getRearLeft() {
189 public CANTalon
getRearRight() {
193 public int getMode() {
198 * Method is a required method that the PID Subsystem uses to return the
199 * calculated PID value to the driver
201 * @param Gives the user the output from the PID algorithm that is calculated
204 * Body: Uses the output, does some filtering and drives the robot
207 protected void usePIDOutput(double output
) {
210 if (DRIVE_MODE
== Constants
.DriveTrain
.ENCODER_MODE
) {
211 double drift
= this.getLeftDistance() - this.getRightDistance();
212 if (Math
.abs(output
) > 0 && Math
.abs(output
) < 0.3)
213 output
= Math
.signum(output
) * 0.3;
215 right
= output
+ drift
* Constants
.DriveTrain
.kp
/ 10;
217 } else if (DRIVE_MODE
== Constants
.DriveTrain
.GYRO_MODE
) {
220 arcadeDrive(0, output
);
226 protected double returnPIDInput() {
227 return sensorFeedback();
231 * Checks the drive mode
233 * @return the current state of the robot in each state Average distance from
234 * both sides of tank drive for Encoder Mode Angle from the gyro in GYRO_MODE
236 private double sensorFeedback() {
237 if (DRIVE_MODE
== Constants
.DriveTrain
.ENCODER_MODE
)
238 return getAvgEncoderDistance();
239 else if (DRIVE_MODE
== Constants
.DriveTrain
.GYRO_MODE
)
240 return -this.getGyroAngle();
241 // counterclockwise is positive on joystick but we want it to be negative
247 * @param left and right setpoints to set to the left and right side of tank
248 * inverted is for Logan, wants the robot to invert all controls left = right
249 * and right = left negative input is required for the regular rotation
250 * because RobotDrive tankdrive method drives inverted
252 public void drive(double left
, double right
) {
253 robotDrive
.tankDrive(-left
, -right
);
254 // dunno why but inverted drive (- values is forward)
255 if (!Constants
.DriveTrain
.inverted
)
256 robotDrive
.tankDrive(-left
, -right
);
258 robotDrive
.tankDrive(right
, left
);
261 public void arcadeDrive(double y
, double twist
) {
262 robotDrive
.arcadeDrive(y
, twist
);
266 * constrains the distance to within -100 and 100 since we aren't going to
267 * drive more than 100 inches
269 * Configure Encoder PID
271 * Sets the setpoint to the PID subsystem
273 public void driveDistance(double dist
, double maxTimeOut
) {
274 dist
= MathLib
.constrain(dist
, -100, 100);
280 * Sets the encoder mode Updates the PID constants sets the tolerance and sets
281 * output/input ranges Enables the PID controllers
283 public void setEncoderPID() {
284 DRIVE_MODE
= Constants
.DriveTrain
.ENCODER_MODE
;
286 this.setAbsoluteTolerance(Constants
.DriveTrain
.encoderTolerance
);
287 this.setOutputRange(-1.0, 1.0);
288 this.setInputRange(-200.0, 200.0);
293 * Sets the Gyro Mode Updates the PID constants, sets the tolerance and sets
294 * output/input ranges Enables the PID controllers
296 private void setGyroPID() {
297 DRIVE_MODE
= Constants
.DriveTrain
.GYRO_MODE
;
299 this.getPIDController().setPID(Constants
.DriveTrain
.gp
,
300 Constants
.DriveTrain
.gi
, Constants
.DriveTrain
.gd
);
302 this.setAbsoluteTolerance(Constants
.DriveTrain
.gyroTolerance
);
303 this.setOutputRange(-1.0, 1.0);
304 this.setInputRange(-360.0, 360.0);
309 * Turning method that should be used repeatedly in a command
311 * First constrains the angle to within -360 and 360 since that is as much as
314 * Configures Gyro PID and sets the setpoint as an angle
316 public void turnAngle(double angle
) {
317 angle
= MathLib
.constrain(angle
, -360, 360);
322 public void setMotorSpeeds(double left
, double right
) {
323 // positive setpoint to left side makes it go backwards
324 // positive setpoint to right side makes it go forwards.
325 frontLeft
.set(-left
);
327 frontRight
.set(right
);
328 rearRight
.set(right
);
332 * @return a value that is the current setpoint for the piston kReverse or
335 public Value
getLeftGearPistonValue() {
336 return leftGearPiston
.get();
340 * @return a value that is the current setpoint for the piston kReverse or
343 public Value
getRightGearPistonValue() {
344 return rightGearPiston
.get();
348 * Changes the ball shift gear assembly to high
350 public void setHighGear() {
351 changeGear(Constants
.DriveTrain
.HIGH_GEAR
);
355 * Changes the ball shift gear assembly to low
357 public void setLowGear() {
358 changeGear(Constants
.DriveTrain
.LOW_GEAR
);
362 * changes the gear to a DoubleSolenoid.Value
364 public void changeGear(DoubleSolenoid
.Value gear
) {
365 leftGearPiston
.set(gear
);
366 rightGearPiston
.set(gear
);