AnalysisFlow.h

/* -*- C++ -*-.*********************************************************************************************
 Author: Ben Strutt
 Email: strutt@physics.ucla.edu

 Description:
             Applies the analysis in sequence
***********************************************************************************************************/


#ifndef ANALYSIS_FLOW_H
#define ANALYSIS_FLOW_H

#include "AnitaDataset.h"
#include "AnalysisReco.h"
#include "AnalysisSettings.h"
#include "TFile.h"
#include "FilterStrategy.h"

class AnitaEventSummary;
class NoiseMonitor;

namespace Acclaim
{

  class CmdLineArgs;


  class AnalysisFlow : public TObject{

  public:

    enum selection{
      kAll,
      kWaisPulser,
      kDecimated
    };


    static const char* selectionAsString(selection sel){
      switch(sel){
      case kAll: return "all";
      case kWaisPulser: return "wais";
      case kDecimated: return "decimated";
      }
    }


    AnalysisFlow(CmdLineArgs* args, FilterStrategy* filterStrat=NULL);


    ~AnalysisFlow();


    void doAnalysis(Long64_t startAtThisEntry=-1);


    AnitaEventSummary* doEntry(Long64_t entry);

    AnitaEventSummary* doEvent(UInt_t eventNumber);

    AnalysisReco* getReco(){return fReco;} 

    FilteredAnitaEvent* getEvent() {return fEv;}

    const AnitaDataset* getData(){return fData;}

    Long64_t firstEntry(){return fFirstEntry;}

    Long64_t lastEntry(){return fLastEntry;}

  protected:
    
    selection fSelection;       
    int fRun;                           
    int fNumDivisions;          
    int fDivision;          

    AnitaDataset* fData;        
    Long64_t fFirstEntry;       
    Long64_t fLastEntry;        

    AnalysisReco* fReco;        
    FilterStrategy* fFilterStrat;   
    AnalysisSettings* fSettings;    
    AnitaEventSummary* fEventSummary;   
    FilteredAnitaEvent* fEv;        
    NoiseMonitor* fNoiseMonitor;    
    UInt_t fLastEventConsidered;    

    TString fOutFileBaseName;       
    TFile* fOutFile;            
    TTree* fSumTree;            

    void prepareDataSet();


    void prepareOutputFiles();

    Bool_t shouldIDoThisEvent(RawAnitaHeader* header, UsefulAdu5Pat* usefulPat);



    Bool_t isPulserWAIS(RawAnitaHeader* header, UsefulAdu5Pat* usefulPat);



    Bool_t isPulserLDB(RawAnitaHeader* header, UsefulAdu5Pat* usefulPat);


    void setPulserFlags(RawAnitaHeader* header, UsefulAdu5Pat* usefulPat, AnitaEventSummary* sum);


    void prepareEverything(const char* preferredSettingsFileName = NULL);

    Bool_t checkForSgeTaskId();


    ANALYSIS_SETTING(Int_t, Debug);


    ANALYSIS_SETTING(Int_t, DoAll);


    ENUM_ANALYSIS_SETTING(AnitaDataset::BlindingStrategy, BlindStrat);


    ANALYSIS_SETTING(Double_t, NoiseTimeScaleSeconds);


    ANALYSIS_SETTING(Int_t, NoiseEvenWaveforms);


    ANALYSIS_SETTING(Int_t, OutFileCompressionLevel);

    ANALYSIS_SETTING(Int_t, OutFileCompressionAlgo);


    ClassDef(AnalysisFlow, 0);
  };

}


#endif