Correlator.h

#ifndef UCORRELATOR_CORRELATOR_H
#define UCORRELATOR_CORRELATOR_H


#include "AnitaConventions.h"
#include "TH2.h"

class FilteredAnitaEvent; 
class AnalysisWaveform; 

#define NANTENNAS NUM_SEAVEYS

namespace UCorrelator
{

  class TrigCache; 

  //This is just to hide the OpenMP stuff from cling
  class CorrelatorLocks; 

  class Correlator
  {
    public:
      Correlator(int nphi, double phimin, double phimax, int ntheta, double theta_lowest, double theta_highest, bool use_bin_center = false, bool scale_by_cos_theta = false, double baseline_weight = 0, double gain_sigma = 0); 

      void compute(const FilteredAnitaEvent * event, AnitaPol::AnitaPol_t pol); 

      const TH2D * getHist() const { return hist; } 
 
      const TH2D * getNorm() const { return norm; } 
      

      TH2D* computeZoomed(double phi, double theta, int nphi, double dphi,  int ntheta, double dtheta, int nant = 0, TH2D * useme = 0); 

      void setDisallowedAntennas(ULong64_t disallowed) { disallowed_antennas = disallowed; } 

      void setAllowedAntennas(ULong64_t allowed) { disallowed_antennas = ~allowed; } 

      void setMaxAntennaMaxPhiDistance(double max_ant_phi) { max_phi = max_ant_phi;  max_phi2 = max_phi * max_phi; } 

      void setGroupDelayFlag(bool flag) { groupDelayFlag = flag; } 


      const AnalysisWaveform * getCorrelationGraph(int ant1, int ant2) { return getCorrelation(ant1,ant2); }

      void setPadFactor(int pad) { pad_factor = pad; } 

      void dumpDeltaTs(const char * file) const; 
      virtual ~Correlator(); 

    private: 
      AnalysisWaveform* padded_waveforms[NANTENNAS]; 
      AnalysisWaveform* correlations[NANTENNAS][NANTENNAS]; 

      TH2D *hist; 
      TH2D *norm; 

      std::vector<TH2D*> hists; 
      std::vector<TH2D*> norms; 

#ifndef NUM_ANITAS
#define NUM_ANITAS 4
#endif
      TrigCache * trigcache[NUM_ANITAS+1]; 
      double rms[NANTENNAS]; 

      double max_phi, max_phi2;
      ULong64_t disallowed_antennas;
      int pad_factor;
      const FilteredAnitaEvent * ev; 
      AnitaPol::AnitaPol_t pol; 
      bool groupDelayFlag; 
      bool use_bin_center; 
      bool scale_cos_theta; 
      double baselineWeight;
      double gainSigma; 

      AnalysisWaveform * getCorrelation(int ant1, int ant2); 
      void doAntennas(int ant1, int ant2, TH2D ** hist, TH2D ** norm, const UCorrelator::TrigCache * tc, const double * center_point  = 0); 
      void reset(); 

      CorrelatorLocks * locks; 
  }; 



}



#endif