aacenc.h 3.23 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
/*
 * AAC encoder
 * Copyright (C) 2008 Konstantin Shishkov
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#ifndef AVCODEC_AACENC_H
#define AVCODEC_AACENC_H

25
#include "libavutil/float_dsp.h"
26 27 28 29
#include "avcodec.h"
#include "put_bits.h"

#include "aac.h"
30
#include "audio_frame_queue.h"
31 32
#include "psymodel.h"

33 34 35 36 37
typedef enum AACCoder {
    AAC_CODER_FAAC = 0,
    AAC_CODER_ANMR,
    AAC_CODER_TWOLOOP,
    AAC_CODER_FAST,
38

39 40
    AAC_CODER_NB,
}AACCoder;
41

42 43
typedef struct AACEncOptions {
    int stereo_mode;
44
    int aac_coder;
45 46
} AACEncOptions;

47 48
struct AACEncContext;

49
typedef struct AACCoefficientsEncoder {
50 51 52 53 54 55 56
    void (*search_for_quantizers)(AVCodecContext *avctx, struct AACEncContext *s,
                                  SingleChannelElement *sce, const float lambda);
    void (*encode_window_bands_info)(struct AACEncContext *s, SingleChannelElement *sce,
                                     int win, int group_len, const float lambda);
    void (*quantize_and_encode_band)(struct AACEncContext *s, PutBitContext *pb, const float *in, int size,
                                     int scale_idx, int cb, const float lambda);
    void (*search_for_ms)(struct AACEncContext *s, ChannelElement *cpe, const float lambda);
57
} AACCoefficientsEncoder;
58 59 60 61 62 63 64

extern AACCoefficientsEncoder ff_aac_coders[];

/**
 * AAC encoder context
 */
typedef struct AACEncContext {
65 66
    AVClass *av_class;
    AACEncOptions options;                       ///< encoding options
67
    PutBitContext pb;
68 69
    FFTContext mdct1024;                         ///< long (1024 samples) frame transform context
    FFTContext mdct128;                          ///< short (128 samples) frame transform context
70
    AVFloatDSPContext fdsp;
71
    float *planar_samples[6];                    ///< saved preprocessed input
72 73

    int samplerate_index;                        ///< MPEG-4 samplerate index
74
    int channels;                                ///< channel count
75
    const uint8_t *chan_map;                     ///< channel configuration map
76 77 78 79 80 81 82 83

    ChannelElement *cpe;                         ///< channel elements
    FFPsyContext psy;
    struct FFPsyPreprocessContext* psypp;
    AACCoefficientsEncoder *coder;
    int cur_channel;
    int last_frame;
    float lambda;
84
    AudioFrameQueue afq;
85
    DECLARE_ALIGNED(16, int,   qcoefs)[96];      ///< quantized coefficients
86
    DECLARE_ALIGNED(32, float, scoefs)[1024];    ///< scaled coefficients
87 88 89 90

    struct {
        float *samples;
    } buffer;
91 92
} AACEncContext;

93 94
extern float ff_aac_pow34sf_tab[428];

95 96
void ff_aac_coder_init_mips(AACEncContext *c);

97
#endif /* AVCODEC_AACENC_H */