JackNewman12/pd.py

## Binary file added pulseview.png

    
          Binary file added
          
            BIN
              
                +45.2 KB

              
  pulseview.png

  
              Show comments
            

   View file


      Loading

      Sorry, something went wrong. Reload?
      Sorry, we cannot display this file.
      Sorry, this file is invalid so it cannot be displayed.
      
          Viewer requires iframe.
      
    
## 73 changes: 73 additions & 0 deletions pd.py
@@ -0,0 +1,73 @@

    import sigrokdecode as srd
import sigrokdecode as srd

    import collections
import collections


    # Matches to find.
# Matches to find.

    # A list of tuples that contain (ByteToMatch, Name)
# A list of tuples that contain (ByteToMatch, Name)

    Matches = [
Matches = [

        ([0x20, 0x20], "Match One!"),
    ([0x20, 0x20], "Match One!"),

        ([0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07], "Match Two!"),
    ([0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07], "Match Two!"),

        ([0xFF, 0xAA, 0xCC, 0x55], "Match Three!"),
    ([0xFF, 0xAA, 0xCC, 0x55], "Match Three!"),

    ]
]


    # Standard Decoder
# Standard Decoder

    class Decoder(srd.Decoder):
class Decoder(srd.Decoder):

        # Decoder stuff that needs to exist for pulseview to be happy
    # Decoder stuff that needs to exist for pulseview to be happy

        api_version = 3
    api_version = 3

        id = "ByteMatch"
    id = "ByteMatch"

        name = "Byte Matcher"
    name = "Byte Matcher"

        longname = "A Byte Matcher"
    longname = "A Byte Matcher"

        desc = "This decoder searches for arbitrary set of bytes"
    desc = "This decoder searches for arbitrary set of bytes"

        license = "MIT"
    license = "MIT"

        inputs = ["spi"]  # CHANGE THE DECODE YOU WANT TO STACK ONTOP OF
    inputs = ["spi"]  # CHANGE THE DECODE YOU WANT TO STACK ONTOP OF

        outputs = []
    outputs = []

        tags = ["IC", "Memory"]
    tags = ["IC", "Memory"]

        annotations = (("Matches", "Matches"),)
    annotations = (("Matches", "Matches"),)

        annotation_rows = (("Matches", "Matches", (0,)),)
    annotation_rows = (("Matches", "Matches", (0,)),)

        options = ()
    options = ()


        def __init__(self):
    def __init__(self):

            # Circular buffers for keeping track of the last few bytes
        # Circular buffers for keeping track of the last few bytes

            # Make the buffers the size of the largest match
        # Make the buffers the size of the largest match

            MaxLength = max([len(bytes) for bytes, name in Matches])
        MaxLength = max([len(bytes) for bytes, name in Matches])

            # The last few bytes
        # The last few bytes

            self.LastBytes = collections.deque(maxlen=MaxLength)
        self.LastBytes = collections.deque(maxlen=MaxLength)

            # The last few start-samples
        # The last few start-samples

            self.LastSamples = collections.deque(maxlen=MaxLength)
        self.LastSamples = collections.deque(maxlen=MaxLength)


        def reset(self):
    def reset(self):

            # Reset the circular buffers
        # Reset the circular buffers

            self.LastBytes.clear()
        self.LastBytes.clear()

            self.LastSamples.clear()
        self.LastSamples.clear()


        def start(self):
    def start(self):

            # Standard Pulseview Stuff
        # Standard Pulseview Stuff

            self.out_ann = self.register(srd.OUTPUT_ANN)
        self.out_ann = self.register(srd.OUTPUT_ANN)


        def decode(self, startSample, endSample, data):
    def decode(self, startSample, endSample, data):

            # Grab the data from your lower level decoder.
        # Grab the data from your lower level decoder.

            ptype, mosi, miso = data
        ptype, mosi, miso = data

            # In this case I only care about full bytes from the decoder (DATA).
        # In this case I only care about full bytes from the decoder (DATA).

            # Do not care about indiviual bits (BITS)
        # Do not care about indiviual bits (BITS)

            if ptype != "DATA":
        if ptype != "DATA":

                return
            return


            # Keep track of the byte we just saw and the sample it started at (for annotations)
        # Keep track of the byte we just saw and the sample it started at (for annotations)

            self.LastBytes.append(miso)
        self.LastBytes.append(miso)

            self.LastSamples.append(startSample)
        self.LastSamples.append(startSample)


            # Just search for a match in the array
        # Just search for a match in the array

            for match in Matches:
        for match in Matches:

                data, annotation = match
            data, annotation = match


                # If the last few bytes in the buffers match our target array
            # If the last few bytes in the buffers match our target array

                if data == list(self.LastBytes)[-1 * len(data) :]:
            if data == list(self.LastBytes)[-1 * len(data) :]:


                    startSample = (
                startSample = (

                        # Cover edge case where we match on the very first byte in the data
                    # Cover edge case where we match on the very first byte in the data

                        self.LastSamples[-1 * len(data)]
                    self.LastSamples[-1 * len(data)]

                        if len(self.LastSamples) >= len(data)
                    if len(self.LastSamples) >= len(data)

                        else 0
                    else 0

                    )
                )


                    # Print the annotation at orignal start sample and the current sample
                # Print the annotation at orignal start sample and the current sample

                    self.put(startSample, endSample, self.out_ann, [0, [annotation]])
                self.put(startSample, endSample, self.out_ann, [0, [annotation]])