2
0
Fork 0
OB-Xd/Modules/gin/3rdparty/avir/avir.h

6199 lines
186 KiB
C++
Executable File

//$ nobt
//$ nocpp
/**
* @file avir.h
*
* @brief The "main" inclusion file with all required classes and functions.
*
* This is the "main" inclusion file for the "AVIR" image resizer. This
* inclusion file contains implementation of the AVIR image resizing algorithm
* in its entirety. Also includes several classes and functions that can be
* useful elsewhere.
*
* AVIR Copyright (c) 2015-2019 Aleksey Vaneev
*
* @mainpage
*
* @section intro_sec Introduction
*
* Description is available at https://github.com/avaneev/avir
*
* AVIR is devoted to women. Your digital photos can look good at any size!
*
* @section license License
*
* AVIR License Agreement
*
* The MIT License (MIT)
*
* Copyright (c) 2015-2019 Aleksey Vaneev
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*
* Please credit the author of this library in your documentation in the
* following way: "AVIR image resizing algorithm designed by Aleksey Vaneev"
*
* @version 2.4
*/
#ifndef AVIR_CIMAGERESIZER_INCLUDED
#define AVIR_CIMAGERESIZER_INCLUDED
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
namespace avir {
/**
* The macro defines AVIR version string.
*/
#define AVIR_VERSION "2.4"
/**
* The macro equals to "pi" constant, fills 53-bit floating point mantissa.
* Undefined at the end of file.
*/
#define AVIR_PI 3.1415926535897932
/**
* The macro equals to "pi divided by 2" constant, fills 53-bit floating
* point mantissa. Undefined at the end of file.
*/
#define AVIR_PId2 1.5707963267948966
/**
* Rounding function, based on the (int) typecast. Biased result. Not suitable
* for numbers >= 2^31.
*
* @param d Value to round.
* @return Rounded value. Some bias may be introduced.
*/
template< class T >
inline T round( const T d )
{
return( d < 0.0 ? -(T) (int) ( (T) 0.5 - d ) : (T) (int) ( d + (T) 0.5 ));
}
/**
* Template function "clamps" (clips) the specified value so that it is not
* lesser than "minv", and not greater than "maxv".
*
* @param Value Value to clamp.
* @param minv Minimal allowed value.
* @param maxv Maximal allowed value.
* @return The clamped value.
*/
template< class T >
inline T clamp( const T& Value, const T minv, const T maxv )
{
if( Value < minv )
{
return( minv );
}
else
if( Value > maxv )
{
return( maxv );
}
else
{
return( Value );
}
}
/**
* Power 2.4 approximation function, designed for sRGB gamma correction.
*
* @param x Argument, in the range 0.09 to 1.
* @return Value raised into power 2.4, approximate.
*/
template< class T >
inline T pow24_sRGB( const T x )
{
const double x2 = x * x;
const double x3 = x2 * x;
const double x4 = x2 * x2;
return( (T) ( 0.0985766365536824 + 0.839474952656502 * x2 +
0.363287814061725 * x3 - 0.0125559718896615 /
( 0.12758338921578 + 0.290283465468235 * x ) -
0.231757513261358 * x - 0.0395365717969074 * x4 ));
}
/**
* Power 1/2.4 approximation function, designed for sRGB gamma correction.
*
* @param x Argument, in the range 0.003 to 1.
* @return Value raised into power 1/2.4, approximate.
*/
template< class T >
inline T pow24i_sRGB( const T x )
{
const double sx = sqrt( x );
const double ssx = sqrt( sx );
const double sssx = sqrt( ssx );
return( (T) ( 0.000213364515060263 + 0.0149409239419218 * x +
0.433973412731747 * sx + ssx * ( 0.659628181609715 * sssx -
0.0380957908841466 - 0.0706476137208521 * sx )));
}
/**
* Function approximately linearizes the sRGB gamma value.
*
* @param s sRGB gamma value, in the range 0 to 1.
* @return Linearized sRGB gamma value, approximated.
*/
template< class T >
inline T convertSRGB2Lin( const T s )
{
const T a = (T) 0.055;
if( s <= (T) 0.04045 )
{
return( s / (T) 12.92 );
}
return( pow24_sRGB(( s + a ) / ( (T) 1 + a )));
}
/**
* Function approximately de-linearizes the linear gamma value.
*
* @param s Linear gamma value, in the range 0 to 1.
* @return sRGB gamma value, approximated.
*/
template< class T >
inline T convertLin2SRGB( const T s )
{
const T a = (T) 0.055;
if( s <= (T) 0.0031308 )
{
return( (T) 12.92 * s );
}
return(( (T) 1 + a ) * pow24i_sRGB( s ) - a );
}
/**
* Function converts (via typecast) specified array of type T1 values of
* length l into array of type T2 values. If T1 is the same as T2, copy
* operation is performed. When copying data at overlapping address spaces,
* "op" should be lower than "ip".
*
* @param ip Input buffer.
* @param[out] op Output buffer.
* @param l The number of elements to copy.
* @param ip Input buffer pointer increment.
* @param op Output buffer pointer increment.
*/
template< class T1, class T2 >
inline void copyArray( const T1* ip, T2* op, int l,
const int ipinc = 1, const int opinc = 1 )
{
while( l > 0 )
{
*op = (T2) *ip;
op += opinc;
ip += ipinc;
l--;
}
}
/**
* Function adds values located in array "ip" to array "op".
*
* @param ip Input buffer.
* @param[out] op Output buffer.
* @param l The number of elements to add.
* @param ip Input buffer pointer increment.
* @param op Output buffer pointer increment.
*/
template< class T1, class T2 >
inline void addArray( const T1* ip, T2* op, int l,
const int ipinc = 1, const int opinc = 1 )
{
while( l > 0 )
{
*op += *ip;
op += opinc;
ip += ipinc;
l--;
}
}
/**
* Function that replicates a set of adjacent elements several times in a row.
* This operation is usually used to replicate pixels at the start or end of
* image's scanline.
*
* @param ip Source array.
* @param ipl Source array length (usually 1..4, but can be any number).
* @param[out] op Destination buffer.
* @param l Number of times the source array should be replicated (the
* destination buffer should be able to hold ipl * l number of elements).
* @param opinc Destination buffer position increment after replicating the
* source array. This value should be equal to at least ipl.
*/
template< class T1, class T2 >
inline void replicateArray( const T1* const ip, const int ipl, T2* op, int l,
const int opinc )
{
if( ipl == 1 )
{
while( l > 0 )
{
op[ 0 ] = ip[ 0 ];
op += opinc;
l--;
}
}
else
if( ipl == 4 )
{
while( l > 0 )
{
op[ 0 ] = ip[ 0 ];
op[ 1 ] = ip[ 1 ];
op[ 2 ] = ip[ 2 ];
op[ 3 ] = ip[ 3 ];
op += opinc;
l--;
}
}
else
if( ipl == 3 )
{
while( l > 0 )
{
op[ 0 ] = ip[ 0 ];
op[ 1 ] = ip[ 1 ];
op[ 2 ] = ip[ 2 ];
op += opinc;
l--;
}
}
else
if( ipl == 2 )
{
while( l > 0 )
{
op[ 0 ] = ip[ 0 ];
op[ 1 ] = ip[ 1 ];
op += opinc;
l--;
}
}
else
{
while( l > 0 )
{
int i;
for( i = 0; i < ipl; i++ )
{
op[ i ] = ip[ i ];
}
op += opinc;
l--;
}
}
}
/**
* Function calculates frequency response of the specified FIR filter at the
* specified circular frequency. Phase can be calculated as atan2( im, re ).
* Function uses computationally-efficient oscillators instead of "cos" and
* "sin" functions.
*
* @param flt FIR filter's coefficients.
* @param fltlen Number of coefficients (taps) in the filter.
* @param th Circular frequency [0; pi].
* @param[out] re0 Resulting real part of the complex frequency response.
* @param[out] im0 Resulting imaginary part of the complex frequency response.
* @param fltlat Filter's latency in samples (taps).
*/
template< class T >
inline void calcFIRFilterResponse( const T* 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;
}
/**
* Function normalizes FIR filter so that its frequency response at DC is
* equal to DCGain.
*
* @param[in,out] p Filter coefficients.
* @param l Filter length.
* @param DCGain Filter's gain at DC.
* @param pstep "p" array step.
*/
template< class T >
inline void normalizeFIRFilter( T* const p, const int l, const double DCGain,
const int pstep = 1 )
{
double s = 0.0;
T* pp = p;
int i = l;
while( i > 0 )
{
s += *pp;
pp += pstep;
i--;
}
s = DCGain / s;
pp = p;
i = l;
while( i > 0 )
{
*pp = (T) ( *pp * s );
pp += pstep;
i--;
}
}
/**
* @brief Memory buffer class for element array storage, with capacity
* tracking.
*
* Allows easier handling of memory blocks allocation and automatic
* deallocation for arrays (buffers) consisting of elements of specified
* class. Tracks buffer's capacity in "int" variable; unsuitable for
* allocation of very large memory blocks (with more than 2 billion elements).
*
* This class manages memory space only - it does not perform element class
* construction (initialization) operations. Buffer's required memory address
* alignment specification is supported.
*
* Uses standard library to allocate and deallocate memory.
*
* @tparam T Buffer element's type.
* @tparam capint Buffer capacity's type to use. Use size_t for large buffers.
*/
template< class T, typename capint = int >
class CBuffer
{
public:
CBuffer()
: Data( nullptr )
, DataAligned( nullptr )
, Capacity( 0 )
, Alignment( 0 )
{
}
/**
* Constructor creates the buffer with the specified capacity.
*
* @param aCapacity Buffer's capacity.
* @param aAlignment Buffer's required memory address alignment. 0 - use
* stdlib's default alignment.
*/
CBuffer( const capint aCapacity, const int aAlignment = 0 )
{
allocinit( aCapacity, aAlignment );
}
CBuffer( const CBuffer& Source )
{
allocinit( Source.Capacity, Source.Alignment );
memcpy( DataAligned, Source.DataAligned, Capacity * sizeof( T ));
}
~CBuffer()
{
freeData();
}
CBuffer& operator = ( const CBuffer& Source )
{
alloc( Source.Capacity, Source.Alignment );
memcpy( DataAligned, Source.DataAligned, Capacity * sizeof( T ));
return( *this );
}
/**
* Function allocates memory so that the specified number of elements
* can be stored in *this buffer object.
*
* @param aCapacity Storage for this number of elements to allocate.
* @param aAlignment Buffer's required memory address alignment,
* power-of-2 values only. 0 - use stdlib's default alignment.
*/
void alloc( const capint aCapacity, const int aAlignment = 0 )
{
freeData();
allocinit( aCapacity, aAlignment );
}
/**
* Function deallocates any previously allocated buffer.
*/
void free()
{
freeData();
Data = nullptr;
DataAligned = nullptr;
Capacity = 0;
Alignment = 0;
}
/**
* @return The capacity of the element buffer.
*/
capint getCapacity() const
{
return( Capacity );
}
/**
* Function "forces" *this buffer to have an arbitary capacity. Calling
* this function invalidates all further operations except deleting *this
* object. This function should not be usually used at all. Function can
* be used to "model" certain buffer capacity without calling a costly
* memory allocation function.
*
* @param NewCapacity A new "forced" capacity.
*/
void forceCapacity( const capint NewCapacity )
{
Capacity = NewCapacity;
}
/**
* Function reallocates *this buffer to a larger size so that it will be
* able to hold the specified number of elements. Downsizing is not
* performed. Alignment is not changed.
*
* @param NewCapacity New (increased) capacity.
* @param DoDataCopy "True" if data in the buffer should be retained.
*/
void increaseCapacity( const capint NewCapacity,
const bool DoDataCopy = true )
{
if( NewCapacity < Capacity )
{
return;
}
if( DoDataCopy )
{
const capint PrevCapacity = Capacity;
T* const PrevData = Data;
T* const PrevDataAligned = DataAligned;
allocinit( NewCapacity, Alignment );
memcpy( DataAligned, PrevDataAligned, PrevCapacity * sizeof( T ));
:: free( PrevData );
}
else
{
:: free( Data );
allocinit( NewCapacity, Alignment );
}
}
/**
* Function "truncates" (reduces) capacity of the buffer without
* reallocating it. Alignment is not changed.
*
* @param NewCapacity New required capacity.
*/
void truncateCapacity( const capint NewCapacity )
{
if( NewCapacity >= Capacity )
{
return;
}
Capacity = NewCapacity;
}
/**
* Function increases capacity so that the specified number of
* elements can be stored. This function increases the previous capacity
* value by third the current capacity value until space for the required
* number of elements is available. Alignment is not changed.
*
* @param ReqCapacity Required capacity.
*/
void updateCapacity( const capint ReqCapacity )
{
if( ReqCapacity <= Capacity )
{
return;
}
capint NewCapacity = Capacity;
while( NewCapacity < ReqCapacity )
{
NewCapacity += NewCapacity / 3 + 1;
}
increaseCapacity( NewCapacity );
}
operator T* () const
{
return( DataAligned );
}
private:
T* Data; ///< Element buffer pointer.
///<
T* DataAligned; ///< Memory address-aligned element buffer pointer.
///<
capint Capacity; ///< Element buffer capacity.
///<
int Alignment; ///< Memory address alignment in use. 0 - use stdlib's
///< default alignment.
///<
/**
* Internal element buffer allocation function used during object
* construction.
*
* @param aCapacity Storage for this number of elements to allocate.
* @param aAlignment Buffer's required memory address alignment. 0 - use
* stdlib's default alignment.
*/
void allocinit( const capint aCapacity, const int aAlignment )
{
if( aAlignment == 0 )
{
Data = (T*) :: malloc( aCapacity * sizeof( T ));
DataAligned = Data;
Alignment = 0;
}
else
{
Data = (T*) :: malloc( aCapacity * sizeof( T ) + aAlignment );
DataAligned = alignptr( Data, aAlignment );
Alignment = aAlignment;
}
Capacity = aCapacity;
}
/**
* Function frees a previously allocated Data buffer.
*/
void freeData()
{
:: free( Data );
}
/**
* Function modifies the specified pointer so that it becomes memory
* address-aligned.
*
* @param ptr Pointer to align.
* @param align Alignment in bytes to apply.
* @return Pointer aligned to align bytes. Works with power-of-2
* alignments only. If no alignment is necessary, "align" bytes will be
* added to the pointer value.
*/
template< class Tp >
inline Tp alignptr( const Tp ptr, const uintptr_t align )
{
return( (Tp) ( (uintptr_t) ptr + align -
( (uintptr_t) ptr & ( align - 1 ))) );
}
};
/**
* Function optimizes the length of the symmetric-odd FIR filter by removing
* left- and rightmost elements that are below specific threshold.
*
* Synthetic test shows that filter gets optimized in 2..3% of cases and in
* each such case optimization reduces filter length by 6..8%. Optimization,
* however, may skew the results of algorithm modeling and complexity
* calculation leading to a choice of a less optimal algorithm.
*
* @param[in,out] Flt Buffer that contains filter being optimized.
* @param[in,out] FltLatency Variable that holds the current latency of the
* filter. May be adjusted on function return.
* @param Threshold Threshold level.
*/
template< class T >
inline void optimizeFIRFilter( CBuffer< T >& Flt, int& FltLatency,
T const Threshold = (T) 0.00001 )
{
int i;
// Optimize length.
for( i = 0; i <= FltLatency; i++ )
{
if( fabs( Flt[ i ]) >= Threshold || i == FltLatency )
{
if( i > 0 )
{
const int NewCapacity = Flt.getCapacity() - i * 2;
copyArray( &Flt[ i ], &Flt[ 0 ], NewCapacity );
Flt.truncateCapacity( NewCapacity );
FltLatency -= i;
}
break;
}
}
}
/**
* @brief Array of structured objects.
*
* Implements allocation of a linear array of objects of class T (which are
* initialized), addressable via operator[]. Each object is created via the
* "operator new". New object insertions are quick since implementation uses
* prior space allocation (capacity), thus not requiring frequent memory block
* reallocations.
*
* @tparam T Array element's type.
*/
template< class T >
class CStructArray
{
public:
CStructArray()
: ItemCount( 0 )
{
}
CStructArray( const CStructArray& Source )
: ItemCount( 0 )
, Items( Source.getItemCount() )
{
while( ItemCount < Source.getItemCount() )
{
Items[ ItemCount ] = new T( Source[ ItemCount ]);
ItemCount++;
}
}
~CStructArray()
{
clear();
}
CStructArray& operator = ( const CStructArray& Source )
{
clear();
const int NewCount = Source.ItemCount;
Items.updateCapacity( NewCount );
while( ItemCount < NewCount )
{
Items[ ItemCount ] = new T( Source[ ItemCount ]);
ItemCount++;
}
return( *this );
}
T& operator []( const int Index )
{
return( *Items[ Index ]);
}
const T& operator []( const int Index ) const
{
return( *Items[ Index ]);
}
/**
* Function creates a new object of type T with the default constructor
* and adds this object to the array.
*
* @return Reference to a newly added object.
*/
T& add()
{
if( ItemCount == Items.getCapacity() )
{
Items.increaseCapacity( ItemCount * 3 / 2 + 1 );
}
Items[ ItemCount ] = new T();
ItemCount++;
return( (*this)[ ItemCount - 1 ]);
}
/**
* Function changes number of allocated items. New items are created with
* the default constructor. If NewCount is below the current item count,
* items that are above NewCount range will be destructed.
*
* @param NewCount New requested item count.
*/
void setItemCount( const int NewCount )
{
if( NewCount > ItemCount )
{
Items.increaseCapacity( NewCount );
while( ItemCount < NewCount )
{
Items[ ItemCount ] = new T();
ItemCount++;
}
}
else
{
while( ItemCount > NewCount )
{
ItemCount--;
delete Items[ ItemCount ];
}
}
}
/**
* Function erases all items of *this array.
*/
void clear()
{
while( ItemCount > 0 )
{
ItemCount--;
delete Items[ ItemCount ];
}
}
/**
* @return The number of allocated items.
*/
int getItemCount() const
{
return( ItemCount );
}
private:
int ItemCount; ///< The number of items available in the array.
///<
CBuffer< T* > Items; ///< Element buffer.
///<
};
/**
* @brief Sine signal generator class.
*
* Class implements sine signal generator without biasing, with
* constructor-based initalization only. This generator uses oscillator
* instead of "sin" function.
*/
class CSineGen
{
public:
/**
* Constructor initializes *this sine signal generator.
*
* @param si Sine function increment, in radians.
* @param ph Starting phase, in radians. Add 0.5 * AVIR_PI for cosine
* function.
*/
CSineGen( const double si, const double ph )
: svalue1( sin( ph ))
, svalue2( sin( ph - si ))
, sincr( 2.0 * cos( si ))
{
}
/**
* @return The next value of the sine function, without biasing.
*/
double generate()
{
const double res = svalue1;
svalue1 = sincr * res - svalue2;
svalue2 = res;
return( res );
}
private:
double svalue1; ///< Current sine value.
///<
double svalue2; ///< Previous sine value.
///<
double sincr; ///< Sine value increment.
///<
};
/**
* @brief Peaked Cosine window function generator class.
*
* Class implements Peaked Cosine window function generator. Generates the
* right-handed half of the window function. The Alpha parameter of this
* window function offers the control of the balance between the early and
* later taps of the filter. E.g. at Alpha=1 both early and later taps are
* attenuated, but at Alpha=4 mostly later taps are attenuated. This offers a
* great control over ringing artifacts produced by a low-pass filter in image
* processing, without compromising achieved image sharpness.
*/
class CDSPWindowGenPeakedCosine
{
public:
/**
* Constructor initializes *this window function generator.
*
* @param aAlpha Alpha parameter, affects the peak shape (peak
* augmentation) of the window function. Should be >= 1.0.
* @param aLen2 Half filter's length (non-truncated).
*/
CDSPWindowGenPeakedCosine( const double aAlpha, const double aLen2 )
: Alpha( aAlpha )
, Len2( aLen2 )
, wn( 0 )
, w1( AVIR_PId2 / Len2, AVIR_PI * 0.5 )
{
}
/**
* @return The next Peaked Cosine window function coefficient.
*/
double generate()
{
const double h = pow( wn / Len2, Alpha );
wn++;
return( w1.generate() * ( 1.0 - h ));
}
private:
double Alpha; ///< Alpha parameter, affects the peak shape of window.
///<
double Len2; ///< Half length of the window function.
///<
int wn; ///< Window function integer position. 0 - center of the
///< window function.
///<
CSineGen w1; ///< Sine-wave generator.
///<
};
/**
* @brief FIR filter-based equalizer generator.
*
* Class implements an object used to generate symmetric-odd FIR filters with
* the specified frequency response (aka paragraphic equalizer). The
* calculated filter is windowed by the Peaked Cosine window function.
*
* In image processing, due to short length of filters being used (6-8 taps)
* the resulting frequency response of the filter is approximate and may be
* mathematically imperfect, but still adequate to the visual requirements.
*
* On a side note, this equalizer generator can be successfully used for audio
* signal equalization as well: for example, it is used in almost the same
* form in Voxengo Marvel GEQ equalizer plug-in.
*
* Filter generation is based on decomposition of frequency range into
* spectral bands, with each band represented by linear and ramp "kernels".
* When the filter is built, these kernels are combined together with
* different weights that approximate the required frequency response.
*/
class CDSPFIREQ
{
public:
/**
* Function initializes *this object with the required parameters. The
* gain of frequencies beyond the MinFreq..MaxFreq range are controlled by
* the first and the last band's gain.
*
* @param SampleRate Processing sample rate (use 2 for image processing).
* @param aFilterLength Required filter length in samples (taps). The
* actual filter length is truncated to an integer value.
* @param aBandCount Number of band crossover points required to control,
* including bands at MinFreq and MaxFreq.
* @param MinFreq Minimal frequency that should be controlled.
* @param MaxFreq Maximal frequency that should be controlled.
* @param IsLogBands "True" if the bands should be spaced logarithmically.
* @param WFAlpha Peaked Cosine window function's Alpha parameter.
*/
void init( const double SampleRate, const double aFilterLength,
const int aBandCount, const double MinFreq, const double MaxFreq,
const bool IsLogBands, const double WFAlpha )
{
FilterLength = aFilterLength;
BandCount = aBandCount;
CenterFreqs.alloc( BandCount );
z = (int) ceil( FilterLength * 0.5 );
zi = z + ( z & 1 );
z2 = z * 2;
CBuffer< double > oscbuf( z2 );
initOscBuf( oscbuf );
CBuffer< double > winbuf( z );
initWinBuf( winbuf, WFAlpha );
UseFirstVirtBand = ( MinFreq > 0.0 );
const int k = zi * ( BandCount + ( UseFirstVirtBand ? 1 : 0 ));
Kernels1.alloc( k );
Kernels2.alloc( k );
double m; // Frequency step multiplier.
double mo; // Frequency step offset (addition).
if( IsLogBands )
{
m = exp( log( MaxFreq / MinFreq ) / ( BandCount - 1 ));
mo = 0.0;
}
else
{
m = 1.0;
mo = ( MaxFreq - MinFreq ) / ( BandCount - 1 );
}
double f = MinFreq;
double x1 = 0.0;
double x2;
int si;
if( UseFirstVirtBand )
{
si = 0;
}
else
{
si = 1;
CenterFreqs[ 0 ] = 0.0;
f = f * m + mo;
}
double* kernbuf1 = &Kernels1[ 0 ];
double* kernbuf2 = &Kernels2[ 0 ];
int i;
for( i = si; i < BandCount; i++ )
{
x2 = f * 2.0 / SampleRate;
CenterFreqs[ i ] = x2;
fillBandKernel( x1, x2, kernbuf1, kernbuf2, oscbuf, winbuf );
kernbuf1 += zi;
kernbuf2 += zi;
x1 = x2;
f = f * m + mo;
}
if( x1 < 1.0 )
{
UseLastVirtBand = true;
fillBandKernel( x1, 1.0, kernbuf1, kernbuf2, oscbuf, winbuf );
}
else
{
UseLastVirtBand = false;
}
}
/**
* @return Filter's length, in samples (taps).
*/
int getFilterLength() const
{
return( z2 - 1 );
}
/**
* @return Filter's latency (group delay), in samples (taps).
*/
int getFilterLatency() const
{
return( z - 1 );
}
/**
* Function creates symmetric-odd FIR filter with the specified gain
* levels at band crossover points.
*
* @param BandGains Array of linear gain levels, count=BandCount specified
* in the init() function.
* @param[out] Filter Output filter buffer, length = getFilterLength().
*/
void buildFilter( const double* const BandGains, double* const Filter )
{
const double* kernbuf1 = &Kernels1[ 0 ];
const double* kernbuf2 = &Kernels2[ 0 ];
double x1 = 0.0;
double y1 = BandGains[ 0 ];
double x2;
double y2;
int i;
int si;
if( UseFirstVirtBand )
{
si = 1;
x2 = CenterFreqs[ 0 ];
y2 = y1;
}
else
{
si = 2;
x2 = CenterFreqs[ 1 ];
y2 = BandGains[ 1 ];
}
copyBandKernel( Filter, kernbuf1, kernbuf2, y1 - y2,
x1 * y2 - x2 * y1 );
kernbuf1 += zi;
kernbuf2 += zi;
x1 = x2;
y1 = y2;
for( i = si; i < BandCount; i++ )
{
x2 = CenterFreqs[ i ];
y2 = BandGains[ i ];
addBandKernel( Filter, kernbuf1, kernbuf2, y1 - y2,
x1 * y2 - x2 * y1 );
kernbuf1 += zi;
kernbuf2 += zi;
x1 = x2;
y1 = y2;
}
if( UseLastVirtBand )
{
addBandKernel( Filter, kernbuf1, kernbuf2, y1 - y2,
x1 * y2 - y1 );
}
for( i = 0; i < z - 1; i++ )
{
Filter[ z + i ] = Filter[ z - 2 - i ];
}
}
/**
* Function calculates filter's length (in samples) and latency depending
* on the required non-truncated filter length.
*
* @param aFilterLength Required filter length in samples (non-truncated).
* @param[out] Latency Resulting latency (group delay) of the filter,
* in samples (taps).
* @return Filter length in samples (taps).
*/
static int calcFilterLength( const double aFilterLength, int& Latency )
{
const int l = (int) ceil( aFilterLength * 0.5 );
Latency = l - 1;
return( l * 2 - 1 );
}
private:
double FilterLength; ///< Length of filter.
///<
int z; ///< Equals (int) ceil( FilterLength * 0.5 ).
///<
int zi; ///< Equals "z" if z is even, or z + 1 if z is odd. Used as a
///< Kernels1 and Kernels2 size multiplier and kernel buffer increment
///< to make sure each kernel buffer is 16-byte aligned.
///<
int z2; ///< Equals z * 2.
///<
int BandCount; ///< Number of controllable bands.
///<
CBuffer< double > CenterFreqs; ///< Center frequencies for all bands,
///< normalized to 0.0-1.0 range.
///<
CBuffer< double > Kernels1; ///< Half-length kernel buffers for each
///< spectral band (linear part).
///<
CBuffer< double > Kernels2; ///< Half-length kernel buffers for each
///< spectral band (ramp part).
///<
bool UseFirstVirtBand; ///< "True" if the first virtual band
///< (between 0.0 and MinFreq) should be used. The first virtual band
///< won't be used if MinFreq equals 0.0.
///<
bool UseLastVirtBand; ///< "True" if the last virtual band (between
///< MaxFreq and SampleRate * 0.5) should be used. The last virtual
///< band won't be used if MaxFreq * 2.0 equals SampleRate.
///<
/**
* Function initializes the "oscbuf" used in the fillBandKernel()
* function.
*
* @param oscbuf Oscillator buffer, length = z * 2.
*/
void initOscBuf( double* oscbuf ) const
{
int i = z;
while( i > 0 )
{
oscbuf[ 0 ] = 0.0;
oscbuf[ 1 ] = 1.0;
oscbuf += 2;
i--;
}
}
/**
* Function initializes window function buffer. This function generates
* Peaked Cosine window function.
*
* @param winbuf Windowing buffer.
* @param Alpha Peaked Cosine alpha parameter.
*/
void initWinBuf( double* winbuf, const double Alpha ) const
{
CDSPWindowGenPeakedCosine wf( Alpha, FilterLength * 0.5 );
int i;
for( i = 1; i <= z; i++ )
{
winbuf[ z - i ] = wf.generate();
}
}
/**
* Function fills first half of symmetric-odd FIR kernel for the band.
* This function should be called successively for adjacent bands.
* Previous band's x2 should be equal to current band's x1. A band kernel
* consists of 2 elements: linear kernel and ramp kernel.
*
* @param x1 Band's left corner frequency (0..1).
* @param x2 Band's right corner frequency (0..1).
* @param kernbuf1 Band kernel buffer 1 (linear part), length = z.
* @param kernbuf2 Band kernel buffer 2 (ramp part), length = z.
* @param oscbuf Oscillation buffer. Before the first call of the
* fillBandKernel() should be initialized with the call of the
* initOscBuf() function.
* @param winbuf Buffer that contains windowing function.
*/
void fillBandKernel( const double x1, const double x2, double* kernbuf1,
double* kernbuf2, double* oscbuf, const double* const winbuf )
{
const double s2_incr = AVIR_PI * x2;
const double s2_coeff = 2.0 * cos( s2_incr );
double s2_value1 = sin( s2_incr * ( -z + 1 ));
double c2_value1 = sin( s2_incr * ( -z + 1 ) + AVIR_PI * 0.5 );
oscbuf[ 0 ] = sin( s2_incr * -z );
oscbuf[ 1 ] = sin( s2_incr * -z + AVIR_PI * 0.5 );
int ks;
for( ks = 1; ks < z; ks++ )
{
const int ks2 = ks * 2;
const double s1_value1 = oscbuf[ ks2 ];
const double c1_value1 = oscbuf[ ks2 + 1 ];
oscbuf[ ks2 ] = s2_value1;
oscbuf[ ks2 + 1 ] = c2_value1;
const double x = AVIR_PI * ( ks - z );
const double v0 = winbuf[ ks - 1 ] / (( x1 - x2 ) * x );
kernbuf1[ ks - 1 ] = ( x2 * s2_value1 - x1 * s1_value1 +
( c2_value1 - c1_value1 ) / x ) * v0;
kernbuf2[ ks - 1 ] = ( s2_value1 - s1_value1 ) * v0;
s2_value1 = s2_coeff * s2_value1 - oscbuf[ ks2 - 2 ];
c2_value1 = s2_coeff * c2_value1 - oscbuf[ ks2 - 1 ];
}
kernbuf1[ z - 1 ] = ( x2 * x2 - x1 * x1 ) / ( x1 - x2 ) * 0.5;
kernbuf2[ z - 1 ] = -1.0;
}
/**
* Function copies band kernel's elements to the output buffer.
*
* @param outbuf Output buffer.
* @param kernbuf1 Kernel buffer 1 (linear part).
* @param kernbuf2 Kernel buffer 2 (ramp part).
* @param c Multiplier for linear kernel element.
* @param d Multiplier for ramp kernel element.
*/
void copyBandKernel( double* outbuf, const double* const kernbuf1,
const double* const kernbuf2, const double c, const double d ) const
{
int ks;
for( ks = 0; ks < z; ks++ )
{
outbuf[ ks ] = c * kernbuf1[ ks ] + d * kernbuf2[ ks ];
}
}
/**
* Function adds band kernel's elements to the output buffer.
*
* @param outbuf Output buffer.
* @param kernbuf1 Kernel buffer 1 (linear part).
* @param kernbuf2 Kernel buffer 2 (ramp part).
* @param c Multiplier for linear kernel element.
* @param d Multiplier for ramp kernel element.
*/
void addBandKernel( double* outbuf, const double* const kernbuf1,
const double* const kernbuf2, const double c, const double d ) const
{
int ks;
for( ks = 0; ks < z; ks++ )
{
outbuf[ ks ] += c * kernbuf1[ ks ] + d * kernbuf2[ ks ];
}
}
};
/**
* @brief Low-pass filter windowed by Peaked Cosine window function.
*
* This class implements calculation of linear-phase symmetric-odd FIR
* low-pass filter windowed by the Peaked Cosine window function, for image
* processing applications.
*/
class CDSPPeakedCosineLPF
{
public:
int fl2; ///< Half filter's length, excluding the peak value. This value
///< can be also used as filter's latency in samples (taps).
///<
int FilterLen; ///< Filter's length in samples (taps).
///<
/**
* Constructor initalizes *this object.
*
* @param aLen2 Half-length (non-truncated) of low-pass filter, in samples
* (taps).
* @param aFreq2 Low-pass filter's corner frequency [0; pi].
* @param aAlpha Peaked Cosine window function Alpha parameter.
*/
CDSPPeakedCosineLPF( const double aLen2, const double aFreq2,
const double aAlpha )
: fl2( (int) ceil( aLen2 ) - 1 )
, FilterLen( fl2 + fl2 + 1 )
, Len2( aLen2 )
, Freq2( aFreq2 )
, Alpha( aAlpha )
{
}
/**
* Function generates a linear-phase low-pass filter windowed by Peaked
* Cosine window function.
*
* @param[out] op Output buffer, length = FilterLen (fl2 * 2 + 1).
* @param DCGain Required gain at DC. The resulting filter will be
* normalized to achieve this DC gain.
*/
template< class T >
void generateLPF( T* op, const double DCGain )
{
CDSPWindowGenPeakedCosine wf( Alpha, Len2 );
CSineGen f2( Freq2, 0.0 );
op += fl2;
T* op2 = op;
f2.generate();
int t = 1;
*op = (T) ( Freq2 * wf.generate() / AVIR_PI );
double s = *op;
while( t <= fl2 )
{
const double v = f2.generate() * wf.generate() / t / AVIR_PI;
op++;
op2--;
*op = (T) v;
*op2 = (T) v;
s += *op + *op2;
t++;
}
t = FilterLen;
s = DCGain / s;
while( t > 0 )
{
*op2 = (T) ( *op2 * s );
op2++;
t--;
}
}
private:
double Len2; ///< Half-length (non-truncated) of low-pass filter, in
///< samples (taps).
///<
double Freq2; ///< Low-pass filter's corner frequency.
///<
double Alpha; ///< Peaked Cosine window function Alpha parameter.
///<
};
/**
* @brief Buffer class for parametrized low-pass filter.
*
* This class extends the CBuffer< double > class by adding several variables
* that define a symmetric-odd FIR low-pass filter windowed by Peaked Cosine
* window function. This class can be used to compare filters without
* comparing their buffer contents.
*/
class CFltBuffer : public CBuffer< double >
{
public:
double Len2; ///< Half-length (non-truncated) of low-pass filters, in
///< samples (taps).
///<
double Freq; ///< Low-pass filter's corner frequency.
///<
double Alpha; ///< Peaked Cosine window function Alpha parameter.
///<
double DCGain; ///< DC gain applied to the filter.
///<
CFltBuffer()
: CBuffer< double >()
, Len2( 0.0 )
, Freq( 0.0 )
, Alpha( 0.0 )
, DCGain( 0.0 )
{
}
/**
* @param b2 Filter buffer to compare *this object to.
* @return Operator returns "true" if both filters have same parameters.
*/
bool operator == ( const CFltBuffer& b2 ) const
{
return( Len2 == b2.Len2 && Freq == b2.Freq && Alpha == b2.Alpha &&
DCGain == b2.DCGain );
}
};
/**
* @brief Sinc function-based fractional delay filter bank.
*
* Class implements storage and initialization of a bank of sinc
* function-based fractional delay filters, expressed as 1st order polynomial
* interpolation coefficients. The filters are produced from a single "long"
* windowed low-pass filter. Also supports 0th-order ("nearest neighbor")
* interpolation.
*
* This class also supports multiplication of each fractional delay filter by
* an external filter (usually a low-pass filter).
*
* @tparam fptype Specifies storage type of the filter coefficients bank. The
* filters are initially calculated using the "double" precision.
*/
template< class fptype >
class CDSPFracFilterBankLin
{
public:
CDSPFracFilterBankLin()
: Order( -1 )
{
}
/**
* Copy constructor copies a limited set of parameters of the source
* filter bank. The actual filters are not copied. Such copying is used
* during filtering steps "modeling" stage. A further init() function
* call is required.
*
* @param s Source filter bank.
*/
void copyInitParams( const CDSPFracFilterBankLin& s )
{
WFLen2 = s.WFLen2;
WFFreq = s.WFFreq;
WFAlpha = s.WFAlpha;
FracCount = s.FracCount;
Order = s.Order;
Alignment = s.Alignment;
SrcFilterLen = s.SrcFilterLen;
FilterLen = s.FilterLen;
FilterSize = s.FilterSize;
IsSrcTableBuilt = false;
ExtFilter = s.ExtFilter;
TableFillFlags.alloc( s.TableFillFlags.getCapacity() );
int i;
// Copy table fill flags, but shifted so that further initialization
// is still possible (such feature should not be used, though).
for( i = 0; i < TableFillFlags.getCapacity(); i++ )
{
TableFillFlags[ i ] = (uint8_t) ( s.TableFillFlags[ i ] << 2 );
}
}
/**
* Operator compares *this filter bank and another filter bank and returns
* "true" if their parameters are equal. Alignment is not taken into
* account.
*
* @param s Filter bank to compare to.
* @return "True" if compared banks have equal parameters.
*/
bool operator == ( const CDSPFracFilterBankLin& s ) const
{
return( Order == s.Order && WFLen2 == s.WFLen2 &&
WFFreq == s.WFFreq && WFAlpha == s.WFAlpha &&
FracCount == s.FracCount && ExtFilter == s.ExtFilter );
}
/**
* Function initializes (builds) the filter bank based on the supplied
* parameters. If the supplied parameters are equal to previously defined
* parameters, function does nothing (alignment is assumed to be never
* changing between the init() function calls).
*
* @param ReqFracCount Required number of fractional delays in the filter
* bank. The minimal value is 2.
* @param ReqOrder Required order of the interpolation polynomial
* (0 or 1).
* @param BaseLen Low-pass filter's base length, in samples (taps).
* Affects the actual length of the filter and its overall steepness.
* @param Cutoff Low-pass filter's normalized cutoff frequency [0; 1].
* @param aWFAlpha Peaked Cosine window function's Alpha parameter.
* @param aExtFilter External filter to apply to each fractional delay
* filter.
* @param aAlignment Memory alignment of the filter bank, power-of-2
* value. 0 - use default stdlib alignment.
* @param FltLenAlign Filter's length alignment, power-of-2 value.
*/
void init( const int ReqFracCount, const int ReqOrder,
const double BaseLen, const double Cutoff, const double aWFAlpha,
const CFltBuffer& aExtFilter, const int aAlignment = 0,
const int FltLenAlign = 1 )
{
double NewWFLen2 = 0.5 * BaseLen * ReqFracCount;
double NewWFFreq = AVIR_PI * Cutoff / ReqFracCount;
double NewWFAlpha = aWFAlpha;
if( ReqOrder == Order && NewWFLen2 == WFLen2 && NewWFFreq == WFFreq &&
NewWFAlpha == WFAlpha && ReqFracCount == FracCount &&
aExtFilter == ExtFilter )
{
IsInitRequired = false;
return;
}
WFLen2 = NewWFLen2;
WFFreq = NewWFFreq;
WFAlpha = NewWFAlpha;
FracCount = ReqFracCount;
Order = ReqOrder;
Alignment = aAlignment;
ExtFilter = aExtFilter;
CDSPPeakedCosineLPF p( WFLen2, WFFreq, WFAlpha );
SrcFilterLen = ( p.fl2 / ReqFracCount + 1 ) * 2;
const int ElementSize = ReqOrder + 1;
FilterLen = SrcFilterLen;
if( ExtFilter.getCapacity() > 0 )
{
FilterLen += ExtFilter.getCapacity() - 1;
}
FilterLen = ( FilterLen + FltLenAlign - 1 ) & ~( FltLenAlign - 1 );
FilterSize = FilterLen * ElementSize;
IsSrcTableBuilt = false;
IsInitRequired = true;
}
/**
* @return The length of each fractional delay filter, in samples (taps).
* Always an even value.
*/
int getFilterLen() const
{
return( FilterLen );
}
/**
* @return The number of fractional filters in use by *this bank.
*/
int getFracCount() const
{
return( FracCount );
}
/**
* @return The order of the interpolation polynomial.
*/
int getOrder() const
{
return( Order );
}
/**
* Function returns the pointer to the specified interpolation table
* filter.
*
* @param i Filter (fractional delay) index, in the range 0 to
* ReqFracCount - 1, inclusive.
* @return Pointer to filter. Higher order polynomial coefficients are
* stored after after previous order coefficients, separated by FilterLen
* elements.
*/
const fptype* getFilter( const int i )
{
if( !IsSrcTableBuilt )
{
buildSrcTable();
}
fptype* const Res = &Table[ i * FilterSize ];
if(( TableFillFlags[ i ] & 2 ) == 0 )
{
createFilter( i );
TableFillFlags[ i ] |= 2;
if( Order > 0 )
{
createFilter( i + 1 );
const fptype* const Res2 = Res + FilterSize;
fptype* const op = Res + FilterLen;
int j;
// Create higher-order interpolation coefficients (linear
// interpolation).
for( j = 0; j < FilterLen; j++ )
{
op[ j ] = Res2[ j ] - Res[ j ];
}
}
}
return( Res );
}
/**
* Function makes sure all fractional delay filters were created.
*/
void createAllFilters()
{
int i;
for( i = 0; i < FracCount; i++ )
{
getFilter( i );
}
}
/**
* Function returns an approximate initialization complexity, expressed in
* the number of multiply-add operations. This includes fractional delay
* filters calculation and multiplication by an external filter. This
* function can only be called after the init() function.
*
* @param FracUseMap Fractional delays use map, each element corresponds
* to a single fractional delay, will be compared to the internal table
* fill flags. This map should include 0 and 1 values only.
* @return The complexity of the initialization, expressed in the number
* of multiply-add operations.
*/
int calcInitComplexity( const CBuffer< uint8_t >& FracUseMap ) const
{
const int FltInitCost = 65; // Cost to initialize a single sample
// of the fractional delay filter.
const int FltUseCost = FilterLen * Order +
SrcFilterLen * ExtFilter.getCapacity(); // Cost to use a single
// fractional delay filter.
const int ucb[ 2 ] = { 0, FltUseCost };
int ic;
int i;
if( IsInitRequired )
{
ic = FracCount * SrcFilterLen * FltInitCost;
for( i = 0; i < FracCount; i++ )
{
ic += ucb[ FracUseMap[ i ]];
}
}
else
{
ic = 0;
for( i = 0; i < FracCount; i++ )
{
if( FracUseMap[ i ] != 0 )
{
ic += ucb[ TableFillFlags[ i ] == 0 ? 1 : 0 ];
}
}
}
return( ic );
}
private:
static const int InterpPoints = 2; ///< The maximal number of points the
///< interpolation is based on.
///<
double WFLen2; ///< Window function's Len2 parameter.
///<
double WFFreq; ///< Window function's Freq parameter.
///<
double WFAlpha; ///< Window function's Alpha parameter.
///<
int FracCount; ///< The required number of fractional delay filters.
///<
int Order; ///< The order of the interpolation polynomial.
///<
int Alignment; ///< The required filter table alignment.
///<
int SrcFilterLen; ///< Length of the "source" filters. This is always an
///< even value.
///<
int FilterLen; ///< Specifies the number of samples (taps) each fractional
///< delay filter has. This is always an even value, adjusted by the
///< FltLenAlign.
///<
int FilterSize; ///< The size of a single filter element, equals
///< FilterLen * ElementSize.
///<
bool IsInitRequired; ///< "True" if SrcTable filter table initialization
///< is required. This value is available only after the call to the
///< init() function.
///<
CBuffer< fptype > Table; ///< Interpolation table, size equals to
///< ReqFracCount * FilterLen * ElementSize.
///<
CBuffer< uint8_t > TableFillFlags; ///< Contains ReqFracCount + 1
///< elements. Bit 0 of every element is 1 if Table already contains
///< the filter from SrcTable filtered by ExtFilter. Bit 1 of every
///< element means higher order coefficients were filled for the
///< filter.
///<
CFltBuffer ExtFilter; ///< External filter that should be applied to every
///< fractional delay filter. Can be empty. Half of this filter's
///< capacity is used as latency (group delay) value of the filter.
///<
CBuffer< double > SrcTable; ///< Source table of delay filters, contains
///< ReqFracCount + 1 elements. This table is used to fill the Table
///< with the actual filters, filtered by an external filter.
///<
bool IsSrcTableBuilt; ///< "True" if the SrcTable was built already. This
///< variable is set to "false" in the init() function.
///<
/**
* Function builds source table used in the createFilter() function.
*/
void buildSrcTable()
{
IsSrcTableBuilt = true;
IsInitRequired = false;
CDSPPeakedCosineLPF p( WFLen2, WFFreq, WFAlpha );
const int BufLen = SrcFilterLen * FracCount + InterpPoints - 1;
const int BufOffs = InterpPoints / 2 - 1;
const int BufCenter = SrcFilterLen * FracCount / 2 + BufOffs;
CBuffer< double > Buf( BufLen );
memset( Buf, 0, ( BufCenter - p.fl2 ) * sizeof( double ));
int i = BufLen - BufCenter - p.fl2 - 1;
memset( &Buf[ BufLen - i ], 0, i * sizeof( double ));
p.generateLPF( &Buf[ BufCenter - p.fl2 ], FracCount );
SrcTable.alloc(( FracCount + 1 ) * SrcFilterLen );
TableFillFlags.alloc( FracCount + 1 );
int j;
double* op0 = SrcTable;
for( i = FracCount; i >= 0; i-- )
{
TableFillFlags[ i ] = 0;
double* p = Buf + BufOffs + i;
for( j = 0; j < SrcFilterLen; j++ )
{
op0[ 0 ] = p[ 0 ];
op0++;
p += FracCount;
}
}
Table.alloc(( FracCount + 1 ) * FilterSize, Alignment );
}
/**
* Function creates the specified filter in the Table by copying it from
* the SrcTable and filtering by ExtFilter. Function does nothing if
* filter was already created.
*
* @param k Filter index to create, in the range 0 to FracCount,
* inclusive.
*/
void createFilter( const int k )
{
if( TableFillFlags[ k ] != 0 )
{
return;
}
TableFillFlags[ k ] |= 1;
const int ExtFilterLatency = ExtFilter.getCapacity() / 2;
const int ResLatency = ExtFilterLatency + SrcFilterLen / 2;
int ResLen = SrcFilterLen;
if( ExtFilter.getCapacity() > 0 )
{
ResLen += ExtFilter.getCapacity() - 1;
}
const int ResOffs = FilterLen / 2 - ResLatency;
fptype* op = &Table[ k * FilterSize ];
int i;
for( i = 0; i < ResOffs; i++ )
{
op[ i ] = 0.0;
}
for( i = ResOffs + ResLen; i < FilterLen; i++ )
{
op[ i ] = 0.0;
}
op += ResOffs;
const double* const srcflt = &SrcTable[ k * SrcFilterLen ];
if( ExtFilter.getCapacity() == 0 )
{
for( i = 0; i < ResLen; i++ )
{
op[ i ] = (fptype) srcflt[ i ];
}
return;
}
// Perform convolution of extflt and srcflt.
const double* const extflt = &ExtFilter[ 0 ];
int j;
for( j = 0; j < ResLen; j++ )
{
int k = 0;
int l = j - ExtFilter.getCapacity() + 1;
int r = l + ExtFilter.getCapacity();
if( l < 0 )
{
k -= l;
l = 0;
}
if( r > SrcFilterLen )
{
r = SrcFilterLen;
}
const double* const extfltb = extflt + k;
const double* const srcfltb = srcflt + l;
double s = 0.0;
l = r - l;
for( i = 0; i < l; i++ )
{
s += extfltb[ i ] * srcfltb[ i ];
}
op[ j ] = (fptype) s;
}
}
};
/**
* @brief Thread pool for multi-threaded image resizing operation.
*
* This base class is used to organize a multi-threaded image resizing
* operation. The thread pool should consist of threads that initially wait
* for a signal. Upon receiving a signal (via the startAllWorkloads()
* function) each previously added thread should execute its workload's
* process() function once, and return to the wait signal state again. The
* thread pool should be also able to efficiently wait for all workloads to
* finish via the waitAllWorkloadsToFinish() function.
*
* The image resizing algorithm makes calls to functions of this class.
*/
class CImageResizerThreadPool
{
public:
CImageResizerThreadPool()
{
}
virtual ~CImageResizerThreadPool()
{
}
/**
* @brief Thread pool's workload object class.
*
* This class should be used as a base class for objects that perform the
* actual work spread over several threads.
*/
class CWorkload
{
public:
virtual ~CWorkload()
{
}
/**
* Function that gets called from the thread when thread pool's
* startAllWorkloads() function is called.
*/
virtual void process() = 0;
};
/**
* @return The suggested number of workloads (and their associated
* threads) to add. The minimal value this function can return is 1. The
* usual value may depend on the number of physical and virtual cores
* present in the system, and on other considerations.
*/
virtual int getSuggestedWorkloadCount() const
{
return( 1 );
}
/**
* Function adds a new workload (and possibly thread) to the thread pool.
* The caller decides how many parallel workloads (and threads) it
* requires, but this number will not exceed the value returned by the
* getSuggestedWorkloadCount() function. It is implementation-specific how
* many workloads to associate with a single thread. But for efficiency
* reasons each workload should be associated with its own thread.
*
* Note that the same set of workload objects will be processed each time
* the startAllWorkloads() function is called. This means that workload
* objects are added only once. The caller changes the state of the
* workload objects and then calls the startAllWorkloads() function to
* process them.
*
* @param Workload Workload object whose process() function will be called
* from within the thread when the startAllWorkloads() function is called.
*/
virtual void addWorkload( CWorkload* const Workload )
{
}
/**
* Function starts all workloads associated with threads previously added
* via the addWorkload() function. It is assumed that this function
* performs the necessary "memory barrier" (or "cache sync") kind of
* operation so that all threads catch up the prior changes made to the
* workload objects during their wait state.
*/
virtual void startAllWorkloads()
{
}
/**
* Function waits for all workloads to finish.
*/
virtual void waitAllWorkloadsToFinish()
{
}
/**
* Function removes all workloads previously added via the addWorkload()
* function. This function gets called only after the
* waitAllWorkloadsToFinish() function call.
*/
virtual void removeAllWorkloads()
{
}
};
/**
* @brief Resizing algorithm parameters structure.
*
* This structure holds all selectable parameters used by the resizing
* algorithm at various stages, for both downsizing and upsizing. There are no
* other parameters exist that can optimize the performance of the resizing
* algorithm. Filter length parameters can take fractional values.
*
* Beside quality, these parameters (except Alpha parameters) directly affect
* the computative cost of the resizing algorithm. It is possible to trade
* the visual quality for computative cost.
*
* Anti-alias filtering during downsizing can be defined as a considerable
* reduction of contrast of smallest features of an image. Unfortunately, such
* de-contrasting partially affects features of all sizes thus producing a
* non-linearity of frequency response. All pre-defined parameter sets are
* described by 3 values separated by slashes. The first value is the
* de-contrasting factor of small features (which are being removed) while
* the second value is the de-contrasting factor of large features (which
* should remain intact), with value of 1 equating to "no contrast change".
* The third value is the optimization score (see below), with value of 0
* equating to the "perfect" linearity of frequency response.
*
* The pre-defined parameter sets offered by this library were auto-optimized
* for the given LPFltBaseLen, IntFltLen and CorrFltAlpha values. The
* optimization goal was to minimize the score: the sum of squares of the
* difference between original and processed images (which was not actually
* resized, k=1). The original image was a 0.5 megapixel uniformly-distributed
* white-noise image with pixel intensities in the 0-1 range. Such goal
* converges very well and produces filtering system with the flattest
* frequency response possible for the given constraints. With this goal,
* increasing the LPFltBaseLen value reduces the general amount of aliasing
* artifacts.
*/
struct CImageResizerParams
{
double CorrFltAlpha; ///< Alpha parameter of the Peaked Cosine window
///< function used on the correction filter. The "usable" values are
///< in the narrow range 1.0 to 1.5.
///<
double CorrFltLen; ///< Correction filter's length in samples (taps). The
///< "usable" range is narrow, 5.5 to 8, as to minimize the
///< "overcorrection" which is mathematically precise, but visually
///< unacceptable.
///<
double IntFltAlpha; ///< Alpha parameter of the Peaked Cosine window
///< function used on the interpolation low-pass filter. The "usable"
///< values are in the range 1.5 to 2.5.
///<
double IntFltCutoff; ///< Interpolation low-pass filter's cutoff frequency
///< (normalized, [0; 1]). The "usable" range is 0.6 to 0.8.
///<
double IntFltLen; ///< Interpolation low-pass filter's length in samples
///< (taps). The length value should be at least 18 or otherwise a
///< "dark grid" artifact will be introduced if a further sharpening
///< is applied. IntFltLen together with other IntFlt parameters
///< should be tuned in a way that produces the flattest frequency
///< response in 0-0.5 normalized frequency range (this range is due
///< to 2X upsampling).
///<
double LPFltAlpha; ///< Alpha parameter of the Peaked Cosine window
///< function used on the low-pass filter. The "usable" values are
///< in the range 1.5 to 6.5.
///<
double LPFltBaseLen; ///< Base length of the low-pass (aka anti-aliasing
///< or reconstruction) filter, in samples (taps), further adjusted by
///< the actual cutoff frequency, upsampling and downsampling factors.
///< The "usable" range is between 6 and 9.
///<
double LPFltCutoffMult; ///< Low-pass filter's cutoff frequency
///< multiplier. This value can be both below and above 1.0 as
///< low-pass filters are inserted on downsampling and upsampling
///< steps and always have corner frequency equal to or below 0.5pi.
///< This multiplier shifts low-pass filter's corner frequency towards
///< lower (if below 1.0) or higher (if above 1.0) frequencies. This
///< multiplier can be way below 1.0 since any additional
///< high-frequency damping will be partially corrected by the
///< correction filter. The "usable" range is 0.3 to 1.0.
///<
CImageResizerParams()
: HBFltAlpha( 1.75395 )
, HBFltCutoff( 0.40356 )
, HBFltLen( 22.00000 )
{
}
double HBFltAlpha; ///< Half-band filter's Alpha. Assigned internally.
///<
double HBFltCutoff; ///< Half-band filter's cutoff point [0; 1]. Assigned
///< internally.
///<
double HBFltLen; ///< Length of the half-band low-pass filter. Assigned
///< internally. Internally used to perform 2X or higher downsampling.
///< These filter parameters should be treated as "technical" and do
///< not require adjustment as they were tuned to suit all
///< combinations of other parameters. This half-band filter provides
///< a wide transition band (for minimal ringing artifacts) and a high
///< stop-band attenuation (for minimal aliasing).
///<
};
/**
* @brief The default set of resizing algorithm parameters
* (10.01/1.029/0.019169).
*
* This is the default set of resizing parameters that was designed to deliver
* a sharp image while still providing a low amount of ringing artifacts, and
* having a reasonable computational cost.
*/
struct CImageResizerParamsDef : public CImageResizerParams
{
CImageResizerParamsDef()
{
CorrFltAlpha = 1.0;//10.01/1.88/1.029(522.43)/0.019169:258648,446808
CorrFltLen = 6.30770;
IntFltAlpha = 2.27825;
IntFltCutoff = 0.75493;
IntFltLen = 18.0;
LPFltAlpha = 3.40127;
LPFltBaseLen = 7.78;
LPFltCutoffMult = 0.78797;
}
};
/**
* @brief Set of resizing algorithm parameters for ultra-low-ringing
* performance (7.69/1.069/0.000245).
*
* This set of resizing algorithm parameters offers the lowest amount of
* ringing this library is capable of providing while still offering a decent
* quality. Low ringing is attained at the expense of higher aliasing
* artifacts and a slightly reduced contrast.
*/
struct CImageResizerParamsULR : public CImageResizerParams
{
CImageResizerParamsULR()
{
CorrFltAlpha = 1.0;//7.69/1.97/1.069(31445.45)/0.000245:258627,436845
CorrFltLen = 5.83280;
IntFltAlpha = 2.11453;
IntFltCutoff = 0.73986;
IntFltLen = 18.0;
LPFltAlpha = 1.73455;
LPFltBaseLen = 6.40;
LPFltCutoffMult = 0.61314;
}
};
/**
* @brief Set of resizing algorithm parameters for low-ringing performance
* (7.86/1.065/0.000106).
*
* This set of resizing algorithm parameters offers a very low-ringing
* performance at the expense of higher aliasing artifacts and a slightly
* reduced contrast.
*/
struct CImageResizerParamsLR : public CImageResizerParams
{
CImageResizerParamsLR()
{
CorrFltAlpha = 1.0;//7.86/1.96/1.065(73865.02)/0.000106:258636,437381
CorrFltLen = 5.87671;
IntFltAlpha = 2.25322;
IntFltCutoff = 0.74090;
IntFltLen = 18.0;
LPFltAlpha = 1.79306;
LPFltBaseLen = 7.00;
LPFltCutoffMult = 0.68881;
}
};
/**
* @brief Set of resizing algorithm parameters for lower-ringing performance
* (8.86/1.046/0.010168).
*
* This set of resizing algorithm parameters offers a lower-ringing
* performance in comparison to the default setting, at the expense of higher
* aliasing artifacts and a slightly reduced contrast.
*/
struct CImageResizerParamsLow : public CImageResizerParams
{
CImageResizerParamsLow()
{
CorrFltAlpha = 1.0;//8.86/1.92/1.046(871.54)/0.010168:258647,442252
CorrFltLen = 6.09757;
IntFltAlpha = 2.36704;
IntFltCutoff = 0.74674;
IntFltLen = 18.0;
LPFltAlpha = 2.19427;
LPFltBaseLen = 7.66;
LPFltCutoffMult = 0.75380;
}
};
/**
* @brief Set of resizing algorithm parameters for low-aliasing
* resizing (11.81/1.012/0.038379).
*
* This set of resizing algorithm parameters offers a considerable
* anti-aliasing performance with a good frequency response linearity (and
* contrast). This is an intermediate setting between the default and Ultra
* parameters.
*/
struct CImageResizerParamsHigh : public CImageResizerParams
{
CImageResizerParamsHigh()
{
CorrFltAlpha = 1.0;//11.81/1.83/1.012(307.84)/0.038379:258660,452719
CorrFltLen = 6.80909;
IntFltAlpha = 2.44917;
IntFltCutoff = 0.75856;
IntFltLen = 18.0;
LPFltAlpha = 4.39527;
LPFltBaseLen = 8.18;
LPFltCutoffMult = 0.79172;
}
};
/**
* @brief Set of resizing algorithm parameters for ultra low-aliasing
* resizing (13.65/1.001/0.000483).
*
* This set of resizing algorithm parameters offers a very considerable
* anti-aliasing performance with a good frequency response linearity (and
* contrast). This set of parameters is computationally expensive and may
* produce ringing artifacts on sharp features.
*/
struct CImageResizerParamsUltra : public CImageResizerParams
{
CImageResizerParamsUltra()
{
CorrFltAlpha = 1.0;//13.65/1.79/1.001(28288.41)/0.000483:258658,457974
CorrFltLen = 7.48060;
IntFltAlpha = 1.93750;
IntFltCutoff = 0.75462;
IntFltLen = 18.0;
LPFltAlpha = 5.55209;
LPFltBaseLen = 8.34;
LPFltCutoffMult = 0.78002;
}
};
/**
* @brief Image resizing variables class.
*
* This is an utility "catch all" class that defines various variables used
* during image resizing. Several variables that are explicitly initialized in
* this class' constructor are also used as additional "input" variables to
* the image resizing function. These variables will not be changed by the
* avir::CImageResizer<>::resizeImage() function.
*/
class CImageResizerVars
{
public:
int ElCount; ///< The number of "fptype" elements used to store 1 pixel.
///<
int ElCountIO; ///< The number of source and destination image's elements
///< used to store 1 pixel.
///<
int fppack; ///< The number of atomic types stored in a single "fptype"
///< element.
///<
int fpalign; ///< Suggested alignment size in bytes. This is not a
///< required alignment, because image resizing algorithm cannot be
///< made to have a strictly aligned data access in all cases (e.g.
///< de-interleaved interpolation cannot perform aligned accesses).
///<
int elalign; ///< Length alignment of arrays of elements. This applies to
///< filters and intermediate buffers: this constant forces filters
///< and scanlines to have a length which is a multiple of this value,
///< for more efficient SIMD implementation.
///<
int packmode; ///< 0 if interleaved packing, 1 if de-interleaved.
///<
int BufLen[ 2 ]; ///< Intermediate buffers' lengths in "fptype" elements.
int BufOffs[ 2 ]; ///< Offsets into the intermediate buffers, used to
///< provide prefix elements required during processing so that no
///< "out of range" access happens. This offset is a multiple of
///< ElCount if pixels are stored in interleaved form.
///<
double k; ///< Resizing step coefficient, updated to reflect the actually
///< used coefficient during resizing.
///<
double o; ///< Starting pixel offset inside the source image, updated to
///< reflect the actually used offset during resizing.
///<
int ResizeStep; ///< Index of the resizing step in the latest filtering
///< steps array.
///<
double InGammaMult; ///< Input gamma multiplier, used to convert input
///< data to 0 to 1 range. 0.0 if no gamma is in use.
///<
double OutGammaMult; ///< Output gamma multiplier, used to convert data to
///< 0 to 255/65535 range. 0.0 if no gamma is in use.
///<
double ox; ///< Start X pixel offset within source image (can be
///< negative). Positive offset moves image to the left.
///<
double oy; ///< Start Y pixel offset within source image (can be
///< negative). Positive offset moves image to the top.
///<
CImageResizerThreadPool* ThreadPool; ///< Thread pool to be used by the
///< image resizing function. Set to NULL to use single-threaded
///< processing.
///<
bool UseSRGBGamma; ///< Perform sRGB gamma linearization (correction).
///<
int BuildMode; ///< The build mode to use, for debugging purposes. Set to
///< -1 to select a minimal-complexity mode automatically. All build
///< modes deliver similar results with minor deviations.
///<
int RndSeed; ///< Random seed parameter. This parameter may be incremented
///< after each random generator initialization. The use of this
///< variable depends on the ditherer implementation.
///<
CImageResizerVars()
: ox( 0.0 )
, oy( 0.0 )
, ThreadPool( nullptr )
, UseSRGBGamma( false )
, BuildMode( -1 )
, RndSeed( 0 )
{
}
};
/**
* @brief Image resizer's filtering step class.
*
* Class defines data to perform a single filtering step over a whole
* horizontal or vertical scanline. Resizing consists of 1 or more steps that
* may be performed before the actual resizing takes place. Filtering may also
* follow a resizing step. Each step must ensure that scanline data contains
* enough pixels to perform the next step (which may be resizing) without
* exceeding scanline's bounds.
*
* A derived class must implement several "const" and "static" functions that
* are used to perform the actual filtering in interleaved or de-interleaved
* mode.
*
* @tparam fptype Floating point type to use for storing pixel elements. SIMD
* types can be used: in this case each element may hold a whole pixel.
* @tparam fptypeatom The atomic type the "fptype" consists of.
*/
template< class fptype, class fptypeatom >
class CImageResizerFilterStep
{
public:
bool IsUpsample; ///< "True" if this step is an upsampling step, "false"
///< if downsampling step. Should be set to "false" if ResampleFactor
///< equals 0.
///<
int ResampleFactor; ///< Resample factor (>=1). If 0, this is a resizing
///< step. This value should be >1 if IsUpsample equals "true".
///<
CBuffer< fptype > Flt; ///< Filter to use at this step.
///<
CFltBuffer FltOrig; ///< Originally-designed filter. This buffer may not
///< be assigned. Assigned by filters that precede the resizing step
///< if such filter is planned to be embedded into the interpolation
///< filter as "external" filter. If IsUpsample=true and this filter
///< buffer is not empty, the upsampling step will not itself apply
///< any filtering over upsampled input scanline.
///<
double DCGain; ///< DC gain which was applied to the filter. Not defined
///< if ResampleFactor = 0.
///<
int FltLatency; ///< Filter's latency (group delay, shift) in pixels.
///<
const CImageResizerVars* Vars; ///< Image resizing-related variables.
///<
int InLen; ///< Input scanline's length in pixels.
///<
int InBuf; ///< Input buffer index, 0 or 1.
///<
int InPrefix; ///< Required input prefix pixels. These prefix pixels will
///< be filled with source scanline's first pixel value. If IsUpsample
///< is "true", this is the additional number of times the first pixel
///< will be filtered before processing scanline, this number is also
///< reflected in the OutPrefix.
///<
int InSuffix; ///< Required input suffix pixels. These suffix pixels will
///< be filled with source scanline's last pixel value. If IsUpsample
///< is "true", this is the additional number of times the last pixel
///< will be filtered before processing scanline, this number is also
///< reflected in the OutSuffix.
///<
int InElIncr; ///< Pixel element increment within the input buffer, used
///< during de-interleaved processing: in this case each image's
///< channel is stored independently, InElIncr elements apart.
///<
int OutLen; ///< Length of the resulting scanline.
///<
int OutBuf; ///< Output buffer index. 0 or 1; 2 for the last step.
///<
int OutPrefix; ///< Required output prefix pixels. These prefix pixels
///< will not be pre-filled with any values. Value is valid only if
///< IsUpsample equals "true".
///<
int OutSuffix; ///< Required input suffix pixels. These suffix pixels will
///< not be pre-filled with any values. Value is valid only if
///< IsUpsample equals "true".
///<
int OutElIncr; ///< Pixel element increment within the output buffer, used
///< during de-interleaved processing. Equals to the InBufElIncr of
///< the next step.
///<
CBuffer< fptype > PrefixDC; ///< DC component fluctuations added at the
///< start of the resulting scanline, used when IsUpsample equals
///< "true".
///<
CBuffer< fptype > SuffixDC; ///< DC component fluctuations added at the
///< end of the resulting scanline, used when IsUpsample equals
///< "true".
///<
int EdgePixelCount; ///< The number of edge pixels added. Affects the
///< initial position within the input scanline, used to produce edge
///< pixels. This variable is used and should be defined when
///< IsUpsample=false and ResampleFactor>0. When assigning this
///< variable it is also necessary to update InPrefix, OutLen and
///< Vars.o variables.
///<
static const int EdgePixelCountDef = 3; ///< The default number of pixels
///< additionally produced at scanline edges during filtering. This is
///< required to reduce edge artifacts.
///<
/**
* @brief Resizing position structure.
*
* Structure holds resizing position and pointer to fractional delay
* filter.
*/
struct CResizePos
{
int SrcPosInt; ///< Source scanline position.
///<
int fti; ///< Fractional delay filter index.
///<
const fptype* ftp; ///< Fractional delay filter pointer.
///<
fptypeatom x; ///< Interpolation coefficient between delay filters.
///<
int SrcOffs; ///< Source scanline offset.
///<
};
/**
* @brief Resizing positions buffer class.
*
* This class combines buffer together with variables that define resizing
* stepping.
*/
class CRPosBuf : public CBuffer< CResizePos >
{
public:
double k; ///< Resizing step.
///<
double o; ///< Resizing offset.
///<
int FracCount; ///< The number of fractional delay filters in a filter
///< bank used together with this buffer.
///<
};
/**
* @brief Resizing positions buffer array class.
*
* This class combines structure array of the CRPosBuf class objects with
* the function that locates or creates buffer with the required resizing
* stepping.
*/
class CRPosBufArray : public CStructArray< CRPosBuf >
{
public:
using CStructArray< CRPosBuf > :: add;
using CStructArray< CRPosBuf > :: getItemCount;
/**
* Function returns the resizing positions buffer with the required
* stepping. If no such buffer exists, it is created.
*
* @param k Resizing step.
* @param o Resizing offset.
* @param FracCount The number of fractional delay filters in a filter
* bank used together with this buffer.
* @return Reference to the CRPosBuf object.
*/
CRPosBuf& getRPosBuf( const double k, const double o,
const int FracCount )
{
int i;
for( i = 0; i < getItemCount(); i++ )
{
CRPosBuf& Buf = (*this)[ i ];
if( Buf.k == k && Buf.o == o && Buf.FracCount == FracCount )
{
return( Buf );
}
}
CRPosBuf& NewBuf = add();
NewBuf.k = k;
NewBuf.o = o;
NewBuf.FracCount = FracCount;
return( NewBuf );
}
};
CRPosBuf* RPosBuf; ///< Resizing positions buffer. Used when
///< ResampleFactor equals 0 (resizing step).
///<
CDSPFracFilterBankLin< fptype >* FltBank; ///< Filter bank in use by *this
///< resizing step.
///<
};
/**
* @brief Interleaved filtering steps implementation class.
*
* This class implements scanline filtering functions in interleaved mode.
* This means that each pixel is processed independently, not in groups.
*
* @tparam fptype Floating point type to use for storing pixel elements. SIMD
* types can be used: in this case each element may hold a whole pixel.
* @tparam fptypeatom The atomic type the "fptype" consists of.
*/
template< class fptype, class fptypeatom >
class CImageResizerFilterStepINL :
public CImageResizerFilterStep< fptype, fptypeatom >
{
public:
using CImageResizerFilterStep< fptype, fptypeatom > :: IsUpsample;
using CImageResizerFilterStep< fptype, fptypeatom > :: ResampleFactor;
using CImageResizerFilterStep< fptype, fptypeatom > :: Flt;
using CImageResizerFilterStep< fptype, fptypeatom > :: FltOrig;
using CImageResizerFilterStep< fptype, fptypeatom > :: FltLatency;
using CImageResizerFilterStep< fptype, fptypeatom > :: Vars;
using CImageResizerFilterStep< fptype, fptypeatom > :: InLen;
using CImageResizerFilterStep< fptype, fptypeatom > :: InPrefix;
using CImageResizerFilterStep< fptype, fptypeatom > :: InSuffix;
using CImageResizerFilterStep< fptype, fptypeatom > :: OutLen;
using CImageResizerFilterStep< fptype, fptypeatom > :: OutPrefix;
using CImageResizerFilterStep< fptype, fptypeatom > :: OutSuffix;
using CImageResizerFilterStep< fptype, fptypeatom > :: PrefixDC;
using CImageResizerFilterStep< fptype, fptypeatom > :: SuffixDC;
using CImageResizerFilterStep< fptype, fptypeatom > :: RPosBuf;
using CImageResizerFilterStep< fptype, fptypeatom > :: FltBank;
using CImageResizerFilterStep< fptype, fptypeatom > :: EdgePixelCount;
/**
* Function performs "packing" of a scanline and type conversion.
* Scanline, depending on the "fptype" can be potentially stored as a
* packed SIMD values having a certain atomic type. If required, the sRGB
* gamma correction is applied.
*
* @param ip Input scanline.
* @param op0 Output scanline.
* @param l0 The number of pixels to "pack".
*/
template< class Tin >
void packScanline( const Tin* ip, fptype* const op0, const int l0 ) const
{
const int ElCount = Vars -> ElCount;
const int ElCountIO = Vars -> ElCountIO;
fptype* op = op0;
int l = l0;
if( !Vars -> UseSRGBGamma )
{
if( ElCountIO == 1 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op;
v[ 0 ] = (fptypeatom) ip[ 0 ];
op += ElCount;
ip++;
l--;
}
}
else
if( ElCountIO == 4 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op;
v[ 0 ] = (fptypeatom) ip[ 0 ];
v[ 1 ] = (fptypeatom) ip[ 1 ];
v[ 2 ] = (fptypeatom) ip[ 2 ];
v[ 3 ] = (fptypeatom) ip[ 3 ];
op += ElCount;
ip += 4;
l--;
}
}
else
if( ElCountIO == 3 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op;
v[ 0 ] = (fptypeatom) ip[ 0 ];
v[ 1 ] = (fptypeatom) ip[ 1 ];
v[ 2 ] = (fptypeatom) ip[ 2 ];
op += ElCount;
ip += 3;
l--;
}
}
else
if( ElCountIO == 2 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op;
v[ 0 ] = (fptypeatom) ip[ 0 ];
v[ 1 ] = (fptypeatom) ip[ 1 ];
op += ElCount;
ip += 2;
l--;
}
}
}
else
{
const fptypeatom gm = (fptypeatom) Vars -> InGammaMult;
if( ElCountIO == 1 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op;
v[ 0 ] = convertSRGB2Lin( (fptypeatom) ip[ 0 ] * gm );
op += ElCount;
ip++;
l--;
}
}
else
if( ElCountIO == 4 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op;
v[ 0 ] = convertSRGB2Lin( (fptypeatom) ip[ 0 ] * gm );
v[ 1 ] = convertSRGB2Lin( (fptypeatom) ip[ 1 ] * gm );
v[ 2 ] = convertSRGB2Lin( (fptypeatom) ip[ 2 ] * gm );
v[ 3 ] = convertSRGB2Lin( (fptypeatom) ip[ 3 ] * gm );
op += ElCount;
ip += 4;
l--;
}
}
else
if( ElCountIO == 3 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op;
v[ 0 ] = convertSRGB2Lin( (fptypeatom) ip[ 0 ] * gm );
v[ 1 ] = convertSRGB2Lin( (fptypeatom) ip[ 1 ] * gm );
v[ 2 ] = convertSRGB2Lin( (fptypeatom) ip[ 2 ] * gm );
op += ElCount;
ip += 3;
l--;
}
}
else
if( ElCountIO == 2 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op;
v[ 0 ] = convertSRGB2Lin( (fptypeatom) ip[ 0 ] * gm );
v[ 1 ] = convertSRGB2Lin( (fptypeatom) ip[ 1 ] * gm );
op += ElCount;
ip += 2;
l--;
}
}
}
const int ZeroCount = ElCount * Vars -> fppack - ElCountIO;
op = op0;
l = l0;
if( ZeroCount == 1 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op + ElCountIO;
v[ 0 ] = (fptypeatom) 0;
op += ElCount;
l--;
}
}
else
if( ZeroCount == 2 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op + ElCountIO;
v[ 0 ] = (fptypeatom) 0;
v[ 1 ] = (fptypeatom) 0;
op += ElCount;
l--;
}
}
else
if( ZeroCount == 3 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) op + ElCountIO;
v[ 0 ] = (fptypeatom) 0;
v[ 1 ] = (fptypeatom) 0;
v[ 2 ] = (fptypeatom) 0;
op += ElCount;
l--;
}
}
}
/**
* Function applies Linear to sRGB gamma correction to the specified
* scanline.
*
* @param p Scanline.
* @param l The number of pixels to de-linearize.
* @param Vars0 Image resizing-related variables.
*/
static void applySRGBGamma( fptype* p, int l,
const CImageResizerVars& Vars0 )
{
const int ElCount = Vars0.ElCount;
const int ElCountIO = Vars0.ElCountIO;
const fptypeatom gm = (fptypeatom) Vars0.OutGammaMult;
if( ElCountIO == 1 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) p;
v[ 0 ] = convertLin2SRGB( v[ 0 ]) * gm;
p += ElCount;
l--;
}
}
else
if( ElCountIO == 4 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) p;
v[ 0 ] = convertLin2SRGB( v[ 0 ]) * gm;
v[ 1 ] = convertLin2SRGB( v[ 1 ]) * gm;
v[ 2 ] = convertLin2SRGB( v[ 2 ]) * gm;
v[ 3 ] = convertLin2SRGB( v[ 3 ]) * gm;
p += ElCount;
l--;
}
}
else
if( ElCountIO == 3 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) p;
v[ 0 ] = convertLin2SRGB( v[ 0 ]) * gm;
v[ 1 ] = convertLin2SRGB( v[ 1 ]) * gm;
v[ 2 ] = convertLin2SRGB( v[ 2 ]) * gm;
p += ElCount;
l--;
}
}
else
if( ElCountIO == 2 )
{
while( l > 0 )
{
fptypeatom* v = (fptypeatom*) p;
v[ 0 ] = convertLin2SRGB( v[ 0 ]) * gm;
v[ 1 ] = convertLin2SRGB( v[ 1 ]) * gm;
p += ElCount;
l--;
}
}
}
/**
* Function converts vertical scanline to horizontal scanline. This
* function is called by the image resizer when image is resized
* vertically. This means that the vertical scanline is stored in the
* same format produced by the packScanline() and maintained by other
* filtering functions.
*
* @param ip Input vertical scanline.
* @param op Output buffer (temporary buffer used during resizing).
* @param SrcLen The number of pixels in the input scanline, also used to
* calculate input buffer increment.
* @param SrcIncr Input buffer increment to the next vertical pixel.
*/
void convertVtoH( const fptype* ip, fptype* op, const int SrcLen,
const int SrcIncr ) const
{
const int ElCount = Vars -> ElCount;
int j;
if( ElCount == 1 )
{
for( j = 0; j < SrcLen; j++ )
{
op[ 0 ] = ip[ 0 ];
ip += SrcIncr;
op++;
}
}
else
if( ElCount == 4 )
{
for( j = 0; j < SrcLen; j++ )
{
op[ 0 ] = ip[ 0 ];
op[ 1 ] = ip[ 1 ];
op[ 2 ] = ip[ 2 ];
op[ 3 ] = ip[ 3 ];
ip += SrcIncr;
op += 4;
}
}
else
if( ElCount == 3 )
{
for( j = 0; j < SrcLen; j++ )
{
op[ 0 ] = ip[ 0 ];
op[ 1 ] = ip[ 1 ];
op[ 2 ] = ip[ 2 ];
ip += SrcIncr;
op += 3;
}
}
else
if( ElCount == 2 )
{
for( j = 0; j < SrcLen; j++ )
{
op[ 0 ] = ip[ 0 ];
op[ 1 ] = ip[ 1 ];
ip += SrcIncr;
op += 2;
}
}
}
/**
* Function performs "unpacking" of a scanline and type conversion
* (truncation is used when floating point is converted to integer).
* Scanline, depending on the "fptype" can be potentially stored as a
* packed SIMD values having a certain atomic type. The unpacking function
* assumes that scanline is stored in the style produced by the
* packScanline() function.
*
* @param ip Input scanline.
* @param op Output scanline.
* @param l The number of pixels to "unpack".
* @param Vars0 Image resizing-related variables.
*/
template< class Tout >
static void unpackScanline( const fptype* ip, Tout* op, int l,
const CImageResizerVars& Vars0 )
{
const int ElCount = Vars0.ElCount;
const int ElCountIO = Vars0.ElCountIO;
if( ElCountIO == 1 )
{
while( l > 0 )
{
const fptypeatom* v = (const fptypeatom*) ip;
op[ 0 ] = (Tout) v[ 0 ];
ip += ElCount;
op++;
l--;
}
}
else
if( ElCountIO == 4 )
{
while( l > 0 )
{
const fptypeatom* v = (const fptypeatom*) ip;
op[ 0 ] = (Tout) v[ 0 ];
op[ 1 ] = (Tout) v[ 1 ];
op[ 2 ] = (Tout) v[ 2 ];
op[ 3 ] = (Tout) v[ 3 ];
ip += ElCount;
op += 4;
l--;
}
}
else
if( ElCountIO == 3 )
{
while( l > 0 )
{
const fptypeatom* v = (const fptypeatom*) ip;
op[ 0 ] = (Tout) v[ 0 ];
op[ 1 ] = (Tout) v[ 1 ];
op[ 2 ] = (Tout) v[ 2 ];
ip += ElCount;
op += 3;
l--;
}
}
else
if( ElCountIO == 2 )
{
while( l > 0 )
{
const fptypeatom* v = (const fptypeatom*) ip;
op[ 0 ] = (Tout) v[ 0 ];
op[ 1 ] = (Tout) v[ 1 ];
ip += ElCount;
op += 2;
l--;
}
}
}
/**
* Function prepares input scanline buffer for *this filtering step.
* Left- and right-most pixels are replicated to make sure no buffer
* overrun happens. Such approach also allows to bypass any pointer
* range checks.
*
* @param Src Source buffer.
*/
void prepareInBuf( fptype* Src ) const
{
if( IsUpsample || InPrefix + InSuffix == 0 )
{
return;
}
const int ElCount = Vars -> ElCount;
replicateArray( Src, ElCount, Src - ElCount, InPrefix, -ElCount );
Src += ( InLen - 1 ) * ElCount;
replicateArray( Src, ElCount, Src + ElCount, InSuffix, ElCount );
}
/**
* Function peforms scanline upsampling with filtering.
*
* @param Src Source scanline buffer (length = this -> InLen). Source
* scanline increment will be equal to ElCount.
* @param Dst Destination scanline buffer.
*/
void doUpsample( const fptype* const Src, fptype* const Dst ) const
{
const int ElCount = Vars -> ElCount;
fptype* op0 = &Dst[ -OutPrefix * ElCount ];
memset( op0, 0, ( OutPrefix + OutLen + OutSuffix ) * ElCount *
sizeof( fptype ));
const fptype* ip = Src;
const int opstep = ElCount * ResampleFactor;
int l;
if( FltOrig.getCapacity() > 0 )
{
// Do not perform filtering, only upsample.
op0 += ( OutPrefix % ResampleFactor ) * ElCount;
l = OutPrefix / ResampleFactor;
if( ElCount == 1 )
{
while( l > 0 )
{
op0[ 0 ] = ip[ 0 ];
op0 += opstep;
l--;
}
l = InLen - 1;
while( l > 0 )
{
op0[ 0 ] = ip[ 0 ];
op0 += opstep;
ip += ElCount;
l--;
}
l = OutSuffix / ResampleFactor;
while( l >= 0 )
{
op0[ 0 ] = ip[ 0 ];
op0 += opstep;
l--;
}
}
else
if( ElCount == 4 )
{
while( l > 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0[ 2 ] = ip[ 2 ];
op0[ 3 ] = ip[ 3 ];
op0 += opstep;
l--;
}
l = InLen - 1;
while( l > 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0[ 2 ] = ip[ 2 ];
op0[ 3 ] = ip[ 3 ];
op0 += opstep;
ip += ElCount;
l--;
}
l = OutSuffix / ResampleFactor;
while( l >= 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0[ 2 ] = ip[ 2 ];
op0[ 3 ] = ip[ 3 ];
op0 += opstep;
l--;
}
}
else
if( ElCount == 3 )
{
while( l > 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0[ 2 ] = ip[ 2 ];
op0 += opstep;
l--;
}
l = InLen - 1;
while( l > 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0[ 2 ] = ip[ 2 ];
op0 += opstep;
ip += ElCount;
l--;
}
l = OutSuffix / ResampleFactor;
while( l >= 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0[ 2 ] = ip[ 2 ];
op0 += opstep;
l--;
}
}
else
if( ElCount == 2 )
{
while( l > 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0 += opstep;
l--;
}
l = InLen - 1;
while( l > 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0 += opstep;
ip += ElCount;
l--;
}
l = OutSuffix / ResampleFactor;
while( l >= 0 )
{
op0[ 0 ] = ip[ 0 ];
op0[ 1 ] = ip[ 1 ];
op0 += opstep;
l--;
}
}
return;
}
const fptype* const f = Flt;
const int flen = Flt.getCapacity();
fptype* op;
int i;
if( ElCount == 1 )
{
l = InPrefix;
while( l > 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ i ] += f[ i ] * ip[ 0 ];
}
op0 += opstep;
l--;
}
l = InLen - 1;
while( l > 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ i ] += f[ i ] * ip[ 0 ];
}
ip += ElCount;
op0 += opstep;
l--;
}
l = InSuffix;
while( l >= 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ i ] += f[ i ] * ip[ 0 ];
}
op0 += opstep;
l--;
}
}
else
if( ElCount == 4 )
{
l = InPrefix;
while( l > 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op[ 2 ] += f[ i ] * ip[ 2 ];
op[ 3 ] += f[ i ] * ip[ 3 ];
op += 4;
}
op0 += opstep;
l--;
}
l = InLen - 1;
while( l > 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op[ 2 ] += f[ i ] * ip[ 2 ];
op[ 3 ] += f[ i ] * ip[ 3 ];
op += 4;
}
ip += ElCount;
op0 += opstep;
l--;
}
l = InSuffix;
while( l >= 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op[ 2 ] += f[ i ] * ip[ 2 ];
op[ 3 ] += f[ i ] * ip[ 3 ];
op += 4;
}
op0 += opstep;
l--;
}
}
else
if( ElCount == 3 )
{
l = InPrefix;
while( l > 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op[ 2 ] += f[ i ] * ip[ 2 ];
op += 3;
}
op0 += opstep;
l--;
}
l = InLen - 1;
while( l > 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op[ 2 ] += f[ i ] * ip[ 2 ];
op += 3;
}
ip += ElCount;
op0 += opstep;
l--;
}
l = InSuffix;
while( l >= 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op[ 2 ] += f[ i ] * ip[ 2 ];
op += 3;
}
op0 += opstep;
l--;
}
}
else
if( ElCount == 2 )
{
l = InPrefix;
while( l > 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op += 2;
}
op0 += opstep;
l--;
}
l = InLen - 1;
while( l > 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op += 2;
}
ip += ElCount;
op0 += opstep;
l--;
}
l = InSuffix;
while( l >= 0 )
{
op = op0;
for( i = 0; i < flen; i++ )
{
op[ 0 ] += f[ i ] * ip[ 0 ];
op[ 1 ] += f[ i ] * ip[ 1 ];
op += 2;
}
op0 += opstep;
l--;
}
}
op = op0;
const fptype* dc = SuffixDC;
l = SuffixDC.getCapacity();
if( ElCount == 1 )
{
for( i = 0; i < l; i++ )
{
op[ i ] += ip[ 0 ] * dc[ i ];
}
}
else
if( ElCount == 4 )
{
while( l > 0 )
{
op[ 0 ] += ip[ 0 ] * dc[ 0 ];
op[ 1 ] += ip[ 1 ] * dc[ 0 ];
op[ 2 ] += ip[ 2 ] * dc[ 0 ];
op[ 3 ] += ip[ 3 ] * dc[ 0 ];
dc++;
op += 4;
l--;
}
}
else
if( ElCount == 3 )
{
while( l > 0 )
{
op[ 0 ] += ip[ 0 ] * dc[ 0 ];
op[ 1 ] += ip[ 1 ] * dc[ 0 ];
op[ 2 ] += ip[ 2 ] * dc[ 0 ];
dc++;
op += 3;
l--;
}
}
else
if( ElCount == 2 )
{
while( l > 0 )
{
op[ 0 ] += ip[ 0 ] * dc[ 0 ];
op[ 1 ] += ip[ 1 ] * dc[ 0 ];
dc++;
op += 2;
l--;
}
}
ip = Src;
op = Dst - InPrefix * opstep;
dc = PrefixDC;
l = PrefixDC.getCapacity();
if( ElCount == 1 )
{
for( i = 0; i < l; i++ )
{
op[ i ] += ip[ 0 ] * dc[ i ];
}
}
else
if( ElCount == 4 )
{
while( l > 0 )
{
op[ 0 ] += ip[ 0 ] * dc[ 0 ];
op[ 1 ] += ip[ 1 ] * dc[ 0 ];
op[ 2 ] += ip[ 2 ] * dc[ 0 ];
op[ 3 ] += ip[ 3 ] * dc[ 0 ];
dc++;
op += 4;
l--;
}
}
else
if( ElCount == 3 )
{
while( l > 0 )
{
op[ 0 ] += ip[ 0 ] * dc[ 0 ];
op[ 1 ] += ip[ 1 ] * dc[ 0 ];
op[ 2 ] += ip[ 2 ] * dc[ 0 ];
dc++;
op += 3;
l--;
}
}
else
if( ElCount == 2 )
{
while( l > 0 )
{
op[ 0 ] += ip[ 0 ] * dc[ 0 ];
op[ 1 ] += ip[ 1 ] * dc[ 0 ];
dc++;
op += 2;
l--;
}
}
}
/**
* Function peforms scanline filtering with optional downsampling.
* Function makes use of the symmetry of the filter.
*
* @param Src Source scanline buffer (length = this -> InLen). Source
* scanline increment will be equal to ElCount.
* @param Dst Destination scanline buffer.
* @param DstIncr Destination scanline buffer increment, used for
* horizontal or vertical scanline stepping.
*/
void doFilter( const fptype* const Src, fptype* Dst,
const int DstIncr ) const
{
const int ElCount = Vars -> ElCount;
const fptype* const f = &Flt[ FltLatency ];
const int flen = FltLatency + 1;
const int ipstep = ElCount * ResampleFactor;
const fptype* ip = Src - EdgePixelCount * ipstep;
const fptype* ip1;
const fptype* ip2;
int l = OutLen;
int i;
if( ElCount == 1 )
{
while( l > 0 )
{
fptype s = f[ 0 ] * ip[ 0 ];
ip1 = ip;
ip2 = ip;
for( i = 1; i < flen; i++ )
{
ip1++;
ip2--;
s += f[ i ] * ( ip1[ 0 ] + ip2[ 0 ]);
}
Dst[ 0 ] = s;
Dst += DstIncr;
ip += ipstep;
l--;
}
}
else
if( ElCount == 4 )
{
while( l > 0 )
{
fptype s1 = f[ 0 ] * ip[ 0 ];
fptype s2 = f[ 0 ] * ip[ 1 ];
fptype s3 = f[ 0 ] * ip[ 2 ];
fptype s4 = f[ 0 ] * ip[ 3 ];
ip1 = ip;
ip2 = ip;
for( i = 1; i < flen; i++ )
{
ip1 += 4;
ip2 -= 4;
s1 += f[ i ] * ( ip1[ 0 ] + ip2[ 0 ]);
s2 += f[ i ] * ( ip1[ 1 ] + ip2[ 1 ]);
s3 += f[ i ] * ( ip1[ 2 ] + ip2[ 2 ]);
s4 += f[ i ] * ( ip1[ 3 ] + ip2[ 3 ]);
}
Dst[ 0 ] = s1;
Dst[ 1 ] = s2;
Dst[ 2 ] = s3;
Dst[ 3 ] = s4;
Dst += DstIncr;
ip += ipstep;
l--;
}
}
else
if( ElCount == 3 )
{
while( l > 0 )
{
fptype s1 = f[ 0 ] * ip[ 0 ];
fptype s2 = f[ 0 ] * ip[ 1 ];
fptype s3 = f[ 0 ] * ip[ 2 ];
ip1 = ip;
ip2 = ip;
for( i = 1; i < flen; i++ )
{
ip1 += 3;
ip2 -= 3;
s1 += f[ i ] * ( ip1[ 0 ] + ip2[ 0 ]);
s2 += f[ i ] * ( ip1[ 1 ] + ip2[ 1 ]);
s3 += f[ i ] * ( ip1[ 2 ] + ip2[ 2 ]);
}
Dst[ 0 ] = s1;
Dst[ 1 ] = s2;
Dst[ 2 ] = s3;
Dst += DstIncr;
ip += ipstep;
l--;
}
}
else
if( ElCount == 2 )
{
while( l > 0 )
{
fptype s1 = f[ 0 ] * ip[ 0 ];
fptype s2 = f[ 0 ] * ip[ 1 ];
ip1 = ip;
ip2 = ip;
for( i = 1; i < flen; i++ )
{
ip1 += 2;
ip2 -= 2;
s1 += f[ i ] * ( ip1[ 0 ] + ip2[ 0 ]);
s2 += f[ i ] * ( ip1[ 1 ] + ip2[ 1 ]);
}
Dst[ 0 ] = s1;
Dst[ 1 ] = s2;
Dst += DstIncr;
ip += ipstep;
l--;
}
}
}
/**
* Function performs resizing of a single scanline. This function does
* not "know" about the length of the source scanline buffer. This buffer
* should be padded with enough pixels so that ( SrcPos - FilterLenD2 ) is
* always >= 0 and ( SrcPos + ( DstLineLen - 1 ) * k + FilterLenD2 + 1 )
* does not exceed source scanline's buffer length. SrcLine's increment is
* assumed to be equal to ElCount.
*
* @param SrcLine Source scanline buffer.
* @param DstLine Destination (resized) scanline buffer.
* @param DstLineIncr Destination scanline position increment, used for
* horizontal or vertical scanline stepping.
* @param xx Temporary buffer, of size FltBank -> getFilterLen(), must be
* aligned by fpclass :: fpalign.
*/
void doResize( const fptype* SrcLine, fptype* DstLine,
const int DstLineIncr, fptype* const ) const
{
const int IntFltLen = FltBank -> getFilterLen();
const int ElCount = Vars -> ElCount;
const typename CImageResizerFilterStep< fptype, fptypeatom > ::
CResizePos* rpos = &(*RPosBuf)[ 0 ];
const typename CImageResizerFilterStep< fptype, fptypeatom > ::
CResizePos* const rpose = rpos + OutLen;
#define AVIR_RESIZE_PART1 \
while( rpos < rpose ) \
{ \
const fptype x = (fptype) rpos -> x; \
const fptype* const ftp = rpos -> ftp; \
const fptype* const ftp2 = ftp + IntFltLen; \
const fptype* Src = SrcLine + rpos -> SrcOffs; \
int i;
#define AVIR_RESIZE_PART1nx \
while( rpos < rpose ) \
{ \
const fptype* const ftp = rpos -> ftp; \
const fptype* Src = SrcLine + rpos -> SrcOffs; \
int i;
#define AVIR_RESIZE_PART2 \
DstLine += DstLineIncr; \
rpos++; \
}
if( FltBank -> getOrder() == 1 )
{
if( ElCount == 1 )
{
AVIR_RESIZE_PART1
fptype sum = 0.0;
for( i = 0; i < IntFltLen; i++ )
{
sum += ( ftp[ i ] + ftp2[ i ] * x ) * Src[ i ];
}
DstLine[ 0 ] = sum;
AVIR_RESIZE_PART2
}
else
if( ElCount == 4 )
{
AVIR_RESIZE_PART1
fptype sum[ 4 ];
sum[ 0 ] = 0.0;
sum[ 1 ] = 0.0;
sum[ 2 ] = 0.0;
sum[ 3 ] = 0.0;
for( i = 0; i < IntFltLen; i++ )
{
const fptype xx = ftp[ i ] + ftp2[ i ] * x;
sum[ 0 ] += xx * Src[ 0 ];
sum[ 1 ] += xx * Src[ 1 ];
sum[ 2 ] += xx * Src[ 2 ];
sum[ 3 ] += xx * Src[ 3 ];
Src += 4;
}
DstLine[ 0 ] = sum[ 0 ];
DstLine[ 1 ] = sum[ 1 ];
DstLine[ 2 ] = sum[ 2 ];
DstLine[ 3 ] = sum[ 3 ];
AVIR_RESIZE_PART2
}
else
if( ElCount == 3 )
{
AVIR_RESIZE_PART1
fptype sum[ 3 ];
sum[ 0 ] = 0.0;
sum[ 1 ] = 0.0;
sum[ 2 ] = 0.0;
for( i = 0; i < IntFltLen; i++ )
{
const fptype xx = ftp[ i ] + ftp2[ i ] * x;
sum[ 0 ] += xx * Src[ 0 ];
sum[ 1 ] += xx * Src[ 1 ];
sum[ 2 ] += xx * Src[ 2 ];
Src += 3;
}
DstLine[ 0 ] = sum[ 0 ];
DstLine[ 1 ] = sum[ 1 ];
DstLine[ 2 ] = sum[ 2 ];
AVIR_RESIZE_PART2
}
else
if( ElCount == 2 )
{
AVIR_RESIZE_PART1
fptype sum[ 2 ];
sum[ 0 ] = 0.0;
sum[ 1 ] = 0.0;
for( i = 0; i < IntFltLen; i++ )
{
const fptype xx = ftp[ i ] + ftp2[ i ] * x;
sum[ 0 ] += xx * Src[ 0 ];
sum[ 1 ] += xx * Src[ 1 ];
Src += 2;
}
DstLine[ 0 ] = sum[ 0 ];
DstLine[ 1 ] = sum[ 1 ];
AVIR_RESIZE_PART2
}
}
else
{
if( ElCount == 1 )
{
AVIR_RESIZE_PART1nx
fptype sum = 0.0;
for( i = 0; i < IntFltLen; i++ )
{
sum += ftp[ i ] * Src[ i ];
}
DstLine[ 0 ] = sum;
AVIR_RESIZE_PART2
}
else
if( ElCount == 4 )
{
AVIR_RESIZE_PART1nx
fptype sum[ 4 ];
sum[ 0 ] = 0.0;
sum[ 1 ] = 0.0;
sum[ 2 ] = 0.0;
sum[ 3 ] = 0.0;
for( i = 0; i < IntFltLen; i++ )
{
const fptype xx = ftp[ i ];
sum[ 0 ] += xx * Src[ 0 ];
sum[ 1 ] += xx * Src[ 1 ];
sum[ 2 ] += xx * Src[ 2 ];
sum[ 3 ] += xx * Src[ 3 ];
Src += 4;
}
DstLine[ 0 ] = sum[ 0 ];
DstLine[ 1 ] = sum[ 1 ];
DstLine[ 2 ] = sum[ 2 ];
DstLine[ 3 ] = sum[ 3 ];
AVIR_RESIZE_PART2
}
else
if( ElCount == 3 )
{
AVIR_RESIZE_PART1nx
fptype sum[ 3 ];
sum[ 0 ] = 0.0;
sum[ 1 ] = 0.0;
sum[ 2 ] = 0.0;
for( i = 0; i < IntFltLen; i++ )
{
const fptype xx = ftp[ i ];
sum[ 0 ] += xx * Src[ 0 ];
sum[ 1 ] += xx * Src[ 1 ];
sum[ 2 ] += xx * Src[ 2 ];
Src += 3;
}
DstLine[ 0 ] = sum[ 0 ];
DstLine[ 1 ] = sum[ 1 ];
DstLine[ 2 ] = sum[ 2 ];
AVIR_RESIZE_PART2
}
else
if( ElCount == 2 )
{
AVIR_RESIZE_PART1nx
fptype sum[ 2 ];
sum[ 0 ] = 0.0;
sum[ 1 ] = 0.0;
for( i = 0; i < IntFltLen; i++ )
{
const fptype xx = ftp[ i ];
sum[ 0 ] += xx * Src[ 0 ];
sum[ 1 ] += xx * Src[ 1 ];
Src += 2;
}
DstLine[ 0 ] = sum[ 0 ];
DstLine[ 1 ] = sum[ 1 ];
AVIR_RESIZE_PART2
}
}
}
#undef AVIR_RESIZE_PART2
#undef AVIR_RESIZE_PART1nx
#undef AVIR_RESIZE_PART1
};
/**
* @brief Image resizer's default dithering class.
*
* This class defines an object that performs rounding, clipping and dithering
* operations over horizontal scanline pixels before scanline is stored in the
* output buffer.
*
* The ditherer should expect the same storage order of the pixels in a
* scanline as used in the "filtering step" class. So, a separate ditherer
* class should be defined for each scanline pixel storage style. The default
* ditherer implements a simple rounding without dithering: it can be used for
* an efficient dithering method which can be multi-threaded.
*
* @tparam fptype Floating point type to use for storing pixel data. SIMD
* types can be used.
*/
template< class fptype >
class CImageResizerDithererDefINL
{
public:
/**
* Function initializes the ditherer object.
*
* @param aLen Scanline length in pixels to process.
* @param aVars Image resizing-related variables.
* @param aTrMul Bit-depth truncation multiplier. 1 - no additional
* truncation.
* @param aPkOut Peak output value allowed.
*/
void init( const int aLen, const CImageResizerVars& aVars,
const double aTrMul, const double aPkOut )
{
Len = aLen;
Vars = &aVars;
LenE = aLen * Vars -> ElCount;
TrMul0 = aTrMul;
PkOut0 = aPkOut;
}
/**
* @return "True" if dithering is recursive relative to scanlines meaning
* multi-threaded execution is not supported by this dithering method.
*/
static bool isRecursive()
{
return( false );
}
/**
* Function performs rounding and clipping operations.
*
* @param ResScanline The buffer containing the final scanline.
*/
void dither( fptype* const ResScanline ) const
{
const fptype c0 = 0.0;
const fptype PkOut = (fptype) PkOut0;
int j;
if( TrMul0 == 1.0 )
{
// Optimization - do not perform bit depth truncation.
for( j = 0; j < LenE; j++ )
{
ResScanline[ j ] = clamp( round( ResScanline[ j ]), c0,
PkOut );
}
}
else
{
const fptype TrMul = (fptype) TrMul0;
for( j = 0; j < LenE; j++ )
{
const fptype z0 = round( ResScanline[ j ] / TrMul ) * TrMul;
ResScanline[ j ] = clamp( z0, c0, PkOut );
}
}
}
protected:
int Len; ///< Scanline's length in pixels.
///<
const CImageResizerVars* Vars; ///< Image resizing-related variables.
///<
int LenE; ///< = LenE * ElCount.
///<
double TrMul0; ///< Bit-depth truncation multiplier.
///<
double PkOut0; ///< Peak output value allowed.
///<
};
/**
* @brief Image resizer's error-diffusion dithering class, interleaved mode.
*
* This ditherer implements error-diffusion dithering which looks good, and
* whose results are compressed by PNG well. This implementation uses
* weighting coefficients obtained via machine optimization and visual
* evaluation.
*
* @tparam fptype Floating point type to use for storing pixel data. SIMD
* types can be used.
*/
template< class fptype >
class CImageResizerDithererErrdINL :
public CImageResizerDithererDefINL< fptype >
{
public:
/**
* Function initializes the ditherer object.
*
* @param aLen Scanline length in pixels to process.
* @param aVars Image resizing-related variables.
* @param aTrMul Bit-depth truncation multiplier. 1 - no additional
* truncation.
* @param aPkOut Peak output value allowed.
*/
void init( const int aLen, const CImageResizerVars& aVars,
const double aTrMul, const double aPkOut )
{
CImageResizerDithererDefINL< fptype > :: init( aLen, aVars, aTrMul,
aPkOut );
ResScanlineDith0.alloc( LenE + Vars -> ElCount, sizeof( fptype ));
ResScanlineDith = ResScanlineDith0 + Vars -> ElCount;
int i;
for( i = 0; i < LenE + Vars -> ElCount; i++ )
{
ResScanlineDith0[ i ] = 0.0;
}
}
static bool isRecursive()
{
return( true );
}
void dither( fptype* const ResScanline )
{
const int ElCount = Vars -> ElCount;
const fptype c0 = 0.0;
const fptype TrMul = (fptype) TrMul0;
const fptype PkOut = (fptype) PkOut0;
int j;
for( j = 0; j < LenE; j++ )
{
ResScanline[ j ] += ResScanlineDith[ j ];
ResScanlineDith[ j ] = 0.0;
}
for( j = 0; j < LenE - ElCount; j++ )
{
// Perform rounding, noise estimation and saturation.
const fptype z0 = round( ResScanline[ j ] / TrMul ) * TrMul;
const fptype Noise = ResScanline[ j ] - z0;
ResScanline[ j ] = clamp( z0, c0, PkOut );
ResScanline[ j + ElCount ] += Noise * (fptype) 0.364842;
ResScanlineDith[ j - ElCount ] += Noise * (fptype) 0.207305;
ResScanlineDith[ j ] += Noise * (fptype) 0.364842;
ResScanlineDith[ j + ElCount ] += Noise * (fptype) 0.063011;
}
while( j < LenE )
{
const fptype z0 = round( ResScanline[ j ] / TrMul ) * TrMul;
const fptype Noise = ResScanline[ j ] - z0;
ResScanline[ j ] = clamp( z0, c0, PkOut );
ResScanlineDith[ j - ElCount ] += Noise * (fptype) 0.207305;
ResScanlineDith[ j ] += Noise * (fptype) 0.364842;
j++;
}
}
protected:
using CImageResizerDithererDefINL< fptype > :: Len;
using CImageResizerDithererDefINL< fptype > :: Vars;
using CImageResizerDithererDefINL< fptype > :: LenE;
using CImageResizerDithererDefINL< fptype > :: TrMul0;
using CImageResizerDithererDefINL< fptype > :: PkOut0;
CBuffer< fptype > ResScanlineDith0; ///< Error diffusion buffer.
///<
fptype* ResScanlineDith; ///< Error diffusion buffer pointer which skips
///< the first ElCount elements.
///<
};
/**
* @brief Floating-point processing definition and abstraction class.
*
* This class defines several constants and typedefs that point to classes
* that should be used by the image resizing algorithm. Such "definition
* class" can be used to define alternative scanline processing algorithms
* (e.g. SIMD) and image scanline packing styles used during processing. This
* class also offers an abstraction layer for dithering, rounding and
* clamping (saturation) operation.
*
* The fpclass_def class can be used to define processing using both SIMD and
* non-SIMD types, but using algorithms that are operate on interleaved pixels
* and non-SIMD optimized themselves.
*
* @tparam afptype Floating point type to use for storing intermediate data
* and variables. For variables that are not used in intensive calculations
* the "double" type is always used. On the latest Intel processors (like
* i7-4770K) there is almost no performance difference between "double" and
* "float". Image quality differences between "double" and "float" are not
* apparent on 8-bit images. At the same time the "float" uses half amount of
* working memory the "double" type uses. SIMD types can be used. The
* functions round() and clamp() in the "avir" or other visible namespace
* should be available for the specified type. SIMD types allow to perform
* resizing of images with more than 4 channels, to be exact 4 * SIMD element
* number (e.g. 16 for float4), without modification of the image resizing
* algorithm required.
* @tparam afptypeatom The atomic type the "afptype" consists of.
* @tparam adith Ditherer class to use during processing.
*/
template< class afptype, class afptypeatom = afptype,
class adith = CImageResizerDithererDefINL< afptype > >
class fpclass_def
{
public:
typedef afptype fptype; ///< Floating-point type to use during processing.
///<
typedef afptypeatom fptypeatom; ///< Atomic type "fptype" consists of.
///<
static const int fppack = sizeof( fptype ) / sizeof( fptypeatom ); ///<
///< The number of atomic types stored in a single "fptype" element.
///<
static const int fpalign = sizeof( fptype ); ///< Suggested alignment size
///< in bytes. This is not a required alignment, because image
///< resizing algorithm cannot be made to have a strictly aligned data
///< access at all steps (e.g. interpolation cannot perform aligned
///< accesses).
///<
static const int elalign = 1; ///< Length alignment of arrays of elements.
///< This applies to filters and intermediate buffers: this constant
///< forces filters and scanlines to have a length which is a multiple
///< of this value, for more efficient SIMD implementation.
///<
static const int packmode = 0; ///< 0 if interleaved packing, 1 if
///< de-interleaved.
///<
typedef CImageResizerFilterStepINL< fptype, fptypeatom > CFilterStep; ///<
///< Filtering step class to use during processing.
///<
typedef adith CDitherer; ///< Ditherer class to use during processing.
///<
};
/**
* @brief Image resizer class.
*
* The object of this class can be used to resize 1-4 channel images to any
* required size. Resizing is performed by utilizing interpolated sinc
* fractional delay filters plus (if necessary) a cascade of built-in
* sinc function-based 2X upsampling or 2X downsampling stages, followed by a
* correction filtering.
*
* Object of this class can be allocated on stack.
*
* @tparam fpclass Floating-point processing definition class to use. See
* avir::fpclass_def for more details.
*/
template< class fpclass = fpclass_def< float > >
class CImageResizer
{
public:
/**
* Constructor initializes the resizer.
*
* @param aResBitDepth Required bit depth of resulting image (1-16). If
* integer value output is used (e.g. uint8_t), the bit depth also affects
* rounding: for example, if aResBitDepth=6 and "Tout" is uint8_t, the
* result will be rounded to 6 most significant bits (2 least significant
* bits truncated, with dithering applied).
* @param aSrcBitDepth Source image's real bit-depth. Set to 0 to use
* aResBitDepth.
* @param aParams Resizing algorithm's parameters to use. Leave out for
* default values. Can be useful when performing automatic optimization of
* parameters.
*/
CImageResizer( const int aResBitDepth = 8, const int aSrcBitDepth = 0,
const CImageResizerParams& aParams = CImageResizerParamsDef() )
: Params( aParams )
, ResBitDepth( aResBitDepth )
{
SrcBitDepth = ( aSrcBitDepth == 0 ? ResBitDepth : aSrcBitDepth );
initFilterBank( FixedFilterBank, 1.0, false, CFltBuffer() );
FixedFilterBank.createAllFilters();
}
/**
* Function resizes image.
*
* @param SrcBuf Source image buffer.
* @param SrcWidth Source image width.
* @param SrcHeight Source image height.
* @param SrcScanlineSize Physical size of source scanline in elements
* (not bytes). If this value is below 1, SrcWidth * ElCountIO will be
* used as the physical source scanline size.
* @param[out] NewBuf Buffer to accept the resized image. Can be equal to
* SrcBuf if the size of the resized image is smaller or equal to source
* image in size.
* @param NewWidth New image width.
* @param NewHeight New image height.
* @param ElCountIO The number of elements (channels) used to store each
* source and destination pixel (1-4).
* @param k Resizing step (one output pixel corresponds to "k" input
* pixels). A downsizing factor if > 1.0; upsizing factor if <= 1.0.
* Multiply by -1 if you would like to bypass "ox" and "oy" adjustment
* which is done by default to produce a centered image. If step value
* equals 0, the step value will be chosen automatically and independently
* for horizontal and vertical resizing.
* @param[in,out] aVars Pointer to variables structure to be passed to the
* image resizing function. Can be NULL. Only variables that are
* initialized in default constructor of this structure are accepted by
* this function. These variables will not be changed by this function.
* All other variables can be modified by this function. The access to
* this object is not thread-safe, each concurrent instance of this
* function should use a separate aVars object.
* @tparam Tin Input buffer element's type. Can be uint8_t (0-255 value
* range), uint16_t (0-65535 value range), float (0.0-1.0 value range),
* double (0.0-1.0 value range). Larger integer types are treated as
* uint16_t. Signed integer types are unsupported.
* @tparam Tout Output buffer element's type. Can be uint8_t (0-255 value
* range), uint16_t (0-65535 value range), float (0.0-1.0 value range),
* double (0.0-1.0 value range). Larger integer types are treated as
* uint16_t. Signed integer types are unsupported.
*/
template< class Tin, class Tout >
void resizeImage( const Tin* const SrcBuf, const int SrcWidth,
const int SrcHeight, int SrcScanlineSize, Tout* const NewBuf,
const int NewWidth, const int NewHeight, const int ElCountIO,
const double k, CImageResizerVars* const aVars = nullptr ) const
{
if( SrcWidth == 0 || SrcHeight == 0 )
{
memset( NewBuf, 0, (size_t) NewWidth * NewHeight *
sizeof( Tout ));
return;
}
else
if( NewWidth == 0 || NewHeight == 0 )
{
return;
}
CImageResizerVars DefVars;
CImageResizerVars& Vars = ( aVars == nullptr ? DefVars : *aVars );
CImageResizerThreadPool DefThreadPool;
CImageResizerThreadPool& ThreadPool = ( Vars.ThreadPool == nullptr ?
DefThreadPool : *Vars.ThreadPool );
// Define resizing steps, also optionally modify offsets so that
// resizing produces a "centered" image.
double kx;
double ky;
double ox = Vars.ox;
double oy = Vars.oy;
if( k == 0.0 )
{
if( NewWidth > SrcWidth )
{
kx = (double) ( SrcWidth - 1 ) / ( NewWidth - 1 );
}
else
{
kx = (double) SrcWidth / NewWidth;
ox += ( kx - 1.0 ) * 0.5;
}
if( NewHeight > SrcHeight )
{
ky = (double) ( SrcHeight - 1 ) / ( NewHeight - 1 );
}
else
{
ky = (double) SrcHeight / NewHeight;
oy += ( ky - 1.0 ) * 0.5;
}
}
else
if( k > 0.0 )
{
kx = k;
ky = k;
if( k > 1.0 )
{
const double ko = ( k - 1.0 ) * 0.5;
ox += ko;
oy += ko;
}
}
else
{
kx = -k;
ky = -k;
}
// Evaluate pre-multipliers used on the output stage.
const bool IsInFloat = ( (Tin) 0.4 != 0 );
const bool IsOutFloat = ( (Tout) 0.4 != 0 );
double OutMul; // Output multiplier.
if( Vars.UseSRGBGamma )
{
if( IsInFloat )
{
Vars.InGammaMult = 1.0;
}
else
{
Vars.InGammaMult =
1.0 / ( sizeof( Tin ) == 1 ? 255.0 : 65535.0 );
}
if( IsOutFloat )
{
Vars.OutGammaMult = 1.0;
}
else
{
Vars.OutGammaMult = ( sizeof( Tout ) == 1 ? 255.0 : 65535.0 );
}
OutMul = 1.0;
}
else
{
if( IsOutFloat )
{
OutMul = 1.0;
}
else
{
OutMul = ( sizeof( Tout ) == 1 ? 255.0 : 65535.0 );
}
if( !IsInFloat )
{
OutMul /= ( sizeof( Tin ) == 1 ? 255.0 : 65535.0 );
}
}
// Fill widely-used variables.
const int ElCount = ( ElCountIO + fpclass :: fppack - 1 ) /
fpclass :: fppack;
const int NewWidthE = NewWidth * ElCount;
if( SrcScanlineSize < 1 )
{
SrcScanlineSize = SrcWidth * ElCountIO;
}
Vars.ElCount = ElCount;
Vars.ElCountIO = ElCountIO;
Vars.fppack = fpclass :: fppack;
Vars.fpalign = fpclass :: fpalign;
Vars.elalign = fpclass :: elalign;
Vars.packmode = fpclass :: packmode;
// Horizontal scanline filtering and resizing.
CDSPFracFilterBankLin< fptype > FltBank;
CFilterSteps FltSteps;
typename CFilterStep :: CRPosBufArray RPosBufArray;
CBuffer< uint8_t > UsedFracMap;
// Perform the filtering steps modeling at various modes, find the
// most efficient mode for both horizontal and vertical resizing.
int UseBuildMode = 1;
const int BuildModeCount =
( FixedFilterBank.getOrder() == 0 ? 4 : 2 );
int m;
if( Vars.BuildMode >= 0 )
{
UseBuildMode = Vars.BuildMode;
}
else
{
int BestScore = 0x7FFFFFFF;
for( m = 0; m < BuildModeCount; m++ )
{
CDSPFracFilterBankLin< fptype > TmpBank;
CFilterSteps TmpSteps;
Vars.k = kx;
Vars.o = ox;
buildFilterSteps( TmpSteps, Vars, TmpBank, OutMul, m, true );
updateFilterStepBuffers( TmpSteps, Vars, RPosBufArray,
SrcWidth, NewWidth );
fillUsedFracMap( TmpSteps[ Vars.ResizeStep ], UsedFracMap );
const int c = calcComplexity( TmpSteps, Vars, UsedFracMap,
SrcHeight );
if( c < BestScore )
{
UseBuildMode = m;
BestScore = c;
}
}
}
// Perform the actual filtering steps building.
Vars.k = kx;
Vars.o = ox;
buildFilterSteps( FltSteps, Vars, FltBank, OutMul, UseBuildMode,
false );
updateFilterStepBuffers( FltSteps, Vars, RPosBufArray, SrcWidth,
NewWidth );
updateBufLenAndRPosPtrs( FltSteps, Vars, NewWidth );
const int ThreadCount = ThreadPool.getSuggestedWorkloadCount();
// Includes the current thread.
CStructArray< CThreadData< Tin, Tout > > td;
td.setItemCount( ThreadCount );
int i;
for( i = 0; i < ThreadCount; i++ )
{
if( i > 0 )
{
ThreadPool.addWorkload( &td[ i ]);
}
td[ i ].init( i, ThreadCount, FltSteps, Vars );
td[ i ].initScanlineQueue( td[ i ].sopResizeH, SrcHeight,
SrcWidth );
}
CBuffer< fptype, size_t > FltBuf( (size_t) NewWidthE * SrcHeight,
fpclass :: fpalign ); // Temporary buffer that receives
// horizontally-filtered and resized image.
for( i = 0; i < SrcHeight; i++ )
{
td[ i % ThreadCount ].addScanlineToQueue(
(void*) &SrcBuf[ (size_t) i * SrcScanlineSize ],
&FltBuf[ (size_t) i * NewWidthE ]);
}
ThreadPool.startAllWorkloads();
td[ 0 ].processScanlineQueue();
ThreadPool.waitAllWorkloadsToFinish();
// Vertical scanline filtering and resizing, reuse previously defined
// filtering steps if possible.
const int PrevUseBuildMode = UseBuildMode;
if( Vars.BuildMode >= 0 )
{
UseBuildMode = Vars.BuildMode;
}
else
{
CImageResizerVars TmpVars( Vars );
int BestScore = 0x7FFFFFFF;
for( m = 0; m < BuildModeCount; m++ )
{
CDSPFracFilterBankLin< fptype > TmpBank;
TmpBank.copyInitParams( FltBank );
CFilterSteps TmpSteps;
TmpVars.k = ky;
TmpVars.o = oy;
buildFilterSteps( TmpSteps, TmpVars, TmpBank, 1.0, m, true );
updateFilterStepBuffers( TmpSteps, TmpVars, RPosBufArray,
SrcHeight, NewHeight );
fillUsedFracMap( TmpSteps[ TmpVars.ResizeStep ],
UsedFracMap );
const int c = calcComplexity( TmpSteps, TmpVars, UsedFracMap,
NewWidth );
if( c < BestScore )
{
UseBuildMode = m;
BestScore = c;
}
}
}
Vars.k = ky;
Vars.o = oy;
if( UseBuildMode == PrevUseBuildMode && ky == kx )
{
if( OutMul != 1.0 )
{
modifyCorrFilterDCGain( FltSteps, 1.0 / OutMul );
}
}
else
{
buildFilterSteps( FltSteps, Vars, FltBank, 1.0, UseBuildMode,
false );
}
updateFilterStepBuffers( FltSteps, Vars, RPosBufArray, SrcHeight,
NewHeight );
updateBufLenAndRPosPtrs( FltSteps, Vars, NewWidth );
if( IsOutFloat && sizeof( FltBuf[ 0 ]) == sizeof( Tout ) &&
fpclass :: packmode == 0 )
{
// In-place output.
for( i = 0; i < ThreadCount; i++ )
{
td[ i ].initScanlineQueue( td[ i ].sopResizeV, NewWidth,
SrcHeight, NewWidthE, NewWidthE );
}
for( i = 0; i < NewWidth; i++ )
{
td[ i % ThreadCount ].addScanlineToQueue(
&FltBuf[ (size_t) i * ElCount ],
(fptype*) (void*) &NewBuf[ (size_t) i * ElCount ]);
}
ThreadPool.startAllWorkloads();
td[ 0 ].processScanlineQueue();
ThreadPool.waitAllWorkloadsToFinish();
ThreadPool.removeAllWorkloads();
return;
}
CBuffer< fptype, size_t > ResBuf( (size_t) NewWidthE * NewHeight,
fpclass :: fpalign );
for( i = 0; i < ThreadCount; i++ )
{
td[ i ].initScanlineQueue( td[ i ].sopResizeV, NewWidth,
SrcHeight, NewWidthE, NewWidthE );
}
const int im = ( fpclass :: packmode == 0 ? ElCount : 1 );
for( i = 0; i < NewWidth; i++ )
{
td[ i % ThreadCount ].addScanlineToQueue(
&FltBuf[ (size_t) i * im ], &ResBuf[ (size_t) i * im ]);
}
ThreadPool.startAllWorkloads();
td[ 0 ].processScanlineQueue();
ThreadPool.waitAllWorkloadsToFinish();
if( IsOutFloat )
{
// Perform output, but skip dithering.
for( i = 0; i < ThreadCount; i++ )
{
td[ i ].initScanlineQueue( td[ i ].sopUnpackH,
NewHeight, NewWidth );
}
for( i = 0; i < NewHeight; i++ )
{
td[ i % ThreadCount ].addScanlineToQueue(
&ResBuf[ (size_t) i * NewWidthE ],
&NewBuf[ (size_t) i * NewWidth * ElCountIO ]);
}
ThreadPool.startAllWorkloads();
td[ 0 ].processScanlineQueue();
ThreadPool.waitAllWorkloadsToFinish();
ThreadPool.removeAllWorkloads();
return;
}
// Perform output with dithering (for integer output only).
int TruncBits; // The number of lower bits to truncate and dither.
int OutRange; // Output range.
if( sizeof( Tout ) == 1 )
{
TruncBits = 8 - ResBitDepth;
OutRange = 255;
}
else
{
TruncBits = 16 - ResBitDepth;
OutRange = 65535;
}
const double PkOut = OutRange;
const double TrMul = ( TruncBits > 0 ?
PkOut / ( OutRange >> TruncBits ) : 1.0 );
if( CDitherer :: isRecursive() )
{
td[ 0 ].getDitherer().init( NewWidth, Vars, TrMul, PkOut );
if( Vars.UseSRGBGamma )
{
for( i = 0; i < NewHeight; i++ )
{
fptype* const ResScanline =
&ResBuf[ (size_t) i * NewWidthE ];
CFilterStep :: applySRGBGamma( ResScanline, NewWidth,
Vars );
td[ 0 ].getDitherer().dither( ResScanline );
CFilterStep :: unpackScanline( ResScanline,
&NewBuf[ (size_t) i * NewWidth * ElCountIO ],
NewWidth, Vars );
}
}
else
{
for( i = 0; i < NewHeight; i++ )
{
fptype* const ResScanline =
&ResBuf[ (size_t) i * NewWidthE ];
td[ 0 ].getDitherer().dither( ResScanline );
CFilterStep :: unpackScanline( ResScanline,
&NewBuf[ (size_t) i * NewWidth * ElCountIO ],
NewWidth, Vars );
}
}
}
else
{
for( i = 0; i < ThreadCount; i++ )
{
td[ i ].initScanlineQueue( td[ i ].sopDitherAndUnpackH,
NewHeight, NewWidth );
td[ i ].getDitherer().init( NewWidth, Vars, TrMul, PkOut );
}
for( i = 0; i < NewHeight; i++ )
{
td[ i % ThreadCount ].addScanlineToQueue(
&ResBuf[ (size_t) i * NewWidthE ],
&NewBuf[ (size_t) i * NewWidth * ElCountIO ]);
}
ThreadPool.startAllWorkloads();
td[ 0 ].processScanlineQueue();
ThreadPool.waitAllWorkloadsToFinish();
}
ThreadPool.removeAllWorkloads();
}
private:
typedef typename fpclass :: fptype fptype; ///< Floating-point type to use
///< during processing.
///<
typedef typename fpclass :: CFilterStep CFilterStep; ///< Filtering step
///< class to use during processing.
///<
typedef typename fpclass :: CDitherer CDitherer; ///< Ditherer class to
///< use during processing.
///<
CImageResizerParams Params; ///< Algorithm's parameters currently in use.
///<
int SrcBitDepth; ///< Bit resolution of the source image.
///<
int ResBitDepth; ///< Bit resolution of the resulting image.
///<
CDSPFracFilterBankLin< fptype > FixedFilterBank; ///< Fractional delay
///< filter bank with fixed characteristics, mainly for upsizing
///< cases.
///<
/**
* @brief Filtering steps array.
*
* The object of this class stores filtering steps together.
*/
typedef CStructArray< CFilterStep > CFilterSteps;
/**
* Function initializes the filter bank in the specified resizing step
* according to the source and resulting image bit depths.
*
* @param FltBank Filter bank to initialize.
* @param CutoffMult Cutoff multiplier, 0 to 1. 1 corresponds to 0.5pi
* cutoff point.
* @param ForceHiOrder "True" if a high-order interpolation should be
* forced which requires considerably less resources for initialization.
* @param ExtFilter External filter to apply to interpolation filter.
*/
void initFilterBank( CDSPFracFilterBankLin< fptype >& FltBank,
const double CutoffMult, const bool ForceHiOrder,
const CFltBuffer& ExtFilter ) const
{
const int IntBitDepth = ( ResBitDepth > SrcBitDepth ? ResBitDepth :
SrcBitDepth );
const double SNR = -6.02 * ( IntBitDepth + 3 );
int UseOrder;
int FracCount; // The number of fractional delay filters sampled by
// the filter bank. This variable affects the signal-to-noise
// ratio at interpolation stage. Theoretically, at UseOrder==1,
// 8-bit image resizing requires 66.2 dB SNR or 11. 16-bit
// resizing requires 114.4 dB SNR or 150. At UseOrder=0 the
// required number of filters is exponentially higher.
if( ForceHiOrder || IntBitDepth > 8 )
{
UseOrder = 1; // -146 dB max
FracCount = (int) ceil( 0.23134052 * exp( -0.058062929 * SNR ));
}
else
{
UseOrder = 0; // -72 dB max
FracCount = (int) ceil( 0.33287686 * exp( -0.11334583 * SNR ));
}
if( FracCount < 2 )
{
FracCount = 2;
}
FltBank.init( FracCount, UseOrder, Params.IntFltLen / CutoffMult,
Params.IntFltCutoff * CutoffMult, Params.IntFltAlpha, ExtFilter,
fpclass :: fpalign, fpclass :: elalign );
}
/**
* Function allocates filter buffer taking "fpclass" alignments into
* account. The allocated buffer may be larger than the requested size: in
* this case the additional elements will be zeroed by this function.
*
* @param Flt Filter buffer.
* @param ReqCapacity The required filter buffer's capacity.
* @param IsModel "True" if filtering steps modeling is performed without
* actual filter allocation.
* @param FltExt If non-NULL this variable will receive the number of
* elements the filter was extended by.
*/
static void allocFilter( CBuffer< fptype >& Flt, const int ReqCapacity,
const bool IsModel = false, int* const FltExt = nullptr )
{
int UseCapacity = ( ReqCapacity + fpclass :: elalign - 1 ) &
~( fpclass :: elalign - 1 );
int Ext = UseCapacity - ReqCapacity;
if( FltExt != nullptr )
{
*FltExt = Ext;
}
if( IsModel )
{
Flt.forceCapacity( UseCapacity );
return;
}
Flt.alloc( UseCapacity, fpclass :: fpalign );
while( Ext > 0 )
{
Ext--;
Flt[ ReqCapacity + Ext ] = 0.0;
}
}
/**
* Function assigns filter parameters to the specified filtering step
* object.
*
* @param fs Filtering step to assign parameter to. This step cannot be
* the last step if ResampleFactor greater than 1 was specified.
* @param IsUpsample "True" if upsampling step. Should be set to "false"
* if FltCutoff is negative.
* @param ResampleFactor Resampling factor of this filter (>=1).
* @param FltCutoff Filter cutoff point. This value will be divided by the
* ResampleFactor if IsUpsample equals "true". If zero value was
* specified, the "half-band" predefined filter will be created. In this
* case the ResampleFactor will modify the filter cutoff point.
* @param DCGain DC gain to apply to the filter. Assigned to filtering
* step's DCGain variable.
* @param UseFltOrig "True" if the originally-designed filter should be
* left in filtering step's FltOrig buffer. Otherwise it will be freed.
* @param IsModel "True" if filtering steps modeling is performed without
* actual filter building.
*/
void assignFilterParams( CFilterStep& fs, const bool IsUpsample,
const int ResampleFactor, const double FltCutoff, const double DCGain,
const bool UseFltOrig, const bool IsModel ) const
{
double FltAlpha;
double Len2;
double Freq;
if( FltCutoff == 0.0 )
{
const double m = 2.0 / ResampleFactor;
FltAlpha = Params.HBFltAlpha;
Len2 = 0.5 * Params.HBFltLen / m;
Freq = AVIR_PI * Params.HBFltCutoff * m;
}
else
{
FltAlpha = Params.LPFltAlpha;
Len2 = 0.25 * Params.LPFltBaseLen / FltCutoff;
Freq = AVIR_PI * Params.LPFltCutoffMult * FltCutoff;
}
if( IsUpsample )
{
Len2 *= ResampleFactor;
Freq /= ResampleFactor;
fs.DCGain = DCGain * ResampleFactor;
}
else
{
fs.DCGain = DCGain;
}
fs.FltOrig.Len2 = Len2;
fs.FltOrig.Freq = Freq;
fs.FltOrig.Alpha = FltAlpha;
fs.FltOrig.DCGain = fs.DCGain;
CDSPPeakedCosineLPF w( Len2, Freq, FltAlpha );
fs.IsUpsample = IsUpsample;
fs.ResampleFactor = ResampleFactor;
fs.FltLatency = w.fl2;
int FltExt; // Filter's extension due to fpclass :: elalign.
if( IsModel )
{
allocFilter( fs.Flt, w.FilterLen, true, &FltExt );
if( UseFltOrig )
{
// Allocate a real buffer even in modeling mode since this
// filter may be copied by the filter bank.
fs.FltOrig.alloc( w.FilterLen );
memset( &fs.FltOrig[ 0 ], 0,
w.FilterLen * sizeof( fs.FltOrig[ 0 ]));
}
}
else
{
fs.FltOrig.alloc( w.FilterLen );
w.generateLPF( &fs.FltOrig[ 0 ], 1.0 );
optimizeFIRFilter( fs.FltOrig, fs.FltLatency );
normalizeFIRFilter( &fs.FltOrig[ 0 ], fs.FltOrig.getCapacity(),
fs.DCGain );
allocFilter( fs.Flt, fs.FltOrig.getCapacity(), false, &FltExt );
copyArray( &fs.FltOrig[ 0 ], &fs.Flt[ 0 ],
fs.FltOrig.getCapacity() );
if( !UseFltOrig )
{
fs.FltOrig.free();
}
}
if( IsUpsample )
{
int l = fs.Flt.getCapacity() - fs.FltLatency - ResampleFactor -
FltExt;
allocFilter( fs.PrefixDC, l, IsModel );
allocFilter( fs.SuffixDC, fs.FltLatency, IsModel );
if( IsModel )
{
return;
}
// Create prefix and suffix "tails" used during upsampling.
const fptype* ip = &fs.Flt[ fs.FltLatency + ResampleFactor ];
copyArray( ip, &fs.PrefixDC[ 0 ], l );
while( true )
{
ip += ResampleFactor;
l -= ResampleFactor;
if( l <= 0 )
{
break;
}
addArray( ip, &fs.PrefixDC[ 0 ], l );
}
l = fs.FltLatency;
fptype* op = &fs.SuffixDC[ 0 ];
copyArray( &fs.Flt[ 0 ], op, l );
while( true )
{
op += ResampleFactor;
l -= ResampleFactor;
if( l <= 0 )
{
break;
}
addArray( &fs.Flt[ 0 ], op, l );
}
}
else
if( !UseFltOrig )
{
fs.EdgePixelCount = fs.EdgePixelCountDef;
}
}
/**
* Function adds a correction filter that tries to achieve a linear
* frequency response at all frequencies. The actual resulting response
* may feature a slight damping of the highest frequencies since a
* suitably short correction filter cannot fix steep high-frequency
* damping.
*
* This function assumes that the resizing step is currently the last
* step, even if it was not inserted yet: this allows placement of the
* correction filter both before and after the resizing step.
*
* @param Steps Filtering steps.
* @param bw Resulting bandwidth relative to the original bandwidth (which
* is 1.0), usually 1/k. Should be <= 1.0.
* @param IsPreCorrection "True" if the filtering step was already created
* and it is first in the Steps array. "True" also adds edge pixels to
* reduce edge artifacts.
* @param IsModel "True" if filtering steps modeling is performed without
* actual filter building.
*/
void addCorrectionFilter( CFilterSteps& Steps, const double bw,
const bool IsPreCorrection, const bool IsModel ) const
{
CFilterStep& fs = ( IsPreCorrection ? Steps[ 0 ] : Steps.add() );
fs.IsUpsample = false;
fs.ResampleFactor = 1;
fs.DCGain = 1.0;
fs.EdgePixelCount = ( IsPreCorrection ? fs.EdgePixelCountDef : 0 );
if( IsModel )
{
allocFilter( fs.Flt, CDSPFIREQ :: calcFilterLength(
Params.CorrFltLen, fs.FltLatency ), true );
return;
}
const int BinCount = 65; // Frequency response bins to control.
const int BinCount1 = BinCount - 1;
double curbw = 1.0; // Bandwidth of the filter at the current step.
int i;
int j;
double re;
double im;
CBuffer< double > Bins( BinCount ); // Adjustment introduced by all
// steps at all frequencies of interest.
for( j = 0; j < BinCount; j++ )
{
Bins[ j ] = 1.0;
}
const int si = ( IsPreCorrection ? 1 : 0 );
for( i = si; i < Steps.getItemCount() - ( si ^ 1 ); i++ )
{
const CFilterStep& fs = Steps[ i ];
if( fs.IsUpsample )
{
curbw *= fs.ResampleFactor;
if( fs.FltOrig.getCapacity() > 0 )
{
continue;
}
}
const double dcg = 1.0 / fs.DCGain; // DC gain correction.
const fptype* Flt;
int FltLen;
if( fs.ResampleFactor == 0 )
{
Flt = fs.FltBank -> getFilter( 0 );
FltLen = fs.FltBank -> getFilterLen();
}
else
{
Flt = &fs.Flt[ 0 ];
FltLen = fs.Flt.getCapacity();
}
// Calculate frequency response adjustment introduced by the
// filter at this step, within the bounds of bandwidth of
// interest.
for( j = 0; j < BinCount; j++ )
{
const double th = AVIR_PI * bw / curbw * j / BinCount1;
calcFIRFilterResponse( Flt, FltLen, th, re, im );
Bins[ j ] /= sqrt( re * re + im * im ) * dcg;
}
if( !fs.IsUpsample && fs.ResampleFactor > 1 )
{
curbw /= fs.ResampleFactor;
}
}
// Calculate filter.
CDSPFIREQ EQ;
EQ.init( bw * 2.0, Params.CorrFltLen, BinCount, 0.0, bw, false,
Params.CorrFltAlpha );
fs.FltLatency = EQ.getFilterLatency();
CBuffer< double > Filter( EQ.getFilterLength() );
EQ.buildFilter( Bins, &Filter[ 0 ]);
normalizeFIRFilter( &Filter[ 0 ], Filter.getCapacity(), 1.0 );
optimizeFIRFilter( Filter, fs.FltLatency );
normalizeFIRFilter( &Filter[ 0 ], Filter.getCapacity(), 1.0 );
allocFilter( fs.Flt, Filter.getCapacity() );
copyArray( &Filter[ 0 ], &fs.Flt[ 0 ], Filter.getCapacity() );
// Print a theoretically achieved final frequency response at various
// feature sizes (from DC to 1 pixel). Values above 255 means features
// become brighter, values below 255 means features become dimmer.
/* const double sbw = ( bw > 1.0 ? 1.0 / bw : 1.0 );
for( j = 0; j < BinCount; j++ )
{
const double th = AVIR_PI * sbw * j / BinCount1;
calcFIRFilterResponse( &fs.Flt[ 0 ], fs.Flt.getCapacity(),
th, re, im );
printf( "%f\n", sqrt( re * re + im * im ) / Bins[ j ] * 255 );
}
printf( "***\n" );*/
}
/**
* Function adds a sharpening filter if image is being upsized. Such
* sharpening allows to spot interpolation filter's stop-band attenuation:
* if attenuation is too weak, a "dark grid" and other artifacts may
* become visible.
*
* It is assumed that 40 decibel stop-band attenuation should be
* considered a required minimum: this allows application of (deliberately
* strong) 64X sharpening without spotting any artifacts.
*
* @param Steps Filtering steps.
* @param bw Resulting bandwidth relative to the original bandwidth (which
* is 1.0), usually 1/k.
* @param IsModel "True" if filtering steps modeling is performed without
* actual filter building.
*/
static void addSharpenTest( CFilterSteps& Steps, const double bw,
const bool IsModel )
{
if( bw <= 1.0 )
{
return;
}
const double FltLen = 10.0 * bw;
CFilterStep& fs = Steps.add();
fs.IsUpsample = false;
fs.ResampleFactor = 1;
fs.DCGain = 1.0;
fs.EdgePixelCount = 0;
if( IsModel )
{
allocFilter( fs.Flt, CDSPFIREQ :: calcFilterLength( FltLen,
fs.FltLatency ), true );
return;
}
const int BinCount = 200;
CBuffer< double > Bins( BinCount );
int Thresh = (int) round( BinCount / bw * 1.75 );
if( Thresh > BinCount )
{
Thresh = BinCount;
}
int j;
for( j = 0; j < Thresh; j++ )
{
Bins[ j ] = 1.0;
}
for( j = Thresh; j < BinCount; j++ )
{
Bins[ j ] = 256.0;
}
CDSPFIREQ EQ;
EQ.init( bw * 2.0, FltLen, BinCount, 0.0, bw, false, 1.7 );
fs.FltLatency = EQ.getFilterLatency();
CBuffer< double > Filter( EQ.getFilterLength() );
EQ.buildFilter( Bins, &Filter[ 0 ]);
normalizeFIRFilter( &Filter[ 0 ], Filter.getCapacity(), 1.0 );
optimizeFIRFilter( Filter, fs.FltLatency );
normalizeFIRFilter( &Filter[ 0 ], Filter.getCapacity(), 1.0 );
allocFilter( fs.Flt, Filter.getCapacity() );
copyArray( &Filter[ 0 ], &fs.Flt[ 0 ], Filter.getCapacity() );
/* for( j = 0; j < BinCount; j++ )
{
const double th = AVIR_PI * j / ( BinCount - 1 );
double re;
double im;
calcFIRFilterResponse( &fs.Flt[ 0 ], fs.Flt.getCapacity(),
th, re, im );
printf( "%f\n", sqrt( re * re + im * im ));
}
printf( "***\n" );*/
}
/**
* Function builds sequence of filtering steps depending on the specified
* resizing coefficient. The last steps included are always the resizing
* step then (possibly) the correction step.
*
* @param Steps Array that receives filtering steps.
* @param[out] Vars Variables object.
* @param FltBank Filter bank to initialize and use.
* @param DCGain The overall DC gain to apply. This DC gain is applied to
* the first filtering step only (upsampling or filtering step).
* @param ModeFlags Build mode flags to use. This is a bitmap of switches
* that enable or disable certain algorithm features.
* @param IsModel "True" if filtering steps modeling is performed without
* the actual filter allocation and building.
*/
void buildFilterSteps( CFilterSteps& Steps, CImageResizerVars& Vars,
CDSPFracFilterBankLin< fptype >& FltBank, const double DCGain,
const int ModeFlags, const bool IsModel ) const
{
Steps.clear();
const bool DoFltAndIntCombo = (( ModeFlags & 1 ) != 0 ); // Do filter
// and interpolator combining.
const bool ForceHiOrderInt = (( ModeFlags & 2 ) != 0 ); // Force use
// of a higher-order interpolation.
const bool UseHalfband = (( ModeFlags & 4 ) != 0 ); // Use half-band
// filter.
const double bw = 1.0 / Vars.k; // Resulting bandwidth.
const int UpsampleFactor = ( (int) floor( Vars.k ) < 2 ? 2 : 1 );
double IntCutoffMult; // Interpolation filter cutoff multiplier.
CFilterStep* ReuseStep; // If not NULL, resizing step should use
// this step object instead of creating a new one.
CFilterStep* ExtFltStep; // Use FltOrig of this step as the external
// filter to applied to the interpolator.
bool IsPreCorrection; // "True" if the correction filter is applied
// first.
double FltCutoff; // Cutoff frequency of the first filtering step.
double corrbw; ///< Bandwidth at the correction step.
if( Vars.k <= 1.0 )
{
IsPreCorrection = true;
FltCutoff = 1.0;
corrbw = 1.0;
Steps.add();
}
else
{
IsPreCorrection = false;
FltCutoff = bw;
corrbw = bw;
}
// Add 1 upsampling or several downsampling filters.
if( UpsampleFactor > 1 )
{
CFilterStep& fs = Steps.add();
assignFilterParams( fs, true, UpsampleFactor, FltCutoff, DCGain,
DoFltAndIntCombo, IsModel );
IntCutoffMult = FltCutoff * 2.0 / UpsampleFactor;
ReuseStep = nullptr;
ExtFltStep = ( DoFltAndIntCombo ? &fs : nullptr );
}
else
{
int DownsampleFactor;
while( true )
{
DownsampleFactor = (int) floor( 0.5 / FltCutoff );
bool DoHBFltAdd;
if( DownsampleFactor > 16 )
{
// Add half-band filter unconditionally in order to keep
// filter lengths lower for more precise frequency
// response and less edge artifacts.
DoHBFltAdd = true;
DownsampleFactor = 16;
}
else
{
DoHBFltAdd = ( UseHalfband && DownsampleFactor > 1 );
}
if( DoHBFltAdd )
{
assignFilterParams( Steps.add(), false, DownsampleFactor,
0.0, 1.0, false, IsModel );
FltCutoff *= DownsampleFactor;
}
else
{
if( DownsampleFactor < 1 )
{
DownsampleFactor = 1;
}
break;
}
}
CFilterStep& fs = Steps.add();
assignFilterParams( fs, false, DownsampleFactor, FltCutoff,
DCGain, DoFltAndIntCombo, IsModel );
IntCutoffMult = FltCutoff / 0.5;
if( DoFltAndIntCombo )
{
ReuseStep = &fs;
ExtFltStep = &fs;
}
else
{
IntCutoffMult *= DownsampleFactor;
ReuseStep = nullptr;
ExtFltStep = nullptr;
}
}
// Insert resizing and correction steps.
CFilterStep& fs = ( ReuseStep == nullptr ? Steps.add() : *ReuseStep );
Vars.ResizeStep = Steps.getItemCount() - 1;
fs.IsUpsample = false;
fs.ResampleFactor = 0;
fs.DCGain = ( ExtFltStep == nullptr ? 1.0 : ExtFltStep -> DCGain );
initFilterBank( FltBank, IntCutoffMult, ForceHiOrderInt,
( ExtFltStep == nullptr ? fs.FltOrig : ExtFltStep -> FltOrig ));
if( FltBank == FixedFilterBank )
{
fs.FltBank = (CDSPFracFilterBankLin< fptype >*) &FixedFilterBank;
}
else
{
fs.FltBank = &FltBank;
}
addCorrectionFilter( Steps, corrbw, IsPreCorrection, IsModel );
//addSharpenTest( Steps, bw, IsModel );
}
/**
* Function extends *this upsampling step so that it produces more
* upsampled pixels that cover the prefix and suffix needs of the next
* step. After the call to this function the InPrefix and InSuffix
* variables of the next step will be set to zero.
*
* @param fs Upsampling filtering step.
* @param NextStep The next step structure.
*/
static void extendUpsample( CFilterStep& fs, CFilterStep& NextStep )
{
fs.InPrefix = ( NextStep.InPrefix + fs.ResampleFactor - 1 ) /
fs.ResampleFactor;
fs.OutPrefix += fs.InPrefix * fs.ResampleFactor;
NextStep.InPrefix = 0;
fs.InSuffix = ( NextStep.InSuffix + fs.ResampleFactor - 1 ) /
fs.ResampleFactor;
fs.OutSuffix += fs.InSuffix * fs.ResampleFactor;
NextStep.InSuffix = 0;
}
/**
* Function fills resizing step's RPosBuf array, excluding the actual
* "ftp" pointers and "SrcOffs" offsets.
*
* This array should be cleared if the resizing step or offset were
* changed. Otherwise this function only fills the elements required to
* cover resizing step's OutLen.
*
* This function is called by the updateFilterStepBuffers() function.
*
* @param fs Resizing step.
* @param Vars Variables object.
*/
static void fillRPosBuf( CFilterStep& fs, const CImageResizerVars& Vars )
{
const int PrevLen = fs.RPosBuf -> getCapacity();
if( fs.OutLen > PrevLen )
{
fs.RPosBuf -> increaseCapacity( fs.OutLen );
}
typename CFilterStep :: CResizePos* rpos = &(*fs.RPosBuf)[ PrevLen ];
const int FracCount = fs.FltBank -> getFracCount();
const double o = Vars.o;
const double k = Vars.k;
int i;
for( i = PrevLen; i < fs.OutLen; i++ )
{
const double SrcPos = o + k * i;
const int SrcPosInt = (int) floor( SrcPos );
const double x = ( SrcPos - SrcPosInt ) * FracCount;
const int fti = (int) x;
rpos -> x = (typename fpclass :: fptypeatom) ( x - fti );
rpos -> fti = fti;
rpos -> SrcPosInt = SrcPosInt;
rpos++;
}
}
/**
* Function updates filtering step buffer lengths depending on the
* specified source and new scanline lengths. This function should be
* called after the buildFilterSteps() function.
*
* @param Steps Array that receives filtering steps.
* @param[out] Vars Variables object, will receive buffer size and length.
* This function expects "k" and "o" variable values that will be
* adjusted by this function.
* @param RPosBufArray Resizing position buffers array, used to obtain
* buffer to initialize and use (will be reused if it is already fully or
* partially filled).
* @param SrcLen Source scanline's length in pixels.
* @param NewLen New scanline's length in pixels.
*/
static void updateFilterStepBuffers( CFilterSteps& Steps,
CImageResizerVars& Vars,
typename CFilterStep :: CRPosBufArray& RPosBufArray, int SrcLen,
const int NewLen )
{
int upstep = -1;
int InBuf = 0;
int i;
for( i = 0; i < Steps.getItemCount(); i++ )
{
CFilterStep& fs = Steps[ i ];
fs.Vars = &Vars;
fs.InLen = SrcLen;
fs.InBuf = InBuf;
fs.OutBuf = ( InBuf + 1 ) & 1;
if( fs.IsUpsample )
{
upstep = i;
Vars.k *= fs.ResampleFactor;
Vars.o *= fs.ResampleFactor;
fs.InPrefix = 0;
fs.InSuffix = 0;
fs.OutLen = fs.InLen * fs.ResampleFactor;
fs.OutPrefix = fs.FltLatency;
fs.OutSuffix = fs.Flt.getCapacity() - fs.FltLatency -
fs.ResampleFactor;
int l0 = fs.OutPrefix + fs.OutLen + fs.OutSuffix;
int l = fs.InLen * fs.ResampleFactor +
fs.SuffixDC.getCapacity();
if( l > l0 )
{
fs.OutSuffix += l - l0;
}
l0 = fs.OutLen + fs.OutSuffix;
if( fs.PrefixDC.getCapacity() > l0 )
{
fs.OutSuffix += fs.PrefixDC.getCapacity() - l0;
}
}
else
if( fs.ResampleFactor == 0 )
{
const int FilterLenD2 = fs.FltBank -> getFilterLen() / 2;
const int FilterLenD21 = FilterLenD2 - 1;
const int ResizeLPix = (int) floor( Vars.o ) - FilterLenD21;
fs.InPrefix = ( ResizeLPix < 0 ? -ResizeLPix : 0 );
const int ResizeRPix = (int) floor( Vars.o +
( NewLen - 1 ) * Vars.k ) + FilterLenD2 + 1;
fs.InSuffix = ( ResizeRPix > fs.InLen ?
ResizeRPix - fs.InLen : 0 );
fs.OutLen = NewLen;
fs.RPosBuf = &RPosBufArray.getRPosBuf( Vars.k, Vars.o,
fs.FltBank -> getFracCount() );
fillRPosBuf( fs, Vars );
}
else
{
Vars.k /= fs.ResampleFactor;
Vars.o /= fs.ResampleFactor;
Vars.o += fs.EdgePixelCount;
fs.InPrefix = fs.FltLatency;
fs.InSuffix = fs.Flt.getCapacity() - fs.FltLatency - 1;
// Additionally extend OutLen to produce more precise edge
// pixels.
fs.OutLen = ( fs.InLen + fs.ResampleFactor - 1 ) /
fs.ResampleFactor + fs.EdgePixelCount;
fs.InSuffix += ( fs.OutLen - 1 ) * fs.ResampleFactor + 1 -
fs.InLen;
fs.InPrefix += fs.EdgePixelCount * fs.ResampleFactor;
fs.OutLen += fs.EdgePixelCount;
}
InBuf = fs.OutBuf;
SrcLen = fs.OutLen;
}
Steps[ Steps.getItemCount() - 1 ].OutBuf = 2;
if( upstep != -1 )
{
extendUpsample( Steps[ upstep ], Steps[ upstep + 1 ]);
}
}
/**
* Function calculates an optimal intermediate buffer length that will
* cover all needs of the specified filtering steps. This function should
* be called after the updateFilterStepBuffers() function.
*
* Function also updates resizing step's RPosBuf pointers to the filter
* bank and SrcOffs values.
*
* @param Steps Filtering steps.
* @param[out] Vars Variables object, will receive buffer size and length.
* @param ResElIncr Resulting (final) element increment, used to produce
* de-interleaved result. For horizontal processing this value is equal
* to last step's OutLen, for vertical processing this value is equal to
* resulting image's width.
*/
static void updateBufLenAndRPosPtrs( CFilterSteps& Steps,
CImageResizerVars& Vars, const int ResElIncr )
{
int MaxPrefix[ 2 ] = { 0, 0 };
int MaxLen[ 2 ] = { 0, 0 };
int i;
for( i = 0; i < Steps.getItemCount(); i++ )
{
CFilterStep& fs = Steps[ i ];
const int ib = fs.InBuf;
if( fs.InPrefix > MaxPrefix[ ib ])
{
MaxPrefix[ ib ] = fs.InPrefix;
}
int l = fs.InLen + fs.InSuffix;
if( l > MaxLen[ ib ])
{
MaxLen[ ib ] = l;
}
fs.InElIncr = fs.InPrefix + l;
if( fs.OutBuf == 2 )
{
break;
}
const int ob = fs.OutBuf;
if( fs.IsUpsample )
{
if( fs.OutPrefix > MaxPrefix[ ob ])
{
MaxPrefix[ ob ] = fs.OutPrefix;
}
l = fs.OutLen + fs.OutSuffix;
if( l > MaxLen[ ob ])
{
MaxLen[ ob ] = l;
}
}
else
{
if( fs.OutLen > MaxLen[ ob ])
{
MaxLen[ ob ] = fs.OutLen;
}
}
}
// Update OutElIncr values of all steps.
for( i = 0; i < Steps.getItemCount(); i++ )
{
CFilterStep& fs = Steps[ i ];
if( fs.OutBuf == 2 )
{
fs.OutElIncr = ResElIncr;
break;
}
CFilterStep& fs2 = Steps[ i + 1 ];
if( fs.IsUpsample )
{
fs.OutElIncr = fs.OutPrefix + fs.OutLen + fs.OutSuffix;
if( fs.OutElIncr > fs2.InElIncr )
{
fs2.InElIncr = fs.OutElIncr;
}
else
{
fs.OutElIncr = fs2.InElIncr;
}
}
else
{
fs.OutElIncr = fs2.InElIncr;
}
}
// Update temporary buffer's length.
for( i = 0; i < 2; i++ )
{
Vars.BufLen[ i ] = MaxPrefix[ i ] + MaxLen[ i ];
Vars.BufOffs[ i ] = MaxPrefix[ i ];
if( Vars.packmode == 0 )
{
Vars.BufOffs[ i ] *= Vars.ElCount;
}
Vars.BufLen[ i ] *= Vars.ElCount;
}
// Update RPosBuf pointers and SrcOffs.
CFilterStep& fs = Steps[ Vars.ResizeStep ];
typename CFilterStep :: CResizePos* rpos = &(*fs.RPosBuf)[ 0 ];
const int em = ( fpclass :: packmode == 0 ? Vars.ElCount : 1 );
const int FilterLenD21 = fs.FltBank -> getFilterLen() / 2 - 1;
for( i = 0; i < fs.OutLen; i++ )
{
rpos -> ftp = fs.FltBank -> getFilter( rpos -> fti );
rpos -> SrcOffs = ( rpos -> SrcPosInt - FilterLenD21 ) * em;
rpos++;
}
}
/**
* Function modifies the overall (DC) gain of the correction filter in the
* pre-built filtering steps array.
*
* @param Steps Filtering steps.
* @param m Multiplier to apply to the correction filter.
*/
void modifyCorrFilterDCGain( CFilterSteps& Steps, const double m ) const
{
CBuffer< fptype >* Flt;
const int z = Steps.getItemCount() - 1;
if( !Steps[ z ].IsUpsample && Steps[ z ].ResampleFactor == 1 )
{
Flt = &Steps[ z ].Flt;
}
else
{
Flt = &Steps[ 0 ].Flt;
}
int i;
for( i = 0; i < Flt -> getCapacity(); i++ )
{
(*Flt)[ i ] = (fptype) ( (double) (*Flt)[ i ] * m );
}
}
/**
* Function builds a map of used fractional delay filters based on the
* resizing positions buffer.
*
* @param fs Resizing step.
* @param[out] UsedFracMap Map of used fractional delay filters.
*/
static void fillUsedFracMap( const CFilterStep& fs,
CBuffer< uint8_t >& UsedFracMap )
{
const int FracCount = fs.FltBank -> getFracCount();
UsedFracMap.increaseCapacity( FracCount, false );
memset( &UsedFracMap[ 0 ], 0, FracCount * sizeof( UsedFracMap[ 0 ]));
typename CFilterStep :: CResizePos* rpos = &(*fs.RPosBuf)[ 0 ];
int i;
for( i = 0; i < fs.OutLen; i++ )
{
UsedFracMap[ rpos -> fti ] |= 1;
rpos++;
}
}
/**
* Function calculates the overall filtering steps complexity per
* scanline. Each complexity unit corresponds to a single multiply-add
* operation. Data copy and pointer math operations are not included in
* this calculation, it is assumed that they correlate to the multiply-add
* operations. Calculation also does not include final rounding, dithering
* and clamping operations since they cannot be optimized out anyway.
*
* Calculation of the CRPosBuf buffer is not included since it cannot be
* avoided.
*
* This function should be called after the updateFilterStepBuffers()
* function.
*
* @param Steps Filtering steps array.
* @param Vars Variables object.
* @param UsedFracMap The map of used fractional delay filters.
* @param ScanlineCount Scanline count.
*/
static int calcComplexity( const CFilterSteps& Steps,
const CImageResizerVars& Vars, const CBuffer< uint8_t >& UsedFracMap,
const int ScanlineCount )
{
int fcnum; // Filter complexity multiplier numerator.
int fcdenom; // Filter complexity multiplier denominator.
if( Vars.packmode != 0 )
{
fcnum = 1;
fcdenom = 1;
}
else
{
// In interleaved processing mode, filters require 1 less
// multiplication per 2 multiply-add instructions.
fcnum = 3;
fcdenom = 4;
}
int s = 0; // Complexity per one scanline.
int s2 = 0; // Complexity per all scanlines.
int i;
for( i = 0; i < Steps.getItemCount(); i++ )
{
const CFilterStep& fs = Steps[ i ];
s2 += 65 * fs.Flt.getCapacity(); // Filter creation complexity.
if( fs.IsUpsample )
{
if( fs.FltOrig.getCapacity() > 0 )
{
continue;
}
s += ( fs.Flt.getCapacity() *
( fs.InPrefix + fs.InLen + fs.InSuffix ) +
fs.SuffixDC.getCapacity() + fs.PrefixDC.getCapacity() ) *
Vars.ElCount;
}
else
if( fs.ResampleFactor == 0 )
{
s += fs.FltBank -> getFilterLen() *
( fs.FltBank -> getOrder() + Vars.ElCount ) * fs.OutLen;
s2 += fs.FltBank -> calcInitComplexity( UsedFracMap );
}
else
{
s += fs.Flt.getCapacity() * Vars.ElCount * fs.OutLen *
fcnum / fcdenom;
}
}
return( s + s2 / ScanlineCount );
}
/**
* @brief Thread-isolated data used for scanline processing.
*
* This structure holds data necessary for image's horizontal or vertical
* scanline processing, including scanline processing queue.
*
* @tparam Tin Source element data type. Intermediate buffers store data
* in floating point format.
* @tparam Tout Destination element data type. Intermediate buffers store
* data in floating point format.
*/
template< class Tin, class Tout >
class CThreadData : public CImageResizerThreadPool :: CWorkload
{
public:
virtual void process()
{
processScanlineQueue();
}
/**
* This enumeration lists possible scanline operations.
*/
enum EScanlineOperation
{
sopResizeH, ///< Resize horizontal scanline.
///<
sopResizeV, ///< Resize vertical scanline.
///<
sopDitherAndUnpackH, ///< Dither and unpack horizontal scanline.
///<
sopUnpackH ///< Unpack horizontal scanline.
///<
};
/**
* Function initializes *this thread data object and assigns certain
* variables provided by the higher level code.
*
* @param aThreadIndex Index of this thread data (0-based).
* @param aThreadCount Total number of threads used during processing.
* @param aSteps Filtering steps.
* @param aVars Image resizer variables.
*/
void init( const int aThreadIndex, const int aThreadCount,
const CFilterSteps& aSteps, const CImageResizerVars& aVars )
{
ThreadIndex = aThreadIndex;
ThreadCount = aThreadCount;
Steps = &aSteps;
Vars = &aVars;
}
/**
* Function initializes scanline processing queue, and updates
* capacities of intermediate buffers.
*
* @param aOp Operation to perform over scanline.
* @param TotalLines The total number of scanlines that will be
* processed by all threads.
* @param aSrcLen Source scanline length in pixels.
* @param aSrcIncr Source scanline buffer increment. Ignored in
* horizontal scanline processing.
* @param aResIncr Resulting scanline buffer increment. Ignored in
* horizontal scanline processing.
*/
void initScanlineQueue( const EScanlineOperation aOp,
const int TotalLines, const int aSrcLen, const int aSrcIncr = 0,
const int aResIncr = 0 )
{
const int l = Vars -> BufLen[ 0 ] + Vars -> BufLen[ 1 ];
if( Bufs.getCapacity() < l )
{
Bufs.alloc( l, fpclass :: fpalign );
}
BufPtrs[ 0 ] = Bufs + Vars -> BufOffs[ 0 ];
BufPtrs[ 1 ] = Bufs + Vars -> BufLen[ 0 ] + Vars -> BufOffs[ 1 ];
int j;
int ml = 0;
for( j = 0; j < Steps -> getItemCount(); j++ )
{
const CFilterStep& fs = (*Steps)[ j ];
if( fs.ResampleFactor == 0 &&
ml < fs.FltBank -> getFilterLen() )
{
ml = fs.FltBank -> getFilterLen();
}
}
TmpFltBuf.alloc( ml, fpclass :: fpalign );
ScanlineOp = aOp;
SrcLen = aSrcLen;
SrcIncr = aSrcIncr;
ResIncr = aResIncr;
QueueLen = 0;
Queue.increaseCapacity(( TotalLines + ThreadCount - 1 ) /
ThreadCount, false );
}
/**
* Function adds a scanline to the queue buffer. The
* initScanlineQueue() function should be called before calling this
* function. The number of calls to this add function should not
* exceed the TotalLines spread over all threads.
*
* @param SrcBuf Source scanline buffer.
* @param ResBuf Resulting scanline buffer.
*/
void addScanlineToQueue( void* const SrcBuf, void* const ResBuf )
{
Queue[ QueueLen ].SrcBuf = SrcBuf;
Queue[ QueueLen ].ResBuf = ResBuf;
QueueLen++;
}
/**
* Function processes all queued scanlines.
*/
void processScanlineQueue()
{
int i;
switch( ScanlineOp )
{
case sopResizeH:
{
for( i = 0; i < QueueLen; i++ )
{
resizeScanlineH( (Tin*) Queue[ i ].SrcBuf,
(fptype*) Queue[ i ].ResBuf );
}
break;
}
case sopResizeV:
{
for( i = 0; i < QueueLen; i++ )
{
resizeScanlineV( (fptype*) Queue[ i ].SrcBuf,
(fptype*) Queue[ i ].ResBuf );
}
break;
}
case sopDitherAndUnpackH:
{
if( Vars -> UseSRGBGamma )
{
for( i = 0; i < QueueLen; i++ )
{
CFilterStep :: applySRGBGamma(
(fptype*) Queue[ i ].SrcBuf, SrcLen, *Vars );
Ditherer.dither( (fptype*) Queue[ i ].SrcBuf );
CFilterStep :: unpackScanline(
(fptype*) Queue[ i ].SrcBuf,
(Tout*) Queue[ i ].ResBuf, SrcLen, *Vars );
}
}
else
{
for( i = 0; i < QueueLen; i++ )
{
Ditherer.dither( (fptype*) Queue[ i ].SrcBuf );
CFilterStep :: unpackScanline(
(fptype*) Queue[ i ].SrcBuf,
(Tout*) Queue[ i ].ResBuf, SrcLen, *Vars );
}
}
break;
}
case sopUnpackH:
{
if( Vars -> UseSRGBGamma )
{
for( i = 0; i < QueueLen; i++ )
{
CFilterStep :: applySRGBGamma(
(fptype*) Queue[ i ].SrcBuf, SrcLen, *Vars );
CFilterStep :: unpackScanline(
(fptype*) Queue[ i ].SrcBuf,
(Tout*) Queue[ i ].ResBuf, SrcLen, *Vars );
}
}
else
{
for( i = 0; i < QueueLen; i++ )
{
CFilterStep :: unpackScanline(
(fptype*) Queue[ i ].SrcBuf,
(Tout*) Queue[ i ].ResBuf, SrcLen, *Vars );
}
}
break;
}
}
}
/**
* Function returns ditherer object associated with *this thread data
* object.
*/
CDitherer& getDitherer()
{
return( Ditherer );
}
private:
int ThreadIndex; ///< Thread index.
///<
int ThreadCount; ///< Thread count.
///<
const CFilterSteps* Steps; ///< Filtering steps.
///<
const CImageResizerVars* Vars; ///< Image resizer variables.
///<
CBuffer< fptype > Bufs; ///< Flip-flop intermediate buffers.
///<
fptype* BufPtrs[ 3 ]; ///< Flip-flop buffer pointers (referenced by
///< filtering step's InBuf and OutBuf indices).
///<
CBuffer< fptype > TmpFltBuf; ///< Temporary buffer used in the
///< doResize() function, aligned by fpclass :: fpalign.
///<
EScanlineOperation ScanlineOp; ///< Operation to perform over
///< scanline.
///<
int SrcLen; ///< Source scanline length in the last queue.
///<
int SrcIncr; ///< Source scanline buffer increment in the last queue.
///<
int ResIncr; ///< Resulting scanline buffer increment in the last
///< queue.
///<
CDitherer Ditherer; ///< Ditherer object to use.
///<
/**
* @brief Scanline processing queue item.
*
* Scanline processing queue item.
*/
struct CQueueItem
{
void* SrcBuf; ///< Source scanline buffer, will by typecasted to
///< Tin or fptype*.
///<
void* ResBuf; ///< Resulting scanline buffer, will by typecasted
///< to Tout or fptype*.
///<
};
CBuffer< CQueueItem > Queue; ///< Scanline processing queue.
///<
int QueueLen; ///< Queue length.
///<
/**
* Function resizes a single horizontal scanline.
*
* @param SrcBuf Source scanline buffer. Can be either horizontal or
* vertical.
* @param ResBuf Resulting scanline buffer.
*/
void resizeScanlineH( const Tin* const SrcBuf, fptype* const ResBuf )
{
(*Steps)[ 0 ].packScanline( SrcBuf, BufPtrs[ 0 ], SrcLen );
BufPtrs[ 2 ] = ResBuf;
int j;
for( j = 0; j < Steps -> getItemCount(); j++ )
{
const CFilterStep& fs = (*Steps)[ j ];
fs.prepareInBuf( BufPtrs[ fs.InBuf ]);
const int DstIncr =
( Vars -> packmode == 0 ? Vars -> ElCount : 1 );
if( fs.ResampleFactor != 0 )
{
if( fs.IsUpsample )
{
fs.doUpsample( BufPtrs[ fs.InBuf ],
BufPtrs[ fs.OutBuf ]);
}
else
{
fs.doFilter( BufPtrs[ fs.InBuf ],
BufPtrs[ fs.OutBuf ], DstIncr );
}
}
else
{
fs.doResize( BufPtrs[ fs.InBuf ], BufPtrs[ fs.OutBuf ],
DstIncr, TmpFltBuf );
}
}
}
/**
* Function resizes a single vertical scanline.
*
* @param SrcBuf Source scanline buffer. Can be either horizontal or
* vertical.
* @param ResBuf Resulting scanline buffer.
*/
void resizeScanlineV( const fptype* const SrcBuf,
fptype* const ResBuf )
{
(*Steps)[ 0 ].convertVtoH( SrcBuf, BufPtrs[ 0 ], SrcLen,
SrcIncr );
BufPtrs[ 2 ] = ResBuf;
int j;
for( j = 0; j < Steps -> getItemCount(); j++ )
{
const CFilterStep& fs = (*Steps)[ j ];
fs.prepareInBuf( BufPtrs[ fs.InBuf ]);
const int DstIncr = ( fs.OutBuf == 2 ? ResIncr :
( Vars -> packmode == 0 ? Vars -> ElCount : 1 ));
if( fs.ResampleFactor != 0 )
{
if( fs.IsUpsample )
{
fs.doUpsample( BufPtrs[ fs.InBuf ],
BufPtrs[ fs.OutBuf ]);
}
else
{
fs.doFilter( BufPtrs[ fs.InBuf ],
BufPtrs[ fs.OutBuf ], DstIncr );
}
}
else
{
fs.doResize( BufPtrs[ fs.InBuf ], BufPtrs[ fs.OutBuf ],
DstIncr, TmpFltBuf );
}
}
}
};
};
#undef AVIR_PI
#undef AVIR_PId2
} // namespace avir
#endif // AVIR_CIMAGERESIZER_INCLUDED