counter: add support for user specified initial counter values

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

diff --git a/decoders/counter/pd.py b/decoders/counter/pd.py
index bb55fa7d060ae4efdaa533ce6630b6176f6aadfa..b8811e589bdf89e58ae8748435b536016f490be9 100644 (file)
--- a/decoders/counter/pd.py
+++ b/decoders/counter/pd.py
@@ -19,8 +19,8 @@
 
 import sigrokdecode as srd
 
-(PIN_DATA, PIN_RESET) = range(2)
-(ROW_EDGE, ROW_WORD, ROW_RESET) = range(3)
+PIN_DATA, PIN_RESET = range(2)
+ROW_EDGE, ROW_WORD, ROW_RESET = range(3)
 
 class Decoder(srd.Decoder):
     api_version = 3
@@ -53,6 +53,8 @@ class Decoder(srd.Decoder):
         {'id': 'divider', 'desc': 'Count divider (word width)', 'default': 0},
         {'id': 'reset_edge', 'desc': 'Edge which clears counters (reset)',
             'default': 'falling', 'values': ('rising', 'falling')},
+        {'id': 'edge_off', 'desc': 'Initial edge counter value', 'default': 0},
+        {'id': 'word_off', 'desc': 'Initial word counter value', 'default': 0},
     )
 
     def __init__(self):
@@ -86,9 +88,9 @@ class Decoder(srd.Decoder):
             cond_reset = len(condition)
             condition.append({PIN_RESET: opt_edge_map[reset_edge]})
 
-        edge_count = 0
+        edge_count = int(self.options['edge_off'])
         edge_start = None
-        word_count = 0
+        word_count = int(self.options['word_off'])
         word_start = None
         while True:
             self.wait(condition)