From daa8e9aa08c8a5ac2b4bb7f5f6b5c5fd3d467f46 Mon Sep 17 00:00:00 2001
From: Arnaud Watlet <arnaud.watlet@umons.ac.be>
Date: Tue, 12 Dec 2023 15:32:09 +0100
Subject: [PATCH] Tries enabling switch_pwr during injection for polarity 0 on
 pwr_batt and pwr_dps5005_fixed

---
 ohmpi/hardware_components/mb_2024_0_2.py | 6 +++---
 ohmpi/hardware_system.py                 | 2 ++
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/ohmpi/hardware_components/mb_2024_0_2.py b/ohmpi/hardware_components/mb_2024_0_2.py
index ce6b9312..deb8c593 100644
--- a/ohmpi/hardware_components/mb_2024_0_2.py
+++ b/ohmpi/hardware_components/mb_2024_0_2.py
@@ -138,19 +138,19 @@ class Tx(Tx_mb_2023):
         assert polarity in [-1, 0, 1]
         self._polarity = polarity
         if polarity == 1:
-            if self.pwr.voltage_adjustable:
+            if not self.pwr.voltage_adjustable:
                 self.pwr_state = 'on'
             self.pin0.value = True
             self.pin1.value = False
             time.sleep(self._activation_delay)
         elif polarity == -1:
-            if self.pwr.voltage_adjustable:
+            if not self.pwr.voltage_adjustable:
                 self.pwr_state = 'on'
             self.pin0.value = False
             self.pin1.value = True
             time.sleep(self._activation_delay)
         else:
-            if self.pwr.voltage_adjustable:
+            if not self.pwr.voltage_adjustable:
                 self.pwr_state = 'off'
             self.pin0.value = False
             self.pin1.value = False
diff --git a/ohmpi/hardware_system.py b/ohmpi/hardware_system.py
index b26dbf99..532026c3 100644
--- a/ohmpi/hardware_system.py
+++ b/ohmpi/hardware_system.py
@@ -137,6 +137,8 @@ class OhmPiHardware:
         if isinstance(self.tx, dict):
             self.tx = tx_module.Tx(**self.tx)
         self.tx.pwr = self.pwr
+        if not self.tx.pwr.voltage_adjustable:
+            self.tx._pwr_latency = 0
         self.tx.polarity = 0
         self.tx.pwr._current_max = current_max
 
-- 
GitLab