decoders/ade77xx/pd.py

   1 ##
   2 ## This file is part of the libsigrokdecode project.
   3 ##
   4 ## Copyright (C) 2017 Karl Palsson <karlp@etactica.com>
   5 ##
   6 ## Permission is hereby granted, free of charge, to any person obtaining a copy
   7 ## of this software and associated documentation files (the "Software"), to deal
   8 ## in the Software without restriction, including without limitation the rights
   9 ## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10 ## copies of the Software, and to permit persons to whom the Software is
  11 ## furnished to do so, subject to the following conditions:
  12 ##
  13 ## The above copyright notice and this permission notice shall be included in all
  14 ## copies or substantial portions of the Software.
  15 ##
  16 ## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17 ## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18 ## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19 ## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20 ## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21 ## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  22 ## SOFTWARE.
  23
  24 import math
  25 import sigrokdecode as srd
  26 from .lists import *
  27
  28 class Decoder(srd.Decoder):
  29     api_version = 3
  30     id = 'ade77xx'
  31     name = 'ADE77xx'
  32     longname = 'Analog Devices ADE77xx'
  33     desc = 'Poly phase multifunction energy metering IC protocol.'
  34     license = 'mit'
  35     inputs = ['spi']
  36     outputs = ['ade77xx']
  37     annotations = (
  38         ('read', 'Register read commands'),
  39         ('write', 'Register write commands'),
  40         ('warning', 'Warnings'),
  41     )
  42     annotation_rows = (
  43         ('read', 'Read', (0,)),
  44         ('write', 'Write', (1,)),
  45         ('warnings', 'Warnings', (2,)),
  46     )
  47
  48     def reset_data(self):
  49         self.expected = 0
  50         self.mosi_bytes, self.miso_bytes = [], []
  51
  52     def __init__(self):
  53         self.ss_cmd, self.es_cmd = 0, 0
  54         self.reset_data()
  55
  56     def start(self):
  57         self.out_ann = self.register(srd.OUTPUT_ANN)
  58
  59     def putx(self, data):
  60         self.put(self.ss_cmd, self.es_cmd, self.out_ann, data)
  61
  62     def put_warn(self, pos, msg):
  63         self.put(pos[0], pos[1], self.out_ann, [2, [msg]])
  64
  65     def decode(self, ss, es, data):
  66         ptype = data[0]
  67         if ptype == 'CS-CHANGE':
  68             # Bear in mind, that CS is optional according to the datasheet.
  69             # If we transition high mid-stream, toss out our data and restart.
  70             cs_old, cs_new = data[1:]
  71             if cs_old is not None and cs_old == 0 and cs_new == 1:
  72                 if len(self.mosi_bytes) > 0 and len(self.mosi_bytes[1:]) < self.expected:
  73                     # Mark short read/write for reg at least!
  74                     self.es_cmd = es
  75                     write, reg = self.cmd & 0x80, self.cmd & 0x7f
  76                     rblob = regs.get(reg)
  77                     idx = 1 if write else 0
  78                     self.putx([idx, ['%s: %s' % (rblob[0], "SHORT")]])
  79                     self.put_warn([self.ss_cmd, es], "Short transfer!")
  80                 self.reset_data()
  81             return
  82
  83         # Don't care about anything else.
  84         if ptype != 'DATA':
  85             return
  86         mosi, miso = data[1:]
  87
  88         if len(self.mosi_bytes) == 0:
  89             self.ss_cmd = ss
  90         self.mosi_bytes.append(mosi)
  91         self.miso_bytes.append(miso)
  92
  93         # A transfer is 2-4 bytes, (command + 1..3 byte reg).
  94         if len(self.mosi_bytes) < 2:
  95             return
  96
  97         self.cmd = self.mosi_bytes[0]
  98         write, reg = self.cmd & 0x80, self.cmd & 0x7f
  99         rblob = regs.get(reg)
 100         if not rblob:
 101             # If you don't have CS, this will _destroy_ comms!
 102             self.put_warn([self.ss_cmd, es], 'Unknown register!')
 103             return
 104
 105         self.expected = math.ceil(rblob[3] / 8)
 106         if len(self.mosi_bytes[1:]) != self.expected:
 107             return
 108         valo, vali = None, None
 109         self.es_cmd = es
 110         if self.expected == 3:
 111             valo = self.mosi_bytes[1] << 16 | self.mosi_bytes[2] << 8 | \
 112                    self.mosi_bytes[3]
 113             vali = self.miso_bytes[1] << 16 | self.miso_bytes[2] << 8 | \
 114                    self.miso_bytes[3]
 115         elif self.expected == 2:
 116             valo = self.mosi_bytes[1] << 8 | self.mosi_bytes[2]
 117             vali = self.miso_bytes[1] << 8 | self.miso_bytes[2]
 118         elif self.expected == 1:
 119             valo = self.mosi_bytes[1]
 120             vali = self.miso_bytes[1]
 121
 122         if write:
 123             self.putx([1, ['%s: %#x' % (rblob[0], valo)]])
 124         else:
 125             self.putx([0, ['%s: %#x' % (rblob[0], vali)]])
 126
 127         self.reset_data()