OSDN Git Service

Probably a good idea to set the new par width after calculating it.
[handbrake-jp/handbrake-jp-git.git] / libhb / hb.c
1 #include "hb.h"
2 #include "hbffmpeg.h"
3
4 struct hb_handle_s
5 {
6     /* The "Check for update" thread */
7     int            build;
8     char           version[16];
9     hb_thread_t  * update_thread;
10
11     /* This thread's only purpose is to check other threads'
12        states */
13     volatile int   die;
14     hb_thread_t  * main_thread;
15     int            pid;
16
17     /* DVD/file scan thread */
18     hb_list_t    * list_title;
19     hb_thread_t  * scan_thread;
20
21     /* The thread which processes the jobs. Others threads are launched
22        from this one (see work.c) */
23     hb_list_t    * jobs;
24     hb_job_t     * current_job;
25     int            job_count;
26     int            job_count_permanent;
27     volatile int   work_die;
28     int            work_error;
29     hb_thread_t  * work_thread;
30
31     int            cpu_count;
32
33     hb_lock_t    * state_lock;
34     hb_state_t     state;
35
36     int            paused;
37     hb_lock_t    * pause_lock;
38     /* For MacGui active queue
39        increments each time the scan thread completes*/
40     int            scanCount;
41     
42     /* Stash of persistent data between jobs, for stuff
43        like correcting frame count and framerate estimates
44        on multi-pass encodes where frames get dropped.     */
45     hb_interjob_t * interjob;
46
47 };
48
49 hb_lock_t *hb_avcodec_lock;
50 hb_work_object_t * hb_objects = NULL;
51
52 static void thread_func( void * );
53
54 void hb_avcodec_init()
55 {
56     hb_avcodec_lock  = hb_lock_init();
57     av_register_all();
58 }
59
60 int hb_avcodec_open(AVCodecContext *avctx, AVCodec *codec)
61 {
62     int ret;
63     hb_lock( hb_avcodec_lock );
64     ret = avcodec_open(avctx, codec);
65     hb_unlock( hb_avcodec_lock );
66     return ret;
67 }
68
69 int hb_avcodec_close(AVCodecContext *avctx)
70 {
71     int ret;
72     hb_lock( hb_avcodec_lock );
73     ret = avcodec_close(avctx);
74     hb_unlock( hb_avcodec_lock );
75     return ret;
76 }
77
78 /**
79  * Registers work objects, by adding the work object to a liked list.
80  * @param w Handle to hb_work_object_t to register.
81  */
82 void hb_register( hb_work_object_t * w )
83 {
84     w->next    = hb_objects;
85     hb_objects = w;
86 }
87
88 /**
89  * libhb initialization routine.
90  * @param verbose HB_DEBUG_NONE or HB_DEBUG_ALL.
91  * @param update_check signals libhb to check for updated version from HandBrake website.
92  * @return Handle to hb_handle_t for use on all subsequent calls to libhb.
93  */
94 hb_handle_t * hb_init( int verbose, int update_check )
95 {
96     hb_handle_t * h = calloc( sizeof( hb_handle_t ), 1 );
97     uint64_t      date;
98
99     /* See hb_deep_log() and hb_log() in common.c */
100     global_verbosity_level = verbose;
101     if( verbose )
102         putenv( "HB_DEBUG=1" );
103
104     /* Check for an update on the website if asked to */
105     h->build = -1;
106
107     if( update_check )
108     {
109         hb_log( "hb_init: checking for updates" );
110         date             = hb_get_date();
111         h->update_thread = hb_update_init( &h->build, h->version );
112
113         for( ;; )
114         {
115             if( hb_thread_has_exited( h->update_thread ) )
116             {
117                 /* Immediate success or failure */
118                 hb_thread_close( &h->update_thread );
119                 break;
120             }
121             if( hb_get_date() > date + 1000 )
122             {
123                 /* Still nothing after one second. Connection problem,
124                    let the thread die */
125                 hb_log( "hb_init: connection problem, not waiting for "
126                         "update_thread" );
127                 break;
128             }
129             hb_snooze( 500 );
130         }
131     }
132
133     /*
134      * Initialise buffer pool
135      */
136     hb_buffer_pool_init();
137
138     /* CPU count detection */
139     hb_log( "hb_init: checking cpu count" );
140     h->cpu_count = hb_get_cpu_count();
141
142     h->list_title = hb_list_init();
143     h->jobs       = hb_list_init();
144
145     h->state_lock  = hb_lock_init();
146     h->state.state = HB_STATE_IDLE;
147
148     h->pause_lock = hb_lock_init();
149
150     h->interjob = calloc( sizeof( hb_interjob_t ), 1 );
151
152     /* libavcodec */
153     hb_avcodec_init();
154
155     /* Start library thread */
156     hb_log( "hb_init: starting libhb thread" );
157     h->die         = 0;
158     h->main_thread = hb_thread_init( "libhb", thread_func, h,
159                                      HB_NORMAL_PRIORITY );
160     hb_register( &hb_sync );
161         hb_register( &hb_decmpeg2 );
162         hb_register( &hb_decvobsub );
163     hb_register( &hb_encvobsub );
164     hb_register( &hb_deccc608 );
165         hb_register( &hb_render );
166         hb_register( &hb_encavcodec );
167         hb_register( &hb_encx264 );
168     hb_register( &hb_enctheora );
169         hb_register( &hb_deca52 );
170         hb_register( &hb_decdca );
171         hb_register( &hb_decavcodec );
172         hb_register( &hb_decavcodecv );
173         hb_register( &hb_decavcodecvi );
174         hb_register( &hb_decavcodecai );
175         hb_register( &hb_declpcm );
176         hb_register( &hb_encfaac );
177         hb_register( &hb_enclame );
178         hb_register( &hb_encvorbis );
179 #ifdef __APPLE__
180         hb_register( &hb_encca_aac );
181 #endif
182
183     return h;
184 }
185
186 /**
187  * libhb initialization routine.
188  * This version is to use when calling the dylib, the macro hb_init isn't available from a dylib call!
189  * @param verbose HB_DEBUG_NONE or HB_DEBUG_ALL.
190  * @param update_check signals libhb to check for updated version from HandBrake website.
191  * @return Handle to hb_handle_t for use on all subsequent calls to libhb.
192  */
193 hb_handle_t * hb_init_dl( int verbose, int update_check )
194 {
195     hb_handle_t * h = calloc( sizeof( hb_handle_t ), 1 );
196     uint64_t      date;
197
198     /* See hb_log() in common.c */
199     if( verbose > HB_DEBUG_NONE )
200     {
201         putenv( "HB_DEBUG=1" );
202     }
203
204     /* Check for an update on the website if asked to */
205     h->build = -1;
206
207     if( update_check )
208     {
209         hb_log( "hb_init: checking for updates" );
210         date             = hb_get_date();
211         h->update_thread = hb_update_init( &h->build, h->version );
212
213         for( ;; )
214         {
215             if( hb_thread_has_exited( h->update_thread ) )
216             {
217                 /* Immediate success or failure */
218                 hb_thread_close( &h->update_thread );
219                 break;
220             }
221             if( hb_get_date() > date + 1000 )
222             {
223                 /* Still nothing after one second. Connection problem,
224                    let the thread die */
225                 hb_log( "hb_init: connection problem, not waiting for "
226                         "update_thread" );
227                 break;
228             }
229             hb_snooze( 500 );
230         }
231     }
232
233     /* CPU count detection */
234     hb_log( "hb_init: checking cpu count" );
235     h->cpu_count = hb_get_cpu_count();
236
237     h->list_title = hb_list_init();
238     h->jobs       = hb_list_init();
239     h->current_job = NULL;
240
241     h->state_lock  = hb_lock_init();
242     h->state.state = HB_STATE_IDLE;
243
244     h->pause_lock = hb_lock_init();
245
246     /* libavcodec */
247     avcodec_init();
248     avcodec_register_all();
249
250     /* Start library thread */
251     hb_log( "hb_init: starting libhb thread" );
252     h->die         = 0;
253     h->main_thread = hb_thread_init( "libhb", thread_func, h,
254                                      HB_NORMAL_PRIORITY );
255
256     hb_register( &hb_sync );
257         hb_register( &hb_decmpeg2 );
258         hb_register( &hb_decvobsub );
259     hb_register( &hb_encvobsub );
260     hb_register( &hb_deccc608 );
261         hb_register( &hb_render );
262         hb_register( &hb_encavcodec );
263         hb_register( &hb_encx264 );
264     hb_register( &hb_enctheora );
265         hb_register( &hb_deca52 );
266         hb_register( &hb_decdca );
267         hb_register( &hb_decavcodec );
268         hb_register( &hb_decavcodecv );
269         hb_register( &hb_decavcodecvi );
270         hb_register( &hb_decavcodecai );
271         hb_register( &hb_declpcm );
272         hb_register( &hb_encfaac );
273         hb_register( &hb_enclame );
274         hb_register( &hb_encvorbis );
275 #ifdef __APPLE__
276         hb_register( &hb_encca_aac );
277 #endif
278
279         return h;
280 }
281
282
283 /**
284  * Returns current version of libhb.
285  * @param h Handle to hb_handle_t.
286  * @return character array of version number.
287  */
288 char * hb_get_version( hb_handle_t * h )
289 {
290     return HB_PROJECT_VERSION;
291 }
292
293 /**
294  * Returns current build of libhb.
295  * @param h Handle to hb_handle_t.
296  * @return character array of build number.
297  */
298 int hb_get_build( hb_handle_t * h )
299 {
300     return HB_PROJECT_BUILD;
301 }
302
303 /**
304  * Checks for needed update.
305  * @param h Handle to hb_handle_t.
306  * @param version Pointer to handle where version will be copied.
307  * @return update indicator.
308  */
309 int hb_check_update( hb_handle_t * h, char ** version )
310 {
311     *version = ( h->build < 0 ) ? NULL : h->version;
312     return h->build;
313 }
314
315 /**
316  * Sets the cpu count to the desired value.
317  * @param h Handle to hb_handle_t
318  * @param cpu_count Number of CPUs to use.
319  */
320 void hb_set_cpu_count( hb_handle_t * h, int cpu_count )
321 {
322     cpu_count    = MAX( 1, cpu_count );
323     cpu_count    = MIN( cpu_count, 8 );
324     h->cpu_count = cpu_count;
325 }
326
327 /**
328  * Deletes current previews associated with titles
329  * @param h Handle to hb_handle_t
330  */
331 void hb_remove_previews( hb_handle_t * h )
332 {
333     char            filename[1024];
334     char            dirname[1024];
335     hb_title_t    * title;
336     int             i, count, len;
337     DIR           * dir;
338     struct dirent * entry;
339
340     memset( dirname, 0, 1024 );
341     hb_get_tempory_directory( h, dirname );
342     dir = opendir( dirname );
343     if (dir == NULL) return;
344
345     count = hb_list_count( h->list_title );
346     while( ( entry = readdir( dir ) ) )
347     {
348         if( entry->d_name[0] == '.' )
349         {
350             continue;
351         }
352         for( i = 0; i < count; i++ )
353         {
354             title = hb_list_item( h->list_title, i );
355             len = snprintf( filename, 1024, "%" PRIxPTR, (intptr_t) title );
356             if (strncmp(entry->d_name, filename, len) == 0)
357             {
358                 snprintf( filename, 1024, "%s/%s", dirname, entry->d_name );
359                 unlink( filename );
360                 break;
361             }
362         }
363     }
364     closedir( dir );
365 }
366
367 /**
368  * Initializes a scan of the by calling hb_scan_init
369  * @param h Handle to hb_handle_t
370  * @param path location of VIDEO_TS folder.
371  * @param title_index Desired title to scan.  0 for all titles.
372  * @param preview_count Number of preview images to generate.
373  * @param store_previews Whether or not to write previews to disk.
374  */
375 void hb_scan( hb_handle_t * h, const char * path, int title_index,
376               int preview_count, int store_previews )
377 {
378     hb_title_t * title;
379
380     /* Clean up from previous scan */
381     hb_remove_previews( h );
382     while( ( title = hb_list_item( h->list_title, 0 ) ) )
383     {
384         hb_list_rem( h->list_title, title );
385         hb_title_close( &title );
386     }
387
388     hb_log( "hb_scan: path=%s, title_index=%d", path, title_index );
389     h->scan_thread = hb_scan_init( h, path, title_index, h->list_title,
390                                    preview_count, store_previews );
391 }
392
393 /**
394  * Returns the list of titles found.
395  * @param h Handle to hb_handle_t
396  * @return Handle to hb_list_t of the title list.
397  */
398 hb_list_t * hb_get_titles( hb_handle_t * h )
399 {
400     return h->list_title;
401 }
402
403 /**
404  * Create preview image of desired title a index of picture.
405  * @param h Handle to hb_handle_t.
406  * @param title Handle to hb_title_t of desired title.
407  * @param picture Index in title.
408  * @param buffer Handle to buufer were inage will be drawn.
409  */
410 void hb_get_preview( hb_handle_t * h, hb_title_t * title, int picture,
411                      uint8_t * buffer )
412 {
413     hb_job_t           * job = title->job;
414     char                 filename[1024];
415     FILE               * file;
416     uint8_t            * buf1, * buf2, * buf3, * buf4, * pen;
417     uint32_t             swsflags;
418     AVPicture            pic_in, pic_preview, pic_deint, pic_crop, pic_scale;
419     struct SwsContext  * context;
420     int                  i;
421     int                  rgb_width = ((job->width + 7) >> 3) << 3;
422     int                  preview_size;
423
424     swsflags = SWS_LANCZOS | SWS_ACCURATE_RND;
425
426     buf1 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
427     buf2 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
428     buf3 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, rgb_width, job->height ) );
429     buf4 = av_malloc( avpicture_get_size( PIX_FMT_RGB32, rgb_width, job->height ) );
430     avpicture_fill( &pic_in, buf1, PIX_FMT_YUV420P,
431                     title->width, title->height );
432     avpicture_fill( &pic_deint, buf2, PIX_FMT_YUV420P,
433                     title->width, title->height );
434     avpicture_fill( &pic_scale, buf3, PIX_FMT_YUV420P,
435                     rgb_width, job->height );
436     avpicture_fill( &pic_preview, buf4, PIX_FMT_RGB32,
437                     rgb_width, job->height );
438
439     // Allocate the AVPicture frames and fill in
440
441     memset( filename, 0, 1024 );
442
443     hb_get_tempory_filename( h, filename, "%" PRIxPTR "%d",
444                              (intptr_t) title, picture );
445
446     file = fopen( filename, "r" );
447     if( !file )
448     {
449         hb_log( "hb_get_preview: fopen failed" );
450         return;
451     }
452
453     fread( buf1, avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height), 1, file );
454     fclose( file );
455
456     if( job->deinterlace )
457     {
458         // Deinterlace and crop
459         avpicture_deinterlace( &pic_deint, &pic_in, PIX_FMT_YUV420P, title->width, title->height );
460         av_picture_crop( &pic_crop, &pic_deint, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
461     }
462     else
463     {
464         // Crop
465         av_picture_crop( &pic_crop, &pic_in, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
466     }
467
468     // Get scaling context
469     context = sws_getContext(title->width  - (job->crop[2] + job->crop[3]),
470                              title->height - (job->crop[0] + job->crop[1]),
471                              PIX_FMT_YUV420P,
472                              rgb_width, job->height, PIX_FMT_YUV420P,
473                              swsflags, NULL, NULL, NULL);
474
475     // Scale
476     sws_scale(context,
477               pic_crop.data, pic_crop.linesize,
478               0, title->height - (job->crop[0] + job->crop[1]),
479               pic_scale.data, pic_scale.linesize);
480
481     // Free context
482     sws_freeContext( context );
483
484     // Get preview context
485     context = sws_getContext(rgb_width, job->height, PIX_FMT_YUV420P,
486                               rgb_width, job->height, PIX_FMT_RGB32,
487                               swsflags, NULL, NULL, NULL);
488
489     // Create preview
490     sws_scale(context,
491               pic_scale.data, pic_scale.linesize,
492               0, job->height,
493               pic_preview.data, pic_preview.linesize);
494
495     // Free context
496     sws_freeContext( context );
497
498     preview_size = pic_preview.linesize[0];
499     pen = buffer;
500     for( i = 0; i < job->height; i++ )
501     {
502         memcpy( pen, buf4 + preview_size * i, 4 * job->width );
503         pen += 4 * job->width;
504     }
505
506     // Clean up
507     avpicture_free( &pic_preview );
508     avpicture_free( &pic_scale );
509     avpicture_free( &pic_deint );
510     avpicture_free( &pic_in );
511 }
512
513  /**
514  * Analyzes a frame to detect interlacing artifacts
515  * and returns true if interlacing (combing) is found.
516  *
517  * Code taken from Thomas Oestreich's 32detect filter
518  * in the Transcode project, with minor formatting changes.
519  *
520  * @param buf         An hb_buffer structure holding valid frame data
521  * @param width       The frame's width in pixels
522  * @param height      The frame's height in pixels
523  * @param color_equal Sensitivity for detecting similar colors
524  * @param color_diff  Sensitivity for detecting different colors
525  * @param threshold   Sensitivity for flagging planes as combed
526  * @param prog_equal  Sensitivity for detecting similar colors on progressive frames
527  * @param prog_diff   Sensitivity for detecting different colors on progressive frames
528  * @param prog_threshold Sensitivity for flagging progressive frames as combed
529  */
530 int hb_detect_comb( hb_buffer_t * buf, int width, int height, int color_equal, int color_diff, int threshold, int prog_equal, int prog_diff, int prog_threshold )
531 {
532     int j, k, n, off, cc_1, cc_2, cc[3];
533         // int flag[3] ; // debugging flag
534     uint16_t s1, s2, s3, s4;
535     cc_1 = 0; cc_2 = 0;
536
537     int offset = 0;
538     
539     if ( buf->flags & 16 )
540     {
541         /* Frame is progressive, be more discerning. */
542         color_diff = prog_diff;
543         color_equal = prog_equal;
544         threshold = prog_threshold;
545     }
546
547     /* One pas for Y, one pass for Cb, one pass for Cr */    
548     for( k = 0; k < 3; k++ )
549     {
550         if( k == 1 )
551         {
552             /* Y has already been checked, now offset by Y's dimensions
553                and divide all the other values by 2, since Cr and Cb
554                are half-size compared to Y.                               */
555             offset = width * height;
556             width >>= 1;
557             height >>= 1;
558         }
559         else if ( k == 2 )
560         {
561             /* Y and Cb are done, so the offset needs to be bumped
562                so it's width*height + (width / 2) * (height / 2)  */
563             offset *= 5/4;
564         }
565
566         for( j = 0; j < width; ++j )
567         {
568             off = 0;
569
570             for( n = 0; n < ( height - 4 ); n = n + 2 )
571             {
572                 /* Look at groups of 4 sequential horizontal lines */
573                 s1 = ( ( buf->data + offset )[ off + j             ] & 0xff );
574                 s2 = ( ( buf->data + offset )[ off + j + width     ] & 0xff );
575                 s3 = ( ( buf->data + offset )[ off + j + 2 * width ] & 0xff );
576                 s4 = ( ( buf->data + offset )[ off + j + 3 * width ] & 0xff );
577
578                 /* Note if the 1st and 2nd lines are more different in
579                    color than the 1st and 3rd lines are similar in color.*/
580                 if ( ( abs( s1 - s3 ) < color_equal ) &&
581                      ( abs( s1 - s2 ) > color_diff ) )
582                         ++cc_1;
583
584                 /* Note if the 2nd and 3rd lines are more different in
585                    color than the 2nd and 4th lines are similar in color.*/
586                 if ( ( abs( s2 - s4 ) < color_equal ) &&
587                      ( abs( s2 - s3 ) > color_diff) )
588                         ++cc_2;
589
590                 /* Now move down 2 horizontal lines before starting over.*/
591                 off += 2 * width;
592             }
593         }
594
595         // compare results
596         /*  The final cc score for a plane is the percentage of combed pixels it contains.
597             Because sensitivity goes down to hundreths of a percent, multiply by 1000
598             so it will be easy to compare against the threhold value which is an integer. */
599         cc[k] = (int)( ( cc_1 + cc_2 ) * 1000.0 / ( width * height ) );
600     }
601
602
603     /* HandBrake is all yuv420, so weight the average percentage of all 3 planes accordingly.*/
604     int average_cc = ( 2 * cc[0] + ( cc[1] / 2 ) + ( cc[2] / 2 ) ) / 3;
605     
606     /* Now see if that average percentage of combed pixels surpasses the threshold percentage given by the user.*/
607     if( average_cc > threshold )
608     {
609 #if 0
610             hb_log("Average %i combed (Threshold %i) %i/%i/%i | PTS: %lld (%fs) %s", average_cc, threshold, cc[0], cc[1], cc[2], buf->start, (float)buf->start / 90000, (buf->flags & 16) ? "Film" : "Video" );
611 #endif
612         return 1;
613     }
614
615 #if 0
616     hb_log("SKIPPED Average %i combed (Threshold %i) %i/%i/%i | PTS: %lld (%fs) %s", average_cc, threshold, cc[0], cc[1], cc[2], buf->start, (float)buf->start / 90000, (buf->flags & 16) ? "Film" : "Video" );
617 #endif
618
619     /* Reaching this point means no combing detected. */
620     return 0;
621
622 }
623
624 /**
625  * Calculates job width and height for anamorphic content,
626  *
627  * @param job Handle to hb_job_t
628  * @param output_width Pointer to returned storage width
629  * @param output_height Pointer to returned storage height
630  * @param output_par_width Pointer to returned pixel width
631  @ param output_par_height Pointer to returned pixel height
632  */
633 void hb_set_anamorphic_size( hb_job_t * job,
634         int *output_width, int *output_height,
635         int *output_par_width, int *output_par_height )
636 {
637     /* Set up some variables to make the math easier to follow. */
638     hb_title_t * title = job->title;
639     int cropped_width = title->width - job->crop[2] - job->crop[3] ;
640     int cropped_height = title->height - job->crop[0] - job->crop[1] ;
641     double storage_aspect = (double)cropped_width / (double)cropped_height;
642     int mod = job->anamorphic.modulus ? job->anamorphic.modulus : 16;
643     double aspect = title->aspect;
644     
645     int pixel_aspect_width  = job->anamorphic.par_width;
646     int pixel_aspect_height = job->anamorphic.par_height;
647
648     /* If a source was really NTSC or PAL and the user specified ITU PAR
649        values, replace the standard PAR values with the ITU broadcast ones. */
650     if( title->width == 720 && job->anamorphic.itu_par )
651     {
652         // convert aspect to a scaled integer so we can test for 16:9 & 4:3
653         // aspect ratios ignoring insignificant differences in the LSBs of
654         // the floating point representation.
655         int iaspect = aspect * 9.;
656
657         /* Handle ITU PARs */
658         if (title->height == 480)
659         {
660             /* It's NTSC */
661             if (iaspect == 16)
662             {
663                 /* It's widescreen */
664                 pixel_aspect_width = 40;
665                 pixel_aspect_height = 33;
666             }
667             else if (iaspect == 12)
668             {
669                 /* It's 4:3 */
670                 pixel_aspect_width = 10;
671                 pixel_aspect_height = 11;
672             }
673         }
674         else if (title->height == 576)
675         {
676             /* It's PAL */
677             if(iaspect == 16)
678             {
679                 /* It's widescreen */
680                 pixel_aspect_width = 16;
681                 pixel_aspect_height = 11;
682             }
683             else if (iaspect == 12)
684             {
685                 /* It's 4:3 */
686                 pixel_aspect_width = 12;
687                 pixel_aspect_height = 11;
688             }
689         }
690     }
691
692     /* Figure out what width the source would display at. */
693     int source_display_width = cropped_width * (double)pixel_aspect_width /
694                                (double)pixel_aspect_height ;
695
696     /*
697        3 different ways of deciding output dimensions:
698         - 1: Strict anamorphic, preserve source dimensions
699         - 2: Loose anamorphic, round to mod16 and preserve storage aspect ratio
700         - 3: Power user anamorphic, specify everything
701     */
702     int width, height;
703     switch( job->anamorphic.mode )
704     {
705         case 1:
706             /* Strict anamorphic */
707             *output_width = cropped_width;
708             *output_height = cropped_height;
709             *output_par_width = title->pixel_aspect_width;
710             *output_par_height = title->pixel_aspect_height;
711         break;
712
713         case 2:
714             /* "Loose" anamorphic.
715                 - Uses mod16-compliant dimensions,
716                 - Allows users to set the width
717             */
718             width = job->width;
719             // height: Gets set later, ignore user job->height value
720
721             /* Gotta handle bounding dimensions.
722                If the width is too big, just reset it with no rescaling.
723                Instead of using the aspect-scaled job height,
724                we need to see if the job width divided by the storage aspect
725                is bigger than the max. If so, set it to the max (this is sloppy).
726                If not, set job height to job width divided by storage aspect.
727             */
728
729             if ( job->maxWidth && (job->maxWidth < job->width) )
730                 width = job->maxWidth;
731
732             /* Time to get picture width that divide cleanly.*/
733             width  = MULTIPLE_MOD( width, mod);
734
735             /* Verify these new dimensions don't violate max height and width settings */
736             if ( job->maxWidth && (job->maxWidth < job->width) )
737                 width = job->maxWidth;
738
739             height = ((double)width / storage_aspect) + 0.5;
740             
741             if ( job->maxHeight && (job->maxHeight < height) )
742                 height = job->maxHeight;
743
744             /* Time to get picture height that divide cleanly.*/
745             height = MULTIPLE_MOD( height, mod);
746
747             /* Verify these new dimensions don't violate max height and width settings */
748             if ( job->maxHeight && (job->maxHeight < height) )
749                 height = job->maxHeight;
750
751             /* The film AR is the source's display width / cropped source height.
752                The output display width is the output height * film AR.
753                The output PAR is the output display width / output storage width. */
754             pixel_aspect_width = height * source_display_width / cropped_height;
755             pixel_aspect_height = width;
756
757             /* Pass the results back to the caller */
758             *output_width = width;
759             *output_height = height;
760         break;
761             
762         case 3:
763             /* Anamorphic 3: Power User Jamboree
764                - Set everything based on specified values */
765             
766             /* Use specified storage dimensions */
767             width = job->width;
768             height = job->height;
769             
770             /* Bind to max dimensions */
771             if( job->maxWidth && width > job->maxWidth )
772                 width = job->maxWidth;
773             if( job->maxHeight && height > job->maxHeight )
774                 height = job->maxHeight;
775             
776             /* Time to get picture dimensions that divide cleanly.*/
777             width  = MULTIPLE_MOD( width, mod);
778             height = MULTIPLE_MOD( height, mod);
779             
780             /* Verify we're still within max dimensions */
781             if( job->maxWidth && width > job->maxWidth )
782                 width = job->maxWidth - (mod/2);
783             if( job->maxHeight && height > job->maxHeight )
784                 height = job->maxHeight - (mod/2);
785                 
786             /* Re-ensure we have picture dimensions that divide cleanly. */
787             width  = MULTIPLE_MOD( width, mod );
788             height = MULTIPLE_MOD( height, mod );
789             
790             /* That finishes the storage dimensions. On to display. */            
791             if( job->anamorphic.dar_width && job->anamorphic.dar_height )
792             {
793                 /* We need to adjust the PAR to produce this aspect. */
794                 pixel_aspect_width = height * job->anamorphic.dar_width / job->anamorphic.dar_height;
795                 pixel_aspect_height = width;
796             }
797             else
798             {
799                 /* If we're doing ana 3 and not specifying a DAR, care needs to be taken.
800                    This indicates a PAR is potentially being set by the interface. But
801                    this is an output PAR, to correct a source, and it should not be assumed
802                    that it properly creates a display aspect ratio when applied to the source,
803                    which could easily be stored in a different resolution. */
804                    
805                 int output_display_width = width * (double)pixel_aspect_width /
806                     (double)pixel_aspect_height;
807                 pixel_aspect_width = output_display_width;
808                 pixel_aspect_height = width;
809             }
810             
811             /* Back to caller */
812             *output_width = width;
813             *output_height = height;
814         break;
815     }
816     
817     /* While x264 is smart enough to reduce fractions on its own, libavcodec
818        needs some help with the math, so lose superfluous factors.            */
819     hb_reduce( output_par_width, output_par_height,
820                pixel_aspect_width, pixel_aspect_height );
821 }
822
823 /**
824  * Calculates job width, height, and cropping parameters.
825  * @param job Handle to hb_job_t.
826  * @param aspect Desired aspect ratio. Value of -1 uses title aspect.
827  * @param pixels Maximum desired pixel count.
828  */
829 void hb_set_size( hb_job_t * job, double aspect, int pixels )
830 {
831     hb_title_t * title = job->title;
832
833     int croppedWidth  = title->width - title->crop[2] - title->crop[3];
834     int croppedHeight = title->height - title->crop[0] - title->crop[1];
835     double croppedAspect = title->aspect * title->height * croppedWidth /
836                            croppedHeight / title->width;
837     int addCrop;
838     int i, w, h;
839
840     if( aspect <= 0 )
841     {
842         /* Keep the best possible aspect ratio */
843         aspect = croppedAspect;
844     }
845
846     /* Crop if necessary to obtain the desired ratio */
847     memcpy( job->crop, title->crop, 4 * sizeof( int ) );
848     if( aspect < croppedAspect )
849     {
850         /* Need to crop on the left and right */
851         addCrop = croppedWidth - aspect * croppedHeight * title->width /
852                     title->aspect / title->height;
853         if( addCrop & 3 )
854         {
855             addCrop = ( addCrop + 1 ) / 2;
856             job->crop[2] += addCrop;
857             job->crop[3] += addCrop;
858         }
859         else if( addCrop & 2 )
860         {
861             addCrop /= 2;
862             job->crop[2] += addCrop - 1;
863             job->crop[3] += addCrop + 1;
864         }
865         else
866         {
867             addCrop /= 2;
868             job->crop[2] += addCrop;
869             job->crop[3] += addCrop;
870         }
871     }
872     else if( aspect > croppedAspect )
873     {
874         /* Need to crop on the top and bottom */
875         addCrop = croppedHeight - croppedWidth * title->aspect *
876             title->height / aspect / title->width;
877         if( addCrop & 3 )
878         {
879             addCrop = ( addCrop + 1 ) / 2;
880             job->crop[0] += addCrop;
881             job->crop[1] += addCrop;
882         }
883         else if( addCrop & 2 )
884         {
885             addCrop /= 2;
886             job->crop[0] += addCrop - 1;
887             job->crop[1] += addCrop + 1;
888         }
889         else
890         {
891             addCrop /= 2;
892             job->crop[0] += addCrop;
893             job->crop[1] += addCrop;
894         }
895     }
896
897     /* Compute a resolution from the number of pixels and aspect */
898     for( i = 0;; i++ )
899     {
900         w = 16 * i;
901         h = MULTIPLE_16( (int)( (double)w / aspect ) );
902         if( w * h > pixels )
903         {
904             break;
905         }
906     }
907     i--;
908     job->width  = 16 * i;
909     job->height = MULTIPLE_16( (int)( (double)job->width / aspect ) );
910 }
911
912 /**
913  * Returns the number of jobs in the queue.
914  * @param h Handle to hb_handle_t.
915  * @return Number of jobs.
916  */
917 int hb_count( hb_handle_t * h )
918 {
919     return hb_list_count( h->jobs );
920 }
921
922 /**
923  * Returns handle to job at index i within the job list.
924  * @param h Handle to hb_handle_t.
925  * @param i Index of job.
926  * @returns Handle to hb_job_t of desired job.
927  */
928 hb_job_t * hb_job( hb_handle_t * h, int i )
929 {
930     return hb_list_item( h->jobs, i );
931 }
932
933 hb_job_t * hb_current_job( hb_handle_t * h )
934 {
935     return( h->current_job );
936 }
937
938 /**
939  * Adds a job to the job list.
940  * @param h Handle to hb_handle_t.
941  * @param job Handle to hb_job_t.
942  */
943 void hb_add( hb_handle_t * h, hb_job_t * job )
944 {
945     hb_job_t      * job_copy;
946     hb_title_t    * title,    * title_copy;
947     hb_chapter_t  * chapter,  * chapter_copy;
948     hb_audio_t    * audio;
949     hb_subtitle_t * subtitle, * subtitle_copy;
950     int             i;
951     char            audio_lang[4];
952
953     /* Copy the title */
954     title      = job->title;
955     title_copy = malloc( sizeof( hb_title_t ) );
956     memcpy( title_copy, title, sizeof( hb_title_t ) );
957
958     title_copy->list_chapter = hb_list_init();
959     for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
960     {
961         chapter      = hb_list_item( title->list_chapter, i );
962         chapter_copy = malloc( sizeof( hb_chapter_t ) );
963         memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
964         hb_list_add( title_copy->list_chapter, chapter_copy );
965     }
966
967     /*
968      * Copy the metadata
969      */
970     if( title->metadata )
971     {
972         title_copy->metadata = malloc( sizeof( hb_metadata_t ) );
973         
974         if( title_copy->metadata ) 
975         {
976             memcpy( title_copy->metadata, title->metadata, sizeof( hb_metadata_t ) );
977
978             /*
979              * Need to copy the artwork seperatly (TODO).
980              */
981             if( title->metadata->coverart )
982             {
983                 title_copy->metadata->coverart = malloc( title->metadata->coverart_size );
984                 if( title_copy->metadata->coverart )
985                 {
986                     memcpy( title_copy->metadata->coverart, title->metadata->coverart,
987                             title->metadata->coverart_size );
988                 } else {
989                     title_copy->metadata->coverart_size = 0; 
990                 }
991             }
992         }
993     }
994
995     /* Copy the audio track(s) we want */
996     title_copy->list_audio = hb_list_init();
997
998     for( i = 0; i < hb_list_count(job->list_audio); i++ )
999     {
1000         if( ( audio = hb_list_item( job->list_audio, i ) ) )
1001         {
1002             hb_list_add( title_copy->list_audio, hb_audio_copy(audio) );
1003         }
1004     }
1005
1006     title_copy->list_subtitle = hb_list_init();
1007
1008     /*
1009      * The following code is confusing, there are three ways in which
1010      * we select subtitles and it depends on whether this is single or
1011      * two pass mode.
1012      *
1013      * subtitle_scan may be enabled, in which case the first pass
1014      * scans all subtitles of that language. The second pass does not
1015      * select any because they are set at the end of the first pass.
1016      *
1017      * native_language may have a preferred language, in which case we
1018      * may be switching the language we want for the subtitles in the
1019      * first pass of a single pass, or the second pass of a two pass.
1020      *
1021      * We may have manually selected a subtitle, in which case that is
1022      * selected in the first pass of a single pass, or the second of a
1023      * two pass.
1024      */
1025     memset( audio_lang, 0, sizeof( audio_lang ) );
1026
1027     if ( job->indepth_scan || job->native_language ) {
1028
1029         /*
1030          * Find the first audio language that is being encoded
1031          */
1032         for( i = 0; i < hb_list_count(job->list_audio); i++ )
1033         {
1034             if( ( audio = hb_list_item( job->list_audio, i ) ) )
1035             {
1036                 strncpy(audio_lang, audio->config.lang.iso639_2, sizeof(audio_lang));
1037                 break;
1038             }
1039         }
1040
1041         /*
1042          * In all cases switch the language if we need to to our native
1043          * language.
1044          */
1045         if( job->native_language )
1046         {
1047             if( strncasecmp( job->native_language, audio_lang,
1048                              sizeof( audio_lang ) ) != 0 )
1049             {
1050
1051                 if( job->pass != 2 )
1052                 {
1053                     hb_log( "Enabled subtitles in native language '%s', audio is in '%s'",
1054                             job->native_language, audio_lang);
1055                 }
1056                 /*
1057                  * The main audio track is not in our native language, so switch
1058                  * the subtitles to use our native language instead.
1059                  */
1060                 strncpy( audio_lang, job->native_language, sizeof( audio_lang ) );
1061             } else {
1062                 /*
1063                  * native language is irrelevent, free it.
1064                  */
1065                 free( job->native_language );
1066                 job->native_language = NULL;
1067             }
1068         }
1069     }
1070
1071     /*
1072      * If doing a subtitle scan then add all the matching subtitles for this
1073      * language.
1074      */
1075     if ( job->indepth_scan )
1076     {
1077         for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1078         {
1079             subtitle = hb_list_item( title->list_subtitle, i );
1080             if( strcmp( subtitle->iso639_2, audio_lang ) == 0 &&
1081                 subtitle->source == VOBSUB )
1082             {
1083                 /*
1084                  * Matched subtitle language with audio language, so
1085                  * add this to our list to scan.
1086                  *
1087                  * We will update the subtitle list on the second pass
1088                  * later after the first pass has completed.
1089                  */
1090                 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1091                 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1092                 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1093                 if ( job->native_language ) {
1094                     /*
1095                      * With native language just select the
1096                      * first match in our langiage, not all of
1097                      * them. Subsequent ones are likely to be commentary
1098                      */
1099                     break;
1100                 }
1101             }
1102         }
1103     } else {
1104         /*
1105          * Not doing a subtitle scan in this pass, but maybe we are in the
1106          * first pass?
1107          */
1108         if( job->pass != 1 && job->native_language )
1109         {
1110             /*
1111              * We are not doing a subtitle scan but do want the
1112              * native langauge subtitle selected, so select it
1113              * for pass 0 or pass 2 of a two pass.
1114              */
1115             for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1116             {
1117                 subtitle = hb_list_item( title->list_subtitle, i );
1118                 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1119                 {
1120                     /*
1121                      * Matched subtitle language with audio language, so
1122                      * add this to our list to scan.
1123                      */
1124                     subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1125                     memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1126                     hb_list_add( title_copy->list_subtitle, subtitle_copy );
1127                     break;
1128                 }
1129             }
1130         } else {
1131             /*
1132              * Manually selected subtitles, in which case only
1133              * bother adding them for pass 0 or pass 2 of a two
1134              * pass.
1135              */
1136             if( job->pass != 1 )
1137             {
1138                 /*
1139                  * Copy all of them from the input job, to the title_copy/job_copy.
1140                  */
1141                 for(  i = 0; i < hb_list_count(job->list_subtitle); i++ ) {
1142                     if( ( subtitle = hb_list_item( job->list_subtitle, i ) ) )
1143                     {
1144                         subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1145                         memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1146                         hb_list_add( title_copy->list_subtitle, subtitle_copy );
1147                     }
1148                 }
1149             }
1150         }
1151     }
1152
1153     /* Copy the job */
1154     job_copy        = calloc( sizeof( hb_job_t ), 1 );
1155     memcpy( job_copy, job, sizeof( hb_job_t ) );
1156     title_copy->job = job_copy;
1157     job_copy->title = title_copy;
1158     job_copy->list_audio = title_copy->list_audio;
1159     job_copy->list_subtitle = title_copy->list_subtitle;   // sharing list between title and job
1160     job_copy->file  = strdup( job->file );
1161     job_copy->h     = h;
1162     job_copy->pause = h->pause_lock;
1163
1164     /* Copy the job filter list */
1165     if( job->filters )
1166     {
1167         int i;
1168         int filter_count = hb_list_count( job->filters );
1169         job_copy->filters = hb_list_init();
1170         for( i = 0; i < filter_count; i++ )
1171         {
1172             /*
1173              * Copy the filters, since the MacGui reuses the global filter objects
1174              * meaning that queued up jobs overwrite the previous filter settings.
1175              * In reality, settings is probably the only field that needs duplicating
1176              * since it's the only value that is ever changed. But name is duplicated
1177              * as well for completeness. Not copying private_data since it gets
1178              * created for each job in renderInit.
1179              */
1180             hb_filter_object_t * filter = hb_list_item( job->filters, i );
1181             hb_filter_object_t * filter_copy = malloc( sizeof( hb_filter_object_t ) );
1182             memcpy( filter_copy, filter, sizeof( hb_filter_object_t ) );
1183             if( filter->name )
1184                 filter_copy->name = strdup( filter->name );
1185             if( filter->settings )
1186                 filter_copy->settings = strdup( filter->settings );
1187             hb_list_add( job_copy->filters, filter_copy );
1188         }
1189     }
1190
1191     /* Add the job to the list */
1192     hb_list_add( h->jobs, job_copy );
1193     h->job_count = hb_count(h);
1194     h->job_count_permanent++;
1195 }
1196
1197 /**
1198  * Removes a job from the job list.
1199  * @param h Handle to hb_handle_t.
1200  * @param job Handle to hb_job_t.
1201  */
1202 void hb_rem( hb_handle_t * h, hb_job_t * job )
1203 {
1204     hb_list_rem( h->jobs, job );
1205
1206     h->job_count = hb_count(h);
1207     if (h->job_count_permanent)
1208         h->job_count_permanent--;
1209
1210     /* XXX free everything XXX */
1211 }
1212
1213 /**
1214  * Starts the conversion process.
1215  * Sets state to HB_STATE_WORKING.
1216  * calls hb_work_init, to launch work thread. Stores handle to work thread.
1217  * @param h Handle to hb_handle_t.
1218  */
1219 void hb_start( hb_handle_t * h )
1220 {
1221     /* XXX Hack */
1222     h->job_count = hb_list_count( h->jobs );
1223     h->job_count_permanent = h->job_count;
1224
1225     hb_lock( h->state_lock );
1226     h->state.state = HB_STATE_WORKING;
1227 #define p h->state.param.working
1228     p.progress  = 0.0;
1229     p.job_cur   = 1;
1230     p.job_count = h->job_count;
1231     p.rate_cur  = 0.0;
1232     p.rate_avg  = 0.0;
1233     p.hours     = -1;
1234     p.minutes   = -1;
1235     p.seconds   = -1;
1236     p.sequence_id = 0;
1237 #undef p
1238     hb_unlock( h->state_lock );
1239
1240     h->paused = 0;
1241
1242     h->work_die    = 0;
1243     h->work_thread = hb_work_init( h->jobs, h->cpu_count,
1244                                    &h->work_die, &h->work_error, &h->current_job );
1245 }
1246
1247 /**
1248  * Pauses the conversion process.
1249  * @param h Handle to hb_handle_t.
1250  */
1251 void hb_pause( hb_handle_t * h )
1252 {
1253     if( !h->paused )
1254     {
1255         hb_lock( h->pause_lock );
1256         h->paused = 1;
1257
1258         hb_lock( h->state_lock );
1259         h->state.state = HB_STATE_PAUSED;
1260         hb_unlock( h->state_lock );
1261     }
1262 }
1263
1264 /**
1265  * Resumes the conversion process.
1266  * @param h Handle to hb_handle_t.
1267  */
1268 void hb_resume( hb_handle_t * h )
1269 {
1270     if( h->paused )
1271     {
1272         hb_unlock( h->pause_lock );
1273         h->paused = 0;
1274     }
1275 }
1276
1277 /**
1278  * Stops the conversion process.
1279  * @param h Handle to hb_handle_t.
1280  */
1281 void hb_stop( hb_handle_t * h )
1282 {
1283     h->work_die = 1;
1284
1285     h->job_count = hb_count(h);
1286     h->job_count_permanent = 0;
1287
1288     hb_resume( h );
1289 }
1290
1291 /**
1292  * Returns the state of the conversion process.
1293  * @param h Handle to hb_handle_t.
1294  * @param s Handle to hb_state_t which to copy the state data.
1295  */
1296 void hb_get_state( hb_handle_t * h, hb_state_t * s )
1297 {
1298     hb_lock( h->state_lock );
1299
1300     memcpy( s, &h->state, sizeof( hb_state_t ) );
1301     if ( h->state.state == HB_STATE_SCANDONE || h->state.state == HB_STATE_WORKDONE )
1302         h->state.state = HB_STATE_IDLE;
1303
1304     hb_unlock( h->state_lock );
1305 }
1306
1307 void hb_get_state2( hb_handle_t * h, hb_state_t * s )
1308 {
1309     hb_lock( h->state_lock );
1310
1311     memcpy( s, &h->state, sizeof( hb_state_t ) );
1312
1313     hb_unlock( h->state_lock );
1314 }
1315
1316 /**
1317  * Called in MacGui in UpdateUI to check
1318  *  for a new scan being completed to set a new source
1319  */
1320 int hb_get_scancount( hb_handle_t * h)
1321  {
1322      return h->scanCount;
1323  }
1324
1325 /**
1326  * Closes access to libhb by freeing the hb_handle_t handle ontained in hb_init.
1327  * @param _h Pointer to handle to hb_handle_t.
1328  */
1329 void hb_close( hb_handle_t ** _h )
1330 {
1331     hb_handle_t * h = *_h;
1332     hb_title_t * title;
1333
1334     h->die = 1;
1335     hb_thread_close( &h->main_thread );
1336
1337     while( ( title = hb_list_item( h->list_title, 0 ) ) )
1338     {
1339         hb_list_rem( h->list_title, title );
1340         if( title->job && title->job->filters )
1341         {
1342             hb_list_close( &title->job->filters );
1343         }
1344         free( title->job );
1345         hb_title_close( &title );
1346     }
1347     hb_list_close( &h->list_title );
1348
1349     hb_list_close( &h->jobs );
1350     hb_lock_close( &h->state_lock );
1351     hb_lock_close( &h->pause_lock );
1352     free( h );
1353     *_h = NULL;
1354
1355 }
1356
1357 /**
1358  * Monitors the state of the update, scan, and work threads.
1359  * Sets scan done state when scan thread exits.
1360  * Sets work done state when work thread exits.
1361  * @param _h Handle to hb_handle_t
1362  */
1363 static void thread_func( void * _h )
1364 {
1365     hb_handle_t * h = (hb_handle_t *) _h;
1366     char dirname[1024];
1367     DIR * dir;
1368     struct dirent * entry;
1369
1370     h->pid = getpid();
1371
1372     /* Create folder for temporary files */
1373     memset( dirname, 0, 1024 );
1374     hb_get_tempory_directory( h, dirname );
1375
1376     hb_mkdir( dirname );
1377
1378     while( !h->die )
1379     {
1380         /* In case the check_update thread hangs, it'll die sooner or
1381            later. Then, we join it here */
1382         if( h->update_thread &&
1383             hb_thread_has_exited( h->update_thread ) )
1384         {
1385             hb_thread_close( &h->update_thread );
1386         }
1387
1388         /* Check if the scan thread is done */
1389         if( h->scan_thread &&
1390             hb_thread_has_exited( h->scan_thread ) )
1391         {
1392             hb_thread_close( &h->scan_thread );
1393
1394             hb_log( "libhb: scan thread found %d valid title(s)",
1395                     hb_list_count( h->list_title ) );
1396             hb_lock( h->state_lock );
1397             h->state.state = HB_STATE_SCANDONE; //originally state.state
1398                         hb_unlock( h->state_lock );
1399                         /*we increment this sessions scan count by one for the MacGui
1400                         to trigger a new source being set */
1401             h->scanCount++;
1402         }
1403
1404         /* Check if the work thread is done */
1405         if( h->work_thread &&
1406             hb_thread_has_exited( h->work_thread ) )
1407         {
1408             hb_thread_close( &h->work_thread );
1409
1410             hb_log( "libhb: work result = %d",
1411                     h->work_error );
1412             hb_lock( h->state_lock );
1413             h->state.state                = HB_STATE_WORKDONE;
1414             h->state.param.workdone.error = h->work_error;
1415
1416             h->job_count = hb_count(h);
1417             if (h->job_count < 1)
1418                 h->job_count_permanent = 0;
1419             hb_unlock( h->state_lock );
1420         }
1421
1422         hb_snooze( 50 );
1423     }
1424
1425     if( h->work_thread )
1426     {
1427         hb_stop( h );
1428         hb_thread_close( &h->work_thread );
1429     }
1430
1431     /* Remove temp folder */
1432     dir = opendir( dirname );
1433     if (dir)
1434     {
1435         while( ( entry = readdir( dir ) ) )
1436         {
1437             char filename[1024];
1438             if( entry->d_name[0] == '.' )
1439             {
1440                 continue;
1441             }
1442             memset( filename, 0, 1024 );
1443             snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
1444             unlink( filename );
1445         }
1446         closedir( dir );
1447         rmdir( dirname );
1448     }
1449 }
1450
1451 /**
1452  * Returns the PID.
1453  * @param h Handle to hb_handle_t
1454  */
1455 int hb_get_pid( hb_handle_t * h )
1456 {
1457     return h->pid;
1458 }
1459
1460 /**
1461  * Sets the current state.
1462  * @param h Handle to hb_handle_t
1463  * @param s Handle to new hb_state_t
1464  */
1465 void hb_set_state( hb_handle_t * h, hb_state_t * s )
1466 {
1467     hb_lock( h->pause_lock );
1468     hb_lock( h->state_lock );
1469     memcpy( &h->state, s, sizeof( hb_state_t ) );
1470     if( h->state.state == HB_STATE_WORKING )
1471     {
1472         /* XXX Hack */
1473         if (h->job_count < 1)
1474             h->job_count_permanent = 1;
1475
1476         h->state.param.working.job_cur =
1477             h->job_count_permanent - hb_list_count( h->jobs );
1478         h->state.param.working.job_count = h->job_count_permanent;
1479
1480         // Set which job is being worked on
1481         if (h->current_job)
1482             h->state.param.working.sequence_id = h->current_job->sequence_id;
1483         else
1484             h->state.param.working.sequence_id = 0;
1485     }
1486     hb_unlock( h->state_lock );
1487     hb_unlock( h->pause_lock );
1488 }
1489
1490 /* Passes a pointer to persistent data */
1491 hb_interjob_t * hb_interjob_get( hb_handle_t * h )
1492 {
1493     return h->interjob;
1494 }