new buffer thingy

core/src/dsp/buffer.h

@@ -214,4 +214,112 @@ namespace dsp {
         std::condition_variable canReadVar;
         std::condition_variable canWriteVar;
     };
+
+#define TEST_BUFFER_SIZE    32
+
+    template <class T>
+    class SampleFrameBuffer : public generic_block<SampleFrameBuffer<T>> {
+    public:
+        SampleFrameBuffer() {}
+
+        SampleFrameBuffer(stream<T>* in) { init(in); }
+
+        ~SampleFrameBuffer() {
+            stop();
+            out.stopWriter();
+            stopWorker = true;
+            cnd.notify_all();
+            if (readWorkerThread.joinable()) { readWorkerThread.join(); }
+        }
+
+        void init(stream<T>* in) {
+            _in = in;
+
+            for (int i = 0; i < TEST_BUFFER_SIZE; i++) {
+                buffers[i] = new T[STREAM_BUFFER_SIZE];
+            }
+
+            generic_block<SampleFrameBuffer<T>>::registerInput(in);
+            generic_block<SampleFrameBuffer<T>>::registerOutput(&out);
+
+            readWorkerThread = std::thread(&SampleFrameBuffer::worker, this);
+        }
+
+        void setInput(stream<T>* in) {
+            std::lock_guard<std::mutex> lck(generic_block<SampleFrameBuffer<T>>::ctrlMtx);
+            generic_block<SampleFrameBuffer<T>>::tempStop();
+            generic_block<SampleFrameBuffer<T>>::unregisterInput(_in);
+            _in = in;
+            generic_block<SampleFrameBuffer<T>>::registerInput(_in);
+            generic_block<SampleFrameBuffer<T>>::tempStart();
+        }
+
+        int run() {
+            // Wait for data
+            int count = _in->read();
+            if (count < 0) { return -1; }
+
+            if (bypass) {
+                memcpy(out.writeBuf, _in->readBuf, count * sizeof(T));
+                _in->flush();
+                if (!out.swap(count)) { return -1; }
+                return count;
+            }
+
+            // Push it on the ring buffer
+            {
+                std::lock_guard<std::mutex> lck(bufMtx);
+                memcpy(buffers[writeCur], _in->readBuf, count * sizeof(T));
+                uintptr_t ptr = (uintptr_t)buffers[writeCur];
+                sizes[writeCur] = count;
+                writeCur++;
+                writeCur = ((writeCur) % TEST_BUFFER_SIZE);
+
+                if (((writeCur - readCur + TEST_BUFFER_SIZE) % TEST_BUFFER_SIZE) >= (TEST_BUFFER_SIZE-2)) {
+                    spdlog::warn("Overflow");
+                }
+            }
+            cnd.notify_all();
+            _in->flush();
+            return count;
+        }
+
+        void worker() {
+            while (true) {
+                // Wait for data
+                std::unique_lock lck(bufMtx);
+                cnd.wait(lck, [this](){ return (((writeCur - readCur + TEST_BUFFER_SIZE) % TEST_BUFFER_SIZE) > 0) || stopWorker; });
+                if (stopWorker) { break; }
+
+                // Write one to output buffer and unlock in preparation to swap buffers
+                int count = sizes[readCur];
+                memcpy(out.writeBuf, buffers[readCur], count * sizeof(T));
+                readCur++;
+                readCur = ((readCur) % TEST_BUFFER_SIZE);
+                lck.unlock();
+
+                // Swap
+                if (!out.swap(count)) { break; }
+            } 
+        }
+
+        stream<T> out;
+
+        int writeCur = 0;
+        int readCur = 0;
+
+        bool bypass = false;
+
+    private:
+        stream<T>* _in;
+
+        std::thread readWorkerThread;
+        std::mutex bufMtx;
+        std::condition_variable cnd;
+        T* buffers[TEST_BUFFER_SIZE];
+        int sizes[TEST_BUFFER_SIZE];
+        
+        bool stopWorker = false;
+
+    };
 };