Skip to main content

Swerve

What is Swerve?

Swerve drive, or swerve, is a type of drivetrain. It has one wheel in each corner of the robot. Each wheel is controlled by two motors, one for driving and one for steering. This allows the robot to turn in place and drive while facing in any direction. Unlike most hyper-mobile drivetrains, it can and should use high-traction wheels, giving it phenomenal pushing power. It has the pushing power of tank drive combined with the mobility of mecanum drive. However, it does have significant downsides, mostly in power issues, cost (both in time and money), and programming issues.

Why Is This Good?

Swerve drive has a number of benefits, both obvious and unobvious. The obvious ones are metioned above: high speed and high manuverability without sacrifices to either. A slighty less obvious benefit is connected to that mobility, and more specifically the rotation. The robot is able to move in a direction while pointing in another, allowing for extreme precision during shooter games. The least obvious benefit is simply the amount of things the prereqs allow. Just getting the wheels moving in the same direction requires a gyroscope and field map, which can easily be paired with a limelight for extreme degrees of autonomy.

Prerequisites

Swerve is annoying to make. It requires a gyroscope, as without it the robot cannot tell which way it is facing. It needs eight motors all synchronized to each other, PID loops to keep the drive motors moving at about the same speed, and more code to make sure everything is pointing the right direction if the robot wants to turn. Here is how to make some of that.

Components

This method of making swerve is built with Phoenix and Kraken motors, uses Java as the programming language, and uses a command-based framework. If you are not using these, treat this as more guidelines for building your own swerve. It uses four files: the Constants file (which also contains the CTRE configs), a Swerve Module file for the code for the individual swerve modules, a Drivetrain file for the actual drivetrain subsystem, and a Drive command.

Swerve Module - Imports

import com.ctre.phoenix6.configs.CANcoderConfiguration;
import com.ctre.phoenix6.configs.TalonFXConfiguration;
import com.ctre.phoenix6.controls.CoastOut;
import com.ctre.phoenix6.controls.NeutralOut;
import com.ctre.phoenix6.controls.PositionDutyCycle;
import com.ctre.phoenix6.controls.StaticBrake;
import com.ctre.phoenix6.controls.VelocityDutyCycle;
import com.ctre.phoenix6.hardware.CANcoder;
import com.ctre.phoenix6.hardware.TalonFX;

import edu.wpi.first.math.MathUtil;
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.wpilibj.DriverStation;
import edu.wpi.first.wpilibj.shuffleboard.ShuffleboardTab;
import frc.robot.settings.Constants.DriveConstants;

Swerve Module

A swerve drive is made up of four modules. Each module has a wheel, two motors, and an encoder. In the code, you need the motors, an encoder, and a Rotation2d. You also need a bunch of variables, like the steering angle and driving speed. These specific varaibles are used in the controlling bit specifically.

  private final TalonFX m_driveMotor;
private final TalonFX m_steerMotor;
private final CANcoder m_steerEncoder;
private final Rotation2d m_steerEncoderOffset;

private double m_desiredSteerAngle;

private double m_desiredDriveSpeed;

private VelocityDutyCycle m_driveControl = new VelocityDutyCycle(0);
private PositionDutyCycle m_steerControl = new PositionDutyCycle(0);
private CoastOut m_coastControl = new CoastOut();
private StaticBrake m_brakeControl = new StaticBrake();
private NeutralOut m_neutralControl = new NeutralOut();

The next bit demonstrates how to get the speed, position, and rotation of the module:

  public SwerveModuleState getState() {
return new SwerveModuleState(
getSpeedMetersPerSecond(), getRotation());
}
public SwerveModulePosition getPosition() {
return new SwerveModulePosition(
getDriveDistanceMeters(), getRotation());
}

As well as the angles of the steering motor. Angles because it needs to have where it is and where it wants to be. This is when one of the variables above shows up.

  public Rotation2d getRotation() {
return Rotation2d.fromRotations(MathUtil.inputModulus(m_steerMotor.getPosition().getValue(), -0.5, 0.5));
}

public double getTargetAngle() {
return m_desiredSteerAngle;
}

