libhb/work.c

   1 /* $Id: work.c,v 1.43 2005/03/17 16:38:49 titer Exp $
   2
   3    This file is part of the HandBrake source code.
   4    Homepage: <http://handbrake.m0k.org/>.
   5    It may be used under the terms of the GNU General Public License. */
   6
   7 #include "hb.h"
   8 #include "a52dec/a52.h"
   9 #include "dca.h"
  10
  11 typedef struct
  12 {
  13     hb_list_t * jobs;
  14     int         cpu_count;
  15     int       * error;
  16     volatile int * die;
  17
  18 } hb_work_t;
  19
  20 static void work_func();
  21 static void do_job( hb_job_t *, int cpu_count );
  22 static void work_loop( void * );
  23
  24 /**
  25  * Allocates work object and launches work thread with work_func.
  26  * @param jobs Handle to hb_list_t.
  27  * @param cpu_count Humber of CPUs found in system.
  28  * @param die Handle to user inititated exit indicator.
  29  * @param error Handle to error indicator.
  30  */
  31 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
  32                             volatile int * die, int * error )
  33 {
  34     hb_work_t * work = calloc( sizeof( hb_work_t ), 1 );
  35
  36     work->jobs      = jobs;
  37     work->cpu_count = cpu_count;
  38     work->die       = die;
  39     work->error     = error;
  40
  41     return hb_thread_init( "work", work_func, work, HB_LOW_PRIORITY );
  42 }
  43
  44 /**
  45  * Iterates through job list and calls do_job for each job.
  46  * @param _work Handle work object.
  47  */
  48 static void work_func( void * _work )
  49 {
  50     hb_work_t  * work = _work;
  51     hb_job_t   * job;
  52
  53     hb_log( "%d job(s) to process", hb_list_count( work->jobs ) );
  54
  55     while( !*work->die && ( job = hb_list_item( work->jobs, 0 ) ) )
  56     {
  57         hb_list_rem( work->jobs, job );
  58         job->die = work->die;
  59         do_job( job, work->cpu_count );
  60     }
  61
  62     *(work->error) = HB_ERROR_NONE;
  63
  64     free( work );
  65 }
  66
  67 static hb_work_object_t * getWork( int id )
  68 {
  69     hb_work_object_t * w;
  70     for( w = hb_objects; w; w = w->next )
  71     {
  72         if( w->id == id )
  73         {
  74             return w;
  75         }
  76     }
  77     return NULL;
  78 }
  79
  80 /**
  81  * Job initialization rountine.
  82  * Initializes fifos.
  83  * Creates work objects for synchronizer, video decoder, video renderer, video decoder, audio decoder, audio encoder, reader, muxer.
  84  * Launches thread for each work object with work_loop.
  85  * Loops while monitoring status of work threads and fifos.
  86  * Exits loop when conversion is done and fifos are empty.
  87  * Closes threads and frees fifos.
  88  * @param job Handle work hb_job_t.
  89  * @param cpu_count number of CPUs found in system.
  90  */
  91 static void do_job( hb_job_t * job, int cpu_count )
  92 {
  93     hb_title_t    * title;
  94     int             i, j;
  95     hb_work_object_t * w;
  96
  97     /* FIXME: This feels really hackish, anything better? */
  98     hb_work_object_t * audio_w = NULL;
  99
 100     hb_audio_t   * audio;
 101     hb_subtitle_t * subtitle;
 102     int done;
 103
 104     title = job->title;
 105
 106     job->list_work = hb_list_init();
 107
 108     hb_log( "starting job" );
 109     hb_log( " + device %s", title->dvd );
 110     hb_log( " + title %d, chapter(s) %d to %d", title->index,
 111             job->chapter_start, job->chapter_end );
 112     if ( job->pixel_ratio == 1 )
 113     {
 114         /* Correct the geometry of the output movie when using PixelRatio */
 115         job->height=title->height-job->crop[0]-job->crop[1];
 116         job->width=title->width-job->crop[2]-job->crop[3];
 117     }
 118
 119         /* Keep width and height within these boundaries */
 120         if (job->maxHeight && (job->height > job->maxHeight) )
 121         {
 122                 job->height = job->maxHeight;
 123                 hb_fix_aspect( job, HB_KEEP_HEIGHT );
 124                 hb_log("Height out of bounds, scaling down to %i", job->maxHeight);
 125                 hb_log("New dimensions %i * %i", job->width, job->height);
 126         }
 127         if (job->maxWidth && (job->width > job->maxWidth) )
 128         {
 129                 job->width = job->maxWidth;
 130                 hb_fix_aspect( job, HB_KEEP_WIDTH );
 131                 hb_log("Width out of bounds, scaling down to %i", job->maxWidth);
 132                 hb_log("New dimensions %i * %i", job->width, job->height);
 133         }
 134
 135     hb_log( " + %dx%d -> %dx%d, crop %d/%d/%d/%d",
 136             title->width, title->height, job->width, job->height,
 137             job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
 138     hb_log( " + deinterlace %s", job->deinterlace ? "on" : "off" );
 139     hb_log( " + grayscale %s", job->grayscale ? "on" : "off" );
 140     if( job->vquality >= 0.0 && job->vquality <= 1.0 )
 141     {
 142         hb_log( " + %.3f fps, video quality %.2f", (float) job->vrate /
 143                 (float) job->vrate_base, job->vquality );
 144     }
 145     else
 146     {
 147         hb_log( " + %.3f fps, video bitrate %d kbps, pass %d",
 148                 (float) job->vrate / (float) job->vrate_base,
 149                 job->vbitrate, job->pass );
 150     }
 151         hb_log (" + PixelRatio: %d, width:%d, height: %d",job->pixel_ratio,job->width, job->height);
 152     job->fifo_mpeg2  = hb_fifo_init( 2048 );
 153     job->fifo_raw    = hb_fifo_init( 8 );
 154     job->fifo_sync   = hb_fifo_init( 8 );
 155     job->fifo_render = hb_fifo_init( 8 );
 156     job->fifo_mpeg4  = hb_fifo_init( 8 );
 157
 158     /* Synchronization */
 159     hb_list_add( job->list_work, ( w = getWork( WORK_SYNC ) ) );
 160     w->fifo_in  = NULL;
 161     w->fifo_out = NULL;
 162
 163     /* Video decoder */
 164     hb_list_add( job->list_work, ( w = getWork( WORK_DECMPEG2 ) ) );
 165     w->fifo_in  = job->fifo_mpeg2;
 166     w->fifo_out = job->fifo_raw;
 167
 168     /* Video renderer */
 169     hb_list_add( job->list_work, ( w = getWork( WORK_RENDER ) ) );
 170     w->fifo_in  = job->fifo_sync;
 171     w->fifo_out = job->fifo_render;
 172
 173     /* Video encoder */
 174     switch( job->vcodec )
 175     {
 176         case HB_VCODEC_FFMPEG:
 177             hb_log( " + encoder FFmpeg" );
 178             w = getWork( WORK_ENCAVCODEC );
 179             break;
 180         case HB_VCODEC_XVID:
 181             hb_log( " + encoder XviD" );
 182             w = getWork( WORK_ENCXVID );
 183             break;
 184         case HB_VCODEC_X264:
 185             hb_log( " + encoder x264" );
 186             w = getWork( WORK_ENCX264 );
 187             break;
 188     }
 189     w->fifo_in  = job->fifo_render;
 190     w->fifo_out = job->fifo_mpeg4;
 191     w->config   = &job->config;
 192     hb_list_add( job->list_work, w );
 193
 194     subtitle = hb_list_item( title->list_subtitle, 0 );
 195     if( subtitle )
 196     {
 197         hb_log( " + subtitle %x, %s", subtitle->id, subtitle->lang );
 198
 199         subtitle->fifo_in  = hb_fifo_init( 8 );
 200         subtitle->fifo_raw = hb_fifo_init( 8 );
 201
 202         hb_list_add( job->list_work, ( w = getWork( WORK_DECSUB ) ) );
 203         w->fifo_in  = subtitle->fifo_in;
 204         w->fifo_out = subtitle->fifo_raw;
 205     }
 206
 207     if( job->acodec & HB_ACODEC_AC3 )
 208     {
 209         hb_log( " + audio AC3 passthrough" );
 210     }
 211     else
 212     {
 213         hb_log( " + audio %d kbps, %d Hz", job->abitrate, job->arate );
 214         hb_log( " + encoder %s", ( job->acodec & HB_ACODEC_FAAC ) ?
 215                 "faac" : ( ( job->acodec & HB_ACODEC_LAME ) ? "lame" :
 216                 "vorbis" ) );
 217     }
 218
 219     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 220     {
 221         audio = hb_list_item( title->list_audio, i );
 222         hb_log( "   + %x, %s", audio->id, audio->lang );
 223
 224                 /* sense-check the current mixdown options */
 225
 226                 /* log the requested mixdown */
 227                 for (j = 0; j < hb_audio_mixdowns_count; j++) {
 228                         if (hb_audio_mixdowns[j].amixdown == job->audio_mixdowns[i]) {
 229                                 hb_log( "     + Requested mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
 230                                 break;
 231                         }
 232                 }
 233
 234         /* sense-check the requested mixdown */
 235
 236         /* audioCodecsSupportMono and audioCodecsSupport6Ch are the same for now,
 237            but this may change in the future, so they are separated for flexibility */
 238         int audioCodecsSupportMono = ((audio->codec == HB_ACODEC_AC3 ||
 239             audio->codec == HB_ACODEC_DCA) && job->acodec == HB_ACODEC_FAAC);
 240         int audioCodecsSupport6Ch =  ((audio->codec == HB_ACODEC_AC3 ||
 241             audio->codec == HB_ACODEC_DCA) && job->acodec == HB_ACODEC_FAAC);
 242
 243         /* find out what the format of our source audio is */
 244         switch (audio->input_channel_layout) {
 245
 246             /* mono sources */
 247             case HB_INPUT_CH_LAYOUT_MONO:
 248                 /* regardless of what stereo mixdown we've requested, a mono source always get mixed down
 249                 to mono if we can, and mixed up to stereo if we can't */
 250                 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 1) {
 251                     job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
 252                 } else {
 253                     job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
 254                 }
 255                 break;
 256
 257             /* stereo input */
 258             case HB_INPUT_CH_LAYOUT_STEREO:
 259                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 260                 /* use stereo if not supported */
 261                 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
 262                     job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
 263                 /* otherwise, preserve stereo regardless of if we requested something higher */
 264                 } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_STEREO) {
 265                     job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
 266                 }
 267                 break;
 268
 269             /* dolby (DPL1 aka Dolby Surround = 4.0 matrix-encoded) input */
 270             /* the A52 flags don't allow for a way to distinguish between DPL1 and DPL2 on a DVD,
 271                so we always assume a DPL1 source for A52_DOLBY */
 272             case HB_INPUT_CH_LAYOUT_DOLBY:
 273                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 274                 /* preserve dolby if not supported */
 275                 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
 276                     job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
 277                 /* otherwise, preserve dolby even if we requested something higher */
 278                 /* a stereo mixdown will still be honoured here */
 279                 } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
 280                     job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
 281                 }
 282                 break;
 283
 284             /* 3F/2R input */
 285             case HB_INPUT_CH_LAYOUT_3F2R:
 286             case HB_INPUT_CH_LAYOUT_3F2RLFE:
 287                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 288                 /* use dpl2 if not supported */
 289                 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
 290                     job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
 291                 } else {
 292                     /* check if we have 3F2R input and also have an LFE - i.e. we have a 5.1 source) */
 293                     if (audio->input_channel_layout == HB_INPUT_CH_LAYOUT_3F2RLFE) {
 294                         /* we have a 5.1 source */
 295                         /* if we requested 6ch, but our audio format doesn't support it, then mix to DPLII instead */
 296                         if (job->audio_mixdowns[i] == HB_AMIXDOWN_6CH && audioCodecsSupport6Ch == 0) {
 297                             job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
 298                         }
 299                     } else {
 300                         /* we have a 5.0 source, so we can't do 6ch conversion
 301                         default to DPL II instead */
 302                         if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBYPLII) {
 303                             job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
 304                         }
 305                     }
 306                 }
 307                 /* all other mixdowns will have been preserved here */
 308                 break;
 309
 310             /* 3F/1R input */
 311             case HB_INPUT_CH_LAYOUT_3F1R:
 312                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 313                 /* use dpl1 if not supported */
 314                 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
 315                     job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
 316                 } else {
 317                     /* we have a 4.0 or 4.1 source, so we can't do DPLII or 6ch conversion
 318                     default to DPL I instead */
 319                     if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
 320                         job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
 321                     }
 322                 }
 323                 /* all other mixdowns will have been preserved here */
 324                 break;
 325
 326             default:
 327                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 328                 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 1) {
 329                     job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
 330                 /* mix everything else down to stereo */
 331                 } else {
 332                     job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
 333                 }
 334
 335         }
 336
 337                 /* log the output mixdown */
 338                 for (j = 0; j < hb_audio_mixdowns_count; j++) {
 339                         if (hb_audio_mixdowns[j].amixdown == job->audio_mixdowns[i]) {
 340                                 hb_log( "     + Actual mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
 341                                 break;
 342                         }
 343                 }
 344
 345                 /* we now know we have a valid mixdown for the input source and the audio output format */
 346                 /* remember the mixdown for this track */
 347                 audio->amixdown = job->audio_mixdowns[i];
 348
 349         audio->config.vorbis.language = audio->lang_simple;
 350
 351                 /* set up the audio work structures */
 352         audio->fifo_in   = hb_fifo_init( 2048 );
 353         audio->fifo_raw  = hb_fifo_init( 8 );
 354         audio->fifo_sync = hb_fifo_init( 8 );
 355         audio->fifo_out  = hb_fifo_init( 8 );
 356
 357         switch( audio->codec )
 358         {
 359             case HB_ACODEC_AC3:
 360                 w = getWork( WORK_DECA52 );
 361                 break;
 362             case HB_ACODEC_DCA:
 363                 w = getWork( WORK_DECDCA );
 364                 break;
 365             case HB_ACODEC_MPGA:
 366                 w = getWork( WORK_DECAVCODEC );
 367                 break;
 368             case HB_ACODEC_LPCM:
 369                 w = getWork( WORK_DECLPCM );
 370                 break;
 371         }
 372         w->fifo_in  = audio->fifo_in;
 373         w->fifo_out = audio->fifo_raw;
 374         w->config   = &audio->config;
 375         w->amixdown = audio->amixdown;
 376
 377         /* FIXME: This feels really hackish, anything better? */
 378         audio_w = calloc( sizeof( hb_work_object_t ), 1 );
 379         audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
 380
 381         hb_list_add( job->list_work, audio_w );
 382
 383         switch( job->acodec )
 384         {
 385             case HB_ACODEC_FAAC:
 386                 w = getWork( WORK_ENCFAAC );
 387                 break;
 388             case HB_ACODEC_LAME:
 389                 w = getWork( WORK_ENCLAME );
 390                 break;
 391             case HB_ACODEC_VORBIS:
 392                 w = getWork( WORK_ENCVORBIS );
 393                 break;
 394         }
 395
 396         if( job->acodec != HB_ACODEC_AC3 )
 397         {
 398             w->fifo_in  = audio->fifo_sync;
 399             w->fifo_out = audio->fifo_out;
 400             w->config   = &audio->config;
 401             w->amixdown = audio->amixdown;
 402
 403             /* FIXME: This feels really hackish, anything better? */
 404             audio_w = calloc( sizeof( hb_work_object_t ), 1 );
 405             audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
 406
 407             hb_list_add( job->list_work, audio_w );
 408         }
 409
 410
 411     }
 412
 413     /* Init read & write threads */
 414     job->reader = hb_reader_init( job );
 415
 416     hb_log( " + output: %s", job->file );
 417     job->muxer = hb_muxer_init( job );
 418
 419     job->done = 0;
 420
 421     /* Launch processing threads */
 422     for( i = 1; i < hb_list_count( job->list_work ); i++ )
 423     {
 424         w = hb_list_item( job->list_work, i );
 425         w->done = &job->done;
 426                 w->thread_sleep_interval = 10;
 427         w->init( w, job );
 428         w->thread = hb_thread_init( w->name, work_loop, w,
 429                                     HB_LOW_PRIORITY );
 430     }
 431
 432     done = 0;
 433     w = hb_list_item( job->list_work, 0 );
 434         w->thread_sleep_interval = 50;
 435     w->init( w, job );
 436     while( !*job->die )
 437     {
 438         if( w->work( w, NULL, NULL ) == HB_WORK_DONE )
 439         {
 440             done = 1;
 441         }
 442         if( done &&
 443             !hb_fifo_size( job->fifo_sync ) &&
 444             !hb_fifo_size( job->fifo_render ) &&
 445             !hb_fifo_size( job->fifo_mpeg4 ) )
 446         {
 447             break;
 448         }
 449         hb_snooze( w->thread_sleep_interval );
 450     }
 451     hb_list_rem( job->list_work, w );
 452     w->close( w );
 453     job->done = 1;
 454
 455     /* Close work objects */
 456     while( ( w = hb_list_item( job->list_work, 0 ) ) )
 457     {
 458         hb_list_rem( job->list_work, w );
 459         hb_thread_close( &w->thread );
 460         w->close( w );
 461
 462         /* FIXME: This feels really hackish, anything better? */
 463         if ( w->id == WORK_DECA52 ||
 464              w->id == WORK_DECDCA ||
 465              w->id == WORK_DECLPCM ||
 466              w->id == WORK_ENCFAAC ||
 467              w->id == WORK_ENCLAME ||
 468              w->id == WORK_ENCVORBIS )
 469         {
 470             free( w );
 471             w = NULL;
 472         }
 473     }
 474
 475     hb_list_close( &job->list_work );
 476
 477     /* Stop read & write threads */
 478     hb_thread_close( &job->reader );
 479     hb_thread_close( &job->muxer );
 480
 481     /* Close fifos */
 482     hb_fifo_close( &job->fifo_mpeg2 );
 483     hb_fifo_close( &job->fifo_raw );
 484     hb_fifo_close( &job->fifo_sync );
 485     hb_fifo_close( &job->fifo_render );
 486     hb_fifo_close( &job->fifo_mpeg4 );
 487     if( subtitle )
 488     {
 489         hb_fifo_close( &subtitle->fifo_in );
 490         hb_fifo_close( &subtitle->fifo_raw );
 491     }
 492     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 493     {
 494         audio = hb_list_item( title->list_audio, i );
 495         hb_fifo_close( &audio->fifo_in );
 496         hb_fifo_close( &audio->fifo_raw );
 497         hb_fifo_close( &audio->fifo_sync );
 498         hb_fifo_close( &audio->fifo_out );
 499     }
 500
 501     hb_title_close( &job->title );
 502     free( job );
 503 }
 504
 505 /**
 506  * Performs the work objects specific work function.
 507  * Loops calling work function for associated work object. Sleeps when fifo is full.
 508  * Monitors work done indicator.
 509  * Exits loop when work indiactor is set.
 510  * @param _w Handle to work object.
 511  */
 512 static void work_loop( void * _w )
 513 {
 514     hb_work_object_t * w = _w;
 515     hb_buffer_t      * buf_in, * buf_out;
 516
 517     while( !*w->done )
 518     {
 519 #if 0
 520         hb_lock( job->pause );
 521         hb_unlock( job->pause );
 522 #endif
 523         if( hb_fifo_is_full( w->fifo_out ) ||
 524 //        if( (hb_fifo_percent_full( w->fifo_out ) > 0.8) ||
 525             !( buf_in = hb_fifo_get( w->fifo_in ) ) )
 526         {
 527             hb_snooze( w->thread_sleep_interval );
 528 //                      w->thread_sleep_interval += 1;
 529             continue;
 530         }
 531 //              w->thread_sleep_interval = MAX(1, (w->thread_sleep_interval - 1));
 532
 533         w->work( w, &buf_in, &buf_out );
 534
 535         // Propogate any chapter breaks for the worker
 536         if( buf_in && buf_out && buf_in->new_chap )
 537         {
 538             printf("WORK: Copying Chapter Break\n");
 539             buf_out->new_chap = 1;
 540         }
 541
 542         if( buf_in )
 543         {
 544             hb_buffer_close( &buf_in );
 545         }
 546         if( buf_out )
 547         {
 548             hb_fifo_push( w->fifo_out, buf_out );
 549         }
 550     }
 551 }