mirror of
https://github.com/BroncBotz3481/YAGSL
synced 2026-06-19 06:21:40 +00:00
479 lines
15 KiB
Java
479 lines
15 KiB
Java
package swervelib;
|
|
|
|
import edu.wpi.first.math.controller.SimpleMotorFeedforward;
|
|
import edu.wpi.first.math.geometry.Rotation2d;
|
|
import edu.wpi.first.math.kinematics.SwerveModulePosition;
|
|
import edu.wpi.first.math.kinematics.SwerveModuleState;
|
|
import edu.wpi.first.math.util.Units;
|
|
import edu.wpi.first.wpilibj.smartdashboard.SmartDashboard;
|
|
import swervelib.encoders.SwerveAbsoluteEncoder;
|
|
import swervelib.math.SwerveMath;
|
|
import swervelib.motors.SwerveMotor;
|
|
import swervelib.parser.SwerveModuleConfiguration;
|
|
import swervelib.simulation.SwerveModuleSimulation;
|
|
import swervelib.telemetry.Alert;
|
|
import swervelib.telemetry.SwerveDriveTelemetry;
|
|
import swervelib.telemetry.SwerveDriveTelemetry.TelemetryVerbosity;
|
|
|
|
/**
|
|
* The Swerve Module class which represents and controls Swerve Modules for the swerve drive.
|
|
*/
|
|
public class SwerveModule
|
|
{
|
|
|
|
/**
|
|
* Swerve module configuration options.
|
|
*/
|
|
public final SwerveModuleConfiguration configuration;
|
|
/**
|
|
* Swerve Motors.
|
|
*/
|
|
private final SwerveMotor angleMotor, driveMotor;
|
|
/**
|
|
* Absolute encoder for swerve drive.
|
|
*/
|
|
private final SwerveAbsoluteEncoder absoluteEncoder;
|
|
/**
|
|
* An {@link Alert} for if pushing the Absolute Encoder offset to the encoder fails.
|
|
*/
|
|
private final Alert encoderOffsetWarning;
|
|
/**
|
|
* An {@link Alert} for if there is no Absolute Encoder on the module.
|
|
*/
|
|
private final Alert noEncoderWarning;
|
|
/**
|
|
* Module number for kinematics, usually 0 to 3. front left -> front right -> back left -> back right.
|
|
*/
|
|
public int moduleNumber;
|
|
/**
|
|
* Feedforward for drive motor during closed loop control.
|
|
*/
|
|
public SimpleMotorFeedforward feedforward;
|
|
/**
|
|
* Maximum speed of the drive motors in meters per second.
|
|
*/
|
|
public double maxSpeed;
|
|
/**
|
|
* Last swerve module state applied.
|
|
*/
|
|
public SwerveModuleState lastState;
|
|
/**
|
|
* Angle offset from the absolute encoder.
|
|
*/
|
|
private double angleOffset;
|
|
/**
|
|
* Simulated swerve module.
|
|
*/
|
|
private SwerveModuleSimulation simModule;
|
|
/**
|
|
* Encoder synchronization queued.
|
|
*/
|
|
private boolean synchronizeEncoderQueued = false;
|
|
|
|
/**
|
|
* Construct the swerve module and initialize the swerve module motors and absolute encoder.
|
|
*
|
|
* @param moduleNumber Module number for kinematics.
|
|
* @param moduleConfiguration Module constants containing CAN ID's and offsets.
|
|
* @param driveFeedforward Drive motor feedforward created by
|
|
* {@link SwerveMath#createDriveFeedforward(double, double, double)}.
|
|
*/
|
|
public SwerveModule(int moduleNumber, SwerveModuleConfiguration moduleConfiguration,
|
|
SimpleMotorFeedforward driveFeedforward)
|
|
{
|
|
// angle = 0;
|
|
// speed = 0;
|
|
// omega = 0;
|
|
// fakePos = 0;
|
|
this.moduleNumber = moduleNumber;
|
|
configuration = moduleConfiguration;
|
|
angleOffset = moduleConfiguration.angleOffset;
|
|
|
|
// Initialize Feedforward for drive motor.
|
|
feedforward = driveFeedforward;
|
|
|
|
// Create motors from configuration and reset them to defaults.
|
|
angleMotor = moduleConfiguration.angleMotor;
|
|
driveMotor = moduleConfiguration.driveMotor;
|
|
angleMotor.factoryDefaults();
|
|
driveMotor.factoryDefaults();
|
|
|
|
// Configure voltage comp, current limit, and ramp rate.
|
|
angleMotor.setVoltageCompensation(configuration.physicalCharacteristics.optimalVoltage);
|
|
driveMotor.setVoltageCompensation(configuration.physicalCharacteristics.optimalVoltage);
|
|
angleMotor.setCurrentLimit(configuration.physicalCharacteristics.angleMotorCurrentLimit);
|
|
driveMotor.setCurrentLimit(configuration.physicalCharacteristics.driveMotorCurrentLimit);
|
|
angleMotor.setLoopRampRate(configuration.physicalCharacteristics.angleMotorRampRate);
|
|
driveMotor.setLoopRampRate(configuration.physicalCharacteristics.driveMotorRampRate);
|
|
|
|
// Config angle encoders
|
|
absoluteEncoder = moduleConfiguration.absoluteEncoder;
|
|
if (absoluteEncoder != null)
|
|
{
|
|
absoluteEncoder.factoryDefault();
|
|
absoluteEncoder.configure(moduleConfiguration.absoluteEncoderInverted);
|
|
angleMotor.setPosition(getAbsolutePosition());
|
|
}
|
|
|
|
// Config angle motor/controller
|
|
angleMotor.configureIntegratedEncoder(moduleConfiguration.conversionFactors.angle);
|
|
angleMotor.configurePIDF(moduleConfiguration.anglePIDF);
|
|
angleMotor.configurePIDWrapping(0, 180);
|
|
angleMotor.setInverted(moduleConfiguration.angleMotorInverted);
|
|
angleMotor.setMotorBrake(false);
|
|
|
|
// Config drive motor/controller
|
|
driveMotor.configureIntegratedEncoder(moduleConfiguration.conversionFactors.drive);
|
|
driveMotor.configurePIDF(moduleConfiguration.velocityPIDF);
|
|
driveMotor.setInverted(moduleConfiguration.driveMotorInverted);
|
|
driveMotor.setMotorBrake(true);
|
|
|
|
driveMotor.burnFlash();
|
|
angleMotor.burnFlash();
|
|
|
|
if (SwerveDriveTelemetry.isSimulation)
|
|
{
|
|
simModule = new SwerveModuleSimulation();
|
|
}
|
|
|
|
lastState = getState();
|
|
|
|
noEncoderWarning = new Alert("Motors",
|
|
"There is no Absolute Encoder on module #" +
|
|
moduleNumber,
|
|
Alert.AlertType.WARNING);
|
|
encoderOffsetWarning = new Alert("Motors",
|
|
"Pushing the Absolute Encoder offset to the encoder failed on module #" +
|
|
moduleNumber,
|
|
Alert.AlertType.WARNING);
|
|
}
|
|
|
|
/**
|
|
* Set the voltage compensation for the swerve module motor.
|
|
*
|
|
* @param optimalVoltage Nominal voltage for operation to output to.
|
|
*/
|
|
public void setAngleMotorVoltageCompensation(double optimalVoltage)
|
|
{
|
|
angleMotor.setVoltageCompensation(optimalVoltage);
|
|
}
|
|
|
|
/**
|
|
* Set the voltage compensation for the swerve module motor.
|
|
*
|
|
* @param optimalVoltage Nominal voltage for operation to output to.
|
|
*/
|
|
public void setDriveMotorVoltageCompensation(double optimalVoltage)
|
|
{
|
|
driveMotor.setVoltageCompensation(optimalVoltage);
|
|
}
|
|
|
|
|
|
/**
|
|
* Queue synchronization of the integrated angle encoder with the absolute encoder.
|
|
*/
|
|
public void queueSynchronizeEncoders()
|
|
{
|
|
if (absoluteEncoder != null)
|
|
{
|
|
synchronizeEncoderQueued = true;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Set the desired state of the swerve module. <br /><b>WARNING: If you are not using one of the functions from
|
|
* {@link SwerveDrive} you may screw up {@link SwerveDrive#kinematics}</b>
|
|
*
|
|
* @param desiredState Desired swerve module state.
|
|
* @param isOpenLoop Whether to use open loop (direct percent) or direct velocity control.
|
|
* @param force Disables optimizations that prevent movement in the angle motor and forces the desired state
|
|
* onto the swerve module.
|
|
*/
|
|
public void setDesiredState(SwerveModuleState desiredState, boolean isOpenLoop, boolean force)
|
|
{
|
|
desiredState = SwerveModuleState.optimize(desiredState, Rotation2d.fromDegrees(getAbsolutePosition()));
|
|
|
|
if (isOpenLoop)
|
|
{
|
|
double percentOutput = desiredState.speedMetersPerSecond / maxSpeed;
|
|
driveMotor.set(percentOutput);
|
|
} else
|
|
{
|
|
// Taken from the CTRE SwerveModule class.
|
|
// https://api.ctr-electronics.com/phoenix6/release/java/src-html/com/ctre/phoenix6/mechanisms/swerve/SwerveModule.html#line.46
|
|
/* From FRC 900's whitepaper, we add a cosine compensator to the applied drive velocity */
|
|
/* To reduce the "skew" that occurs when changing direction */
|
|
double steerMotorError = desiredState.angle.getDegrees() - getAbsolutePosition();
|
|
/* If error is close to 0 rotations, we're already there, so apply full power */
|
|
/* If the error is close to 0.25 rotations, then we're 90 degrees, so movement doesn't help us at all */
|
|
double cosineScalar = Math.cos(Units.degreesToRadians(steerMotorError));
|
|
/* Make sure we don't invert our drive, even though we shouldn't ever target over 90 degrees anyway */
|
|
if (cosineScalar < 0.0 || desiredState.speedMetersPerSecond == 0)
|
|
{
|
|
cosineScalar = 0.0;
|
|
}
|
|
|
|
double velocity = desiredState.speedMetersPerSecond * (cosineScalar);
|
|
driveMotor.setReference(velocity, feedforward.calculate(velocity));
|
|
}
|
|
|
|
/* // Not necessary anymore.
|
|
// If we are forcing the angle
|
|
if (!force)
|
|
{
|
|
// Prevents module rotation if speed is less than 1%
|
|
SwerveMath.antiJitter(desiredState, lastState, Math.min(maxSpeed, 4));
|
|
}
|
|
*/
|
|
|
|
// Prevent module rotation if angle is the same as the previous angle.
|
|
// Synchronize encoders if queued and send in the current position as the value from the absolute encoder.
|
|
if (absoluteEncoder != null && synchronizeEncoderQueued)
|
|
{
|
|
double absoluteEncoderPosition = getAbsolutePosition();
|
|
angleMotor.setPosition(absoluteEncoderPosition);
|
|
angleMotor.setReference(desiredState.angle.getDegrees(), 0, absoluteEncoderPosition);
|
|
synchronizeEncoderQueued = false;
|
|
} else
|
|
{
|
|
angleMotor.setReference(desiredState.angle.getDegrees(), 0);
|
|
}
|
|
|
|
lastState = desiredState;
|
|
|
|
if (SwerveDriveTelemetry.isSimulation)
|
|
{
|
|
simModule.updateStateAndPosition(desiredState);
|
|
}
|
|
|
|
if (SwerveDriveTelemetry.verbosity == TelemetryVerbosity.HIGH)
|
|
{
|
|
SmartDashboard.putNumber("Module[" + configuration.name + "] Speed Setpoint", desiredState.speedMetersPerSecond);
|
|
SmartDashboard.putNumber("Module[" + configuration.name + "] Angle Setpoint", desiredState.angle.getDegrees());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Set the angle for the module.
|
|
*
|
|
* @param angle Angle in degrees.
|
|
*/
|
|
public void setAngle(double angle)
|
|
{
|
|
angleMotor.setReference(angle, 0);
|
|
lastState.angle = Rotation2d.fromDegrees(angle);
|
|
}
|
|
|
|
/**
|
|
* Get the Swerve Module state.
|
|
*
|
|
* @return Current SwerveModule state.
|
|
*/
|
|
public SwerveModuleState getState()
|
|
{
|
|
double velocity;
|
|
Rotation2d azimuth;
|
|
if (!SwerveDriveTelemetry.isSimulation)
|
|
{
|
|
velocity = driveMotor.getVelocity();
|
|
azimuth = Rotation2d.fromDegrees(getAbsolutePosition());
|
|
} else
|
|
{
|
|
return simModule.getState();
|
|
}
|
|
return new SwerveModuleState(velocity, azimuth);
|
|
}
|
|
|
|
/**
|
|
* Get the position of the swerve module.
|
|
*
|
|
* @return {@link SwerveModulePosition} of the swerve module.
|
|
*/
|
|
public SwerveModulePosition getPosition()
|
|
{
|
|
double position;
|
|
Rotation2d azimuth;
|
|
if (!SwerveDriveTelemetry.isSimulation)
|
|
{
|
|
position = driveMotor.getPosition();
|
|
azimuth = Rotation2d.fromDegrees(getAbsolutePosition());
|
|
} else
|
|
{
|
|
return simModule.getPosition();
|
|
}
|
|
return new SwerveModulePosition(position, azimuth);
|
|
}
|
|
|
|
/**
|
|
* Get the absolute position. Falls back to relative position on reading failure.
|
|
*
|
|
* @return Absolute encoder angle in degrees in the range [0, 360).
|
|
*/
|
|
public double getAbsolutePosition()
|
|
{
|
|
double angle;
|
|
if (absoluteEncoder != null)
|
|
{
|
|
angle = absoluteEncoder.getAbsolutePosition() - angleOffset;
|
|
if (absoluteEncoder.readingError)
|
|
{
|
|
angle = getRelativePosition();
|
|
}
|
|
} else
|
|
{
|
|
angle = getRelativePosition();
|
|
}
|
|
angle %= 360;
|
|
if (angle < 0.0)
|
|
{
|
|
angle += 360;
|
|
}
|
|
|
|
return angle;
|
|
}
|
|
|
|
/**
|
|
* Get the relative angle in degrees.
|
|
*
|
|
* @return Angle in degrees.
|
|
*/
|
|
public double getRelativePosition()
|
|
{
|
|
return angleMotor.getPosition();
|
|
}
|
|
|
|
/**
|
|
* Set the brake mode.
|
|
*
|
|
* @param brake Set the brake mode.
|
|
*/
|
|
public void setMotorBrake(boolean brake)
|
|
{
|
|
driveMotor.setMotorBrake(brake);
|
|
}
|
|
|
|
/**
|
|
* Set the conversion factor for the angle/azimuth motor controller.
|
|
*
|
|
* @param conversionFactor Angle motor conversion factor for PID, should be generated from
|
|
* {@link SwerveMath#calculateDegreesPerSteeringRotation(double, double)} or calculated.
|
|
*/
|
|
public void setAngleMotorConversionFactor(double conversionFactor)
|
|
{
|
|
angleMotor.configureIntegratedEncoder(conversionFactor);
|
|
}
|
|
|
|
/**
|
|
* Set the conversion factor for the drive motor controller.
|
|
*
|
|
* @param conversionFactor Drive motor conversion factor for PID, should be generated from
|
|
* {@link SwerveMath#calculateMetersPerRotation(double, double, double)} or calculated.
|
|
*/
|
|
public void setDriveMotorConversionFactor(double conversionFactor)
|
|
{
|
|
driveMotor.configureIntegratedEncoder(conversionFactor);
|
|
}
|
|
|
|
/**
|
|
* Get the angle {@link SwerveMotor} for the {@link SwerveModule}.
|
|
*
|
|
* @return {@link SwerveMotor} for the angle/steering motor of the module.
|
|
*/
|
|
public SwerveMotor getAngleMotor()
|
|
{
|
|
return angleMotor;
|
|
}
|
|
|
|
/**
|
|
* Get the drive {@link SwerveMotor} for the {@link SwerveModule}.
|
|
*
|
|
* @return {@link SwerveMotor} for the drive motor of the module.
|
|
*/
|
|
public SwerveMotor getDriveMotor()
|
|
{
|
|
return driveMotor;
|
|
}
|
|
|
|
/**
|
|
* Get the {@link SwerveAbsoluteEncoder} for the {@link SwerveModule}.
|
|
*
|
|
* @return {@link SwerveAbsoluteEncoder} for the swerve module.
|
|
*/
|
|
public SwerveAbsoluteEncoder getAbsoluteEncoder()
|
|
{
|
|
return absoluteEncoder;
|
|
}
|
|
|
|
/**
|
|
* Fetch the {@link SwerveModuleConfiguration} for the {@link SwerveModule} with the parsed configurations.
|
|
*
|
|
* @return {@link SwerveModuleConfiguration} for the {@link SwerveModule}.
|
|
*/
|
|
public SwerveModuleConfiguration getConfiguration()
|
|
{
|
|
return configuration;
|
|
}
|
|
|
|
/**
|
|
* Push absolute encoder offset in the memory of the encoder or controller. Also removes the internal angle offset.
|
|
*/
|
|
public void pushOffsetsToControllers()
|
|
{
|
|
if (absoluteEncoder != null)
|
|
{
|
|
if (absoluteEncoder.setAbsoluteEncoderOffset(angleOffset))
|
|
{
|
|
angleOffset = 0;
|
|
} else
|
|
{
|
|
encoderOffsetWarning.set(true);
|
|
}
|
|
} else
|
|
{
|
|
noEncoderWarning.set(true);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Restore internal offset in YAGSL and either sets absolute encoder offset to 0 or restores old value.
|
|
*/
|
|
public void restoreInternalOffset()
|
|
{
|
|
absoluteEncoder.setAbsoluteEncoderOffset(0);
|
|
angleOffset = configuration.angleOffset;
|
|
}
|
|
|
|
/**
|
|
* Get if the last Absolute Encoder had a read issue, such as it does not exist.
|
|
*
|
|
* @return If the last Absolute Encoder had a read issue, or absolute encoder does not exist.
|
|
*/
|
|
public boolean getAbsoluteEncoderReadIssue()
|
|
{
|
|
if (absoluteEncoder == null)
|
|
{
|
|
return true;
|
|
} else
|
|
{
|
|
return absoluteEncoder.readingError;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Update data sent to {@link SmartDashboard}.
|
|
*/
|
|
public void updateTelemetry()
|
|
{
|
|
if (absoluteEncoder != null)
|
|
{
|
|
SmartDashboard.putNumber("Module[" + configuration.name + "] Raw Absolute Encoder",
|
|
absoluteEncoder.getAbsolutePosition());
|
|
}
|
|
SmartDashboard.putNumber("Module[" + configuration.name + "] Raw Angle Encoder", angleMotor.getPosition());
|
|
SmartDashboard.putNumber("Module[" + configuration.name + "] Raw Drive Encoder", driveMotor.getPosition());
|
|
SmartDashboard.putNumber("Module[" + configuration.name + "] Adjusted Absolute Encoder", getAbsolutePosition());
|
|
SmartDashboard.putNumber("Module[" + configuration.name + "] Absolute Encoder Read Issue",
|
|
getAbsoluteEncoderReadIssue() ? 1 : 0);
|
|
}
|
|
}
|