gtk2_ardour/editor_export_audio.cc

   1 /*
   2     Copyright (C) 2001 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 /* Note: public Editor methods are documented in public_editor.h */
  21
  22 #include <unistd.h>
  23 #include <climits>
  24
  25 #include <gtkmm/messagedialog.h>
  26
  27 #include "export_dialog.h"
  28 #include "editor.h"
  29 #include "public_editor.h"
  30 #include "selection.h"
  31 #include "time_axis_view.h"
  32 #include "audio_time_axis.h"
  33 #include "audio_region_view.h"
  34
  35 #include "pbd/pthread_utils.h"
  36 #include "ardour/types.h"
  37 #include "ardour/audio_track.h"
  38 #include "ardour/audiofilesource.h"
  39 #include "ardour/audio_diskstream.h"
  40 #include "ardour/audioregion.h"
  41 #include "ardour/audioplaylist.h"
  42 #include "ardour/chan_count.h"
  43 #include "ardour/session_directory.h"
  44 #include "ardour/source_factory.h"
  45 #include "ardour/audiofilesource.h"
  46
  47 #include "i18n.h"
  48
  49 using namespace std;
  50 using namespace ARDOUR;
  51 using namespace PBD;
  52 using namespace Gtk;
  53
  54 void
  55 Editor::export_audio ()
  56 {
  57         ExportDialog dialog (*this);
  58         dialog.set_session (session);
  59         dialog.run();
  60 }
  61
  62 void
  63 Editor::export_selection ()
  64 {
  65         ExportSelectionDialog dialog (*this);
  66         dialog.set_session (session);
  67         dialog.run();
  68 }
  69
  70 void
  71 Editor::export_range ()
  72 {
  73         Marker* marker;
  74
  75         if ((marker = reinterpret_cast<Marker *> (marker_menu_item->get_data ("marker"))) == 0) {
  76                 fatal << _("programming error: marker canvas item has no marker object pointer!") << endmsg;
  77                 /*NOTREACHED*/
  78         }
  79
  80         Location* l;
  81         bool is_start;
  82
  83         if (((l = find_location_from_marker (marker, is_start)) != 0) && (l->end() > l->start())) {
  84                 ExportRangeDialog dialog (*this, l->id().to_s());
  85                 dialog.set_session (session);
  86                 dialog.run();
  87         }
  88 }
  89
  90 /** Export the first selected region */
  91 void
  92 Editor::export_region ()
  93 {
  94         if (selection->regions.empty()) {
  95                 return;
  96         }
  97
  98         try {
  99                 boost::shared_ptr<Region> r = selection->regions.front()->region();
 100                 AudioRegion & region (dynamic_cast<AudioRegion &> (*r));
 101
 102                 RouteTimeAxisView & rtv (dynamic_cast<RouteTimeAxisView &> (selection->regions.front()->get_time_axis_view()));
 103                 AudioTrack & track (dynamic_cast<AudioTrack &> (*rtv.route()));
 104
 105                 ExportRegionDialog dialog (*this, region, track);
 106                 dialog.set_session (session);
 107                 dialog.run();
 108
 109         } catch (std::bad_cast & e) {
 110                 error << "Exporting Region failed!" << endmsg;
 111                 return;
 112         }
 113 }
 114
 115 int
 116 Editor::write_region_selection (RegionSelection& regions)
 117 {
 118         for (RegionSelection::iterator i = regions.begin(); i != regions.end(); ++i) {
 119                 // FIXME
 120                 AudioRegionView* arv = dynamic_cast<AudioRegionView*>(*i);
 121                 if (arv)
 122                         if (write_region ("", arv->audio_region()) == false)
 123                                 return -1;
 124         }
 125
 126         return 0;
 127 }
 128
 129 void
 130 Editor::bounce_region_selection ()
 131 {
 132         for (RegionSelection::iterator i = selection->regions.begin(); i != selection->regions.end(); ++i) {
 133
 134                 boost::shared_ptr<Region> region ((*i)->region());
 135                 RouteTimeAxisView* rtv = dynamic_cast<RouteTimeAxisView*>(&(*i)->get_time_axis_view());
 136                 Track* track = dynamic_cast<Track*>(rtv->route().get());
 137
 138                 InterThreadInfo itt;
 139
 140                 itt.done = false;
 141                 itt.cancel = false;
 142                 itt.progress = 0.0f;
 143
 144                 boost::shared_ptr<Region> r = track->bounce_range (region->position(), region->position() + region->length(), itt);
 145                 cerr << "Result of bounce of "
 146                      << region->name() << " len = " << region->length()
 147                      << " was "
 148                      << r->name() << " len = " << r->length()
 149                      << endl;
 150         }
 151 }
 152
 153 bool
 154 Editor::write_region (string path, boost::shared_ptr<AudioRegion> region)
 155 {
 156         boost::shared_ptr<AudioFileSource> fs;
 157         const nframes64_t chunk_size = 4096;
 158         nframes64_t to_read;
 159         Sample buf[chunk_size];
 160         gain_t gain_buffer[chunk_size];
 161         nframes64_t pos;
 162         char s[PATH_MAX+1];
 163         uint32_t cnt;
 164         vector<boost::shared_ptr<AudioFileSource> > sources;
 165         uint32_t nchans;
 166
 167         const string sound_directory = session->session_directory().sound_path().to_string();
 168
 169         nchans = region->n_channels();
 170
 171         /* don't do duplicate of the entire source if that's what is going on here */
 172
 173         if (region->start() == 0 && region->length() == region->source_length(0)) {
 174                 /* XXX should link(2) to create a new inode with "path" */
 175                 return true;
 176         }
 177
 178         if (path.length() == 0) {
 179
 180                 for (uint32_t n=0; n < nchans; ++n) {
 181
 182                         for (cnt = 0; cnt < 999999; ++cnt) {
 183                                 if (nchans == 1) {
 184                                         snprintf (s, sizeof(s), "%s/%s_%" PRIu32 ".wav", sound_directory.c_str(),
 185                                                   legalize_for_path(region->name()).c_str(), cnt);
 186                                 }
 187                                 else {
 188                                         snprintf (s, sizeof(s), "%s/%s_%" PRIu32 "-%" PRId32 ".wav", sound_directory.c_str(),
 189                                                   legalize_for_path(region->name()).c_str(), cnt, n);
 190                                 }
 191
 192                                 path = s;
 193
 194                                 if (::access (path.c_str(), F_OK) != 0) {
 195                                         break;
 196                                 }
 197                         }
 198
 199                         if (cnt == 999999) {
 200                                 error << "" << endmsg;
 201                                 goto error_out;
 202                         }
 203
 204
 205
 206                         try {
 207                                 fs = boost::dynamic_pointer_cast<AudioFileSource> (
 208                                                 SourceFactory::createWritable (DataType::AUDIO, *session,
 209                                                                 path, true,
 210                                                                 false, session->frame_rate()));
 211                         }
 212
 213                         catch (failed_constructor& err) {
 214                                 goto error_out;
 215                         }
 216
 217                         sources.push_back (fs);
 218                 }
 219         }
 220         else {
 221                 /* TODO: make filesources based on passed path */
 222
 223         }
 224
 225         to_read = region->length();
 226         pos = region->position();
 227
 228         while (to_read) {
 229                 nframes64_t this_time;
 230
 231                 this_time = min (to_read, chunk_size);
 232
 233                 for (vector<boost::shared_ptr<AudioFileSource> >::iterator src=sources.begin(); src != sources.end(); ++src) {
 234
 235                         fs = (*src);
 236
 237                         if (region->read_at (buf, buf, gain_buffer, pos, this_time) != this_time) {
 238                                 break;
 239                         }
 240
 241                         if (fs->write (buf, this_time) != this_time) {
 242                                 error << "" << endmsg;
 243                                 goto error_out;
 244                         }
 245                 }
 246
 247                 to_read -= this_time;
 248                 pos += this_time;
 249         }
 250
 251         time_t tnow;
 252         struct tm* now;
 253         time (&tnow);
 254         now = localtime (&tnow);
 255
 256         for (vector<boost::shared_ptr<AudioFileSource> >::iterator src = sources.begin(); src != sources.end(); ++src) {
 257                 (*src)->update_header (0, *now, tnow);
 258                 (*src)->mark_immutable ();
 259         }
 260
 261         return true;
 262
 263 error_out:
 264
 265         for (vector<boost::shared_ptr<AudioFileSource> >::iterator i = sources.begin(); i != sources.end(); ++i) {
 266                 (*i)->mark_for_remove ();
 267         }
 268
 269         return 0;
 270 }
 271
 272 int
 273 Editor::write_audio_selection (TimeSelection& ts)
 274 {
 275         int ret = 0;
 276
 277         if (selection->tracks.empty()) {
 278                 return 0;
 279         }
 280
 281         for (TrackSelection::iterator i = selection->tracks.begin(); i != selection->tracks.end(); ++i) {
 282
 283                 AudioTimeAxisView* atv;
 284
 285                 if ((atv = dynamic_cast<AudioTimeAxisView*>(*i)) == 0) {
 286                         continue;
 287                 }
 288
 289                 if (atv->is_audio_track()) {
 290
 291                         boost::shared_ptr<AudioPlaylist> playlist = boost::dynamic_pointer_cast<AudioPlaylist>(atv->get_diskstream()->playlist());
 292
 293                         if (playlist && write_audio_range (*playlist, atv->get_diskstream()->n_channels(), ts) == 0) {
 294                                 ret = -1;
 295                                 break;
 296                         }
 297                 }
 298         }
 299
 300         return ret;
 301 }
 302
 303 bool
 304 Editor::write_audio_range (AudioPlaylist& playlist, const ChanCount& count, list<AudioRange>& range)
 305 {
 306         boost::shared_ptr<AudioFileSource> fs;
 307         const nframes64_t chunk_size = 4096;
 308         nframes64_t nframes;
 309         Sample buf[chunk_size];
 310         gain_t gain_buffer[chunk_size];
 311         nframes64_t pos;
 312         char s[PATH_MAX+1];
 313         uint32_t cnt;
 314         string path;
 315         vector<boost::shared_ptr<AudioFileSource> > sources;
 316
 317         const string sound_directory = session->session_directory().sound_path().to_string();
 318
 319         uint32_t channels = count.n_audio();
 320
 321         for (uint32_t n=0; n < channels; ++n) {
 322
 323                 for (cnt = 0; cnt < 999999; ++cnt) {
 324                         if (channels == 1) {
 325                                 snprintf (s, sizeof(s), "%s/%s_%" PRIu32 ".wav", sound_directory.c_str(),
 326                                           legalize_for_path(playlist.name()).c_str(), cnt);
 327                         }
 328                         else {
 329                                 snprintf (s, sizeof(s), "%s/%s_%" PRIu32 "-%" PRId32 ".wav", sound_directory.c_str(),
 330                                           legalize_for_path(playlist.name()).c_str(), cnt, n);
 331                         }
 332
 333                         if (::access (s, F_OK) != 0) {
 334                                 break;
 335                         }
 336                 }
 337
 338                 if (cnt == 999999) {
 339                         error << "" << endmsg;
 340                         goto error_out;
 341                 }
 342
 343                 path = s;
 344
 345                 try {
 346                         fs = boost::dynamic_pointer_cast<AudioFileSource> (
 347                                         SourceFactory::createWritable (DataType::AUDIO, *session,
 348                                                         path, true,
 349                                                         false, session->frame_rate()));
 350                 }
 351
 352                 catch (failed_constructor& err) {
 353                         goto error_out;
 354                 }
 355
 356                 sources.push_back (fs);
 357
 358         }
 359
 360
 361         for (list<AudioRange>::iterator i = range.begin(); i != range.end();) {
 362
 363                 nframes = (*i).length();
 364                 pos = (*i).start;
 365
 366                 while (nframes) {
 367                         nframes64_t this_time;
 368
 369                         this_time = min (nframes, chunk_size);
 370
 371                         for (uint32_t n=0; n < channels; ++n) {
 372
 373                                 fs = sources[n];
 374
 375                                 if (playlist.read (buf, buf, gain_buffer, pos, this_time, n) != this_time) {
 376                                         break;
 377                                 }
 378
 379                                 if (fs->write (buf, this_time) != this_time) {
 380                                         goto error_out;
 381                                 }
 382                         }
 383
 384                         nframes -= this_time;
 385                         pos += this_time;
 386                 }
 387
 388                 list<AudioRange>::iterator tmp = i;
 389                 ++tmp;
 390
 391                 if (tmp != range.end()) {
 392
 393                         /* fill gaps with silence */
 394
 395                         nframes = (*tmp).start - (*i).end;
 396
 397                         while (nframes) {
 398
 399                                 nframes64_t this_time = min (nframes, chunk_size);
 400                                 memset (buf, 0, sizeof (Sample) * this_time);
 401
 402                                 for (uint32_t n=0; n < channels; ++n) {
 403
 404                                         fs = sources[n];
 405                                         if (fs->write (buf, this_time) != this_time) {
 406                                                 goto error_out;
 407                                         }
 408                                 }
 409
 410                                 nframes -= this_time;
 411                         }
 412                 }
 413
 414                 i = tmp;
 415         }
 416
 417         time_t tnow;
 418         struct tm* now;
 419         time (&tnow);
 420         now = localtime (&tnow);
 421
 422         for (vector<boost::shared_ptr<AudioFileSource> >::iterator s = sources.begin(); s != sources.end(); ++s) {
 423                 (*s)->update_header (0, *now, tnow);
 424                 (*s)->mark_immutable ();
 425                 // do we need to ref it again?
 426         }
 427
 428         return true;
 429
 430 error_out:
 431         /* unref created files */
 432
 433         for (vector<boost::shared_ptr<AudioFileSource> >::iterator i = sources.begin(); i != sources.end(); ++i) {
 434                 (*i)->mark_for_remove ();
 435         }
 436
 437         return false;
 438 }
 439
 440 void
 441 Editor::write_selection ()
 442 {
 443         if (!selection->time.empty()) {
 444                 write_audio_selection (selection->time);
 445         } else if (!selection->regions.empty()) {
 446                 write_region_selection (selection->regions);
 447         }
 448 }