From 1d46ad771e2d4fc9a265d449a37edd43b3c80d8e Mon Sep 17 00:00:00 2001
From: awatlet <arnaud.watlet@umons.ac.be>
Date: Mon, 9 Oct 2023 15:45:45 +0200
Subject: [PATCH] test pulse

---
 ohmpi/hardware_components/mb_2024_0_2.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/ohmpi/hardware_components/mb_2024_0_2.py b/ohmpi/hardware_components/mb_2024_0_2.py
index 1d916dac..48fcfd2d 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):
-- 
GitLab