cannam@154: /* Copyright (c) 2007-2008 CSIRO cannam@154: Copyright (c) 2007-2009 Xiph.Org Foundation cannam@154: Copyright (c) 2008 Gregory Maxwell cannam@154: Written by Jean-Marc Valin and Gregory Maxwell */ cannam@154: /* cannam@154: Redistribution and use in source and binary forms, with or without cannam@154: modification, are permitted provided that the following conditions cannam@154: are met: cannam@154: cannam@154: - Redistributions of source code must retain the above copyright cannam@154: notice, this list of conditions and the following disclaimer. cannam@154: cannam@154: - Redistributions in binary form must reproduce the above copyright cannam@154: notice, this list of conditions and the following disclaimer in the cannam@154: documentation and/or other materials provided with the distribution. cannam@154: cannam@154: THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS cannam@154: ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT cannam@154: LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR cannam@154: A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER cannam@154: OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, cannam@154: EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, cannam@154: PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR cannam@154: PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF cannam@154: LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING cannam@154: NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS cannam@154: SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. cannam@154: */ cannam@154: cannam@154: #ifdef HAVE_CONFIG_H cannam@154: #include "config.h" cannam@154: #endif cannam@154: cannam@154: #include "celt.h" cannam@154: #include "modes.h" cannam@154: #include "rate.h" cannam@154: #include "os_support.h" cannam@154: #include "stack_alloc.h" cannam@154: #include "quant_bands.h" cannam@154: #include "cpu_support.h" cannam@154: cannam@154: static const opus_int16 eband5ms[] = { cannam@154: /*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 */ cannam@154: 0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 12, 14, 16, 20, 24, 28, 34, 40, 48, 60, 78, 100 cannam@154: }; cannam@154: cannam@154: /* Alternate tuning (partially derived from Vorbis) */ cannam@154: #define BITALLOC_SIZE 11 cannam@154: /* Bit allocation table in units of 1/32 bit/sample (0.1875 dB SNR) */ cannam@154: static const unsigned char band_allocation[] = { cannam@154: /*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 */ cannam@154: 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, cannam@154: 90, 80, 75, 69, 63, 56, 49, 40, 34, 29, 20, 18, 10, 0, 0, 0, 0, 0, 0, 0, 0, cannam@154: 110,100, 90, 84, 78, 71, 65, 58, 51, 45, 39, 32, 26, 20, 12, 0, 0, 0, 0, 0, 0, cannam@154: 118,110,103, 93, 86, 80, 75, 70, 65, 59, 53, 47, 40, 31, 23, 15, 4, 0, 0, 0, 0, cannam@154: 126,119,112,104, 95, 89, 83, 78, 72, 66, 60, 54, 47, 39, 32, 25, 17, 12, 1, 0, 0, cannam@154: 134,127,120,114,103, 97, 91, 85, 78, 72, 66, 60, 54, 47, 41, 35, 29, 23, 16, 10, 1, cannam@154: 144,137,130,124,113,107,101, 95, 88, 82, 76, 70, 64, 57, 51, 45, 39, 33, 26, 15, 1, cannam@154: 152,145,138,132,123,117,111,105, 98, 92, 86, 80, 74, 67, 61, 55, 49, 43, 36, 20, 1, cannam@154: 162,155,148,142,133,127,121,115,108,102, 96, 90, 84, 77, 71, 65, 59, 53, 46, 30, 1, cannam@154: 172,165,158,152,143,137,131,125,118,112,106,100, 94, 87, 81, 75, 69, 63, 56, 45, 20, cannam@154: 200,200,200,200,200,200,200,200,198,193,188,183,178,173,168,163,158,153,148,129,104, cannam@154: }; cannam@154: cannam@154: #ifndef CUSTOM_MODES_ONLY cannam@154: #ifdef FIXED_POINT cannam@154: #include "static_modes_fixed.h" cannam@154: #else cannam@154: #include "static_modes_float.h" cannam@154: #endif cannam@154: #endif /* CUSTOM_MODES_ONLY */ cannam@154: cannam@154: #ifndef M_PI cannam@154: #define M_PI 3.141592653 cannam@154: #endif cannam@154: cannam@154: #ifdef CUSTOM_MODES cannam@154: cannam@154: /* Defining 25 critical bands for the full 0-20 kHz audio bandwidth cannam@154: Taken from http://ccrma.stanford.edu/~jos/bbt/Bark_Frequency_Scale.html */ cannam@154: #define BARK_BANDS 25 cannam@154: static const opus_int16 bark_freq[BARK_BANDS+1] = { cannam@154: 0, 100, 200, 300, 400, cannam@154: 510, 630, 770, 920, 1080, cannam@154: 1270, 1480, 1720, 2000, 2320, cannam@154: 2700, 3150, 3700, 4400, 5300, cannam@154: 6400, 7700, 9500, 12000, 15500, cannam@154: 20000}; cannam@154: cannam@154: static opus_int16 *compute_ebands(opus_int32 Fs, int frame_size, int res, int *nbEBands) cannam@154: { cannam@154: opus_int16 *eBands; cannam@154: int i, j, lin, low, high, nBark, offset=0; cannam@154: cannam@154: /* All modes that have 2.5 ms short blocks use the same definition */ cannam@154: if (Fs == 400*(opus_int32)frame_size) cannam@154: { cannam@154: *nbEBands = sizeof(eband5ms)/sizeof(eband5ms[0])-1; cannam@154: eBands = opus_alloc(sizeof(opus_int16)*(*nbEBands+1)); cannam@154: for (i=0;i<*nbEBands+1;i++) cannam@154: eBands[i] = eband5ms[i]; cannam@154: return eBands; cannam@154: } cannam@154: /* Find the number of critical bands supported by our sampling rate */ cannam@154: for (nBark=1;nBark= Fs) cannam@154: break; cannam@154: cannam@154: /* Find where the linear part ends (i.e. where the spacing is more than min_width */ cannam@154: for (lin=0;lin= res) cannam@154: break; cannam@154: cannam@154: low = (bark_freq[lin]+res/2)/res; cannam@154: high = nBark-lin; cannam@154: *nbEBands = low+high; cannam@154: eBands = opus_alloc(sizeof(opus_int16)*(*nbEBands+2)); cannam@154: cannam@154: if (eBands==NULL) cannam@154: return NULL; cannam@154: cannam@154: /* Linear spacing (min_width) */ cannam@154: for (i=0;i0) cannam@154: offset = eBands[low-1]*res - bark_freq[lin-1]; cannam@154: /* Spacing follows critical bands */ cannam@154: for (i=0;i frame_size) cannam@154: eBands[*nbEBands] = frame_size; cannam@154: for (i=1;i<*nbEBands-1;i++) cannam@154: { cannam@154: if (eBands[i+1]-eBands[i] < eBands[i]-eBands[i-1]) cannam@154: { cannam@154: eBands[i] -= (2*eBands[i]-eBands[i-1]-eBands[i+1])/2; cannam@154: } cannam@154: } cannam@154: /* Remove any empty bands. */ cannam@154: for (i=j=0;i<*nbEBands;i++) cannam@154: if(eBands[i+1]>eBands[j]) cannam@154: eBands[++j]=eBands[i+1]; cannam@154: *nbEBands=j; cannam@154: cannam@154: for (i=1;i<*nbEBands;i++) cannam@154: { cannam@154: /* Every band must be smaller than the last band. */ cannam@154: celt_assert(eBands[i]-eBands[i-1]<=eBands[*nbEBands]-eBands[*nbEBands-1]); cannam@154: /* Each band must be no larger than twice the size of the previous one. */ cannam@154: celt_assert(eBands[i+1]-eBands[i]<=2*(eBands[i]-eBands[i-1])); cannam@154: } cannam@154: cannam@154: return eBands; cannam@154: } cannam@154: cannam@154: static void compute_allocation_table(CELTMode *mode) cannam@154: { cannam@154: int i, j; cannam@154: unsigned char *allocVectors; cannam@154: int maxBands = sizeof(eband5ms)/sizeof(eband5ms[0])-1; cannam@154: cannam@154: mode->nbAllocVectors = BITALLOC_SIZE; cannam@154: allocVectors = opus_alloc(sizeof(unsigned char)*(BITALLOC_SIZE*mode->nbEBands)); cannam@154: if (allocVectors==NULL) cannam@154: return; cannam@154: cannam@154: /* Check for standard mode */ cannam@154: if (mode->Fs == 400*(opus_int32)mode->shortMdctSize) cannam@154: { cannam@154: for (i=0;inbEBands;i++) cannam@154: allocVectors[i] = band_allocation[i]; cannam@154: mode->allocVectors = allocVectors; cannam@154: return; cannam@154: } cannam@154: /* If not the standard mode, interpolate */ cannam@154: /* Compute per-codec-band allocation from per-critical-band matrix */ cannam@154: for (i=0;inbEBands;j++) cannam@154: { cannam@154: int k; cannam@154: for (k=0;k mode->eBands[j]*(opus_int32)mode->Fs/mode->shortMdctSize) cannam@154: break; cannam@154: } cannam@154: if (k>maxBands-1) cannam@154: allocVectors[i*mode->nbEBands+j] = band_allocation[i*maxBands + maxBands-1]; cannam@154: else { cannam@154: opus_int32 a0, a1; cannam@154: a1 = mode->eBands[j]*(opus_int32)mode->Fs/mode->shortMdctSize - 400*(opus_int32)eband5ms[k-1]; cannam@154: a0 = 400*(opus_int32)eband5ms[k] - mode->eBands[j]*(opus_int32)mode->Fs/mode->shortMdctSize; cannam@154: allocVectors[i*mode->nbEBands+j] = (a0*band_allocation[i*maxBands+k-1] cannam@154: + a1*band_allocation[i*maxBands+k])/(a0+a1); cannam@154: } cannam@154: } cannam@154: } cannam@154: cannam@154: /*printf ("\n"); cannam@154: for (i=0;inbEBands;j++) cannam@154: printf ("%d ", allocVectors[i*mode->nbEBands+j]); cannam@154: printf ("\n"); cannam@154: } cannam@154: exit(0);*/ cannam@154: cannam@154: mode->allocVectors = allocVectors; cannam@154: } cannam@154: cannam@154: #endif /* CUSTOM_MODES */ cannam@154: cannam@154: CELTMode *opus_custom_mode_create(opus_int32 Fs, int frame_size, int *error) cannam@154: { cannam@154: int i; cannam@154: #ifdef CUSTOM_MODES cannam@154: CELTMode *mode=NULL; cannam@154: int res; cannam@154: opus_val16 *window; cannam@154: opus_int16 *logN; cannam@154: int LM; cannam@154: int arch = opus_select_arch(); cannam@154: ALLOC_STACK; cannam@154: #if !defined(VAR_ARRAYS) && !defined(USE_ALLOCA) cannam@154: if (global_stack==NULL) cannam@154: goto failure; cannam@154: #endif cannam@154: #endif cannam@154: cannam@154: #ifndef CUSTOM_MODES_ONLY cannam@154: for (i=0;iFs && cannam@154: (frame_size<shortMdctSize*static_mode_list[i]->nbShortMdcts) cannam@154: { cannam@154: if (error) cannam@154: *error = OPUS_OK; cannam@154: return (CELTMode*)static_mode_list[i]; cannam@154: } cannam@154: } cannam@154: } cannam@154: #endif /* CUSTOM_MODES_ONLY */ cannam@154: cannam@154: #ifndef CUSTOM_MODES cannam@154: if (error) cannam@154: *error = OPUS_BAD_ARG; cannam@154: return NULL; cannam@154: #else cannam@154: cannam@154: /* The good thing here is that permutation of the arguments will automatically be invalid */ cannam@154: cannam@154: if (Fs < 8000 || Fs > 96000) cannam@154: { cannam@154: if (error) cannam@154: *error = OPUS_BAD_ARG; cannam@154: return NULL; cannam@154: } cannam@154: if (frame_size < 40 || frame_size > 1024 || frame_size%2!=0) cannam@154: { cannam@154: if (error) cannam@154: *error = OPUS_BAD_ARG; cannam@154: return NULL; cannam@154: } cannam@154: /* Frames of less than 1ms are not supported. */ cannam@154: if ((opus_int32)frame_size*1000 < Fs) cannam@154: { cannam@154: if (error) cannam@154: *error = OPUS_BAD_ARG; cannam@154: return NULL; cannam@154: } cannam@154: cannam@154: if ((opus_int32)frame_size*75 >= Fs && (frame_size%16)==0) cannam@154: { cannam@154: LM = 3; cannam@154: } else if ((opus_int32)frame_size*150 >= Fs && (frame_size%8)==0) cannam@154: { cannam@154: LM = 2; cannam@154: } else if ((opus_int32)frame_size*300 >= Fs && (frame_size%4)==0) cannam@154: { cannam@154: LM = 1; cannam@154: } else cannam@154: { cannam@154: LM = 0; cannam@154: } cannam@154: cannam@154: /* Shorts longer than 3.3ms are not supported. */ cannam@154: if ((opus_int32)(frame_size>>LM)*300 > Fs) cannam@154: { cannam@154: if (error) cannam@154: *error = OPUS_BAD_ARG; cannam@154: return NULL; cannam@154: } cannam@154: cannam@154: mode = opus_alloc(sizeof(CELTMode)); cannam@154: if (mode==NULL) cannam@154: goto failure; cannam@154: mode->Fs = Fs; cannam@154: cannam@154: /* Pre/de-emphasis depends on sampling rate. The "standard" pre-emphasis cannam@154: is defined as A(z) = 1 - 0.85*z^-1 at 48 kHz. Other rates should cannam@154: approximate that. */ cannam@154: if(Fs < 12000) /* 8 kHz */ cannam@154: { cannam@154: mode->preemph[0] = QCONST16(0.3500061035f, 15); cannam@154: mode->preemph[1] = -QCONST16(0.1799926758f, 15); cannam@154: mode->preemph[2] = QCONST16(0.2719968125f, SIG_SHIFT); /* exact 1/preemph[3] */ cannam@154: mode->preemph[3] = QCONST16(3.6765136719f, 13); cannam@154: } else if(Fs < 24000) /* 16 kHz */ cannam@154: { cannam@154: mode->preemph[0] = QCONST16(0.6000061035f, 15); cannam@154: mode->preemph[1] = -QCONST16(0.1799926758f, 15); cannam@154: mode->preemph[2] = QCONST16(0.4424998650f, SIG_SHIFT); /* exact 1/preemph[3] */ cannam@154: mode->preemph[3] = QCONST16(2.2598876953f, 13); cannam@154: } else if(Fs < 40000) /* 32 kHz */ cannam@154: { cannam@154: mode->preemph[0] = QCONST16(0.7799987793f, 15); cannam@154: mode->preemph[1] = -QCONST16(0.1000061035f, 15); cannam@154: mode->preemph[2] = QCONST16(0.7499771125f, SIG_SHIFT); /* exact 1/preemph[3] */ cannam@154: mode->preemph[3] = QCONST16(1.3333740234f, 13); cannam@154: } else /* 48 kHz */ cannam@154: { cannam@154: mode->preemph[0] = QCONST16(0.8500061035f, 15); cannam@154: mode->preemph[1] = QCONST16(0.0f, 15); cannam@154: mode->preemph[2] = QCONST16(1.f, SIG_SHIFT); cannam@154: mode->preemph[3] = QCONST16(1.f, 13); cannam@154: } cannam@154: cannam@154: mode->maxLM = LM; cannam@154: mode->nbShortMdcts = 1<shortMdctSize = frame_size/mode->nbShortMdcts; cannam@154: res = (mode->Fs+mode->shortMdctSize)/(2*mode->shortMdctSize); cannam@154: cannam@154: mode->eBands = compute_ebands(Fs, mode->shortMdctSize, res, &mode->nbEBands); cannam@154: if (mode->eBands==NULL) cannam@154: goto failure; cannam@154: #if !defined(SMALL_FOOTPRINT) cannam@154: /* Make sure we don't allocate a band larger than our PVQ table. cannam@154: 208 should be enough, but let's be paranoid. */ cannam@154: if ((mode->eBands[mode->nbEBands] - mode->eBands[mode->nbEBands-1])< cannam@154: 208) { cannam@154: goto failure; cannam@154: } cannam@154: #endif cannam@154: cannam@154: mode->effEBands = mode->nbEBands; cannam@154: while (mode->eBands[mode->effEBands] > mode->shortMdctSize) cannam@154: mode->effEBands--; cannam@154: cannam@154: /* Overlap must be divisible by 4 */ cannam@154: mode->overlap = ((mode->shortMdctSize>>2)<<2); cannam@154: cannam@154: compute_allocation_table(mode); cannam@154: if (mode->allocVectors==NULL) cannam@154: goto failure; cannam@154: cannam@154: window = (opus_val16*)opus_alloc(mode->overlap*sizeof(opus_val16)); cannam@154: if (window==NULL) cannam@154: goto failure; cannam@154: cannam@154: #ifndef FIXED_POINT cannam@154: for (i=0;ioverlap;i++) cannam@154: window[i] = Q15ONE*sin(.5*M_PI* sin(.5*M_PI*(i+.5)/mode->overlap) * sin(.5*M_PI*(i+.5)/mode->overlap)); cannam@154: #else cannam@154: for (i=0;ioverlap;i++) cannam@154: 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)))); cannam@154: #endif cannam@154: mode->window = window; cannam@154: cannam@154: logN = (opus_int16*)opus_alloc(mode->nbEBands*sizeof(opus_int16)); cannam@154: if (logN==NULL) cannam@154: goto failure; cannam@154: cannam@154: for (i=0;inbEBands;i++) cannam@154: logN[i] = log2_frac(mode->eBands[i+1]-mode->eBands[i], BITRES); cannam@154: mode->logN = logN; cannam@154: cannam@154: compute_pulse_cache(mode, mode->maxLM); cannam@154: cannam@154: if (clt_mdct_init(&mode->mdct, 2*mode->shortMdctSize*mode->nbShortMdcts, cannam@154: mode->maxLM, arch) == 0) cannam@154: goto failure; cannam@154: cannam@154: if (error) cannam@154: *error = OPUS_OK; cannam@154: cannam@154: return mode; cannam@154: failure: cannam@154: if (error) cannam@154: *error = OPUS_ALLOC_FAIL; cannam@154: if (mode!=NULL) cannam@154: opus_custom_mode_destroy(mode); cannam@154: return NULL; cannam@154: #endif /* !CUSTOM_MODES */ cannam@154: } cannam@154: cannam@154: #ifdef CUSTOM_MODES cannam@154: void opus_custom_mode_destroy(CELTMode *mode) cannam@154: { cannam@154: int arch = opus_select_arch(); cannam@154: cannam@154: if (mode == NULL) cannam@154: return; cannam@154: #ifndef CUSTOM_MODES_ONLY cannam@154: { cannam@154: int i; cannam@154: for (i=0;ieBands); cannam@154: opus_free((unsigned char*)mode->allocVectors); cannam@154: cannam@154: opus_free((opus_val16*)mode->window); cannam@154: opus_free((opus_int16*)mode->logN); cannam@154: cannam@154: opus_free((opus_int16*)mode->cache.index); cannam@154: opus_free((unsigned char*)mode->cache.bits); cannam@154: opus_free((unsigned char*)mode->cache.caps); cannam@154: clt_mdct_clear(&mode->mdct, arch); cannam@154: cannam@154: opus_free((CELTMode *)mode); cannam@154: } cannam@154: #endif