diff trunk/src/Modules/Features/ModuleGaussians.h @ 268:e14c70d1b171

- Initial add of support code and modules. Not everything is working yet.
author tomwalters
date Fri, 12 Feb 2010 12:31:23 +0000
parents
children c26222c51fb7
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/trunk/src/Modules/Features/ModuleGaussians.h	Fri Feb 12 12:31:23 2010 +0000
@@ -0,0 +1,99 @@
+// 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);
+
+  //! \brief Reset the internal state of the module
+  void Reset();
+
+ private:
+    /*! \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
+   */
+  float m_fParamVar;
+
+  /*! \brief posterior probability expansion exponent
+   */
+  float m_fParamPosteriorExp;
+
+  /*! \brief Maximum Number of iterations
+   */
+  unsigned int m_iParamMaxIt;
+
+  /*! \brief convergence criterion
+   */
+  float m_fParamPriorsConverged;
+
+  /*! \brief The amplitudes of the components (priors)
+   */
+  vector<float> m_pA;
+
+  /*! \brief The means of the components (priors)
+   */
+  vector<float> m_pMu;
+
+  /*! \brief The spectral profile of the incoming buffer
+   */
+  vector<float> m_pSpectralProfile;
+
+  int m_iNumChannels;
+};
+}  // namespace aimc
+
+#endif  // _AIMC_MODULE_FEATURES_GAUSSIANS_H_