drive_choreo.h

Go to the documentation of this file.

 
 
#pragma once
 
#include <choreo/lib/ChoreoSwerveCommand.h>
#include <frc/geometry/Pose2d.h>
#include <frc2/command/Command.h>
#include <frc2/command/CommandHelper.h>
#include <units/angle.h>
#include <units/length.h>
#include <wpi/DataLog.h>
 
#include <chrono>
#include <string>
 
#include "subsystems/swerve_drive_subsystem.h"
 
class DriveChoreo : public frc2::CommandHelper<frc2::Command, DriveChoreo> {
 public:
  DriveChoreo(SwerveDriveSubsystem& drive, const std::string& trajectoryName, const bool initializeOdometry = false);
 
  void Initialize() override;
 
  void Execute() override;
 
  void End(bool interrupted) override;
 
  bool IsFinished() override;
 
  [[nodiscard]] static bool IsAtEndPoint(SwerveDriveSubsystem& drive,
                                         const std::string& trajectoryName,
                                         const units::inch_t translationalTolerance = 6_in,
                                         const units::degree_t rotationalTolerance = 1.0_deg);
 
  [[nodiscard]] static units::inch_t EndpointShotDistance(const std::string& trajectoryName);
 
 private:
  SwerveDriveSubsystem& m_Drive;
  const choreolib::ChoreoTrajectory m_trajectory;
  choreolib::ChoreoTrajectory m_orientedTrajectory;
  choreolib::ChoreoSwerveCommand m_ChoreoCommand;
  const bool m_initializeOdometry;
  std::chrono::time_point<std::chrono::steady_clock> m_startTime;
  wpi::log::StructLogEntry<frc::Pose2d> m_desiredAutoPositionLogger;
  wpi::log::StructArrayLogEntry<frc::Pose2d> m_autoTrajectoryLogger;
};