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.m0k.org/>.
   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 #ifndef MIN
  22 #define MIN( a, b ) ( (a) > (b) ? (b) : (a) )
  23 #endif
  24 #ifndef MAX
  25 #define MAX( a, b ) ( (a) > (b) ? (a) : (b) )
  26 #endif
  27
  28 #define EVEN( a )        ( (a) + ( (a) & 1 ) )
  29 #define MULTIPLE_16( a ) ( 16 * ( ( (a) + 8 ) / 16 ) )
  30
  31 typedef struct hb_handle_s hb_handle_t;
  32 typedef struct hb_list_s hb_list_t;
  33 typedef struct hb_rate_s hb_rate_t;
  34 typedef struct hb_mixdown_s hb_mixdown_t;
  35 typedef struct hb_job_s  hb_job_t;
  36 typedef struct hb_title_s hb_title_t;
  37 typedef struct hb_chapter_s hb_chapter_t;
  38 typedef struct hb_audio_s hb_audio_t;
  39 typedef struct hb_subtitle_s hb_subtitle_t;
  40 typedef struct hb_state_s hb_state_t;
  41 typedef union  hb_esconfig_u     hb_esconfig_t;
  42 typedef struct hb_work_private_s hb_work_private_t;
  43 typedef struct hb_work_object_s  hb_work_object_t;
  44 typedef struct hb_buffer_s hb_buffer_t;
  45 typedef struct hb_fifo_s hb_fifo_t;
  46 typedef struct hb_lock_s hb_lock_t;
  47
  48 #include "ports.h"
  49 #ifdef __LIBHB__
  50 #include "internal.h"
  51 #endif
  52
  53 hb_list_t * hb_list_init();
  54 int         hb_list_count( hb_list_t * );
  55 void        hb_list_add( hb_list_t *, void * );
  56 void        hb_list_rem( hb_list_t *, void * );
  57 void      * hb_list_item( hb_list_t *, int );
  58 void        hb_list_close( hb_list_t ** );
  59
  60 void hb_reduce( int *x, int *y, int num, int den );
  61
  62 #define HB_KEEP_WIDTH  0
  63 #define HB_KEEP_HEIGHT 1
  64 void hb_fix_aspect( hb_job_t * job, int keep );
  65
  66 int hb_calc_bitrate( hb_job_t *, int size );
  67
  68 struct hb_rate_s
  69 {
  70     char * string;
  71     int    rate;
  72 };
  73
  74 struct hb_mixdown_s
  75 {
  76     char * human_readable_name;
  77     char * internal_name;
  78     char * short_name;
  79     int    amixdown;
  80 };
  81
  82 #define HB_ASPECT_BASE 9
  83 #define HB_VIDEO_RATE_BASE   27000000
  84
  85 extern hb_rate_t    hb_video_rates[];
  86 extern int          hb_video_rates_count;
  87 extern hb_rate_t    hb_audio_rates[];
  88 extern int          hb_audio_rates_count;
  89 extern int          hb_audio_rates_default;
  90 extern hb_rate_t    hb_audio_bitrates[];
  91 extern int          hb_audio_bitrates_count;
  92 extern int          hb_audio_bitrates_default;
  93 extern hb_mixdown_t hb_audio_mixdowns[];
  94 extern int          hb_audio_mixdowns_count;
  95 int hb_mixdown_get_mixdown_from_short_name( const char * short_name );
  96 const char * hb_mixdown_get_short_name_from_mixdown( int amixdown );
  97
  98 /******************************************************************************
  99  * hb_job_t: settings to be filled by the UI
 100  *****************************************************************************/
 101 struct hb_job_s
 102 {
 103     /* Pointer to the title to be ripped */
 104     hb_title_t    * title;
 105
 106     /* Chapter selection */
 107     int             chapter_start;
 108     int             chapter_end;
 109
 110         /* Include chapter marker track in mp4? */
 111     int             chapter_markers;
 112
 113     /* Picture settings:
 114          crop:                must be multiples of 2 (top/bottom/left/right)
 115          deinterlace:         0 or 1
 116          width:               must be a multiple of 16
 117          height:              must be a multiple of 16
 118          keep_ratio:          used by UIs
 119          pixel_ratio:         store pixel aspect ratio in the video
 120          pixel_aspect_width:  numerator for pixel aspect ratio
 121          pixel_aspect_height: denominator for pixel aspect ratio
 122                  maxWidth:                        keep width below this
 123                  maxHeight:                       keep height below this */
 124
 125     int             crop[4];
 126     int             deinterlace;
 127     int             width;
 128     int             height;
 129     int             keep_ratio;
 130     int             grayscale;
 131     int             pixel_ratio;
 132     int             pixel_aspect_width;
 133     int             pixel_aspect_height;
 134         int                             maxWidth;
 135         int                             maxHeight;
 136
 137
 138     /* Video settings:
 139          vcodec:            output codec
 140          vquality:          output quality (0.0..1.0)
 141                             if < 0.0 or > 1.0, bitrate is used instead
 142          vbitrate:          output bitrate (kbps)
 143          pass:              0, 1 or 2
 144          vrate, vrate_base: output framerate is vrate / vrate_base
 145                  h264_level:            boolean for whether or not we're encoding for iPod
 146                  crf:                           boolean for whether to use constant rate factor with x264
 147                  x264opts:                      string of extra x264 options
 148                  areBframes:            boolean to note if b-frames are included in x264opts */
 149 #define HB_VCODEC_MASK   0x0000FF
 150 #define HB_VCODEC_FFMPEG 0x000001
 151 #define HB_VCODEC_XVID   0x000002
 152 #define HB_VCODEC_X264   0x000004
 153
 154     int             vcodec;
 155     float           vquality;
 156     int             vbitrate;
 157     int             vrate;
 158     int             vrate_base;
 159     int             pass;
 160     int             h264_13;
 161         int                             h264_level;
 162         int                             crf;
 163         char                *x264opts;
 164         int                             areBframes;
 165
 166     /* Audio tracks:
 167          audios:          Indexes in hb_title_t's audios list, starting from 0.
 168                           -1 indicates the end of the list
 169              audio_mixdowns:  The mixdown to be used for each audio track in audios[] */
 170
 171 /* define some masks, used to extract the various information from the HB_AMIXDOWN_XXXX values */
 172 #define HB_AMIXDOWN_DCA_FORMAT_MASK             0x00FFF000
 173 #define HB_AMIXDOWN_A52_FORMAT_MASK             0x00000FF0
 174 #define HB_AMIXDOWN_DISCRETE_CHANNEL_COUNT_MASK 0x0000000F
 175
 176 /* define the HB_AMIXDOWN_XXXX values */
 177
 178 #define HB_AMIXDOWN_MONO                        0x01000001
 179 // DCA_FORMAT of DCA_MONO                  = 0    = 0x000
 180 // A52_FORMAT of A52_MONO                  = 1    = 0x01
 181 // discrete channel count of 1
 182
 183 #define HB_AMIXDOWN_STEREO                      0x02002022
 184 // DCA_FORMAT of DCA_STEREO                = 2    = 0x002
 185 // A52_FORMAT of A52_STEREO                = 2    = 0x02
 186 // discrete channel count of 2
 187
 188 #define HB_AMIXDOWN_DOLBY                       0x042070A2
 189 // DCA_FORMAT of DCA_3F1R | DCA_OUT_DPLI   = 519  = 0x207
 190 // A52_FORMAT of A52_DOLBY                 = 10   = 0x0A
 191 // discrete channel count of 2
 192
 193 #define HB_AMIXDOWN_DOLBYPLII                   0x084094A2
 194 // DCA_FORMAT of DCA_3F2R | DCA_OUT_DPLII  = 1033 = 0x409
 195 // A52_FORMAT of A52_DOLBY | A52_USE_DPLII = 74   = 0x4A
 196 // discrete channel count of 2
 197
 198 #define HB_AMIXDOWN_6CH                         0x10089176
 199 // DCA_FORMAT of DCA_3F2R | DCA_LFE        = 137  = 0x089
 200 // A52_FORMAT of A52_3F2R | A52_LFE        = 23   = 0x17
 201 // discrete channel count of 6
 202
 203 /* define some macros to extract the various information from the HB_AMIXDOWN_XXXX values */
 204 #define HB_AMIXDOWN_GET_DCA_FORMAT( a ) ( ( a & HB_AMIXDOWN_DCA_FORMAT_MASK ) >> 12 )
 205 #define HB_AMIXDOWN_GET_A52_FORMAT( a ) ( ( a & HB_AMIXDOWN_A52_FORMAT_MASK ) >> 4 )
 206 #define HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT( a ) ( ( a & HB_AMIXDOWN_DISCRETE_CHANNEL_COUNT_MASK ) )
 207
 208     int             audios[8];
 209         int             audio_mixdowns[8];
 210
 211     /* Audio settings:
 212          acodec:   output codec
 213          abitrate: output bitrate (kbps)
 214          arate:    output samplerate (Hz)
 215        HB_ACODEC_AC3 means pass-through, then abitrate and arate are
 216        ignored */
 217 #define HB_ACODEC_MASK   0x00FF00
 218 #define HB_ACODEC_FAAC   0x000100
 219 #define HB_ACODEC_LAME   0x000200
 220 #define HB_ACODEC_VORBIS 0x000400
 221 #define HB_ACODEC_AC3    0x000800
 222 #define HB_ACODEC_MPGA   0x001000
 223 #define HB_ACODEC_LPCM   0x002000
 224 #define HB_ACODEC_DCA    0x004000
 225     int             acodec;
 226     int             abitrate;
 227     int             arate;
 228
 229     /* Subtitle settings:
 230          subtitle: index in hb_title_t's subtitles list, starting
 231          from 0. -1 means no subtitle */
 232     int             subtitle;
 233     int                         subtitleSmartAdjust;
 234
 235     /* Muxer settings
 236          mux:  output file format
 237          file: file path */
 238 #define HB_MUX_MASK 0xFF0000
 239 #define HB_MUX_MP4  0x010000
 240 #define HB_MUX_PSP  0x020000
 241 #define HB_MUX_AVI  0x040000
 242 #define HB_MUX_OGM  0x080000
 243 #define HB_MUX_IPOD 0x100000
 244
 245     int             mux;
 246     const char          * file;
 247
 248 #ifdef __LIBHB__
 249     /* Internal data */
 250     hb_handle_t   * h;
 251     hb_lock_t     * pause;
 252     volatile int  * die;
 253     volatile int    done;
 254
 255     hb_fifo_t     * fifo_mpeg2;   /* MPEG-2 video ES */
 256     hb_fifo_t     * fifo_raw;     /* Raw pictures */
 257     hb_fifo_t     * fifo_sync;    /* Raw pictures, framerate corrected */
 258     hb_fifo_t     * fifo_render;  /* Raw pictures, scaled */
 259     hb_fifo_t     * fifo_mpeg4;   /* MPEG-4 video ES */
 260
 261     hb_thread_t   * reader;
 262     hb_thread_t   * muxer;
 263
 264     hb_list_t     * list_work;
 265
 266     hb_esconfig_t config;
 267
 268     hb_mux_data_t * mux_data;
 269 #endif
 270 };
 271
 272 struct hb_audio_s
 273 {
 274     int  id;
 275     char lang[1024];
 276     char lang_simple[1024];
 277     char iso639_2[4];
 278     int  codec;
 279     int  rate;
 280     int  bitrate;
 281         /* src_discrete_front_channels: The # of discrete front channels in the source audio
 282            src_discrete_rear_channels:  The # of discrete rear channels in the source audio
 283            src_discrete_lfe_channels:   The # of discrete lfe channels in the source audio
 284            src_encoded_front_channels:  The # of front channels encoded into the source audio
 285            src_encoded_rear_channels:   The # of rear channels encoded into the source audio */
 286     int src_discrete_front_channels;
 287     int src_discrete_rear_channels;
 288         int src_discrete_lfe_channels;
 289     int src_encoded_front_channels;
 290     int src_encoded_rear_channels;
 291
 292     /* ac3flags is only set when the source audio format is HB_ACODEC_AC3 */
 293     int ac3flags;
 294
 295     /* dcaflags is only set when the source audio format is HB_ACODEC_DCA */
 296     int dcaflags;
 297
 298 #ifdef __LIBHB__
 299     /* Internal data */
 300     hb_fifo_t * fifo_in;   /* AC3/MPEG/LPCM ES */
 301     hb_fifo_t * fifo_raw;  /* Raw audio */
 302     hb_fifo_t * fifo_sync; /* Resampled, synced raw audio */
 303     hb_fifo_t * fifo_out;  /* MP3/AAC/Vorbis ES */
 304
 305     hb_esconfig_t config;
 306     hb_mux_data_t * mux_data;
 307
 308         /* amixdown is the mixdown format to be used for this audio track */
 309         int amixdown;
 310
 311 #endif
 312 };
 313
 314 struct hb_chapter_s
 315 {
 316     int      index;
 317     int      cell_start;
 318     int      cell_end;
 319     int      block_start;
 320     int      block_end;
 321     int      block_count;
 322
 323     /* Visual-friendly duration */
 324     int      hours;
 325     int      minutes;
 326     int      seconds;
 327
 328     /* Exact duration (in 1/90000s) */
 329     uint64_t duration;
 330
 331     /* Optional chapter title */
 332     char     title[1024];
 333 };
 334
 335 struct hb_subtitle_s
 336 {
 337     int  id;
 338     char lang[1024];
 339     char iso639_2[4];
 340
 341 #ifdef __LIBHB__
 342     /* Internal data */
 343     hb_fifo_t * fifo_in;  /* SPU ES */
 344     hb_fifo_t * fifo_raw; /* Decodec SPU */
 345 #endif
 346 };
 347
 348 struct hb_title_s
 349 {
 350     char        dvd[1024];
 351     char        name[1024];
 352     int         index;
 353     int         vts;
 354     int         ttn;
 355     int         cell_start;
 356     int         cell_end;
 357     int         block_start;
 358     int         block_end;
 359     int         block_count;
 360
 361     /* Visual-friendly duration */
 362     int         hours;
 363     int         minutes;
 364     int         seconds;
 365
 366     /* Exact duration (in 1/90000s) */
 367     uint64_t    duration;
 368
 369     int         width;
 370     int         height;
 371     int         aspect;
 372     int         rate;
 373     int         rate_base;
 374     int         crop[4];
 375
 376     uint32_t    palette[16];
 377
 378     hb_list_t * list_chapter;
 379     hb_list_t * list_audio;
 380     hb_list_t * list_subtitle;
 381
 382     /* Job template for this title */
 383     hb_job_t  * job;
 384 };
 385
 386
 387 struct hb_state_s
 388 {
 389 #define HB_STATE_IDLE     1
 390 #define HB_STATE_SCANNING 2
 391 #define HB_STATE_SCANDONE 4
 392 #define HB_STATE_WORKING  8
 393 #define HB_STATE_PAUSED   16
 394 #define HB_STATE_WORKDONE 32
 395 #define HB_STATE_MUXING   64
 396     int state;
 397
 398     union
 399     {
 400         struct
 401         {
 402             /* HB_STATE_SCANNING */
 403             int title_cur;
 404             int title_count;
 405         } scanning;
 406
 407         struct
 408         {
 409             /* HB_STATE_WORKING */
 410             float progress;
 411             int   job_cur;
 412             int   job_count;
 413             float rate_cur;
 414             float rate_avg;
 415             int   hours;
 416             int   minutes;
 417             int   seconds;
 418         } working;
 419
 420         struct
 421         {
 422             /* HB_STATE_WORKDONE */
 423 #define HB_ERROR_NONE     0
 424 #define HB_ERROR_CANCELED 1
 425 #define HB_ERROR_UNKNOWN  2
 426             int error;
 427         } workdone;
 428
 429         struct
 430         {
 431             /* HB_STATE_MUXING */
 432             float progress;
 433         } muxing;
 434     } param;
 435 };
 436
 437 struct hb_work_object_s
 438 {
 439     int                 id;
 440     char              * name;
 441
 442 #ifdef __LIBHB__
 443     int              (* init)  ( hb_work_object_t *, hb_job_t * );
 444     int              (* work)  ( hb_work_object_t *, hb_buffer_t **,
 445                                  hb_buffer_t ** );
 446     void             (* close) ( hb_work_object_t * );
 447
 448     hb_fifo_t         * fifo_in;
 449     hb_fifo_t         * fifo_out;
 450     hb_esconfig_t     * config;
 451
 452         /* amixdown is the mixdown format to be used if the work object is an audio track */
 453         int               amixdown;
 454
 455     hb_work_private_t * private_data;
 456
 457     hb_thread_t       * thread;
 458     volatile int      * done;
 459
 460     hb_work_object_t  * next;
 461         int                               thread_sleep_interval;
 462 #endif
 463 };
 464
 465 extern hb_work_object_t hb_sync;
 466 extern hb_work_object_t hb_decmpeg2;
 467 extern hb_work_object_t hb_decsub;
 468 extern hb_work_object_t hb_render;
 469 extern hb_work_object_t hb_encavcodec;
 470 extern hb_work_object_t hb_encxvid;
 471 extern hb_work_object_t hb_encx264;
 472 extern hb_work_object_t hb_deca52;
 473 extern hb_work_object_t hb_decdca;
 474 extern hb_work_object_t hb_decavcodec;
 475 extern hb_work_object_t hb_declpcm;
 476 extern hb_work_object_t hb_encfaac;
 477 extern hb_work_object_t hb_enclame;
 478 extern hb_work_object_t hb_encvorbis;
 479
 480 #endif