1 /* $Id: common.c,v 1.15 2005/03/17 19:22:47 titer Exp $
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. */
13 /**********************************************************************
15 *********************************************************************/
16 hb_rate_t hb_video_rates
[] =
17 { { "5", 5400000 }, { "10", 2700000 }, { "12", 2250000 },
18 { "15", 1800000 }, { "23.976", 1126125 }, { "24", 1125000 },
19 { "25", 1080000 }, { "29.97", 900900 } };
20 int hb_video_rates_count
= sizeof( hb_video_rates
) /
23 hb_rate_t hb_audio_rates
[] =
24 { { "22.05", 22050 }, { "24", 24000 }, { "32", 32000 },
25 { "44.1", 44100 }, { "48", 48000 } };
26 int hb_audio_rates_count
= sizeof( hb_audio_rates
) /
28 int hb_audio_rates_default
= 3; /* 44100 Hz */
30 hb_rate_t hb_audio_bitrates
[] =
31 { { "32", 32 }, { "40", 40 }, { "48", 48 }, { "56", 56 },
32 { "64", 64 }, { "80", 80 }, { "96", 96 }, { "112", 112 },
33 { "128", 128 }, { "160", 160 }, { "192", 192 }, { "224", 224 },
34 { "256", 256 }, { "320", 320 }, { "384", 384 } };
35 int hb_audio_bitrates_count
= sizeof( hb_audio_bitrates
) /
37 int hb_audio_bitrates_default
= 8; /* 128 kbps */
39 static hb_error_handler_t
*error_handler
= NULL
;
41 hb_mixdown_t hb_audio_mixdowns
[] =
42 { { "Mono", "HB_AMIXDOWN_MONO", "mono", HB_AMIXDOWN_MONO
},
43 { "Stereo", "HB_AMIXDOWN_STEREO", "stereo", HB_AMIXDOWN_STEREO
},
44 { "Dolby Surround", "HB_AMIXDOWN_DOLBY", "dpl1", HB_AMIXDOWN_DOLBY
},
45 { "Dolby Pro Logic II", "HB_AMIXDOWN_DOLBYPLII", "dpl2", HB_AMIXDOWN_DOLBYPLII
},
46 { "6-channel discrete", "HB_AMIXDOWN_6CH", "6ch", HB_AMIXDOWN_6CH
},
47 { "AC-3 Pass-through", "HB_AMIXDOWN_AC3", "ac-3", HB_AMIXDOWN_AC3
},
48 { "Dolby PLII + AC-3", "HB_AMIXDOWN_DOLBYPLII_AC3", "dpl2+ac3", HB_AMIXDOWN_DOLBYPLII_AC3
},
50 int hb_audio_mixdowns_count
= sizeof( hb_audio_mixdowns
) /
51 sizeof( hb_mixdown_t
);
53 int hb_mixdown_get_mixdown_from_short_name( const char * short_name
)
56 for (i
= 0; i
< hb_audio_mixdowns_count
; i
++)
58 if (strcmp(hb_audio_mixdowns
[i
].short_name
, short_name
) == 0)
60 return hb_audio_mixdowns
[i
].amixdown
;
66 const char * hb_mixdown_get_short_name_from_mixdown( int amixdown
)
69 for (i
= 0; i
< hb_audio_mixdowns_count
; i
++)
71 if (hb_audio_mixdowns
[i
].amixdown
== amixdown
)
73 return hb_audio_mixdowns
[i
].short_name
;
79 /**********************************************************************
81 **********************************************************************
82 * Given a numerator (num) and a denominator (den), reduce them to an
83 * equivalent fraction and store the result in x and y.
84 *********************************************************************/
85 void hb_reduce( int *x
, int *y
, int num
, int den
)
87 int lower
= MIN( num
, den
);
91 for( i
= lower
- 1; i
> 1; --i
)
93 if( ( num
% i
== 0 ) && ( den
% i
== 0 ) )
102 /**********************************************************************
104 **********************************************************************
105 * Given the output width (if HB_KEEP_WIDTH) or height
106 * (HB_KEEP_HEIGHT) and the current crop values, calculates the
107 * correct height or width in order to respect the DVD aspect ratio
108 *********************************************************************/
109 void hb_fix_aspect( hb_job_t
* job
, int keep
)
111 hb_title_t
* title
= job
->title
;
115 Widths and heights must be multiples of 16 and greater than or
117 Crop values must be multiples of 2, greater than or equal to 0
118 and less than half of the dimension */
119 job
->width
= MULTIPLE_16( job
->width
);
120 job
->height
= MULTIPLE_16( job
->height
);
121 job
->width
= MAX( 16, job
->width
);
122 job
->height
= MAX( 16, job
->height
);
123 for( i
= 0; i
< 4; i
++ )
125 job
->crop
[i
] = EVEN( job
->crop
[i
] );
126 job
->crop
[i
] = MAX( 0, job
->crop
[i
] );
130 job
->crop
[i
] = MIN( job
->crop
[i
], ( title
->height
/ 2 ) - 2 );
135 job
->crop
[i
] = MIN( job
->crop
[i
], ( title
->width
/ 2 ) - 2 );
139 if( keep
== HB_KEEP_WIDTH
)
141 job
->height
= MULTIPLE_16(
142 (uint64_t) job
->width
* title
->width
* HB_ASPECT_BASE
*
143 ( title
->height
- job
->crop
[0] - job
->crop
[1] ) /
144 ( (uint64_t) title
->height
* title
->aspect
*
145 ( title
->width
- job
->crop
[2] - job
->crop
[3] ) ) );
146 job
->height
= MAX( 16, job
->height
);
150 job
->width
= MULTIPLE_16(
151 (uint64_t) job
->height
* title
->height
* title
->aspect
*
152 ( title
->width
- job
->crop
[2] - job
->crop
[3] ) /
153 ( (uint64_t) title
->width
* HB_ASPECT_BASE
*
154 ( title
->height
- job
->crop
[0] - job
->crop
[1] ) ) );
155 job
->width
= MAX( 16, job
->width
);
159 /**********************************************************************
161 **********************************************************************
163 *********************************************************************/
164 int hb_calc_bitrate( hb_job_t
* job
, int size
)
166 int64_t avail
= (int64_t) size
* 1024 * 1024;
169 int samples_per_frame
;
172 hb_title_t
* title
= job
->title
;
173 hb_chapter_t
* chapter
;
176 /* How many overhead bytes are used for each frame
196 /* How many audio samples we put in each frame */
197 switch( job
->acodec
)
200 case HB_ACODEC_VORBIS
:
201 samples_per_frame
= 1024;
204 samples_per_frame
= 1152;
207 samples_per_frame
= 1536;
213 /* Get the duration in seconds */
215 for( i
= job
->chapter_start
; i
<= job
->chapter_end
; i
++ )
217 chapter
= hb_list_item( title
->list_chapter
, i
- 1 );
218 length
+= chapter
->duration
;
224 avail
-= length
* job
->vrate
* overhead
/ job
->vrate_base
;
226 for( i
= 0; job
->audios
[i
] >= 0; i
++ )
230 if( job
->acodec
& HB_ACODEC_AC3
||
231 job
->audio_mixdowns
[i
] == HB_AMIXDOWN_AC3
)
234 * For AC-3 we take the bitrate from the input audio
235 * bitrate as we are simply passing it through.
237 audio
= hb_list_item( title
->list_audio
, job
->audios
[i
] );
238 abitrate
= audio
->bitrate
/ 8;
243 * Where we are transcoding the audio we use the destination
246 abitrate
= job
->abitrate
* 1000 / 8;
248 avail
-= length
* abitrate
;
251 avail
-= length
* job
->arate
* overhead
/ samples_per_frame
;
259 return ( avail
/ ( 125 * length
) );
262 /**********************************************************************
263 * hb_list implementation
264 **********************************************************************
265 * Basic and slow, but enough for what we need
266 *********************************************************************/
268 #define HB_LIST_DEFAULT_SIZE 20
272 /* Pointers to items in the list */
275 /* How many (void *) allocated in 'items' */
278 /* How many valid pointers in 'items' */
282 /**********************************************************************
284 **********************************************************************
285 * Allocates an empty list ready for HB_LIST_DEFAULT_SIZE items
286 *********************************************************************/
287 hb_list_t
* hb_list_init()
291 l
= calloc( sizeof( hb_list_t
), 1 );
292 l
->items
= calloc( HB_LIST_DEFAULT_SIZE
* sizeof( void * ), 1 );
293 l
->items_alloc
= HB_LIST_DEFAULT_SIZE
;
298 /**********************************************************************
300 **********************************************************************
301 * Returns the number of items currently in the list
302 *********************************************************************/
303 int hb_list_count( hb_list_t
* l
)
305 return l
->items_count
;
308 /**********************************************************************
310 **********************************************************************
311 * Adds an item at the end of the list, making it bigger if necessary.
312 * Can safely be called with a NULL pointer to add, it will be ignored.
313 *********************************************************************/
314 void hb_list_add( hb_list_t
* l
, void * p
)
321 if( l
->items_count
== l
->items_alloc
)
323 /* We need a bigger boat */
324 l
->items_alloc
+= HB_LIST_DEFAULT_SIZE
;
325 l
->items
= realloc( l
->items
,
326 l
->items_alloc
* sizeof( void * ) );
329 l
->items
[l
->items_count
] = p
;
333 /**********************************************************************
335 **********************************************************************
336 * Remove an item from the list. Bad things will happen if called
337 * with a NULL pointer or if the item is not in the list.
338 *********************************************************************/
339 void hb_list_rem( hb_list_t
* l
, void * p
)
343 /* Find the item in the list */
344 for( i
= 0; i
< l
->items_count
; i
++ )
346 if( l
->items
[i
] == p
)
352 /* Shift all items after it sizeof( void * ) bytes earlier */
353 memmove( &l
->items
[i
], &l
->items
[i
+1],
354 ( l
->items_count
- i
- 1 ) * sizeof( void * ) );
359 /**********************************************************************
361 **********************************************************************
362 * Returns item at position i, or NULL if there are not that many
364 *********************************************************************/
365 void * hb_list_item( hb_list_t
* l
, int i
)
367 if( i
< 0 || i
>= l
->items_count
)
375 /**********************************************************************
377 **********************************************************************
378 * Assuming all items are of type hb_buffer_t, returns the total
379 * number of bytes in the list
380 *********************************************************************/
381 int hb_list_bytes( hb_list_t
* l
)
388 for( i
= 0; i
< hb_list_count( l
); i
++ )
390 buf
= hb_list_item( l
, i
);
391 ret
+= buf
->size
- buf
->cur
;
397 /**********************************************************************
399 **********************************************************************
400 * Assuming all items are of type hb_buffer_t, copy <size> bytes from
401 * the list to <dst>, keeping the list unmodified.
402 *********************************************************************/
403 void hb_list_seebytes( hb_list_t
* l
, uint8_t * dst
, int size
)
410 for( i
= 0, copied
= 0; copied
< size
; i
++ )
412 buf
= hb_list_item( l
, i
);
413 copying
= MIN( buf
->size
- buf
->cur
, size
- copied
);
414 memcpy( &dst
[copied
], &buf
->data
[buf
->cur
], copying
);
419 /**********************************************************************
421 **********************************************************************
422 * Assuming all items are of type hb_buffer_t, copy <size> bytes from
423 * the list to <dst>. What's copied is removed from the list.
424 * The variable pointed by <pts> is set to the PTS of the buffer the
425 * first byte has been got from.
426 * The variable pointed by <pos> is set to the position of that byte
428 *********************************************************************/
429 void hb_list_getbytes( hb_list_t
* l
, uint8_t * dst
, int size
,
430 uint64_t * pts
, uint64_t * pos
)
437 /* So we won't have to deal with NULL pointers */
438 uint64_t dummy1
, dummy2
;
440 if( !pts
) pts
= &dummy1
;
441 if( !pos
) pos
= &dummy2
;
443 for( copied
= 0, has_pts
= 0; copied
< size
; )
445 buf
= hb_list_item( l
, 0 );
446 copying
= MIN( buf
->size
- buf
->cur
, size
- copied
);
447 memcpy( &dst
[copied
], &buf
->data
[buf
->cur
], copying
);
457 if( buf
->cur
>= buf
->size
)
459 hb_list_rem( l
, buf
);
460 hb_buffer_close( &buf
);
467 /**********************************************************************
469 **********************************************************************
470 * Assuming all items are of type hb_buffer_t, close them all and
472 *********************************************************************/
473 void hb_list_empty( hb_list_t
** _l
)
478 while( ( b
= hb_list_item( l
, 0 ) ) )
481 hb_buffer_close( &b
);
487 /**********************************************************************
489 **********************************************************************
490 * Free memory allocated by hb_list_init. Does NOT free contents of
491 * items still in the list.
492 *********************************************************************/
493 void hb_list_close( hb_list_t
** _l
)
503 /**********************************************************************
505 **********************************************************************
506 * If verbose mode is one, print message with timestamp. Messages
507 * longer than 180 characters are stripped ;p
508 *********************************************************************/
509 void hb_log( char * log
, ... )
511 char string
[362]; /* 360 chars + \n + \0 */
516 if( !getenv( "HB_DEBUG" ) )
518 /* We don't want to print it */
524 now
= localtime( &_now
);
525 sprintf( string
, "[%02d:%02d:%02d] ",
526 now
->tm_hour
, now
->tm_min
, now
->tm_sec
);
528 /* Convert the message to a string */
529 va_start( args
, log
);
530 vsnprintf( string
+ 11, 169, log
, args
);
533 /* Add the end of line */
534 strcat( string
, "\n" );
537 fprintf( stderr
, "%s", string
);
540 /**********************************************************************
542 **********************************************************************
543 * Using whatever output is available display this error.
544 *********************************************************************/
545 void hb_error( char * log
, ... )
547 char string
[181]; /* 180 chars + \0 */
550 /* Convert the message to a string */
551 va_start( args
, log
);
552 vsnprintf( string
, 180, log
, args
);
556 * Got the error in a single string, send it off to be dispatched.
560 error_handler( string
);
566 void hb_register_error_handler( hb_error_handler_t
* handler
)
568 error_handler
= handler
;
571 /**********************************************************************
573 **********************************************************************
575 *********************************************************************/
576 hb_title_t
* hb_title_init( char * dvd
, int index
)
580 t
= calloc( sizeof( hb_title_t
), 1 );
583 t
->list_audio
= hb_list_init();
584 t
->list_chapter
= hb_list_init();
585 t
->list_subtitle
= hb_list_init();
586 strcat( t
->dvd
, dvd
);
591 /**********************************************************************
593 **********************************************************************
595 *********************************************************************/
596 void hb_title_close( hb_title_t
** _t
)
598 hb_title_t
* t
= *_t
;
600 hb_chapter_t
* chapter
;
601 hb_subtitle_t
* subtitle
;
603 while( ( audio
= hb_list_item( t
->list_audio
, 0 ) ) )
605 hb_list_rem( t
->list_audio
, audio
);
608 hb_list_close( &t
->list_audio
);
610 while( ( chapter
= hb_list_item( t
->list_chapter
, 0 ) ) )
612 hb_list_rem( t
->list_chapter
, chapter
);
615 hb_list_close( &t
->list_chapter
);
617 while( ( subtitle
= hb_list_item( t
->list_subtitle
, 0 ) ) )
619 hb_list_rem( t
->list_subtitle
, subtitle
);
622 hb_list_close( &t
->list_subtitle
);
628 /**********************************************************************
630 **********************************************************************
632 *********************************************************************/
633 void hb_filter_close( hb_filter_object_t
** _f
)
635 hb_filter_object_t
* f
= *_f
;
637 f
->close( f
->private_data
);