libhb/common.h

   1 /* $Id: common.h,v 1.51 2005/11/04 13:09:40 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 #ifndef HB_COMMON_H
   8 #define HB_COMMON_H
   9
  10 #include <math.h>
  11 #include <stdio.h>
  12 #include <stdlib.h>
  13 #include <stdarg.h>
  14 #include <string.h>
  15 #include <unistd.h>
  16 #include <inttypes.h>
  17 #include <sys/types.h>
  18 #include <sys/stat.h>
  19 #include <dirent.h>
  20
  21 #if defined( __GNUC__ ) && !(defined( _WIN32 ) || defined( __MINGW32__ ))
  22 #   define HB_WPRINTF(s,v) __attribute__((format(printf,s,v)))
  23 #else
  24 #   define HB_WPRINTF(s,v)
  25 #endif
  26
  27 #if defined( SYS_MINGW )
  28 #   define fseek fseeko64
  29 #   define ftell ftello64
  30 #   undef  fseeko
  31 #   define fseeko fseeko64
  32 #   undef  ftello
  33 #   define ftello ftello64
  34 #   define flockfile(...)
  35 #   define funlockfile(...)
  36 #   define getc_unlocked getc
  37 #   undef  off_t
  38 #   define off_t off64_t
  39 #endif
  40
  41 #ifndef MIN
  42 #define MIN( a, b ) ( (a) > (b) ? (b) : (a) )
  43 #endif
  44 #ifndef MAX
  45 #define MAX( a, b ) ( (a) > (b) ? (a) : (b) )
  46 #endif
  47
  48 #define EVEN( a )        ( (a) + ( (a) & 1 ) )
  49 #define MULTIPLE_16( a ) ( 16 * ( ( (a) + 8 ) / 16 ) )
  50 #define MULTIPLE_MOD( a, b ) ((b==1)?a:( b * ( ( (a) + (b / 2) - 1) / b ) ))
  51
  52 #define HB_DVD_READ_BUFFER_SIZE 2048
  53
  54 typedef struct hb_handle_s hb_handle_t;
  55 typedef struct hb_list_s hb_list_t;
  56 typedef struct hb_rate_s hb_rate_t;
  57 typedef struct hb_mixdown_s hb_mixdown_t;
  58 typedef struct hb_job_s  hb_job_t;
  59 typedef struct hb_title_s hb_title_t;
  60 typedef struct hb_chapter_s hb_chapter_t;
  61 typedef struct hb_audio_s hb_audio_t;
  62 typedef struct hb_audio_config_s hb_audio_config_t;
  63 typedef struct hb_subtitle_s hb_subtitle_t;
  64 typedef struct hb_subtitle_config_s hb_subtitle_config_t;
  65 typedef struct hb_metadata_s hb_metadata_t;
  66 typedef struct hb_state_s hb_state_t;
  67 typedef union  hb_esconfig_u     hb_esconfig_t;
  68 typedef struct hb_work_private_s hb_work_private_t;
  69 typedef struct hb_work_object_s  hb_work_object_t;
  70 typedef struct hb_filter_private_s hb_filter_private_t;
  71 typedef struct hb_filter_object_s  hb_filter_object_t;
  72 typedef struct hb_buffer_s hb_buffer_t;
  73 typedef struct hb_fifo_s hb_fifo_t;
  74 typedef struct hb_lock_s hb_lock_t;
  75
  76 #include "ports.h"
  77 #ifdef __LIBHB__
  78 #include "internal.h"
  79 #define PRIVATE
  80 #else
  81 #define PRIVATE const
  82 #endif
  83
  84 hb_list_t * hb_list_init();
  85 int         hb_list_count( hb_list_t * );
  86 void        hb_list_add( hb_list_t *, void * );
  87 void        hb_list_rem( hb_list_t *, void * );
  88 void      * hb_list_item( hb_list_t *, int );
  89 void        hb_list_close( hb_list_t ** );
  90
  91 void hb_reduce( int *x, int *y, int num, int den );
  92
  93 #define HB_KEEP_WIDTH  0
  94 #define HB_KEEP_HEIGHT 1
  95 void hb_fix_aspect( hb_job_t * job, int keep );
  96
  97 int hb_calc_bitrate( hb_job_t *, int size );
  98
  99 hb_audio_t *hb_audio_copy(const hb_audio_t *src);
 100 void hb_audio_config_init(hb_audio_config_t * audiocfg);
 101 int hb_audio_add(const hb_job_t * job, const hb_audio_config_t * audiocfg);
 102 hb_audio_config_t * hb_list_audio_config_item(hb_list_t * list, int i);
 103
 104 int hb_subtitle_add(const hb_job_t * job, const hb_subtitle_config_t * subtitlecfg, int track);
 105 int hb_srt_add(const hb_job_t * job, const hb_subtitle_config_t * subtitlecfg,
 106                const char *lang);
 107
 108
 109 struct hb_rate_s
 110 {
 111     char * string;
 112     int    rate;
 113 };
 114
 115 struct hb_mixdown_s
 116 {
 117     char * human_readable_name;
 118     char * internal_name;
 119     char * short_name;
 120     int    amixdown;
 121 };
 122
 123 struct hb_subtitle_config_s
 124 {
 125     enum subdest { RENDERSUB, PASSTHRUSUB } dest;
 126     int  force;
 127     int  default_track;
 128     char src_filename[128];
 129     char src_codeset[40];
 130     int64_t offset;
 131 };
 132
 133 #define HB_VIDEO_RATE_BASE   27000000
 134
 135 extern hb_rate_t    hb_video_rates[];
 136 extern int          hb_video_rates_count;
 137 extern hb_rate_t    hb_audio_rates[];
 138 extern int          hb_audio_rates_count;
 139 extern int          hb_audio_rates_default;
 140 extern hb_rate_t    hb_audio_bitrates[];
 141 extern int          hb_audio_bitrates_count;
 142 extern int          hb_audio_bitrates_default;
 143 extern hb_mixdown_t hb_audio_mixdowns[];
 144 extern int          hb_audio_mixdowns_count;
 145 int hb_mixdown_get_mixdown_from_short_name( const char * short_name );
 146 const char * hb_mixdown_get_short_name_from_mixdown( int amixdown );
 147
 148 /******************************************************************************
 149  * hb_job_t: settings to be filled by the UI
 150  *****************************************************************************/
 151 struct hb_job_s
 152 {
 153     /* ID assigned by UI so it can groups job passes together */
 154     int             sequence_id;
 155
 156     /* Pointer to the title to be ripped */
 157     hb_title_t    * title;
 158
 159     /* Chapter selection */
 160     int             chapter_start;
 161     int             chapter_end;
 162
 163         /* Include chapter marker track in mp4? */
 164     int             chapter_markers;
 165
 166     /* Picture settings:
 167          crop:                must be multiples of 2 (top/bottom/left/right)
 168          deinterlace:         0 or 1
 169          width:               must be a multiple of 2
 170          height:              must be a multiple of 2
 171          keep_ratio:          used by UIs
 172          grayscale:           black and white encoding
 173          pixel_ratio:         store pixel aspect ratio in the video
 174          pixel_aspect_width:  numerator for pixel aspect ratio
 175          pixel_aspect_height: denominator for pixel aspect ratio
 176          modulus:             set a number for dimensions to be multiples of
 177          maxWidth:            keep width below this
 178          maxHeight:           keep height below this */
 179     int             crop[4];
 180     int             deinterlace;
 181     hb_list_t     * filters;
 182     int             width;
 183     int             height;
 184     int             keep_ratio;
 185     int             grayscale;
 186
 187     struct
 188     {
 189         int             mode;
 190         int             itu_par;
 191         int             par_width;
 192         int             par_height;
 193         int             dar_width;
 194         int             dar_height;
 195         int             keep_display_aspect;
 196     } anamorphic;
 197
 198     int             modulus;
 199     int             maxWidth;
 200     int             maxHeight;
 201
 202     /* Video settings:
 203          vcodec:            output codec
 204          vquality:          output quality (0.0..1.0)
 205                             if < 0.0 or > 1.0, bitrate is used instead,
 206                             except with x264, to use its real QP/RF scale
 207          vbitrate:          output bitrate (kbps)
 208          vrate, vrate_base: output framerate is vrate / vrate_base
 209          cfr:               0 (vfr), 1 (cfr), 2 (pfr) [see render.c]
 210          pass:              0, 1 or 2 (or -1 for scan)
 211          h264_level:        vestigial boolean to decide if we're encoding for iPod
 212          x264opts:          string of extra x264 options
 213          areBframes:        boolean to note if b-frames are included in x264opts */
 214 #define HB_VCODEC_MASK   0x0000FF
 215 #define HB_VCODEC_FFMPEG 0x000001
 216 #define HB_VCODEC_X264   0x000002
 217 #define HB_VCODEC_THEORA 0x000004
 218
 219     int             vcodec;
 220     float           vquality;
 221     int             vbitrate;
 222     int             vrate;
 223     int             vrate_base;
 224     int             vfr;
 225     int             cfr;
 226     int             pass;
 227     int             h264_13;
 228     int             h264_level;
 229     char            *x264opts;
 230     int             areBframes;
 231     int             color_matrix;
 232
 233     /* List of audio settings. */
 234     hb_list_t     * list_audio;
 235
 236     /* Subtitles
 237      */
 238     hb_list_t     * list_subtitle;
 239
 240     /* Muxer settings
 241          mux:  output file format
 242          file: file path */
 243 #define HB_MUX_MASK 0xFF0000
 244 #define HB_MUX_MP4  0x010000
 245 #define HB_MUX_PSP  0x020000
 246 #define HB_MUX_AVI  0x040000
 247 #define HB_MUX_OGM  0x080000
 248 #define HB_MUX_IPOD 0x100000
 249 #define HB_MUX_MKV  0x200000
 250
 251     int             mux;
 252     const char          * file;
 253
 254     /* Allow MP4 files > 4 gigs */
 255     int             largeFileSize;
 256     int             mp4_optimize;
 257     int             ipod_atom;
 258
 259     int                     indepth_scan;
 260     hb_subtitle_config_t    select_subtitle_config;
 261
 262     int             angle;              // dvd angle to encode
 263     int             frame_to_start;     // declare eof when we hit this frame
 264     int64_t         pts_to_start;       // drop frames until  we pass this pts
 265                                         //  in the time-linearized input stream
 266     int             frame_to_stop;      // declare eof when we hit this frame
 267     int64_t         pts_to_stop;        // declare eof when we pass this pts in
 268                                         //  the time-linearized input stream
 269     int             start_at_preview;   // if non-zero, encoding will start
 270                                         //  at the position of preview n
 271     int             seek_points;        //  out of N previews
 272     uint32_t        frames_to_skip;     // decode but discard this many frames
 273                                         //  initially (for frame accurate positioning
 274                                         //  to non-I frames).
 275
 276 #ifdef __LIBHB__
 277     /* Internal data */
 278     hb_handle_t   * h;
 279     hb_lock_t     * pause;
 280     volatile int  * die;
 281     volatile int    done;
 282
 283     uint64_t        st_pause_date;
 284     uint64_t        st_paused;
 285
 286     hb_fifo_t     * fifo_mpeg2;   /* MPEG-2 video ES */
 287     hb_fifo_t     * fifo_raw;     /* Raw pictures */
 288     hb_fifo_t     * fifo_sync;    /* Raw pictures, framerate corrected */
 289     hb_fifo_t     * fifo_render;  /* Raw pictures, scaled */
 290     hb_fifo_t     * fifo_mpeg4;   /* MPEG-4 video ES */
 291
 292     hb_thread_t   * reader;
 293
 294     hb_list_t     * list_work;
 295
 296     hb_esconfig_t config;
 297
 298     hb_mux_data_t * mux_data;
 299 #endif
 300 };
 301
 302 /* Audio starts here */
 303 /* Audio Codecs */
 304 #define HB_ACODEC_MASK   0x00FF00
 305 #define HB_ACODEC_FAAC   0x000100
 306 #define HB_ACODEC_LAME   0x000200
 307 #define HB_ACODEC_VORBIS 0x000400
 308 #define HB_ACODEC_AC3    0x000800
 309 #define HB_ACODEC_MPGA   0x001000
 310 #define HB_ACODEC_LPCM   0x002000
 311 #define HB_ACODEC_DCA    0x004000
 312 #define HB_ACODEC_FFMPEG 0x008000
 313 #define HB_ACODEC_CA_AAC 0x010000
 314
 315 /* Audio Mixdown */
 316 /* define some masks, used to extract the various information from the HB_AMIXDOWN_XXXX values */
 317 #define HB_AMIXDOWN_DCA_FORMAT_MASK             0x00FFF000
 318 #define HB_AMIXDOWN_A52_FORMAT_MASK             0x00000FF0
 319 #define HB_AMIXDOWN_DISCRETE_CHANNEL_COUNT_MASK 0x0000000F
 320 /* define the HB_AMIXDOWN_XXXX values */
 321 #define HB_AMIXDOWN_MONO                        0x01000001
 322 // DCA_FORMAT of DCA_MONO                  = 0    = 0x000
 323 // A52_FORMAT of A52_MONO                  = 1    = 0x01
 324 // discrete channel count of 1
 325 #define HB_AMIXDOWN_STEREO                      0x02002022
 326 // DCA_FORMAT of DCA_STEREO                = 2    = 0x002
 327 // A52_FORMAT of A52_STEREO                = 2    = 0x02
 328 // discrete channel count of 2
 329 #define HB_AMIXDOWN_DOLBY                       0x042070A2
 330 // DCA_FORMAT of DCA_3F1R | DCA_OUT_DPLI   = 519  = 0x207
 331 // A52_FORMAT of A52_DOLBY                 = 10   = 0x0A
 332 // discrete channel count of 2
 333 #define HB_AMIXDOWN_DOLBYPLII                   0x084094A2
 334 // DCA_FORMAT of DCA_3F2R | DCA_OUT_DPLII  = 1033 = 0x409
 335 // A52_FORMAT of A52_DOLBY | A52_USE_DPLII = 74   = 0x4A
 336 // discrete channel count of 2
 337 #define HB_AMIXDOWN_6CH                         0x10089176
 338 // DCA_FORMAT of DCA_3F2R | DCA_LFE        = 137  = 0x089
 339 // A52_FORMAT of A52_3F2R | A52_LFE        = 23   = 0x17
 340 // discrete channel count of 6
 341 /* define some macros to extract the various information from the HB_AMIXDOWN_XXXX values */
 342 #define HB_AMIXDOWN_GET_DCA_FORMAT( a ) ( ( a & HB_AMIXDOWN_DCA_FORMAT_MASK ) >> 12 )
 343 #define HB_AMIXDOWN_GET_A52_FORMAT( a ) ( ( a & HB_AMIXDOWN_A52_FORMAT_MASK ) >> 4 )
 344 #define HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT( a ) ( ( a & HB_AMIXDOWN_DISCRETE_CHANNEL_COUNT_MASK ) )
 345
 346 /* Input Channel Layout */
 347 /* define some masks, used to extract the various information from the HB_AMIXDOWN_XXXX values */
 348 #define HB_INPUT_CH_LAYOUT_DISCRETE_FRONT_MASK  0x00F0000
 349 #define HB_INPUT_CH_LAYOUT_DISCRETE_REAR_MASK   0x000F000
 350 #define HB_INPUT_CH_LAYOUT_DISCRETE_LFE_MASK    0x0000F00
 351 #define HB_INPUT_CH_LAYOUT_DISCRETE_NO_LFE_MASK 0xFFFF0FF
 352 #define HB_INPUT_CH_LAYOUT_ENCODED_FRONT_MASK   0x00000F0
 353 #define HB_INPUT_CH_LAYOUT_ENCODED_REAR_MASK    0x000000F
 354 /* define the input channel layouts used to describe the channel layout of this audio */
 355 #define HB_INPUT_CH_LAYOUT_MONO    0x0110010
 356 #define HB_INPUT_CH_LAYOUT_STEREO  0x0220020
 357 #define HB_INPUT_CH_LAYOUT_DOLBY   0x0320031
 358 #define HB_INPUT_CH_LAYOUT_3F      0x0430030
 359 #define HB_INPUT_CH_LAYOUT_2F1R    0x0521021
 360 #define HB_INPUT_CH_LAYOUT_3F1R    0x0631031
 361 #define HB_INPUT_CH_LAYOUT_2F2R    0x0722022
 362 #define HB_INPUT_CH_LAYOUT_3F2R    0x0832032
 363 #define HB_INPUT_CH_LAYOUT_4F2R    0x0942042
 364 #define HB_INPUT_CH_LAYOUT_3F4R    0x0a34034
 365 #define HB_INPUT_CH_LAYOUT_HAS_LFE 0x0000100
 366 /* define some macros to extract the various information from the HB_AMIXDOWN_XXXX values */
 367 #define HB_INPUT_CH_LAYOUT_GET_DISCRETE_FRONT_COUNT( a ) ( ( a & HB_INPUT_CH_LAYOUT_DISCRETE_FRONT_MASK ) >> 16 )
 368 #define HB_INPUT_CH_LAYOUT_GET_DISCRETE_REAR_COUNT( a )  ( ( a & HB_INPUT_CH_LAYOUT_DISCRETE_REAR_MASK ) >> 12 )
 369 #define HB_INPUT_CH_LAYOUT_GET_DISCRETE_LFE_COUNT( a )   ( ( a & HB_INPUT_CH_LAYOUT_DISCRETE_LFE_MASK ) >> 8 )
 370 #define HB_INPUT_CH_LAYOUT_GET_DISCRETE_COUNT( a ) ( ( ( a & HB_INPUT_CH_LAYOUT_DISCRETE_FRONT_MASK ) >> 16 ) + ( ( a & HB_INPUT_CH_LAYOUT_DISCRETE_REAR_MASK ) >> 12 ) + ( ( a & HB_INPUT_CH_LAYOUT_DISCRETE_LFE_MASK ) >> 8 ) )
 371 #define HB_INPUT_CH_LAYOUT_GET_ENCODED_FRONT_COUNT( a )   ( ( a & HB_INPUT_CH_LAYOUT_ENCODED_FRONT_MASK ) >> 4 )
 372 #define HB_INPUT_CH_LAYOUT_GET_ENCODED_REAR_COUNT( a )   ( ( a & HB_INPUT_CH_LAYOUT_ENCODED_REAR_MASK ) )
 373
 374 struct hb_audio_config_s
 375 {
 376     /* Output */
 377     struct
 378     {
 379             int track;      /* Output track number */
 380             uint32_t codec;  /* Output audio codec.
 381                                  * HB_ACODEC_AC3 means pass-through, then bitrate and samplerate
 382                                  * are ignored.
 383                                  */
 384             int samplerate; /* Output sample rate (Hz) */
 385             int bitrate;    /* Output bitrate (kbps) */
 386             int mixdown;    /* The mixdown format to be used for this audio track (see HB_AMIXDOWN_*) */
 387             double dynamic_range_compression; /* Amount of DRC that gets applied to this track */
 388             char * name;    /* Output track name */
 389     } out;
 390
 391     /* Input */
 392     struct
 393     {
 394         int track;                /* Input track number */
 395         PRIVATE uint32_t codec;   /* Input audio codec */
 396         PRIVATE uint32_t codec_param; /* per-codec config info */
 397         PRIVATE uint32_t version; /* Bitsream version */
 398         PRIVATE uint32_t mode;    /* Bitstream mode, codec dependent encoding */
 399         PRIVATE int samplerate; /* Input sample rate (Hz) */
 400         PRIVATE int bitrate;    /* Input bitrate (kbps) */
 401         PRIVATE int channel_layout;
 402         /* channel_layout is the channel layout of this audio this is used to
 403         * provide a common way of describing the source audio
 404         */
 405     } in;
 406
 407     /* Misc. */
 408     union
 409     {
 410         PRIVATE int ac3;    /* flags.ac3 is only set when the source audio format is HB_ACODEC_AC3 */
 411         PRIVATE int dca;    /* flags.dca is only set when the source audio format is HB_ACODEC_DCA */
 412     } flags;
 413 #define AUDIO_F_DOLBY (1 << 31)  /* set if source uses Dolby Surround */
 414
 415     struct
 416     {
 417         PRIVATE char description[1024];
 418         PRIVATE char simple[1024];
 419         PRIVATE char iso639_2[4];
 420         PRIVATE uint8_t type; /* normal, visually impared, directors */
 421     } lang;
 422 };
 423
 424 #ifdef __LIBHB__
 425 struct hb_audio_s
 426 {
 427     int id;
 428
 429     hb_audio_config_t config;
 430
 431     struct {
 432         hb_fifo_t * fifo_in;   /* AC3/MPEG/LPCM ES */
 433         hb_fifo_t * fifo_raw;  /* Raw audio */
 434         hb_fifo_t * fifo_sync; /* Resampled, synced raw audio */
 435         hb_fifo_t * fifo_out;  /* MP3/AAC/Vorbis ES */
 436
 437         hb_esconfig_t config;
 438         hb_mux_data_t * mux_data;
 439         hb_fifo_t     * scan_cache;
 440     } priv;
 441 };
 442 #endif
 443
 444 struct hb_chapter_s
 445 {
 446     int      index;
 447     int      pgcn;
 448     int      pgn;
 449     int      cell_start;
 450     int      cell_end;
 451     int      block_start;
 452     int      block_end;
 453     int      block_count;
 454
 455     /* Visual-friendly duration */
 456     int      hours;
 457     int      minutes;
 458     int      seconds;
 459
 460     /* Exact duration (in 1/90000s) */
 461     uint64_t duration;
 462
 463     /* Optional chapter title */
 464     char     title[1024];
 465 };
 466
 467 struct hb_subtitle_s
 468 {
 469     int  id;
 470     int  track;
 471
 472     hb_subtitle_config_t config;
 473
 474     enum subtype { PICTURESUB, TEXTSUB } format;
 475     enum subsource { VOBSUB, SRTSUB, CC608SUB, CC708SUB } source;
 476     char lang[1024];
 477     char iso639_2[4];
 478     uint8_t type; /* Closed Caption, Childrens, Directors etc */
 479
 480     int hits;     /* How many hits/occurrences of this subtitle */
 481     int forced_hits; /* How many forced hits in this subtitle */
 482
 483 #ifdef __LIBHB__
 484     /* Internal data */
 485     hb_fifo_t * fifo_in;  /* SPU ES */
 486     hb_fifo_t * fifo_raw; /* Decoded SPU */
 487     hb_fifo_t * fifo_sync;/* Synced */
 488     hb_fifo_t * fifo_out; /* Correct Timestamps, ready to be muxed */
 489     hb_mux_data_t * mux_data;
 490 #endif
 491 };
 492
 493 struct hb_metadata_s
 494 {
 495     char  name[255];
 496     char  artist[255];
 497     char  composer[255];
 498     char  release_date[255];
 499     char  comment[1024];
 500     char  album[255];
 501     char  genre[255];
 502     uint32_t coverart_size;
 503     uint8_t *coverart;
 504 };
 505
 506 struct hb_title_s
 507 {
 508     enum { HB_DVD_TYPE, HB_STREAM_TYPE } type;
 509     char        path[1024];
 510     char        name[1024];
 511     int         index;
 512     int         vts;
 513     int         ttn;
 514     int         cell_start;
 515     int         cell_end;
 516     int         block_start;
 517     int         block_end;
 518     int         block_count;
 519     int         angle_count;
 520
 521     /* Visual-friendly duration */
 522     int         hours;
 523     int         minutes;
 524     int         seconds;
 525
 526     /* Exact duration (in 1/90000s) */
 527     uint64_t    duration;
 528
 529     double      aspect;             // aspect ratio for the title's video
 530     double      container_aspect;   // aspect ratio from container (0 if none)
 531     int         width;
 532     int         height;
 533     int         pixel_aspect_width;
 534     int         pixel_aspect_height;
 535     int         rate;
 536     int         rate_base;
 537     int         crop[4];
 538     enum { HB_MPEG2_PS_DEMUXER = 0, HB_MPEG2_TS_DEMUXER, HB_NULL_DEMUXER } demuxer;
 539     int         detected_interlacing;
 540     int         video_id;               /* demuxer stream id for video */
 541     int         video_codec;            /* worker object id of video codec */
 542     int         video_codec_param;      /* codec specific config */
 543     const char  *video_codec_name;
 544     int         video_bitrate;
 545     const char  *container_name;
 546     int         data_rate;
 547
 548     uint32_t    palette[16];
 549
 550     hb_metadata_t *metadata;
 551
 552     hb_list_t * list_chapter;
 553     hb_list_t * list_audio;
 554     hb_list_t * list_subtitle;
 555
 556     /* Job template for this title */
 557     hb_job_t  * job;
 558
 559     uint32_t    flags;
 560                 // set if video stream doesn't have IDR frames
 561 #define         HBTF_NO_IDR (1 << 0)
 562 };
 563
 564
 565 struct hb_state_s
 566 {
 567 #define HB_STATE_IDLE     1
 568 #define HB_STATE_SCANNING 2
 569 #define HB_STATE_SCANDONE 4
 570 #define HB_STATE_WORKING  8
 571 #define HB_STATE_PAUSED   16
 572 #define HB_STATE_WORKDONE 32
 573 #define HB_STATE_MUXING   64
 574 #define HB_STATE_SEARCHING 128
 575     int state;
 576
 577     union
 578     {
 579         struct
 580         {
 581             /* HB_STATE_SCANNING */
 582             int title_cur;
 583             int title_count;
 584         } scanning;
 585
 586         struct
 587         {
 588             /* HB_STATE_WORKING */
 589             float progress;
 590             int   job_cur;
 591             int   job_count;
 592             float rate_cur;
 593             float rate_avg;
 594             int   hours;
 595             int   minutes;
 596             int   seconds;
 597             int   sequence_id;
 598         } working;
 599
 600         struct
 601         {
 602             /* HB_STATE_WORKDONE */
 603 #define HB_ERROR_NONE     0
 604 #define HB_ERROR_CANCELED 1
 605 #define HB_ERROR_UNKNOWN  2
 606             int error;
 607         } workdone;
 608
 609         struct
 610         {
 611             /* HB_STATE_MUXING */
 612             float progress;
 613         } muxing;
 614     } param;
 615 };
 616
 617 typedef struct hb_work_info_s
 618 {
 619     const char *name;
 620     int     profile;
 621     int     level;
 622     int     bitrate;
 623     int     rate;
 624     int     rate_base;
 625     int     flags;
 626     int     version;
 627     int     mode;
 628     union {
 629         struct {    // info only valid for video decoders
 630             int     width;
 631             int     height;
 632             int     pixel_aspect_width;
 633             int     pixel_aspect_height;
 634             double  aspect;
 635         };
 636         struct {    // info only valid for audio decoders
 637             int     channel_layout;
 638         };
 639     };
 640 } hb_work_info_t;
 641
 642 struct hb_work_object_s
 643 {
 644     int                 id;
 645     char              * name;
 646
 647 #ifdef __LIBHB__
 648     int              (* init)  ( hb_work_object_t *, hb_job_t * );
 649     int              (* work)  ( hb_work_object_t *, hb_buffer_t **,
 650                                  hb_buffer_t ** );
 651     void             (* close) ( hb_work_object_t * );
 652     /* the info entry point is used by scan to get bitstream information
 653      * during a decode (i.e., it should only be called after at least one
 654      * call to the 'work' entry point). currently it's only called for
 655      * video streams & can be null for other work objects. */
 656     int              (* info)  ( hb_work_object_t *, hb_work_info_t * );
 657     /* the bitstream info entry point is used by scan to get bitstream
 658      * information from a buffer. it doesn't have to be called during a
 659      * decode (it can be called even if init & work haven't been).
 660      * currently it's only called for audio streams & can be null for
 661      * other work objects. */
 662     int              (* bsinfo)  ( hb_work_object_t *, const hb_buffer_t *,
 663                                    hb_work_info_t * );
 664
 665     hb_fifo_t         * fifo_in;
 666     hb_fifo_t         * fifo_out;
 667     hb_esconfig_t     * config;
 668
 669     /* Pointer hb_audio_t so we have access to the info in the audio worker threads. */
 670     hb_audio_t        * audio;
 671
 672     /* Pointer hb_subtitle_t so we have access to the info in the subtitle worker threads. */
 673     hb_subtitle_t     * subtitle;
 674
 675     hb_work_private_t * private_data;
 676
 677     hb_thread_t       * thread;
 678     volatile int      * done;
 679     int                 status;
 680     int                 codec_param;
 681     hb_title_t        * title;
 682
 683     hb_work_object_t  * next;
 684     int                 thread_sleep_interval;
 685 #endif
 686 };
 687
 688 extern hb_work_object_t hb_sync_video;
 689 extern hb_work_object_t hb_sync_audio;
 690 extern hb_work_object_t hb_decmpeg2;
 691 extern hb_work_object_t hb_decvobsub;
 692 extern hb_work_object_t hb_encvobsub;
 693 extern hb_work_object_t hb_deccc608;
 694 extern hb_work_object_t hb_decsrtsub;
 695 extern hb_work_object_t hb_render;
 696 extern hb_work_object_t hb_encavcodec;
 697 extern hb_work_object_t hb_encx264;
 698 extern hb_work_object_t hb_enctheora;
 699 extern hb_work_object_t hb_deca52;
 700 extern hb_work_object_t hb_decdca;
 701 extern hb_work_object_t hb_decavcodec;
 702 extern hb_work_object_t hb_decavcodecv;
 703 extern hb_work_object_t hb_decavcodecvi;
 704 extern hb_work_object_t hb_decavcodecai;
 705 extern hb_work_object_t hb_declpcm;
 706 extern hb_work_object_t hb_encfaac;
 707 extern hb_work_object_t hb_enclame;
 708 extern hb_work_object_t hb_encvorbis;
 709 extern hb_work_object_t hb_muxer;
 710 extern hb_work_object_t hb_encca_aac;
 711
 712 #define FILTER_OK      0
 713 #define FILTER_DELAY   1
 714 #define FILTER_FAILED  2
 715 #define FILTER_DROP    3
 716
 717 struct hb_filter_object_s
 718 {
 719     int                     id;
 720     char                  * name;
 721     char                  * settings;
 722
 723 #ifdef __LIBHB__
 724     hb_filter_private_t* (* init)  ( int, int, int, char * );
 725
 726     int                  (* work)  ( const hb_buffer_t *, hb_buffer_t **,
 727                                      int, int, int, hb_filter_private_t * );
 728
 729     void                 (* close) ( hb_filter_private_t * );
 730
 731     hb_filter_private_t   * private_data;
 732     //hb_buffer_t           * buffer;
 733 #endif
 734 };
 735
 736 extern hb_filter_object_t hb_filter_detelecine;
 737 extern hb_filter_object_t hb_filter_deinterlace;
 738 extern hb_filter_object_t hb_filter_deblock;
 739 extern hb_filter_object_t hb_filter_denoise;
 740 extern hb_filter_object_t hb_filter_decomb;
 741 extern hb_filter_object_t hb_filter_rotate;
 742
 743 typedef void hb_error_handler_t( const char *errmsg );
 744
 745 extern void hb_register_error_handler( hb_error_handler_t * handler );
 746
 747 char * hb_strdup_printf( char * fmt, ... );
 748
 749 #endif