27f728ad6dc80c1a21d20723d5eec697edf30a34
[goodguy/cinelerra.git] / cinelerra-5.1 / cinelerra / bdcreate.C
1 #include "asset.h"
2 #include "bchash.h"
3 #include "bdcreate.h"
4 #include "clip.h"
5 #include "edl.h"
6 #include "edit.h"
7 #include "edits.h"
8 #include "edlsession.h"
9 #include "file.h"
10 #include "filexml.h"
11 #include "interlacemodes.h"
12 #include "keyframe.h"
13 #include "labels.h"
14 #include "mainerror.h"
15 #include "mainundo.h"
16 #include "mwindow.h"
17 #include "mwindowgui.h"
18 #include "plugin.h"
19 #include "pluginset.h"
20 #include "preferences.h"
21 #include "rescale.h"
22 #include "track.h"
23 #include "tracks.h"
24
25 #include <unistd.h>
26 #include <fcntl.h>
27 #include <errno.h>
28 #include <sys/stat.h>
29 #include <sys/statfs.h>
30
31 // BD Creation
32
33 // selected by timezone
34 #define BD_1920x1080_2997i      0
35 #define BD_1920x1080_25i        3
36
37 static struct bd_format {
38         const char *name;
39         int w, h;
40         double framerate;
41         int wide, interlaced;
42 } bd_formats[] = {
43 // framerates are frames, not fields, per second, *=not standard
44         { "1920x1080 29.97i",   1920,1080, 29.97,  1, ILACE_MODE_TOP_FIRST },
45         { "1920x1080 29.97p*",  1920,1080, 29.97,  1, ILACE_MODE_NOTINTERLACED },
46         { "1920x1080 24p",      1920,1080, 24.,    1, ILACE_MODE_NOTINTERLACED },
47         { "1920x1080 25i",      1920,1080, 25.,    1, ILACE_MODE_TOP_FIRST },
48         { "1920x1080 23.976p",  1920,1080, 23.976, 1, ILACE_MODE_NOTINTERLACED },
49         { "1440x1080 29.97i",   1440,1080, 29.97, -1, ILACE_MODE_TOP_FIRST },
50         { "1440x1080 25i",      1440,1080, 25.,   -1, ILACE_MODE_TOP_FIRST },
51         { "1440x1080 24p",      1440,1080, 24.,   -1, ILACE_MODE_NOTINTERLACED },
52         { "1440x1080 23.976p",  1440,1080, 23.976,-1, ILACE_MODE_NOTINTERLACED },
53         { "1280x720  59.94p",   1280,720,  59.94,  1, ILACE_MODE_NOTINTERLACED },
54         { "1280x720  50p",      1280,720,  50.,    1, ILACE_MODE_NOTINTERLACED },
55         { "1280x720  24p",      1280,720,  24.,    1, ILACE_MODE_NOTINTERLACED },
56         { "1280x720  23.976p",  1280,720,  23.976, 1, ILACE_MODE_NOTINTERLACED },
57         { "720x576   25i",       720,576,  25.,    0, ILACE_MODE_BOTTOM_FIRST },
58         { "720x576   25p*",      720,576,  25.,    0, ILACE_MODE_NOTINTERLACED },
59         { "720x480   29.97i",    720,480,  29.97,  0, ILACE_MODE_BOTTOM_FIRST },
60         { "720x480   29.97p*",   720,480,  29.97,  0, ILACE_MODE_NOTINTERLACED },
61 };
62
63 const int64_t CreateBD_Thread::BD_SIZE = 25000000000;
64 const int CreateBD_Thread::BD_STREAMS = 1;
65 const int CreateBD_Thread::BD_WIDTH = 1920;
66 const int CreateBD_Thread::BD_HEIGHT = 1080;
67 const double CreateBD_Thread::BD_WIDE_ASPECT_WIDTH = 16.;
68 const double CreateBD_Thread::BD_WIDE_ASPECT_HEIGHT = 9.;
69 const double CreateBD_Thread::BD_ASPECT_WIDTH = 4.;
70 const double CreateBD_Thread::BD_ASPECT_HEIGHT = 3.;
71 const double CreateBD_Thread::BD_FRAMERATE = 24000. / 1001.;
72 //const int CreateBD_Thread::BD_MAX_BITRATE = 40000000;
73 const int CreateBD_Thread::BD_MAX_BITRATE = 10000000;
74 const int CreateBD_Thread::BD_CHANNELS = 2;
75 const int CreateBD_Thread::BD_WIDE_CHANNELS = 6;
76 const double CreateBD_Thread::BD_SAMPLERATE = 48000;
77 const double CreateBD_Thread::BD_KAUDIO_RATE = 192;
78 const int CreateBD_Thread::BD_INTERLACE_MODE = ILACE_MODE_NOTINTERLACED;
79
80 CreateBD_MenuItem::CreateBD_MenuItem(MWindow *mwindow)
81  : BC_MenuItem(_("BD Render..."), _("Ctrl-d"), 'd')
82 {
83         set_ctrl(1);
84         this->mwindow = mwindow;
85 }
86
87 int CreateBD_MenuItem::handle_event()
88 {
89         mwindow->create_bd->start();
90         return 1;
91 }
92
93
94 CreateBD_Thread::CreateBD_Thread(MWindow *mwindow)
95  : BC_DialogThread()
96 {
97         this->mwindow = mwindow;
98         this->gui = 0;
99         this->use_deinterlace = 0;
100         this->use_scale = Rescale::none;
101         this->use_histogram = 0;
102         this->use_inverse_telecine = 0;
103         this->use_wide_audio = 0;
104         this->use_resize_tracks = 0;
105         this->use_label_chapters = 0;
106
107         this->bd_size = BD_SIZE;
108         this->bd_width = BD_WIDTH;
109         this->bd_height = BD_HEIGHT;
110         this->bd_aspect_width = BD_WIDE_ASPECT_WIDTH;
111         this->bd_aspect_height = BD_WIDE_ASPECT_HEIGHT;
112         this->bd_framerate = BD_FRAMERATE;
113         this->bd_samplerate = BD_SAMPLERATE;
114         this->bd_max_bitrate = BD_MAX_BITRATE;
115         this->bd_kaudio_rate = BD_KAUDIO_RATE;
116         this->max_w = this->max_h = 0;
117 }
118
119 CreateBD_Thread::~CreateBD_Thread()
120 {
121         close_window();
122 }
123
124 int CreateBD_Thread::get_udfs_mount(char *udfs, char *mopts, char *mntpt)
125 {
126         int ret = 0;
127 // default: mount -t udf -o loop $1/bd.udfs $1/udfs
128         strcpy(udfs,"$1/bd.udfs");
129         strcpy(mopts,"-t udf -o loop $1/bd.udfs ");
130         strcpy(mntpt,"$1/udfs");
131         const char *home = getenv("HOME");
132         if( !home ) return ret;
133         FILE *fp = fopen("/etc/fstab","r");
134         if( !fp ) return ret;
135         int len = strlen(home);
136         char line[BCTEXTLEN], typ[BCTEXTLEN], file[BCTEXTLEN];
137         char mpnt[BCTEXTLEN], opts[BCTEXTLEN];
138         while( fgets(line,sizeof(line),fp) ) {
139 // search "/etc/fstab" for: $HOME/img_file $HOME/bluray udf noauto,loop,rw,user
140                 if( line[0] == '#' || line[0] == ';' ) continue;
141                 if( sscanf(line,"%s %s %s %s ", &file[0], &mpnt[0], &typ[0], &opts[0]) != 4 )
142                         continue;
143                 if( strcmp("udf", typ) ) continue;
144                 if( strncmp(home, file, len) || file[len] != '/' ) continue;
145                 if( strncmp(home, mpnt, len) ) continue;
146                 if( strcmp(&mpnt[len], "/bluray") ) continue;
147                 int loop = 0, user = 0, rw = 0, noauto = 0;
148                 char *op = opts, *ep = op + sizeof(opts)-1;
149                 while( op < ep && *op ) {
150                         char opt[BCTEXTLEN], *cp = opt;
151                         while( op < ep && *op && (*cp=*op++)!=',' ) ++cp;
152                         *cp = 0;
153                         if( !strcmp("loop", opt) ) loop = 1;
154                         else if( !strcmp("user", opt) ) user = 1;
155                         else if( !strcmp("noauto", opt) ) noauto = 1;
156                         else if( !strcmp("rw", opt) ) rw = 1;
157                 }
158                 if( loop && user && rw && noauto ) {
159                         strcpy(udfs, file);
160                         strcpy(mopts, "");
161                         strcpy(mntpt, mpnt);
162                         ret = 1;
163                         break;
164                 }
165         }
166         fclose(fp);
167         return ret;
168 }
169
170 int CreateBD_Thread::create_bd_jobs(ArrayList<BatchRenderJob*> *jobs, const char *asset_dir)
171 {
172         EDL *edl = mwindow->edl;
173         if( !edl || !edl->session ) {
174                 char msg[BCTEXTLEN];
175                 sprintf(msg, _("No EDL/Session"));
176                 MainError::show_error(msg);
177                 return 1;
178         }
179         EDLSession *session = edl->session;
180
181         double total_length = edl->tracks->total_length();
182         if( total_length <= 0 ) {
183                 char msg[BCTEXTLEN];
184                 sprintf(msg, _("No content: %s"), asset_title);
185                 MainError::show_error(msg);
186                 return 1;
187         }
188
189         if( mkdir(asset_dir, 0777) ) {
190                 char err[BCTEXTLEN], msg[BCTEXTLEN];
191                 strerror_r(errno, err, sizeof(err));
192                 sprintf(msg, _("Unable to create directory: %s\n-- %s"), asset_dir, err);
193                 MainError::show_error(msg);
194                 return 1;
195         }
196
197         double old_samplerate = session->sample_rate;
198         double old_framerate = session->frame_rate;
199
200         session->video_channels = BD_STREAMS;
201         session->video_tracks = BD_STREAMS;
202         session->frame_rate = bd_framerate;
203         session->output_w = bd_width;
204         session->output_h = bd_height;
205         session->aspect_w = bd_aspect_width;
206         session->aspect_h = bd_aspect_height;
207         session->sample_rate = bd_samplerate;
208         session->audio_channels = session->audio_tracks =
209                 use_wide_audio ? BD_WIDE_CHANNELS : BD_CHANNELS;
210         session->interlace_mode = bd_interlace_mode;
211
212         char script_filename[BCTEXTLEN];
213         sprintf(script_filename, "%s/bd.sh", asset_dir);
214         int fd = open(script_filename, O_WRONLY+O_CREAT+O_TRUNC, 0755);
215         FILE *fp = fdopen(fd, "w");
216         if( !fp ) {
217                 char err[BCTEXTLEN], msg[BCTEXTLEN];
218                 strerror_r(errno, err, sizeof(err));
219                 sprintf(msg, _("Unable to save: %s\n-- %s"), script_filename, err);
220                 MainError::show_error(msg);
221                 return 1;
222         }
223         char udfs[BCTEXTLEN], mopts[BCTEXTLEN], mntpt[BCTEXTLEN];
224         int is_usr_mnt = get_udfs_mount(udfs, mopts, mntpt);
225         const char *exec_path = File::get_cinlib_path();
226         fprintf(fp,"#!/bin/bash -ex\n");
227         fprintf(fp,"PATH=$PATH:%s\n",exec_path);
228         fprintf(fp,"mkdir -p $1/udfs\n");
229         fprintf(fp,"sz=`du -cb $1/bd.m2ts* | tail -1 | sed -e 's/[ \t].*//'`\n");
230         fprintf(fp,"blks=$((sz/2048 + 4096))\n");
231         fprintf(fp,"rm -f %s\n", udfs);
232         fprintf(fp,"mkudffs -b 2048 %s $blks\n", udfs);
233         fprintf(fp,"mount %s%s\n", mopts, mntpt);
234         fprintf(fp,"bdwrite %s $1/bd.m2ts*\n",mntpt);
235         fprintf(fp,"umount %s\n",mntpt);
236         if( is_usr_mnt )
237                 fprintf(fp,"mv -f %s $1/bd.udfs\n", udfs);
238         fprintf(fp,"echo To burn bluray, load writable media and run:\n");
239         fprintf(fp,"echo for WORM: growisofs -dvd-compat -Z /dev/bd=$1/bd.udfs\n");
240         fprintf(fp,"echo for RW:   dd if=$1/bd.udfs of=/dev/bd bs=2048000\n");
241         fprintf(fp,"kill $$\n");
242         fprintf(fp,"\n");
243         fclose(fp);
244
245         session->audio_channels = session->audio_tracks =
246                 !use_wide_audio ? BD_CHANNELS : BD_WIDE_CHANNELS;
247         for( int i=0; i<MAX_CHANNELS; ++i )
248                 session->achannel_positions[i] = default_audio_channel_position(i, session->audio_channels);
249         int audio_mapping = edl->tracks->recordable_audio_tracks() == BD_WIDE_CHANNELS &&
250                 !use_wide_audio ? MWindow::AUDIO_5_1_TO_2 : MWindow::AUDIO_1_TO_1;
251         mwindow->remap_audio(audio_mapping);
252
253         double new_samplerate = session->sample_rate;
254         double new_framerate = session->frame_rate;
255         edl->retrack();
256         edl->rechannel();
257         edl->resample(old_samplerate, new_samplerate, TRACK_AUDIO);
258         edl->resample(old_framerate, new_framerate, TRACK_VIDEO);
259
260         int64_t aud_size = ((bd_kaudio_rate * total_length)/8 + 1000-1) * 1000;
261         int64_t vid_size = bd_size*0.96 - aud_size;
262         int64_t vid_bitrate = (vid_size * 8) / total_length;
263         vid_bitrate /= 1000;  vid_bitrate *= 1000;
264         if( vid_bitrate > bd_max_bitrate ) vid_bitrate = bd_max_bitrate;
265
266         char xml_filename[BCTEXTLEN];
267         sprintf(xml_filename, "%s/bd.xml", asset_dir);
268         FileXML xml_file;
269         edl->save_xml(&xml_file, xml_filename);
270         xml_file.terminate_string();
271         if( xml_file.write_to_file(xml_filename) ) {
272                 char msg[BCTEXTLEN];
273                 sprintf(msg, _("Unable to save: %s"), xml_filename);
274                 MainError::show_error(msg);
275                 return 1;
276         }
277
278         BatchRenderJob *job = new BatchRenderJob(mwindow->preferences, use_label_chapters);
279         jobs->append(job);
280         strcpy(&job->edl_path[0], xml_filename);
281         Asset *asset = job->asset;
282
283         asset->layers = BD_STREAMS;
284         asset->frame_rate = session->frame_rate;
285         asset->width = session->output_w;
286         asset->height = session->output_h;
287         asset->aspect_ratio = session->aspect_w / session->aspect_h;
288         asset->interlace_mode = session->interlace_mode;
289
290         char option_path[BCTEXTLEN];
291         sprintf(&asset->path[0],"%s/bd.m2ts", asset_dir);
292         asset->format = FILE_FFMPEG;
293         strcpy(asset->fformat, "m2ts");
294
295         asset->audio_data = 1;
296         strcpy(asset->acodec, "bluray.m2ts");
297         //mwindow->defaults->get("DEFAULT_BLURAY_ACODEC", asset->acodec);
298         FFMPEG::set_option_path(option_path, "audio/%s", asset->acodec);
299         FFMPEG::load_options(option_path, asset->ff_audio_options,
300                          sizeof(asset->ff_audio_options));
301         asset->ff_audio_bitrate = bd_kaudio_rate * 1000;
302
303         asset->video_data = 1;
304         strcpy(asset->vcodec, "bluray.m2ts");
305         //mwindow->defaults->get("DEFAULT_BLURAY_VCODEC", asset->vcodec);
306         FFMPEG::set_option_path(option_path, "video/%s", asset->vcodec);
307         FFMPEG::load_options(option_path, asset->ff_video_options,
308                  sizeof(asset->ff_video_options));
309         const char *opts = 0;
310         switch( asset->interlace_mode ) {
311         case ILACE_MODE_TOP_FIRST:    opts = ":tff\n";  break;
312         case ILACE_MODE_BOTTOM_FIRST: opts = ":bff\n";  break;
313         }
314         if( opts ) {
315                 int len = strlen(asset->ff_video_options);
316                 char *cp = asset->ff_video_options + len-1;
317                 strncpy(cp, opts, sizeof(asset->ff_video_options)-len);
318         }
319         asset->ff_video_bitrate = vid_bitrate;
320         asset->ff_video_quality = -1;
321
322         job = new BatchRenderJob(mwindow->preferences, 0, 0);
323         jobs->append(job);
324         job->edl_path[0] = '@';
325         strcpy(&job->edl_path[1], script_filename);
326         strcpy(&job->asset->path[0], asset_dir);
327
328         return 0;
329 }
330
331 void CreateBD_Thread::handle_close_event(int result)
332 {
333         if( result ) return;
334         mwindow->defaults->update("WORK_DIRECTORY", tmp_path);
335         mwindow->batch_render->load_defaults(mwindow->defaults);
336         mwindow->undo->update_undo_before();
337         KeyFrame keyframe;  char data[BCTEXTLEN];
338         if( use_deinterlace ) {
339                 sprintf(data,"<DEINTERLACE MODE=1>");
340                 keyframe.set_data(data);
341                 insert_video_plugin("Deinterlace", &keyframe);
342         }
343         if( use_inverse_telecine ) {
344                 sprintf(data,"<IVTC FRAME_OFFSET=0 FIRST_FIELD=0 "
345                         "AUTOMATIC=1 AUTO_THRESHOLD=2.0e+00 PATTERN=2>");
346                 keyframe.set_data(data);
347                 insert_video_plugin("Inverse Telecine", &keyframe);
348         }
349         if( use_scale != Rescale::none ) {
350                 double bd_aspect = bd_aspect_height > 0 ? bd_aspect_width / bd_aspect_height : 1;
351
352                 Tracks *tracks = mwindow->edl->tracks;
353                 for( Track *vtrk=tracks->first; vtrk; vtrk=vtrk->next ) {
354                         if( vtrk->data_type != TRACK_VIDEO ) continue;
355                         if( !vtrk->record ) continue;
356                         vtrk->expand_view = 1;
357                         PluginSet *plugin_set = new PluginSet(mwindow->edl, vtrk);
358                         vtrk->plugin_set.append(plugin_set);
359                         Edits *edits = vtrk->edits;
360                         for( Edit *edit=edits->first; edit; edit=edit->next ) {
361                                 Indexable *indexable = edit->get_source();
362                                 if( !indexable ) continue;
363                                 Rescale in(indexable);
364                                 Rescale out(bd_width, bd_height, bd_aspect);
365                                 float src_w, src_h, dst_w, dst_h;
366                                 in.rescale(out,use_scale, src_w,src_h, dst_w,dst_h);
367                                 sprintf(data,"<SCALERATIO TYPE=%d"
368                                         " IN_W=%d IN_H=%d IN_ASPECT_RATIO=%f"
369                                         " OUT_W=%d OUT_H=%d OUT_ASPECT_RATIO=%f"
370                                         " SRC_X=%f SRC_Y=%f SRC_W=%f SRC_H=%f"
371                                         " DST_X=%f DST_Y=%f DST_W=%f DST_H=%f>", use_scale,
372                                         in.w, in.h, in.aspect, out.w, out.h, out.aspect,
373                                         0., 0., src_w, src_h, 0., 0., dst_w, dst_h);
374                                 keyframe.set_data(data);
375                                 plugin_set->insert_plugin(_("Scale Ratio"),
376                                         edit->startproject, edit->length,
377                                         PLUGIN_STANDALONE, 0, &keyframe, 0);
378                         }
379                         vtrk->optimize();
380                 }
381         }
382         if( use_resize_tracks )
383                 resize_tracks();
384         if( use_histogram ) {
385 #if 0
386                 sprintf(data, "<HISTOGRAM OUTPUT_MIN_0=0 OUTPUT_MAX_0=1 "
387                         "OUTPUT_MIN_1=0 OUTPUT_MAX_1=1 "
388                         "OUTPUT_MIN_2=0 OUTPUT_MAX_2=1 "
389                         "OUTPUT_MIN_3=0 OUTPUT_MAX_3=1 "
390                         "AUTOMATIC=0 THRESHOLD=9.0-01 PLOT=0 SPLIT=0>"
391                         "<POINTS></POINTS><POINTS></POINTS><POINTS></POINTS>"
392                         "<POINTS><POINT X=6.0e-02 Y=0>"
393                                 "<POINT X=9.4e-01 Y=1></POINTS>");
394 #else
395                 sprintf(data, "<HISTOGRAM AUTOMATIC=0 THRESHOLD=1.0e-01 "
396                         "PLOT=0 SPLIT=0 W=440 H=500 PARADE=0 MODE=3 "
397                         "LOW_OUTPUT_0=0 HIGH_OUTPUT_0=1 LOW_INPUT_0=0 HIGH_INPUT_0=1 GAMMA_0=1 "
398                         "LOW_OUTPUT_1=0 HIGH_OUTPUT_1=1 LOW_INPUT_1=0 HIGH_INPUT_1=1 GAMMA_1=1 "
399                         "LOW_OUTPUT_2=0 HIGH_OUTPUT_2=1 LOW_INPUT_2=0 HIGH_INPUT_2=1 GAMMA_2=1 "
400                         "LOW_OUTPUT_3=0 HIGH_OUTPUT_3=1 LOW_INPUT_3=0.044 HIGH_INPUT_3=0.956 "
401                         "GAMMA_3=1>");
402 #endif
403                 keyframe.set_data(data);
404                 insert_video_plugin("Histogram", &keyframe);
405         }
406
407         char asset_dir[BCTEXTLEN], jobs_path[BCTEXTLEN];
408         snprintf(asset_dir, sizeof(asset_dir), "%s/%s", tmp_path, asset_title);
409         snprintf(jobs_path, sizeof(jobs_path), "%s/bd.jobs", asset_dir);
410         mwindow->batch_render->reset(jobs_path);
411         int ret = create_bd_jobs(&mwindow->batch_render->jobs, asset_dir);
412         mwindow->undo->update_undo_after(_("create bd"), LOAD_ALL);
413         mwindow->resync_guis();
414         if( ret ) return;
415         mwindow->batch_render->save_jobs();
416         mwindow->batch_render->start();
417 }
418
419 BC_Window* CreateBD_Thread::new_gui()
420 {
421         strcpy(tmp_path, "/tmp");
422         mwindow->defaults->get("WORK_DIRECTORY", tmp_path);
423         memset(asset_title,0,sizeof(asset_title));
424         time_t dt;      time(&dt);
425         struct tm dtm;  localtime_r(&dt, &dtm);
426         sprintf(asset_title, "bd_%02d%02d%02d-%02d%02d%02d",
427                 dtm.tm_year+1900, dtm.tm_mon+1, dtm.tm_mday,
428                 dtm.tm_hour, dtm.tm_min, dtm.tm_sec);
429         use_deinterlace = 0;
430         use_scale = Rescale::none;
431         use_histogram = 0;
432         use_inverse_telecine = 0;
433         use_wide_audio = 0;
434         use_resize_tracks = 0;
435         use_label_chapters = 0;
436         use_standard = !strcmp(mwindow->default_std(),"NTSC") ?
437                  BD_1920x1080_2997i : BD_1920x1080_25i;
438         bd_size = BD_SIZE;
439         bd_width = BD_WIDTH;
440         bd_height = BD_HEIGHT;
441         bd_aspect_width = BD_WIDE_ASPECT_WIDTH;
442         bd_aspect_height = BD_WIDE_ASPECT_HEIGHT;
443         bd_framerate = BD_FRAMERATE;
444         bd_samplerate = BD_SAMPLERATE;
445         bd_max_bitrate = BD_MAX_BITRATE;
446         bd_kaudio_rate = BD_KAUDIO_RATE;
447         bd_interlace_mode = BD_INTERLACE_MODE;
448         max_w = 0; max_h = 0;
449
450         int has_standard = -1;
451         if( mwindow->edl ) {
452                 EDLSession *session = mwindow->edl->session;
453 // match the session to any known standard
454                 for( int i=0; i<(int)(sizeof(bd_formats)/sizeof(bd_formats[0])); ++i ) {
455                         if( !EQUIV(session->frame_rate, bd_formats[i].framerate) ) continue;
456                         if( session->output_w != bd_formats[i].w ) continue;
457                         if( session->output_h != bd_formats[i].h ) continue;
458                         has_standard = i;  break;
459                 }
460         }
461         if( has_standard >= 0 )
462                 use_standard = has_standard;
463
464         option_presets();
465         int scr_x = mwindow->gui->get_screen_x(0, -1);
466         int scr_w = mwindow->gui->get_screen_w(0, -1);
467         int scr_h = mwindow->gui->get_screen_h(0, -1);
468         int w = 500, h = 280;
469         int x = scr_x + scr_w/2 - w/2, y = scr_h/2 - h/2;
470
471         gui = new CreateBD_GUI(this, x, y, w, h);
472         gui->create_objects();
473
474         if( getuid() != 0 ) {
475                 mwindow->show_warning(
476                         &mwindow->preferences->bd_warn_root,
477                         _("Must be root to mount UDFS images\n"));
478         }
479         return gui;
480 }
481
482
483 CreateBD_OK::CreateBD_OK(CreateBD_GUI *gui, int x, int y)
484  : BC_OKButton(x, y)
485 {
486         this->gui = gui;
487         set_tooltip(_("end setup, start batch render"));
488 }
489
490 CreateBD_OK::~CreateBD_OK()
491 {
492 }
493
494 int CreateBD_OK::button_press_event()
495 {
496         if(get_buttonpress() == 1 && is_event_win() && cursor_inside()) {
497                 gui->set_done(0);
498                 return 1;
499         }
500         return 0;
501 }
502
503 int CreateBD_OK::keypress_event()
504 {
505         return 0;
506 }
507
508
509 CreateBD_Cancel::CreateBD_Cancel(CreateBD_GUI *gui, int x, int y)
510  : BC_CancelButton(x, y)
511 {
512         this->gui = gui;
513 }
514
515 CreateBD_Cancel::~CreateBD_Cancel()
516 {
517 }
518
519 int CreateBD_Cancel::button_press_event()
520 {
521         if(get_buttonpress() == 1 && is_event_win() && cursor_inside()) {
522                 gui->set_done(1);
523                 return 1;
524         }
525         return 0;
526 }
527
528
529 CreateBD_DiskSpace::CreateBD_DiskSpace(CreateBD_GUI *gui, int x, int y)
530  : BC_Title(x, y, "", MEDIUMFONT, GREEN)
531 {
532         this->gui = gui;
533 }
534
535 CreateBD_DiskSpace::~CreateBD_DiskSpace()
536 {
537 }
538
539 int64_t CreateBD_DiskSpace::tmp_path_space()
540 {
541         const char *path = gui->thread->tmp_path;
542         if( access(path,R_OK+W_OK) ) return 0;
543         struct statfs sfs;
544         if( statfs(path, &sfs) ) return 0;
545         return (int64_t)sfs.f_bsize * sfs.f_bfree;
546 }
547
548 void CreateBD_DiskSpace::update()
549 {
550         static const char *suffix[] = { "", "KB", "MB", "GB", "TB", "PB" };
551         int64_t disk_space = tmp_path_space();
552         double media_size = 100e9, msz = 0, m = 1;
553         char sfx[BCSTRLEN];
554         if( sscanf(gui->media_size->get_text(), "%lf%s", &msz, sfx) == 2 ) {
555                 int i = sizeof(suffix)/sizeof(suffix[0]);
556                 while( --i >= 0 && strcmp(sfx, suffix[i]) );
557                 while( --i >= 0 ) m *= 1000;
558                 media_size = msz * m;
559         }
560         int color = disk_space < media_size*2 ? RED : GREEN;
561         int i = 0;
562         for( int64_t space=disk_space; i<5 && (space/=1000)>0; disk_space=space, ++i );
563         char text[BCTEXTLEN];
564         sprintf(text, "%s%3jd%s", _("disk space: "), disk_space, suffix[i]);
565         gui->disk_space->BC_Title::update(text);
566         gui->disk_space->set_color(color);
567 }
568
569 CreateBD_TmpPath::CreateBD_TmpPath(CreateBD_GUI *gui, int x, int y, int w)
570  : BC_TextBox(x, y, w, 1, -(int)sizeof(gui->thread->tmp_path),
571                 gui->thread->tmp_path, 1, MEDIUMFONT)
572 {
573         this->gui = gui;
574 }
575
576 CreateBD_TmpPath::~CreateBD_TmpPath()
577 {
578 }
579
580 int CreateBD_TmpPath::handle_event()
581 {
582         get_text();
583         gui->disk_space->update();
584         return 1;
585 }
586
587
588 CreateBD_AssetTitle::CreateBD_AssetTitle(CreateBD_GUI *gui, int x, int y, int w)
589  : BC_TextBox(x, y, w, 1, -(int)sizeof(gui->thread->asset_title),
590                 gui->thread->asset_title, 1, MEDIUMFONT)
591 {
592         this->gui = gui;
593 }
594
595 CreateBD_AssetTitle::~CreateBD_AssetTitle()
596 {
597 }
598
599 int CreateBD_AssetTitle::handle_event()
600 {
601         get_text();
602         return 1;
603 }
604
605 CreateBD_Deinterlace::CreateBD_Deinterlace(CreateBD_GUI *gui, int x, int y)
606  : BC_CheckBox(x, y, &gui->thread->use_deinterlace, _("Deinterlace"))
607 {
608         this->gui = gui;
609 }
610
611 CreateBD_Deinterlace::~CreateBD_Deinterlace()
612 {
613 }
614
615 int CreateBD_Deinterlace::handle_event()
616 {
617         if( get_value() ) {
618                 gui->need_inverse_telecine->set_value(0);
619                 gui->thread->use_inverse_telecine = 0;
620         }
621         return BC_CheckBox::handle_event();
622 }
623
624
625 CreateBD_InverseTelecine::CreateBD_InverseTelecine(CreateBD_GUI *gui, int x, int y)
626  : BC_CheckBox(x, y, &gui->thread->use_inverse_telecine, _("Inverse Telecine"))
627 {
628         this->gui = gui;
629 }
630
631 CreateBD_InverseTelecine::~CreateBD_InverseTelecine()
632 {
633 }
634
635 int CreateBD_InverseTelecine::handle_event()
636 {
637         if( get_value() ) {
638                 gui->need_deinterlace->set_value(0);
639                 gui->thread->use_deinterlace = 0;
640         }
641         return BC_CheckBox::handle_event();
642 }
643
644
645 CreateBD_ResizeTracks::CreateBD_ResizeTracks(CreateBD_GUI *gui, int x, int y)
646  : BC_CheckBox(x, y, &gui->thread->use_resize_tracks, _("Resize Tracks"))
647 {
648         this->gui = gui;
649 }
650
651 CreateBD_ResizeTracks::~CreateBD_ResizeTracks()
652 {
653 }
654
655
656 CreateBD_Histogram::CreateBD_Histogram(CreateBD_GUI *gui, int x, int y)
657  : BC_CheckBox(x, y, &gui->thread->use_histogram, _("Histogram"))
658 {
659         this->gui = gui;
660 }
661
662 CreateBD_Histogram::~CreateBD_Histogram()
663 {
664 }
665
666 CreateBD_LabelChapters::CreateBD_LabelChapters(CreateBD_GUI *gui, int x, int y)
667  : BC_CheckBox(x, y, &gui->thread->use_label_chapters, _("Chapters at Labels"))
668 {
669         this->gui = gui;
670 }
671
672 CreateBD_LabelChapters::~CreateBD_LabelChapters()
673 {
674 }
675
676 CreateBD_WideAudio::CreateBD_WideAudio(CreateBD_GUI *gui, int x, int y)
677  : BC_CheckBox(x, y, &gui->thread->use_wide_audio, _("Audio 5.1"))
678 {
679         this->gui = gui;
680 }
681
682 CreateBD_WideAudio::~CreateBD_WideAudio()
683 {
684 }
685
686
687 CreateBD_GUI::CreateBD_GUI(CreateBD_Thread *thread, int x, int y, int w, int h)
688  : BC_Window(_(PROGRAM_NAME ": Create BD"), x, y, w, h, 50, 50, 1, 0, 1)
689 {
690         this->thread = thread;
691         at_x = at_y = tmp_x = tmp_y = 0;
692         ok_x = ok_y = ok_w = ok_h = 0;
693         cancel_x = cancel_y = cancel_w = cancel_h = 0;
694         asset_title = 0;
695         tmp_path = 0;
696         btmp_path = 0;
697         disk_space = 0;
698         standard = 0;
699         scale = 0;
700         need_deinterlace = 0;
701         need_inverse_telecine = 0;
702         need_resize_tracks = 0;
703         need_histogram = 0;
704         non_standard = 0;
705         need_wide_audio = 0;
706         need_label_chapters = 0;
707         ok = 0;
708         cancel = 0;
709 }
710
711 CreateBD_GUI::~CreateBD_GUI()
712 {
713 }
714
715 void CreateBD_GUI::create_objects()
716 {
717         lock_window("CreateBD_GUI::create_objects");
718         int pady = BC_TextBox::calculate_h(this, MEDIUMFONT, 0, 1) + 5;
719         int padx = BC_Title::calculate_w(this, (char*)"X", MEDIUMFONT);
720         int x = padx/2, y = pady/2;
721         BC_Title *title = new BC_Title(x, y, _("Title:"), MEDIUMFONT, YELLOW);
722         add_subwindow(title);
723         at_x = x + title->get_w();  at_y = y;
724         asset_title = new CreateBD_AssetTitle(this, at_x, at_y, get_w()-at_x-10);
725         add_subwindow(asset_title);
726         y += title->get_h() + pady/2;
727         title = new BC_Title(x, y, _("Work path:"), MEDIUMFONT, YELLOW);
728         add_subwindow(title);
729         tmp_x = x + title->get_w();  tmp_y = y;
730         tmp_path = new CreateBD_TmpPath(this, tmp_x, tmp_y,  get_w()-tmp_x-35);
731         add_subwindow(tmp_path);
732         btmp_path = new BrowseButton(thread->mwindow->theme, this, tmp_path,
733                 tmp_x+tmp_path->get_w(), tmp_y, "/tmp",
734                 _("Work path"), _("Select a Work directory:"), 1);
735         add_subwindow(btmp_path);
736         y += title->get_h() + pady/2;
737         disk_space = new CreateBD_DiskSpace(this, x, y);
738         add_subwindow(disk_space);
739         int x0 = get_w() - 170;
740         title = new BC_Title(x0, y, _("Media:"), MEDIUMFONT, YELLOW);
741         add_subwindow(title);
742         int x1 =  x0+title->get_w()+padx;
743         media_size = new CreateBD_MediaSize(this, x1, y);
744         media_size->create_objects();
745         media_sizes.append(new BC_ListBoxItem("25GB"));
746         media_sizes.append(new BC_ListBoxItem("50GB"));
747         media_size->update_list(&media_sizes);
748         media_size->update(media_sizes[0]->get_text());
749         disk_space->update();
750         y += disk_space->get_h() + pady/2;
751         title = new BC_Title(x, y, _("Format:"), MEDIUMFONT, YELLOW);
752         add_subwindow(title);
753         standard = new CreateBD_Format(this, title->get_w() + padx, y);
754         add_subwindow(standard);
755         standard->create_objects();
756         standard->set_text(bd_formats[thread->use_standard].name);
757         x0 -= 30;
758         title = new BC_Title(x0, y, _("Scale:"), MEDIUMFONT, YELLOW);
759         add_subwindow(title);
760         x1 = x0+title->get_w()+padx;
761         scale = new CreateBD_Scale(this, x1, y);
762         add_subwindow(scale);
763         scale->create_objects();
764         y += standard->get_h() + pady/2;
765         need_deinterlace = new CreateBD_Deinterlace(this, x, y);
766         add_subwindow(need_deinterlace);
767         x1 = x + 170; //, x2 = x1 + 150;
768         need_inverse_telecine = new CreateBD_InverseTelecine(this, x1, y);
769         add_subwindow(need_inverse_telecine);
770         y += need_deinterlace->get_h() + pady/2;
771         need_histogram = new CreateBD_Histogram(this, x, y);
772         add_subwindow(need_histogram);
773         need_wide_audio = new CreateBD_WideAudio(this, x1, y);
774         add_subwindow(need_wide_audio);
775         y += need_histogram->get_h() + pady/2;
776         non_standard = new BC_Title(x, y+5, "", MEDIUMFONT, RED);
777         add_subwindow(non_standard);
778         need_resize_tracks = new CreateBD_ResizeTracks(this, x1, y);
779         add_subwindow(need_resize_tracks);
780 //      need_label_chapters = new CreateBD_LabelChapters(this, x2, y);
781 //      add_subwindow(need_label_chapters);
782         ok_w = BC_OKButton::calculate_w();
783         ok_h = BC_OKButton::calculate_h();
784         ok_x = 10;
785         ok_y = get_h() - ok_h - 10;
786         ok = new CreateBD_OK(this, ok_x, ok_y);
787         add_subwindow(ok);
788         cancel_w = BC_CancelButton::calculate_w();
789         cancel_h = BC_CancelButton::calculate_h();
790         cancel_x = get_w() - cancel_w - 10,
791         cancel_y = get_h() - cancel_h - 10;
792         cancel = new CreateBD_Cancel(this, cancel_x, cancel_y);
793         add_subwindow(cancel);
794         show_window();
795         unlock_window();
796 }
797
798 int CreateBD_GUI::resize_event(int w, int h)
799 {
800         asset_title->reposition_window(at_x, at_y, get_w()-at_x-10);
801         tmp_path->reposition_window(tmp_x, tmp_y,  get_w()-tmp_x-35);
802         btmp_path->reposition_window(tmp_x+tmp_path->get_w(), tmp_y);
803         ok_y = h - ok_h - 10;
804         ok->reposition_window(ok_x, ok_y);
805         cancel_x = w - cancel_w - 10,
806         cancel_y = h - cancel_h - 10;
807         cancel->reposition_window(cancel_x, cancel_y);
808         return 0;
809 }
810
811 int CreateBD_GUI::translation_event()
812 {
813         return 1;
814 }
815
816 int CreateBD_GUI::close_event()
817 {
818         set_done(1);
819         return 1;
820 }
821
822 void CreateBD_GUI::update()
823 {
824         scale->set_value(thread->use_scale);
825         need_deinterlace->set_value(thread->use_deinterlace);
826         need_inverse_telecine->set_value(thread->use_inverse_telecine);
827         need_resize_tracks->set_value(thread->use_resize_tracks);
828         need_histogram->set_value(thread->use_histogram);
829         need_wide_audio->set_value(thread->use_wide_audio);
830 //      need_label_chapters->set_value(thread->use_label_chapters);
831 }
832
833 int CreateBD_Thread::
834 insert_video_plugin(const char *title, KeyFrame *default_keyframe)
835 {
836         Tracks *tracks = mwindow->edl->tracks;
837         for( Track *vtrk=tracks->first; vtrk; vtrk=vtrk->next ) {
838                 if( vtrk->data_type != TRACK_VIDEO ) continue;
839                 if( !vtrk->record ) continue;
840                 vtrk->expand_view = 1;
841                 PluginSet *plugin_set = new PluginSet(mwindow->edl, vtrk);
842                 vtrk->plugin_set.append(plugin_set);
843                 Edits *edits = vtrk->edits;
844                 for( Edit *edit=edits->first; edit; edit=edit->next ) {
845                         plugin_set->insert_plugin(_(title),
846                                 edit->startproject, edit->length,
847                                 PLUGIN_STANDALONE, 0, default_keyframe, 0);
848                 }
849                 vtrk->optimize();
850         }
851         return 0;
852 }
853
854 int CreateBD_Thread::
855 resize_tracks()
856 {
857         Tracks *tracks = mwindow->edl->tracks;
858         int trk_w = max_w, trk_h = max_h;
859         if( trk_w < bd_width ) trk_w = bd_width;
860         if( trk_h < bd_height ) trk_h = bd_height;
861         for( Track *vtrk=tracks->first; vtrk; vtrk=vtrk->next ) {
862                 if( vtrk->data_type != TRACK_VIDEO ) continue;
863                 if( !vtrk->record ) continue;
864                 vtrk->track_w = trk_w;
865                 vtrk->track_h = trk_h;
866         }
867         return 0;
868 }
869
870 int CreateBD_Thread::
871 option_presets()
872 {
873 // reset only probed options
874         use_deinterlace = 0;
875         use_scale = Rescale::none;
876         use_resize_tracks = 0;
877         use_wide_audio = 0;
878         use_label_chapters = 0;
879
880         if( !mwindow->edl ) return 1;
881
882         bd_width = bd_formats[use_standard].w;
883         bd_height = bd_formats[use_standard].h;
884         bd_framerate = bd_formats[use_standard].framerate;
885         int wide = bd_formats[use_standard].wide;
886         bd_aspect_width  = wide < 0 ? 1. :
887                 wide > 0 ? BD_WIDE_ASPECT_WIDTH  : BD_ASPECT_WIDTH;
888         bd_aspect_height = wide < 0 ? 1. :
889                 wide > 0 ? BD_WIDE_ASPECT_HEIGHT : BD_ASPECT_HEIGHT;
890         bd_interlace_mode = bd_formats[use_standard].interlaced;
891         double bd_aspect = bd_aspect_width / bd_aspect_height;
892
893         Tracks *tracks = mwindow->edl->tracks;
894         max_w = 0;  max_h = 0;
895         int has_deinterlace = 0, has_scale = 0;
896         for( Track *trk=tracks->first; trk; trk=trk->next ) {
897                 if( !trk->record ) continue;
898                 Edits *edits = trk->edits;
899                 switch( trk->data_type ) {
900                 case TRACK_VIDEO:
901                         for( Edit *edit=edits->first; edit; edit=edit->next ) {
902                                 if( edit->silence() ) continue;
903                                 Indexable *indexable = edit->get_source();
904                                 int w = indexable->get_w();
905                                 if( w > max_w ) max_w = w;
906                                 if( w != bd_width ) use_scale = Rescale::scaled;
907                                 int h = indexable->get_h();
908                                 if( h > max_h ) max_h = h;
909                                 if( h != bd_height ) use_scale = Rescale::scaled;
910                                 float aw, ah;
911                                 MWindow::create_aspect_ratio(aw, ah, w, h);
912                                 double aspect = ah > 0 ? aw / ah : 1;
913                                 if( wide >= 0 && !EQUIV(aspect, bd_aspect) )
914                                         use_scale = Rescale::scaled;
915                         }
916                         for( int i=0; i<trk->plugin_set.size(); ++i ) {
917                                 for( Plugin *plugin = (Plugin*)trk->plugin_set[i]->first;
918                                                 plugin; plugin=(Plugin*)plugin->next ) {
919                                         if( !strcmp(plugin->title, "Deinterlace") )
920                                                 has_deinterlace = 1;
921                                         if( !strcmp(plugin->title, "Auto Scale") ||
922                                             !strcmp(plugin->title, "Scale Ratio") ||
923                                             !strcmp(plugin->title, "Scale") )
924                                                 has_scale = 1;
925                                 }
926                         }
927                         break;
928                 }
929         }
930         if( has_scale )
931                 use_scale = Rescale::none;
932         if( use_scale != Rescale::none ) {
933                 if( max_w != bd_width ) use_resize_tracks = 1;
934                 if( max_h != bd_height ) use_resize_tracks = 1;
935         }
936         for( Track *trk=tracks->first; trk && !use_resize_tracks; trk=trk->next ) {
937                 if( !trk->record ) continue;
938                 switch( trk->data_type ) {
939                 case TRACK_VIDEO:
940                         if( trk->track_w != max_w ) use_resize_tracks = 1;
941                         if( trk->track_h != max_h ) use_resize_tracks = 1;
942                         break;
943                 }
944         }
945         if( !has_deinterlace && max_h > 2*bd_height ) use_deinterlace = 1;
946         // Labels *labels = mwindow->edl->labels;
947         // use_label_chapters = labels && labels->first ? 1 : 0;
948
949         if( tracks->recordable_audio_tracks() == BD_WIDE_CHANNELS )
950                 use_wide_audio = 1;
951
952         return 0;
953 }
954
955
956 CreateBD_FormatItem::CreateBD_FormatItem(CreateBD_Format *popup,
957                 int standard, const char *name)
958  : BC_MenuItem(name)
959 {
960         this->popup = popup;
961         this->standard = standard;
962 }
963
964 CreateBD_FormatItem::~CreateBD_FormatItem()
965 {
966 }
967
968 int CreateBD_FormatItem::handle_event()
969 {
970         const char *text = get_text();
971         int standard = this->standard;
972         if( standard < 0 ) {
973                 BC_SubMenu *submenu = get_submenu();
974                 CreateBD_FormatItem *sub_item0 =
975                         (CreateBD_FormatItem *)submenu->get_item(0);
976                 standard = sub_item0->standard;
977                 text = sub_item0->get_text();
978         }
979         popup->gui->thread->use_standard = standard;
980         popup->set_text(text);
981         int n = strlen(text)-1;
982         int not_standard = n >= 0 && text[n] == '*' ? 1 : 0;
983         popup->gui->non_standard->update(not_standard ? _("* non-standard format") : "", 0);
984         return popup->handle_event();
985 }
986
987
988 CreateBD_Format::CreateBD_Format(CreateBD_GUI *gui, int x, int y)
989  : BC_PopupMenu(x, y, 180, bd_formats[gui->thread->use_standard].name, 1)
990 {
991         this->gui = gui;
992 }
993
994 CreateBD_Format::~CreateBD_Format()
995 {
996 }
997
998 void CreateBD_Format::create_objects()
999 {
1000         BC_SubMenu *submenu = 0;
1001         CreateBD_FormatItem *item = 0;
1002         int ww = 0, hh = 0;
1003         for( int i=0; i<(int)(sizeof(bd_formats)/sizeof(bd_formats[0])); ++i ) {
1004                 if( ww != bd_formats[i].w || hh != bd_formats[i].h ) {
1005                         ww = bd_formats[i].w;  hh = bd_formats[i].h;
1006                         char string[BCSTRLEN];
1007                         sprintf(string, "%dx%d", ww,hh);
1008                         add_item(item = new CreateBD_FormatItem(this, -1, string));
1009                         item->add_submenu(submenu = new BC_SubMenu());
1010                 }
1011                 submenu->add_submenuitem(new CreateBD_FormatItem(this, i, bd_formats[i].name));
1012         }
1013 }
1014
1015 int CreateBD_Format::handle_event()
1016 {
1017         gui->thread->option_presets();
1018         gui->update();
1019         return 1;
1020 }
1021
1022
1023 CreateBD_ScaleItem::CreateBD_ScaleItem(CreateBD_Scale *popup,
1024                 int scale, const char *text)
1025  : BC_MenuItem(text)
1026 {
1027         this->popup = popup;
1028         this->scale = scale;
1029 }
1030
1031 CreateBD_ScaleItem::~CreateBD_ScaleItem()
1032 {
1033 }
1034
1035 int CreateBD_ScaleItem::handle_event()
1036 {
1037         popup->gui->thread->use_scale = scale;
1038         popup->set_value(scale);
1039         return popup->handle_event();
1040 }
1041
1042
1043 CreateBD_Scale::CreateBD_Scale(CreateBD_GUI *gui, int x, int y)
1044  : BC_PopupMenu(x, y, 100, "", 1)
1045 {
1046         this->gui = gui;
1047 }
1048
1049 CreateBD_Scale::~CreateBD_Scale()
1050 {
1051 }
1052
1053 void CreateBD_Scale::create_objects()
1054 {
1055
1056         for( int i=0; i<(int)Rescale::n_scale_types; ++i ) {
1057                 add_item(new CreateBD_ScaleItem(this, i, Rescale::scale_types[i]));
1058         }
1059         set_value(gui->thread->use_scale);
1060 }
1061
1062 int CreateBD_Scale::handle_event()
1063 {
1064         gui->update();
1065         return 1;
1066 }
1067
1068
1069 CreateBD_MediaSize::CreateBD_MediaSize(CreateBD_GUI *gui, int x, int y)
1070  : BC_PopupTextBox(gui, 0, 0, x, y, 70,50)
1071 {
1072         this->gui = gui;
1073 }
1074
1075 CreateBD_MediaSize::~CreateBD_MediaSize()
1076 {
1077 }
1078
1079 int CreateBD_MediaSize::handle_event()
1080 {
1081         gui->disk_space->update();
1082         return 1;
1083 }
1084