PitchInverseProblem.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 PITCHINVERSEPROBLEM_H
#define PITCHINVERSEPROBLEM_H

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

#include "PeakDetection.h"
#include "PeakInterpolation.h"

#include <Eigen/LU>

class PitchInverseProblem {
protected:
  int _fftSize;
  int _halfSize;
  Real _lowFrequency;
  Real _highFrequency;
  int _pitchCount;
  int _harmonicCount;
  int _frequencyCandidateCount;
  Real _peakWidth;

  Real _samplerate;

  Real _tMin;
  Real _tMax;
  Real _alpha;
  Real _beta;
  Real _inharmonicity;
  Real _regularisation;

  MatrixXR _projectionMatrix;
  MatrixXR _inverseProjectionMatrix;

  PeakDetection _peak;
  PeakInterpolation _peakInterp;

  void harmonicWeight(MatrixXR f, Real fMin, Real fMax, int harmonicIndex, MatrixXR* result);
  void harmonicSpread(MatrixXR f, Real fMin, Real fMax, int harmonicIndex, MatrixXR* result);
  void harmonicPosition(MatrixXR f, Real fMin, Real fMax, int harmonicIndex, MatrixXR* result);
  Real harmonicWeight(Real f, Real fMin, Real fMax, int harmonicIndex);
  Real harmonicSpread(Real f, Real fMin, Real fMax, int harmonicIndex);
  Real harmonicPosition(Real f, Real fMin, Real fMax, int harmonicIndex);

public:
  PitchInverseProblem(int fftSize = 1024, Real lowFrequency = 50.0, Real highFrequency = 2100.0, Real samplerate = 44100.0, int pitchCount = 5, int harmonicCount = 10, int frequencyCandidateCount = -1, Real peakWidth = 8);

  ~PitchInverseProblem();

  void reset();
  void setup();

  void process(const MatrixXR& spectrum, MatrixXR* pitches, MatrixXR* saliencies, MatrixXR* frequencies);

  void projectionMatrix(MatrixXR* matrix) const;

  Real lowFrequency() const;  

  void setLowFrequency( Real frequency, bool callSetup = true );

  Real highFrequency() const;  

  void setHighFrequency( Real frequency, bool callSetup = true );

  Real samplerate() const;  

  void setSamplerate( Real frequency, bool callSetup = true );

  int fftSize() const;

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

  int frequencyCandidateCount() const;

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

  int peakWidth() const;
  
  void setPeakWidth( int width, bool callSetup = true );

  int pitchCount() const;

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

  int harmonicCount() const;

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

};

#endif  /* PITCHINVERSEPROBLEM_H */

---