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