123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206 |
- #ifndef R8BBASE_INCLUDED
- #define R8BBASE_INCLUDED
- #include <stdlib.h>
- #include <stdint.h>
- #include <string.h>
- #include <math.h>
- #include "r8bconf.h"
- #if defined( _WIN32 )
- #include <windows.h>
- #else
- #include <pthread.h>
- #endif
- #if defined( __SSE4_2__ ) || defined( __SSE4_1__ ) || \
- defined( __SSSE3__ ) || defined( __SSE3__ ) || defined( __SSE2__ ) || \
- defined( __x86_64__ ) || defined( _M_AMD64 ) || defined( _M_X64 ) || \
- defined( __amd64 )
- #include <immintrin.h>
- #define R8B_SSE2
- #define R8B_SIMD_ISH
- #elif defined( __aarch64__ ) || defined( __arm64__ ) || defined( __ARM_NEON )
- #include <arm_neon.h>
- #define R8B_NEON
- #if !defined( __APPLE__ )
- #define R8B_SIMD_ISH
- #endif
- #endif
- namespace r8b {
- #define R8B_VERSION "5.6"
- #define R8B_PI 3.14159265358979324
- #define R8B_2PI 6.28318530717958648
- #define R8B_3PI 9.42477796076937972
- #define R8B_PId2 1.57079632679489662
- #define R8BNOCTOR( ClassName ) \
- private: \
- ClassName( const ClassName& ) { } \
- ClassName& operator = ( const ClassName& ) { return( *this ); }
- class CStdClassAllocator
- {
- public:
-
- void* operator new( size_t, void* p )
- {
- return( p );
- }
-
- void* operator new( size_t n )
- {
- return( :: malloc( n ));
- }
-
- void* operator new[]( size_t n )
- {
- return( :: malloc( n ));
- }
-
- void operator delete( void* p )
- {
- :: free( p );
- }
-
- void operator delete[]( void* p )
- {
- :: free( p );
- }
- };
- class CStdMemAllocator : public CStdClassAllocator
- {
- public:
-
- static void* allocmem( const size_t Size )
- {
- return( :: malloc( Size ));
- }
-
- static void* reallocmem( void* p, const size_t Size )
- {
- return( :: realloc( p, Size ));
- }
-
- static void freemem( void* p )
- {
- :: free( p );
- }
- };
- template< typename T >
- inline T* alignptr( T* const ptr, const uintptr_t align )
- {
- return( (T*) (( (uintptr_t) ptr + align - 1 ) & ~( align - 1 )));
- }
- template< typename T >
- class CFixedBuffer : public R8B_MEMALLOCCLASS
- {
- R8BNOCTOR( CFixedBuffer );
- public:
- CFixedBuffer()
- : Data0( NULL )
- , Data( NULL )
- {
- }
-
- CFixedBuffer( const int Capacity )
- {
- R8BASSERT( Capacity > 0 || Capacity == 0 );
- Data0 = allocmem( Capacity * sizeof( T ) + Alignment );
- Data = (T*) alignptr( Data0, Alignment );
- R8BASSERT( Data0 != NULL || Capacity == 0 );
- }
- ~CFixedBuffer()
- {
- freemem( Data0 );
- }
-
- void alloc( const int Capacity )
- {
- R8BASSERT( Capacity > 0 || Capacity == 0 );
- freemem( Data0 );
- Data0 = allocmem( Capacity * sizeof( T ) + Alignment );
- Data = (T*) alignptr( Data0, Alignment );
- R8BASSERT( Data0 != NULL || Capacity == 0 );
- }
-
- void realloc( const int PrevCapacity, const int NewCapacity )
- {
- R8BASSERT( PrevCapacity >= 0 );
- R8BASSERT( NewCapacity >= 0 );
- void* const NewData0 = allocmem( NewCapacity * sizeof( T ) +
- Alignment );
- T* const NewData = (T*) alignptr( NewData0, Alignment );
- const size_t CopySize = ( PrevCapacity > NewCapacity ?
- NewCapacity : PrevCapacity ) * sizeof( T );
- if( CopySize > 0 )
- {
- memcpy( NewData, Data, CopySize );
- }
- freemem( Data0 );
- Data0 = NewData0;
- Data = NewData;
- R8BASSERT( Data0 != NULL || NewCapacity == 0 );
- }
-
- void free()
- {
- freemem( Data0 );
- Data0 = NULL;
- Data = NULL;
- }
-
- T* getPtr() const
- {
- return( Data );
- }
-
- operator T* () const
- {
- return( Data );
- }
- private:
- static const size_t Alignment = 64;
-
-
- void* Data0;
-
- T* Data;
-
- };
- template< class T >
- class CPtrKeeper
- {
- R8BNOCTOR( CPtrKeeper );
- public:
- CPtrKeeper()
- : Object( NULL )
- {
- }
-
- template< class T2 >
- CPtrKeeper( T2 const aObject )
- : Object( aObject )
- {
- }
- ~CPtrKeeper()
- {
- delete Object;
- }
-
- template< class T2 >
- void operator = ( T2 const aObject )
- {
- reset();
- Object = aObject;
- }
-
- T operator -> () const
- {
- return( Object );
- }
-
- operator T () const
- {
- return( Object );
- }
-
- void reset()
- {
- T DelObj = Object;
- Object = NULL;
- delete DelObj;
- }
-
- T unkeep()
- {
- T ResObject = Object;
- Object = NULL;
- return( ResObject );
- }
- private:
- T Object;
-
- };
- class CSyncObject
- {
- R8BNOCTOR( CSyncObject );
- public:
- CSyncObject()
- {
- #if defined( _WIN32 )
- InitializeCriticalSectionAndSpinCount( &CritSec, 4000 );
- #else
- pthread_mutexattr_t MutexAttrs;
- pthread_mutexattr_init( &MutexAttrs );
- pthread_mutexattr_settype( &MutexAttrs, PTHREAD_MUTEX_RECURSIVE );
- pthread_mutex_init( &Mutex, &MutexAttrs );
- pthread_mutexattr_destroy( &MutexAttrs );
- #endif
- }
- ~CSyncObject()
- {
- #if defined( _WIN32 )
- DeleteCriticalSection( &CritSec );
- #else
- pthread_mutex_destroy( &Mutex );
- #endif
- }
-
- void acquire()
- {
- #if defined( _WIN32 )
- EnterCriticalSection( &CritSec );
- #else
- pthread_mutex_lock( &Mutex );
- #endif
- }
-
- void release()
- {
- #if defined( _WIN32 )
- LeaveCriticalSection( &CritSec );
- #else
- pthread_mutex_unlock( &Mutex );
- #endif
- }
- private:
- #if defined( _WIN32 )
- CRITICAL_SECTION CritSec;
-
-
- #else
- pthread_mutex_t Mutex;
-
- #endif
- };
- class CSyncKeeper
- {
- R8BNOCTOR( CSyncKeeper );
- public:
- CSyncKeeper()
- : SyncObj( NULL )
- {
- }
-
- CSyncKeeper( CSyncObject* const aSyncObj )
- : SyncObj( aSyncObj )
- {
- if( SyncObj != NULL )
- {
- SyncObj -> acquire();
- }
- }
-
- CSyncKeeper( CSyncObject& aSyncObj )
- : SyncObj( &aSyncObj )
- {
- SyncObj -> acquire();
- }
- ~CSyncKeeper()
- {
- if( SyncObj != NULL )
- {
- SyncObj -> release();
- }
- }
- protected:
- CSyncObject* SyncObj;
-
- };
- #define R8BSYNC( SyncObject ) R8BSYNC_( SyncObject, __LINE__ )
- #define R8BSYNC_( SyncObject, id ) R8BSYNC__( SyncObject, id )
- #define R8BSYNC__( SyncObject, id ) CSyncKeeper SyncKeeper##id( SyncObject )
- class CSineGen
- {
- public:
- CSineGen()
- {
- }
-
- CSineGen( const double si, const double ph )
- : svalue1( sin( ph ))
- , svalue2( sin( ph - si ))
- , sincr( 2.0 * cos( si ))
- {
- }
-
- CSineGen( const double si, const double ph, const double g )
- : svalue1( sin( ph ) * g )
- , svalue2( sin( ph - si ) * g )
- , sincr( 2.0 * cos( si ))
- {
- }
-
- void init( const double si, const double ph )
- {
- svalue1 = sin( ph );
- svalue2 = sin( ph - si );
- sincr = 2.0 * cos( si );
- }
-
- void init( const double si, const double ph, const double g )
- {
- svalue1 = sin( ph ) * g;
- svalue2 = sin( ph - si ) * g;
- sincr = 2.0 * cos( si );
- }
-
- double generate()
- {
- const double res = svalue1;
- svalue1 = sincr * res - svalue2;
- svalue2 = res;
- return( res );
- }
- private:
- double svalue1;
-
- double svalue2;
-
- double sincr;
-
- };
- inline int getBitOccupancy( const int v )
- {
- static const uint8_t OccupancyTable[] =
- {
- 1, 1, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4,
- 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
- 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
- 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
- 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
- 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
- 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
- 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
- 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
- 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
- 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
- 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
- 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
- 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
- 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
- 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8
- };
- const int tt = v >> 16;
- if( tt != 0 )
- {
- const int t = v >> 24;
- return( t != 0 ? 24 + OccupancyTable[ t & 0xFF ] :
- 16 + OccupancyTable[ tt ]);
- }
- else
- {
- const int t = v >> 8;
- return( t != 0 ? 8 + OccupancyTable[ t ] : OccupancyTable[ v ]);
- }
- }
- inline void calcFIRFilterResponse( const double* flt, int fltlen,
- const double th, double& re0, double& im0, const int fltlat = 0 )
- {
- const double sincr = 2.0 * cos( th );
- double cvalue1;
- double svalue1;
- if( fltlat == 0 )
- {
- cvalue1 = 1.0;
- svalue1 = 0.0;
- }
- else
- {
- cvalue1 = cos( -fltlat * th );
- svalue1 = sin( -fltlat * th );
- }
- double cvalue2 = cos( -( fltlat + 1 ) * th );
- double svalue2 = sin( -( fltlat + 1 ) * th );
- double re = 0.0;
- double im = 0.0;
- while( fltlen > 0 )
- {
- re += cvalue1 * flt[ 0 ];
- im += svalue1 * flt[ 0 ];
- flt++;
- fltlen--;
- double tmp = cvalue1;
- cvalue1 = sincr * cvalue1 - cvalue2;
- cvalue2 = tmp;
- tmp = svalue1;
- svalue1 = sincr * svalue1 - svalue2;
- svalue2 = tmp;
- }
- re0 = re;
- im0 = im;
- }
- inline void calcFIRFilterResponseAndGroupDelay( const double* const flt,
- const int fltlen, const double th, double& re, double& im, double& gd )
- {
-
- calcFIRFilterResponse( flt, fltlen, th, re, im );
-
- const int Count = 2;
- const double thd2 = 1e-9;
- double ths[ Count ] = { th - thd2, th + thd2 };
- if( ths[ 0 ] < 0.0 )
- {
- ths[ 0 ] = 0.0;
- }
- if( ths[ 1 ] > R8B_PI )
- {
- ths[ 1 ] = R8B_PI;
- }
- double ph1[ Count ];
- int i;
- for( i = 0; i < Count; i++ )
- {
- double re1;
- double im1;
- calcFIRFilterResponse( flt, fltlen, ths[ i ], re1, im1 );
- ph1[ i ] = atan2( im1, re1 );
- }
- if( fabs( ph1[ 1 ] - ph1[ 0 ]) > R8B_PI )
- {
- if( ph1[ 1 ] > ph1[ 0 ])
- {
- ph1[ 1 ] -= R8B_2PI;
- }
- else
- {
- ph1[ 1 ] += R8B_2PI;
- }
- }
- const double thd = ths[ 1 ] - ths[ 0 ];
- gd = ( ph1[ 1 ] - ph1[ 0 ]) / thd;
- }
- inline void normalizeFIRFilter( double* const p, const int l,
- const double DCGain, const int pstep = 1 )
- {
- R8BASSERT( l > 0 );
- R8BASSERT( pstep != 0 );
- double s = 0.0;
- double* pp = p;
- int i = l;
- while( i > 0 )
- {
- s += *pp;
- pp += pstep;
- i--;
- }
- s = DCGain / s;
- pp = p;
- i = l;
- while( i > 0 )
- {
- *pp *= s;
- pp += pstep;
- i--;
- }
- }
- inline void calcSpline3p8Coeffs( double* const c, const double xm3,
- const double xm2, const double xm1, const double x0, const double x1,
- const double x2, const double x3, const double x4 )
- {
- c[ 0 ] = x0;
- c[ 1 ] = ( 61.0 * ( x1 - xm1 ) + 16.0 * ( xm2 - x2 ) +
- 3.0 * ( x3 - xm3 )) / 76.0;
- c[ 2 ] = ( 106.0 * ( xm1 + x1 ) + 10.0 * x3 + 6.0 * xm3 - 3.0 * x4 -
- 29.0 * ( xm2 + x2 ) - 167.0 * x0 ) / 76.0;
- c[ 3 ] = ( 91.0 * ( x0 - x1 ) + 45.0 * ( x2 - xm1 ) +
- 13.0 * ( xm2 - x3 ) + 3.0 * ( x4 - xm3 )) / 76.0;
- }
- inline void calcSpline2p8Coeffs( double* const c, const double xm3,
- const double xm2, const double xm1, const double x0, const double x1,
- const double x2, const double x3, const double x4 )
- {
- c[ 0 ] = x0;
- c[ 1 ] = ( 61.0 * ( x1 - xm1 ) + 16.0 * ( xm2 - x2 ) +
- 3.0 * ( x3 - xm3 )) / 76.0;
- c[ 2 ] = ( 106.0 * ( xm1 + x1 ) + 10.0 * x3 + 6.0 * xm3 - 3.0 * x4 -
- 29.0 * ( xm2 + x2 ) - 167.0 * x0 ) / 76.0;
- }
- inline void calcSpline3p4Coeffs( double* const c, const double* const y )
- {
- c[ 0 ] = y[ 1 ];
- c[ 1 ] = 0.5 * ( y[ 2 ] - y[ 0 ]);
- c[ 2 ] = y[ 0 ] - 2.5 * y[ 1 ] + y[ 2 ] + y[ 2 ] - 0.5 * y[ 3 ];
- c[ 3 ] = 0.5 * ( y[ 3 ] - y[ 0 ] ) + 1.5 * ( y[ 1 ] - y[ 2 ]);
- }
- inline void calcSpline3p6Coeffs( double* const c, const double* const y )
- {
- c[ 0 ] = y[ 2 ];
- c[ 1 ] = ( 11.0 * ( y[ 3 ] - y[ 1 ]) + 2.0 * ( y[ 0 ] - y[ 4 ])) / 14.0;
- c[ 2 ] = ( 20.0 * ( y[ 1 ] + y[ 3 ]) + 2.0 * y[ 5 ] - 4.0 * y[ 0 ] -
- 7.0 * y[ 4 ] - 31.0 * y[ 2 ]) / 14.0;
- c[ 3 ] = ( 17.0 * ( y[ 2 ] - y[ 3 ]) + 9.0 * ( y[ 4 ] - y[ 1 ]) +
- 2.0 * ( y[ 0 ] - y[ 5 ])) / 14.0;
- }
- #if !defined( min )
- template< typename T >
- inline T min( const T& v1, const T& v2 )
- {
- return( v1 < v2 ? v1 : v2 );
- }
- #endif
- #if !defined( max )
- template< typename T >
- inline T max( const T& v1, const T& v2 )
- {
- return( v1 > v2 ? v1 : v2 );
- }
- #endif
- inline double clampr( const double Value, const double minv,
- const double maxv )
- {
- if( Value < minv )
- {
- return( minv );
- }
- else
- if( Value > maxv )
- {
- return( maxv );
- }
- else
- {
- return( Value );
- }
- }
- inline double sqr( const double x )
- {
- return( x * x );
- }
- inline double pows( const double v, const double p )
- {
- return( v < 0.0 ? -pow( -v, p ) : pow( v, p ));
- }
- inline double gauss( const double v )
- {
- return( exp( -( v * v )));
- }
- inline double asinh( const double v )
- {
- return( log( v + sqrt( v * v + 1.0 )));
- }
- inline double besselI0( const double x )
- {
- const double ax = fabs( x );
- double y;
- if( ax < 3.75 )
- {
- y = x / 3.75;
- y *= y;
- return( 1.0 + y * ( 3.5156229 + y * ( 3.0899424 + y * ( 1.2067492 +
- y * ( 0.2659732 + y * ( 0.360768e-1 + y * 0.45813e-2 ))))));
- }
- y = 3.75 / ax;
- return( exp( ax ) / sqrt( ax ) * ( 0.39894228 + y * ( 0.1328592e-1 +
- y * ( 0.225319e-2 + y * ( -0.157565e-2 + y * ( 0.916281e-2 +
- y * ( -0.2057706e-1 + y * ( 0.2635537e-1 + y * ( -0.1647633e-1 +
- y * 0.392377e-2 )))))))));
- }
- }
- #endif
|