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