Castor2026/components/agitator.py

25 lines
638 B
Python

import magicbot
from phoenix6.hardware import TalonFX
from phoenix6.configs.talon_fx_configs import TalonFXConfiguration
from phoenix6.signals import NeutralModeValue
from phoenix6.controls import VoltageOut
class Agitator:
motor: TalonFX
voltage = magicbot.will_reset_to(0.0)
def execute(self):
self.motor.set_control(VoltageOut(self.voltage))
def setup(self):
config = TalonFXConfiguration()
config.motor_output.neutral_mode = NeutralModeValue.BRAKE
self.motor.configurator.apply(config)
def spin(self):
self.voltage = 1.0
def stop(self):
self.voltage = 0.0