TwoPole.h

#ifndef STK_TWOPOLE_H
#define STK_TWOPOLE_H
 
#include "Filter.h"
 
namespace stk {
 
/***************************************************/
/***************************************************/
 
class TwoPole : public Filter
{
 public:
 
  TwoPole( void );
 
  ~TwoPole();
 
  void ignoreSampleRateChange( bool ignore = true ) { ignoreSampleRateChange_ = ignore; };
 
  void setB0( StkFloat b0 ) { b_[0] = b0; };
 
  void setA1( StkFloat a1 ) { a_[1] = a1; };
 
  void setA2( StkFloat a2 ) { a_[2] = a2; };
 
  void setCoefficients( StkFloat b0, StkFloat a1, StkFloat a2, bool clearState = false );
 
 
  void setResonance(StkFloat frequency, StkFloat radius, bool normalize = false);
 
  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:
 
  virtual void sampleRateChanged( StkFloat newRate, StkFloat oldRate );
};
 
inline StkFloat TwoPole :: tick( StkFloat input )
{
  inputs_[0] = gain_ * input;
  lastFrame_[0] = b_[0] * inputs_[0] - a_[1] * outputs_[1] - a_[2] * outputs_[2];
  outputs_[2] = outputs_[1];
  outputs_[1] = lastFrame_[0];
 
  return lastFrame_[0];
}
 
inline StkFrames& TwoPole :: tick( StkFrames& frames, unsigned int channel )
{
#if defined(_STK_DEBUG_)
  if ( channel >= frames.channels() ) {
    oStream_ << "TwoPole::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_[0] * inputs_[0] - a_[1] * outputs_[1] - a_[2] * outputs_[2];
    outputs_[2] = outputs_[1];
    outputs_[1] = *samples;
  }
 
  lastFrame_[0] = outputs_[1];
  return frames;
}
 
inline StkFrames& TwoPole :: tick( StkFrames& iFrames, StkFrames& oFrames, unsigned int iChannel, unsigned int oChannel )
{
#if defined(_STK_DEBUG_)
  if ( iChannel >= iFrames.channels() || oChannel >= oFrames.channels() ) {
    oStream_ << "TwoPole::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_[0] * inputs_[0] - a_[1] * outputs_[1] - a_[2] * outputs_[2];
    outputs_[2] = outputs_[1];
    outputs_[1] = *oSamples;
  }
 
  lastFrame_[0] = outputs_[1];
  return iFrames;
}
 
} // stk namespace
 
#endif