/home/marsyas/marsyas/src/marsyas/Selector.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.
*/

//
// Brought to you by sness@sness.net
//

#include "Selector.h"

using namespace std;
using namespace Marsyas;

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

Selector::Selector(const Selector& a) : MarSystem(a)
{
  ctrl_enabled_ = getctrl("mrs_realvec/enabled");
}

Selector::~Selector()
{
}


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

void
Selector::addControls()
{
  addctrl("mrs_natural/disable", -1);
  setctrlState("mrs_natural/disable", true);
  addctrl("mrs_natural/enable", -1);
  setctrlState("mrs_natural/enable", true);

  addctrl("mrs_realvec/enabled", realvec(), ctrl_enabled_);
}

void
Selector::myUpdate(MarControlPtr sender)
{
  MRSDIAG("Selector.cpp - Selector:myUpdate");

  MarControlAccessor acc(ctrl_enabled_);
  mrs_realvec& enabled = acc.to<mrs_realvec>();

  //
  // If the enabled realvec has not been created, create it now.
  // Enable all observations by default.
  //
  if ((mrs_natural)enabled.getSize() < inObservations_) 
    {
      enabled.create(inObservations_);
      enabled.setval(1.0);
    }

  //
  // Disable any observations that the user asks to be disabled
  //
  disable_ = getctrl("mrs_natural/disable")->to<mrs_natural>();
  if (disable_ != -1 && disable_ < inObservations_)
    {
      enabled(disable_) = 0.0;
      setctrl("mrs_natural/disable", -1);
    }
  else
    setctrl("mrs_natural/disable", -1);

  
  //
  // Enable any observations that the user asks to be disabled
  // 
  enable_ = getctrl("mrs_natural/enable")->to<mrs_natural>();
  if (enable_ != -1 && enable_ < inObservations_)
    {
      enabled(enable_) = 1.0;
      setctrl("mrs_natural/enable", -1);
    }
  else
    setctrl("mrs_natural/enable", -1);

  //
  // Count how many of the observations are enabled
  //
  mrs_natural total_enabled = 0;
  for (mrs_natural i=0; i < enabled.getSize(); ++i) {
    if (enabled(i) > 0.1) // sness - Just in case of floating point roundoff
      total_enabled++;
  }
  
  ctrl_onObservations_->setValue(total_enabled, NOUPDATE);
  ctrl_onSamples_->setValue(ctrl_inSamples_, NOUPDATE);
  ctrl_osrate_->setValue(ctrl_israte_, NOUPDATE);

}

void
Selector::myProcess(realvec& in, realvec& out)
{
  MarControlAccessor acc(ctrl_enabled_);
  mrs_realvec& enabled = acc.to<mrs_realvec>();

  mrs_natural outIndex = 0;
    mrs_natural t,o;

  //
  // Copy all the input observations and samples to the output except
  // for any observations that are not enabled.
  //
  for (o=0; o < inObservations_; o++)
    if (enabled(o)) {
      for (t = 0; t < inSamples_; t++)
        {
          out(outIndex,t) = in(o,t);
        }
      outIndex++;
    }
}

---