vtl: audio-export progress-bar tweaks
[ardour.git] / gtk2_ardour / export_video_dialog.cc
1 /*
2     Copyright (C) 2010 Paul Davis
3     Author: Robin Gareus <robin@gareus.org>
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License
16     along with this program; if not, write to the Free Software
17     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18
19 */
20 #ifdef WITH_VIDEOTIMELINE
21
22 #include <cstdio>
23 #include <string>
24 #include <sstream>
25 #include <iomanip>
26
27 #include <unistd.h>
28 #include <sys/types.h>
29 #include <sys/stat.h>
30 #include <fcntl.h>
31
32 #include <sigc++/bind.h>
33 #include <libgen.h>
34
35 #include "pbd/error.h"
36 #include "pbd/convert.h"
37 #include "gtkmm2ext/utils.h"
38 #include "ardour/session_directory.h"
39 #include "ardour/profile.h"
40 #include "ardour/template_utils.h"
41 #include "ardour/session.h"
42 #include "ardour_ui.h"
43 #include "gui_thread.h"
44
45 #include "ardour/export_handler.h"
46 #include "ardour/export_status.h"
47 #include "ardour/export_timespan.h"
48 #include "ardour/export_channel_configuration.h"
49 #include "ardour/export_format_specification.h"
50 #include "ardour/export_filename.h"
51 #include "ardour/route.h"
52 #include "ardour/session_metadata.h"
53 #include "ardour/broadcast_info.h"
54
55 #include "utils.h"
56 #include "opts.h"
57 #include "export_video_dialog.h"
58 #include "utils_videotl.h"
59 #include "i18n.h"
60
61 using namespace Gtk;
62 using namespace std;
63 using namespace PBD;
64 using namespace ARDOUR;
65
66 ExportVideoDialog::ExportVideoDialog (PublicEditor& ed, Session* s)
67         : ArdourDialog (_("Export Video File "))
68         , editor (ed)
69         , outfn_path_label (_("File:"), Gtk::ALIGN_LEFT)
70         , outfn_browse_button (_("Browse"))
71         , invid_path_label (_("Video:"), Gtk::ALIGN_LEFT)
72         , invid_browse_button (_("Browse"))
73         , transcode_button (_("Export"))
74         , abort_button (_("Abort"))
75         , scale_checkbox (_("Scale Video (W x H):"))
76         , width_adjustment (768, 128, 1920, 1, 16, 0)
77         , width_spinner (width_adjustment)
78         , height_adjustment (576, 128, 1920, 1, 16, 0)
79         , height_spinner (height_adjustment)
80         , aspect_checkbox (_("Set Aspect Ratio:"))
81         , normalize_checkbox (_("Normalize Audio"))
82         , twopass_checkbox (_("2 Pass Encoding"))
83         , optimizations_checkbox (_("Codec Optimizations:"))
84         , optimizations_label ("-")
85         , deinterlace_checkbox (_("Deinterlace"))
86         , bframes_checkbox (_("Use [2] B-frames (MPEG 2 or 4 only)"))
87         , fps_checkbox (_("Override FPS (Default is to retain FPS from the input video file):"))
88         , meta_checkbox (_("Include Session Metadata"))
89 #if 1 /* tentative debug mode */
90         , debug_checkbox (_("Enable Debug Mode: Print ffmpeg Command & Output to stdout."))
91 #endif
92 {
93         set_session (s);
94
95         set_name ("ExportVideoDialog");
96         set_position (Gtk::WIN_POS_MOUSE);
97         set_modal (true);
98         set_skip_taskbar_hint (true);
99         set_resizable (false);
100
101         Gtk::Label* l;
102         vbox = manage (new VBox);
103         VBox* options_box = manage (new VBox);
104         HBox* path_hbox;
105
106         /* check if ffmpeg can be found */
107         transcoder = new TranscodeFfmpeg("");
108         if (!transcoder->ffexec_ok()) {
109                 l = manage (new Label (_("No ffprobe or ffmpeg executables could be found on this system. Video Export is not possible until you install those tools. See the Log widow for more information."), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
110                 l->set_line_wrap();
111                 vbox->pack_start (*l, false, false, 8);
112                 get_vbox()->pack_start (*vbox, false, false);
113                 add_button (Stock::OK, RESPONSE_CANCEL);
114                 show_all_children ();
115                 return;
116         }
117         delete transcoder; transcoder = 0;
118
119         l = manage (new Label (_("<b>Output:</b>"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
120         l->set_use_markup ();
121         vbox->pack_start (*l, false, false, 4);
122
123         path_hbox = manage (new HBox);
124         path_hbox->pack_start (outfn_path_label, false, false, 3);
125         path_hbox->pack_start (outfn_path_entry, true, true, 3);
126         path_hbox->pack_start (outfn_browse_button, false, false, 3);
127         vbox->pack_start (*path_hbox, false, false, 2);
128
129         l = manage (new Label (_("<b>Input:</b>"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
130         l->set_use_markup ();
131         vbox->pack_start (*l, false, false, 4);
132
133         path_hbox = manage (new HBox);
134         path_hbox->pack_start (invid_path_label, false, false, 3);
135         path_hbox->pack_start (invid_path_entry, true, true, 3);
136         path_hbox->pack_start (invid_browse_button, false, false, 3);
137         vbox->pack_start (*path_hbox, false, false, 2);
138
139         path_hbox = manage (new HBox);
140         l = manage (new Label (_("Audio:"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
141         path_hbox->pack_start (*l, false, false, 3);
142         l = manage (new Label (_("Master Bus"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
143         path_hbox->pack_start (*l, false, false, 2);
144         vbox->pack_start (*path_hbox, false, false, 2);
145
146         insnd_combo.set_name ("PaddedButton");
147         insnd_combo.append_text("from ardour session-start to session-end");
148
149         frameoffset_t av_offset = ARDOUR_UI::instance()->video_timeline->get_offset();
150         if (av_offset < 0 ) {
151                 insnd_combo.append_text("from 00:00:00:00 to video-end");
152         } else {
153                 insnd_combo.append_text("from video-start to video-end");
154         }
155         insnd_combo.set_active(0);
156
157         outfn_path_entry.set_width_chars(38);
158         outfn_path_entry.set_text (_session->session_directory().export_path() + G_DIR_SEPARATOR +"export.avi");
159
160         XMLNode* node = _session->extra_xml (X_("Videotimeline"));
161         if (node && node->property(X_("Filename"))) {
162                 std::string filename = node->property(X_("Filename"))->value();
163                 if (filename.at(0) != G_DIR_SEPARATOR) {
164                         filename = Glib::build_filename (_session->session_directory().video_path(), filename);
165                 }
166                 invid_path_entry.set_text (filename);
167         }
168
169         l = manage (new Label (_("<b>Settings:</b>"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
170         l->set_use_markup ();
171         options_box->pack_start (*l, false, true, 4);
172
173         Table* t = manage (new Table (4, 12));
174         t->set_spacings (4);
175         int ty = 0;
176         options_box->pack_start (*t, true, true, 4);
177         l = manage (new Label (_("Range:"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
178         t->attach (*l, 0, 1, ty, ty+1);
179         t->attach (insnd_combo, 1, 4, ty, ty+1); ty++;
180         l = manage (new Label (_("Preset:"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
181         t->attach (*l, 0, 1, ty, ty+1);
182         t->attach (preset_combo, 1, 4, ty, ty+1); ty++;
183         l = manage (new Label (_("Video Codec:"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
184         t->attach (*l, 0, 1, ty, ty+1);
185         t->attach (video_codec_combo, 1, 2, ty, ty+1);
186         l = manage (new Label (_("Video KBit/s:"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
187         t->attach (*l, 2, 3, ty, ty+1);
188         t->attach (video_bitrate_combo, 3, 4, ty, ty+1); ty++;
189         l = manage (new Label (_("Audio Codec:"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
190         t->attach (*l, 0, 1, ty, ty+1);
191         t->attach (audio_codec_combo, 1, 2, ty, ty+1);
192         l = manage (new Label (_("Audio KBit/s:"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
193         t->attach (*l, 2, 3, ty, ty+1);
194         t->attach (audio_bitrate_combo, 3, 4, ty, ty+1); ty++;
195         l = manage (new Label (_("Audio Samplerate:"), Gtk::ALIGN_LEFT, Gtk::ALIGN_CENTER, false));
196         t->attach (*l, 0, 1, ty, ty+1);
197         t->attach (audio_samplerate_combo, 1, 2, ty, ty+1);
198         t->attach (normalize_checkbox, 2, 4, ty, ty+1); ty++;
199         t->attach (scale_checkbox, 0, 2, ty, ty+1);
200         t->attach (width_spinner, 2, 3, ty, ty+1);
201         t->attach (height_spinner, 3, 4, ty, ty+1); ty++;
202         t->attach (fps_checkbox, 0, 3, ty, ty+1);
203         t->attach (fps_combo, 3, 4, ty, ty+1); ty++;
204         t->attach (twopass_checkbox, 0, 2, ty, ty+1);
205         t->attach (aspect_checkbox, 2, 3, ty, ty+1);
206         t->attach (aspect_combo, 3, 4, ty, ty+1); ty++;
207         t->attach (bframes_checkbox, 0, 2, ty, ty+1);
208         t->attach (deinterlace_checkbox, 2, 4, ty, ty+1); ty++;
209         t->attach (meta_checkbox, 2, 4, ty, ty+1); ty++;
210         t->attach (optimizations_checkbox, 0, 1, ty, ty+1);
211         t->attach (optimizations_label, 1, 4, ty, ty+1); ty++;
212 #if 1 /* tentative debug mode */
213         t->attach (debug_checkbox, 0, 4, ty, ty+1); ty++;
214 #endif
215
216         preset_combo.set_name ("PaddedButton");
217         preset_combo.append_text("none");
218         preset_combo.append_text("dvd-mp2");
219         preset_combo.append_text("dvd-NTSC");
220         preset_combo.append_text("dvd-PAL");
221         preset_combo.append_text("flv");
222         preset_combo.append_text("mpeg4");
223         preset_combo.append_text("ogg");
224         preset_combo.append_text("you-tube");
225         preset_combo.set_active(0);
226
227         audio_codec_combo.set_name ("PaddedButton");
228         audio_codec_combo.append_text("ac3");
229         audio_codec_combo.append_text("aac");
230         audio_codec_combo.append_text("libmp3lame");
231         audio_codec_combo.append_text("libvorbis");
232         audio_codec_combo.append_text("mp2");
233         audio_codec_combo.append_text("pcm_s16le");
234         audio_codec_combo.set_active(2);
235
236         video_codec_combo.set_name ("PaddedButton");
237         video_codec_combo.append_text("flv");
238         video_codec_combo.append_text("libtheora");
239         video_codec_combo.append_text("mjpeg");
240         video_codec_combo.append_text("mpeg2video");
241         video_codec_combo.append_text("mpeg4");
242         video_codec_combo.append_text("x264 (baseline)");
243         video_codec_combo.append_text("x264 (hq)");
244         video_codec_combo.append_text("vpx (webm)");
245         video_codec_combo.append_text("copy");
246         video_codec_combo.set_active(4);
247
248         audio_bitrate_combo.set_name ("PaddedButton");
249         audio_bitrate_combo.append_text("64k");
250         audio_bitrate_combo.append_text("128k");
251         audio_bitrate_combo.append_text("192k");
252         audio_bitrate_combo.append_text("256k");
253         audio_bitrate_combo.append_text("320k");
254         audio_bitrate_combo.set_active(2);
255
256         audio_samplerate_combo.set_name ("PaddedButton");
257         audio_samplerate_combo.append_text("22050");
258         audio_samplerate_combo.append_text("44100");
259         audio_samplerate_combo.append_text("48000");
260         audio_samplerate_combo.set_active(2);
261
262         video_bitrate_combo.set_name ("PaddedButton");
263         video_bitrate_combo.append_text("200k");
264         video_bitrate_combo.append_text("800k");
265         video_bitrate_combo.append_text("2000k");
266         video_bitrate_combo.append_text("5000k");
267         video_bitrate_combo.append_text("8000k");
268         video_bitrate_combo.append_text("retain");
269         video_bitrate_combo.set_active(3);
270
271         fps_combo.set_name ("PaddedButton");
272         fps_combo.append_text("23.976");
273         fps_combo.append_text("24");
274         fps_combo.append_text("24.976");
275         fps_combo.append_text("25");
276         fps_combo.append_text("29.97");
277         fps_combo.append_text("30");
278         fps_combo.append_text("59.94");
279         fps_combo.append_text("60");
280         float tcfps = _session->timecode_frames_per_second();
281         if      (fabs(tcfps - 23.976) < 0.01) { fps_combo.set_active(0); }
282         else if (fabs(tcfps - 24.0  ) < 0.01) { fps_combo.set_active(1); }
283         else if (fabs(tcfps - 24.976) < 0.01) { fps_combo.set_active(2); }
284         else if (fabs(tcfps - 25.0  ) < 0.01) { fps_combo.set_active(3); }
285         else if (fabs(tcfps - 29.97 ) < 0.01) { fps_combo.set_active(4); }
286         else if (fabs(tcfps - 30.0  ) < 0.01) { fps_combo.set_active(5); }
287         else if (fabs(tcfps - 59.94 ) < 0.01) { fps_combo.set_active(6); }
288         else if (fabs(tcfps - 60.0  ) < 0.01) { fps_combo.set_active(7); }
289         else { fps_combo.set_active(5); }
290
291         aspect_combo.set_name ("PaddedButton");
292         aspect_combo.append_text("4:3");
293         aspect_combo.append_text("16:9");
294         aspect_combo.set_active(1);
295
296         optimizations_checkbox.set_sensitive(false);
297         scale_checkbox_toggled();
298         aspect_checkbox_toggled();
299         fps_checkbox_toggled();
300         video_codec_combo_changed();
301
302         vbox->pack_start (*options_box, false, true, 4);
303         get_vbox()->set_spacing (4);
304         get_vbox()->pack_start (*vbox, false, false);
305
306         progress_box = manage (new VBox);
307         progress_box->pack_start (pbar, false, false);
308         progress_box->pack_start (abort_button, false, false);
309         get_vbox()->pack_start (*progress_box, false, false);
310
311         scale_checkbox.signal_toggled().connect (sigc::mem_fun (*this, &ExportVideoDialog::scale_checkbox_toggled));
312         aspect_checkbox.signal_toggled().connect (sigc::mem_fun (*this, &ExportVideoDialog::aspect_checkbox_toggled));
313         fps_checkbox.signal_toggled().connect (sigc::mem_fun (*this, &ExportVideoDialog::fps_checkbox_toggled));
314         preset_combo.signal_changed().connect (sigc::mem_fun (*this, &ExportVideoDialog::preset_combo_changed));
315         video_codec_combo.signal_changed().connect (sigc::mem_fun (*this, &ExportVideoDialog::video_codec_combo_changed));
316         outfn_browse_button.signal_clicked().connect (sigc::mem_fun (*this, &ExportVideoDialog::open_outfn_dialog));
317         invid_browse_button.signal_clicked().connect (sigc::mem_fun (*this, &ExportVideoDialog::open_invid_dialog));
318         transcode_button.signal_clicked().connect (sigc::mem_fun (*this, &ExportVideoDialog::launch_export));
319         abort_button.signal_clicked().connect (sigc::mem_fun (*this, &ExportVideoDialog::abort_clicked));
320
321         cancel_button = add_button (Stock::CANCEL, RESPONSE_CANCEL);
322         get_action_area()->pack_start (transcode_button, false, false);
323         show_all_children ();
324         progress_box->hide();
325 }
326
327 ExportVideoDialog::~ExportVideoDialog ()
328 {
329         if (transcoder) { delete transcoder; transcoder = 0;}
330 }
331
332 void
333 ExportVideoDialog::on_show ()
334 {
335         Dialog::on_show ();
336 }
337
338 void
339 ExportVideoDialog::abort_clicked ()
340 {
341         aborted = true;
342         if (transcoder) {
343                 transcoder->cancel();
344         }
345 }
346
347 void
348 ExportVideoDialog::update_progress (framecnt_t c, framecnt_t a)
349 {
350         if (a == 0 || c > a) {
351                 pbar.set_pulse_step(.1);
352                 pbar.pulse();
353                 return;
354         }
355         pbar.set_fraction ((double)c / (double) a);
356 }
357
358
359 gint
360 ExportVideoDialog::audio_progress_display ()
361 {
362         std::string status_text;
363         float progress = 0.0;
364                 if (status->normalizing) {
365                         pbar.set_text (_("Normalizing audio"));
366                         progress = ((float) status->current_normalize_cycle) / status->total_normalize_cycles;
367                 } else {
368                         pbar.set_text (_("Exporting audio"));
369                         progress = ((float) status->processed_frames_current_timespan) / status->total_frames_current_timespan;
370                 }
371                 if (progress < previous_progress) {
372                         // Work around gtk bug
373                         pbar.hide();
374                         pbar.show();
375                 }
376                 previous_progress = progress;
377                 pbar.set_fraction (progress);
378         return TRUE;
379 }
380
381 void
382 ExportVideoDialog::finished ()
383 {
384         if (aborted) {
385                 unlink(outfn_path_entry.get_text().c_str());
386                 unlink (insnd.c_str());
387                 warning << _("Video Export Failed or Was Aborted") << endmsg;
388                 Gtk::Dialog::response(RESPONSE_CANCEL);
389         } else if (twopass) {
390                 twopass = false;
391                 if (transcoder) { delete transcoder; transcoder = 0;}
392                 encode_pass(2);
393         } else {
394                 if (twopass_checkbox.get_active()) {
395                         std::string outfn = outfn_path_entry.get_text();
396                         std::string p2log = Glib::path_get_dirname (outfn) + G_DIR_SEPARATOR + "ffmpeg2pass";
397                         unlink (p2log.c_str());
398                 }
399                 unlink (insnd.c_str());
400                 Gtk::Dialog::response(RESPONSE_ACCEPT);
401         }
402 }
403
404 void
405 ExportVideoDialog::launch_export ()
406 {
407         std::string outfn = outfn_path_entry.get_text();
408         if (!confirm_video_outfn(outfn)) { return; }
409
410         vbox->hide();
411         cancel_button->hide();
412         transcode_button.hide();
413         pbar.set_size_request(300,-1);
414         pbar.set_text(_("Exporting Audio.."));
415         progress_box->show();
416         aborted = false;
417         twopass = twopass_checkbox.get_active();
418
419         /* export audio track */
420         ExportTimespanPtr tsp = _session->get_export_handler()->add_timespan();
421         boost::shared_ptr<ExportChannelConfiguration> ccp = _session->get_export_handler()->add_channel_config();
422         boost::shared_ptr<ARDOUR::ExportFilename> fnp = _session->get_export_handler()->add_filename();
423         boost::shared_ptr<AudioGrapher::BroadcastInfo> b;
424         XMLTree tree;
425         std::string vtl_samplerate = audio_samplerate_combo.get_active_text();
426         std::string vtl_normalize = normalize_checkbox.get_active()?"true":"false";
427         tree.read_buffer(std::string(
428 "<?xml version=\"1.0\" encoding=\"UTF-8\"?>"
429 "<ExportFormatSpecification name=\"VTL-WAV-16\" id=\"3094591e-ccb9-4385-a93f-c9955ffeb1f0\">"
430 "  <Encoding id=\"F_WAV\" type=\"T_Sndfile\" extension=\"wav\" name=\"WAV\" has-sample-format=\"true\" channel-limit=\"256\"/>"
431 "  <SampleRate rate=\""+ vtl_samplerate +"\"/>"
432 "  <SRCQuality quality=\"SRC_SincBest\"/>"
433 "  <EncodingOptions>"
434 "    <Option name=\"sample-format\" value=\"SF_16\"/>"
435 "    <Option name=\"dithering\" value=\"D_None\"/>"
436 "    <Option name=\"tag-metadata\" value=\"true\"/>"
437 "    <Option name=\"tag-support\" value=\"false\"/>"
438 "    <Option name=\"broadcast-info\" value=\"false\"/>"
439 "  </EncodingOptions>"
440 "  <Processing>"
441 "    <Normalize enabled=\""+ vtl_normalize +"\" target=\"0\"/>"
442 "    <Silence>"
443 "      <Start>"
444 "        <Trim enabled=\"false\"/>"
445 "        <Add enabled=\"false\">"
446 "          <Duration format=\"Timecode\" hours=\"0\" minutes=\"0\" seconds=\"0\" frames=\"0\"/>"
447 "        </Add>"
448 "      </Start>"
449 "      <End>"
450 "        <Trim enabled=\"false\"/>"
451 "        <Add enabled=\"false\">"
452 "          <Duration format=\"Timecode\" hours=\"0\" minutes=\"0\" seconds=\"0\" frames=\"0\"/>"
453 "        </Add>"
454 "      </End>"
455 "    </Silence>"
456 "  </Processing>"
457 "</ExportFormatSpecification>"
458 ));
459         boost::shared_ptr<ExportFormatSpecification> fmp = _session->get_export_handler()->add_format(*tree.root());
460
461         /* set up range */
462         framepos_t start, end;
463         start = end = 0;
464         if (insnd_combo.get_active_row_number() == 1) {
465                 transcoder = new TranscodeFfmpeg(invid_path_entry.get_text());
466                 if (transcoder->probe_ok() && transcoder->get_fps() > 0) {
467                         end = transcoder->get_duration() * _session->nominal_frame_rate() / transcoder->get_fps();
468                 } else {
469                         warning << _("Export Video: Cannot query duration of video-file, using duration from timeline instead.") << endmsg;
470                         end = ARDOUR_UI::instance()->video_timeline->get_duration();
471                 }
472                 if (transcoder) {delete transcoder; transcoder = 0;}
473
474                 frameoffset_t av_offset = ARDOUR_UI::instance()->video_timeline->get_offset();
475 #if 0 /* DEBUG */
476                 printf("audio-range -- AV offset: %lld\n", av_offset);
477 #endif
478                 if (av_offset > 0) {
479                         start = av_offset;
480                 }
481                 end += av_offset;
482         }
483         if (end <= 0) {
484                 start = _session->current_start_frame();
485                 end   = _session->current_end_frame();
486         }
487 #if 0 /* DEBUG */
488         printf("audio export-range %lld -> %lld\n", start, end);
489 #endif
490
491         tsp->set_range (start, end);
492         tsp->set_name ("mysession");
493         tsp->set_range_id ("session");
494
495         /* add master outs as default */
496         IO* master_out = _session->master_out()->output().get();
497         if (!master_out) {
498                 warning << _("Export Video: No Master Out Ports to Connect for Audio Export") << endmsg;
499                 Gtk::Dialog::response(RESPONSE_CANCEL);
500                 return;
501         }
502         for (uint32_t n = 0; n < master_out->n_ports().n_audio(); ++n) {
503                 PortExportChannel * channel = new PortExportChannel ();
504                 channel->add_port (master_out->audio (n));
505                 ExportChannelPtr chan_ptr (channel);
506                 ccp->register_channel (chan_ptr);
507         }
508
509         /* outfile */
510         fnp->set_timespan(tsp);
511         fnp->set_label("vtl");
512         fnp->include_label = true;
513         insnd = fnp->get_path(fmp);
514
515         /* do sound export */
516         _session->get_export_handler()->add_export_config (tsp, ccp, fmp, fnp, b);
517         _session->get_export_handler()->do_export();
518         status = _session->get_export_status ();
519
520         audio_progress_connection = Glib::signal_timeout().connect (sigc::mem_fun(*this, &ExportVideoDialog::audio_progress_display), 100);
521         previous_progress = 0.0;
522         while (status->running) {
523                 if (aborted) { status->abort(); }
524                 if (gtk_events_pending()) {
525                         gtk_main_iteration ();
526                 } else {
527                         usleep (10000);
528                 }
529         }
530         audio_progress_connection.disconnect();
531         status->finish ();
532         if (status->aborted()) {
533                 unlink (insnd.c_str());
534                 Gtk::Dialog::response(RESPONSE_CANCEL);
535                 return;
536         }
537         pbar.set_text (_("Encoding Video.."));
538         encode_pass(1);
539 }
540
541 void
542 ExportVideoDialog::encode_pass (int pass)
543 {
544         std::string outfn = outfn_path_entry.get_text();
545         std::string invid = invid_path_entry.get_text();
546
547         transcoder = new TranscodeFfmpeg(invid);
548         if (!transcoder->ffexec_ok()) {
549                 /* ffmpeg binary was not found. TranscodeFfmpeg prints a warning */
550                 unlink (insnd.c_str());
551                 Gtk::Dialog::response(RESPONSE_CANCEL);
552                 return;
553         }
554         if (!transcoder->probe_ok()) {
555                 /* video input file can not be read */
556                 warning << _("Export Video: Video input file cannot be read.") << endmsg;
557           unlink (insnd.c_str());
558           Gtk::Dialog::response(RESPONSE_CANCEL);
559           return;
560         }
561
562         std::string preset = preset_combo.get_active_text();
563         FFSettings ffs ; /* = transcoder->default_encoder_settings(); */
564         ffs.clear();
565
566         if (fps_checkbox.get_active()) {
567           ffs["-r"] = fps_combo.get_active_text();
568         }
569
570         if (scale_checkbox.get_active()) {
571                 ffs["-s"] = string_compose("%1x%2", width_spinner.get_value(), height_spinner.get_value());
572         }
573         ffs["-vcodec"] = video_codec_combo.get_active_text();
574         ffs["-acodec"] = audio_codec_combo.get_active_text();
575
576         if (video_bitrate_combo.get_active_text() == "retain" ) {
577                 ffs["-sameq"]  = "-y"; // we use '-y' as dummy parameter for non key/value options
578         } else {
579                 ffs["-b:v"]  = video_bitrate_combo.get_active_text();
580         }
581         ffs["-b:a"] = audio_bitrate_combo.get_active_text();
582
583         if (audio_codec_combo.get_active_text() == "aac" ) {
584                 ffs["-strict"] = "-2";
585         }
586
587         if (video_codec_combo.get_active_text() == "x264 (hq)" ) {
588                 ffs["-vcodec"] = "libx264";
589                 ffs["-vprofile"] = "high";
590         }
591         else if (video_codec_combo.get_active_text() == "x264 (baseline)" ) {
592                 ffs["-vcodec"] = "libx264";
593                 ffs["-vpre"] = "baseline";
594         }
595         else if (video_codec_combo.get_active_text() == "vpx (webm)" ) {
596                 ffs["-vcodec"] = "libvpx";
597                 ffs["-g"] = "120";
598                 ffs["-qmin"] = "11";
599                 ffs["-qmax"] = "51";
600         }
601
602         if (optimizations_checkbox.get_active()) {
603           if (video_codec_combo.get_active_text() == "mpeg2video") {
604                         ffs["-mbd"] = "rd";
605                         ffs["-trellis"] = "2";
606                         ffs["-cmp"] = "2";
607                         ffs["-subcmp"] = "2";
608                 }
609                 else if (video_codec_combo.get_active_text() == "mpeg4") {
610                         ffs["-mbd"] = "rd";
611                         ffs["-flags"] = "+mv4+aic";
612                         ffs["-trellis"] = "2";
613                         ffs["-cmp"] = "2";
614                         ffs["-subcmp"] = "2";
615                         ffs["-g"] = "300";
616                 }
617                 else if (video_codec_combo.get_active_text() == "flv") {
618                         ffs["-mbd"] = "2";
619                         ffs["-cmp"] = "2";
620                         ffs["-subcmp"] = "2";
621                         ffs["-trellis"] = "2";
622                         ffs["-flags"] = "+aic+mv0+mv4";
623                         ffs["-g"] = "160";
624                 }
625         }
626
627         if (bframes_checkbox.get_active() && (
628                    video_codec_combo.get_active_text() == "mpeg2video"
629                 || video_codec_combo.get_active_text() == "mpeg4"
630                 )) {
631                 ffs["-bf"] = "2";
632         }
633
634         if (preset == "dvd-PAL") {
635                 ffs.clear(); /* ignore all prev settings */
636                 ffs["-target"] = "pal-dvd";
637                 ffs["-aspect"] = "4:3"; /* required for DVD - may be overridden below */
638         }
639         else if (preset == "dvd-NTSC") {
640                 ffs.clear(); /* ignore all prev settings */
641                 ffs["-target"] = "ntsc-dvd";
642                 ffs["-aspect"] = "4:3"; /* required for DVD - may be overridden below */
643         }
644
645         if (aspect_checkbox.get_active()) {
646                 ffs["-aspect"] = aspect_combo.get_active_text();
647         }
648         if (deinterlace_checkbox.get_active()) {
649                 ffs["-deinterlace"] = "-y"; // we use '-y' as dummy parameter for non key/value options
650         }
651
652         bool map = true;
653         if (pass == 1 && twopass) {
654                 pbar.set_text (_("Encoding Video.. Pass 1/2"));
655                 map = false;
656                 ffs["-pass"] = "1";
657                 ffs["-an"] = "-y";
658                 ffs["-passlogfile"] =  Glib::path_get_dirname (outfn) + G_DIR_SEPARATOR + "ffmpeg2pass";
659                 ffs["-f"] = get_file_extension(invid).empty()?"mov":get_file_extension(invid);
660 #ifdef _OS_WIN32
661                 outfn = "NUL";
662 #else
663                 outfn = "/dev/null";
664 #endif
665         } else if (pass == 2) {
666                 pbar.set_text (_("Encoding Video.. Pass 2/2"));
667                 ffs["-pass"] = "2";
668                 ffs["-passlogfile"] =  Glib::path_get_dirname (outfn) + G_DIR_SEPARATOR + "ffmpeg2pass";
669         }
670
671         frameoffset_t av_offset = ARDOUR_UI::instance()->video_timeline->get_offset();
672         double duration_s  = 0;
673
674         if (insnd_combo.get_active_row_number() == 0) {
675                 framecnt_t duration_f = _session->current_end_frame() - _session->current_start_frame();
676                 duration_s = (double)duration_f / (double)_session->nominal_frame_rate();
677         } else {
678                 framecnt_t duration_f = ARDOUR_UI::instance()->video_timeline->get_duration();
679                 if (av_offset < 0 ) {
680                         duration_f += av_offset;
681                 }
682                 duration_s = (double)duration_f / (double)_session->nominal_frame_rate();
683         }
684
685         std::ostringstream osstream; osstream << duration_s;
686         ffs["-t"] = osstream.str();
687         if (fps_checkbox.get_active()) {
688                 transcoder->set_duration(duration_s * atof(fps_combo.get_active_text()));
689         } else {
690                 transcoder->set_duration(duration_s * transcoder->get_fps());
691         }
692
693         if (insnd_combo.get_active_row_number() == 0) {
694                 const framepos_t start = _session->current_start_frame();
695                 const framepos_t snend = _session->current_end_frame();
696                 const frameoffset_t vid_duration = ARDOUR_UI::instance()->video_timeline->get_duration();
697
698 #if 0 /* DEBUG */
699                 printf("AV offset: %lld Vid-len: %lld Vid-end: %lld || start:%lld || end:%lld\n",
700                                 av_offset, vid_duration, av_offset+vid_duration, start, snend); // XXX
701 #endif
702
703                 if (av_offset > start && av_offset + vid_duration < snend) {
704                         transcoder->set_leadinout((av_offset - start) / (double)_session->nominal_frame_rate(),
705                                 (snend - (av_offset + vid_duration)) / (double)_session->nominal_frame_rate());
706                 } else if (av_offset > start) {
707                         transcoder->set_leadinout((av_offset - start) / (double)_session->nominal_frame_rate(), 0);
708                 } else if (av_offset + vid_duration < snend) {
709                         transcoder->set_leadinout(0, (snend - (av_offset + vid_duration)) / (double)_session->nominal_frame_rate());
710                         transcoder->set_avoffset((av_offset - start) / (double)_session->nominal_frame_rate());
711                 } else {
712                         transcoder->set_avoffset((av_offset - start) / (double)_session->nominal_frame_rate());
713                 }
714         } else if (av_offset < 0) {
715                 /* from 00:00:00:00 to video-end */
716                 transcoder->set_avoffset(av_offset / (double)_session->nominal_frame_rate());
717         }
718
719         FFSettings meta = transcoder->default_meta_data();
720         if (meta_checkbox.get_active()) {
721                 ARDOUR::SessionMetadata * session_data = ARDOUR::SessionMetadata::Metadata();
722                 if (session_data->year() > 0 ) {
723                         std::ostringstream osstream; osstream << session_data->year();
724                         meta["year"] = osstream.str();
725                 }
726                 if (session_data->track_number() > 0 ) {
727                         std::ostringstream osstream; osstream << session_data->track_number();
728                         meta["track"] = osstream.str();
729                 }
730                 if (session_data->disc_number() > 0 ) {
731                         std::ostringstream osstream; osstream << session_data->disc_number();
732                         meta["disc"] = osstream.str();
733                 }
734                 if (!session_data->title().empty())     {meta["title"] = session_data->title();}
735                 if (!session_data->artist().empty())    {meta["author"] = session_data->artist();}
736                 if (!session_data->album_artist().empty()) {meta["album_artist"] = session_data->album_artist();}
737                 if (!session_data->album().empty())     {meta["album"] = session_data->album();}
738                 if (!session_data->genre().empty())     {meta["genre"] = session_data->genre();}
739                 if (!session_data->composer().empty())  {meta["composer"] = session_data->composer();}
740                 if (!session_data->comment().empty())   {meta["comment"] = session_data->comment();}
741                 if (!session_data->copyright().empty()) {meta["copyright"] = session_data->copyright();}
742                 if (!session_data->subtitle().empty())  {meta["description"] = session_data->subtitle();}
743         }
744
745 #if 1 /* tentative debug mode */
746         if (debug_checkbox.get_active()) {
747                 transcoder->set_debug(true);
748         }
749 #endif
750
751         transcoder->Progress.connect(*this, invalidator (*this), boost::bind (&ExportVideoDialog::update_progress , this, _1, _2), gui_context());
752         transcoder->Finished.connect(*this, invalidator (*this), boost::bind (&ExportVideoDialog::finished, this), gui_context());
753         if (!transcoder->encode(outfn, insnd, invid, ffs, meta, map)) {
754                 ARDOUR_UI::instance()->popup_error(_("Transcoding failed."));
755                 Gtk::Dialog::response(RESPONSE_CANCEL);
756                 return;
757         }
758 }
759
760 void
761 ExportVideoDialog::change_file_extension (std::string ext)
762 {
763         outfn_path_entry.set_text (
764                 strip_file_extension(outfn_path_entry.get_text()) + ext
765         );
766 }
767
768 void
769 ExportVideoDialog::scale_checkbox_toggled ()
770 {
771         width_spinner.set_sensitive(scale_checkbox.get_active());
772         height_spinner.set_sensitive(scale_checkbox.get_active());
773 }
774
775 void
776 ExportVideoDialog::fps_checkbox_toggled ()
777 {
778         fps_combo.set_sensitive(fps_checkbox.get_active());
779 }
780
781 void
782 ExportVideoDialog::aspect_checkbox_toggled ()
783 {
784         aspect_combo.set_sensitive(aspect_checkbox.get_active());
785 }
786
787 void
788 ExportVideoDialog::video_codec_combo_changed ()
789 {
790         if ((  video_codec_combo.get_active_text() == "mpeg4"
791              ||video_codec_combo.get_active_text() == "mpeg2video"
792                         ) && !(
793                preset_combo.get_active_text() == "dvd-PAL"
794              ||preset_combo.get_active_text() == "dvd-NTSC"
795            )) {
796                 bframes_checkbox.set_sensitive(true);
797                 optimizations_checkbox.set_sensitive(true);
798                 if (video_codec_combo.get_active_text() == "mpeg2video") {
799                         optimizations_label.set_text("-mbd rd -trellis 2 -cmp 2 -subcmp 2"); // mpeg2
800                 } else if (video_codec_combo.get_active_text() == "mpeg4") {
801                         optimizations_label.set_text("-mbd rd -flags +mv4+aic -trellis 2 -cmp 2 -subcmp 2 -g 300"); // mpeg4
802                 } else {
803                         optimizations_label.set_text("-mbd 2 -cmp 2 -subcmp 2 -trellis 2 -flags +aic+mv0+mv4 -g 160"); // flv
804                 }
805         } else {
806                 bframes_checkbox.set_sensitive(false);
807                 bframes_checkbox.set_active(false);
808                 optimizations_checkbox.set_sensitive(false);
809                 optimizations_checkbox.set_active(false);
810                 optimizations_label.set_text("-");
811         }
812 }
813
814 void
815 ExportVideoDialog::preset_combo_changed ()
816 {
817         std::string p = preset_combo.get_active_text();
818         scale_checkbox.set_sensitive(true);
819
820         if (p == "flv") {
821                 change_file_extension(".flv");
822                 audio_codec_combo.set_active(1);
823                 video_codec_combo.set_active(0);
824                 audio_bitrate_combo.set_active(1);
825                 video_bitrate_combo.set_active(1);
826                 audio_samplerate_combo.set_active(1);
827         }
828         else if (p == "you-tube") {
829                 change_file_extension(".avi");
830                 audio_codec_combo.set_active(2);
831                 video_codec_combo.set_active(5);
832                 audio_bitrate_combo.set_active(1);
833                 video_bitrate_combo.set_active(2);
834                 if (_session->nominal_frame_rate() == 48000 || _session->nominal_frame_rate() == 96000) {
835                         audio_samplerate_combo.set_active(2);
836                 } else {
837                         audio_samplerate_combo.set_active(1);
838                 }
839         }
840         else if (p == "ogg") {
841                 change_file_extension(".ogv");
842                 audio_codec_combo.set_active(3);
843                 video_codec_combo.set_active(1);
844                 audio_bitrate_combo.set_active(2);
845                 video_bitrate_combo.set_active(2);
846                 if (_session->nominal_frame_rate() == 48000 || _session->nominal_frame_rate() == 96000) {
847                         audio_samplerate_combo.set_active(2);
848                 } else {
849                         audio_samplerate_combo.set_active(1);
850                 }
851         }
852         else if (p == "dvd-mp2") {
853                 change_file_extension(".mpg");
854                 audio_codec_combo.set_active(4);
855                 video_codec_combo.set_active(3);
856                 audio_bitrate_combo.set_active(3);
857                 video_bitrate_combo.set_active(3);
858                 audio_samplerate_combo.set_active(2);
859         }
860         else if (p == "dvd-NTSC" || p == "dvd-PAL") {
861                 change_file_extension(".mpg");
862                 audio_codec_combo.set_active(5);
863                 video_codec_combo.set_active(3);
864                 audio_bitrate_combo.set_active(3);
865                 video_bitrate_combo.set_active(3);
866                 audio_samplerate_combo.set_active(2);
867
868                 scale_checkbox.set_active(false);
869                 scale_checkbox.set_sensitive(false);
870         }
871         else if (p == "mpeg4") {
872                 change_file_extension(".mp4");
873                 audio_codec_combo.set_active(0);
874                 video_codec_combo.set_active(4);
875                 audio_bitrate_combo.set_active(3);
876                 video_bitrate_combo.set_active(3);
877                 if (_session->nominal_frame_rate() == 48000 || _session->nominal_frame_rate() == 96000) {
878                         audio_samplerate_combo.set_active(2);
879                 } else {
880                         audio_samplerate_combo.set_active(1);
881                 }
882         }
883
884         if (p == "none") {
885                 audio_codec_combo.set_sensitive(true);
886                 video_codec_combo.set_sensitive(true);
887                 audio_bitrate_combo.set_sensitive(true);
888                 video_bitrate_combo.set_sensitive(true);
889                 audio_samplerate_combo.set_sensitive(true);
890         } else {
891                 audio_codec_combo.set_sensitive(false);
892                 video_codec_combo.set_sensitive(false);
893                 audio_bitrate_combo.set_sensitive(false);
894                 video_bitrate_combo.set_sensitive(false);
895                 audio_samplerate_combo.set_sensitive(false);
896         }
897
898         Gtk::Table *t = (Gtk::Table*) preset_combo.get_parent();
899         Gtk::Table_Helpers::TableList c = t->children();
900         Gtk::Table_Helpers::TableList::iterator it;
901         if (p == "dvd-PAL" || p == "dvd-NTSC") {
902                 for (it = c.begin(); it != c.end(); ++it) {
903                         int row = it->get_top_attach();
904                         if (row == 2 || row == 3 || row== 5 || row== 6 || row == 9) {
905                                 it->get_widget()->hide();
906                         }
907                 }
908         } else {
909                 for (it = c.begin(); it != c.end(); ++it) {
910                         int row = it->get_top_attach();
911                         if (row == 2 || row == 3 || row== 5 || row== 6 || row == 9) {
912                                 it->get_widget()->show();
913                         }
914                 }
915         }
916
917         video_codec_combo_changed();
918 }
919
920 void
921 ExportVideoDialog::open_outfn_dialog ()
922 {
923         Gtk::FileChooserDialog dialog(_("Save Exported Video File"), Gtk::FILE_CHOOSER_ACTION_SAVE);
924         dialog.set_filename (outfn_path_entry.get_text());
925
926         dialog.add_button(Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
927         dialog.add_button(Gtk::Stock::OK, Gtk::RESPONSE_OK);
928
929         int result = dialog.run();
930
931         if (result == Gtk::RESPONSE_OK) {
932                 std::string filename = dialog.get_filename();
933
934                 if (filename.length()) {
935                         outfn_path_entry.set_text (filename);
936                 }
937         }
938 }
939
940 void
941 ExportVideoDialog::open_invid_dialog ()
942 {
943         Gtk::FileChooserDialog dialog(_("Save Exported Video File"), Gtk::FILE_CHOOSER_ACTION_SAVE);
944         dialog.set_filename (invid_path_entry.get_text());
945
946         dialog.add_button(Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
947         dialog.add_button(Gtk::Stock::OK, Gtk::RESPONSE_OK);
948
949         int result = dialog.run();
950
951         if (result == Gtk::RESPONSE_OK) {
952                 std::string filename = dialog.get_filename();
953
954                 if (filename.length()) {
955                         invid_path_entry.set_text (filename);
956                 }
957         }
958 }
959
960 #endif /* WITH_VIDEOTIMELINE */