TwoZero.h

#ifndef STK_TWOZERO_H
#define STK_TWOZERO_H
 
#include "Filter.h"
 
namespace stk {
 
/***************************************************/
/***************************************************/
 
class TwoZero : public Filter
{
 public:
  TwoZero();
 
  ~TwoZero();
 
  void ignoreSampleRateChange( bool ignore = true ) { ignoreSampleRateChange_ = ignore; };
 
  void setB0( StkFloat b0 ) { b_[0] = b0; };
 
  void setB1( StkFloat b1 ) { b_[1] = b1; };
 
  void setB2( StkFloat b2 ) { b_[2] = b2; };
 
  void setCoefficients( StkFloat b0, StkFloat b1, StkFloat b2, bool clearState = false );
 
 
  void setNotch( StkFloat frequency, StkFloat radius );
 
  StkFloat lastOut( void ) const { return lastFrame_[0]; };
 
  StkFloat tick( StkFloat input );
 
 
  StkFrames& tick( StkFrames& frames, unsigned int channel = 0 );
 
 
  StkFrames& tick( StkFrames& iFrames, StkFrames &oFrames, unsigned int iChannel = 0, unsigned int oChannel = 0 );
 
 protected:
 
  void sampleRateChanged( StkFloat newRate, StkFloat oldRate );
};
 
inline StkFloat TwoZero :: tick( StkFloat input )
{
  inputs_[0] = gain_ * input;
  lastFrame_[0] = b_[2] * inputs_[2] + b_[1] * inputs_[1] + b_[0] * inputs_[0];
  inputs_[2] = inputs_[1];
  inputs_[1] = inputs_[0];
 
  return lastFrame_[0];
}
 
inline StkFrames& TwoZero :: tick( StkFrames& frames, unsigned int channel )
{
#if defined(_STK_DEBUG_)
  if ( channel >= frames.channels() ) {
    oStream_ << "TwoZero::tick(): channel and StkFrames arguments are incompatible!";
    handleError( StkError::FUNCTION_ARGUMENT );
  }
#endif
 
  StkFloat *samples = &frames[channel];
  unsigned int hop = frames.channels();
  for ( unsigned int i=0; i<frames.frames(); i++, samples += hop ) {
    inputs_[0] = gain_ * *samples;
    *samples = b_[2] * inputs_[2] + b_[1] * inputs_[1] + b_[0] * inputs_[0];
    inputs_[2] = inputs_[1];
    inputs_[1] = inputs_[0];
  }
 
  lastFrame_[0] = *(samples-hop);
  return frames;
}
 
inline StkFrames& TwoZero :: tick( StkFrames& iFrames, StkFrames& oFrames, unsigned int iChannel, unsigned int oChannel )
{
#if defined(_STK_DEBUG_)
  if ( iChannel >= iFrames.channels() || oChannel >= oFrames.channels() ) {
    oStream_ << "TwoZero::tick(): channel and StkFrames arguments are incompatible!";
    handleError( StkError::FUNCTION_ARGUMENT );
  }
#endif
 
  StkFloat *iSamples = &iFrames[iChannel];
  StkFloat *oSamples = &oFrames[oChannel];
  unsigned int iHop = iFrames.channels(), oHop = oFrames.channels();
  for ( unsigned int i=0; i<iFrames.frames(); i++, iSamples += iHop, oSamples += oHop ) {
    inputs_[0] = gain_ * *iSamples;
    *oSamples = b_[2] * inputs_[2] + b_[1] * inputs_[1] + b_[0] * inputs_[0];
    inputs_[2] = inputs_[1];
    inputs_[1] = inputs_[0];
  }
 
  lastFrame_[0] = *(oSamples-oHop);
  return iFrames;
}
 
} // stk namespace
 
#endif