NMF.h

/*                                                         
** Copyright (C) 2008, 2009 Ricard Marxer <email@ricardmarxer.com>
**                                                                  
** 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.
*/                                                                          

#ifndef NMF_H
#define NMF_H

#include "Typedefs.h"
#include "Debug.h"

class NMF {
protected:
  // Internal parameters
  int _inputSize;
  int _componentCount;

  int _iterationCount;
  
  Real _epsilon;
  
  // Internal variables
  MatrixXR _xOverWH;
  MatrixXR _norms;

public:
  NMF(int inputSize = 1024, int componentCount = 3, int iterationCount = 10, Real epsilon = 1e-9);

  ~NMF();

  void reset();
  void setup();

  void process(const MatrixXR& frames, MatrixXR* gains, MatrixXR* components);

  int inputSize() const;  

  void setInputSize( int size, bool callSetup = true );

  int componentCount() const;

  void setComponentCount( int count, bool callSetup = true );

  int iterationCount() const;

  void setIterationCount( int count, bool callSetup = true );

  Real epsilon() const;

  void setEpsilon( Real epsilon, bool callSetup = true );

};

#endif  /* NMF_H */

---