Commit | Line | Data |
---|---|---|
aa160dfd ME |
1 | package org.usfirst.frc.team3501.robot.commands.driving; |
2 | ||
f0a71840 | 3 | import org.usfirst.frc.team3501.robot.Constants; |
8af76159 | 4 | import org.usfirst.frc.team3501.robot.Constants.Direction; |
aa160dfd | 5 | import org.usfirst.frc.team3501.robot.Robot; |
f0a71840 CZ |
6 | import org.usfirst.frc.team3501.robot.subsystems.DriveTrain; |
7 | import org.usfirst.frc.team3501.robot.utils.PIDController; | |
aa160dfd ME |
8 | |
9 | import edu.wpi.first.wpilibj.command.Command; | |
10 | ||
11 | /** | |
9b9d6bed ME |
12 | * This command turns the robot for a specified angle in specified direction - |
13 | * either left or right | |
14 | * | |
f0a71840 CZ |
15 | * parameters: angle: the robot will turn - in degrees direction: the robot will |
16 | * turn - either right or left maxTimeOut: the max time this command will be | |
17 | * allowed to run on for | |
aa160dfd ME |
18 | */ |
19 | public class TurnForAngle extends Command { | |
f0a71840 | 20 | private DriveTrain driveTrain = Robot.getDriveTrain(); |
aa160dfd | 21 | |
f0a71840 CZ |
22 | Direction direction; |
23 | private double maxTimeOut; | |
24 | private PIDController gyroController; | |
25 | ||
26 | private double target; | |
27 | private double gyroP; | |
28 | private double gyroI; | |
29 | private double gyroD; | |
30 | ||
571fb5c9 CZ |
31 | private double zeroAngle; |
32 | ||
f0a71840 | 33 | public TurnForAngle(double angle, Direction direction, double maxTimeOut) { |
d17d868d | 34 | requires(Robot.getDriveTrain()); |
f0a71840 CZ |
35 | this.direction = direction; |
36 | this.maxTimeOut = maxTimeOut; | |
571fb5c9 | 37 | this.target = Math.abs(angle); |
f0a71840 | 38 | |
c846487e CZ |
39 | this.gyroP = driveTrain.turnP; |
40 | this.gyroI = driveTrain.turnI; | |
41 | this.gyroD = driveTrain.turnD; | |
f0a71840 | 42 | |
01ea2879 | 43 | this.gyroController = Robot.getDriveTrain().getDriveController(); |
f0a71840 CZ |
44 | this.gyroController.setDoneRange(1); |
45 | this.gyroController.setMinDoneCycles(5); | |
d17d868d | 46 | } |
47 | ||
d17d868d | 48 | @Override |
49 | protected void initialize() { | |
f0a71840 | 50 | this.driveTrain.resetEncoders(); |
f0a71840 | 51 | this.gyroController.setSetPoint(this.target); |
c726d04a | 52 | this.zeroAngle = driveTrain.getAngle(); |
d17d868d | 53 | } |
54 | ||
d17d868d | 55 | @Override |
56 | protected void execute() { | |
f0a71840 CZ |
57 | double xVal = 0; |
58 | ||
59 | xVal = this.gyroController | |
571fb5c9 | 60 | .calcPID(Math.abs(this.driveTrain.getAngle() - this.zeroAngle)); |
f0a71840 | 61 | |
571fb5c9 CZ |
62 | double leftDrive = 0; |
63 | double rightDrive = 0; | |
f0a71840 CZ |
64 | if (direction == Constants.Direction.RIGHT) { |
65 | leftDrive = xVal; | |
66 | rightDrive = -xVal; | |
571fb5c9 | 67 | } else if (direction == Constants.Direction.LEFT) { |
f0a71840 CZ |
68 | leftDrive = -xVal; |
69 | rightDrive = xVal; | |
70 | } | |
71 | ||
72 | this.driveTrain.setMotorValues(leftDrive, rightDrive); | |
d17d868d | 73 | } |
74 | ||
d17d868d | 75 | @Override |
76 | protected boolean isFinished() { | |
f0a71840 | 77 | return timeSinceInitialized() >= maxTimeOut || gyroController.isDone(); |
d17d868d | 78 | } |
79 | ||
d17d868d | 80 | @Override |
81 | protected void end() { | |
82 | } | |
83 | ||
d17d868d | 84 | @Override |
85 | protected void interrupted() { | |
f0a71840 | 86 | end(); |
d17d868d | 87 | } |
aa160dfd | 88 | } |