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