static int encode_init(AVCodecContext *avctx) { DCAContext *c = avctx->priv_data; uint64_t layout = avctx->channel_layout; int i, min_frame_bits; c->fullband_channels = c->channels = avctx->channels; c->lfe_channel = (avctx->channels == 3 || avctx->channels == 6); c->band_interpolation = band_interpolation[1]; c->band_spectrum = band_spectrum[1]; c->worst_quantization_noise = -2047; c->worst_noise_ever = -2047; if (!layout) { av_log(avctx, AV_LOG_WARNING, "No channel layout specified. The " "encoder will guess the layout, but it " "might be incorrect.\n"); layout = av_get_default_channel_layout(avctx->channels); } switch (layout) { case AV_CH_LAYOUT_MONO: c->channel_config = 0; break; case AV_CH_LAYOUT_STEREO: c->channel_config = 2; break; case AV_CH_LAYOUT_2_2: c->channel_config = 8; break; case AV_CH_LAYOUT_5POINT0: c->channel_config = 9; break; case AV_CH_LAYOUT_5POINT1: c->channel_config = 9; break; default: av_log(avctx, AV_LOG_ERROR, "Unsupported channel layout!\n"); return AVERROR_PATCHWELCOME; } if (c->lfe_channel) c->fullband_channels--; for (i = 0; i < 9; i++) { if (sample_rates[i] == avctx->sample_rate) break; } if (i == 9) return AVERROR(EINVAL); c->samplerate_index = i; if (avctx->bit_rate < 32000 || avctx->bit_rate > 3840000) { av_log(avctx, AV_LOG_ERROR, "Bit rate %i not supported.", avctx->bit_rate); return AVERROR(EINVAL); } for (i = 0; dca_bit_rates[i] < avctx->bit_rate; i++) ; c->bitrate_index = i; avctx->bit_rate = dca_bit_rates[i]; c->frame_bits = FFALIGN((avctx->bit_rate * 512 + avctx->sample_rate - 1) / avctx->sample_rate, 32); min_frame_bits = 132 + (493 + 28 * 32) * c->fullband_channels + c->lfe_channel * 72; if (c->frame_bits < min_frame_bits || c->frame_bits > (DCA_MAX_FRAME_SIZE << 3)) return AVERROR(EINVAL); c->frame_size = (c->frame_bits + 7) / 8; avctx->frame_size = 32 * SUBBAND_SAMPLES; if (!cos_table[0]) { int j, k; for (i = 0; i < 2048; i++) { cos_table[i] = (int32_t)(0x7fffffff * cos(M_PI * i / 1024)); cb_to_level[i] = (int32_t)(0x7fffffff * pow(10, -0.005 * i)); } /* FIXME: probably incorrect */ for (i = 0; i < 256; i++) { lfe_fir_64i[i] = (int32_t)(0x01ffffff * lfe_fir_64[i]); lfe_fir_64i[511 - i] = (int32_t)(0x01ffffff * lfe_fir_64[i]); } for (i = 0; i < 512; i++) { band_interpolation[0][i] = (int32_t)(ULLN(0x1000000000) * fir_32bands_perfect[i]); band_interpolation[1][i] = (int32_t)(ULLN(0x1000000000) * fir_32bands_nonperfect[i]); } for (i = 0; i < 9; i++) { for (j = 0; j < AUBANDS; j++) { for (k = 0; k < 256; k++) { double freq = sample_rates[i] * (k + 0.5) / 512; auf[i][j][k] = (int32_t)(10 * (hom(freq) + gammafilter(j, freq))); } } } for (i = 0; i < 256; i++) { double add = 1 + pow(10, -0.01 * i); cb_to_add[i] = (int32_t)(100 * log10(add)); } for (j = 0; j < 8; j++) { double accum = 0; for (i = 0; i < 512; i++) { double reconst = fir_32bands_perfect[i] * ((i & 64) ? (-1) : 1); accum += reconst * cos(2 * M_PI * (i + 0.5 - 256) * (j + 0.5) / 512); } band_spectrum[0][j] = (int32_t)(200 * log10(accum)); } for (j = 0; j < 8; j++) { double accum = 0; for (i = 0; i < 512; i++) { double reconst = fir_32bands_nonperfect[i] * ((i & 64) ? (-1) : 1); accum += reconst * cos(2 * M_PI * (i + 0.5 - 256) * (j + 0.5) / 512); } band_spectrum[1][j] = (int32_t)(200 * log10(accum)); } } return 0; }
static int encode_init(AVCodecContext *avctx) { DCAEncContext *c = avctx->priv_data; uint64_t layout = avctx->channel_layout; int i, j, min_frame_bits; c->fullband_channels = c->channels = avctx->channels; c->lfe_channel = (avctx->channels == 3 || avctx->channels == 6); c->band_interpolation = band_interpolation[1]; c->band_spectrum = band_spectrum[1]; c->worst_quantization_noise = -2047; c->worst_noise_ever = -2047; if (!layout) { av_log(avctx, AV_LOG_WARNING, "No channel layout specified. The " "encoder will guess the layout, but it " "might be incorrect.\n"); layout = av_get_default_channel_layout(avctx->channels); } switch (layout) { case AV_CH_LAYOUT_MONO: c->channel_config = 0; break; case AV_CH_LAYOUT_STEREO: c->channel_config = 2; break; case AV_CH_LAYOUT_2_2: c->channel_config = 8; break; case AV_CH_LAYOUT_5POINT0: c->channel_config = 9; break; case AV_CH_LAYOUT_5POINT1: c->channel_config = 9; break; default: av_log(avctx, AV_LOG_ERROR, "Unsupported channel layout!\n"); return AVERROR_PATCHWELCOME; } if (c->lfe_channel) { c->fullband_channels--; c->channel_order_tab = channel_reorder_lfe[c->channel_config]; } else { c->channel_order_tab = channel_reorder_nolfe[c->channel_config]; } for (i = 0; i < MAX_CHANNELS; i++) { for (j = 0; j < DCA_CODE_BOOKS; j++) { c->quant_index_sel[i][j] = ff_dca_quant_index_group_size[j]; } /* 6 - no Huffman */ c->bit_allocation_sel[i] = 6; } for (i = 0; i < 9; i++) { if (sample_rates[i] == avctx->sample_rate) break; } if (i == 9) return AVERROR(EINVAL); c->samplerate_index = i; if (avctx->bit_rate < 32000 || avctx->bit_rate > 3840000) { av_log(avctx, AV_LOG_ERROR, "Bit rate %"PRId64" not supported.", (int64_t)avctx->bit_rate); return AVERROR(EINVAL); } for (i = 0; ff_dca_bit_rates[i] < avctx->bit_rate; i++) ; c->bitrate_index = i; c->frame_bits = FFALIGN((avctx->bit_rate * 512 + avctx->sample_rate - 1) / avctx->sample_rate, 32); min_frame_bits = 132 + (493 + 28 * 32) * c->fullband_channels + c->lfe_channel * 72; if (c->frame_bits < min_frame_bits || c->frame_bits > (DCA_MAX_FRAME_SIZE << 3)) return AVERROR(EINVAL); c->frame_size = (c->frame_bits + 7) / 8; avctx->frame_size = 32 * SUBBAND_SAMPLES; if (!cos_table[0]) { int j, k; cos_table[0] = 0x7fffffff; cos_table[512] = 0; cos_table[1024] = -cos_table[0]; for (i = 1; i < 512; i++) { cos_table[i] = (int32_t)(0x7fffffff * cos(M_PI * i / 1024)); cos_table[1024-i] = -cos_table[i]; cos_table[1024+i] = -cos_table[i]; cos_table[2048-i] = cos_table[i]; } for (i = 0; i < 2048; i++) { cb_to_level[i] = (int32_t)(0x7fffffff * ff_exp10(-0.005 * i)); } for (k = 0; k < 32; k++) { for (j = 0; j < 8; j++) { lfe_fir_64i[64 * j + k] = (int32_t)(0xffffff800000ULL * ff_dca_lfe_fir_64[8 * k + j]); lfe_fir_64i[64 * (7-j) + (63 - k)] = (int32_t)(0xffffff800000ULL * ff_dca_lfe_fir_64[8 * k + j]); } } for (i = 0; i < 512; i++) { band_interpolation[0][i] = (int32_t)(0x1000000000ULL * ff_dca_fir_32bands_perfect[i]); band_interpolation[1][i] = (int32_t)(0x1000000000ULL * ff_dca_fir_32bands_nonperfect[i]); } for (i = 0; i < 9; i++) { for (j = 0; j < AUBANDS; j++) { for (k = 0; k < 256; k++) { double freq = sample_rates[i] * (k + 0.5) / 512; auf[i][j][k] = (int32_t)(10 * (hom(freq) + gammafilter(j, freq))); } } } for (i = 0; i < 256; i++) { double add = 1 + ff_exp10(-0.01 * i); cb_to_add[i] = (int32_t)(100 * log10(add)); } for (j = 0; j < 8; j++) { double accum = 0; for (i = 0; i < 512; i++) { double reconst = ff_dca_fir_32bands_perfect[i] * ((i & 64) ? (-1) : 1); accum += reconst * cos(2 * M_PI * (i + 0.5 - 256) * (j + 0.5) / 512); } band_spectrum[0][j] = (int32_t)(200 * log10(accum)); } for (j = 0; j < 8; j++) { double accum = 0; for (i = 0; i < 512; i++) { double reconst = ff_dca_fir_32bands_nonperfect[i] * ((i & 64) ? (-1) : 1); accum += reconst * cos(2 * M_PI * (i + 0.5 - 256) * (j + 0.5) / 512); } band_spectrum[1][j] = (int32_t)(200 * log10(accum)); } } return 0; }