This part finds the motor offset: how much the steer motor needs to be offset from 0 to drive in a straight line. The other method is to manually position every motor offset of every module you make and then put them in as constants.

public double findOffset() {
return MathUtil.inputModulus(
(m_steerEncoder.getPosition().getValue()+m_steerEncoderOffset.getRotations()),
-0.5,
0.5);
}

This section of code shows how to get the actual speed and velocity. As usual, this is divided between the current state and the target state. It also has the distance of the drive motor, which is very helpful for autonomous.

  public double getTargetSpeedMetersPerSecond() {
return m_desiredDriveSpeed;
}

public double getSpeedMetersPerSecond() {
return (m_driveMotor.getVelocity().getValue() * DriveConstants.DRIVETRAIN_ROTATIONS_TO_METERS);
}

public double getDriveDistanceMeters() {
return (m_driveMotor.getPosition().getValue() * DriveConstants.DRIVETRAIN_ROTATIONS_TO_METERS);
}

These functions are useful for

  public TalonFX getDriveMotor(){
return m_driveMotor;
}

public TalonFX getSteerMotor(){
return m_steerMotor;
}

The last part is the code that actually gives all of the targets. It also optimizes the reference state to avoid spinning further than 90 degrees.

  public void setDesiredState(SwerveModuleState desiredState) {

if (desiredState.angle == null) {
DriverStation.reportWarning("Cannot set module angle to null.", true);
}
SwerveModuleState state =
SwerveModuleState.optimize(desiredState, getRotation());

m_desiredSteerAngle = MathUtil.inputModulus(state.angle.getRotations(), -0.5, 0.5);
m_desiredDriveSpeed = state.speedMetersPerSecond / DriveConstants.DRIVETRAIN_ROTATIONS_TO_METERS;

if (Math.abs(m_desiredDriveSpeed) <= 0.001) {
m_driveMotor.setControl(m_neutralControl);
m_steerMotor.setControl(m_steerControl.withPosition(m_desiredSteerAngle));
} else {
m_driveMotor.setControl(m_driveControl.withVelocity(m_desiredDriveSpeed).withFeedForward(m_desiredDriveSpeed/DriveConstants.MAX_VELOCITY_RPS_EMPIRICAL));
m_steerMotor.setControl(m_steerControl.withPosition(m_desiredSteerAngle));
}
}

One important thing to know is that for this part specifically, the rule about not copy-pasting anything you don't know the function of doesn't quite apply.

Drivetrain - Imports

Please do not just copy-paste, your code might have something else.

import static frc.robot.settings.Constants.DriveConstants.BL_DRIVE_MOTOR_ID;
import static frc.robot.settings.Constants.DriveConstants.BL_STEER_ENCODER_ID;
import static frc.robot.settings.Constants.DriveConstants.BL_STEER_MOTOR_ID;
import static frc.robot.settings.Constants.DriveConstants.BR_DRIVE_MOTOR_ID;
import static frc.robot.settings.Constants.DriveConstants.BR_STEER_ENCODER_ID;
import static frc.robot.settings.Constants.DriveConstants.BR_STEER_MOTOR_ID;
import static frc.robot.settings.Constants.DriveConstants.CANIVORE_DRIVETRAIN;
import static frc.robot.settings.Constants.DriveConstants.DRIVETRAIN_PIGEON_ID;
import static frc.robot.settings.Constants.DriveConstants.DRIVE_ODOMETRY_ORIGIN;
import static frc.robot.settings.Constants.DriveConstants.FL_DRIVE_MOTOR_ID;
import static frc.robot.settings.Constants.DriveConstants.FL_STEER_ENCODER_ID;
import static frc.robot.settings.Constants.DriveConstants.FL_STEER_MOTOR_ID;
import static frc.robot.settings.Constants.DriveConstants.FR_DRIVE_MOTOR_ID;
import static frc.robot.settings.Constants.DriveConstants.FR_STEER_ENCODER_ID;
import static frc.robot.settings.Constants.DriveConstants.FR_STEER_MOTOR_ID;
import static frc.robot.settings.Constants.ShooterConstants.OFFSET_MULTIPLIER;

