FFmpeg
atrac.c
Go to the documentation of this file.
1 /*
2  * common functions for the ATRAC family of decoders
3  *
4  * Copyright (c) 2006-2013 Maxim Poliakovski
5  * Copyright (c) 2006-2008 Benjamin Larsson
6  *
7  * This file is part of FFmpeg.
8  *
9  * FFmpeg is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * FFmpeg is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with FFmpeg; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22  */
23 
24 /**
25  * @file
26  */
27 
28 #include <math.h>
29 #include <stddef.h>
30 #include <stdio.h>
31 #include <string.h>
32 
33 #include "libavutil/thread.h"
34 
35 #include "avcodec.h"
36 #include "atrac.h"
37 
39 static float qmf_window[48];
40 
41 static const float qmf_48tap_half[24] = {
42  -0.00001461907, -0.00009205479,-0.000056157569,0.00030117269,
43  0.0002422519, -0.00085293897,-0.0005205574, 0.0020340169,
44  0.00078333891, -0.0042153862, -0.00075614988, 0.0078402944,
45  -0.000061169922,-0.01344162, 0.0024626821, 0.021736089,
46  -0.007801671, -0.034090221, 0.01880949, 0.054326009,
47  -0.043596379, -0.099384367, 0.13207909, 0.46424159
48 };
49 
50 static av_cold void atrac_generate_tables(void)
51 {
52  /* Generate scale factors */
53  for (int i = 0; i < 64; i++)
54  ff_atrac_sf_table[i] = pow(2.0, (i - 15) / 3.0);
55 
56  /* Generate the QMF window. */
57  for (int i = 0; i < 24; i++) {
58  float s = qmf_48tap_half[i] * 2.0;
59  qmf_window[i] = qmf_window[47 - i] = s;
60  }
61 }
62 
64 {
65  static AVOnce init_static_once = AV_ONCE_INIT;
66  ff_thread_once(&init_static_once, atrac_generate_tables);
67 }
68 
70  int loc_scale)
71 {
72  int i;
73 
74  gctx->loc_scale = loc_scale;
75  gctx->loc_size = 1 << loc_scale;
76  gctx->id2exp_offset = id2exp_offset;
77 
78  /* Generate gain level table. */
79  for (i = 0; i < 16; i++)
80  gctx->gain_tab1[i] = powf(2.0, id2exp_offset - i);
81 
82  /* Generate gain interpolation table. */
83  for (i = -15; i < 16; i++)
84  gctx->gain_tab2[i + 15] = powf(2.0, -1.0f / gctx->loc_size * i);
85 }
86 
87 void ff_atrac_gain_compensation(AtracGCContext *gctx, float *in, float *prev,
88  AtracGainInfo *gc_now, AtracGainInfo *gc_next,
89  int num_samples, float *out)
90 {
91  float lev, gc_scale, gain_inc;
92  int i, pos, lastpos;
93 
94  gc_scale = gc_next->num_points ? gctx->gain_tab1[gc_next->lev_code[0]]
95  : 1.0f;
96 
97  if (!gc_now->num_points) {
98  for (pos = 0; pos < num_samples; pos++)
99  out[pos] = in[pos] * gc_scale + prev[pos];
100  } else {
101  pos = 0;
102 
103  for (i = 0; i < gc_now->num_points; i++) {
104  lastpos = gc_now->loc_code[i] << gctx->loc_scale;
105 
106  lev = gctx->gain_tab1[gc_now->lev_code[i]];
107  gain_inc = gctx->gain_tab2[(i + 1 < gc_now->num_points ? gc_now->lev_code[i + 1]
108  : gctx->id2exp_offset) -
109  gc_now->lev_code[i] + 15];
110 
111  /* apply constant gain level and overlap */
112  for (; pos < lastpos; pos++)
113  out[pos] = (in[pos] * gc_scale + prev[pos]) * lev;
114 
115  /* interpolate between two different gain levels */
116  for (; pos < lastpos + gctx->loc_size; pos++) {
117  out[pos] = (in[pos] * gc_scale + prev[pos]) * lev;
118  lev *= gain_inc;
119  }
120  }
121 
122  for (; pos < num_samples; pos++)
123  out[pos] = in[pos] * gc_scale + prev[pos];
124  }
125 
126  /* copy the overlapping part into the delay buffer */
127  memcpy(prev, &in[num_samples], num_samples * sizeof(float));
128 }
129 
130 void ff_atrac_iqmf(float *inlo, float *inhi, unsigned int nIn, float *pOut,
131  float *delayBuf, float *temp)
132 {
133  int i, j;
134  float *p1, *p3;
135 
136  memcpy(temp, delayBuf, 46*sizeof(float));
137 
138  p3 = temp + 46;
139 
140  /* loop1 */
141  for(i=0; i<nIn; i+=2){
142  p3[2*i+0] = inlo[i ] + inhi[i ];
143  p3[2*i+1] = inlo[i ] - inhi[i ];
144  p3[2*i+2] = inlo[i+1] + inhi[i+1];
145  p3[2*i+3] = inlo[i+1] - inhi[i+1];
146  }
147 
148  /* loop2 */
149  p1 = temp;
150  for (j = nIn; j != 0; j--) {
151  float s1 = 0.0;
152  float s2 = 0.0;
153 
154  for (i = 0; i < 48; i += 2) {
155  s1 += p1[i] * qmf_window[i];
156  s2 += p1[i+1] * qmf_window[i+1];
157  }
158 
159  pOut[0] = s2;
160  pOut[1] = s1;
161 
162  p1 += 2;
163  pOut += 2;
164  }
165 
166  /* Update the delay buffer. */
167  memcpy(delayBuf, temp + nIn*2, 46*sizeof(float));
168 }
out
FILE * out
Definition: movenc.c:54
AtracGCContext::gain_tab2
float gain_tab2[31]
gain compensation interpolation table
Definition: atrac.h:46
thread.h
qmf_window
static float qmf_window[48]
Definition: atrac.c:39
ff_atrac_init_gain_compensation
av_cold void ff_atrac_init_gain_compensation(AtracGCContext *gctx, int id2exp_offset, int loc_scale)
Initialize gain compensation context.
Definition: atrac.c:69
ff_thread_once
static int ff_thread_once(char *control, void(*routine)(void))
Definition: thread.h:175
av_cold
#define av_cold
Definition: attributes.h:90
s
#define s(width, name)
Definition: cbs_vp9.c:257
atrac.h
s1
#define s1
Definition: regdef.h:38
AtracGainInfo::num_points
int num_points
number of gain control points
Definition: atrac.h:36
AtracGCContext
Gain compensation context structure.
Definition: atrac.h:44
f
#define f(width, name)
Definition: cbs_vp9.c:255
AV_ONCE_INIT
#define AV_ONCE_INIT
Definition: thread.h:173
ff_atrac_sf_table
float ff_atrac_sf_table[64]
Definition: atrac.c:38
AtracGainInfo
Gain control parameters for one subband.
Definition: atrac.h:35
AtracGCContext::loc_scale
int loc_scale
scale of location code = 2^loc_scale samples
Definition: atrac.h:48
AVOnce
#define AVOnce
Definition: thread.h:172
for
for(j=16;j >0;--j)
Definition: h264pred_template.c:469
s2
#define s2
Definition: regdef.h:39
powf
#define powf(x, y)
Definition: libm.h:50
AtracGainInfo::loc_code
int loc_code[7]
location of gain control points
Definition: atrac.h:38
ff_atrac_generate_tables
av_cold void ff_atrac_generate_tables(void)
Generate common tables.
Definition: atrac.c:63
in
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) #define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac) { } void ff_audio_convert_free(AudioConvert **ac) { if(! *ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);} AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map) { AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method !=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2) { ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc) { av_free(ac);return NULL;} return ac;} in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar) { ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar ? ac->channels :1;} else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;} int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in) { int use_generic=1;int len=in->nb_samples;int p;if(ac->dc) { av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> in
Definition: audio_convert.c:326
AtracGCContext::loc_size
int loc_size
size of location code in samples
Definition: atrac.h:49
i
int i
Definition: input.c:407
AtracGCContext::gain_tab1
float gain_tab1[16]
gain compensation level table
Definition: atrac.h:45
lev
static LevelCodes lev[4+3+3]
Definition: clearvideo.c:85
avcodec.h
atrac_generate_tables
static av_cold void atrac_generate_tables(void)
Definition: atrac.c:50
pos
unsigned int pos
Definition: spdifenc.c:412
AtracGCContext::id2exp_offset
int id2exp_offset
offset for converting level index into level exponent
Definition: atrac.h:47
AtracGainInfo::lev_code
int lev_code[7]
level at corresponding control point
Definition: atrac.h:37
ff_atrac_gain_compensation
void ff_atrac_gain_compensation(AtracGCContext *gctx, float *in, float *prev, AtracGainInfo *gc_now, AtracGainInfo *gc_next, int num_samples, float *out)
Apply gain compensation and perform the MDCT overlapping part.
Definition: atrac.c:87
temp
else temp
Definition: vf_mcdeint.c:259
qmf_48tap_half
static const float qmf_48tap_half[24]
Definition: atrac.c:41
ff_atrac_iqmf
void ff_atrac_iqmf(float *inlo, float *inhi, unsigned int nIn, float *pOut, float *delayBuf, float *temp)
Quadrature mirror synthesis filter.
Definition: atrac.c:130