123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552 |
- /*
- * This file is part of the UCB release of Plan 9. It is subject to the license
- * terms in the LICENSE file found in the top-level directory of this
- * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No
- * part of the UCB release of Plan 9, including this file, may be copied,
- * modified, propagated, or distributed except according to the terms contained
- * in the LICENSE file.
- */
- /*
- * lame utility library include file
- *
- * Copyright (c) 1999 Albert L Faber
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Library General Public
- * License as published by the Free Software Foundation; either
- * version 2 of the License, or (at your option) any later version.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Library General Public License for more details.
- *
- * You should have received a copy of the GNU Library General Public
- * License along with this library; if not, write to the
- * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
- * Boston, MA 02111-1307, USA.
- */
- #ifndef LAME_UTIL_H
- #define LAME_UTIL_H
- #ifdef HUGE_VAL /* math.h already seen? */
- #ifndef fabs
- #define fabs(x) ((double)((x) < 0? -(x): (x)))
- #endif
- #endif
- /***********************************************************************
- *
- * Global Include Files
- *
- ***********************************************************************/
- #include "machine.h"
- #include "encoder.h"
- #include "lame.h"
- #include "lame-analysis.h"
- #include "id3tag.h"
- /***********************************************************************
- *
- * Global Definitions
- *
- ***********************************************************************/
- #ifndef FALSE
- #define FALSE 0
- #endif
- #ifndef TRUE
- #define TRUE (!FALSE)
- #endif
- #ifdef UINT_MAX
- # define MAX_U_32_NUM UINT_MAX
- #else
- # define MAX_U_32_NUM 0xFFFFFFFF
- #endif
- #ifndef PI
- # ifdef M_PI
- # define PI M_PI
- # else
- # define PI 3.14159265358979323846
- # endif
- #endif
- #ifdef M_LN2
- # define LOG2 M_LN2
- #else
- # define LOG2 0.69314718055994530942
- #endif
- #ifdef M_LN10
- # define LOG10 M_LN10
- #else
- # define LOG10 2.30258509299404568402
- #endif
- #ifdef M_SQRT2
- # define SQRT2 M_SQRT2
- #else
- # define SQRT2 1.41421356237309504880
- #endif
- #define HAN_SIZE 512
- #define CRC16_POLYNOMIAL 0x8005
- #define MAX_BITS 4095
- /* "bit_stream.h" Definitions */
- #define BUFFER_SIZE LAME_MAXMP3BUFFER
- #define Min(A, B) ((A) < (B) ? (A) : (B))
- #define Max(A, B) ((A) > (B) ? (A) : (B))
- /***********************************************************************
- *
- * Global Type Definitions
- *
- ***********************************************************************/
- /* "bit_stream.h" Type Definitions */
- typedef struct bit_stream_struc {
- unsigned char *buf; /* bit stream buffer */
- int buf_size; /* size of buffer (in number of bytes) */
- int totbit; /* bit counter of bit stream */
- int buf_byte_idx; /* pointer to top byte in buffer */
- int buf_bit_idx; /* pointer to top bit of top byte in buffer */
-
- /* format of file in rd mode (BINARY/ASCII) */
- } Bit_stream_struc;
- #include "l3side.h"
- /* variables used for --nspsytune */
- typedef struct {
- int use; /* indicates the use of exp_nspsytune */
- int safejoint; /* safe joint stereo mode */
- FLOAT last_en_subshort[4][9];
- FLOAT last_attack_intensity[4][9];
- FLOAT last_thm[4][SBMAX_s][3];
- int last_attacks[4][3];
- FLOAT pe_l[4],pe_s[4];
- FLOAT pefirbuf[19];
- FLOAT bass,alto,treble;
- } nsPsy_t;
- typedef struct
- {
- int sum; // what we have seen so far
- int seen; // how many frames we have seen in this chunk
- int want; // how many frames we want to collect into one chunk
- int pos; // actual position in our bag
- int size; // size of our bag
- int *bag; // pointer to our bag
- } VBR_seek_info_t;
- /**
- * ATH related stuff, if something new ATH related has to be added,
- * please plugg it here into the ATH_t struct
- */
- typedef struct
- {
- int use_adjust; // do we want to use the auto adjustment yes/no
- FLOAT8 adjust; // lowering based on peak volume, 1 = no lowering
- FLOAT8 adjust_limit; // limit for dynamic ATH adjust
- FLOAT8 decay; // determined to lower x dB each second
- FLOAT8 l[SBMAX_l]; // ATH for sfbs in long blocks
- FLOAT8 s[SBMAX_s]; // ATH for sfbs in short blocks
- FLOAT8 cb[CBANDS]; // ATH for convolution bands
- } ATH_t;
- /* Guest structure, only temporarly here */
- typedef enum {
- coding_MPEG_Layer_1 = 1,
- coding_MPEG_Layer_2 = 2,
- coding_MPEG_Layer_3 = 3,
- coding_MPEG_AAC = 4,
- coding_Ogg_Vorbis = 5,
- coding_MPEG_plus = 6
- } coding_t;
- #define MAX_CHANNELS 2
- typedef struct {
- unsigned long Class_ID; /* Class ID to recognize a resample_t
- object */
- long double sample_freq_in; /* Input sample frequency in Hz */
- long double sample_freq_out; /* requested Output sample frequency in Hz */
- float lowpass_freq; /* lowpass frequency, this is the -6 dB
- point */
- int scale_in; /* the resampling is actually done by
- scale_out: */
- int scale_out; /* frequency is
- samplefreq_in * scale_out / scal */
- int taps; /* number of taps for every FIR resample
- filter */
- sample_t** fir; /* the FIR resample filters:
- fir [scale_out] [taps */
- void* firfree; /* start address of the alloced memory for
- fir, */
- unsigned char* src_step;
- sample_t* in_old [MAX_CHANNELS];
- // uint64_t sample_count [MAX_CHANNELS];
- unsigned fir_stepper [MAX_CHANNELS];
- int inp_stepper [MAX_CHANNELS];
- } resample_t;
- typedef struct {
- /********************************************************************
- * internal variables NOT set by calling program, and should not be *
- * modified by the calling program *
- ********************************************************************/
-
- /*
- * Some remarks to the Class_ID field:
- * The Class ID is an Identifier for a pointer to this struct.
- * It is very unlikely that a pointer to lame_global_flags has the same 32 bits
- * in it's structure (large and other special properties, for instance prime).
- *
- * To test that the structure is right and initialized, use:
- * if ( gfc -> Class_ID == LAME_ID ) ...
- * Other remark:
- * If you set a flag to 0 for uninit data and 1 for init data, the right test
- * should be "if (flag == 1)" and NOT "if (flag)". Unintended modification
- * of this element will be otherwise misinterpreted as an init.
- */
-
- #define LAME_ID 0xFFF88E3B
- unsigned long Class_ID;
- struct {
- void (*msgf) (const char *format, va_list ap);
- void (*debugf)(const char *format, va_list ap);
- void (*errorf)(const char *format, va_list ap);
- } report;
-
- int lame_encode_frame_init;
- int iteration_init_init;
- int fill_buffer_resample_init;
- int psymodel_init;
- int padding; /* padding for the current frame? */
- int mode_gr; /* granules per frame */
- int channels_in; /* number of channels in the input data stream (PCM or decoded PCM) */
- int channels_out; /* number of channels in the output data stream (not used for decoding) */
- resample_t* resample_in; /* context for coding (PCM=>MP3) resampling */
- resample_t* resample_out; /* context for decoding (MP3=>PCM) resampling */
- long double samplefreq_in;
- long double samplefreq_out;
- #ifndef MFSIZE
- # define MFSIZE ( 3*1152 + ENCDELAY - MDCTDELAY )
- #endif
- #ifdef KLEMM_44
- sample_t* mfbuf [MAX_CHANNELS];
- #else
- sample_t mfbuf [2] [MFSIZE];
- #endif
- size_t frame_size; /* size of one frame in samples per channel */
- lame_global_flags* gfp; /* needed as long as the frame encoding functions must access gfp (all needed information can be added to gfc) */
- coding_t coding; /* MPEG Layer 1/2/3, Ogg Vorbis, MPEG AAC, ... */
- unsigned long frame_count; /* Number of frames coded, 2^32 > 3 years */
- int mf_samples_to_encode;
- int mf_size;
- float ampl; /* amplification at the end of the current chunk (1. = 0 dB) */
- float last_ampl; /* amplification at the end of the last chunk (1. = 0 dB) */
- int VBR_min_bitrate; /* min bitrate index */
- int VBR_max_bitrate; /* max bitrate index */
- float resample_ratio; /* input_samp_rate/output_samp_rate */
- int bitrate_index;
- int samplerate_index;
- int mode_ext;
- /* lowpass and highpass filter control */
- float lowpass1,lowpass2; /* normalized frequency bounds of passband */
- float highpass1,highpass2; /* normalized frequency bounds of passband */
-
- /* polyphase filter (filter_type=0) */
- int lowpass_band; /* zero bands >= lowpass_band in the polyphase filterbank */
- int highpass_band; /* zero bands <= highpass_band */
- int lowpass_start_band; /* amplify bands between start */
- int lowpass_end_band; /* and end for lowpass */
- int highpass_start_band; /* amplify bands between start */
- int highpass_end_band; /* and end for highpass */
- int filter_type; /* 0=polyphase filter, 1= FIR filter 2=MDCT filter(bad)*/
- int quantization; /* 0 = ISO formual, 1=best amplitude */
- int noise_shaping; /* 0 = none
- 1 = ISO AAC model
- 2 = allow scalefac_select=1
- */
- int noise_shaping_amp; /* 0 = ISO model: amplify all distorted bands
- 1 = amplify only most distorted band
- 2 = amplify bands using?
- 3 = amplify bands using?
- */
- int psymodel; /* 1 = gpsycho. 0 = none */
- int noise_shaping_stop; /* 0 = stop at over=0, all scalefacs amplified or
- a scalefac has reached max value
- 1 = stop when all scalefacs amplified or
- a scalefac has reached max value
- 2 = stop when all scalefacs amplified
- */
- int use_best_huffman; /* 0 = no. 1=outside loop 2=inside loop(slow) */
- /* variables used by lame.c */
- Bit_stream_struc bs;
- III_side_info_t l3_side;
- FLOAT8 ms_ratio[2];
- /* used for padding */
- int frac_SpF;
- int slot_lag;
- /* optional ID3 tags, used in id3tag.c */
- struct id3tag_spec tag_spec;
- /* variables used by quantize.c */
- int OldValue[2];
- int CurrentStep;
- FLOAT8 decay;
- FLOAT8 masking_lower;
- char bv_scf[576];
-
- int sfb21_extra; /* will be set in lame_init_params */
-
- int is_mpeg1; /* 1 for MPEG-1, 0 for MPEG-2(.5) */
- #ifndef KLEMM_44
- /* variables used by util.c */
- /* BPC = maximum number of filter convolution windows to precompute */
- #define BPC 320
- sample_t *inbuf_old [2];
- sample_t *blackfilt [2*BPC+1];
- FLOAT8 itime[2];
- #endif
- int sideinfo_len;
- /* variables for newmdct.c */
- FLOAT8 sb_sample[2][2][18][SBLIMIT];
- FLOAT8 amp_lowpass[32];
- FLOAT8 amp_highpass[32];
- /* variables for bitstream.c */
- /* mpeg1: buffer=511 bytes smallest frame: 96-38(sideinfo)=58
- * max number of frames in reservoir: 8
- * mpeg2: buffer=255 bytes. smallest frame: 24-23bytes=1
- * with VBR, if you are encoding all silence, it is possible to
- * have 8kbs/24khz frames with 1byte of data each, which means we need
- * to buffer up to 255 headers! */
- /* also, max_header_buf has to be a power of two */
- #define MAX_HEADER_BUF 256
- #define MAX_HEADER_LEN 40 /* max size of header is 38 */
- struct {
- int write_timing;
- int ptr;
- char buf[MAX_HEADER_LEN];
- } header[MAX_HEADER_BUF];
- int h_ptr;
- int w_ptr;
- int ancillary_flag;
-
- /* variables for reservoir.c */
- int ResvSize; /* in bits */
- int ResvMax; /* in bits */
-
- scalefac_struct scalefac_band;
- /* DATA FROM PSYMODEL.C */
- /* The static variables "r", "phi_sav", "new", "old" and "oldest" have */
- /* to be remembered for the unpredictability measure. For "r" and */
- /* "phi_sav", the first index from the left is the channel select and */
- /* the second index is the "age" of the data. */
- FLOAT8 minval[CBANDS];
- FLOAT8 nb_1[4][CBANDS], nb_2[4][CBANDS];
- FLOAT8 s3_s[CBANDS][CBANDS];
- FLOAT8 s3_l[CBANDS][CBANDS];
- III_psy_xmin thm[4];
- III_psy_xmin en[4];
-
- /* unpredictability calculation
- */
- int cw_upper_index;
- int cw_lower_index;
- FLOAT ax_sav[4][2][HBLKSIZE];
- FLOAT bx_sav[4][2][HBLKSIZE];
- FLOAT rx_sav[4][2][HBLKSIZE];
- FLOAT cw[HBLKSIZE];
- /* fft and energy calculation */
- FLOAT wsamp_L[2][BLKSIZE];
- FLOAT energy[HBLKSIZE];
- FLOAT wsamp_S[2][3][BLKSIZE_s];
- FLOAT energy_s[3][HBLKSIZE_s];
- FLOAT tot_ener[4];
-
- /* fft.c */
- FLOAT window[BLKSIZE], window_s[BLKSIZE_s/2];
-
-
- /* Scale Factor Bands */
- FLOAT8 w1_l[SBMAX_l], w2_l[SBMAX_l];
- FLOAT8 w1_s[SBMAX_s], w2_s[SBMAX_s];
- FLOAT8 mld_l[SBMAX_l],mld_s[SBMAX_s];
- int bu_l[SBMAX_l],bo_l[SBMAX_l] ;
- int bu_s[SBMAX_s],bo_s[SBMAX_s] ;
- int npart_l,npart_s;
- int npart_l_orig,npart_s_orig;
-
- int s3ind[CBANDS][2];
- int s3ind_s[CBANDS][2];
- FLOAT8 SNR_s[CBANDS];
- int numlines_s[CBANDS];
- int numlines_l[CBANDS];
-
- /* frame analyzer */
- FLOAT energy_save[4][HBLKSIZE];
- FLOAT8 pe_save[4];
- FLOAT8 ers_save[4];
-
- /* simple statistics */
- int bitrate_stereoMode_Hist [16] [4+1];
- /* ratios */
- FLOAT8 pe[4];
- FLOAT8 ms_ratio_s_old,ms_ratio_l_old;
- FLOAT8 ms_ener_ratio_old;
- /* block type */
- int blocktype_old[2];
- /* used by the frame analyzer */
- plotting_data *pinfo;
- /* CPU features */
- struct {
- unsigned int i387 : 1; /* FPU is a normal Intel CPU */
- unsigned int MMX : 1; /* Pentium MMX, Pentium II...IV, K6, K6-2,
- K6-III, Athlon */
- unsigned int AMD_3DNow : 1; /* K6-2, K6-III, Athlon */
- unsigned int SIMD : 1; /* Pentium III, Pentium 4 */
- unsigned int SIMD2 : 1; /* Pentium 4, K8 */
- } CPU_features;
-
- /* functions to replace with CPU feature optimized versions in takehiro.c */
- int (*choose_table)(const int *ix, const int *end, int *s);
-
- nsPsy_t nsPsy; /* variables used for --nspsytune */
-
- unsigned crcvalue;
-
- VBR_seek_info_t VBR_seek_table; // used for Xing VBR header
-
- ATH_t *ATH; // all ATH related stuff
-
-
- } lame_internal_flags;
-
- /***********************************************************************
- *
- * Global Function Prototype Declarations
- *
- ***********************************************************************/
- void freegfc(lame_internal_flags *gfc);
- extern int BitrateIndex(int, int,int);
- extern int FindNearestBitrate(int,int,int);
- extern int map2MP3Frequency(int freq);
- extern int SmpFrqIndex(int, int*);
- extern FLOAT8 ATHformula(FLOAT8 f,lame_global_flags *gfp);
- extern FLOAT8 freq2bark(FLOAT8 freq);
- extern FLOAT8 freq2cbw(FLOAT8 freq);
- extern void freorder(int scalefac_band[],FLOAT8 ix_orig[576]);
- void disable_FPE(void);
- extern void
- getframebits(lame_global_flags *gfp, int *bitsPerFrame, int *mean_bits);
- void fill_buffer(lame_global_flags *gfp,
- sample_t *mfbuf[2],
- sample_t *in_buffer[2],
- int nsamples, int *n_in, int *n_out);
- int fill_buffer_resample (
- lame_global_flags *gfp,
- sample_t* outbuf,
- int desired_len,
- sample_t* inbuf,
- int len,
- int* num_used,
- int channels );
- extern int has_i387 ( void );
- extern int has_MMX ( void );
- extern int has_3DNow ( void );
- extern int has_SIMD ( void );
- extern int has_SIMD2 ( void );
- extern void updateStats (lame_internal_flags *gfc);
- /***********************************************************************
- *
- * Macros about Message Printing and Exit
- *
- ***********************************************************************/
- extern void lame_errorf(const lame_internal_flags *gfc, const char *, ...);
- extern void lame_debugf(const lame_internal_flags *gfc, const char *, ...);
- extern void lame_msgf (const lame_internal_flags *gfc, const char *, ...);
- #define DEBUGF lame_debugf
- #define ERRORF lame_errorf
- #define MSGF lame_msgf
- int select_kth_int(int b[], int N, int k);
- #endif /* LAME_UTIL_H */
|