import java.util.Arrays;
import java.util.Collections;
import java.util.Optional;

import javax.swing.text.html.Option;

import com.ctre.phoenix6.hardware.Pigeon2;

import edu.wpi.first.math.estimator.SwerveDrivePoseEstimator;
import edu.wpi.first.math.geometry.Pose2d;
import edu.wpi.first.math.geometry.Rotation2d;
import edu.wpi.first.math.geometry.Translation2d;
import edu.wpi.first.math.kinematics.ChassisSpeeds;
import edu.wpi.first.math.kinematics.SwerveDriveKinematics;
import edu.wpi.first.math.kinematics.SwerveModulePosition;
import edu.wpi.first.math.kinematics.SwerveModuleState;
import edu.wpi.first.wpilibj.DriverStation;
import edu.wpi.first.wpilibj.DriverStation.Alliance;
import edu.wpi.first.wpilibj.Preferences;
import edu.wpi.first.wpilibj.Timer;
import edu.wpi.first.wpilibj.smartdashboard.Field2d;
import edu.wpi.first.wpilibj.smartdashboard.SmartDashboard;
import edu.wpi.first.wpilibj2.command.InstantCommand;
import edu.wpi.first.wpilibj2.command.SubsystemBase;
import frc.robot.LimelightHelpers;
import frc.robot.settings.Constants;
import frc.robot.settings.LimelightValues;
import frc.robot.settings.Constants.CTREConfigs;
import frc.robot.settings.Constants.Vision;
import frc.robot.settings.Constants.DriveConstants.Offsets;
import frc.robot.settings.Constants.DriveConstants.Positions;

Drivetrain

The first step is to create a set of CTRE Configs and Kinematics, arrays for the swerve modules and rotations, and some field stuff. I'm leaving the numbers here because these ones specifically don't matter that much.

public static final CTREConfigs ctreConfig = new CTREConfigs();
public SwerveDriveKinematics kinematics = DriveConstants.kinematics;

private final Pigeon2 pigeon = new Pigeon2(DRIVETRAIN_PIGEON_ID, CANIVORE_DRIVETRAIN);

/**
* These are our modules. We initialize them in the constructor.
* 0 = Front Left
* 1 = Front Right
* 2 = Back Left
* 3 = Back Right
*/
private final SwerveModule[] modules;
private final Rotation2d[] lastAngles;
private int accumulativeLoops;

private final SwerveDrivePoseEstimator odometer;
private final Field2d m_field = new Field2d();

Limelight limelight; //This is optional, but helpful.

Then we have the constructor. This first part has the array, field, offset, and limelight setup. Offsets are actually a thing that haven't been mentioned yet. Because of how humans work and the way serve modules are built, it is basically impossible to have every single robot naturally be both at 0 rotations and facing forwards. Offsets are how much from zero the steer motor has to be offset.

