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