comparison atrac1.c @ 10266:8695a5f212fc libavcodec

Factorize duplicated code in at1_imdct_block()
author vitor
date Thu, 24 Sep 2009 21:24:58 +0000
parents d884c0b7af2f
children ec0e313a00aa
comparison
equal deleted inserted replaced
10265:9acd9b34b2b9 10266:8695a5f212fc
106 { 106 {
107 int band_num, band_samples, log2_block_count, nbits, num_blocks, block_size; 107 int band_num, band_samples, log2_block_count, nbits, num_blocks, block_size;
108 unsigned int start_pos, ref_pos = 0, pos = 0; 108 unsigned int start_pos, ref_pos = 0, pos = 0;
109 109
110 for (band_num = 0; band_num < AT1_QMF_BANDS; band_num++) { 110 for (band_num = 0; band_num < AT1_QMF_BANDS; band_num++) {
111 float *prev_buf;
112 int j;
113
111 band_samples = samples_per_band[band_num]; 114 band_samples = samples_per_band[band_num];
112 log2_block_count = su->log2_block_count[band_num]; 115 log2_block_count = su->log2_block_count[band_num];
113 116
114 /* number of mdct blocks in the current QMF band: 1 - for long mode */ 117 /* number of mdct blocks in the current QMF band: 1 - for long mode */
115 /* 4 for short mode(low/middle bands) and 8 for short mode(high band)*/ 118 /* 4 for short mode(low/middle bands) and 8 for short mode(high band)*/
116 num_blocks = 1 << log2_block_count; 119 num_blocks = 1 << log2_block_count;
117 120
121 if (num_blocks == 1) {
118 /* mdct block size in samples: 128 (long mode, low & mid bands), */ 122 /* mdct block size in samples: 128 (long mode, low & mid bands), */
119 /* 256 (long mode, high band) and 32 (short mode, all bands) */ 123 /* 256 (long mode, high band) and 32 (short mode, all bands) */
120 block_size = band_samples >> log2_block_count; 124 block_size = band_samples >> log2_block_count;
121 125
122 /* calc transform size in bits according to the block_size_mode */ 126 /* calc transform size in bits according to the block_size_mode */
123 nbits = mdct_long_nbits[band_num] - log2_block_count; 127 nbits = mdct_long_nbits[band_num] - log2_block_count;
124 128
125 if (nbits != 5 && nbits != 7 && nbits != 8) 129 if (nbits != 5 && nbits != 7 && nbits != 8)
126 return -1; 130 return -1;
127
128 if (num_blocks == 1) {
129 /* long blocks */
130 at1_imdct(q, &q->spec[pos], &su->spectrum[0][ref_pos], nbits, band_num);
131 pos += block_size; // move to the next mdct block in the spectrum
132
133 /* overlap and window long blocks */
134 q->dsp.vector_fmul_window(q->bands[band_num], &su->spectrum[1][ref_pos + band_samples - 16],
135 &su->spectrum[0][ref_pos], ff_sine_32, 0, 16);
136 memcpy(q->bands[band_num] + 32, &su->spectrum[0][ref_pos + 16], 240 * sizeof(float));
137 } else { 131 } else {
138 /* short blocks */ 132 block_size = 32;
139 float *prev_buf; 133 nbits = 5;
134 }
135
140 start_pos = 0; 136 start_pos = 0;
141 prev_buf = &su->spectrum[1][ref_pos + band_samples - 16]; 137 prev_buf = &su->spectrum[1][ref_pos + band_samples - 16];
142 for (; num_blocks != 0; num_blocks--) { 138 for (j=0; j < num_blocks; j++) {
143 at1_imdct(q, &q->spec[pos], &su->spectrum[0][ref_pos + start_pos], 5, band_num); 139 at1_imdct(q, &q->spec[pos], &su->spectrum[0][ref_pos + start_pos], nbits, band_num);
144 140
145 /* overlap and window between short blocks */ 141 /* overlap and window */
146 q->dsp.vector_fmul_window(&q->bands[band_num][start_pos], prev_buf, 142 q->dsp.vector_fmul_window(&q->bands[band_num][start_pos], prev_buf,
147 &su->spectrum[0][ref_pos + start_pos], ff_sine_32, 0, 16); 143 &su->spectrum[0][ref_pos + start_pos], ff_sine_32, 0, 16);
148 144
149 prev_buf = &su->spectrum[0][ref_pos+start_pos + 16]; 145 prev_buf = &su->spectrum[0][ref_pos+start_pos + 16];
150 start_pos += 32; // use hardcoded block_size 146 start_pos += block_size;
151 pos += 32; 147 pos += block_size;
152 } 148 }
153 } 149
150 if (num_blocks == 1)
151 memcpy(q->bands[band_num] + 32, &su->spectrum[0][ref_pos + 16], 240 * sizeof(float));
152
154 ref_pos += band_samples; 153 ref_pos += band_samples;
155 } 154 }
156 155
157 /* Swap buffers so the mdct overlap works */ 156 /* Swap buffers so the mdct overlap works */
158 FFSWAP(float*, su->spectrum[0], su->spectrum[1]); 157 FFSWAP(float*, su->spectrum[0], su->spectrum[1]);