public DrivetrainSubsystem() {
this.limelight=Limelight.getInstance();

Preferences.initDouble("FL offset", 0);
Preferences.initDouble("FR offset", 0);
Preferences.initDouble("BL offset", 0);
Preferences.initDouble("BR offset", 0);
PathPlannerLogging.setLogActivePathCallback((poses) -> m_field.getObject("path").setPoses(poses));
SmartDashboard.putData("Field", m_field);
SmartDashboard.putBoolean("Vision/force use limelight", false);
modules = new SwerveModule[4];
lastAngles = new Rotation2d[] {new Rotation2d(), new Rotation2d(), new Rotation2d(), new Rotation2d()}; // manually make empty angles to avoid null errors.

The next step is to make all of the modules and the Pose Estimator. The pose estimator is the thing that lets the code and robot know what way the wheels are pointing, where the robot is pointing, and where the robot is.

modules[0] = new SwerveModule(
"FL",
FL_DRIVE_MOTOR_ID,
FL_STEER_MOTOR_ID,
FL_STEER_ENCODER_ID,
Rotation2d.fromRotations(Preferences.getDouble("FL offset", 0)),
CANIVORE_DRIVETRAIN);
modules[1] = new SwerveModule(
"FR",
FR_DRIVE_MOTOR_ID,
FR_STEER_MOTOR_ID,
FR_STEER_ENCODER_ID,
Rotation2d.fromRotations(Preferences.getDouble("FR offset", 0)),
CANIVORE_DRIVETRAIN);
modules[2] = new SwerveModule(
"BL",
BL_DRIVE_MOTOR_ID,
BL_STEER_MOTOR_ID,
BL_STEER_ENCODER_ID,
Rotation2d.fromRotations(Preferences.getDouble("BL offset", 0)),
CANIVORE_DRIVETRAIN);
modules[3] = new SwerveModule(
"BR",
BR_DRIVE_MOTOR_ID,
BR_STEER_MOTOR_ID,
BR_STEER_ENCODER_ID,
Rotation2d.fromRotations(Preferences.getDouble("BR offset", 0)),
CANIVORE_DRIVETRAIN);

odometer = new SwerveDrivePoseEstimator(
kinematics,
getGyroscopeRotation(),
getModulePositions(),
DRIVE_ODOMETRY_ORIGIN);
}

The next bit resets what the gyro thinks is 0 degrees.

public void zeroGyroscope() {
if(DriverStation.getAlliance().isPresent() && DriverStation.getAlliance().get() == Alliance.Red) {
zeroGyroscopeTo(180);
} else {
zeroGyroscopeTo(0);
}
}
public void zeroGyroscopeTo(double angleDeg) {
resetOdometry(new Pose2d(getPose().getTranslation(), Rotation2d.fromDegrees(angleDeg)));
}

This segment is a giant list of things that the robot needs to get. It should be noted that getHeadingLooped gets the heading out of 360 rather than the accumulative heading.

public double getHeadingLooped() {
accumulativeLoops = (int) (getHeadingDegrees()/180); //finding the amount of times that 360 goes into the heading, as an int
return getHeadingDegrees()-180*(accumulativeLoops);
}
public Rotation2d getGyroscopeRotation() {
return pigeon.getRotation2d();
}
public Rotation2d getOdometryRotation() {
return odometer.getEstimatedPosition().getRotation();
}
public double getHeadingDegrees() {
return odometer.getEstimatedPosition().getRotation().getDegrees();
}
public ChassisSpeeds getChassisSpeeds() {
return kinematics.toChassisSpeeds(getModuleStates());
}

This section is how the swerve modules are used. It gets the positions and states of the modules (using arrays to apply to all of them at once) and sets the offsets (see earlier).

    public SwerveModulePosition[] getModulePositions() {
SwerveModulePosition[] positions = new SwerveModulePosition[4];
for (int i = 0; i < 4; i++) positions[i] = modules[i].getPosition();
return positions;
}
public SwerveModuleState[] getModuleStates() {
SwerveModuleState[] states = new SwerveModuleState[4];
for (int i = 0; i < 4; i++) states[i] = modules[i].getState();
return states;
}
public void setEncoderOffsets() {
Preferences.setDouble("FL offset", modules[0].findOffset());
Preferences.setDouble("FR offset", modules[1].findOffset());
Preferences.setDouble("BL offset", modules[2].findOffset());
Preferences.setDouble("BR offset", modules[3].findOffset());
}

This part gets and resets the pose and gyroscope.

    public Pose2d getPose() {
return odometer.getEstimatedPosition();
}
public void resetOdometry(Pose2d pose) {
odometer.resetPosition(getGyroscopeRotation(), getModulePositions(), pose);
}

