Files
2025RidgeScape/src/main/java/frc4388/robot/commands/GotoPositionCommand.java
T

148 lines
3.9 KiB
Java
Raw Normal View History

2025-01-09 12:57:04 -07:00
package frc4388.robot.commands;
import edu.wpi.first.math.geometry.Translation2d;
import edu.wpi.first.wpilibj.smartdashboard.SmartDashboard;
2025-01-09 12:57:04 -07:00
import edu.wpi.first.wpilibj2.command.Command;
import frc4388.robot.subsystems.SwerveDrive;
import frc4388.robot.subsystems.Vision;
import frc4388.utility.Gains;
2025-01-09 12:57:04 -07:00
import frc4388.utility.UtilityStructs.AutoRecordingControllerFrame;
import frc4388.utility.UtilityStructs.AutoRecordingFrame;
import frc4388.utility.controller.VirtualController;
public class GotoPositionCommand extends Command {
private Translation2d translation2d= new Translation2d(0,0);
static Gains gains = new Gains(3,0,0);
static double tolerance = 0;
private PID xPID = new PID(gains, 0);
private PID yPID = new PID(gains, 0);
2025-01-09 12:57:04 -07:00
SwerveDrive swerveDrive;
Vision vision;
2025-01-09 12:57:04 -07:00
/**
* Command to drive robot to position.
* @param SwerveDrive m_robotSwerveDrive
*/
public GotoPositionCommand(SwerveDrive swerveDrive, Vision vision) {
2025-01-09 12:57:04 -07:00
this.swerveDrive = swerveDrive;
this.vision = vision;
addRequirements(swerveDrive);
2025-01-09 12:57:04 -07:00
}
@Override
public void initialize() {
xPID.initialize();
yPID.initialize();
2025-01-09 12:57:04 -07:00
}
2025-01-09 12:57:04 -07:00
@Override
public void execute() {
double xerr = translation2d.getX() - vision.getPose2d().getX();
double yerr = translation2d.getY() - vision.getPose2d().getY();
2025-01-09 12:57:04 -07:00
SmartDashboard.putNumber("PID X Error", xerr);
SmartDashboard.putNumber("PID Y Error", yerr);
2025-01-09 12:57:04 -07:00
double xoutput = xPID.update(xerr);
double youtput = yPID.update(yerr);
Translation2d leftStick = new Translation2d(
Math.max(Math.min(youtput, 1), -1),
Math.max(Math.min(xoutput, 1), -1)
);
Translation2d rightStick = new Translation2d();
SmartDashboard.putNumber("PID X Output", xoutput);
SmartDashboard.putNumber("PID Y Output", youtput);
swerveDrive.driveWithInput(leftStick, rightStick, true);
2025-01-09 12:57:04 -07:00
}
// @Override
// public void end(boolean interrupted) {
// }
2025-01-09 12:57:04 -07:00
// @Override
// public double getError() {
// return e;
// }
// @Override
// public void runWithOutput(double output) {
// // TODO Auto-generated method stub
// Translation2d leftStick = new Translation2d(Math.max(Math.min(output, 1), -1),0);
// Translation2d rightStick = new Translation2d();
// // System.out.println("Output = " + -output);
// SmartDashboard.putNumber("PID Output", output);
// swerveDrive.driveWithInput(leftStick, rightStick, true);
// }
private class PID {
protected Gains gains;
private double output = 0;
private double tolerance = 0;
/** Creates a new PelvicInflammatoryDisease. */
public PID(double kp, double ki, double kd, double kf, double tolerance) {
gains = new Gains(kp, ki, kd, kf, 0);
this.tolerance = tolerance;
}
public PID(Gains gains, double tolerance) {
this.gains = gains;
this.tolerance = tolerance;
}
// Called when the command is initially scheduled.
public final void initialize() {
output = 0;
}
private double prevError, cumError = 0;
// Called every time the scheduler runs while the command is scheduled.
public double update(double error) {
cumError += error * .02; // 20 ms
double delta = error - prevError;
output = error * gains.kP;
output += cumError * gains.kI;
output += delta * gains.kD;
output += gains.kF;
return output;
}
// // Returns true when the command should end.
// public final boolean isFinished() {
// return Math.abs(getError()) < tolerance;
// }
2025-01-09 12:57:04 -07:00
}
}