/home/marsyas/marsyas/src/marsyas/PeakSynthFFT.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 "common.h"
#include "PeakSynthFFT.h"
#include "peakView.h"

using std::ostringstream;
using namespace Marsyas;

//#define MTLB_DBG_LOG

PeakSynthFFT::PeakSynthFFT(mrs_string name):MarSystem("PeakSynthFFT", name)
{
 
  addControls();
}

PeakSynthFFT::PeakSynthFFT(const PeakSynthFFT& a) : MarSystem(a)
{
    ctrl_Peaks_ = getctrl("mrs_realvec/peaks");
    ctrl_NbChannels_ = getctrl("mrs_natural/nbChannels");
}

PeakSynthFFT::~PeakSynthFFT()
{
}

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

void 
PeakSynthFFT::addControls()
{
    addctrl("mrs_realvec/peaks", realvec(), ctrl_Peaks_);
    addctrl("mrs_natural/nbChannels", 1, ctrl_NbChannels_);
    setctrlState("mrs_natural/nbChannels", true);
    addctrl("mrs_string/panning", "MARSYAS_EMPTY");
    setctrlState("mrs_string/panning", true);
}

void
PeakSynthFFT::myUpdate(MarControlPtr sender)
{
    MarSystem::myUpdate(sender);
  setctrl("mrs_natural/onSamples", getctrl("mrs_natural/inSamples")->to<mrs_natural>()*getctrl("mrs_natural/nbChannels")->to<mrs_natural>());

    realvec conv(4);
    conv.setval(-1);
    if(getctrl("mrs_string/panning")->to<mrs_string>() != "MARSYAS_EMPTY")
    {
    string2parameters(getctrl("mrs_string/panning")->to<mrs_string>(), conv, '_');
    }
    fgVolume_ = conv(0); 
  fgPanning_ = conv(1);
    bgVolume_ = conv(2); 
  bgPanning_ = conv(3);

    mask_.create(getctrl("mrs_natural/inObservations")->to<mrs_natural>()/2+1);
    lastMask_.create(getctrl("mrs_natural/inObservations")->to<mrs_natural>()/2+1);
    lastMask_.setval(0);
}

void
PeakSynthFFT::generateMask(mrs_natural type)
{
    mrs_natural i, j;
    realvec peaks = ctrl_Peaks_->to<mrs_realvec>();

    //cout << peaks;
    mrs_natural nbPeaks = peaks.getSize()/peakView::nbPkParameters;

    // intialize background
    if(bgVolume_ != -1)
    {
        mask_.setval(bgVolume_);
        if(type == 1)
            mask_*=(1-bgPanning_)/2;
        if(type == 2)
            mask_*=(1+bgPanning_)/2;
    }
    else
        mask_.setval(0);

    // set level info for foreground clusters
    for (i=0 ; i<onObservations_/2 + 1; ++i)
    {
        for(j=0 ; j<nbPeaks; j++)
        {
            if(peaks(j+peakView::pkGroup*nbPeaks) <= -1)
                continue;
            if (i>=peaks(j+nbPeaks*peakView::pkBinLow)*onObservations_ && i<=peaks(j+nbPeaks*peakView::pkBinHigh)*onObservations_)
            {
                mrs_real vol, pan;
                if(fgVolume_ != -1)
                {
                    // use global info
                    vol = fgVolume_;
                    pan = fgPanning_;
                }
                else
                {
                    // use peaks info
                    vol = peaks(j+nbPeaks*peakView::pkVolume);
                    pan = peaks(j+nbPeaks*peakView::pkPan);
                }

                mask_(i) = vol;
                // left
                if(type == 1)
                    mask_(i) *= (1-pan)/2;
                // right
                if(type == 2)
                    mask_(i) *= (1+pan)/2;
                break;
            }
        }
    }
//  cout << mask_;
}

void
PeakSynthFFT::lpfMask()
{
    mrs_real gain = 0.8, deltaGain=.3;

    for (mrs_natural i=0 ; i<mask_.getSize() ; ++i)
    {
        mrs_real g=gain-deltaGain*(mask_.getSize()-i)/mask_.getSize();
    mask_(i) = g*mask_(i)+(1-g)*lastMask_(i);
    }
    lastMask_ = mask_;
}

void 
PeakSynthFFT::myProcess(realvec& in, realvec& out)
{   
    mrs_natural nbChannels = ctrl_NbChannels_->to<mrs_natural>();
    mrs_natural t,o;
    //cout << ctrl_Peaks_->to<mrs_realvec>();
    for (t = 0; t < onSamples_; t++)
    {
        generateMask(t+(nbChannels-1));

#ifdef MARSYAS_MATLAB
#ifdef MTLB_DBG_LOG
        MATLAB_PUT(mask_, "mask");
        MATLAB_EVAL("figure(71),plot(mask),axis('tight'),grid on");
#endif
#endif
        lpfMask();
        for (o=0; o < onObservations_/2+1; o++)
        {
            //apply PeakSynthFFT to all channels
            out(o,t) = mask_(o) * in(o, 0);     
        }
        for (o=onObservations_/2+1; o < onObservations_; o++)
        {
            //apply PeakSynthFFT to all channels
            out(o,t) = in(o, 0);        
        }
    }
//  cout << out;
}







    

---