comparison trunk/src/Modules/SNR/ModuleNoise.h @ 305:ed91095d9240

-New AIMCopy main for the SSI features (temporary hack till I get a working module load system)
-LocalMax strobe criterion. This is faster and better than the parabola version, which still seems buggy.
-Noise generator module. Adds noise to a signal. Uses boost for the random number generator.
-New options for the SSI
-Slice now respects all its flags (oops!).
-MATLAB functions for visualisation
-Scripts for generating data to view in MATLAB
-Script to download and build HTK - useful for running experiments

+// Copyright 2010, Thomas Walters
+//
+// 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/>.
+
+/*!
+ * \author Thomas Walters <[e-mail]>
+ * \date created 2010/02/24
+ * \version \$Id$
+ */
+
+#ifndef AIMC_MODULES_SNR_NOISE_H_
+#define AIMC_MODULES_SNR_NOISE_H_
+
+#include <boost/random.hpp>
+
+#include "Support/Module.h"
+
+namespace aimc {
+class ModuleNoise : public Module {
+ public:
+  explicit ModuleNoise(Parameters *pParam);
+  virtual ~ModuleNoise();
+
+  /*! \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);
+
+  float sample_rate_;
+  int buffer_length_;
+  int channel_count_;
+
+  float multiplier_;
+
+  // Random number generator which yeilds Gaussian-distributed values by
+  // The generator is a Mersenne twister
+  boost::variate_generator<boost::mt19937,
+                           boost::normal_distribution<float> >
+                           gaussian_variate_;
+};
+}  // namespace aimc
+
+#endif  // AIMC_MODULES_SNR_NOISE_H_