git.sesse.net Git - ffmpeg/blob - libavcodec/opusenc_psy.h

   1 /*
   2  * Opus encoder
   3  * Copyright (c) 2017 Rostislav Pehlivanov <atomnuker@gmail.com>
   4  *
   5  * This file is part of FFmpeg.
   6  *
   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.
  11  *
  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.
  16  *
  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
  20  */
  21
  22 #ifndef AVCODEC_OPUSENC_PSY_H
  23 #define AVCODEC_OPUSENC_PSY_H
  24
  25 #include "libavutil/mem_internal.h"
  26
  27 #include "opusenc.h"
  28 #include "opusenc_utils.h"
  29 #include "libavfilter/window_func.h"
  30
  31 /* Each step is 2.5ms */
  32 typedef struct OpusPsyStep {
  33     int   index; /* Current index */
  34     int   silence;
  35     float energy[OPUS_MAX_CHANNELS][CELT_MAX_BANDS]; /* Masking effects included */
  36     float tone[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];   /* Tonality */
  37     float stereo[CELT_MAX_BANDS];                    /* IS/MS compatibility */
  38     float change_amp[OPUS_MAX_CHANNELS][CELT_MAX_BANDS]; /* Jump over last frame */
  39     float total_change; /* Total change */
  40
  41     float *bands[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
  42     float coeffs[OPUS_MAX_CHANNELS][OPUS_BLOCK_SIZE(CELT_BLOCK_960)];
  43 } OpusPsyStep;
  44
  45 typedef struct OpusBandExcitation {
  46     float excitation;
  47     float excitation_dist;
  48     float excitation_init;
  49 } OpusBandExcitation;
  50
  51 typedef struct PsyChain {
  52     int start;
  53     int end;
  54 } PsyChain;
  55
  56 typedef struct OpusPsyContext {
  57     AVCodecContext *avctx;
  58     AVFloatDSPContext *dsp;
  59     struct FFBufQueue *bufqueue;
  60     OpusEncOptions *options;
  61
  62     PsyChain cs[128];
  63     int cs_num;
  64
  65     OpusBandExcitation ex[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
  66     FFBesselFilter bfilter_lo[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
  67     FFBesselFilter bfilter_hi[OPUS_MAX_CHANNELS][CELT_MAX_BANDS];
  68
  69     OpusPsyStep *steps[FF_BUFQUEUE_SIZE + 1];
  70     int max_steps;
  71
  72     float *window[CELT_BLOCK_NB];
  73     MDCT15Context *mdct[CELT_BLOCK_NB];
  74     int bsize_analysis;
  75
  76     DECLARE_ALIGNED(32, float, scratch)[2048];
  77
  78     /* Stats */
  79     float rc_waste;
  80     float avg_is_band;
  81     int64_t dual_stereo_used;
  82     int64_t total_packets_out;
  83
  84     /* State */
  85     FFBesselFilter lambda_lp;
  86     OpusPacketInfo p;
  87     int redo_analysis;
  88     int buffered_steps;
  89     int steps_to_process;
  90     int eof;
  91     float lambda;
  92     int *inflection_points;
  93     int inflection_points_count;
  94 } OpusPsyContext;
  95
  96 int  ff_opus_psy_process           (OpusPsyContext *s, OpusPacketInfo *p);
  97 void ff_opus_psy_celt_frame_init   (OpusPsyContext *s, CeltFrame *f, int index);
  98 int  ff_opus_psy_celt_frame_process(OpusPsyContext *s, CeltFrame *f, int index);
  99 void ff_opus_psy_postencode_update (OpusPsyContext *s, CeltFrame *f, OpusRangeCoder *rc);
 100
 101 int  ff_opus_psy_init(OpusPsyContext *s, AVCodecContext *avctx,
 102                       struct FFBufQueue *bufqueue, OpusEncOptions *options);
 103 void ff_opus_psy_signal_eof(OpusPsyContext *s);
 104 int  ff_opus_psy_end(OpusPsyContext *s);
 105
 106 #endif /* AVCODEC_OPUSENC_PSY_H */