libs/ardour/st_stretch.cc

   1 /*
   2     Copyright (C) 2004-2007 Paul Davis
   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 <algorithm>
  21 #include <cmath>
  22
  23 #include "pbd/error.h"
  24
  25 #include "ardour/types.h"
  26 #include "ardour/stretch.h"
  27 #include "ardour/audiofilesource.h"
  28 #include "ardour/session.h"
  29 #include "ardour/audioregion.h"
  30
  31 #include "i18n.h"
  32
  33 using namespace std;
  34 using namespace ARDOUR;
  35 using namespace PBD;
  36 using namespace soundtouch;
  37
  38 STStretch::STStretch (Session& s, TimeFXRequest& req)
  39         : Filter (s)
  40         , tsr (req)
  41 {
  42         float percentage;
  43
  44         /* the soundtouch code wants a *tempo* change percentage, which is
  45            of opposite sign to the length change.
  46         */
  47
  48         percentage = -tsr.time_fraction;
  49
  50         st.setSampleRate (s.frame_rate());
  51         st.setChannels (1);
  52         st.setTempoChange (percentage);
  53         st.setPitchSemiTones (0);
  54         st.setRateChange (0);
  55
  56         st.setSetting(SETTING_USE_QUICKSEEK, tsr.quick_seek);
  57         st.setSetting(SETTING_USE_AA_FILTER, tsr.antialias);
  58
  59         tsr.progress = 0.0f;
  60 }
  61
  62 STStretch::~STStretch ()
  63 {
  64 }
  65
  66 int
  67 STStretch::run (boost::shared_ptr<Region> a_region)
  68 {
  69         SourceList nsrcs;
  70         framecnt_t total_frames;
  71         framecnt_t done;
  72         int ret = -1;
  73         const framecnt_t bufsize = 16384;
  74         gain_t *gain_buffer = 0;
  75         Sample *buffer = 0;
  76         char suffix[32];
  77         string new_name;
  78         string::size_type at;
  79
  80         tsr.progress = 0.0f;
  81         tsr.done = false;
  82
  83         boost::shared_ptr<AudioRegion> region = boost::dynamic_pointer_cast<AudioRegion>(a_region);
  84
  85         total_frames = region->length() * region->n_channels();
  86         done = 0;
  87
  88         /* the name doesn't need to be super-precise, but allow for 2 fractional
  89            digits just to disambiguate close but not identical stretches.
  90         */
  91
  92         snprintf (suffix, sizeof (suffix), "@%d", (int) floor (tsr.time_fraction * 100.0f));
  93
  94         /* create new sources */
  95
  96         if (make_new_sources (region, nsrcs, suffix)) {
  97                 goto out;
  98         }
  99
 100         gain_buffer = new gain_t[bufsize];
 101         buffer = new Sample[bufsize];
 102
 103         // soundtouch throws runtime_error on error
 104
 105         try {
 106                 for (uint32_t i = 0; i < nsrcs.size(); ++i) {
 107
 108                         boost::shared_ptr<AudioSource> asrc
 109                                 = boost::dynamic_pointer_cast<AudioSource>(nsrcs[i]);
 110
 111                         framepos_t pos = 0;
 112                         framecnt_t this_read = 0;
 113
 114                         st.clear();
 115
 116                         while (!tsr.cancel && pos < region->length()) {
 117                                 framecnt_t this_time;
 118
 119                                 this_time = min (bufsize, region->length() - pos);
 120
 121                                 /* read from the master (original) sources for the region,
 122                                    not the ones currently in use, in case it's already been
 123                                    subject to timefx.
 124                                 */
 125
 126                                 if ((this_read = region->master_read_at (buffer, buffer, gain_buffer, pos + region->position(), this_time)) != this_time) {
 127                                         error << string_compose (_("tempoize: error reading data from %1"), asrc->name()) << endmsg;
 128                                         goto out;
 129                                 }
 130
 131                                 pos += this_read;
 132                                 done += this_read;
 133
 134                                 tsr.progress = (float) done / total_frames;
 135
 136                                 st.putSamples (buffer, this_read);
 137
 138                                 while ((this_read = st.receiveSamples (buffer, bufsize)) > 0 && !tsr.cancel) {
 139                                         if (asrc->write (buffer, this_read) != this_read) {
 140                                                 error << string_compose (_("error writing tempo-adjusted data to %1"), asrc->name()) << endmsg;
 141                                                 goto out;
 142                                         }
 143                                 }
 144                         }
 145
 146                         if (!tsr.cancel) {
 147                                 st.flush ();
 148                         }
 149
 150                         while (!tsr.cancel && (this_read = st.receiveSamples (buffer, bufsize)) > 0) {
 151                                 if (asrc->write (buffer, this_read) != this_read) {
 152                                         error << string_compose (_("error writing tempo-adjusted data to %1"), asrc->name()) << endmsg;
 153                                         goto out;
 154                                 }
 155                         }
 156                 }
 157
 158         } catch (runtime_error& err) {
 159                 error << _("timefx code failure. please notify ardour-developers.") << endmsg;
 160                 error << err.what() << endmsg;
 161                 goto out;
 162         }
 163
 164         new_name = region->name();
 165         at = new_name.find ('@');
 166
 167         // remove any existing stretch indicator
 168
 169         if (at != string::npos && at > 2) {
 170                 new_name = new_name.substr (0, at - 1);
 171         }
 172
 173         new_name += suffix;
 174
 175         ret = finish (region, nsrcs, new_name);
 176
 177         /* now reset ancestral data for each new region */
 178
 179         for (vector<boost::shared_ptr<Region> >::iterator x = results.begin(); x != results.end(); ++x) {
 180                 framepos_t astart = (*x)->ancestral_start();
 181                 framepos_t alength = (*x)->ancestral_length();
 182                 framepos_t start;
 183                 framecnt_t length;
 184
 185                 // note: tsr.fraction is a percentage of original length. 100 = no change,
 186                 // 50 is half as long, 200 is twice as long, etc.
 187
 188                 float stretch = (*x)->stretch() * (tsr.time_fraction/100.0);
 189
 190                 start = (framepos_t) floor (astart + ((astart - (*x)->start()) / stretch));
 191                 length = (framecnt_t) floor (alength / stretch);
 192
 193                 (*x)->set_ancestral_data (start, length, stretch, (*x)->shift());
 194         }
 195
 196   out:
 197
 198         delete [] gain_buffer;
 199         delete [] buffer;
 200
 201         if (ret || tsr.cancel) {
 202                 for (SourceList::iterator si = nsrcs.begin(); si != nsrcs.end(); ++si) {
 203                         (*si)->mark_for_remove ();
 204                 }
 205         }
 206
 207         tsr.done = true;
 208
 209         return ret;
 210 }