1 /* Copyright (c) 2007-2008 CSIRO 2 Copyright (c) 2007-2009 Xiph.Org Foundation 3 Copyright (c) 2008 Gregory Maxwell 4 Written by Jean-Marc Valin and Gregory Maxwell */ 5 /* 6 Redistribution and use in source and binary forms, with or without 7 modification, are permitted provided that the following conditions 8 are met: 9 10 - Redistributions of source code must retain the above copyright 11 notice, this list of conditions and the following disclaimer. 12 13 - Redistributions in binary form must reproduce the above copyright 14 notice, this list of conditions and the following disclaimer in the 15 documentation and/or other materials provided with the distribution. 16 17 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20 A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 21 OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 22 EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 23 PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 24 PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 25 LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 26 NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 27 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 */ 29 30 #ifdef HAVE_CONFIG_H 31 #include "config.h" 32 #endif 33 34 #include "celt.h" 35 #include "modes.h" 36 #include "rate.h" 37 #include "os_support.h" 38 #include "stack_alloc.h" 39 #include "quant_bands.h" 40 #include "cpu_support.h" 41 42 static const opus_int16 eband5ms[] = { 43 /*0 200 400 600 800 1k 1.2 1.4 1.6 2k 2.4 2.8 3.2 4k 4.8 5.6 6.8 8k 9.6 12k 15.6 */ 44 0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 12, 14, 16, 20, 24, 28, 34, 40, 48, 60, 78, 100 45 }; 46 47 /* Alternate tuning (partially derived from Vorbis) */ 48 #define BITALLOC_SIZE 11 49 /* Bit allocation table in units of 1/32 bit/sample (0.1875 dB SNR) */ 50 static const unsigned char band_allocation[] = { 51 /*0 200 400 600 800 1k 1.2 1.4 1.6 2k 2.4 2.8 3.2 4k 4.8 5.6 6.8 8k 9.6 12k 15.6 */ 52 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 53 90, 80, 75, 69, 63, 56, 49, 40, 34, 29, 20, 18, 10, 0, 0, 0, 0, 0, 0, 0, 0, 54 110,100, 90, 84, 78, 71, 65, 58, 51, 45, 39, 32, 26, 20, 12, 0, 0, 0, 0, 0, 0, 55 118,110,103, 93, 86, 80, 75, 70, 65, 59, 53, 47, 40, 31, 23, 15, 4, 0, 0, 0, 0, 56 126,119,112,104, 95, 89, 83, 78, 72, 66, 60, 54, 47, 39, 32, 25, 17, 12, 1, 0, 0, 57 134,127,120,114,103, 97, 91, 85, 78, 72, 66, 60, 54, 47, 41, 35, 29, 23, 16, 10, 1, 58 144,137,130,124,113,107,101, 95, 88, 82, 76, 70, 64, 57, 51, 45, 39, 33, 26, 15, 1, 59 152,145,138,132,123,117,111,105, 98, 92, 86, 80, 74, 67, 61, 55, 49, 43, 36, 20, 1, 60 162,155,148,142,133,127,121,115,108,102, 96, 90, 84, 77, 71, 65, 59, 53, 46, 30, 1, 61 172,165,158,152,143,137,131,125,118,112,106,100, 94, 87, 81, 75, 69, 63, 56, 45, 20, 62 200,200,200,200,200,200,200,200,198,193,188,183,178,173,168,163,158,153,148,129,104, 63 }; 64 65 #ifndef CUSTOM_MODES_ONLY 66 #ifdef FIXED_POINT 67 #include "static_modes_fixed.h" 68 #else 69 #include "static_modes_float.h" 70 #endif 71 #endif /* CUSTOM_MODES_ONLY */ 72 73 #ifndef M_PI 74 #define M_PI 3.141592653 75 #endif 76 77 #ifdef CUSTOM_MODES 78 79 /* Defining 25 critical bands for the full 0-20 kHz audio bandwidth 80 Taken from http://ccrma.stanford.edu/~jos/bbt/Bark_Frequency_Scale.html */ 81 #define BARK_BANDS 25 82 static const opus_int16 bark_freq[BARK_BANDS+1] = { 83 0, 100, 200, 300, 400, 84 510, 630, 770, 920, 1080, 85 1270, 1480, 1720, 2000, 2320, 86 2700, 3150, 3700, 4400, 5300, 87 6400, 7700, 9500, 12000, 15500, 88 20000}; 89 90 static opus_int16 *compute_ebands(opus_int32 Fs, int frame_size, int res, int *nbEBands) 91 { 92 opus_int16 *eBands; 93 int i, j, lin, low, high, nBark, offset=0; 94 95 /* All modes that have 2.5 ms short blocks use the same definition */ 96 if (Fs == 400*(opus_int32)frame_size) 97 { 98 *nbEBands = sizeof(eband5ms)/sizeof(eband5ms[0])-1; 99 eBands = opus_alloc(sizeof(opus_int16)*(*nbEBands+1)); 100 for (i=0;i<*nbEBands+1;i++) 101 eBands[i] = eband5ms[i]; 102 return eBands; 103 } 104 /* Find the number of critical bands supported by our sampling rate */ 105 for (nBark=1;nBark<BARK_BANDS;nBark++) 106 if (bark_freq[nBark+1]*2 >= Fs) 107 break; 108 109 /* Find where the linear part ends (i.e. where the spacing is more than min_width */ 110 for (lin=0;lin<nBark;lin++) 111 if (bark_freq[lin+1]-bark_freq[lin] >= res) 112 break; 113 114 low = (bark_freq[lin]+res/2)/res; 115 high = nBark-lin; 116 *nbEBands = low+high; 117 eBands = opus_alloc(sizeof(opus_int16)*(*nbEBands+2)); 118 119 if (eBands==NULL) 120 return NULL; 121 122 /* Linear spacing (min_width) */ 123 for (i=0;i<low;i++) 124 eBands[i] = i; 125 if (low>0) 126 offset = eBands[low-1]*res - bark_freq[lin-1]; 127 /* Spacing follows critical bands */ 128 for (i=0;i<high;i++) 129 { 130 int target = bark_freq[lin+i]; 131 /* Round to an even value */ 132 eBands[i+low] = (target+offset/2+res)/(2*res)*2; 133 offset = eBands[i+low]*res - target; 134 } 135 /* Enforce the minimum spacing at the boundary */ 136 for (i=0;i<*nbEBands;i++) 137 if (eBands[i] < i) 138 eBands[i] = i; 139 /* Round to an even value */ 140 eBands[*nbEBands] = (bark_freq[nBark]+res)/(2*res)*2; 141 if (eBands[*nbEBands] > frame_size) 142 eBands[*nbEBands] = frame_size; 143 for (i=1;i<*nbEBands-1;i++) 144 { 145 if (eBands[i+1]-eBands[i] < eBands[i]-eBands[i-1]) 146 { 147 eBands[i] -= (2*eBands[i]-eBands[i-1]-eBands[i+1])/2; 148 } 149 } 150 /* Remove any empty bands. */ 151 for (i=j=0;i<*nbEBands;i++) 152 if(eBands[i+1]>eBands[j]) 153 eBands[++j]=eBands[i+1]; 154 *nbEBands=j; 155 156 for (i=1;i<*nbEBands;i++) 157 { 158 /* Every band must be smaller than the last band. */ 159 celt_assert(eBands[i]-eBands[i-1]<=eBands[*nbEBands]-eBands[*nbEBands-1]); 160 /* Each band must be no larger than twice the size of the previous one. */ 161 celt_assert(eBands[i+1]-eBands[i]<=2*(eBands[i]-eBands[i-1])); 162 } 163 164 return eBands; 165 } 166 167 static void compute_allocation_table(CELTMode *mode) 168 { 169 int i, j; 170 unsigned char *allocVectors; 171 int maxBands = sizeof(eband5ms)/sizeof(eband5ms[0])-1; 172 173 mode->nbAllocVectors = BITALLOC_SIZE; 174 allocVectors = opus_alloc(sizeof(unsigned char)*(BITALLOC_SIZE*mode->nbEBands)); 175 if (allocVectors==NULL) 176 return; 177 178 /* Check for standard mode */ 179 if (mode->Fs == 400*(opus_int32)mode->shortMdctSize) 180 { 181 for (i=0;i<BITALLOC_SIZE*mode->nbEBands;i++) 182 allocVectors[i] = band_allocation[i]; 183 mode->allocVectors = allocVectors; 184 return; 185 } 186 /* If not the standard mode, interpolate */ 187 /* Compute per-codec-band allocation from per-critical-band matrix */ 188 for (i=0;i<BITALLOC_SIZE;i++) 189 { 190 for (j=0;j<mode->nbEBands;j++) 191 { 192 int k; 193 for (k=0;k<maxBands;k++) 194 { 195 if (400*(opus_int32)eband5ms[k] > mode->eBands[j]*(opus_int32)mode->Fs/mode->shortMdctSize) 196 break; 197 } 198 if (k>maxBands-1) 199 allocVectors[i*mode->nbEBands+j] = band_allocation[i*maxBands + maxBands-1]; 200 else { 201 opus_int32 a0, a1; 202 a1 = mode->eBands[j]*(opus_int32)mode->Fs/mode->shortMdctSize - 400*(opus_int32)eband5ms[k-1]; 203 a0 = 400*(opus_int32)eband5ms[k] - mode->eBands[j]*(opus_int32)mode->Fs/mode->shortMdctSize; 204 allocVectors[i*mode->nbEBands+j] = (a0*band_allocation[i*maxBands+k-1] 205 + a1*band_allocation[i*maxBands+k])/(a0+a1); 206 } 207 } 208 } 209 210 /*printf ("\n"); 211 for (i=0;i<BITALLOC_SIZE;i++) 212 { 213 for (j=0;j<mode->nbEBands;j++) 214 printf ("%d ", allocVectors[i*mode->nbEBands+j]); 215 printf ("\n"); 216 } 217 exit(0);*/ 218 219 mode->allocVectors = allocVectors; 220 } 221 222 #endif /* CUSTOM_MODES */ 223 224 CELTMode *opus_custom_mode_create(opus_int32 Fs, int frame_size, int *error) 225 { 226 int i; 227 #ifdef CUSTOM_MODES 228 CELTMode *mode=NULL; 229 int res; 230 opus_val16 *window; 231 opus_int16 *logN; 232 int LM; 233 int arch = opus_select_arch(); 234 ALLOC_STACK; 235 #if !defined(VAR_ARRAYS) && !defined(USE_ALLOCA) 236 if (global_stack==NULL) 237 goto failure; 238 #endif 239 #endif 240 241 #ifndef CUSTOM_MODES_ONLY 242 for (i=0;i<TOTAL_MODES;i++) 243 { 244 int j; 245 for (j=0;j<4;j++) 246 { 247 if (Fs == static_mode_list[i]->Fs && 248 (frame_size<<j) == static_mode_list[i]->shortMdctSize*static_mode_list[i]->nbShortMdcts) 249 { 250 if (error) 251 *error = OPUS_OK; 252 return (CELTMode*)static_mode_list[i]; 253 } 254 } 255 } 256 #endif /* CUSTOM_MODES_ONLY */ 257 258 #ifndef CUSTOM_MODES 259 if (error) 260 *error = OPUS_BAD_ARG; 261 return NULL; 262 #else 263 264 /* The good thing here is that permutation of the arguments will automatically be invalid */ 265 266 if (Fs < 8000 || Fs > 96000) 267 { 268 if (error) 269 *error = OPUS_BAD_ARG; 270 return NULL; 271 } 272 if (frame_size < 40 || frame_size > 1024 || frame_size%2!=0) 273 { 274 if (error) 275 *error = OPUS_BAD_ARG; 276 return NULL; 277 } 278 /* Frames of less than 1ms are not supported. */ 279 if ((opus_int32)frame_size*1000 < Fs) 280 { 281 if (error) 282 *error = OPUS_BAD_ARG; 283 return NULL; 284 } 285 286 if ((opus_int32)frame_size*75 >= Fs && (frame_size%16)==0) 287 { 288 LM = 3; 289 } else if ((opus_int32)frame_size*150 >= Fs && (frame_size%8)==0) 290 { 291 LM = 2; 292 } else if ((opus_int32)frame_size*300 >= Fs && (frame_size%4)==0) 293 { 294 LM = 1; 295 } else 296 { 297 LM = 0; 298 } 299 300 /* Shorts longer than 3.3ms are not supported. */ 301 if ((opus_int32)(frame_size>>LM)*300 > Fs) 302 { 303 if (error) 304 *error = OPUS_BAD_ARG; 305 return NULL; 306 } 307 308 mode = opus_alloc(sizeof(CELTMode)); 309 if (mode==NULL) 310 goto failure; 311 mode->Fs = Fs; 312 313 /* Pre/de-emphasis depends on sampling rate. The "standard" pre-emphasis 314 is defined as A(z) = 1 - 0.85*z^-1 at 48 kHz. Other rates should 315 approximate that. */ 316 if(Fs < 12000) /* 8 kHz */ 317 { 318 mode->preemph[0] = QCONST16(0.3500061035f, 15); 319 mode->preemph[1] = -QCONST16(0.1799926758f, 15); 320 mode->preemph[2] = QCONST16(0.2719968125f, SIG_SHIFT); /* exact 1/preemph[3] */ 321 mode->preemph[3] = QCONST16(3.6765136719f, 13); 322 } else if(Fs < 24000) /* 16 kHz */ 323 { 324 mode->preemph[0] = QCONST16(0.6000061035f, 15); 325 mode->preemph[1] = -QCONST16(0.1799926758f, 15); 326 mode->preemph[2] = QCONST16(0.4424998650f, SIG_SHIFT); /* exact 1/preemph[3] */ 327 mode->preemph[3] = QCONST16(2.2598876953f, 13); 328 } else if(Fs < 40000) /* 32 kHz */ 329 { 330 mode->preemph[0] = QCONST16(0.7799987793f, 15); 331 mode->preemph[1] = -QCONST16(0.1000061035f, 15); 332 mode->preemph[2] = QCONST16(0.7499771125f, SIG_SHIFT); /* exact 1/preemph[3] */ 333 mode->preemph[3] = QCONST16(1.3333740234f, 13); 334 } else /* 48 kHz */ 335 { 336 mode->preemph[0] = QCONST16(0.8500061035f, 15); 337 mode->preemph[1] = QCONST16(0.0f, 15); 338 mode->preemph[2] = QCONST16(1.f, SIG_SHIFT); 339 mode->preemph[3] = QCONST16(1.f, 13); 340 } 341 342 mode->maxLM = LM; 343 mode->nbShortMdcts = 1<<LM; 344 mode->shortMdctSize = frame_size/mode->nbShortMdcts; 345 res = (mode->Fs+mode->shortMdctSize)/(2*mode->shortMdctSize); 346 347 mode->eBands = compute_ebands(Fs, mode->shortMdctSize, res, &mode->nbEBands); 348 if (mode->eBands==NULL) 349 goto failure; 350 #if !defined(SMALL_FOOTPRINT) 351 /* Make sure we don't allocate a band larger than our PVQ table. 352 208 should be enough, but let's be paranoid. */ 353 if ((mode->eBands[mode->nbEBands] - mode->eBands[mode->nbEBands-1])<<LM > 354 208) { 355 goto failure; 356 } 357 #endif 358 359 mode->effEBands = mode->nbEBands; 360 while (mode->eBands[mode->effEBands] > mode->shortMdctSize) 361 mode->effEBands--; 362 363 /* Overlap must be divisible by 4 */ 364 mode->overlap = ((mode->shortMdctSize>>2)<<2); 365 366 compute_allocation_table(mode); 367 if (mode->allocVectors==NULL) 368 goto failure; 369 370 window = (opus_val16*)opus_alloc(mode->overlap*sizeof(opus_val16)); 371 if (window==NULL) 372 goto failure; 373 374 #ifndef FIXED_POINT 375 for (i=0;i<mode->overlap;i++) 376 window[i] = Q15ONE*sin(.5*M_PI* sin(.5*M_PI*(i+.5)/mode->overlap) * sin(.5*M_PI*(i+.5)/mode->overlap)); 377 #else 378 for (i=0;i<mode->overlap;i++) 379 window[i] = MIN32(32767,floor(.5+32768.*sin(.5*M_PI* sin(.5*M_PI*(i+.5)/mode->overlap) * sin(.5*M_PI*(i+.5)/mode->overlap)))); 380 #endif 381 mode->window = window; 382 383 logN = (opus_int16*)opus_alloc(mode->nbEBands*sizeof(opus_int16)); 384 if (logN==NULL) 385 goto failure; 386 387 for (i=0;i<mode->nbEBands;i++) 388 logN[i] = log2_frac(mode->eBands[i+1]-mode->eBands[i], BITRES); 389 mode->logN = logN; 390 391 compute_pulse_cache(mode, mode->maxLM); 392 393 if (clt_mdct_init(&mode->mdct, 2*mode->shortMdctSize*mode->nbShortMdcts, 394 mode->maxLM, arch) == 0) 395 goto failure; 396 397 if (error) 398 *error = OPUS_OK; 399 400 return mode; 401 failure: 402 if (error) 403 *error = OPUS_ALLOC_FAIL; 404 if (mode!=NULL) 405 opus_custom_mode_destroy(mode); 406 return NULL; 407 #endif /* !CUSTOM_MODES */ 408 } 409 410 #ifdef CUSTOM_MODES 411 void opus_custom_mode_destroy(CELTMode *mode) 412 { 413 int arch = opus_select_arch(); 414 415 if (mode == NULL) 416 return; 417 #ifndef CUSTOM_MODES_ONLY 418 { 419 int i; 420 for (i=0;i<TOTAL_MODES;i++) 421 { 422 if (mode == static_mode_list[i]) 423 { 424 return; 425 } 426 } 427 } 428 #endif /* CUSTOM_MODES_ONLY */ 429 opus_free((opus_int16*)mode->eBands); 430 opus_free((opus_int16*)mode->allocVectors); 431 432 opus_free((opus_val16*)mode->window); 433 opus_free((opus_int16*)mode->logN); 434 435 opus_free((opus_int16*)mode->cache.index); 436 opus_free((unsigned char*)mode->cache.bits); 437 opus_free((unsigned char*)mode->cache.caps); 438 clt_mdct_clear(&mode->mdct, arch); 439 440 opus_free((CELTMode *)mode); 441 } 442 #endif 443