Update _PMBus.py

### Description - add arg to allow the PHASE to not be included
slaclab · Nov 4, 2024 · 91997a8 · 91997a8
1 parent 266afd6
commit 91997a8
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/python/surf/protocols/i2c/_PMBus.py b/python/surf/protocols/i2c/_PMBus.py
@@ -11,7 +11,7 @@
 import pyrogue as pr
 
 class PMBus(pr.Device):
-    def __init__(self, simpleDisplay = True, dynamicAddr=False, **kwargs):
+    def __init__(self, simpleDisplay = True, includePhase = True, dynamicAddr=False, **kwargs):
         super().__init__(**kwargs)
 
         self.add(pr.RemoteVariable(
@@ -73,13 +73,14 @@ def __init__(self, simpleDisplay = True, dynamicAddr=False, **kwargs):
             hidden       = simpleDisplay,
         ))
 
-        self.add(pr.RemoteVariable(
-            name         = 'PHASE',
-            offset       = (4*0x04),
-            bitSize      = 8,
-            mode         = 'RW',
-            hidden       = simpleDisplay,
-        ))
+        if includePhase:
+            self.add(pr.RemoteVariable(
+                name         = 'PHASE',
+                offset       = (4*0x04),
+                bitSize      = 8,
+                mode         = 'RW',
+                hidden       = simpleDisplay,
+            ))
 
         self.add(pr.RemoteVariable(
             name         = 'WRITE_PROTECT',