Next we have the giant list of functions that actually lets one drive the robot. pointWheelsForward sets the speed and rotation of the modules to 0. pointWheelsInward does a similar thing, but instead of setting all of the wheels forward it points them all towards the center of the robot. This is very useful as a demonstration tool for potential team members to get across why we use swerve drive. Drive is what actually makes the robot drive, setting all of the module states. Stop just turns everything off without reseting the wheel directions.

    public void pointWheelsForward() {
for (int i = 0; i < 4; i++) {
setModule(i, new SwerveModuleState(0, new Rotation2d()));
}
}
public void pointWheelsInward() {
setModule(0, new SwerveModuleState(0, Rotation2d.fromDegrees(-135)));
setModule(1, new SwerveModuleState(0, Rotation2d.fromDegrees(135)));
setModule(2, new SwerveModuleState(0, Rotation2d.fromDegrees(-45)));
setModule(3, new SwerveModuleState(0, Rotation2d.fromDegrees(45)));
}
public void drive(ChassisSpeeds chassisSpeeds) {

SwerveModuleState[] desiredStates = kinematics.toSwerveModuleStates(ChassisSpeeds.discretize(chassisSpeeds, 0.02));
double maxSpeed = Collections.max(Arrays.asList(desiredStates)).speedMetersPerSecond;
if (maxSpeed <= DriveConstants.DRIVE_DEADBAND_MPS) {
for (int i = 0; i < 4; i++) {
stop();
}
} else {
setModuleStates(desiredStates);
}
}
/**
* Sets all module drive speeds to 0, but leaves the wheel angles where they were.
*/
public void stop() {
for (int i = 0; i < 4; i++) {
modules[i].setDesiredState(new SwerveModuleState(0, lastAngles[i]));
}
}

This part sets the swerve module states. I'm not sure why these functions are after their use cases in the code, but they are.

public void setModuleStates(SwerveModuleState[] desiredStates) {
SwerveDriveKinematics.desaturateWheelSpeeds(desiredStates, DriveConstants.MAX_VELOCITY_METERS_PER_SECOND);
for (int i = 0; i < 4; i++) {
setModule(i, desiredStates[i]);
}
}
private void setModule(int i, SwerveModuleState desiredState) {
modules[i].setDesiredState(desiredState);
lastAngles[i] = desiredState.angle;
}

This section updates the odometry, either from vision pose or Apriltags.

public void updateOdometry() {
odometer.updateWithTime(Timer.getFPGATimestamp(), getGyroscopeRotation(), getModulePositions());
}
/**
* Provide the odometry a vision pose estimate, only if there is a trustworthy pose available.
* <p>
* Each time a vision pose is supplied, the odometry pose estimation will change a little,
* larger pose shifts will take multiple calls to complete.
*/
public void updateOdometryWithVision() {
PoseEstimate estimate = limelight.getTrustedPose(getPose());
if (estimate != null) {
odometer.addVisionMeasurement(new Pose2d(estimate.pose.getTranslation(), getOdometryRotation()), estimate.timestampSeconds);
RobotState.getInstance().LimelightsUpdated = true;
} else {
RobotState.getInstance().LimelightsUpdated = false;
}
}
/**
* Set the odometry using the current apriltag estimate, disregarding the pose trustworthyness.
* <p>
* You only need to run this once for it to take effect.
*/
public void forceUpdateOdometryWithVision() {
PoseEstimate estimate = limelight.getValidPose();
if (estimate != null) {
resetOdometry(estimate.pose);
} else {
System.err.println("No valid limelight estimate to reset from. (Drivetrain.forceUpdateOdometryWithVision)");
}
}

These last chunks are all in the Periodic method.

    @Override
public void periodic() {
if (DriverStation.getAlliance().isPresent()) {
SmartDashboard.putString("alliance:", DriverStation.getAlliance().get().toString());
}
updateOdometry();
if (Preferences.getBoolean("Use Limelight", false)) {
if (SmartDashboard.getBoolean("Vision/force use limelight", false)) {
forceUpdateOdometryWithVision();
} else {
updateOdometryWithVision();
}
} else {
RobotState.getInstance().LimelightsUpdated = false;
}

m_field.setRobotPose(odometer.getEstimatedPosition());
SmartDashboard.putNumber("Robot Angle", getOdometryRotation().getDegrees());
SmartDashboard.putString("Robot Location", getPose().getTranslation().toString());
SmartDashboard.putNumber("TESTING robot angle difference", getSpeakerAngleDifference());
if (getSpeakerAngleDifference()<DriveConstants.ALLOWED_ERROR) {
runsValid++;
} else {
runsValid = 0;
}

This part is still in periodic. It's for data logging and keeping track of numbers.

        for (int i = 0; i < 4; i++) {
motorLoggers[i].log(modules[i].getDriveMotor());
}
SmartDashboard.putNumber("DRIVETRAIN/forward speed", getChassisSpeeds().vxMetersPerSecond);
SmartDashboard.putNumber("DRIVETRAIN/rotational speed", Math.toDegrees(getChassisSpeeds().omegaRadiansPerSecond));
SmartDashboard.putNumber("DRIVETRAIN/gyroscope rotation degrees", getPose().getRotation().getDegrees());
for (int i = 0; i < 4; i++) {
motorLoggers[i].log(modules[i].getDriveMotor());
}
}

