sda2506: Use SrdIntEnum for pins.

[libsigrokdecode.git] / decoders / sda2506 / pd.py

diff --git a/decoders/sda2506/pd.py b/decoders/sda2506/pd.py
index 5b071d88d57d3a3cadaa6a4d4bb949403f396373..6df57554f69f0176facddd0d06682616bfcdfef7 100644 (file)
--- a/decoders/sda2506/pd.py
+++ b/decoders/sda2506/pd.py
@@ -19,6 +19,9 @@
 
 import re
 import sigrokdecode as srd
 
 import re
 import sigrokdecode as srd

+from common.srdhelper import SrdIntEnum
+
+Pin = SrdIntEnum.from_str('Pin', 'CLK DATA CE')

 
 ann_cmdbit, ann_databit, ann_cmd, ann_data, ann_warning = range(5)
 
 
 ann_cmdbit, ann_databit, ann_cmd, ann_data, ann_warning = range(5)
 


@@ -88,8 +91,8 @@ class Decoder(srd.Decoder):
 
     def decode(self):
         while True:
 
     def decode(self):
         while True:

-            # Wait for CLK edge or CE edge.
-            clk, d, ce = self.wait([{0: 'e'}, {2: 'e'}])
+            # Wait for CLK edge or CE# edge.
+            clk, d, ce = self.wait([{Pin.CLK: 'e'}, {Pin.CE: 'e'}])

 
             if self.matched[0] and ce == 1 and clk == 1:
                 # Rising clk edge and command mode.
 
             if self.matched[0] and ce == 1 and clk == 1:
                 # Rising clk edge and command mode.