do not allow smf_source's reads to stomp on cached read_end position in parent class...
[ardour.git] / libs / ardour / ardour / midi_ring_buffer.h
index aaad5769a32a20f3182f2ce4f2359696acd131da..f879aa75341e668fa4c4cd335e2adbfd8a4ca27d 100644 (file)
 
 #include <iostream>
 #include <algorithm>
-#include <ardour/types.h>
-#include <ardour/buffer.h>
+#include "ardour/types.h"
+#include "ardour/buffer.h"
+#include "evoral/EventRingBuffer.hpp"
 
 namespace ARDOUR {
 
+class MidiBuffer;
 
-/* FIXME: this is probably too much inlined code */
-
-
-/** A RingBuffer.
- * Read/Write realtime safe.
- * Single-reader Single-writer thread safe.
- *
- * This is Raul::RingBuffer, lifted for MIDIRingBuffer to inherit from as it works
- * a bit differently than PBD::Ringbuffer.  This could/should be replaced with
- * the PBD ringbuffer to decrease code size, but this code is tested and known to
- * work, so here it sits for now...
- *
- * Ignore this class, use MidiRingBuffer.
- */
-template <typename T>
-class MidiRingBufferBase {
-public:
-
-       /** @param size Size in bytes.
-        */
-       MidiRingBufferBase(size_t size)
-               : _size(size)
-               , _buf(new T[size])
-       {
-               reset();
-               assert(read_space() == 0);
-               assert(write_space() == size - 1);
-       }
-       
-       virtual ~MidiRingBufferBase() {
-               delete[] _buf;
-       }
-
-       /** Reset(empty) the ringbuffer.
-        * NOT thread safe.
-        */
-       void reset() {
-               g_atomic_int_set(&_write_ptr, 0);
-               g_atomic_int_set(&_read_ptr, 0);
-       }
-
-       size_t write_space() const {
-               
-               const size_t w = g_atomic_int_get(&_write_ptr);
-               const size_t r = g_atomic_int_get(&_read_ptr);
-               
-               if (w > r) {
-                       return ((r - w + _size) % _size) - 1;
-               } else if(w < r) {
-                       return (r - w) - 1;
-               } else {
-                       return _size - 1;
-               }
-       }
-       
-       size_t read_space() const {
-               
-               const size_t w = g_atomic_int_get(&_write_ptr);
-               const size_t r = g_atomic_int_get(&_read_ptr);
-               
-               if (w > r) {
-                       return w - r;
-               } else {
-                       return (w - r + _size) % _size;
-               }
-       }
-
-       size_t capacity() const { return _size; }
-
-       size_t peek(size_t size, T* dst);
-       bool   full_peek(size_t size, T* dst);
-
-       size_t read(size_t size, T* dst);
-       bool   full_read(size_t size, T* dst);
-       
-       void   write(size_t size, const T* src);
-
-protected:
-       mutable gint _write_ptr;
-       mutable gint _read_ptr;
-       
-       size_t _size; ///< Size (capacity) in bytes
-       T*  _buf;  ///< size, event, size, event...
-};
-
-
-/** Peek at the ringbuffer (read w/o advancing read pointer).
+/** A RingBuffer for (MIDI) events.
  *
- * Note that a full read may not be done if the data wraps around.
- * Caller must check return value and call again if necessary, or use the 
- * full_peek method which does this automatically.
- */
-template<typename T>
-size_t
-MidiRingBufferBase<T>::peek(size_t size, T* dst)
-{
-       const size_t priv_read_ptr = g_atomic_int_get(&_read_ptr);
-
-       const size_t read_size = (priv_read_ptr + size < _size)
-                       ? size
-                       : _size - priv_read_ptr;
-       
-       memcpy(dst, &_buf[priv_read_ptr], read_size);
-        
-       return read_size;
-}
-
-
-template<typename T>
-bool
-MidiRingBufferBase<T>::full_peek(size_t size, T* dst)
-{
-       if (read_space() < size)
-               return false;
-
-       const size_t read_size = peek(size, dst);
-       
-       if (read_size < size)
-               peek(size - read_size, dst + read_size);
-
-       return true;
-}
-
-
-/** Read from the ringbuffer.
+ * This is simply a wrapper around a raw ringbuffer which writes/reads events
+ * as flat placked blobs.
+ * The buffer looks like this:
  *
- * Note that a full read may not be done if the data wraps around.
- * Caller must check return value and call again if necessary, or use the 
- * full_read method which does this automatically.
+ * [timestamp][type][size][size bytes of raw MIDI][timestamp][type][size](etc...)
  */
 template<typename T>
-size_t
-MidiRingBufferBase<T>::read(size_t size, T* dst)
-{
-       const size_t priv_read_ptr = g_atomic_int_get(&_read_ptr);
-
-       const size_t read_size = (priv_read_ptr + size < _size)
-                       ? size
-                       : _size - priv_read_ptr;
-       
-       memcpy(dst, &_buf[priv_read_ptr], read_size);
-        
-       g_atomic_int_set(&_read_ptr, (priv_read_ptr + read_size) % _size);
-
-       return read_size;
-}
-
-
-template<typename T>
-bool
-MidiRingBufferBase<T>::full_read(size_t size, T* dst)
-{
-       if (read_space() < size)
-               return false;
-
-       const size_t read_size = read(size, dst);
-       
-       if (read_size < size)
-               read(size - read_size, dst + read_size);
-
-       return true;
-}
-
-
-template<typename T>
-inline void
-MidiRingBufferBase<T>::write(size_t size, const T* src)
-{
-       const size_t priv_write_ptr = g_atomic_int_get(&_write_ptr);
-       
-       if (priv_write_ptr + size <= _size) {
-               memcpy(&_buf[priv_write_ptr], src, size);
-        g_atomic_int_set(&_write_ptr, (priv_write_ptr + size) % _size);
-       } else {
-               const size_t this_size = _size - priv_write_ptr;
-               assert(this_size < size);
-               assert(priv_write_ptr + this_size <= _size);
-               memcpy(&_buf[priv_write_ptr], src, this_size);
-               memcpy(&_buf[0], src+this_size, size - this_size);
-        g_atomic_int_set(&_write_ptr, size - this_size);
-       }
-}
-
-
-/* ******************************************************************** */
-       
-
-/** A MIDI RingBuffer.
- *
- * This is timestamps and MIDI packed sequentially into a single buffer, similarly
- * to LV2 MIDI.  The buffer looks like this:
- *
- * [timestamp][size][size bytes of raw MIDI][timestamp][size][etc..]
- */
-class MidiRingBuffer : public MidiRingBufferBase<Byte> {
+class MidiRingBuffer : public Evoral::EventRingBuffer<T> {
 public:
-
        /** @param size Size in bytes.
         */
        MidiRingBuffer(size_t size)
-               : MidiRingBufferBase<Byte>(size), _channel_mask(0xFFFF), _force_channel(-1)
+               : Evoral::EventRingBuffer<T>(size)
+               , _channel_mask(0x0000FFFF)
        {}
 
-       size_t write(double time, size_t size, const Byte* buf);
-       bool   read(double* time, size_t* size, Byte* buf);
-
-       bool   read_prefix(double* time, size_t* size);
-       bool   read_contents(size_t size, Byte* buf);
+       inline bool read_prefix(T* time, Evoral::EventType* type, uint32_t* size);
+       inline bool read_contents(uint32_t size, uint8_t* buf);
 
        size_t read(MidiBuffer& dst, nframes_t start, nframes_t end, nframes_t offset=0);
-       
-       /**
-        * @param channel_mask each bit in channel_mask represents a midi channel: bit 0 = channel 0,
-        *                     bit 1 = channel 1 etc. the read and write methods will only allow
-        *                     events to pass, whose channel bit is 1.
+       void dump(std::ostream& dst);
+       
+       /** Set the channel filtering mode.
+        * @param mask If mode is FilterChannels, each bit represents a midi channel:
+        *     bit 0 = channel 0, bit 1 = channel 1 etc. the read and write methods will only
+        *     process events whose channel bit is 1.
+        *     If mode is ForceChannel, mask is simply a channel number which all events will
+        *     be forced to while reading.
         */
-       void      set_channel_mask(uint16_t channel_mask) { g_atomic_int_set(&_channel_mask, channel_mask); }
-       uint16_t  get_channel_mask() { return g_atomic_int_get(&_channel_mask); }
+       void set_channel_mode(ChannelMode mode, uint16_t mask) {
+               g_atomic_int_set(&_channel_mask, (uint32_t(mode) << 16) | uint32_t(mask));
+       }
+
+       ChannelMode get_channel_mode() const {
+               return static_cast<ChannelMode>((g_atomic_int_get(&_channel_mask) & 0xFFFF0000) >> 16);
+       }
        
-       /**
-        * @param channel if negative, forcing channels is deactivated and filtering channels
-        *                is activated, if positive, the LSB of channel is the channel number
-        *                of the channel all events are forced into and filtering is deactivated
-        */
-       void      set_force_channel(int8_t channel) { g_atomic_int_set(&_force_channel, channel); }
-       int8_t    get_force_channel() { return g_atomic_int_get(&_force_channel); }
+       uint16_t get_channel_mask() const {
+               return g_atomic_int_get(&_channel_mask) & 0x0000FFFF;
+       }
        
 protected:
-       inline bool is_channel_event(Byte event_type_byte) {
+       inline bool is_channel_event(uint8_t event_type_byte) {
                // mask out channel information
                event_type_byte &= 0xF0;
                // midi channel events range from 0x80 to 0xE0
@@ -263,171 +81,35 @@ protected:
        }
        
 private:
-       volatile uint16_t _channel_mask;
-       volatile int8_t   _force_channel;
+       volatile uint32_t _channel_mask; // 16 bits mode, 16 bits mask
 };
 
 
-inline bool
-MidiRingBuffer::read(double* time, size_t* size, Byte* buf)
-{
-       bool success = MidiRingBufferBase<Byte>::full_read(sizeof(double), (Byte*)time);
-       if (success)
-               success = MidiRingBufferBase<Byte>::full_read(sizeof(size_t), (Byte*)size);
-       if (success)
-               success = MidiRingBufferBase<Byte>::full_read(*size, buf);
-
-       return success;
-}
-
-
 /** Read the time and size of an event.  This call MUST be immediately proceeded
- * by a call to read_contents (or the read pointer will be garabage).
+ * by a call to read_contents (or the read pointer will be garbage).
  */
+template<typename T>
 inline bool
-MidiRingBuffer::read_prefix(double* time, size_t* size)
+MidiRingBuffer<T>::read_prefix(T* time, Evoral::EventType* type, uint32_t* size)
 {
-       bool success = MidiRingBufferBase<Byte>::full_read(sizeof(double), (Byte*)time);
+       bool success = Evoral::EventRingBuffer<T>::full_read(sizeof(T), (uint8_t*)time);
        if (success)
-               success = MidiRingBufferBase<Byte>::full_read(sizeof(size_t), (Byte*)size);
+               success = Evoral::EventRingBuffer<T>::full_read(sizeof(Evoral::EventType), (uint8_t*)type);
+       if (success)
+               success = Evoral::EventRingBuffer<T>::full_read(sizeof(uint32_t), (uint8_t*)size);
 
        return success;
 }
 
 
-/** Read the contenst of an event.  This call MUST be immediately preceeded
- * by a call to read_prefix (or the returned even will be garabage).
+/** Read the content of an event.  This call MUST be immediately preceded
+ * by a call to read_prefix (or the returned even will be garbage).
  */
+template<typename T>
 inline bool
-MidiRingBuffer::read_contents(size_t size, Byte* buf)
-{
-       return MidiRingBufferBase<Byte>::full_read(size, buf);
-}
-
-
-inline size_t
-MidiRingBuffer::write(double time, size_t size, const Byte* buf)
+MidiRingBuffer<T>::read_contents(uint32_t size, uint8_t* buf)
 {
-       //printf("MRB - write %#X %d %d with time %lf\n",
-       //              buf[0], buf[1], buf[2], time);
-       
-       assert(size > 0);
-       
-       if(is_channel_event(buf[0]) && (g_atomic_int_get(&_force_channel) < 0)) {
-               // filter events for channels
-               Byte channel_nr = buf[0] & 0x0F;
-               if( !(g_atomic_int_get(&_channel_mask) & (1L << channel_nr)) )  {
-                       return 0;
-               }
-       }
-
-       if (write_space() < (sizeof(double) + sizeof(size_t) + size)) {
-               return 0;
-       } else {
-               MidiRingBufferBase<Byte>::write(sizeof(double), (Byte*)&time);
-               MidiRingBufferBase<Byte>::write(sizeof(size_t), (Byte*)&size);
-               if(is_channel_event(buf[0]) && (g_atomic_int_get(&_force_channel) >= 0)) {
-                       assert(size == 3);
-                       Byte tmp_buf[3];
-                       //force event into channel
-                       tmp_buf[0] = (buf[0] & 0xF0) | (g_atomic_int_get(&_force_channel) & 0x0F);
-                       tmp_buf[1] = buf[1];
-                       tmp_buf[2] = buf[2];
-                       MidiRingBufferBase<Byte>::write(size, tmp_buf);
-               } else {
-                       MidiRingBufferBase<Byte>::write(size, buf);
-               }
-               return size;
-       }
-
-}
-
-
-/** Read a block of MIDI events from buffer.
- *
- * Timestamps of events returned are relative to start (ie event with stamp 0
- * occurred at start), with offset added.
- */
-inline size_t
-MidiRingBuffer::read(MidiBuffer& dst, nframes_t start, nframes_t end, nframes_t offset)
-{
-       if (read_space() == 0)
-               return 0;
-
-       MIDI::Event ev;
-
-       size_t count = 0;
-
-       //printf("MRB - read %u .. %u + %u\n", start, end, offset);
-
-       while (read_space() > sizeof(double) + sizeof(size_t)) {
-       
-               full_peek(sizeof(double), (Byte*)&ev.time());
-       
-               if (ev.time() > end)
-                       break;
-               
-               bool success = MidiRingBufferBase<Byte>::full_read(sizeof(double), (Byte*)&ev.time());
-               if (success) {
-                       success = MidiRingBufferBase<Byte>::full_read(sizeof(size_t), (Byte*)&ev.size());
-               }
-
-               if (!success) {
-                       std::cerr << "MRB: READ ERROR (time/size)" << std::endl;
-                       continue;
-               }
-               
-               Byte first_event_byte;
-               if(success) {
-                       success = full_peek(sizeof(Byte), &first_event_byte);
-               }
-                               
-               // could this ever happen???
-               if (!success) {
-                       std::cerr << "MRB: PEEK ERROR (first event byte)" << std::endl;
-                       continue;
-               }
-               
-               // filter events for channels
-               // filtering is only active, if forcing channels is not active
-               if(is_channel_event(first_event_byte) && (g_atomic_int_get(&_force_channel) < 0)) {
-                       Byte channel_nr = first_event_byte & 0x0F;
-                       if( !(g_atomic_int_get(&_channel_mask) & (1L << channel_nr)) )  {
-                               return 0;
-                       }
-               }
-
-               if (ev.time() >= start) {
-                       ev.time() -= start;
-                       // TODO: Right now there come MIDI Events with empty buffer
-                       if(!ev.buffer()) {
-                               std::cerr << "MidiRingBuffer::read WARNING: Skipping MIDI Event with NULL buffer pointer " 
-                                    << " and length " << int(ev.size()) << std::endl; 
-                               return 0;
-                       }
-                       
-                       Byte* write_loc = dst.reserve(ev.time(), ev.size());
-                       
-                       success = MidiRingBufferBase<Byte>::full_read(ev.size(), write_loc);
-               
-                       if (success) {
-                               if(is_channel_event(first_event_byte) && (g_atomic_int_get(&_force_channel) >= 0)) {
-                                       write_loc[0] = (write_loc[0] & 0xF0) | (g_atomic_int_get(&_force_channel) & 0x0F);
-                               }
-                               ++count;
-                               //printf("MRB - read event at time %lf\n", ev.time());
-                       } else {
-                               std::cerr << "MRB: READ ERROR (data)" << std::endl;
-                       }
-                       
-               } else {
-                       printf("MRB - SKIPPING EVENT AT TIME %f\n", ev.time());
-               }
-       }
-       
-       //printf("(R) read space: %zu\n", read_space());
-
-       return count;
+       return Evoral::EventRingBuffer<T>::full_read(size, buf);
 }