1885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org/* Copyright (c) 2007-2008 CSIRO 2885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org Copyright (c) 2007-2009 Xiph.Org Foundation 3885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org Written by Jean-Marc Valin */ 4885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org/* 5885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org Redistribution and use in source and binary forms, with or without 6885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org modification, are permitted provided that the following conditions 7885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org are met: 8885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 9885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org - Redistributions of source code must retain the above copyright 10885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org notice, this list of conditions and the following disclaimer. 11885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 12885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org - Redistributions in binary form must reproduce the above copyright 13885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org notice, this list of conditions and the following disclaimer in the 14885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org documentation and/or other materials provided with the distribution. 15885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 16885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 17885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 18885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 19885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 20885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 21885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 22885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 23885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 24885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 25885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 26885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org*/ 28885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 29885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#ifndef RATE_H 30885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#define RATE_H 31885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 32885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#define MAX_PSEUDO 40 33885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#define LOG_MAX_PSEUDO 6 34885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 35885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#define MAX_PULSES 128 36885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 37885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#define MAX_FINE_BITS 8 38885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 39885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#define FINE_OFFSET 21 40885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#define QTHETA_OFFSET 4 41885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#define QTHETA_OFFSET_TWOPHASE 16 42885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 43885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#include "cwrs.h" 44885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#include "modes.h" 45885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 46885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.orgvoid compute_pulse_cache(CELTMode *m, int LM); 47885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 483c3902f0ac13428394f14f78f0fab05ef3468d69tlegrand@google.comstatic OPUS_INLINE int get_pulses(int i) 49885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org{ 50885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org return i<8 ? i : (8 + (i&7)) << ((i>>3)-1); 51885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org} 52885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 533c3902f0ac13428394f14f78f0fab05ef3468d69tlegrand@google.comstatic OPUS_INLINE int bits2pulses(const CELTMode *m, int band, int LM, int bits) 54885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org{ 55885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org int i; 56885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org int lo, hi; 57885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org const unsigned char *cache; 58885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 59885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org LM++; 60885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org cache = m->cache.bits + m->cache.index[LM*m->nbEBands+band]; 61885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 62885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org lo = 0; 63885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org hi = cache[0]; 64885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org bits--; 65885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org for (i=0;i<LOG_MAX_PSEUDO;i++) 66885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org { 67885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org int mid = (lo+hi+1)>>1; 68885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org /* OPT: Make sure this is implemented with a conditional move */ 69885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org if ((int)cache[mid] >= bits) 70885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org hi = mid; 71885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org else 72885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org lo = mid; 73885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org } 74885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org if (bits- (lo == 0 ? -1 : (int)cache[lo]) <= (int)cache[hi]-bits) 75885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org return lo; 76885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org else 77885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org return hi; 78885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org} 79885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 803c3902f0ac13428394f14f78f0fab05ef3468d69tlegrand@google.comstatic OPUS_INLINE int pulses2bits(const CELTMode *m, int band, int LM, int pulses) 81885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org{ 82885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org const unsigned char *cache; 83885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 84885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org LM++; 85885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org cache = m->cache.bits + m->cache.index[LM*m->nbEBands+band]; 86885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org return pulses == 0 ? 0 : cache[pulses]+1; 87885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org} 88885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 89885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org/** Compute the pulse allocation, i.e. how many pulses will go in each 90885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org * band. 91885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org @param m mode 92885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org @param offsets Requested increase or decrease in the number of bits for 93885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org each band 94885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org @param total Number of bands 95885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org @param pulses Number of pulses per band (returned) 96885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org @return Total number of bits allocated 97885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org*/ 98885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.orgint compute_allocation(const CELTMode *m, int start, int end, const int *offsets, const int *cap, int alloc_trim, int *intensity, int *dual_stero, 99e3ea049fcaee2247e45f0ce793d4313babb4ef69tlegrand@chromium.org opus_int32 total, opus_int32 *balance, int *pulses, int *ebits, int *fine_priority, int C, int LM, ec_ctx *ec, int encode, int prev, int signalBandwidth); 100885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org 101885f2ff5a7a7d6a73432d26a6c0ae9147e6b452sergeyu@chromium.org#endif 102