mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
dca59147e1
This also updates the bazel scripts to behave more like the C++ and Java examples, and updates the copybara scripts to be able to sync up `mostrobotpy`
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
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 typing
|
|
import commands2
|
|
from subsystems.drivetrain import Drivetrain
|
|
|
|
|
|
class ArcadeDrive(commands2.Command):
|
|
def __init__(
|
|
self,
|
|
drive: Drivetrain,
|
|
xaxisVelocitySupplier: typing.Callable[[], float],
|
|
zaxisRotateSupplier: typing.Callable[[], float],
|
|
) -> None:
|
|
"""Creates a new ArcadeDrive. This command will drive your robot according to the velocity supplier
|
|
lambdas. This command does not terminate.
|
|
|
|
:param drivetrain: The drivetrain subsystem on which this command will run
|
|
:param xaxisVelocitySupplier: Callable supplier of forward/backward velocity
|
|
:param zaxisRotateSupplier: Callable supplier of rotational velocity
|
|
"""
|
|
|
|
self.drive = drive
|
|
self.xaxisVelocitySupplier = xaxisVelocitySupplier
|
|
self.zaxisRotateSupplier = zaxisRotateSupplier
|
|
|
|
self.addRequirements(self.drive)
|
|
|
|
def execute(self) -> None:
|
|
self.drive.arcadeDrive(self.xaxisVelocitySupplier(), self.zaxisRotateSupplier())
|