mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
31 lines
908 B
Python
31 lines
908 B
Python
#
|
|
# Copyright (c) FIRST and other WPILib contributors.
|
|
# Open Source Software; you can modify and/or share it under the terms of
|
|
# the WPILib BSD license file in the root directory of this project.
|
|
#
|
|
|
|
import commands2
|
|
|
|
from subsystems.drivesubsystem import DriveSubsystem
|
|
|
|
|
|
class DriveDistance(commands2.Command):
|
|
def __init__(self, inches: float, speed: float, drive: DriveSubsystem) -> None:
|
|
self.distance = inches
|
|
self.speed = speed
|
|
self.drive = drive
|
|
self.addRequirements(drive)
|
|
|
|
def initialize(self) -> None:
|
|
self.drive.resetEncoders()
|
|
self.drive.arcadeDrive(self.speed, 0)
|
|
|
|
def execute(self) -> None:
|
|
self.drive.arcadeDrive(self.speed, 0)
|
|
|
|
def end(self, interrupted: bool) -> None:
|
|
self.drive.arcadeDrive(0, 0)
|
|
|
|
def isFinished(self) -> bool:
|
|
return abs(self.drive.getAverageEncoderDistance()) >= self.distance
|