decoders/timing/pd.py

   1 ##
   2 ## This file is part of the libsigrokdecode project.
   3 ##
   4 ## Copyright (C) 2014 Torsten Duwe <duwe@suse.de>
   5 ## Copyright (C) 2014 Sebastien Bourdelin <sebastien.bourdelin@savoirfairelinux.com>
   6 ##
   7 ## This program is free software; you can redistribute it and/or modify
   8 ## it under the terms of the GNU General Public License as published by
   9 ## the Free Software Foundation; either version 2 of the License, or
  10 ## (at your option) any later version.
  11 ##
  12 ## This program is distributed in the hope that it will be useful,
  13 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 ## GNU General Public License for more details.
  16 ##
  17 ## You should have received a copy of the GNU General Public License
  18 ## along with this program; if not, see <http://www.gnu.org/licenses/>.
  19 ##
  20
  21 import sigrokdecode as srd
  22 from collections import deque
  23
  24 class SamplerateError(Exception):
  25     pass
  26
  27 def normalize_time(t):
  28     if abs(t) >= 1.0:
  29         return '%.3f s  (%.3f Hz)' % (t, (1/t))
  30     elif abs(t) >= 0.001:
  31         if 1/t/1000 < 1:
  32             return '%.3f ms (%.3f Hz)' % (t * 1000.0, (1/t))
  33         else:
  34             return '%.3f ms (%.3f kHz)' % (t * 1000.0, (1/t)/1000)
  35     elif abs(t) >= 0.000001:
  36         if 1/t/1000/1000 < 1:
  37             return '%.3f μs (%.3f kHz)' % (t * 1000.0 * 1000.0, (1/t)/1000)
  38         else:
  39             return '%.3f μs (%.3f MHz)' % (t * 1000.0 * 1000.0, (1/t)/1000/1000)
  40     elif abs(t) >= 0.000000001:
  41         if 1/t/1000/1000/1000:
  42             return '%.3f ns (%.3f MHz)' % (t * 1000.0 * 1000.0 * 1000.0, (1/t)/1000/1000)
  43         else:
  44             return '%.3f ns (%.3f GHz)' % (t * 1000.0 * 1000.0 * 1000.0, (1/t)/1000/1000/1000)
  45     else:
  46         return '%f' % t
  47
  48 def terse_times(t):
  49     if abs(t) >= 1e0:
  50         t *= 1e0
  51         return ['{:.0f}s'.format(t), '{:.0f}'.format(t)]
  52     if abs(t) >= 1e-3:
  53         t *= 1e3
  54         return ['{:.0f}ms'.format(t), '{:.0f}'.format(t)]
  55     if abs(t) >= 1e-6:
  56         t *= 1e6
  57         return ['{:.0f}us'.format(t), '{:.0f}'.format(t)]
  58     if abs(t) >= 1e-9:
  59         t *= 1e9
  60         return ['{:.0f}ns'.format(t), '{:.0f}'.format(t)]
  61     return ['{:f}'.format(t = t)]
  62
  63 class Pin:
  64     (DATA,) = range(1)
  65
  66 class Ann:
  67     (TIME, TERSE, AVG, DELTA,) = range(4)
  68
  69 class Decoder(srd.Decoder):
  70     api_version = 3
  71     id = 'timing'
  72     name = 'Timing'
  73     longname = 'Timing calculation with frequency and averaging'
  74     desc = 'Calculate time between edges.'
  75     license = 'gplv2+'
  76     inputs = ['logic']
  77     outputs = []
  78     tags = ['Clock/timing', 'Util']
  79     channels = (
  80         {'id': 'data', 'name': 'Data', 'desc': 'Data line'},
  81     )
  82     annotations = (
  83         ('time', 'Time'),
  84         ('terse', 'Terse'),
  85         ('average', 'Average'),
  86         ('delta', 'Delta'),
  87     )
  88     annotation_rows = (
  89         ('times', 'Times', (Ann.TIME, Ann.TERSE,)),
  90         ('averages', 'Averages', (Ann.AVG,)),
  91         ('deltas', 'Deltas', (Ann.DELTA,)),
  92     )
  93     options = (
  94         { 'id': 'avg_period', 'desc': 'Averaging period', 'default': 100 },
  95         { 'id': 'edge', 'desc': 'Edges to check', 'default': 'any', 'values': ('any', 'rising', 'falling') },
  96         { 'id': 'delta', 'desc': 'Show delta from last', 'default': 'no', 'values': ('yes', 'no') },
  97         { 'id': 'terse', 'desc': 'Show periods in terse format', 'default': 'no', 'values': ('yes', 'no') },
  98     )
  99
 100     def __init__(self):
 101         self.reset()
 102
 103     def reset(self):
 104         self.samplerate = None
 105
 106     def metadata(self, key, value):
 107         if key == srd.SRD_CONF_SAMPLERATE:
 108             self.samplerate = value
 109
 110     def start(self):
 111         self.out_ann = self.register(srd.OUTPUT_ANN)
 112
 113     def decode(self):
 114         if not self.samplerate:
 115             raise SamplerateError('Cannot decode without samplerate.')
 116         edge = self.options['edge']
 117         avg_period = self.options['avg_period']
 118         terse = self.options['terse'] == 'yes'
 119         ss = None
 120         last_n = deque()
 121         last_t = None
 122         while True:
 123             if edge == 'rising':
 124                 pin = self.wait({Pin.DATA: 'r'})
 125             elif edge == 'falling':
 126                 pin = self.wait({Pin.DATA: 'f'})
 127             else:
 128                 pin = self.wait({Pin.DATA: 'e'})
 129
 130             if not ss:
 131                 ss = self.samplenum
 132                 continue
 133             es = self.samplenum
 134             samples = es - ss
 135             t = samples / self.samplerate
 136
 137             if t > 0:
 138                 last_n.append(t)
 139             if len(last_n) > avg_period:
 140                 last_n.popleft()
 141
 142             if terse:
 143                 self.put(ss, es, self.out_ann, [Ann.TERSE, terse_times(t)])
 144             else:
 145                 self.put(ss, es, self.out_ann, [Ann.TIME, [normalize_time(t)]])
 146             if avg_period > 0:
 147                 self.put(ss, es, self.out_ann,
 148                          [Ann.AVG, [normalize_time(sum(last_n) / len(last_n))]])
 149             if last_t and self.options['delta'] == 'yes':
 150                 self.put(ss, es, self.out_ann,
 151                          [Ann.DELTA, [normalize_time(t - last_t)]])
 152
 153             last_t = t
 154             ss = es