CTRE Configs

The Company CTRE has a large amount of things one can do with the Talon's integrated motor controllers and other things the company makes. All of these are extremely important for swerve. These should probably go in Constants. The first chunk makes configurations that can be implemented. (The brackets have matches at the bottom.)

public static final class CTREConfigs {
public TalonFXConfiguration driveMotorConfig;
public TalonFXConfiguration steerMotorConfig;
public CANcoderConfiguration steerEncoderConfig;
public Pigeon2Configuration pigeon2Config;

public CTREConfigs() {
driveMotorConfig = new TalonFXConfiguration();
steerMotorConfig = new TalonFXConfiguration();
steerEncoderConfig = new CANcoderConfiguration();
pigeon2Config = new Pigeon2Configuration();

Each of the following sections is a specific configuration. This one is the steering motor:

      steerMotorConfig.Feedback.RotorToSensorRatio = 1/DriveConstants.DRIVETRAIN_STEER_REDUCTION;
steerMotorConfig.MotorOutput.Inverted = DriveConstants.DRIVETRAIN_STEER_INVERTED;
steerMotorConfig.Slot0.kP = DriveConstants.k_STEER_P;
steerMotorConfig.Slot0.kI = DriveConstants.k_STEER_I;
steerMotorConfig.Slot0.kD = DriveConstants.k_STEER_D;
steerMotorConfig.Slot0.kS = DriveConstants.k_STEER_FF_S;
steerMotorConfig.Slot0.kV = DriveConstants.k_STEER_FF_V;
steerMotorConfig.Voltage.PeakForwardVoltage = 12;
steerMotorConfig.Voltage.PeakReverseVoltage = -12;
steerMotorConfig.Feedback.FeedbackSensorSource = FeedbackSensorSourceValue.RemoteCANcoder;
steerMotorConfig.ClosedLoopGeneral.ContinuousWrap = true;

This the drive motor:

      driveMotorConfig.Feedback.SensorToMechanismRatio = 1/DriveConstants.DRIVETRAIN_DRIVE_REDUCTION;
driveMotorConfig.MotorOutput.Inverted = DriveConstants.DRIVETRAIN_DRIVE_INVERTED;
driveMotorConfig.MotorOutput.DutyCycleNeutralDeadband = 0.0;
driveMotorConfig.ClosedLoopRamps.DutyCycleClosedLoopRampPeriod = DriveConstants.DRIVE_MOTOR_RAMP;
driveMotorConfig.Slot0.kP = DriveConstants.k_DRIVE_P;
driveMotorConfig.Slot0.kI = DriveConstants.k_DRIVE_I;
driveMotorConfig.Slot0.kD = DriveConstants.k_DRIVE_D;
driveMotorConfig.Slot0.kS = DriveConstants.k_DRIVE_FF_S;
driveMotorConfig.Slot0.kV = DriveConstants.k_DRIVE_FF_V;
driveMotorConfig.Voltage.PeakForwardVoltage = 12;
driveMotorConfig.Voltage.PeakReverseVoltage = -12;
driveMotorConfig.MotorOutput.NeutralMode = NeutralModeValue.Brake;

The steering encoder and pigeon also have some:

      //  Steer encoder.
steerEncoderConfig.MagnetSensor.AbsoluteSensorRange = AbsoluteSensorRangeValue.Signed_PlusMinusHalf;

// Pigeon 2.
pigeon2Config.MountPose.MountPosePitch = 0;
pigeon2Config.MountPose.MountPoseRoll = 0;
pigeon2Config.MountPose.MountPoseYaw = 0;
}
}

