ShiftRegister.h

#pragma once
#include <Bela.h> // this class acts on the BelaContext directly
#include <vector>

class ShiftRegister
{
public:
        struct Pins {
                unsigned int data; 
                unsigned int clock; 
                unsigned int latch; 
        };

        ShiftRegister();
        ShiftRegister(const Pins& pins, unsigned int maxSize);
         void setup(const Pins& pins, unsigned int maxSize);
        bool dataReady();
        virtual void process(BelaContext* context, unsigned int n) = 0;
protected:
        Pins pins;
        typedef enum {
                kStart,
                kTransmitting,
                kStop,
                kIdle,
        } State;
 
        State state = kStop;
        std::vector<bool> data;
        bool pinModeSet = false;
        bool notified = true;
};
 
class ShiftRegisterOut : public ShiftRegister
{
public:
        void setClockPeriod(unsigned int period);
        void setData(const std::vector<bool>& dataBuf);
        void setData(const bool* dataBuf, unsigned int length);
        void process(BelaContext* context);
        void process(BelaContext* context, unsigned int n) override;
private:
        unsigned int period = 2;
        unsigned int currentDataFrame;
        unsigned int currentStopFrame;
};
 
class ShiftRegisterIn : public ShiftRegister
{
public:
        bool dataReceived();
        const std::vector<bool>& getData();
        void process(BelaContext* context, unsigned int n) override;
private:
        bool pastLatch = false;
        bool pastClock = false;
};