view src/Modules/SSI/ModuleSSI.h @ 33:f8fe1aadf097

-Modified AIMCopy for slices experiment
-Added gen_features script to just generate features for a given SNR

// 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/19
 * \version \$Id$
 */

#ifndef AIMC_MODULES_SSI_SSI_H_
#define AIMC_MODULES_SSI_SSI_H_

#include "Support/Module.h"

namespace aimc {
using std::vector;
class ModuleSSI : public Module {
 public:
  explicit ModuleSSI(Parameters *pParam);
  virtual ~ModuleSSI();

  /*! \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);

  int ExtractPitchIndex(const SignalBank &input) const;

  float sample_rate_;
  int buffer_length_;
  int channel_count_;
  int ssi_width_samples_;
  float ssi_width_cycles_;
  float pivot_cf_;

  bool do_pitch_cutoff_;
  bool weight_by_cutoff_;
  bool weight_by_scaling_;
  bool log_cycles_axis_;
  float pitch_search_start_ms_;
};
}  // namespace aimc

#endif  // AIMC_MODULES_SSI_SSI_H_