3 * Copyright (c) 2017 Rostislav Pehlivanov <atomnuker@gmail.com>
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 #ifndef AVCODEC_OPUSENC_PSY_H
23 #define AVCODEC_OPUSENC_PSY_H
26 #include "opusenc_utils.h"
27 #include "libavfilter/window_func.h"
29 /* Each step is 2.5ms */
30 typedef struct OpusPsyStep {
31 int index; /* Current index */
33 float energy[OPUS_MAX_CHANNELS][CELT_MAX_BANDS]; /* Masking effects included */
34 float tone[OPUS_MAX_CHANNELS][CELT_MAX_BANDS]; /* Tonality */
35 float stereo[CELT_MAX_BANDS]; /* IS/MS compatibility */
36 float change_amp[OPUS_MAX_CHANNELS][CELT_MAX_BANDS]; /* Jump over last frame */
37 float total_change; /* Total change */
39 float *bands[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
40 float coeffs[OPUS_MAX_CHANNELS][OPUS_BLOCK_SIZE(CELT_BLOCK_960)];
43 typedef struct OpusBandExcitation {
45 float excitation_dist;
46 float excitation_init;
49 typedef struct PsyChain {
54 typedef struct OpusPsyContext {
55 AVCodecContext *avctx;
56 AVFloatDSPContext *dsp;
57 struct FFBufQueue *bufqueue;
58 OpusEncOptions *options;
63 OpusBandExcitation ex[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
64 FFBesselFilter bfilter_lo[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
65 FFBesselFilter bfilter_hi[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
67 OpusPsyStep *steps[FF_BUFQUEUE_SIZE + 1];
70 float *window[CELT_BLOCK_NB];
71 MDCT15Context *mdct[CELT_BLOCK_NB];
74 DECLARE_ALIGNED(32, float, scratch)[2048];
79 int64_t dual_stereo_used;
80 int64_t total_packets_out;
83 FFBesselFilter lambda_lp;
90 int *inflection_points;
91 int inflection_points_count;
94 int ff_opus_psy_process (OpusPsyContext *s, OpusPacketInfo *p);
95 void ff_opus_psy_celt_frame_init (OpusPsyContext *s, CeltFrame *f, int index);
96 int ff_opus_psy_celt_frame_process(OpusPsyContext *s, CeltFrame *f, int index);
97 void ff_opus_psy_postencode_update (OpusPsyContext *s, CeltFrame *f, OpusRangeCoder *rc);
99 int ff_opus_psy_init(OpusPsyContext *s, AVCodecContext *avctx,
100 struct FFBufQueue *bufqueue, OpusEncOptions *options);
101 void ff_opus_psy_signal_eof(OpusPsyContext *s);
102 int ff_opus_psy_end(OpusPsyContext *s);
104 #endif /* AVCODEC_OPUSENC_PSY_H */