Required Constants

Here is the list of the constants required to get this working. It is in a seperate page because the list is very very long.

The Drive Command - Imports

package frc.robot.commands;

import java.util.function.BooleanSupplier;
import java.util.function.DoubleSupplier;

import edu.wpi.first.math.kinematics.ChassisSpeeds;
import edu.wpi.first.wpilibj.DriverStation;
import edu.wpi.first.wpilibj.DriverStation.Alliance;
import edu.wpi.first.wpilibj2.command.Command;
import frc.robot.settings.Constants.DriveConstants;
import frc.robot.subsystems.DrivetrainSubsystem;

The Drive Command

This is the command that actually lets us drive the robot. The first bit just makes the suppliers needed and sets up a drivetrain for it to call. Normal class stuff.

public class Drive extends Command {
private final DrivetrainSubsystem drivetrain;
private final BooleanSupplier robotCentricMode;
private final DoubleSupplier translationXSupplier;
private final DoubleSupplier translationYSupplier;
private final DoubleSupplier rotationSupplier;
private int invert;

Then we have the constructor, which does normal command constructor things like adding requirements.

    public Drive(DrivetrainSubsystem drivetrainSubsystem,
BooleanSupplier robotCentricMode,
DoubleSupplier translationXSupplier,
DoubleSupplier translationYSupplier,
DoubleSupplier rotationSupplier) {
this.drivetrain = drivetrainSubsystem;
this.robotCentricMode = robotCentricMode;
this.translationXSupplier = translationXSupplier;
this.translationYSupplier = translationYSupplier;
this.rotationSupplier = rotationSupplier;
addRequirements(drivetrainSubsystem);
}

The Execute method has three parts. The first is only here because the first year this was disassembled we had an asymetrical field. It inverts the motors depending on the alliance.

    @Override
public void execute() {
// You can use `new ChassisSpeeds(...)` for robot-oriented movement instead of field-oriented movement
if(DriverStation.getAlliance().get() == Alliance.Red) {
invert = -1;
} else {
invert = 1;
}

The second and third parts are basically the same, and handle the actual driving part. One works when the robot is in robot-centric mode, the other works when the robot is in field-centric mode.

if (robotCentricMode.getAsBoolean()) {
drivetrain.drive(new ChassisSpeeds(
translationXSupplier.getAsDouble() * DriveConstants.MAX_VELOCITY_METERS_PER_SECOND * invert,
translationYSupplier.getAsDouble() * DriveConstants.MAX_VELOCITY_METERS_PER_SECOND * invert,
rotationSupplier.getAsDouble() * DriveConstants.MAX_ANGULAR_VELOCITY_RADIANS_PER_SECOND
));
} else {
drivetrain.drive(
ChassisSpeeds.fromFieldRelativeSpeeds(
translationXSupplier.getAsDouble() * DriveConstants.MAX_VELOCITY_METERS_PER_SECOND * invert,
translationYSupplier.getAsDouble() * DriveConstants.MAX_VELOCITY_METERS_PER_SECOND * invert,
rotationSupplier.getAsDouble() * DriveConstants.MAX_ANGULAR_VELOCITY_RADIANS_PER_SECOND,
drivetrain.getPose().getRotation()
)
);
}
}

Finally, the end method just turns everything off. And now you know how to code a swerve drive, have fun!

    @Override
public void end(boolean interrupted) {
drivetrain.drive(new ChassisSpeeds(0.0, 0.0, 0.0));
}
}

Problems

Problem: My wheels are spinning randomly!

This issue shows up a lot whenever new swerve modules show up. One possible cause has to do with inversions - done improperly, inverting the modules will also invert the PID loop, causing it to make the modules as unsynched as possible. This is very funny, and also pretty easy to fix. Just double check where the inversion is occuring.