/home/marsyas/marsyas/src/marsyas/SNR.cpp

/*
** Copyright (C) 1998-2006 George Tzanetakis <gtzan@cs.uvic.ca>
**
** This program is free software; you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation; either version 2 of the License, or
** (at your option) any later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software
** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/

#include "SNR.h"

using namespace std;
using namespace Marsyas;

SNR::SNR(mrs_string name):MarSystem("SNR", name)
{
    addControls();
    nsum_ = 0.0;
    nbsum_ = 0.0;
    psum_ = 0.0;
    dsum_ = 0.0;
}

SNR::SNR(const SNR& a) : MarSystem(a)
{
    ctrl_mode_ = getctrl("mrs_string/mode");
    ctrl_done_ = getctrl("mrs_bool/done");
    nsum_ = 0.0;
    dsum_ = 0.0;
    nbsum_ = 0.0;
    psum_ = 0.0;
}


SNR::~SNR()
{
}

MarSystem*
SNR::clone() const
{
    return new SNR(*this);
}

void
SNR::addControls()
{
  addctrl("mrs_string/mode", "standard", ctrl_mode_);
  addctrl("mrs_bool/done", false, ctrl_done_);
  
}

void
SNR::myUpdate(MarControlPtr sender)
{
  (void) sender;
  MRSDIAG("SNR.cpp - SNR:myUpdate");
  ctrl_onSamples_->setValue((mrs_natural)1, NOUPDATE);
  ctrl_onObservations_->setValue((mrs_natural)2, NOUPDATE);
  ctrl_osrate_->setValue(ctrl_israte_, NOUPDATE);
  ctrl_onObsNames_->setValue("SNR_" + ctrl_inObsNames_->to<mrs_string>() , NOUPDATE);


}

void
SNR::myProcess(realvec& in, realvec& out)
{
    const mrs_string& mode = ctrl_mode_->to<mrs_string>();
    mrs_bool accumulateNow = true;

    out.setval (0.);

    mrs_real    nsum    = 0, 
        nbsum   = 0,
        psum    = 0,
        diff    = 0,
        dsum    = 0;

    for (mrs_natural t = 0; t < inSamples_; t++)
    {
        nsum    += (in(0,t) * in(0,t));
        nbsum   += (in(1,t) * in(1,t));
        psum    += (in(0,t) * in(1,t));
        diff    = (in(0,t) - in(1,t));
        dsum    += (diff * diff);
    }

    if (mode == "checkRef4Silence")
    {
        if (nbsum/inSamples_ < 1e-6) //-60dB
            accumulateNow   = false;
    }

    if (accumulateNow)
    {
        nsum_   += nsum;
        nbsum_  += nbsum;
        psum_   += psum;
        dsum_   += dsum;
    }

    if (nsum_ != 0 && dsum_ != 0)
        out(0,0) = 10. * log10(nsum_ / dsum_);

    if (nsum_ != 0 && nbsum_ != 0)
        r_ = (psum_ / sqrt(nsum_ * nbsum_));
    else
        r_ = 0;

    out(1,0) = 10. * log10(1 / (1 - (r_ * r_)));

    if (ctrl_done_->to<mrs_bool>() == true) 
    {
        nsum_ = 0.0;
        nbsum_ = 0.0;
        dsum_ = 0.0;
        psum_ = 0.0;
    }



}

---