libhb/internal.h

   1 /* $Id: internal.h,v 1.41 2005/11/25 15:05:25 titer Exp $
   2
   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. */
   6
   7 /***********************************************************************
   8  * common.c
   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
  13 {
  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
  17 } hb_debug_level_t;
  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);
  20
  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 ** );
  26
  27 hb_title_t * hb_title_init( char * dvd, int index );
  28 void         hb_title_close( hb_title_t ** );
  29
  30 void         hb_filter_close( hb_filter_object_t ** );
  31
  32 /***********************************************************************
  33  * hb.c
  34  **********************************************************************/
  35 int  hb_get_pid( hb_handle_t * );
  36 void hb_set_state( hb_handle_t *, hb_state_t * );
  37
  38 /***********************************************************************
  39  * fifo.c
  40  **********************************************************************/
  41 struct hb_buffer_s
  42 {
  43     int           size;
  44     int           alloc;
  45     uint8_t *     data;
  46     int           cur;
  47
  48     int64_t       sequence;
  49
  50     int           id;
  51     int64_t       start;
  52     int64_t       stop;
  53     int           new_chap;
  54
  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
  63     uint8_t       frametype;
  64     uint16_t       flags;
  65
  66     /* Holds the output PTS from x264, for use by b-frame offsets in muxmp4.c */
  67     int64_t     renderOffset;
  68
  69     int           x;
  70     int           y;
  71     int           width;
  72     int           height;
  73
  74     hb_buffer_t * sub;
  75
  76     hb_buffer_t * next;
  77 };
  78
  79 void hb_buffer_pool_init( void );
  80 void hb_buffer_pool_free( void );
  81
  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 );
  87
  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 ** );
  98
  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 )
 102 {
 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 );
 112 }
 113
 114 // this routine 'moves' data from src to dst by interchanging 'data',
 115 // 'size' & 'alloc' between them and copying the rest of the fields
 116 // from src to dst.
 117 static inline void hb_buffer_swap_copy( hb_buffer_t *src, hb_buffer_t *dst )
 118 {
 119     uint8_t *data  = dst->data;
 120     int      size  = dst->size;
 121     int      alloc = dst->alloc;
 122
 123     *dst = *src;
 124
 125     src->data  = data;
 126     src->size  = size;
 127     src->alloc = alloc;
 128 }
 129
 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 );
 144
 145 /***********************************************************************
 146  * mpegdemux.c
 147  **********************************************************************/
 148 typedef struct {
 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 */
 153 } hb_psdemux_t;
 154
 155 typedef int (*hb_muxer_t)(hb_buffer_t *, hb_list_t *, hb_psdemux_t*);
 156
 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 * );
 160
 161 extern const hb_muxer_t hb_demux[];
 162
 163 /***********************************************************************
 164  * decmetadata.c
 165  **********************************************************************/
 166 extern void decmetadata( hb_title_t *title );
 167
 168 /***********************************************************************
 169  * dvd.c
 170  **********************************************************************/
 171 typedef union  hb_dvd_s hb_dvd_t;
 172 typedef struct hb_stream_s hb_stream_t;
 173
 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 );
 186
 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 * );
 194
 195
 196 void       * hb_ffmpeg_context( int codec_param );
 197 void       * hb_ffmpeg_avstream( int codec_param );
 198
 199 /***********************************************************************
 200  * Work objects
 201  **********************************************************************/
 202 #define HB_CONFIG_MAX_SIZE 8192
 203 union hb_esconfig_u
 204 {
 205
 206     struct
 207     {
 208         uint8_t bytes[HB_CONFIG_MAX_SIZE];
 209         int     length;
 210     } mpeg4;
 211
 212         struct
 213         {
 214             uint8_t  sps[HB_CONFIG_MAX_SIZE];
 215             int       sps_length;
 216             uint8_t  pps[HB_CONFIG_MAX_SIZE];
 217             int       pps_length;
 218         uint32_t init_delay;
 219         } h264;
 220
 221     struct
 222     {
 223         uint8_t headers[3][HB_CONFIG_MAX_SIZE];
 224     } theora;
 225
 226     struct
 227     {
 228         uint8_t bytes[HB_CONFIG_MAX_SIZE];
 229         int     length;
 230     } aac;
 231
 232     struct
 233     {
 234         uint8_t headers[3][HB_CONFIG_MAX_SIZE];
 235         char *language;
 236     } vorbis;
 237
 238     struct
 239     {
 240         /* ac3flags stores the flags from the AC3 source, as found in scan.c */
 241         int     ac3flags;
 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)];
 246     } a52;
 247
 248     struct
 249     {
 250         /* dcaflags stores the flags from the DCA source, as found in scan.c */
 251         int  dcaflags;
 252     } dca;
 253
 254 };
 255
 256 enum
 257 {
 258     WORK_SYNC = 1,
 259     WORK_DECMPEG2,
 260     WORK_DECCC608,
 261     WORK_DECVOBSUB,
 262     WORK_DECSRTSUB,
 263     WORK_ENCVOBSUB,
 264     WORK_RENDER,
 265     WORK_ENCAVCODEC,
 266     WORK_ENCX264,
 267     WORK_ENCTHEORA,
 268     WORK_DECA52,
 269     WORK_DECDCA,
 270     WORK_DECAVCODEC,
 271     WORK_DECAVCODECV,
 272     WORK_DECAVCODECVI,
 273     WORK_DECAVCODECAI,
 274     WORK_DECLPCM,
 275     WORK_ENCFAAC,
 276     WORK_ENCLAME,
 277     WORK_ENCVORBIS,
 278     WORK_ENC_CA_AAC
 279 };
 280
 281 enum
 282 {
 283     FILTER_DEINTERLACE = 1,
 284     FILTER_DEBLOCK,
 285     FILTER_DENOISE,
 286     FILTER_DETELECINE,
 287     FILTER_DECOMB
 288 };
 289
 290 extern hb_work_object_t * hb_objects;
 291
 292 #define HB_WORK_IDLE     0
 293 #define HB_WORK_OK       1
 294 #define HB_WORK_ERROR    2
 295 #define HB_WORK_DONE     3
 296
 297 /***********************************************************************
 298  * Muxers
 299  **********************************************************************/
 300 typedef struct hb_mux_object_s hb_mux_object_t;
 301 typedef struct hb_mux_data_s   hb_mux_data_t;
 302
 303 #define HB_MUX_COMMON \
 304     int (*init)      ( hb_mux_object_t * ); \
 305     int (*mux)       ( hb_mux_object_t *, hb_mux_data_t *, \
 306                        hb_buffer_t * ); \
 307     int (*end)       ( hb_mux_object_t * );
 308
 309 #define DECLARE_MUX( a ) \
 310     hb_mux_object_t  * hb_mux_##a##_init( hb_job_t * );
 311
 312 DECLARE_MUX( mp4 );
 313 DECLARE_MUX( avi );
 314 DECLARE_MUX( ogm );
 315 DECLARE_MUX( mkv );
 316