58 #define MONO 0x1000001
59 #define STEREO 0x1000002
60 #define JOINT_STEREO 0x1000003
61 #define MC_COOK 0x2000000 // multichannel Cook, not supported
63 #define SUBBAND_SIZE 20
64 #define MAX_SUBPACKETS 5
90 float mono_previous_buffer1[1024];
91 float mono_previous_buffer2[1024];
107 int *subband_coef_index,
int *subband_coef_sign,
114 float *decode_buffer,
115 float *mlt_buffer1,
float *mlt_buffer2);
118 cook_gains *gains_ptr,
float *previous_buffer);
121 int gain_index,
int gain_index_next);
172 for (i = -63; i < 64; i++) {
183 for (i = 0; i < 23; i++)
194 for (i = 0; i < 13; i++) {
200 for (i = 0; i < 7; i++) {
229 for (j = 0; j < mlt_size; j++)
238 av_log2(mlt_size) + 1);
252 for (i = 0; i < 5; i++)
258 #define DECODE_BYTES_PAD1(bytes) (3 - ((bytes) + 3) % 4)
259 #define DECODE_BYTES_PAD2(bytes) ((bytes) % 4 + DECODE_BYTES_PAD1(2 * (bytes)))
281 static inline int decode_bytes(
const uint8_t *inbuffer, uint8_t *out,
int bytes)
283 static const uint32_t
tab[4] = {
290 uint32_t *obuf = (uint32_t *) out;
297 off = (intptr_t) inbuffer & 3;
298 buf = (
const uint32_t *) (inbuffer - off);
301 for (i = 0; i < bytes / 4; i++)
302 obuf[i] = c ^ buf[i];
324 for (i = 0; i < 13; i++)
326 for (i = 0; i < 7; i++)
358 gaininfo[i++] = gain;
371 int *quant_index_table)
375 quant_index_table[0] =
get_bits(&q->
gb, 6) - 6;
391 quant_index_table[i] = quant_index_table[i - 1] + j - 12;
392 if (quant_index_table[i] > 63 || quant_index_table[i] < -63) {
394 "Invalid quantizer %d at position %d, outside [-63, 63] range\n",
395 quant_index_table[i], i);
412 int *category,
int *category_index)
414 int exp_idx, bias, tmpbias1, tmpbias2, bits_left, num_bits,
index,
v, i, j;
418 int tmp_categorize_array[128 * 2];
430 memset(&exp_index1, 0,
sizeof(exp_index1));
431 memset(&exp_index2, 0,
sizeof(exp_index2));
432 memset(&tmp_categorize_array, 0,
sizeof(tmp_categorize_array));
437 for (i = 32; i > 0; i = i / 2) {
441 exp_idx = av_clip((i - quant_index_table[index] + bias) / 2, 0, 7);
445 if (num_bits >= bits_left - 32)
452 exp_idx = av_clip((bias - quant_index_table[i]) / 2, 0, 7);
454 exp_index1[i] = exp_idx;
455 exp_index2[i] = exp_idx;
457 tmpbias1 = tmpbias2 = num_bits;
460 if (tmpbias1 + tmpbias2 > 2 * bits_left) {
464 if (exp_index1[i] < 7) {
465 v = (-2 * exp_index1[i]) - quant_index_table[i] + bias;
474 tmp_categorize_array[tmp_categorize_array1_idx++] =
index;
482 if (exp_index2[i] > 0) {
483 v = (-2 * exp_index2[i]) - quant_index_table[i] + bias;
492 tmp_categorize_array[--tmp_categorize_array2_idx] =
index;
500 category[i] = exp_index2[i];
503 category_index[i] = tmp_categorize_array[tmp_categorize_array2_idx++];
520 int idx = category_index[i];
537 int *subband_coef_index,
int *subband_coef_sign,
544 if (subband_coef_index[i]) {
546 if (subband_coef_sign[i])
566 int *subband_coef_index,
int *subband_coef_sign)
569 int vlc, vd, tmp, result;
573 for (i = 0; i <
vpr_tab[category]; i++) {
579 for (j = vd - 1; j >= 0; j--) {
581 subband_coef_index[vd * i + j] = vlc - tmp * (
kmax_tab[category] + 1);
584 for (j = 0; j < vd; j++) {
585 if (subband_coef_index[i * vd + j]) {
590 subband_coef_sign[i * vd + j] = 0;
593 subband_coef_sign[i * vd + j] = 0;
610 int *quant_index_table,
float *mlt_buffer)
622 index = category[band];
623 if (category[band] < 7) {
624 if (
unpack_SQVH(q, p, category[band], subband_coef_index, subband_coef_sign)) {
627 category[band + j] = 7;
631 memset(subband_coef_index, 0,
sizeof(subband_coef_index));
632 memset(subband_coef_sign, 0,
sizeof(subband_coef_sign));
635 subband_coef_index, subband_coef_sign,
653 int category_index[128];
654 int quant_index_table[102];
658 memset(&category, 0,
sizeof(category));
659 memset(&category_index, 0,
sizeof(category_index));
664 categorize(q, p, quant_index_table, category, category_index);
681 int gain_index,
int gain_index_next)
685 fc1 =
pow2tab[gain_index + 63];
687 if (gain_index == gain_index_next) {
691 fc2 = q->
gain_table[11 + (gain_index_next - gain_index)];
708 cook_gains *gains_ptr,
float *previous_buffer)
720 inbuffer[i] = inbuffer[i] * fc * q->
mlt_window[i] -
736 cook_gains *gains_ptr,
float *previous_buffer)
745 q->
imlt_window(q, buffer1, gains_ptr, previous_buffer);
748 for (i = 0; i < 8; i++)
749 if (gains_ptr->
now[i] || gains_ptr->
now[i + 1])
751 gains_ptr->
now[i], gains_ptr->
now[i + 1]);
754 memcpy(previous_buffer, buffer0,
772 int length = end - start + 1;
778 for (i = 0; i < length; i++)
781 for (i = 0; i < length; i++)
800 float *decode_buffer,
801 float *mlt_buffer1,
float *mlt_buffer2)
806 mlt_buffer1[SUBBAND_SIZE * subband + j] = f1 * decode_buffer[tmp_idx];
807 mlt_buffer2[SUBBAND_SIZE * subband + j] = f2 * decode_buffer[tmp_idx];
826 const float *cplscale;
828 memset(decouple_tab, 0,
sizeof(decouple_tab));
832 memset(mlt_buffer1, 0, 1024 *
sizeof(*mlt_buffer1));
833 memset(mlt_buffer2, 0, 1024 *
sizeof(*mlt_buffer2));
841 mlt_buffer1[i * 20 + j] = decode_buffer[i * 40 + j];
842 mlt_buffer2[i * 20 + j] = decode_buffer[i * 40 + 20 + j];
851 idx -= decouple_tab[cpl_tmp];
853 f1 = cplscale[decouple_tab[cpl_tmp] + 1];
855 q->
decouple(q, p, i, f1, f2, decode_buffer, mlt_buffer1, mlt_buffer2);
871 const uint8_t *inbuffer,
898 out[chan + q->
nb_channels * j] = av_clipf(output[j], -1.0, 1.0);
915 cook_gains *gains_ptr,
float *previous_buffer,
916 float *out,
int chan)
918 imlt_gain(q, decode_buffer, gains_ptr, previous_buffer);
933 const uint8_t *inbuffer,
float *outbuffer)
935 int sub_packet_size = p->
size;
979 int *got_frame_ptr,
AVPacket *avpkt)
981 const uint8_t *buf = avpkt->
data;
982 int buf_size = avpkt->
size;
989 if (buf_size < avctx->block_align)
1010 "frame subpacket size total > avctx->block_align!\n");
1021 "subpacket[%i] size %i js %i %i block_align %i\n",
1050 #define PRINT(a, b) av_log(q->avctx, AV_LOG_ERROR, " %s = %d\n", a, b);
1075 for (i = 0; i < 32; i++)
1076 if (mask & (1 << i))
1089 const uint8_t *edata_ptr = avctx->
extradata;
1090 const uint8_t *edata_ptr_end = edata_ptr + avctx->
extradata_size;
1093 unsigned int channel_mask = 0;
1098 if (extradata_size <= 0) {
1116 while (edata_ptr < edata_ptr_end) {
1119 if (extradata_size >= 8) {
1123 extradata_size -= 8;
1125 if (extradata_size >= 8) {
1126 bytestream_get_be32(&edata_ptr);
1129 extradata_size -= 8;
1182 if (extradata_size >= 4)
1286 "unknown amount of samples_per_channel = %d\n",
1301 dump_cook_context(q);