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
   9 typedef struct
  10 {
  11     hb_list_t * jobs;
  12     int         cpu_count;
  13     int       * error;
  14     volatile int * die;
  15
  16 } hb_work_t;
  17
  18 static void work_func();
  19 static void do_job( hb_job_t *, int cpu_count );
  20 static void job_loop( void * );
  21
  22 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
  23                             volatile int * die, int * error )
  24 {
  25     hb_work_t * work = calloc( sizeof( hb_work_t ), 1 );
  26
  27     work->jobs      = jobs;
  28     work->cpu_count = cpu_count;
  29     work->die       = die;
  30     work->error     = error;
  31
  32     return hb_thread_init( "work", work_func, work, HB_LOW_PRIORITY );
  33 }
  34
  35 static void work_func( void * _work )
  36 {
  37     hb_work_t  * work = _work;
  38     hb_job_t   * job;
  39
  40     hb_log( "%d job(s) to process", hb_list_count( work->jobs ) );
  41
  42     while( !*work->die && ( job = hb_list_item( work->jobs, 0 ) ) )
  43     {
  44         hb_list_rem( work->jobs, job );
  45         job->die = work->die;
  46         do_job( job, work->cpu_count );
  47     }
  48
  49     *(work->error) = HB_ERROR_NONE;
  50
  51     free( work );
  52 }
  53
  54 static hb_work_object_t * getWork( int id )
  55 {
  56     switch( id )
  57     {
  58         case WORK_SYNC:       return &hb_sync;
  59         case WORK_DECMPEG2:   return &hb_decmpeg2;
  60         case WORK_DECSUB:     return &hb_decsub;
  61         case WORK_RENDER:     return &hb_render;
  62         case WORK_ENCAVCODEC: return &hb_encavcodec;
  63         case WORK_ENCXVID:    return &hb_encxvid;
  64         case WORK_ENCX264:    return &hb_encx264;
  65         case WORK_DECA52:     return &hb_deca52;
  66         case WORK_DECAVCODEC: return &hb_decavcodec;
  67         case WORK_DECLPCM:    return &hb_declpcm;
  68         case WORK_ENCFAAC:    return &hb_encfaac;
  69         case WORK_ENCLAME:    return &hb_enclame;
  70         case WORK_ENCVORBIS:  return &hb_encvorbis;
  71     }
  72     return NULL;
  73 }
  74
  75 static void do_job( hb_job_t * job, int cpu_count )
  76 {
  77     hb_title_t    * title;
  78     int             i;
  79     hb_thread_t   * threads[8];
  80     hb_work_object_t * w;
  81     uint64_t time_total;
  82     hb_audio_t   * audio;
  83     hb_subtitle_t * subtitle;
  84
  85     title = job->title;
  86
  87     job->list_work = hb_list_init();
  88
  89     hb_log( "starting job" );
  90     hb_log( " + device %s", title->dvd );
  91     hb_log( " + title %d, chapter(s) %d to %d", title->index,
  92             job->chapter_start, job->chapter_end );
  93     hb_log( " + %dx%d -> %dx%d, crop %d/%d/%d/%d",
  94             title->width, title->height, job->width, job->height,
  95             job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
  96     hb_log( " + deinterlace %s", job->deinterlace ? "on" : "off" );
  97     hb_log( " + grayscale %s", job->grayscale ? "on" : "off" );
  98     if( job->vquality >= 0.0 && job->vquality <= 1.0 )
  99     {
 100         hb_log( " + %.3f fps, video quality %.2f", (float) job->vrate /
 101                 (float) job->vrate_base, job->vquality );
 102     }
 103     else
 104     {
 105         hb_log( " + %.3f fps, video bitrate %d kbps, pass %d",
 106                 (float) job->vrate / (float) job->vrate_base,
 107                 job->vbitrate, job->pass );
 108     }
 109
 110     job->fifo_mpeg2  = hb_fifo_init( 2048 );
 111     job->fifo_raw    = hb_fifo_init( 8 );
 112     job->fifo_sync   = hb_fifo_init( 8 );
 113     job->fifo_render = hb_fifo_init( 8 );
 114     job->fifo_mpeg4  = hb_fifo_init( 8 );
 115
 116     /* Synchronization */
 117     hb_list_add( job->list_work, ( w = getWork( WORK_SYNC ) ) );
 118     w->fifo_in  = NULL;
 119     w->fifo_out = NULL;
 120
 121     /* Video decoder */
 122     hb_list_add( job->list_work, ( w = getWork( WORK_DECMPEG2 ) ) );
 123     w->fifo_in  = job->fifo_mpeg2;
 124     w->fifo_out = job->fifo_raw;
 125
 126     /* Video renderer */
 127     hb_list_add( job->list_work, ( w = getWork( WORK_RENDER ) ) );
 128     w->fifo_in  = job->fifo_sync;
 129     w->fifo_out = job->fifo_render;
 130
 131     /* Video encoder */
 132     switch( job->vcodec )
 133     {
 134         case HB_VCODEC_FFMPEG:
 135             hb_log( " + encoder FFmpeg" );
 136             w = getWork( WORK_ENCAVCODEC );
 137             break;
 138         case HB_VCODEC_XVID:
 139             hb_log( " + encoder XviD" );
 140             w = getWork( WORK_ENCXVID );
 141             break;
 142         case HB_VCODEC_X264:
 143             hb_log( " + encoder x264" );
 144             w = getWork( WORK_ENCX264 );
 145             break;
 146     }
 147     w->fifo_in  = job->fifo_render;
 148     w->fifo_out = job->fifo_mpeg4;
 149     w->config   = &job->config;
 150     hb_list_add( job->list_work, w );
 151
 152     subtitle = hb_list_item( title->list_subtitle, 0 );
 153     if( subtitle )
 154     {
 155         hb_log( " + subtitle %x, %s", subtitle->id, subtitle->lang );
 156
 157         subtitle->fifo_in  = hb_fifo_init( 8 );
 158         subtitle->fifo_raw = hb_fifo_init( 8 );
 159
 160         hb_list_add( job->list_work, ( w = getWork( WORK_DECSUB ) ) );
 161         w->fifo_in  = subtitle->fifo_in;
 162         w->fifo_out = subtitle->fifo_raw;
 163     }
 164
 165     if( job->acodec & HB_ACODEC_AC3 )
 166     {
 167         hb_log( " + audio AC3 passthrough" );
 168     }
 169     else
 170     {
 171         hb_log( " + audio %d kbps, %d Hz", job->abitrate, job->arate );
 172         hb_log( " + encoder %s", ( job->acodec & HB_ACODEC_FAAC ) ?
 173                 "faac" : ( ( job->acodec & HB_ACODEC_LAME ) ? "lame" :
 174                 "vorbis" ) );
 175     }
 176
 177     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 178     {
 179         audio = hb_list_item( title->list_audio, i );
 180         hb_log( "   + %x, %s", audio->id, audio->lang );
 181
 182         audio->fifo_in   = hb_fifo_init( 2048 );
 183         audio->fifo_raw  = hb_fifo_init( 8 );
 184         audio->fifo_sync = hb_fifo_init( 8 );
 185         audio->fifo_out  = hb_fifo_init( 8 );
 186
 187         switch( audio->codec )
 188         {
 189             case HB_ACODEC_AC3:
 190                 w = getWork( WORK_DECA52 );
 191                 break;
 192             case HB_ACODEC_MPGA:
 193                 w = getWork( WORK_DECAVCODEC );
 194                 break;
 195             case HB_ACODEC_LPCM:
 196                 w = getWork( WORK_DECLPCM );
 197                 break;
 198         }
 199         w->fifo_in  = audio->fifo_in;
 200         w->fifo_out = audio->fifo_raw;
 201         hb_list_add( job->list_work, w );
 202
 203         switch( job->acodec )
 204         {
 205             case HB_ACODEC_FAAC:
 206                 w = getWork( WORK_ENCFAAC );
 207                 break;
 208             case HB_ACODEC_LAME:
 209                 w = getWork( WORK_ENCLAME );
 210                 break;
 211             case HB_ACODEC_VORBIS:
 212                 w = getWork( WORK_ENCVORBIS );
 213                 break;
 214         }
 215         if( job->acodec != HB_ACODEC_AC3 )
 216         {
 217             w->fifo_in  = audio->fifo_sync;
 218             w->fifo_out = audio->fifo_out;
 219             w->config   = &audio->config;
 220             hb_list_add( job->list_work, w );
 221         }
 222     }
 223
 224     /* Init read & write threads */
 225     job->reader = hb_reader_init( job );
 226
 227     hb_log( " + output: %s", job->file );
 228     job->muxer = hb_muxer_init( job );
 229
 230     for( i = 0; i < hb_list_count( job->list_work ); i++ )
 231     {
 232         w       = hb_list_item( job->list_work, i );
 233         w->lock = hb_lock_init();
 234         w->used = 0;
 235         w->time = 0;
 236         w->init( w, job );
 237     }
 238
 239     job->done = 0;
 240
 241     /* Launch processing threads */
 242     for( i = 0; i < cpu_count; i++ )
 243     {
 244         char thread_name[16];
 245         if( cpu_count - 1 )
 246         {
 247             snprintf( thread_name, 16, "cpu killer %d", i + 1 );
 248         }
 249         else
 250         {
 251             snprintf( thread_name, 16, "cpu killer" );
 252         }
 253         threads[i] = hb_thread_init( thread_name, job_loop, job,
 254                                      HB_LOW_PRIORITY );
 255     }
 256
 257     while( !*job->die && !job->done )
 258     {
 259         hb_snooze( 500 );
 260     }
 261
 262     for( i = 0; i < cpu_count; i++ )
 263     {
 264         hb_thread_close( &threads[i] );
 265     }
 266
 267     /* Stop read & write threads */
 268     hb_thread_close( &job->reader );
 269     hb_thread_close( &job->muxer );
 270
 271     /* Stats */
 272     time_total = 0;
 273     for( i = 0; i < hb_list_count( job->list_work ); i++ )
 274     {
 275         w = hb_list_item( job->list_work, i );
 276         time_total += w->time;
 277     }
 278     for( i = 0; i < hb_list_count( job->list_work ); i++ )
 279     {
 280         w = hb_list_item( job->list_work, i );
 281         hb_log( "%s: %.2f %%", w->name,
 282                 100.0 * (float) w->time / (float) time_total );
 283     }
 284
 285     /* Close work objects */
 286     while( ( w = hb_list_item( job->list_work, 0 ) ) )
 287     {
 288         hb_list_rem( job->list_work, w );
 289         hb_lock_close( &w->lock );
 290         w->close( w );
 291     }
 292
 293     /* Close fifos */
 294     hb_fifo_close( &job->fifo_mpeg2 );
 295     hb_fifo_close( &job->fifo_raw );
 296     hb_fifo_close( &job->fifo_sync );
 297     hb_fifo_close( &job->fifo_render );
 298     hb_fifo_close( &job->fifo_mpeg4 );
 299     if( subtitle )
 300     {
 301         hb_fifo_close( &subtitle->fifo_in );
 302         hb_fifo_close( &subtitle->fifo_raw );
 303     }
 304     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 305     {
 306         audio = hb_list_item( title->list_audio, i );
 307         hb_fifo_close( &audio->fifo_in );
 308         hb_fifo_close( &audio->fifo_raw );
 309         hb_fifo_close( &audio->fifo_sync );
 310         hb_fifo_close( &audio->fifo_out );
 311     }
 312 }
 313
 314 static int lock( hb_work_object_t * w )
 315 {
 316     hb_lock( w->lock );
 317     if( w->used )
 318     {
 319         hb_unlock( w->lock );
 320         return 0;
 321     }
 322     w->used = 1;
 323     hb_unlock( w->lock );
 324     return 1;
 325 }
 326
 327 static void unlock( hb_work_object_t * w )
 328 {
 329     hb_lock( w->lock );
 330     w->used = 0;
 331     hb_unlock( w->lock );
 332 }
 333
 334 static void job_loop( void * _job )
 335 {
 336     hb_job_t         * job = _job;
 337     hb_buffer_t      * buf_in, * buf_out;
 338     hb_work_object_t * w;
 339     int                work_count;
 340     int                act;
 341     int                i;
 342     uint64_t           date;
 343     int                done;
 344
 345     work_count = hb_list_count( job->list_work );
 346     act        = 0;
 347     done       = 0;
 348
 349     while( !*job->die && !job->done )
 350     {
 351         /* Handle synchronization, resampling, framerate change,
 352            etc */
 353         w = hb_list_item( job->list_work, 0 );
 354         if( lock( w ) )
 355         {
 356             date = hb_get_date();
 357             if( w->work( w, NULL, NULL ) == HB_WORK_DONE )
 358             {
 359                 done = 1;
 360             }
 361             w->time += hb_get_date() - date;
 362             unlock( w );
 363         }
 364
 365         for( i = 1; !*job->die && !job->done && i < work_count; i++ )
 366         {
 367             w = hb_list_item( job->list_work, i );
 368             if( !lock( w ) )
 369                 continue;
 370
 371             for( ;; )
 372             {
 373                 hb_lock( job->pause );
 374                 hb_unlock( job->pause );
 375
 376                 if( hb_fifo_is_full( w->fifo_out ) ||
 377                     !( buf_in = hb_fifo_get( w->fifo_in ) ) )
 378                 {
 379                     break;
 380                 }
 381
 382                 date = hb_get_date();
 383                 w->work( w, &buf_in, &buf_out );
 384                 w->time += hb_get_date() - date;
 385                 if( buf_in )
 386                 {
 387                     hb_buffer_close( &buf_in );
 388                 }
 389                 if( buf_out )
 390                 {
 391                     act = 1;
 392                     hb_fifo_push( w->fifo_out, buf_out );
 393                 }
 394             }
 395
 396             unlock( w );
 397         }
 398
 399         if( done &&
 400             !hb_fifo_size( job->fifo_sync ) &&
 401             !hb_fifo_size( job->fifo_render ) &&
 402             hb_fifo_size( job->fifo_mpeg4 ) < 2 )
 403         {
 404             job->done = 1;
 405             break;
 406         }
 407
 408         /* If we did nothing, wait a bit before trying again */
 409         if( !act )
 410         {
 411             hb_snooze( 50 );
 412         }
 413         act = 0;
 414     }
 415 }