cinelerra-5.1/cinelerra/proxy.C

   1
   2 /*
   3  * CINELERRA
   4  * Copyright (C) 2015 Adam Williams <broadcast at earthling dot net>
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  *
  20  */
  21
  22 #include "assets.h"
  23 #include "audiodevice.h"
  24 #include "bcsignals.h"
  25 #include "cache.h"
  26 #include "clip.h"
  27 #include "cstrdup.h"
  28 #include "edl.h"
  29 #include "edlsession.h"
  30 #include "errorbox.h"
  31 #include "file.h"
  32 #include "filesystem.h"
  33 #include "formattools.h"
  34 #include "language.h"
  35 #include "mainerror.h"
  36 #include "mainindexes.h"
  37 #include "mainprogress.h"
  38 #include "mainundo.h"
  39 #include "mutex.h"
  40 #include "mwindow.h"
  41 #include "mwindowgui.h"
  42 #include "overlayframe.h"
  43 #include "preferences.h"
  44 #include "proxy.h"
  45 #include "renderengine.h"
  46 #include "theme.h"
  47 #include "transportque.h"
  48 #include "vrender.h"
  49
  50 #define WIDTH xS(400)
  51 #define HEIGHT yS(330)
  52 #define MAX_SCALE 16
  53
  54 ProxyMenuItem::ProxyMenuItem(MWindow *mwindow)
  55  : BC_MenuItem(_("Proxy settings..."),  _("Alt-r"), 'r')
  56 {
  57         this->mwindow = mwindow;
  58         set_alt();
  59         dialog = 0;
  60 }
  61 ProxyMenuItem::~ProxyMenuItem()
  62 {
  63         delete dialog;
  64 }
  65
  66 void ProxyMenuItem::create_objects()
  67 {
  68         dialog = new ProxyDialog(mwindow);
  69 }
  70
  71 int ProxyMenuItem::handle_event()
  72 {
  73         mwindow->gui->unlock_window();
  74         dialog->start();
  75         mwindow->gui->lock_window("ProxyMenuItem::handle_event");
  76
  77         return 1;
  78 }
  79
  80 ProxyDialog::ProxyDialog(MWindow *mwindow)
  81 {
  82         this->mwindow = mwindow;
  83         gui = 0;
  84         asset = new Asset;
  85
  86 // quicker than some, not as good as others
  87         asset->format = FILE_FFMPEG;
  88         strcpy(asset->fformat, "mpeg");
  89         strcpy(asset->vcodec, "mpeg.mpeg");
  90         asset->ff_video_bitrate = 2000000;
  91         asset->video_data = 1;
  92
  93         bzero(size_text, sizeof(char*) * MAX_SIZES);
  94         bzero(size_factors, sizeof(int) * MAX_SIZES);
  95         size_text[0] = cstrdup(_("Original size"));
  96         size_factors[0] = 1;
  97         total_sizes = 1;
  98 }
  99
 100 ProxyDialog::~ProxyDialog()
 101 {
 102         close_window();
 103         for( int i=0; i<MAX_SIZES; ++i ) delete [] size_text[i];
 104         asset->remove_user();
 105 }
 106
 107 BC_Window* ProxyDialog::new_gui()
 108 {
 109         asset->format = FILE_FFMPEG;
 110         asset->load_defaults(mwindow->defaults, "PROXY_", 1, 1, 0, 0, 0);
 111         mwindow->gui->lock_window("ProxyDialog::new_gui");
 112         int cx, cy;
 113         mwindow->gui->get_abs_cursor(cx, cy);
 114         gui = new ProxyWindow(mwindow, this, cx - WIDTH/2, cy - HEIGHT/2);
 115         gui->create_objects();
 116         mwindow->gui->unlock_window();
 117         return gui;
 118 }
 119
 120 void ProxyDialog::scale_to_text(char *string, int scale)
 121 {
 122         strcpy(string, size_text[0]);
 123         for( int i = 0; i < total_sizes; i++ ) {
 124                 if( scale == size_factors[i] ) {
 125                         strcpy(string, size_text[i]);
 126                         break;
 127                 }
 128         }
 129 }
 130
 131
 132 void ProxyDialog::calculate_sizes()
 133 {
 134         for( int i=1; i<total_sizes; ++i ) {
 135                 delete [] size_text[i];
 136                 size_text[i] = 0;
 137                 size_factors[i] = 0;
 138         }
 139         total_sizes = 1;
 140
 141         if( !use_scaler ) {
 142 // w,h should stay even for yuv
 143                 int ow = orig_w, oh = orig_h;
 144                 if( BC_CModels::is_yuv(mwindow->edl->session->color_model) ) {
 145                         ow /= 2;  oh /= 2;
 146                 }
 147                 for( int i=2; i<MAX_SCALE; ++i ) {
 148                         if( (ow % i) != 0 ) continue;
 149                         if( (oh % i) != 0 ) continue;
 150                         size_factors[total_sizes++] = i;
 151                 }
 152         }
 153         else {
 154                 size_factors[total_sizes++] = 2;   size_factors[total_sizes++] = 3;
 155                 size_factors[total_sizes++] = 8;   size_factors[total_sizes++] = 12;
 156                 size_factors[total_sizes++] = 16;  size_factors[total_sizes++] = 24;
 157                 size_factors[total_sizes++] = 32;
 158         }
 159         for( int i=1; i<total_sizes; ++i ) {
 160                 char string[BCTEXTLEN];
 161                 sprintf(string, "1/%d", size_factors[i]);
 162                 size_text[i] = cstrdup(string);
 163         }
 164 }
 165
 166 void ProxyDialog::handle_close_event(int result)
 167 {
 168         if( result ) return;
 169         if( !File::renders_video(asset) ) {
 170                 eprintf(_("Specified format does not render video"));
 171                 return;
 172         }
 173         mwindow->edl->session->proxy_auto_scale = auto_scale;
 174         mwindow->edl->session->proxy_beep = beep;
 175         mwindow->edl->session->proxy_disabled_scale = 1;
 176         mwindow->gui->lock_window("ProxyDialog::handle_close_event");
 177         mwindow->gui->update_proxy_toggle();
 178         mwindow->gui->unlock_window();
 179         asset->save_defaults(mwindow->defaults, "PROXY_", 1, 1, 0, 0, 0);
 180         result = mwindow->to_proxy(asset, new_scale, use_scaler);
 181         if( result >= 0 && beep > 0 && new_scale != 1 ) {
 182                 if( !result ) {
 183                         mwindow->beep(4000., 0.5, beep);
 184                         usleep(250000);
 185                         mwindow->beep(1000., 0.5, beep);
 186                         usleep(250000);
 187                         mwindow->beep(4000., 0.5, beep);
 188                 }
 189                 else
 190                         mwindow->beep(2000., 2.0, beep);
 191         }
 192 }
 193
 194 void ProxyRender::to_proxy_path(char *new_path, Indexable *indexable, int scale)
 195 {
 196 // path is already a proxy
 197         if( strstr(indexable->path, ".proxy") ) return;
 198         strcpy(new_path, indexable->path);
 199         char prxy[BCSTRLEN];
 200         int n = sprintf(prxy, ".proxy%d", scale);
 201 // insert proxy, path.sfx => path.proxy#-sfx.ext
 202         char *ep = new_path + strlen(new_path);
 203         char *sfx = strrchr(new_path, '.');
 204         if( sfx ) {
 205                 char *bp = ep, *cp = (ep += n);
 206                 while( --bp > sfx ) *--cp = *bp;
 207                 *--cp = '-';
 208         }
 209         else {
 210                 sfx = ep;  ep += n;
 211         }
 212         for( char *cp=prxy; --n>=0; ++cp ) *sfx++ = *cp;
 213         *ep++ = '.';
 214         const char *ext = indexable->get_video_frames() < 0 ?  "png" :
 215                 format_asset->format == FILE_FFMPEG ?
 216                         format_asset->fformat :
 217                         File::get_tag(format_asset->format);
 218         while( *ext ) *ep++ = *ext++;
 219         *ep = 0;
 220 //printf("ProxyRender::to_proxy_path %d %s %s\n", __LINE__, new_path), asset->path);
 221 }
 222
 223 int ProxyRender::from_proxy_path(char *new_path, Indexable *indexable, int scale)
 224 {
 225         char prxy[BCTEXTLEN];
 226         int n = sprintf(prxy, ".proxy%d", scale);
 227         strcpy(new_path, indexable->path);
 228         char *ptr = strstr(new_path, prxy);
 229         if( !ptr || (ptr[n] != '-' && ptr[n] != '.') ) return 1;
 230 // remove proxy, path.proxy#-sfx.ext => path.sfx
 231         char *ext = strrchr(ptr, '.');
 232         if( !ext ) ext = ptr + strlen(ptr);
 233         char *cp = ptr + n;
 234         for( *cp='.'; cp<ext; ++cp ) *ptr++ = *cp;
 235         *ptr = 0;
 236         return 0;
 237 }
 238
 239 ProxyRender::ProxyRender(MWindow *mwindow, Asset *format_asset, int asset_scale)
 240 {
 241         this->mwindow = mwindow;
 242         this->format_asset = format_asset;
 243         this->asset_scale = asset_scale;
 244         progress = 0;
 245         counter_lock = new Mutex("ProxyDialog::counter_lock");
 246         total_rendered = 0;
 247         failed = 0;  canceled = 0;
 248 }
 249
 250 ProxyRender::~ProxyRender()
 251 {
 252         delete progress;
 253         delete counter_lock;
 254
 255         for( int i=0,n=orig_idxbls.size(); i<n; ++i ) orig_idxbls[i]->remove_user();
 256         for( int i=0,n=orig_proxies.size(); i<n; ++i ) orig_proxies[i]->remove_user();
 257         for( int i=0,n=needed_idxbls.size(); i<n; ++i ) needed_idxbls[i]->remove_user();
 258         for( int i=0,n=needed_proxies.size(); i<n; ++i ) needed_proxies[i]->remove_user();
 259 }
 260
 261 Asset *ProxyRender::add_original(Indexable *idxbl, int new_scale)
 262 {
 263         if( !idxbl->have_video() ) return 0;
 264 // don't proxy proxies
 265         if( strstr(idxbl->path,".proxy") ) return 0;
 266         char new_path[BCTEXTLEN];
 267         to_proxy_path(new_path, idxbl, new_scale);
 268 // don't proxy if not readable
 269         if( idxbl->is_asset && access(idxbl->path, R_OK) ) return 0;
 270 // add to orig_idxbls & orig_proxies if it isn't already there.
 271         int got_it = 0;
 272         for( int i = 0; !got_it && i<orig_proxies.size(); ++i )
 273                 got_it = !strcmp(orig_proxies[i]->path, new_path);
 274         if( got_it ) return 0;
 275         Assets *edl_assets = mwindow->edl->assets;
 276         Asset *proxy = edl_assets->get_asset(new_path);
 277         if( !proxy ) {
 278                 proxy = new Asset(new_path);
 279 // new compression parameters
 280                 int64_t video_frames = idxbl->get_video_frames();
 281                 if( video_frames < 0 ) {
 282                         proxy->format = FILE_PNG;
 283                         proxy->png_use_alpha = 1;
 284                         proxy->video_length = -1;
 285                 }
 286                 else {
 287                         proxy->copy_format(format_asset, 0);
 288                         proxy->video_length = video_frames;
 289                 }
 290                 proxy->folder_no = AW_PROXY_FOLDER;
 291                 proxy->audio_data = 0;
 292                 proxy->video_data = 1;
 293                 proxy->layers = 1;
 294                 proxy->width = idxbl->get_w() / new_scale;
 295                 if( proxy->width & 1 ) ++proxy->width;
 296                 proxy->actual_width = proxy->width;
 297                 proxy->height = idxbl->get_h() / new_scale;
 298                 if( proxy->height & 1 ) ++proxy->height;
 299                 proxy->actual_height = proxy->height;
 300                 proxy->frame_rate = idxbl->get_frame_rate();
 301         }
 302         orig_proxies.append(proxy);
 303         idxbl->add_user();
 304         orig_idxbls.append(idxbl);
 305         return proxy;
 306 }
 307
 308 void ProxyRender::add_needed(Indexable *idxbl, Asset *proxy)
 309 {
 310         needed_idxbls.append(idxbl);
 311         idxbl->add_user();
 312         needed_proxies.append(proxy);
 313         proxy->add_user();
 314 }
 315
 316 void ProxyRender::update_progress()
 317 {
 318         counter_lock->lock();
 319         ++total_rendered;
 320         counter_lock->unlock();
 321         progress->update(total_rendered);
 322 }
 323
 324 int ProxyRender::is_canceled()
 325 {
 326         return progress->is_cancelled();
 327 }
 328
 329 int ProxyRender::create_needed_proxies(int new_scale)
 330 {
 331         if( !needed_proxies.size() ) return 0;
 332         total_rendered = 0;
 333         failed = 0;  canceled = 0;
 334
 335 // create proxy assets which don't already exist
 336         int64_t total_len = 0;
 337         for( int i = 0; i < needed_idxbls.size(); i++ ) {
 338                 total_len += needed_idxbls[i]->get_video_frames();
 339         }
 340
 341 // start progress bar.  MWindow is locked inside this
 342         progress = mwindow->mainprogress->
 343                 start_progress(_("Creating proxy files..."), total_len);
 344
 345         ProxyFarm engine(mwindow, this, &needed_idxbls, &needed_proxies);
 346         engine.process_packages();
 347 printf("proxy: failed=%d canceled=%d\n", failed, progress->is_cancelled());
 348
 349 // stop progress bar
 350         canceled = progress->is_cancelled();
 351         progress->stop_progress();
 352         delete progress;  progress = 0;
 353
 354         if( failed && !canceled ) {
 355                 eprintf(_("Error making proxy."));
 356         }
 357         return !failed && !canceled ? 0 : 1;
 358 }
 359
 360
 361 ProxyWindow::ProxyWindow(MWindow *mwindow, ProxyDialog *dialog, int x, int y)
 362  : BC_Window(_(PROGRAM_NAME ": Proxy settings"), x, y, WIDTH, HEIGHT,
 363                 -1, -1, 0, 0, 1)
 364 {
 365         this->mwindow = mwindow;
 366         this->dialog = dialog;
 367         format_tools = 0;
 368 }
 369
 370 ProxyWindow::~ProxyWindow()
 371 {
 372         lock_window("ProxyWindow::~ProxyWindow");
 373         delete format_tools;
 374         unlock_window();
 375 }
 376
 377
 378 void ProxyWindow::create_objects()
 379 {
 380         lock_window("ProxyWindow::create_objects");
 381         int margin = mwindow->theme->widget_border;
 382         int lmargin = margin + xS(10);
 383
 384         dialog->use_scaler = mwindow->edl->session->proxy_use_scaler;
 385         dialog->orig_scale = mwindow->edl->session->proxy_scale;
 386         dialog->auto_scale = mwindow->edl->session->proxy_auto_scale;
 387         dialog->beep = mwindow->edl->session->proxy_beep;
 388         dialog->new_scale = dialog->orig_scale;
 389         dialog->orig_w = mwindow->edl->session->output_w;
 390         dialog->orig_h = mwindow->edl->session->output_h;
 391         if( !dialog->use_scaler ) {
 392                 dialog->orig_w *= dialog->orig_scale;
 393                 dialog->orig_h *= dialog->orig_scale;
 394         }
 395
 396         int x = lmargin;
 397         int y = margin+yS(10);
 398         add_subwindow(use_scaler = new ProxyUseScaler(this, x, y));
 399         y += use_scaler->get_h() + margin;
 400
 401         BC_Title *text;
 402         add_subwindow(text = new BC_Title(x, y, _("Scale factor:")));
 403         x += text->get_w() + margin;
 404
 405         int popupmenu_w = BC_PopupMenu::calculate_w(get_text_width(MEDIUMFONT, dialog->size_text[0])+xS(15));
 406         add_subwindow(scale_factor = new ProxyMenu(mwindow, this, x, y, popupmenu_w, ""));
 407         scale_factor->update_sizes();
 408         x += scale_factor->get_w() + margin;
 409
 410         ProxyTumbler *tumbler;
 411         add_subwindow(tumbler = new ProxyTumbler(mwindow, this, x, y));
 412         y += tumbler->get_h() + margin;
 413
 414         x = lmargin;
 415         add_subwindow(text = new BC_Title(x, y, _("New media dimensions: ")));
 416         x += text->get_w() + margin;
 417         add_subwindow(new_dimensions = new BC_Title(x, y, ""));
 418         y += new_dimensions->get_h() + margin;
 419
 420         x = lmargin;
 421         add_subwindow(text = new BC_Title(x, y, _("Active Scale: ")));
 422         x += text->get_w() + margin;
 423         add_subwindow(active_scale = new BC_Title(x, y, ""));
 424         y += active_scale->get_h() + margin;
 425
 426         x = lmargin;  y += yS(25);
 427         format_tools = new ProxyFormatTools(mwindow, this, dialog->asset);
 428         format_tools->create_objects(x, y, 0, 1, 0, 0, 0, 1, 0, 1, // skip the path
 429                 0, 0);
 430
 431         x = lmargin;
 432         add_subwindow(auto_scale = new ProxyAutoScale(this, x, y));
 433         y += auto_scale->get_h() + margin;
 434         add_subwindow(beep_on_done = new ProxyBeepOnDone(this, x, y));
 435         x += beep_on_done->get_w() + margin + xS(10);
 436         add_subwindow(new BC_Title(x, y+yS(10), _("Beep on done volume")));
 437 //      y += beep_on_done->get_h() + margin;
 438
 439         update();
 440
 441         add_subwindow(new BC_OKButton(this));
 442         add_subwindow(new BC_CancelButton(this));
 443         show_window(1);
 444         unlock_window();
 445 }
 446
 447 ProxyFormatTools::ProxyFormatTools(MWindow *mwindow, ProxyWindow *pwindow, Asset *asset)
 448  : FormatTools(mwindow, pwindow, asset)
 449 {
 450         this->pwindow = pwindow;
 451 }
 452
 453 void ProxyFormatTools::update_format()
 454 {
 455         asset->save_defaults(mwindow->defaults, "PROXY_", 1, 1, 0, 0, 0);
 456         FormatTools::update_format();
 457         pwindow->use_scaler->update();
 458 }
 459
 460 void ProxyWindow::update()
 461 {
 462         char string[BCSTRLEN];
 463         int new_w = dialog->orig_w / dialog->new_scale;
 464         if( new_w & 1 ) ++new_w;
 465         int new_h = dialog->orig_h / dialog->new_scale;
 466         if( new_h & 1 ) ++new_h;
 467         sprintf(string, "%dx%d", new_w, new_h);
 468         new_dimensions->update(string);
 469         dialog->scale_to_text(string, dialog->new_scale);
 470         scale_factor->set_text(string);
 471         use_scaler->update();
 472         auto_scale->update();
 473         int scale = mwindow->edl->session->proxy_scale;
 474         if( scale == 1 ) scale = mwindow->edl->session->proxy_disabled_scale;
 475         sprintf(string, scale>1 ? "1/%d" : "%d", scale);
 476         active_scale->update(string);
 477 }
 478
 479
 480 ProxyUseScaler::ProxyUseScaler(ProxyWindow *pwindow, int x, int y)
 481  : BC_CheckBox(x, y, pwindow->dialog->use_scaler, _("Use scaler   (FFMPEG only)"))
 482 {
 483         this->pwindow = pwindow;
 484 }
 485
 486 void ProxyUseScaler::update()
 487 {
 488         ProxyDialog *dialog = pwindow->dialog;
 489         if( dialog->asset->format != FILE_FFMPEG ) dialog->use_scaler = 0;
 490         BC_CheckBox::update(dialog->use_scaler);
 491         int scaler_avail = dialog->asset->format == FILE_FFMPEG ? 1 : 0;
 492         if( !scaler_avail &&  enabled ) disable();
 493         if( scaler_avail  && !enabled ) enable();
 494 }
 495
 496 int ProxyUseScaler::handle_event()
 497 {
 498         pwindow->dialog->new_scale = 1;
 499         pwindow->dialog->use_scaler = get_value();
 500         pwindow->scale_factor->update_sizes();
 501         pwindow->update();
 502         return 1;
 503 }
 504
 505 ProxyAutoScale::ProxyAutoScale(ProxyWindow *pwindow, int x, int y)
 506  : BC_CheckBox(x, y, pwindow->dialog->auto_scale, _("Auto proxy/scale media loads"))
 507 {
 508         this->pwindow = pwindow;
 509 }
 510
 511 void ProxyAutoScale::update()
 512 {
 513         ProxyDialog *dialog = pwindow->dialog;
 514         if( dialog->new_scale == 1 ) dialog->auto_scale = 0;
 515         BC_CheckBox::update(dialog->auto_scale);
 516         int can_auto_proxy = dialog->new_scale != 1 ? 1 : 0;
 517         if( !can_auto_proxy &&  enabled ) disable();
 518         if( can_auto_proxy  && !enabled ) enable();
 519 }
 520
 521 int ProxyAutoScale::handle_event()
 522 {
 523         pwindow->dialog->auto_scale = get_value();
 524         pwindow->update();
 525         return 1;
 526 }
 527
 528 ProxyBeepOnDone::ProxyBeepOnDone(ProxyWindow *pwindow, int x, int y)
 529  : BC_FPot(x, y, pwindow->dialog->beep*100.f, 0.f, 100.f)
 530 {
 531         this->pwindow = pwindow;
 532 }
 533
 534 int ProxyBeepOnDone::handle_event()
 535 {
 536         pwindow->dialog->beep = get_value()/100.f;
 537         pwindow->update();
 538         return 1;
 539 }
 540
 541
 542 ProxyMenu::ProxyMenu(MWindow *mwindow, ProxyWindow *pwindow,
 543                 int x, int y, int w, const char *text)
 544  : BC_PopupMenu(x, y, w, text, 1)
 545 {
 546         this->mwindow = mwindow;
 547         this->pwindow = pwindow;
 548 }
 549
 550 void ProxyMenu::update_sizes()
 551 {
 552         while( total_items() > 0 ) del_item(0);
 553         ProxyDialog *dialog = pwindow->dialog;
 554         dialog->calculate_sizes();
 555         for( int i=0; i < dialog->total_sizes; i++ )
 556                 add_item(new BC_MenuItem(dialog->size_text[i]));
 557 }
 558
 559 int ProxyMenu::handle_event()
 560 {
 561         ProxyDialog *dialog = pwindow->dialog;
 562         for( int i = 0; i < dialog->total_sizes; i++ ) {
 563                 if( !strcmp(get_text(), pwindow->dialog->size_text[i]) ) {
 564                         dialog->new_scale = pwindow->dialog->size_factors[i];
 565                         pwindow->update();
 566                         break;
 567                 }
 568         }
 569         return 1;
 570 }
 571
 572
 573 ProxyTumbler::ProxyTumbler(MWindow *mwindow, ProxyWindow *pwindow, int x, int y)
 574  : BC_Tumbler(x, y, 0)
 575 {
 576         this->mwindow = mwindow;
 577         this->pwindow = pwindow;
 578 }
 579
 580 int ProxyTumbler::handle_up_event()
 581 {
 582         if( pwindow->dialog->new_scale > 1 ) {
 583                 int i;
 584                 for( i = 0; i < pwindow->dialog->total_sizes; i++ ) {
 585                         if( pwindow->dialog->new_scale == pwindow->dialog->size_factors[i] ) {
 586                                 i--;
 587                                 pwindow->dialog->new_scale = pwindow->dialog->size_factors[i];
 588                                 pwindow->update();
 589                                 return 1;
 590                         }
 591                 }
 592         }
 593
 594         return 0;
 595 }
 596
 597 int ProxyTumbler::handle_down_event()
 598 {
 599         int i;
 600         for( i = 0; i < pwindow->dialog->total_sizes - 1; i++ ) {
 601                 if( pwindow->dialog->new_scale == pwindow->dialog->size_factors[i] ) {
 602                         i++;
 603                         pwindow->dialog->new_scale = pwindow->dialog->size_factors[i];
 604                         pwindow->update();
 605                         return 1;
 606                 }
 607         }
 608
 609         return 0;
 610 }
 611
 612
 613
 614 ProxyPackage::ProxyPackage()
 615  : LoadPackage()
 616 {
 617 }
 618
 619 ProxyClient::ProxyClient(MWindow *mwindow,
 620                 ProxyRender *proxy_render, ProxyFarm *server)
 621  : LoadClient(server)
 622 {
 623         this->mwindow = mwindow;
 624         this->proxy_render = proxy_render;
 625         render_engine = 0;
 626         video_cache = 0;
 627         src_file = 0;
 628 }
 629 ProxyClient::~ProxyClient()
 630 {
 631         delete render_engine;
 632         delete video_cache;
 633         delete src_file;
 634 }
 635
 636 void ProxyClient::process_package(LoadPackage *ptr)
 637 {
 638         if( proxy_render->failed ) return;
 639         if( proxy_render->is_canceled() ) return;
 640
 641         EDL *edl = mwindow->edl;
 642         Preferences *preferences = mwindow->preferences;
 643         ProxyPackage *package = (ProxyPackage*)ptr;
 644         Indexable *orig = package->orig_idxbl;
 645         Asset *proxy = package->proxy_asset;
 646 //printf("%s %s\n", orig->path, proxy->path);
 647         VRender *vrender = 0;
 648         int jobs = proxy_render->needed_proxies.size();
 649         int processors = preferences->project_smp / jobs + 1, result = 0;
 650
 651 // each cpu should process at least about 1 MB, or it thrashes
 652         int size = edl->session->output_w * edl->session->output_h * 4;
 653         int cpus = size / 0x100000 + 1;
 654         if( processors > cpus ) processors = cpus;
 655
 656         if( orig->is_asset ) {
 657                 src_file = new File;
 658                 src_file->set_processors(processors);
 659                 src_file->set_preload(edl->session->playback_preload);
 660                 src_file->set_subtitle(edl->session->decode_subtitles ?
 661                         edl->session->subtitle_number : -1);
 662                 src_file->set_interpolate_raw(edl->session->interpolate_raw);
 663                 src_file->set_white_balance_raw(edl->session->white_balance_raw);
 664                 if( src_file->open_file(preferences, (Asset*)orig, 1, 0) != FILE_OK )
 665                         result = 1;
 666         }
 667         else {
 668                 TransportCommand command;
 669                 command.command = CURRENT_FRAME;
 670                 command.get_edl()->copy_all((EDL *)orig);
 671                 command.change_type = CHANGE_ALL;
 672                 command.realtime = 0;
 673                 render_engine = new RenderEngine(0, preferences, 0, 0);
 674                 render_engine->set_vcache(video_cache = new CICache(preferences));
 675                 render_engine->arm_command(&command);
 676                 if( !(vrender = render_engine->vrender) )
 677                         result = 1;
 678         }
 679         if( result ) {
 680 // go to the next asset if the reader fails
 681 //              proxy_render->failed = 1;
 682                 return;
 683         }
 684
 685         File dst_file;
 686         dst_file.set_processors(processors);
 687         result = dst_file.open_file(preferences, proxy, 0, 1);
 688         if( result ) {
 689                 proxy_render->failed = 1;
 690                 ::remove(proxy->path);
 691                 return;
 692         }
 693
 694         dst_file.start_video_thread(1, edl->session->color_model,
 695                         processors > 1 ? 2 : 1, 0);
 696
 697         int src_w = orig->get_w(), src_h = orig->get_h();
 698         VFrame src_frame(src_w,src_h, edl->session->color_model);
 699
 700         OverlayFrame scaler(processors);
 701         int64_t video_length = orig->get_video_frames();
 702         if( video_length < 0 ) video_length = 1;
 703
 704         for( int64_t i=0; i<video_length &&
 705              !proxy_render->failed && !proxy_render->is_canceled(); ++i ) {
 706                 if( orig->is_asset ) {
 707                         src_file->set_video_position(i, 0);
 708                         result = src_file->read_frame(&src_frame);
 709                 }
 710                 else
 711                         result = vrender->process_buffer(&src_frame, i, 0);
 712 //printf("result=%d\n", result);
 713
 714                 if( result ) {
 715 // go to the next asset if the reader fails
 716 //                      proxy_render->failed = 1;
 717                         break;
 718                 }
 719
 720 // have to write after getting the video buffer or it locks up
 721                 VFrame ***dst_frames = dst_file.get_video_buffer();
 722                 VFrame *dst_frame = dst_frames[0][0];
 723                 int dst_w = dst_frame->get_w(), dst_h = dst_frame->get_h();
 724                 scaler.overlay(dst_frame, &src_frame,
 725                         0,0, src_w,src_h, 0,0, dst_w,dst_h,
 726                         1.0, TRANSFER_REPLACE, NEAREST_NEIGHBOR);
 727                 result = dst_file.write_video_buffer(1);
 728                 if( result ) {
 729 // only fail if the writer fails
 730                         proxy_render->failed = 1;
 731                         break;
 732                 }
 733                 proxy_render->update_progress();
 734         }
 735         if( !proxy_render->failed && !proxy_render->is_canceled() ) {
 736                 Asset *asset = edl->assets->update(proxy);
 737                 asset->proxy_scale = proxy_render->asset_scale;
 738                 int scale = asset->proxy_scale;
 739                 if( !scale ) scale = 1;
 740                 asset->width = asset->actual_width * scale;
 741                 asset->height = asset->actual_height * scale;
 742                 mwindow->mainindexes->add_next_asset(0, asset);
 743                 mwindow->mainindexes->start_build();
 744         }
 745         else
 746                 ::remove(proxy->path);
 747 }
 748
 749
 750 ProxyFarm::ProxyFarm(MWindow *mwindow, ProxyRender *proxy_render,
 751                 ArrayList<Indexable*> *orig_idxbls,
 752                 ArrayList<Asset*> *proxy_assets)
 753  : LoadServer(MIN(mwindow->preferences->processors, proxy_assets->size()),
 754         proxy_assets->size())
 755 {
 756         this->mwindow = mwindow;
 757         this->proxy_render = proxy_render;
 758         this->orig_idxbls = orig_idxbls;
 759         this->proxy_assets = proxy_assets;
 760 }
 761
 762 void ProxyFarm::init_packages()
 763 {
 764         for( int i = 0; i < get_total_packages(); i++ ) {
 765                 ProxyPackage *package = (ProxyPackage*)get_package(i);
 766                 package->proxy_asset = proxy_assets->get(i);
 767                 package->orig_idxbl = orig_idxbls->get(i);
 768         }
 769 }
 770
 771 LoadClient* ProxyFarm::new_client()
 772 {
 773         return new ProxyClient(mwindow, proxy_render, this);
 774 }
 775
 776 LoadPackage* ProxyFarm::new_package()
 777 {
 778         return new ProxyPackage;
 779 }
 780
 781
 782 ProxyBeep::ProxyBeep(MWindow *mwindow)
 783  : Thread(1, 0, 0)
 784 {
 785         this->mwindow = mwindow;
 786         audio = new AudioDevice(mwindow);
 787         playing_audio = 0;
 788         interrupted = -1;
 789 }
 790
 791 ProxyBeep::~ProxyBeep()
 792 {
 793         stop(0);
 794         delete audio;
 795 }
 796
 797 void ProxyBeep::run()
 798 {
 799         int channels = 2;
 800         int64_t bfrsz = BEEP_SAMPLE_RATE;
 801         EDL *edl = mwindow->edl;
 802         EDLSession *session = edl->session;
 803         AudioOutConfig *aconfig = session->playback_config->aconfig;
 804         audio->open_output(aconfig, BEEP_SAMPLE_RATE, bfrsz, channels, 0);
 805         audio->start_playback();
 806
 807         double out0[bfrsz], out1[bfrsz], *out[2] = { out0, out1 };
 808         const double two_pi = 2*M_PI;
 809         int64_t audio_len = BEEP_SAMPLE_RATE * secs;
 810         const double dt = two_pi * freq/BEEP_SAMPLE_RATE;
 811         double th = 0;
 812
 813         audio_pos = 0;
 814         playing_audio = 1;
 815         while( !interrupted ) {
 816                 int len = audio_len - audio_pos;
 817                 if( len <= 0 ) break;
 818                 if( len > bfrsz ) len = bfrsz;
 819                 int k = audio_pos;
 820                 for( int i=0; i<len; ++i,++k,th+=dt ) {
 821                         double t = th - two_pi;
 822                         if( t >= 0 ) th = t;
 823                         out0[i] = out1[i] = sin(th) * gain;
 824                 }
 825                 audio->write_buffer(out, channels, len);
 826                 audio_pos = k;
 827         }
 828
 829         if( !interrupted )
 830                 audio->set_last_buffer();
 831         audio->stop_audio(interrupted ? 0 : 1);
 832         playing_audio = 0;
 833
 834         audio->close_all();
 835 }
 836
 837 void ProxyBeep::start()
 838 {
 839         if( running() ) return;
 840         audio_pos = -1;
 841         interrupted = 0;
 842         Thread::start();
 843 }
 844
 845 void ProxyBeep::stop(int wait)
 846 {
 847         if( running() && !interrupted ) {
 848                 interrupted = 1;
 849                 audio->stop_audio(wait);
 850         }
 851         Thread::join();
 852 }
 853
 854 void ProxyBeep::tone(double freq, double secs, double gain)
 855 {
 856         stop(0);
 857         this->freq = freq;
 858         this->secs = secs;
 859         this->gain = gain;
 860         start();
 861 }
 862