AtFilterFFT.h

Go to the documentation of this file.

#ifndef ATFFTFILTER_H
#define ATFFTFILTER_H
 
#include "AtFilter.h"
 
#include <Rtypes.h>
#include <TVirtualFFT.h> // Annoyingly required for ROOT to generate a dictionary (even without IO)
 
#include <map>
#include <memory>
#include <vector>
 
class AtPad;
class AtPadFFT;
class AtRawEvent;
struct AtPadReference;
 
class AtFilterFFT : public AtFilter {
 
   // Forward declares and alias declerations
public:
   struct AtFreqRange {
      Int_t fBeginFreq;
      Double_t fBeginFact;
 
      Int_t fEndFreq;
      Double_t fEndFact;
      friend bool operator<(const AtFreqRange &lhs, const AtFreqRange &rhs);
   };
   using FreqRanges = std::vector<AtFreqRange>;
 
protected:
   FreqRanges fFreqRanges;
   std::map<Int_t, Double_t> fFactors;
 
   std::unique_ptr<TVirtualFFT> fFFT{nullptr};
   std::unique_ptr<TVirtualFFT> fFFTbackward{nullptr};
 
   Bool_t fSaveTransform{false};
   Bool_t fSubtractBackground{true};
 
   AtRawEvent *fInputEvent{nullptr};
   // AtRawEvent *fFilteredEvent{nullptr};
   static constexpr Int_t fTransformSize = 512;
 
public:
   AtFilterFFT() = default;
   ~AtFilterFFT() = default;
 
   bool AddFreqRange(AtFreqRange range); // Range is inclusive
   void SetSaveTransform(bool saveTransform) { fSaveTransform = saveTransform; }
   void SetSubtractBackground(bool subtractBackground) { fSubtractBackground = subtractBackground; }
 
   bool GetSaveTransform() { return fSaveTransform; }
   bool GetSubtractBackground() { return fSubtractBackground; }
 
   const FreqRanges &GetFreqRanges() { return fFreqRanges; }
   void DumpFactors();
 
   void Init() override;
   void InitEvent(AtRawEvent *event = nullptr) override;
   void Filter(AtPad *pad, AtPadReference *padReference) override;
   bool IsGoodEvent() override { return true; }
   void SetLowPass(int order, int cuttoff);
 
protected:
   virtual std::unique_ptr<AtPadFFT> applyFrequencyCutsAndSetInverseFFT();
 
private:
   bool isValidFreqRange(const AtFreqRange &range);
   bool doesFreqRangeOverlap(const AtFreqRange &range);
   double getFilterKernel(int freq, int fFilterOrder, int fCutoffFreq);
};
 
#endif //#ifndef ATFFTFILTER_H