view src/Modules/Features/ModuleGaussians.h @ 3:decdac21cfc2

- Imported file input using libsndfile from old AIM-C and updated to the new API - Modified the Module base class to propogate Reset() calls down the module chain. - This required changing all Reset() functions in subclasses to ResetInternal() - Removed some unneeded imports from the Gaussians test
author tomwalters
date Tue, 16 Feb 2010 18:00:16 +0000
parents e91769e84be1
children fcbf85ce59fb
line wrap: on
line source
// Copyright 2008-2010, University of Cambridge
//
// AIM-C: A C++ implementation of the Auditory Image Model
// http://www.acousticscale.org/AIMC
//
// 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 3 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, see <http://www.gnu.org/licenses/>.

/*! \file ModuleGaussians.h
 *  \brief Gaussian features
 */

/*! \author Tom Walters <tcw24@cam.ac.uk>
 *  \date created 2008/06/23
 *  \version \$Id: ModuleGaussians.h 2 2010-02-02 12:59:50Z tcw $
 */

#ifndef _AIMC_MODULE_FEATURES_GAUSSIANS_H_
#define _AIMC_MODULE_FEATURES_GAUSSIANS_H_

#include <vector>

#include "Support/Module.h"
#include "Support/Parameters.h"
#include "Support/SignalBank.h"

/*! \class ModuleGaussians "Modules/Features/ModuleGaussians.h"
 *  \brief
 */
namespace aimc {
using std::vector;
class ModuleGaussians : public Module
{
 public:
  ModuleGaussians(Parameters *pParam);
  virtual ~ModuleGaussians();

  //! \brief Process a buffer
  virtual void Process(const SignalBank &input);

 private:
  //! \brief Reset the internal state of the module
  virtual void ResetInternal();

  /*! \brief Prepare the module
   *  \param input Input signal
   *  \param output true on success false on failure
   */
  virtual bool InitializeInternal(const SignalBank &input);

  bool RubberGMMCore(int iNumComponents, bool bDoInit);

  /*! \brief Number of Gaussian Components
   */
  int m_iParamNComp;

  /*! \brief Constant variance of Gaussians
   */
  double m_fParamVar;

  /*! \brief posterior probability expansion exponent
   */
  double m_fParamPosteriorExp;

  /*! \brief Maximum Number of iterations
   */
  int m_iParamMaxIt;

  /*! \brief convergence criterion
   */
  double m_fParamPriorsConverged;

  /*! \brief The amplitudes of the components (priors)
   */
  vector<double> m_pA;

  /*! \brief The means of the components (priors)
   */
  vector<double> m_pMu;

  /*! \brief The spectral profile of the incoming buffer
   */
  vector<double> m_pSpectralProfile;

  int m_iNumChannels;
};
}  // namespace aimc

#endif  // _AIMC_MODULE_FEATURES_GAUSSIANS_H_