1 /* $Id: internal.h,v 1.41 2005/11/25 15:05:25 titer Exp $
3 This file is part of the HandBrake source code.
4 Homepage: <http://handbrake.fr/>.
5 It may be used under the terms of the GNU General Public License. */
7 /***********************************************************************
9 **********************************************************************/
10 void hb_log( char * log, ... ) HB_WPRINTF(1,2);
11 extern int global_verbosity_level; // Global variable for hb_deep_log
12 typedef enum hb_debug_level_s
14 HB_SUPPORT_LOG = 1, // helpful in tech support
15 HB_HOUSEKEEPING_LOG = 2, // stuff we hate scrolling through
16 HB_GRANULAR_LOG = 3 // sample-by-sample
18 void hb_deep_log( hb_debug_level_t level, char * log, ... ) HB_WPRINTF(2,3);
19 void hb_error( char * fmt, ...) HB_WPRINTF(1,2);
21 int hb_list_bytes( hb_list_t * );
22 void hb_list_seebytes( hb_list_t * l, uint8_t * dst, int size );
23 void hb_list_getbytes( hb_list_t * l, uint8_t * dst, int size,
24 uint64_t * pts, uint64_t * pos );
25 void hb_list_empty( hb_list_t ** );
27 hb_title_t * hb_title_init( char * dvd, int index );
28 void hb_title_close( hb_title_t ** );
30 void hb_filter_close( hb_filter_object_t ** );
32 /***********************************************************************
34 **********************************************************************/
35 int hb_get_pid( hb_handle_t * );
36 void hb_set_state( hb_handle_t *, hb_state_t * );
38 /***********************************************************************
40 **********************************************************************/
55 #define HB_FRAME_IDR 0x01
56 #define HB_FRAME_I 0x02
57 #define HB_FRAME_AUDIO 0x04
58 #define HB_FRAME_P 0x10
59 #define HB_FRAME_B 0x20
60 #define HB_FRAME_BREF 0x40
61 #define HB_FRAME_KEY 0x0F
62 #define HB_FRAME_REF 0xF0
66 /* Holds the output PTS from x264, for use by b-frame offsets in muxmp4.c */
79 void hb_buffer_pool_init( void );
80 void hb_buffer_pool_free( void );
82 hb_buffer_t * hb_buffer_init( int size );
83 void hb_buffer_realloc( hb_buffer_t *, int size );
84 void hb_buffer_close( hb_buffer_t ** );
85 void hb_buffer_copy_settings( hb_buffer_t * dst,
86 const hb_buffer_t * src );
88 hb_fifo_t * hb_fifo_init();
89 int hb_fifo_size( hb_fifo_t * );
90 int hb_fifo_is_full( hb_fifo_t * );
91 float hb_fifo_percent_full( hb_fifo_t * f );
92 hb_buffer_t * hb_fifo_get( hb_fifo_t * );
93 hb_buffer_t * hb_fifo_see( hb_fifo_t * );
94 hb_buffer_t * hb_fifo_see2( hb_fifo_t * );
95 void hb_fifo_push( hb_fifo_t *, hb_buffer_t * );
96 void hb_fifo_push_head( hb_fifo_t *, hb_buffer_t * );
97 void hb_fifo_close( hb_fifo_t ** );
99 // this routine gets a buffer for an uncompressed YUV420 video frame
100 // with dimensions width x height.
101 static inline hb_buffer_t * hb_video_buffer_init( int width, int height )
103 // Y requires w x h bytes. U & V each require (w+1)/2 x
104 // (h+1)/2 bytes (the "+1" is to round up). We shift rather
105 // than divide by 2 since the compiler can't know these ints
106 // are positive so it generates very expensive integer divides
107 // if we do "/2". The code here matches the calculation for
108 // PIX_FMT_YUV420P in ffmpeg's avpicture_fill() which is required
109 // for most of HB's filters to work right.
110 return hb_buffer_init( width * height + ( ( width+1 ) >> 1 ) *
111 ( ( height+1 ) >> 1 ) * 2 );
114 // this routine 'moves' data from src to dst by interchanging 'data',
115 // 'size' & 'alloc' between them and copying the rest of the fields
117 static inline void hb_buffer_swap_copy( hb_buffer_t *src, hb_buffer_t *dst )
119 uint8_t *data = dst->data;
120 int size = dst->size;
121 int alloc = dst->alloc;
130 /***********************************************************************
131 * Threads: update.c, scan.c, work.c, reader.c, muxcommon.c
132 **********************************************************************/
133 hb_thread_t * hb_update_init( int * build, char * version );
134 hb_thread_t * hb_scan_init( hb_handle_t *, const char * path,
135 int title_index, hb_list_t * list_title,
136 int preview_count, int store_previews );
137 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
138 volatile int * die, int * error, hb_job_t ** job );
139 hb_thread_t * hb_reader_init( hb_job_t * );
140 hb_thread_t * hb_muxer_init( hb_job_t * );
141 hb_work_object_t * hb_get_work( int );
142 hb_work_object_t * hb_codec_decoder( int );
143 hb_work_object_t * hb_codec_encoder( int );
145 /***********************************************************************
147 **********************************************************************/
149 int64_t last_scr; /* unadjusted SCR from most recent pack */
150 int64_t last_pts; /* last pts we saw */
151 int scr_changes; /* number of SCR discontinuities */
152 int dts_drops; /* number of drops because DTS too far from SCR */
155 typedef int (*hb_muxer_t)(hb_buffer_t *, hb_list_t *, hb_psdemux_t*);
157 int hb_demux_ps( hb_buffer_t * ps_buf, hb_list_t * es_list, hb_psdemux_t * );
158 int hb_demux_ss( hb_buffer_t * ps_buf, hb_list_t * es_list, hb_psdemux_t * );
159 int hb_demux_null( hb_buffer_t * ps_buf, hb_list_t * es_list, hb_psdemux_t * );
161 extern const hb_muxer_t hb_demux[];
163 /***********************************************************************
165 **********************************************************************/
166 extern void decmetadata( hb_title_t *title );
168 /***********************************************************************
170 **********************************************************************/
171 typedef union hb_dvd_s hb_dvd_t;
172 typedef struct hb_stream_s hb_stream_t;
174 hb_dvd_t * hb_dvd_init( char * path );
175 int hb_dvd_title_count( hb_dvd_t * );
176 hb_title_t * hb_dvd_title_scan( hb_dvd_t *, int title );
177 int hb_dvd_start( hb_dvd_t *, hb_title_t *title, int chapter );
178 void hb_dvd_stop( hb_dvd_t * );
179 int hb_dvd_seek( hb_dvd_t *, float );
180 int hb_dvd_read( hb_dvd_t *, hb_buffer_t * );
181 int hb_dvd_chapter( hb_dvd_t * );
182 int hb_dvd_is_break( hb_dvd_t * d );
183 void hb_dvd_close( hb_dvd_t ** );
184 int hb_dvd_angle_count( hb_dvd_t * d );
185 void hb_dvd_set_angle( hb_dvd_t * d, int angle );
187 hb_stream_t * hb_stream_open( char * path, hb_title_t *title );
188 void hb_stream_close( hb_stream_t ** );
189 hb_title_t * hb_stream_title_scan( hb_stream_t *);
190 int hb_stream_read( hb_stream_t *, hb_buffer_t *);
191 int hb_stream_seek( hb_stream_t *, float );
192 int hb_stream_seek_chapter( hb_stream_t *, int );
193 int hb_stream_chapter( hb_stream_t * );
196 void * hb_ffmpeg_context( int codec_param );
197 void * hb_ffmpeg_avstream( int codec_param );
199 /***********************************************************************
201 **********************************************************************/
202 #define HB_CONFIG_MAX_SIZE 8192
208 uint8_t bytes[HB_CONFIG_MAX_SIZE];
214 uint8_t sps[HB_CONFIG_MAX_SIZE];
216 uint8_t pps[HB_CONFIG_MAX_SIZE];
223 uint8_t headers[3][HB_CONFIG_MAX_SIZE];
228 uint8_t bytes[HB_CONFIG_MAX_SIZE];
234 uint8_t headers[3][HB_CONFIG_MAX_SIZE];
240 /* ac3flags stores the flags from the AC3 source, as found in scan.c */
242 // next two items are used by the bsinfo routine to accumulate small
243 // frames until we have enough to validate the crc.
244 int len; // space currently used in 'buf'
245 uint8_t buf[HB_CONFIG_MAX_SIZE-sizeof(int)];
250 /* dcaflags stores the flags from the DCA source, as found in scan.c */
283 FILTER_DEINTERLACE = 1,
290 extern hb_work_object_t * hb_objects;
292 #define HB_WORK_IDLE 0
294 #define HB_WORK_ERROR 2
295 #define HB_WORK_DONE 3
297 /***********************************************************************
299 **********************************************************************/
300 typedef struct hb_mux_object_s hb_mux_object_t;
301 typedef struct hb_mux_data_s hb_mux_data_t;
303 #define HB_MUX_COMMON \
304 int (*init) ( hb_mux_object_t * ); \
305 int (*mux) ( hb_mux_object_t *, hb_mux_data_t *, \
307 int (*end) ( hb_mux_object_t * );
309 #define DECLARE_MUX( a ) \
310 hb_mux_object_t * hb_mux_##a##_init( hb_job_t * );