cwVectOps.h : Many additions and changes.

This commit is contained in:
kevin 2021-08-15 16:00:40 -04:00
parent 297ab58212
commit a5c7f41ecf

View File

@ -11,13 +11,37 @@ namespace cw
// //
template< typename T0 > template< typename T0 >
void print( const T0* v0, unsigned n, const char* fmt, const char* label=nullptr ) void print( const T0* v0, unsigned n, const char* fmt, const char* label=nullptr, unsigned colN=0 )
{ {
bool newline_fl = false;
if( label != nullptr ) if( label != nullptr )
{
printf("%s : ",label); printf("%s : ",label);
if( colN && n > colN )
{
printf("\n");
newline_fl = true;
}
}
if( colN == 0 )
colN = n;
for(unsigned i=0; i<n; ++i) for(unsigned i=0; i<n; ++i)
{
printf(fmt,v0[i]); printf(fmt,v0[i]);
newline_fl = false;
if( (n+1) % colN == 0 )
{
printf("\n");
newline_fl = true;
}
}
if( !newline_fl )
printf("\n"); printf("\n");
} }
@ -119,6 +143,7 @@ namespace cw
return acc; return acc;
} }
//================================================================================================================== //==================================================================================================================
// Arithmetic // Arithmetic
// //
@ -194,17 +219,17 @@ namespace cw
} }
template< typename T0, typename T1 > template< typename T0, typename T1 >
void div( T0* v0, const T1& scalar, unsigned n ) void div( T0* v0, const T1& denom, unsigned n )
{ {
for(unsigned i=0; i<n; ++i) for(unsigned i=0; i<n; ++i)
v0[i] /= scalar; v0[i] /= denom;
} }
template< typename T0, typename T1 > template< typename T0, typename T1 >
void div( T0* y0, const T0* v0, const T1& scalar, unsigned n ) void div( T0* y0, const T0* v0, const T1& denom, unsigned n )
{ {
for(unsigned i=0; i<n; ++i) for(unsigned i=0; i<n; ++i)
y0[i] = v0[i] / scalar; y0[i] = v0[i] / denom;
} }
template< typename T0, typename T1 > template< typename T0, typename T1 >
@ -240,7 +265,7 @@ namespace cw
// //
// Fill y[0:min(n,cnt)] with values {beg,beg+step,beg+2*step .... beg+(cnt-1)*step}} // Fill y[0:min(n,cnt)] with values {beg,beg+step,beg+2*step .... beg+(cnt-1)*step}}
template< typename T > template< typename T >
void seq( T* y, unsigned n, const T& beg, const T& cnt, const T& step=1 ) void seq( T* y, unsigned n, const T& beg, const T& cnt, const T& step )
{ {
if( cnt < n ) if( cnt < n )
n = cnt; n = cnt;
@ -260,6 +285,7 @@ namespace cw
return y; return y;
} }
// Fill y[0:dn] with [beg+0,beg+1, ... beg+dn]
template< typename T > template< typename T >
T seq( T* dbp, unsigned dn, const T& beg, const T& incr ) T seq( T* dbp, unsigned dn, const T& beg, const T& incr )
{ {
@ -271,7 +297,7 @@ namespace cw
} }
template< typename T > template< typename T >
T cumsum( const T* v, unsigned n ) T sum( const T* v, unsigned n )
{ {
T y = 0; T y = 0;
for(unsigned i=0; i<n; ++i) for(unsigned i=0; i<n; ++i)
@ -281,7 +307,7 @@ namespace cw
} }
template< typename T > template< typename T >
T cumprod( const T* v, unsigned n ) T prod( const T* v, unsigned n )
{ {
T y = 1; T y = 1;
for(unsigned i=0; i<n; ++i) for(unsigned i=0; i<n; ++i)
@ -290,7 +316,28 @@ namespace cw
} }
template< typename T0, typename T1 >
T0 sum_sq_diff( const T0* v0, const T1* v1, unsigned n )
{
T0 sum = 0;
for(unsigned i=0; i<n; ++i)
sum += (v0[i]-v1[i]) * (v0[i]-v1[i]);
return sum;
}
//==================================================================================================================
// Statistics
//
template< typename T >
T mean( const T* v, unsigned n )
{
if( n == 0 )
return 0;
return sum(v,n)/n;
}
//================================================================================================================== //==================================================================================================================
// Signal Processing // Signal Processing
@ -316,6 +363,46 @@ namespace cw
return init_idx; return init_idx;
} }
template< typename T >
T* ampl_to_db( T* dbp, const T* sbp, unsigned dn, T minDb=-1000 )
{
T minVal = pow(10.0,minDb/20.0);
T* dp = dbp;
T* ep = dp + dn;
for(; dp<ep; ++dp,++sbp)
*dp = *sbp<minVal ? minDb : 20.0 * log10(*sbp);
return dbp;
}
template< typename T >
T* db_to_ampl( T* dbp, const T* sbp, unsigned dn, T minDb=-1000 )
{
T* dp = dbp;
T* ep = dp + dn;
for(; dp<ep; ++dp,++sbp)
*dp = pow(10.0,*sbp/20.0);
return dbp;
}
template< typename T >
T rms( const T* x, unsigned xN )
{
T rms = 0;
if( xN > 0 )
{
T x0[ xN ];
mul(x0,x,x,xN);
rms = std::sqrt(sum(x0,xN)/(T)xN);
}
return rms;
}
} }