target lock thing on smart dash

This commit is contained in:
aarav18
2022-03-17 19:41:20 -06:00
parent b4bf4296de
commit d9b49b9673
2 changed files with 28 additions and 17 deletions
@@ -187,9 +187,9 @@ public class Shoot extends CommandBase {
// Returns true when the command should end. // Returns true when the command should end.
@Override @Override
public boolean isFinished() { public boolean isFinished() {
if (simMode) { // if (simMode) {
return isAimedInTolerance; return isAimedInTolerance;
} // }
return false; // return false;
} }
} }
@@ -9,6 +9,7 @@ import java.util.ArrayList;
import org.opencv.core.Point; import org.opencv.core.Point;
import edu.wpi.first.math.geometry.Pose2d; import edu.wpi.first.math.geometry.Pose2d;
import edu.wpi.first.wpilibj.motorcontrol.MotorController;
import edu.wpi.first.wpilibj.smartdashboard.SmartDashboard; import edu.wpi.first.wpilibj.smartdashboard.SmartDashboard;
import edu.wpi.first.wpilibj2.command.CommandBase; import edu.wpi.first.wpilibj2.command.CommandBase;
@@ -40,6 +41,10 @@ public class TrackTarget extends CommandBase {
Pose2d pos = new Pose2d(); Pose2d pos = new Pose2d();
ArrayList<Point> points = new ArrayList<>(); ArrayList<Point> points = new ArrayList<>();
private boolean targetLocked = false;
private double velocityTolerance = 100.0;
private double hoodTolerance = 5.0;
double m=0; double m=0;
double b=0; double b=0;
boolean isExecuted = false; boolean isExecuted = false;
@@ -67,62 +72,68 @@ public class TrackTarget extends CommandBase {
@Override @Override
public void execute() { public void execute() {
//m_targetAngle = angleToCenter(x, y, m_drive.getRegGyro().getDegrees()); //m_targetAngle = angleToCenter(x, y, m_drive.getRegGyro().getDegrees());
SmartDashboard.putBoolean("Target Locked", this.targetLocked);
try { try {
m_visionOdometry.setLEDs(true); m_visionOdometry.setLEDs(true);
points = m_visionOdometry.getTargetPoints(); points = m_visionOdometry.getTargetPoints();
Point average = VisionOdometry.averagePoint(points); Point average = VisionOdometry.averagePoint(points);
for(Point point : points) { for(Point point : points) {
Vector2D difference = new Vector2D(point); Vector2D difference = new Vector2D(point);
difference.subtract(new Vector2D(average)); difference.subtract(new Vector2D(average));
if(difference.magnitude() < VisionConstants.POINTS_THRESHOLD) if(difference.magnitude() < VisionConstants.POINTS_THRESHOLD)
points.remove(point); points.remove(point);
} }
average = VisionOdometry.averagePoint(points); average = VisionOdometry.averagePoint(points);
DesmosServer.putPoint("average", average); DesmosServer.putPoint("average", average);
for(int i = 0; i < points.size(); i++) { for(int i = 0; i < points.size(); i++) {
DesmosServer.putPoint("Point" + i, points.get(i)); DesmosServer.putPoint("Point" + i, points.get(i));
} }
output = (average.x - VisionConstants.LIME_HIXELS/2.d) / VisionConstants.LIME_HIXELS; output = (average.x - VisionConstants.LIME_HIXELS/2.d) / VisionConstants.LIME_HIXELS;
output *= 4; output *= 4;
// output *= 0.5; // output *= 0.5;
DesmosServer.putDouble("output", output); DesmosServer.putDouble("output", output);
m_turret.runTurretWithInput(output); m_turret.runTurretWithInput(output);
double y_rot = average.y / VisionConstants.LIME_VIXELS; double y_rot = average.y / VisionConstants.LIME_VIXELS;
y_rot *= Math.toRadians(VisionConstants.V_FOV); y_rot *= Math.toRadians(VisionConstants.V_FOV);
y_rot -= Math.toRadians(VisionConstants.V_FOV) / 2; y_rot -= Math.toRadians(VisionConstants.V_FOV) / 2;
y_rot += Math.toRadians(VisionConstants.LIME_ANGLE); y_rot += Math.toRadians(VisionConstants.LIME_ANGLE);
double distance = (VisionConstants.TARGET_HEIGHT - VisionConstants.LIME_HEIGHT) / Math.tan(y_rot); double distance = (VisionConstants.TARGET_HEIGHT - VisionConstants.LIME_HEIGHT) / Math.tan(y_rot);
DesmosServer.putDouble("distance", distance); DesmosServer.putDouble("distance", distance);
updateRegressionDesmos(); updateRegressionDesmos();
double regressedDistance = distanceRegression(distance); double regressedDistance = distanceRegression(distance);
regressedDistance += VisionConstants.EDGE_TO_CENTER + VisionConstants.LIMELIGHT_RADIUS; regressedDistance += VisionConstants.EDGE_TO_CENTER + VisionConstants.LIMELIGHT_RADIUS;
DesmosServer.putDouble("distanceReg", regressedDistance); DesmosServer.putDouble("distanceReg", regressedDistance);
//Vision odemetry circle fit based pose estimate //Vision odometry circle fit based pose estimate
// Point targetOffset = m_visionOdometry.getTargetOffset(); // Point targetOffset = m_visionOdometry.getTargetOffset();
// DesmosServer.putPoint("targetOff", targetOffset); // DesmosServer.putPoint("targetOff", targetOffset);
vel = m_boomBoom.getVelocity(regressedDistance + 30); vel = m_boomBoom.getVelocity(regressedDistance + 30);
hood = m_boomBoom.getHood(regressedDistance + 30); hood = m_boomBoom.getHood(regressedDistance + 30);
// m_boomBoom.runDrumShooter(vel); // m_boomBoom.runDrumShooter(vel);
m_boomBoom.runDrumShooterVelocityPID(vel); m_boomBoom.runDrumShooterVelocityPID(vel);
m_hood.runAngleAdjustPID(hood); m_hood.runAngleAdjustPID(hood);
double currentDrumVel = this.m_boomBoom.m_shooterFalconLeft.getSelectedSensorVelocity();
double currentHood = this.m_hood.getEncoderPosition();
this.targetLocked = (Math.abs(currentDrumVel - vel) < velocityTolerance) && (Math.abs(currentHood - hood) < hoodTolerance);
SmartDashboard.putNumber("Regressed Distance", regressedDistance); SmartDashboard.putNumber("Regressed Distance", regressedDistance);
SmartDashboard.putNumber("Distance", distance); SmartDashboard.putNumber("Distance", distance);
SmartDashboard.putNumber("Hood Target Angle Track", hood); SmartDashboard.putNumber("Hood Target Angle Track", hood);
SmartDashboard.putNumber("Vel Target Track", vel); SmartDashboard.putNumber("Vel Target Track", vel);
// isExecuted = true; // isExecuted = true;
} }
catch (Exception e){ catch (Exception e){