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, ... );
  11 void hb_error( char * fmt, ...);
  12
  13 int  hb_list_bytes( hb_list_t * );
  14 void hb_list_seebytes( hb_list_t * l, uint8_t * dst, int size );
  15 void hb_list_getbytes( hb_list_t * l, uint8_t * dst, int size,
  16                        uint64_t * pts, uint64_t * pos );
  17 void hb_list_empty( hb_list_t ** );
  18
  19 hb_title_t * hb_title_init( char * dvd, int index );
  20 void         hb_title_close( hb_title_t ** );
  21
  22 void         hb_filter_close( hb_filter_object_t ** );
  23
  24 /***********************************************************************
  25  * hb.c
  26  **********************************************************************/
  27 int  hb_get_pid( hb_handle_t * );
  28 void hb_set_state( hb_handle_t *, hb_state_t * );
  29
  30 /***********************************************************************
  31  * fifo.c
  32  **********************************************************************/
  33 struct hb_buffer_s
  34 {
  35     int           size;
  36     int           alloc;
  37     uint8_t *     data;
  38     int           cur;
  39
  40     int64_t       sequence;
  41
  42     int           id;
  43     int64_t       start;
  44     int64_t       stop;
  45     int           new_chap;
  46
  47 #define HB_FRAME_IDR    0x01
  48 #define HB_FRAME_I      0x02
  49 #define HB_FRAME_AUDIO  0x04
  50 #define HB_FRAME_P      0x10
  51 #define HB_FRAME_B      0x20
  52 #define HB_FRAME_BREF   0x40
  53 #define HB_FRAME_KEY    0x0F
  54 #define HB_FRAME_REF    0xF0
  55     uint8_t       frametype;
  56     uint16_t       flags;
  57
  58     /* Holds the output PTS from x264, for use by b-frame offsets in muxmp4.c */
  59     int64_t     renderOffset;
  60
  61     int           x;
  62     int           y;
  63     int           width;
  64     int           height;
  65
  66     hb_buffer_t * sub;
  67
  68     hb_buffer_t * next;
  69 };
  70
  71 void hb_buffer_pool_init( void );
  72 void hb_buffer_pool_free( void );
  73
  74 hb_buffer_t * hb_buffer_init( int size );
  75 void          hb_buffer_realloc( hb_buffer_t *, int size );
  76 void          hb_buffer_close( hb_buffer_t ** );
  77 void          hb_buffer_copy_settings( hb_buffer_t * dst,
  78                                        const hb_buffer_t * src );
  79
  80 hb_fifo_t   * hb_fifo_init();
  81 int           hb_fifo_size( hb_fifo_t * );
  82 int           hb_fifo_is_full( hb_fifo_t * );
  83 float         hb_fifo_percent_full( hb_fifo_t * f );
  84 hb_buffer_t * hb_fifo_get( hb_fifo_t * );
  85 hb_buffer_t * hb_fifo_see( hb_fifo_t * );
  86 hb_buffer_t * hb_fifo_see2( hb_fifo_t * );
  87 void          hb_fifo_push( hb_fifo_t *, hb_buffer_t * );
  88 void          hb_fifo_push_head( hb_fifo_t *, hb_buffer_t * );
  89 void          hb_fifo_close( hb_fifo_t ** );
  90
  91 /***********************************************************************
  92  * Threads: update.c, scan.c, work.c, reader.c, muxcommon.c
  93  **********************************************************************/
  94 hb_thread_t * hb_update_init( int * build, char * version );
  95 hb_thread_t * hb_scan_init( hb_handle_t *, const char * path,
  96                             int title_index, hb_list_t * list_title );
  97 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
  98                             volatile int * die, int * error, hb_job_t ** job );
  99 hb_thread_t  * hb_reader_init( hb_job_t * );
 100 hb_thread_t  * hb_muxer_init( hb_job_t * );
 101 hb_work_object_t * hb_get_work( int );
 102 hb_work_object_t * hb_codec_decoder( int );
 103 hb_work_object_t * hb_codec_encoder( int );
 104
 105 /***********************************************************************
 106  * mpegdemux.c
 107  **********************************************************************/
 108 typedef struct {
 109     int64_t last_scr;       /* unadjusted SCR from most recent pack */
 110     int     scr_changes;    /* number of SCR discontinuities */
 111     int     flaky_clock;    /* try to compensate for PCR drops */
 112     int     dts_drops;      /* number of drops because DTS too far from SCR */
 113 } hb_psdemux_t;
 114
 115 int hb_demux_ps( hb_buffer_t * ps_buf, hb_list_t * es_list, hb_psdemux_t * );
 116 int hb_demux_null( hb_buffer_t * ps_buf, hb_list_t * es_list, hb_psdemux_t * );
 117
 118 /***********************************************************************
 119  * dvd.c
 120  **********************************************************************/
 121 typedef struct hb_dvd_s hb_dvd_t;
 122 typedef struct hb_stream_s hb_stream_t;
 123
 124 hb_dvd_t *   hb_dvd_init( char * path );
 125 int          hb_dvd_title_count( hb_dvd_t * );
 126 hb_title_t * hb_dvd_title_scan( hb_dvd_t *, int title );
 127 int          hb_dvd_start( hb_dvd_t *, int title, int chapter );
 128 void         hb_dvd_stop( hb_dvd_t * );
 129 int          hb_dvd_seek( hb_dvd_t *, float );
 130 int          hb_dvd_read( hb_dvd_t *, hb_buffer_t * );
 131 int          hb_dvd_chapter( hb_dvd_t * );
 132 int          hb_dvd_is_break( hb_dvd_t * d );
 133 void         hb_dvd_close( hb_dvd_t ** );
 134
 135 hb_stream_t * hb_stream_open( char * path, hb_title_t *title );
 136 void             hb_stream_close( hb_stream_t ** );
 137 hb_title_t * hb_stream_title_scan( hb_stream_t *);
 138 int          hb_stream_read( hb_stream_t *, hb_buffer_t *);
 139 int          hb_stream_seek( hb_stream_t *, float );
 140
 141 void       * hb_ffmpeg_context( int codec_param );
 142 void       * hb_ffmpeg_avstream( int codec_param );
 143
 144 /***********************************************************************
 145  * Work objects
 146  **********************************************************************/
 147 #define HB_CONFIG_MAX_SIZE 8192
 148 union hb_esconfig_u
 149 {
 150
 151     struct
 152     {
 153         uint8_t bytes[HB_CONFIG_MAX_SIZE];
 154         int     length;
 155     } mpeg4;
 156
 157         struct
 158         {
 159             uint8_t  sps[HB_CONFIG_MAX_SIZE];
 160             int       sps_length;
 161             uint8_t  pps[HB_CONFIG_MAX_SIZE];
 162             int       pps_length;
 163         uint32_t init_delay;
 164         } h264;
 165
 166     struct
 167     {
 168         uint8_t headers[3][HB_CONFIG_MAX_SIZE];
 169     } theora;
 170
 171     struct
 172     {
 173         uint8_t bytes[HB_CONFIG_MAX_SIZE];
 174         int     length;
 175     } aac;
 176
 177     struct
 178     {
 179         uint8_t headers[3][HB_CONFIG_MAX_SIZE];
 180         char *language;
 181     } vorbis;
 182
 183     struct
 184     {
 185         /* ac3flags stores the flags from the AC3 source, as found in scan.c */
 186         int  ac3flags;
 187     } a52;
 188
 189     struct
 190     {
 191         /* dcaflags stores the flags from the DCA source, as found in scan.c */
 192         int  dcaflags;
 193     } dca;
 194
 195 };
 196
 197 enum
 198 {
 199     WORK_SYNC = 1,
 200     WORK_DECMPEG2,
 201     WORK_DECSUB,
 202     WORK_RENDER,
 203     WORK_ENCAVCODEC,
 204     WORK_ENCXVID,
 205     WORK_ENCX264,
 206     WORK_ENCTHEORA,
 207     WORK_DECA52,
 208     WORK_DECDCA,
 209     WORK_DECAVCODEC,
 210     WORK_DECAVCODECV,
 211     WORK_DECAVCODECVI,
 212     WORK_DECAVCODECAI,
 213     WORK_DECLPCM,
 214     WORK_ENCFAAC,
 215     WORK_ENCLAME,
 216     WORK_ENCVORBIS
 217 };
 218
 219 enum
 220 {
 221     FILTER_DEINTERLACE = 1,
 222     FILTER_DEBLOCK,
 223     FILTER_DENOISE,
 224     FILTER_DETELECINE,
 225     FILTER_DECOMB
 226 };
 227
 228 extern hb_work_object_t * hb_objects;
 229
 230 #define HB_WORK_IDLE     0
 231 #define HB_WORK_OK       1
 232 #define HB_WORK_ERROR    2
 233 #define HB_WORK_DONE     3
 234
 235 /***********************************************************************
 236  * Muxers
 237  **********************************************************************/
 238 typedef struct hb_mux_object_s hb_mux_object_t;
 239 typedef struct hb_mux_data_s   hb_mux_data_t;
 240
 241 #define HB_MUX_COMMON \
 242     int (*init)      ( hb_mux_object_t * ); \
 243     int (*mux)       ( hb_mux_object_t *, hb_mux_data_t *, \
 244                        hb_buffer_t * ); \
 245     int (*end)       ( hb_mux_object_t * );
 246
 247 #define DECLARE_MUX( a ) \
 248     hb_mux_object_t  * hb_mux_##a##_init( hb_job_t * );
 249
 250 DECLARE_MUX( mp4 );
 251 DECLARE_MUX( avi );
 252 DECLARE_MUX( ogm );
 253 DECLARE_MUX( mkv );
 254