untested joystick recording

This commit is contained in:
Abhi
2023-02-10 18:12:56 -07:00
parent c9a4274c82
commit f9b237592f
@@ -4,53 +4,75 @@
package frc4388.robot.commands; package frc4388.robot.commands;
import java.io.File;
import java.io.IOException; import java.io.IOException;
import java.io.PrintWriter;
import java.nio.file.Files; import java.nio.file.Files;
import java.nio.file.Path; import java.nio.file.Path;
import java.util.HashMap;
import java.util.function.Supplier; import java.util.function.Supplier;
import edu.wpi.first.math.geometry.Translation2d;
import edu.wpi.first.wpilibj2.command.CommandBase; import edu.wpi.first.wpilibj2.command.CommandBase;
import frc4388.robot.subsystems.SwerveDrive;
import frc4388.utility.RobotTime; import frc4388.utility.RobotTime;
import frc4388.utility.controller.DeadbandedXboxController; import frc4388.utility.controller.DeadbandedXboxController;
public class JoystickRecorder extends CommandBase { public class JoystickRecorder extends CommandBase {
DeadbandedXboxController joystick;
Supplier<DeadbandedXboxController> joystickSupplier;
RobotTime time; SwerveDrive swerve;
Supplier<Double> leftXSupplier;
Supplier<Double> leftYSupplier;
Supplier<Double> rightXSupplier;
Supplier<Double> rightYSupplier;
HashMap<Long, double[]> timedInput;
private long startTime;
/** Creates a new JoystickRecorder. */ /** Creates a new JoystickRecorder. */
public JoystickRecorder(Supplier<DeadbandedXboxController> joystickSupplier) { public JoystickRecorder(SwerveDrive swerve, Supplier<Double> leftXSupplier, Supplier<Double> leftYSupplier, Supplier<Double> rightXSupplier, Supplier<Double> rightYSupplier) {
// Use addRequirements() here to declare subsystem dependencies. // Use addRequirements() here to declare subsystem dependencies.
this.joystickSupplier = joystickSupplier; this.swerve = swerve;
time = RobotTime.getInstance(); this.leftXSupplier = leftXSupplier;
this.leftYSupplier = leftYSupplier;
this.rightXSupplier = rightXSupplier;
this.rightYSupplier = rightYSupplier;
} }
// Called when the command is initially scheduled. // Called when the command is initially scheduled.
@Override @Override
public void initialize() { public void initialize() {
joystick = joystickSupplier.get(); timedInput.put((long) 0, new double[] {0.0, 0.0, 0.0, 0.0});
startTime = System.currentTimeMillis();
} }
// Called every time the scheduler runs while the command is scheduled. // Called every time the scheduler runs while the command is scheduled.
@Override @Override
public void execute() { public void execute() {
double leftX = joystick.getLeftX(); double[] inputs = new double[] {leftXSupplier.get(), leftYSupplier.get(), rightXSupplier.get(), rightYSupplier.get()};
double leftY = joystick.getLeftY(); timedInput.put(System.currentTimeMillis() - startTime, inputs);
double rightX = joystick.getRightX();
double rightY = joystick.getRightY();
try { swerve.driveWithInput(new Translation2d(inputs[0], inputs[1]), new Translation2d(-inputs[2], inputs[3]), true);
Files.writeString(Path.of("JoystickInputs.txt"), "Time: " + time.m_robotTime + ", leftX: " + leftX + ", " + "leftY: " + leftY + ", " + "rightX: " + rightX + ", " + "rightY: " + rightY + "\n");
} catch (IOException e) {
e.printStackTrace();
}
} }
// Called once the command ends or is interrupted. // Called once the command ends or is interrupted.
@Override @Override
public void end(boolean interrupted) {} public void end(boolean interrupted) {
File output = new File("JoystickInputs.txt");
try(PrintWriter writer = new PrintWriter(output)) {
for(long millis : timedInput.keySet()) {
writer.println("time: " + millis + ", leftX: " + timedInput.get(millis)[0] + ", leftY: " + timedInput.get(millis)[1] + ", rightX: " + timedInput.get(millis)[2] + ", rightY: " + timedInput.get(millis)[3]);
}
writer.close();
} catch(IOException e) {
e.printStackTrace();
}
}
// Returns true when the command should end. // Returns true when the command should end.
@Override @Override