comparison src/file_comparer.hpp @ 0:add35537fdbb tip

Initial import

+//  Copyright 2011, Ian Hobson.
+//
+//  This file is part of gpsynth.
+//
+//  gpsynth 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.
+//
+//  gpsynth 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 gpsynth in the file COPYING. 
+//  If not, see http://www.gnu.org/licenses/.
+
+// FileComparer - Compares sounds against a target set of audio features that
+// have been retrieved from FeatureExtractor
+
+#pragma once
+
+#include "feature_extractor.hpp"
+
+#include "boost/shared_ptr.hpp"
+
+#include <map>
+#include <string>
+#include <vector>
+
+namespace dsp {
+
+struct FeatureComparerInterface {
+  virtual ~FeatureComparerInterface() {}
+
+  virtual void AnalyzeTarget() = 0;
+  virtual Value Compare() = 0;
+  virtual FeatureComparerInterface* Clone() const = 0;
+  virtual void SetExtractor(FeatureExtractor* extractor) = 0;
+  virtual int ID() const = 0;
+};
+  
+typedef boost::shared_ptr<FeatureComparerInterface> FeatureComparerPtr; 
+
+
+class FileComparer {
+public:
+  
+  struct Feature {
+    enum ID {
+      Pitch,
+      Energy,
+      MFCCs,
+      DeltaMFCCs,
+      DoubleDeltaMFCCs,
+      Magnitude,
+      LogMagnitude,
+      SpectralCentroid,
+      SpectralSpread,
+      SpectralFlux
+    };
+  };
+
+private:
+  std::string target_file_;
+  Value target_duration_;
+  std::vector<FeatureComparerPtr> features_;
+  FeatureExtractor extractor_;
+  std::map<std::string, Feature::ID> feature_names_;
+  
+public:
+  // copy constructor
+  FileComparer(const FileComparer& other);
+  FileComparer(int window_size = 1024,
+               int hop_size = 256);
+  FileComparer(const std::string& feature_list,
+               int window_size = 1024,
+               int hop_size = 256);
+  
+  void SetTargetFile(const std::string& target_file);
+  Value CompareFile(const std::string& file_path);
+  void SetFeatureExtractorSettings(int window_size, int hop_size);
+  
+  Value TargetDuration() const { return target_duration_; }
+  
+  void EnableFeature(Feature::ID feature, bool enable = true);
+  void EnableFeatures(const std::vector<Feature::ID>& features,
+                      bool enable = true);
+  void EnableFeatures(std::string feature_name_list, bool enable = true);
+  void SetFeatures(const std::vector<Feature::ID>& features);
+};
+
+} // dsp namespace