diff --git a/ohmpi/hardware_components/mb_2024_0_2.py b/ohmpi/hardware_components/mb_2024_0_2.py index 1d916dac7c70e1d5f070f07eaedd719d84db5083..48fcfd2d5df5ed9901b5af6123c41314720285cc 100644 --- a/ohmpi/hardware_components/mb_2024_0_2.py +++ b/ohmpi/hardware_components/mb_2024_0_2.py @@ -80,6 +80,29 @@ class Tx(Tx_mb_2023): Tx_mb_2023.inject(self, polarity=polarity, injection_duration=injection_duration) self.pin6.value = False + @property + def polarity(self): + return self._polarity + + @polarity.setter + def polarity(self, polarity): + print(polarity) + assert polarity in [-1, 0, 1] + print(f'asserted polarity: {self.polarity}') + self._polarity = polarity + if polarity == 1: + self.pin0.value = True + self.pin1.value = False + time.sleep(self._activation_delay) + elif polarity == -1: + self.pin0.value = False + self.pin1.value = True + time.sleep(self._activation_delay) + else: + self.pin0.value = False + self.pin1.value = False + time.sleep(self._release_delay) + class Rx(Rx_mb_2023): def __init__(self, **kwargs):