decoders/ade7758/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 = 2
  30     id = 'ade7758'
  31     name = 'ADE7758'
  32     longname = 'Analog Devices ADE7758'
  33     desc = 'Poly phase multifunction energy metering IC protocol.'
  34     license = 'mit'
  35     inputs = ['spi']
  36     outputs = ['ade7758']
  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(self):
  49         self.expected = 0
  50         self.mosi_bytes = []
  51         self.miso_bytes = []
  52
  53     def __init__(self):
  54         self.ss_cmd, self.es_cmd = 0, 0
  55         self.reset()
  56
  57     def start(self):
  58         self.out_ann = self.register(srd.OUTPUT_ANN)
  59
  60     def putx(self, data):
  61         self.put(self.ss_cmd, self.es_cmd, self.out_ann, data)
  62
  63     def put_warn(self, pos, msg):
  64         self.put(pos[0], pos[1], self.out_ann, [2, [msg]])
  65
  66     def decode(self, ss, es, data):
  67         ptype = data[0]
  68         if ptype == 'CS-CHANGE':
  69             # Bear in mind, that CS is optional according to the datasheet.
  70             # If we transition high mid-stream, toss out our data and restart.
  71             cs_old, cs_new = data[1:]
  72             if cs_old is not None and cs_old == 0 and cs_new == 1:
  73                 if len(self.mosi_bytes) > 0 and len(self.mosi_bytes[1:]) < self.expected:
  74                     # Mark short read/write for reg at least!
  75                     self.es_cmd = es
  76                     write = self.cmd & 0x80
  77                     reg = self.cmd & 0x7f
  78                     rblob = regs.get(reg)
  79                     if write:
  80                         self.putx([1, ['%s: %s' % (rblob[0], "SHORT")]])
  81                     else:
  82                         self.putx([0, ['%s: %s' % (rblob[0], "SHORT")]])
  83
  84                     self.put_warn([self.ss_cmd, es], "Short transfer!")
  85                 self.reset()
  86             return
  87
  88         # Don't care about anything else.
  89         if ptype != 'DATA':
  90             return
  91         mosi, miso = data[1:]
  92
  93         if len(self.mosi_bytes) == 0:
  94             self.ss_cmd = ss
  95         self.mosi_bytes.append(mosi)
  96         self.miso_bytes.append(miso)
  97
  98         # A transfer is 2-4 bytes, (command + 1..3 byte reg).
  99         if len(self.mosi_bytes) < 2:
 100             return
 101
 102         self.cmd = self.mosi_bytes[0]
 103         write = self.cmd & 0x80
 104         reg = self.cmd & 0x7f
 105         rblob = regs.get(reg)
 106         if not rblob:
 107             # If you don't have CS, this will _destroy_ comms!
 108             self.put_warn([self.ss_cmd, es], 'Unknown register!')
 109             return
 110
 111         self.expected = math.ceil(rblob[3] / 8)
 112         if len(self.mosi_bytes[1:]) != self.expected:
 113             return
 114         valo = None
 115         vali = None
 116         self.es_cmd = es
 117         if self.expected == 3:
 118             valo = self.mosi_bytes[1] << 16 | self.mosi_bytes[2] << 8 | \
 119                    self.mosi_bytes[3]
 120             vali = self.miso_bytes[1] << 16 | self.miso_bytes[2] << 8 | \
 121                    self.miso_bytes[3]
 122         elif self.expected == 2:
 123             valo = self.mosi_bytes[1] << 8 | self.mosi_bytes[2]
 124             vali = self.miso_bytes[1] << 8 | self.miso_bytes[2]
 125         elif self.expected == 1:
 126             valo = self.mosi_bytes[1]
 127             vali = self.miso_bytes[1]
 128
 129         if write:
 130             self.putx([1, ['%s: %#x' % (rblob[0], valo)]])
 131         else:
 132             self.putx([0, ['%s: %#x' % (rblob[0], vali)]])
 133         self.reset()