4 * Copyright (C) 2015 Adam Williams <broadcast at earthling dot net>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 #include "bcsignals.h"
26 #include "confirmsave.h"
29 #include "edlsession.h"
32 #include "filesystem.h"
33 #include "formattools.h"
35 #include "mainerror.h"
36 #include "mainprogress.h"
40 #include "mwindowgui.h"
41 #include "overlayframe.h"
42 #include "preferences.h"
44 #include "renderengine.h"
46 #include "transportque.h"
53 ProxyMenuItem::ProxyMenuItem(MWindow *mwindow)
54 : BC_MenuItem(_("Proxy settings..."), _("Alt-r"), 'r')
56 this->mwindow = mwindow;
60 ProxyMenuItem::~ProxyMenuItem()
65 void ProxyMenuItem::create_objects()
67 dialog = new ProxyDialog(mwindow);
70 int ProxyMenuItem::handle_event()
72 mwindow->gui->unlock_window();
74 mwindow->gui->lock_window("ProxyMenuItem::handle_event");
79 ProxyDialog::ProxyDialog(MWindow *mwindow)
81 this->mwindow = mwindow;
84 bzero(size_text, sizeof(char*) * MAX_SIZES);
85 bzero(size_factors, sizeof(int) * MAX_SIZES);
89 ProxyDialog::~ProxyDialog()
92 for( int i=0; i<MAX_SIZES; ++i ) delete [] size_text[i];
96 BC_Window* ProxyDialog::new_gui()
98 asset->format = FILE_FFMPEG;
99 asset->load_defaults(mwindow->defaults, "PROXY_", 1, 1, 0, 0, 0);
100 mwindow->gui->lock_window("ProxyDialog::new_gui");
102 mwindow->gui->get_abs_cursor(cx, cy);
103 gui = new ProxyWindow(mwindow, this, cx - WIDTH/2, cy - HEIGHT/2);
104 gui->create_objects();
105 mwindow->gui->unlock_window();
109 void ProxyDialog::scale_to_text(char *string, int scale)
111 strcpy(string, size_text[0]);
112 for( int i = 0; i < total_sizes; i++ ) {
113 if( scale == size_factors[i] ) {
114 strcpy(string, size_text[i]);
121 void ProxyDialog::calculate_sizes()
123 for( int i=1; i<total_sizes; ++i ) {
124 delete [] size_text[i];
130 int orig_w = mwindow->edl->session->output_w * orig_scale;
131 int orig_h = mwindow->edl->session->output_h * orig_scale;
134 // w,h should stay even for yuv
135 int ow = orig_w, oh = orig_h;
136 if( BC_CModels::is_yuv(mwindow->edl->session->color_model) ) {
139 for( int i=2; i<MAX_SCALE; ++i ) {
140 if( (ow % i) != 0 ) continue;
141 if( (oh % i) != 0 ) continue;
142 size_factors[total_sizes++] = i;
146 size_factors[total_sizes++] = 2; size_factors[total_sizes++] = 3;
147 size_factors[total_sizes++] = 8; size_factors[total_sizes++] = 12;
148 size_factors[total_sizes++] = 16; size_factors[total_sizes++] = 24;
149 size_factors[total_sizes++] = 32;
151 for( int i=1; i<total_sizes; ++i ) {
152 char string[BCTEXTLEN];
153 sprintf(string, "1/%d", size_factors[i]);
154 size_text[i] = cstrdup(string);
158 void ProxyDialog::handle_close_event(int result)
160 asset->save_defaults(mwindow->defaults, "PROXY_", 1, 1, 0, 0, 0);
166 void ProxyDialog::to_proxy()
168 ArrayList<Indexable*> orig_idxbls;
169 ArrayList<Indexable*> proxy_assets;
171 EDL *edl = mwindow->edl;
172 mwindow->edl->Garbage::add_user();
173 mwindow->save_backup();
174 mwindow->undo->update_undo_before(_("proxy"), this);
175 ProxyRender proxy_render(mwindow, asset);
177 // revert project to original size from current size
178 // remove all session proxy assets at the at the current proxy_scale
179 int proxy_scale = edl->session->proxy_scale;
180 if( proxy_scale > 1 ) {
181 Asset *orig_asset = edl->assets->first;
182 for( ; orig_asset; orig_asset=orig_asset->next ) {
183 char new_path[BCTEXTLEN];
184 proxy_render.to_proxy_path(new_path, orig_asset, proxy_scale);
185 // test if proxy asset was already added to proxy_assets
187 for( int i = 0; !got_it && i<proxy_assets.size(); ++i )
188 got_it = !strcmp(proxy_assets[i]->path, new_path);
189 if( got_it ) continue;
190 Asset *proxy_asset = edl->assets->get_asset(new_path);
191 if( !proxy_asset ) continue;
192 // add pointer to existing EDL asset if it exists
193 // EDL won't delete it unless it's the same pointer.
194 proxy_assets.append(proxy_asset);
195 proxy_asset->add_user();
196 orig_idxbls.append(orig_asset);
197 orig_asset->add_user();
200 // convert from the proxy assets to the original assets
201 int proxy_auto_scale = edl->session->proxy_auto_scale;
202 mwindow->set_proxy(0, 1, proxy_auto_scale, &proxy_assets, &orig_idxbls);
204 // remove the references
205 for( int i=0; i<proxy_assets.size(); ++i ) {
206 Asset *proxy = (Asset *) proxy_assets[i];
207 proxy->width = proxy->actual_width;
208 proxy->height = proxy->actual_height;
209 proxy->remove_user();
211 proxy_assets.remove_all();
212 for( int i = 0; i < orig_idxbls.size(); i++ )
213 orig_idxbls[i]->remove_user();
214 orig_idxbls.remove_all();
217 ArrayList<char *> confirm_paths; // test for new files
218 confirm_paths.set_array_delete();
220 // convert to new size if not original size
221 if( new_scale != 1 ) {
223 Asset *orig = mwindow->edl->assets->first;
224 for( ; orig; orig=orig->next ) {
225 Asset *proxy = proxy_render.add_original(orig, new_scale);
226 if( !proxy ) continue;
227 int exists = fs.get_size(proxy->path) > 0 ? 1 : 0;
228 int got_it = exists && // if proxy exists, and is newer than orig
229 fs.get_date(proxy->path) > fs.get_date(orig->path) ? 1 : 0;
231 if( exists ) // prompt user to overwrite
232 confirm_paths.append(cstrdup(proxy->path));
233 proxy_render.add_needed(orig, proxy);
239 // test for existing files
240 if( confirm_paths.size() ) {
241 result = ConfirmSave::test_files(mwindow, &confirm_paths);
242 confirm_paths.remove_all_objects();
246 result = proxy_render.create_needed_proxies(new_scale);
248 if( !result ) // resize project
249 mwindow->set_proxy(use_scaler, new_scale, auto_scale,
250 &proxy_render.orig_idxbls, &proxy_render.orig_proxies);
252 mwindow->undo->update_undo_after(_("proxy"), LOAD_ALL);
253 mwindow->edl->Garbage::remove_user();
254 mwindow->restart_brender();
256 mwindow->gui->lock_window("ProxyDialog::to_proxy");
257 mwindow->update_project(LOAD_ALL);
258 mwindow->gui->unlock_window();
262 void ProxyRender::to_proxy_path(char *new_path, Indexable *indexable, int scale)
264 // path is already a proxy
265 if( strstr(indexable->path, ".proxy") ) return;
266 strcpy(new_path, indexable->path);
268 int n = sprintf(prxy, ".proxy%d", scale);
269 // insert proxy, path.sfx => path.proxy#-sfx.ext
270 char *ep = new_path + strlen(new_path);
271 char *sfx = strrchr(new_path, '.');
273 char *bp = ep, *cp = (ep += n);
274 while( --bp > sfx ) *--cp = *bp;
280 for( char *cp=prxy; --n>=0; ++cp ) *sfx++ = *cp;
281 const char *ext = format_asset->format == FILE_FFMPEG ?
282 format_asset->fformat : File::get_tag(format_asset->format);
284 while( *ext ) *ep++ = *ext++;
286 //printf("ProxyRender::to_proxy_path %d %s %s\n", __LINE__, new_path), asset->path);
289 void ProxyRender::from_proxy_path(char *new_path, Asset *asset, int scale)
291 char prxy[BCTEXTLEN];
292 int n = sprintf(prxy, ".proxy%d", scale);
293 strcpy(new_path, asset->path);
294 char *ptr = strstr(asset->path, prxy);
295 if( !ptr || (ptr[n] != '-' && ptr[n] != '.') ) return;
296 // remove proxy, path.proxy#-sfx.ext => path.sfx
297 char *ext = strrchr(ptr, '.');
298 if( !ext ) ext = ptr + strlen(ptr);
300 for( *cp='.'; cp<ext; ++cp ) *ptr++ = *cp;
304 ProxyRender::ProxyRender(MWindow *mwindow, Asset *format_asset)
306 this->mwindow = mwindow;
307 this->format_asset = format_asset;
309 counter_lock = new Mutex("ProxyDialog::counter_lock");
311 failed = 0; canceled = 0;
314 ProxyRender::~ProxyRender()
319 for( int i=0,n=orig_idxbls.size(); i<n; ++i ) orig_idxbls[i]->remove_user();
320 for( int i=0,n=orig_proxies.size(); i<n; ++i ) orig_proxies[i]->remove_user();
321 for( int i=0,n=needed_idxbls.size(); i<n; ++i ) needed_idxbls[i]->remove_user();
322 for( int i=0,n=needed_proxies.size(); i<n; ++i ) needed_proxies[i]->remove_user();
325 Asset *ProxyRender::add_original(Indexable *idxbl, int new_scale)
327 if( !idxbl->have_video() ) return 0;
328 if( idxbl->get_video_frames() <= 0 ) return 0;
329 // don't proxy proxies
330 if( strstr(idxbl->path,".proxy") ) return 0;
331 char new_path[BCTEXTLEN];
332 to_proxy_path(new_path, idxbl, new_scale);
333 // don't proxy if not readable
334 if( idxbl->is_asset && access(idxbl->path, R_OK) ) return 0;
335 // add to orig_idxbls & orig_proxies if it isn't already there.
337 for( int i = 0; !got_it && i<orig_proxies.size(); ++i )
338 got_it = !strcmp(orig_proxies[i]->path, new_path);
339 if( got_it ) return 0;
340 Assets *edl_assets = mwindow->edl->assets;
341 Asset *proxy = edl_assets->get_asset(new_path);
343 proxy = new Asset(new_path);
344 // new compression parameters
345 proxy->copy_format(format_asset, 0);
346 proxy->awindow_folder = AW_PROXY_FOLDER;
347 proxy->audio_data = 0;
348 proxy->video_data = 1;
350 proxy->width = idxbl->get_w() / new_scale;
351 if( proxy->width & 1 ) ++proxy->width;
352 proxy->actual_width = proxy->width;
353 proxy->height = idxbl->get_h() / new_scale;
354 if( proxy->height & 1 ) ++proxy->height;
355 proxy->actual_height = proxy->height;
356 proxy->frame_rate = idxbl->get_frame_rate();
357 proxy->video_length = idxbl->get_video_frames();
358 edl_assets->append(proxy);
361 orig_proxies.append(proxy);
363 orig_idxbls.append(idxbl);
367 void ProxyRender::add_needed(Indexable *idxbl, Asset *proxy)
369 needed_idxbls.append(idxbl);
371 needed_proxies.append(proxy);
375 void ProxyRender::update_progress()
377 counter_lock->lock();
379 counter_lock->unlock();
380 progress->update(total_rendered);
383 int ProxyRender::is_canceled()
385 return progress->is_cancelled();
388 int ProxyRender::create_needed_proxies(int new_scale)
390 if( !needed_proxies.size() ) return 0;
392 failed = 0; canceled = 0;
394 // create proxy assets which don't already exist
395 int64_t total_len = 0;
396 for( int i = 0; i < needed_idxbls.size(); i++ ) {
397 total_len += needed_idxbls[i]->get_video_frames();
400 // start progress bar. MWindow is locked inside this
401 progress = mwindow->mainprogress->
402 start_progress(_("Creating proxy files..."), total_len);
404 ProxyFarm engine(mwindow, this, &needed_idxbls, &needed_proxies);
405 engine.process_packages();
406 printf("failed=%d canceled=%d\n", failed, progress->is_cancelled());
409 canceled = progress->is_cancelled();
410 progress->stop_progress();
411 delete progress; progress = 0;
413 if( failed && !canceled ) {
414 eprintf("Error making proxy.");
417 return !failed && !canceled ? 0 : 1;
421 ProxyWindow::ProxyWindow(MWindow *mwindow, ProxyDialog *dialog, int x, int y)
422 : BC_Window(_(PROGRAM_NAME ": Proxy settings"), x, y, WIDTH, HEIGHT,
425 this->mwindow = mwindow;
426 this->dialog = dialog;
430 ProxyWindow::~ProxyWindow()
432 lock_window("ProxyWindow::~ProxyWindow");
438 void ProxyWindow::create_objects()
440 lock_window("ProxyWindow::create_objects");
441 int margin = mwindow->theme->widget_border;
443 dialog->use_scaler = mwindow->edl->session->proxy_use_scaler;
444 dialog->orig_scale = mwindow->edl->session->proxy_scale;
445 dialog->auto_scale = mwindow->edl->session->proxy_auto_scale;
446 dialog->new_scale = dialog->orig_scale;
450 add_subwindow(use_scaler = new ProxyUseScaler(this, x, y));
451 y += use_scaler->get_h() + margin;
454 add_subwindow(text = new BC_Title(x, y, _("Scale factor:")));
455 x += text->get_w() + margin;
457 dialog->size_text[0] = cstrdup(_("Original size"));
458 dialog->size_factors[0] = 1;
459 dialog->total_sizes = 1;
460 int popupmenu_w = BC_PopupMenu::calculate_w(get_text_width(MEDIUMFONT, dialog->size_text[0]));
461 add_subwindow(scale_factor = new ProxyMenu(mwindow, this, x, y, popupmenu_w, ""));
462 scale_factor->update_sizes();
463 x += scale_factor->get_w() + margin;
465 ProxyTumbler *tumbler;
466 add_subwindow(tumbler = new ProxyTumbler(mwindow, this, x, y));
467 y += tumbler->get_h() + margin;
470 add_subwindow(text = new BC_Title(x, y, _("New media dimensions: ")));
471 x += text->get_w() + margin;
472 add_subwindow(new_dimensions = new BC_Title(x, y, ""));
473 y += new_dimensions->get_h() + margin;
476 format_tools = new ProxyFormatTools(mwindow, this, dialog->asset);
477 format_tools->create_objects(x, y, 0, 1, 0, 0, 0, 1, 0, 1, // skip the path
481 add_subwindow(auto_scale = new ProxyAutoScale(this, x, y));
482 y += auto_scale->get_h() + margin;
486 add_subwindow(new BC_OKButton(this));
487 add_subwindow(new BC_CancelButton(this));
492 ProxyFormatTools::ProxyFormatTools(MWindow *mwindow, ProxyWindow *pwindow, Asset *asset)
493 : FormatTools(mwindow, pwindow, asset)
495 this->pwindow = pwindow;
498 void ProxyFormatTools::update_format()
500 FormatTools::update_format();
501 pwindow->use_scaler->update();
504 void ProxyWindow::update()
506 // preview the new size
507 char string[BCTEXTLEN];
508 //printf("ProxyWindow::update %d %d %d %d %d\n",
509 // __LINE__, mwindow->edl->session->output_w, mwindow->edl->session->output_h,
510 // dialog->orig_scale, dialog->new_scale);
511 int orig_w = mwindow->edl->session->output_w * dialog->orig_scale;
512 int orig_h = mwindow->edl->session->output_h * dialog->orig_scale;
513 int new_w = orig_w / dialog->new_scale;
514 if( new_w & 1 ) ++new_w;
515 int new_h = orig_h / dialog->new_scale;
516 if( new_h & 1 ) ++new_h;
517 sprintf(string, "%dx%d", new_w, new_h);
518 new_dimensions->update(string);
519 dialog->scale_to_text(string, dialog->new_scale);
520 scale_factor->set_text(string);
521 use_scaler->update();
522 auto_scale->update();
526 ProxyUseScaler::ProxyUseScaler(ProxyWindow *pwindow, int x, int y)
527 : BC_CheckBox(x, y, pwindow->dialog->use_scaler, _("Use scaler (FFMPEG only)"))
529 this->pwindow = pwindow;
532 void ProxyUseScaler::update()
534 ProxyDialog *dialog = pwindow->dialog;
535 if( dialog->asset->format != FILE_FFMPEG ) dialog->use_scaler = 0;
536 BC_CheckBox::update(dialog->use_scaler);
537 int scaler_avail = dialog->asset->format == FILE_FFMPEG ? 1 : 0;
538 if( !scaler_avail && enabled ) disable();
539 if( scaler_avail && !enabled ) enable();
542 int ProxyUseScaler::handle_event()
544 pwindow->dialog->new_scale = 1;
545 pwindow->dialog->use_scaler = get_value();
546 pwindow->scale_factor->update_sizes();
551 ProxyAutoScale::ProxyAutoScale(ProxyWindow *pwindow, int x, int y)
552 : BC_CheckBox(x, y, pwindow->dialog->use_scaler, _("Auto proxy/scale media loads"))
554 this->pwindow = pwindow;
557 void ProxyAutoScale::update()
559 ProxyDialog *dialog = pwindow->dialog;
560 if( dialog->new_scale == 1 ) dialog->auto_scale = 0;
561 BC_CheckBox::update(dialog->auto_scale);
562 int can_auto_proxy = dialog->new_scale != 1 ? 1 : 0;
563 if( !can_auto_proxy && enabled ) disable();
564 if( can_auto_proxy && !enabled ) enable();
567 int ProxyAutoScale::handle_event()
569 pwindow->dialog->auto_scale = get_value();
575 ProxyMenu::ProxyMenu(MWindow *mwindow, ProxyWindow *pwindow,
576 int x, int y, int w, const char *text)
577 : BC_PopupMenu(x, y, w, text, 1)
579 this->mwindow = mwindow;
580 this->pwindow = pwindow;
583 void ProxyMenu::update_sizes()
585 while( total_items() > 0 ) del_item(0);
586 ProxyDialog *dialog = pwindow->dialog;
587 dialog->calculate_sizes();
588 for( int i=0; i < dialog->total_sizes; i++ )
589 add_item(new BC_MenuItem(dialog->size_text[i]));
592 int ProxyMenu::handle_event()
594 ProxyDialog *dialog = pwindow->dialog;
595 for( int i = 0; i < dialog->total_sizes; i++ ) {
596 if( !strcmp(get_text(), pwindow->dialog->size_text[i]) ) {
597 dialog->new_scale = pwindow->dialog->size_factors[i];
598 if( dialog->new_scale == 1 ) dialog->use_scaler = 0;
607 ProxyTumbler::ProxyTumbler(MWindow *mwindow, ProxyWindow *pwindow, int x, int y)
608 : BC_Tumbler(x, y, 0)
610 this->mwindow = mwindow;
611 this->pwindow = pwindow;
614 int ProxyTumbler::handle_up_event()
616 if( pwindow->dialog->new_scale > 1 ) {
618 for( i = 0; i < pwindow->dialog->total_sizes; i++ ) {
619 if( pwindow->dialog->new_scale == pwindow->dialog->size_factors[i] ) {
621 pwindow->dialog->new_scale = pwindow->dialog->size_factors[i];
631 int ProxyTumbler::handle_down_event()
634 for( i = 0; i < pwindow->dialog->total_sizes - 1; i++ ) {
635 if( pwindow->dialog->new_scale == pwindow->dialog->size_factors[i] ) {
637 pwindow->dialog->new_scale = pwindow->dialog->size_factors[i];
648 ProxyPackage::ProxyPackage()
653 ProxyClient::ProxyClient(MWindow *mwindow,
654 ProxyRender *proxy_render, ProxyFarm *server)
657 this->mwindow = mwindow;
658 this->proxy_render = proxy_render;
663 ProxyClient::~ProxyClient()
665 delete render_engine;
670 void ProxyClient::process_package(LoadPackage *ptr)
672 if( proxy_render->failed ) return;
673 if( proxy_render->is_canceled() ) return;
675 EDL *edl = mwindow->edl;
676 Preferences *preferences = mwindow->preferences;
677 ProxyPackage *package = (ProxyPackage*)ptr;
678 Indexable *orig = package->orig_idxbl;
679 Asset *proxy = package->proxy_asset;
680 //printf("%s %s\n", orig->path, proxy->path);
681 VRender *vrender = 0;
682 int jobs = proxy_render->needed_proxies.size();
683 int processors = preferences->project_smp / jobs + 1, result = 0;
685 if( orig->is_asset ) {
687 src_file->set_processors(processors);
688 src_file->set_preload(edl->session->playback_preload);
689 src_file->set_subtitle(edl->session->decode_subtitles ?
690 edl->session->subtitle_number : -1);
691 src_file->set_interpolate_raw(edl->session->interpolate_raw);
692 src_file->set_white_balance_raw(edl->session->white_balance_raw);
693 if( src_file->open_file(preferences, (Asset*)orig, 1, 0) != FILE_OK )
697 TransportCommand command;
698 command.command = CURRENT_FRAME;
699 command.get_edl()->copy_all((EDL *)orig);
700 command.change_type = CHANGE_ALL;
701 command.realtime = 0;
702 render_engine = new RenderEngine(0, preferences, 0, 0);
703 render_engine->set_vcache(video_cache = new CICache(preferences));
704 render_engine->arm_command(&command);
705 if( !(vrender = render_engine->vrender) )
709 // go to the next asset if the reader fails
710 // proxy_render->failed = 1;
715 dst_file.set_processors(processors);
716 result = dst_file.open_file(preferences, proxy, 0, 1);
718 proxy_render->failed = 1;
722 dst_file.start_video_thread(1, edl->session->color_model,
723 processors > 1 ? 2 : 1, 0);
725 int src_w = orig->get_w(), src_h = orig->get_h();
726 VFrame src_frame(src_w,src_h, edl->session->color_model);
728 OverlayFrame scaler(processors);
730 for( int64_t i=0, length=orig->get_video_frames(); i<length &&
731 !proxy_render->failed && !proxy_render->is_canceled(); ++i ) {
732 if( orig->is_asset ) {
733 src_file->set_video_position(i, 0);
734 result = src_file->read_frame(&src_frame);
737 result = vrender->process_buffer(&src_frame, i, 0);
738 //printf("result=%d\n", result);
741 // go to the next asset if the reader fails
742 // proxy_render->failed = 1;
746 // have to write after getting the video buffer or it locks up
747 VFrame ***dst_frames = dst_file.get_video_buffer();
748 VFrame *dst_frame = dst_frames[0][0];
749 int dst_w = dst_frame->get_w(), dst_h = dst_frame->get_h();
750 scaler.overlay(dst_frame, &src_frame,
751 0,0, src_w,src_h, 0,0, dst_w,dst_h,
752 1.0, TRANSFER_REPLACE, NEAREST_NEIGHBOR);
753 result = dst_file.write_video_buffer(1);
755 // only fail if the writer fails
756 proxy_render->failed = 1;
759 proxy_render->update_progress();
764 ProxyFarm::ProxyFarm(MWindow *mwindow, ProxyRender *proxy_render,
765 ArrayList<Indexable*> *orig_idxbls,
766 ArrayList<Asset*> *proxy_assets)
767 : LoadServer(MIN(mwindow->preferences->processors, proxy_assets->size()),
768 proxy_assets->size())
770 this->mwindow = mwindow;
771 this->proxy_render = proxy_render;
772 this->orig_idxbls = orig_idxbls;
773 this->proxy_assets = proxy_assets;
776 void ProxyFarm::init_packages()
778 for( int i = 0; i < get_total_packages(); i++ ) {
779 ProxyPackage *package = (ProxyPackage*)get_package(i);
780 package->proxy_asset = proxy_assets->get(i);
781 package->orig_idxbl = orig_idxbls->get(i);
785 LoadClient* ProxyFarm::new_client()
787 return new ProxyClient(mwindow, proxy_render, this);
790 LoadPackage* ProxyFarm::new_package()
792 return new ProxyPackage;