107 lines
3.3 KiB
C
107 lines
3.3 KiB
C
/*
|
|
* Opus encoder
|
|
* Copyright (c) 2017 Rostislav Pehlivanov <atomnuker@gmail.com>
|
|
*
|
|
* 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_OPUSENC_PSY_H
|
|
#define AVCODEC_OPUSENC_PSY_H
|
|
|
|
#include "libavutil/mem_internal.h"
|
|
|
|
#include "opusenc.h"
|
|
#include "opusenc_utils.h"
|
|
#include "libavfilter/window_func.h"
|
|
|
|
/* Each step is 2.5ms */
|
|
typedef struct OpusPsyStep {
|
|
int index; /* Current index */
|
|
int silence;
|
|
float energy[OPUS_MAX_CHANNELS][CELT_MAX_BANDS]; /* Masking effects included */
|
|
float tone[OPUS_MAX_CHANNELS][CELT_MAX_BANDS]; /* Tonality */
|
|
float stereo[CELT_MAX_BANDS]; /* IS/MS compatibility */
|
|
float change_amp[OPUS_MAX_CHANNELS][CELT_MAX_BANDS]; /* Jump over last frame */
|
|
float total_change; /* Total change */
|
|
|
|
float *bands[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
|
|
float coeffs[OPUS_MAX_CHANNELS][OPUS_BLOCK_SIZE(CELT_BLOCK_960)];
|
|
} OpusPsyStep;
|
|
|
|
typedef struct OpusBandExcitation {
|
|
float excitation;
|
|
float excitation_dist;
|
|
float excitation_init;
|
|
} OpusBandExcitation;
|
|
|
|
typedef struct PsyChain {
|
|
int start;
|
|
int end;
|
|
} PsyChain;
|
|
|
|
typedef struct OpusPsyContext {
|
|
AVCodecContext *avctx;
|
|
AVFloatDSPContext *dsp;
|
|
struct FFBufQueue *bufqueue;
|
|
OpusEncOptions *options;
|
|
|
|
PsyChain cs[128];
|
|
int cs_num;
|
|
|
|
OpusBandExcitation ex[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
|
|
FFBesselFilter bfilter_lo[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
|
|
FFBesselFilter bfilter_hi[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
|
|
|
|
OpusPsyStep *steps[FF_BUFQUEUE_SIZE + 1];
|
|
int max_steps;
|
|
|
|
float *window[CELT_BLOCK_NB];
|
|
MDCT15Context *mdct[CELT_BLOCK_NB];
|
|
int bsize_analysis;
|
|
|
|
DECLARE_ALIGNED(32, float, scratch)[2048];
|
|
|
|
/* Stats */
|
|
float rc_waste;
|
|
float avg_is_band;
|
|
int64_t dual_stereo_used;
|
|
int64_t total_packets_out;
|
|
|
|
/* State */
|
|
FFBesselFilter lambda_lp;
|
|
OpusPacketInfo p;
|
|
int redo_analysis;
|
|
int buffered_steps;
|
|
int steps_to_process;
|
|
int eof;
|
|
float lambda;
|
|
int *inflection_points;
|
|
int inflection_points_count;
|
|
} OpusPsyContext;
|
|
|
|
int ff_opus_psy_process (OpusPsyContext *s, OpusPacketInfo *p);
|
|
void ff_opus_psy_celt_frame_init (OpusPsyContext *s, CeltFrame *f, int index);
|
|
int ff_opus_psy_celt_frame_process(OpusPsyContext *s, CeltFrame *f, int index);
|
|
void ff_opus_psy_postencode_update (OpusPsyContext *s, CeltFrame *f, OpusRangeCoder *rc);
|
|
|
|
int ff_opus_psy_init(OpusPsyContext *s, AVCodecContext *avctx,
|
|
struct FFBufQueue *bufqueue, OpusEncOptions *options);
|
|
void ff_opus_psy_signal_eof(OpusPsyContext *s);
|
|
int ff_opus_psy_end(OpusPsyContext *s);
|
|
|
|
#endif /* AVCODEC_OPUSENC_PSY_H */
|