src/lib/audio_buffers.cc

   1 /*
   2     Copyright (C) 2012-2013 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include <cassert>
  21 #include <cstring>
  22 #include <cmath>
  23 #include <stdexcept>
  24 #include "audio_buffers.h"
  25
  26 using std::bad_alloc;
  27 using boost::shared_ptr;
  28
  29 /** Construct an AudioBuffers.  Audio data is undefined after this constructor.
  30  *  @param channels Number of channels.
  31  *  @param frames Number of frames to reserve space for.
  32  */
  33 AudioBuffers::AudioBuffers (int channels, int frames)
  34 {
  35         allocate (channels, frames);
  36 }
  37
  38 /** Copy constructor.
  39  *  @param other Other AudioBuffers; data is copied.
  40  */
  41 AudioBuffers::AudioBuffers (AudioBuffers const & other)
  42 {
  43         allocate (other._channels, other._frames);
  44         copy_from (&other, other._frames, 0, 0);
  45 }
  46
  47 AudioBuffers::AudioBuffers (boost::shared_ptr<const AudioBuffers> other)
  48 {
  49         allocate (other->_channels, other->_frames);
  50         copy_from (other.get(), other->_frames, 0, 0);
  51 }
  52
  53 AudioBuffers &
  54 AudioBuffers::operator= (AudioBuffers const & other)
  55 {
  56         if (this == &other) {
  57                 return *this;
  58         }
  59
  60         deallocate ();
  61         allocate (other._channels, other._frames);
  62         copy_from (&other, other._frames, 0, 0);
  63
  64         return *this;
  65 }
  66
  67 /** AudioBuffers destructor */
  68 AudioBuffers::~AudioBuffers ()
  69 {
  70         deallocate ();
  71 }
  72
  73 void
  74 AudioBuffers::allocate (int channels, int frames)
  75 {
  76         assert (frames >= 0);
  77         assert (channels >= 0);
  78
  79         _channels = channels;
  80         _frames = frames;
  81         _allocated_frames = frames;
  82
  83         _data = static_cast<float**> (malloc (_channels * sizeof (float *)));
  84         if (!_data) {
  85                 throw bad_alloc ();
  86         }
  87
  88         for (int i = 0; i < _channels; ++i) {
  89                 _data[i] = static_cast<float*> (malloc (frames * sizeof (float)));
  90                 if (!_data[i]) {
  91                         throw bad_alloc ();
  92                 }
  93         }
  94 }
  95
  96 void
  97 AudioBuffers::deallocate ()
  98 {
  99         for (int i = 0; i < _channels; ++i) {
 100                 free (_data[i]);
 101         }
 102
 103         free (_data);
 104 }
 105
 106 /** @param c Channel index.
 107  *  @return Buffer for this channel.
 108  */
 109 float*
 110 AudioBuffers::data (int c) const
 111 {
 112         assert (c >= 0 && c < _channels);
 113         return _data[c];
 114 }
 115
 116 /** Set the number of frames that these AudioBuffers will report themselves
 117  *  as having.  If we reduce the number of frames, the `lost' frames will
 118  *  be silenced.
 119  *  @param f Frames; must be less than or equal to the number of allocated frames.
 120  */
 121 void
 122 AudioBuffers::set_frames (int f)
 123 {
 124         assert (f <= _allocated_frames);
 125
 126         for (int c = 0; c < _channels; ++c) {
 127                 for (int i = f; i < _frames; ++i) {
 128                         _data[c][i] = 0;
 129                 }
 130         }
 131
 132         _frames = f;
 133 }
 134
 135 /** Make all samples on all channels silent */
 136 void
 137 AudioBuffers::make_silent ()
 138 {
 139         for (int i = 0; i < _channels; ++i) {
 140                 make_silent (i);
 141         }
 142 }
 143
 144 /** Make all samples on a given channel silent.
 145  *  @param c Channel.
 146  */
 147 void
 148 AudioBuffers::make_silent (int c)
 149 {
 150         assert (c >= 0 && c < _channels);
 151
 152         for (int i = 0; i < _frames; ++i) {
 153                 _data[c][i] = 0;
 154         }
 155 }
 156
 157 void
 158 AudioBuffers::make_silent (int from, int frames)
 159 {
 160         assert ((from + frames) <= _allocated_frames);
 161
 162         for (int c = 0; c < _channels; ++c) {
 163                 for (int i = from; i < (from + frames); ++i) {
 164                         _data[c][i] = 0;
 165                 }
 166         }
 167 }
 168
 169 /** Copy data from another AudioBuffers to this one.  All channels are copied.
 170  *  @param from AudioBuffers to copy from; must have the same number of channels as this.
 171  *  @param frames_to_copy Number of frames to copy.
 172  *  @param read_offset Offset to read from in `from'.
 173  *  @param write_offset Offset to write to in `to'.
 174  */
 175 void
 176 AudioBuffers::copy_from (AudioBuffers const * from, int frames_to_copy, int read_offset, int write_offset)
 177 {
 178         assert (from->channels() == channels());
 179
 180         assert (from);
 181         assert (read_offset >= 0 && (read_offset + frames_to_copy) <= from->_allocated_frames);
 182         assert (write_offset >= 0 && (write_offset + frames_to_copy) <= _allocated_frames);
 183
 184         for (int i = 0; i < _channels; ++i) {
 185                 memcpy (_data[i] + write_offset, from->_data[i] + read_offset, frames_to_copy * sizeof(float));
 186         }
 187 }
 188
 189 /** Move audio data around.
 190  *  @param from Offset to move from.
 191  *  @param to Offset to move to.
 192  *  @param frames Number of frames to move.
 193  */
 194
 195 void
 196 AudioBuffers::move (int from, int to, int frames)
 197 {
 198         if (frames == 0) {
 199                 return;
 200         }
 201
 202         assert (from >= 0);
 203         assert (from < _frames);
 204         assert (to >= 0);
 205         assert (to < _frames);
 206         assert (frames > 0);
 207         assert (frames <= _frames);
 208         assert ((from + frames) <= _frames);
 209         assert ((to + frames) <= _allocated_frames);
 210
 211         for (int i = 0; i < _channels; ++i) {
 212                 memmove (_data[i] + to, _data[i] + from, frames * sizeof(float));
 213         }
 214 }
 215
 216 /** Add data from from `from', `from_channel' to our channel `to_channel'.
 217  *  @param gain Linear gain to apply to the data before it is added.
 218  */
 219 void
 220 AudioBuffers::accumulate_channel (AudioBuffers const * from, int from_channel, int to_channel, float gain)
 221 {
 222         int const N = frames ();
 223         assert (from->frames() == N);
 224         assert (to_channel <= _channels);
 225
 226         float* s = from->data (from_channel);
 227         float* d = _data[to_channel];
 228
 229         for (int i = 0; i < N; ++i) {
 230                 *d++ += (*s++) * gain;
 231         }
 232 }
 233
 234 /** Ensure we have space for at least a certain number of frames.  If we extend
 235  *  the buffers, fill the new space with silence.
 236  */
 237 void
 238 AudioBuffers::ensure_size (int frames)
 239 {
 240         if (_allocated_frames >= frames) {
 241                 return;
 242         }
 243
 244         for (int i = 0; i < _channels; ++i) {
 245                 _data[i] = static_cast<float*> (realloc (_data[i], frames * sizeof (float)));
 246                 if (!_data[i]) {
 247                         throw bad_alloc ();
 248                 }
 249                 for (int j = _allocated_frames; j < frames; ++j) {
 250                         _data[i][j] = 0;
 251                 }
 252         }
 253
 254         _allocated_frames = frames;
 255 }
 256
 257 void
 258 AudioBuffers::accumulate_frames (AudioBuffers const * from, int read_offset, int write_offset, int frames)
 259 {
 260         assert (_channels == from->channels ());
 261
 262         for (int i = 0; i < _channels; ++i) {
 263                 for (int j = 0; j < frames; ++j) {
 264                         _data[i][j + write_offset] += from->data()[i][j + read_offset];
 265                 }
 266         }
 267 }
 268
 269 /** @param dB gain in dB */
 270 void
 271 AudioBuffers::apply_gain (float dB)
 272 {
 273         float const linear = pow (10, dB / 20);
 274
 275         for (int i = 0; i < _channels; ++i) {
 276                 for (int j = 0; j < _frames; ++j) {
 277                         _data[i][j] *= linear;
 278                 }
 279         }
 280 }