SpectralDataManager.h

Go to the documentation of this file.

/**********************************************************************
 
  Audacity: A Digital Audio Editor
 
  SpectralDataManager.h
 
  Edward Hui
 
*******************************************************************//*******************************************************************/
 
#include "TrackSpectrumTransformer.h"
#include "Effect.h"
#include "tracks/playabletrack/wavetrack/ui/SpectrumView.h"
 
class WaveChannel;
 
class SpectralDataManager{
public:
   SpectralDataManager();
   ~SpectralDataManager();
   static bool ProcessTracks(AudacityProject &project);
   static int FindFrequencySnappingBin(const WaveChannel &channel,
      long long startSC, int hopSize, double threshold, int targetFreqBin);
 
   static std::vector<int> FindHighestFrequencyBins(WaveChannel &wc,
                                          long long int startSC,
                                          int hopSize,
                                          double threshold,
                                          int targetFreqBin);
private:
   class Worker;
   struct Setting;
};
 
class SpectralDataManager::Worker
      : public TrackSpectrumTransformer{
public:
   Worker(WaveChannel *pChannel, const Setting &setting);
   ~Worker();
 
   struct MyWindow : public Window
   {
      MyWindow(size_t windowSize)
            : Window{ windowSize }
            , mSpectrums(windowSize / 2 + 1)
            , mGains(windowSize / 2 + 1)
      {
 
      }
      ~MyWindow() override;
 
      FloatVector mSpectrums;
      FloatVector mGains;
   };
 
   bool Process(const WaveChannel &channel,
      const std::shared_ptr<SpectralData> &sDataPtr);
   int ProcessSnapping(const WaveChannel &channel,
      long long int startSC, int hopSize, size_t winSize,
      double threshold, int targetFreqBin);
   std::vector<int> ProcessOvertones(const WaveChannel &channel,
      long long int startSC, int hopSize, size_t winSize,
      double threshold, int targetFreqBin);
 
protected:
   MyWindow &NthWindow(int nn) {
      return static_cast<MyWindow&>(Nth(nn));
   }
   std::unique_ptr<Window> NewWindow(size_t windowSize) override;
   bool DoStart() override;
   static bool Processor(SpectrumTransformer &transformer);
   static bool OvertonesProcessor(SpectrumTransformer &transformer);
   static bool SnappingProcessor(SpectrumTransformer &transformer);
   bool DoFinish() override;
 
private:
   bool ApplyEffectToSelection();
   std::shared_ptr<SpectralData> mpSpectralData;
   int mWindowCount { 0 };
   double mSnapSamplingRate;
   double mSnapThreshold;
   double mOvertonesThreshold;
   std::vector<int> mOvertonesTargetFreqBin;
   int mSnapTargetFreqBin;
   int mSnapReturnFreqBin { -1 };
   long long mStartHopNum { 0 };
};