libs/ardour/ardour/buffer_set.h

   1 /*
   2     Copyright (C) 2006 Paul Davis
   3
   4     This program is free software; you can redistribute it and/or modify it
   5     under the terms of the GNU General Public License as published by the Free
   6     Software Foundation; either version 2 of the License, or (at your option)
   7     any later version.
   8
   9     This program is distributed in the hope that it will be useful, but WITHOUT
  10     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11     FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12     for more details.
  13
  14     You should have received a copy of the GNU General Public License along
  15     with this program; if not, write to the Free Software Foundation, Inc.,
  16     675 Mass Ave, Cambridge, MA 02139, USA.
  17 */
  18
  19 #ifndef __ardour_buffer_set_h__
  20 #define __ardour_buffer_set_h__
  21
  22 #include <cassert>
  23 #include <vector>
  24 #include <ardour/chan_count.h>
  25 #include <ardour/data_type.h>
  26
  27 namespace ARDOUR {
  28
  29 class Buffer;
  30 class AudioBuffer;
  31 class MidiBuffer;
  32 class PortSet;
  33
  34
  35 /** A set of buffers of various types.
  36  *
  37  * These are mainly accessed from Session and passed around as scratch buffers
  38  * (eg as parameters to run() methods) to do in-place signal processing.
  39  *
  40  * There are two types of counts associated with a BufferSet - available,
  41  * and the 'use count'.  Available is the actual number of allocated buffers
  42  * (and so is the maximum acceptable value for the use counts).
  43  *
  44  * The use counts are how things determine the form of their input and inform
  45  * others the form of their output (eg what they did to the BufferSet).
  46  * Setting the use counts is realtime safe.
  47  */
  48 class BufferSet
  49 {
  50 public:
  51         BufferSet();
  52         ~BufferSet();
  53
  54         void clear();
  55
  56         void attach_buffers(PortSet& ports);
  57
  58         void ensure_buffers(const ChanCount& count, size_t buffer_capacity);
  59         void ensure_buffers(DataType type, size_t num_buffers, size_t buffer_capacity);
  60
  61         const ChanCount& available() const { return _available; }
  62         ChanCount&       available()       { return _available; }
  63
  64         const ChanCount& count() const { return _count; }
  65         ChanCount&       count()       { return _count; }
  66
  67         void set_count(const ChanCount& count) { _count = count; }
  68
  69         size_t buffer_capacity(DataType type) const;
  70
  71         Buffer& get(DataType type, size_t i)
  72         {
  73                 assert(i <= _count.get(type));
  74                 return *_buffers[type.to_index()][i];
  75         }
  76
  77         AudioBuffer& get_audio(size_t i)
  78         {
  79                 return (AudioBuffer&)get(DataType::AUDIO, i);
  80         }
  81
  82         MidiBuffer& get_midi(size_t i)
  83         {
  84                 return (MidiBuffer&)get(DataType::MIDI, i);
  85         }
  86
  87         void read_from(BufferSet& in, jack_nframes_t nframes, jack_nframes_t offset=0)
  88         {
  89                 throw; // FIXME: implement this with spiffy DataType iterator etc.
  90         }
  91
  92         // ITERATORS
  93
  94         // FIXME: this is a filthy copy-and-paste mess
  95         // FIXME: litter these with assertions
  96
  97         class audio_iterator {
  98         public:
  99
 100                 AudioBuffer& operator*()  { return _set.get_audio(_index); }
 101                 AudioBuffer* operator->() { return &_set.get_audio(_index); }
 102                 audio_iterator& operator++() { ++_index; return *this; } // yes, prefix only
 103                 bool operator==(const audio_iterator& other) { return (_index == other._index); }
 104                 bool operator!=(const audio_iterator& other) { return (_index != other._index); }
 105
 106         private:
 107                 friend class BufferSet;
 108
 109                 audio_iterator(BufferSet& list, size_t index) : _set(list), _index(index) {}
 110
 111                 BufferSet& _set;
 112                 size_t    _index;
 113         };
 114
 115         audio_iterator audio_begin() { return audio_iterator(*this, 0); }
 116         audio_iterator audio_end()   { return audio_iterator(*this, _count.get(DataType::AUDIO)); }
 117
 118         class iterator {
 119         public:
 120
 121                 Buffer& operator*()  { return _set.get(_type, _index); }
 122                 Buffer* operator->() { return &_set.get(_type, _index); }
 123                 iterator& operator++() { ++_index; return *this; } // yes, prefix only
 124                 bool operator==(const iterator& other) { return (_index == other._index); }
 125                 bool operator!=(const iterator& other) { return (_index != other._index); }
 126
 127         private:
 128                 friend class BufferSet;
 129
 130                 iterator(BufferSet& list, DataType type, size_t index)
 131                         : _set(list), _type(type), _index(index) {}
 132
 133                 BufferSet& _set;
 134                 DataType   _type;
 135                 size_t     _index;
 136         };
 137
 138         iterator begin(DataType type) { return iterator(*this, type, 0); }
 139         iterator end(DataType type)   { return iterator(*this, type, _count.get(type)); }
 140
 141
 142 private:
 143         typedef std::vector<Buffer*> BufferVec;
 144
 145         /// Vector of vectors, indexed by DataType::to_index()
 146         std::vector<BufferVec> _buffers;
 147
 148         /// Use counts (there may be more actual buffers than this)
 149         ChanCount _count;
 150
 151         /// Available counts (number of buffers actually allocated)
 152         ChanCount _available;
 153
 154         /// Whether we (don't) 'own' the contained buffers (are a mirror of a PortSet)
 155         bool _is_mirror;
 156 };
 157
 158
 159 } // namespace ARDOUR
 160
 161 #endif // __ardour_buffer_set_h__