numuCC4piMultiPiSelection.hxx

#ifndef numuCC4piMultiPiSelection_h
#define numuCC4piMultiPiSelection_h

#include "SelectionBase.hxx"
#include "Parameters.hxx"
#include "ToyBoxTracker.hxx"
#include "EventBoxId.hxx"
#include "SystId.hxx"
#include "SubDetId.hxx"

#include "ToFSenseCorrector.hxx"

#include "numuCC4piSelection.hxx"

class numuCC4piMultiPiSelection: public SelectionBase{
public:
    numuCC4piMultiPiSelection(bool forceBreak=true);
    virtual ~numuCC4piMultiPiSelection(){}
    
    //---- These are mandatory functions
    void DefineSteps();
    void DefineDetectorFV();
    inline ToyBoxB* MakeToyBox();
    bool FillEventSummary(AnaEventC& event, Int_t allCutsPassed[]);
    SampleId::SampleEnum GetSampleEnum(){return SampleId::kFGD1NuMuCC0Pi;}
    bool IsRelevantRecObjectForSystematic(const AnaEventC& event, AnaRecObjectC* recObj, SystId_h systId, Int_t branch) const;
    bool IsRelevantRecObjectForSystematicInToy(const AnaEventC&, const ToyBoxB&, AnaRecObjectC*, SystId_h systId, Int_t branch=0) const;
    bool IsRelevantTrueObjectForSystematic(const AnaEventC& event, AnaTrueObjectC* trueObj, SystId_h systId, Int_t branch) const;
    bool IsRelevantTrueObjectForSystematicInToy(const AnaEventC&, const ToyBoxB&, AnaTrueObjectC*, SystId_h systId, Int_t branch=0) const;
    bool IsRelevantSystematic(const AnaEventC& event, const ToyBoxB& box, SystId_h systId, Int_t branch) const;
    void InitializeEvent(AnaEventC& event);
    bool CheckRedoSelection(const AnaEventC& event, const ToyBoxB& PreviousToyBox, Int_t& redoFromStep);
    
private:
    ToFSenseCorrector _tofCorrector;
    
};

class ToyBoxCC4piMultiPi: public ToyBoxCC4pi{
    
public:
    
    ToyBoxCC4piMultiPi() {
        MainTrack = NULL;
        LowAngle.clear(); HighAngle.clear();
        FwdTracks.clear();      BwdTracks.clear();      HAFwdTracks.clear();      HABwdTracks.clear();
        FwdTracks_Veto.clear(); BwdTracks_Veto.clear(); HAFwdTracks_Veto.clear(); HABwdTracks_Veto.clear();
        FwdTracks_PID.clear();  BwdTracks_PID.clear();  HAFwdTracks_PID.clear();  HABwdTracks_PID.clear();
                
        nPositivePionTPCtracks = 0;
        nNegativePionTPCtracks = 0;
        nMichelElectrons = 0;
        nIsoFGDPiontracks = 0;
        nPosPi0TPCtracks = 0;
        nElPi0TPCtracks = 0;
        nIsoFGDElPi0tracks = 0;
        nPosPions = 0;
        nOtherPions = 0;
        
        // For ECal Pi0 veto
        ISOEcal.clear();
        HObject = NULL;
        Pi0Ecaltrack  = NULL; 
        nPi0Ecaltracks = 0;

        for( int i = 0; i < 100; i ++){
            PositivePionTPCtracks[i] = NULL;
            NegativePionTPCtracks[i] = NULL;
            MichelElectrons[i] = NULL;
            IsoFGDPiontracks[i] = NULL;
            PosPi0TPCtracks[i] = NULL;
            ElPi0TPCtracks[i] = NULL;
            IsoFGDElPi0tracks[i] = NULL;
        }
    }
    
    virtual void Reset(){
        ResetBase();
        MainTrack = NULL;
        LowAngle.clear(); HighAngle.clear();
        FwdTracks.clear();      BwdTracks.clear();      HAFwdTracks.clear();      HABwdTracks.clear();
        FwdTracks_Veto.clear(); BwdTracks_Veto.clear(); HAFwdTracks_Veto.clear(); HABwdTracks_Veto.clear();
        FwdTracks_PID.clear();  BwdTracks_PID.clear();  HAFwdTracks_PID.clear();  HABwdTracks_PID.clear(); 
                
        for( int i = 0; i <nPositivePionTPCtracks ; i ++)
          PositivePionTPCtracks[i] = NULL;
        for( int i = 0; i <nNegativePionTPCtracks ; i ++)
          NegativePionTPCtracks[i] = NULL;
        for( int i = 0; i <nMichelElectrons ; i ++)
          MichelElectrons[i] = NULL;
        for( int i = 0; i <nIsoFGDPiontracks ; i ++)
          IsoFGDPiontracks[i] = NULL;
        for( int i = 0; i <nPosPi0TPCtracks ; i ++)
          PosPi0TPCtracks[i] = NULL;
        for( int i = 0; i <nElPi0TPCtracks ; i ++)
          ElPi0TPCtracks[i] = NULL;
        for( int i = 0; i <nIsoFGDElPi0tracks ; i ++)
          IsoFGDElPi0tracks[i] = NULL;

        nPositivePionTPCtracks = 0;
        nNegativePionTPCtracks = 0;
        nMichelElectrons = 0;
        nIsoFGDPiontracks = 0;
        nPosPi0TPCtracks = 0;
        nElPi0TPCtracks = 0;
        nIsoFGDElPi0tracks = 0;
        nPosPions = 0;
        nOtherPions = 0;

        ISOEcal.clear();
        HObject = NULL;
        Pi0Ecaltrack  = NULL;
        nPi0Ecaltracks = 0;
    }
    
    virtual ~ToyBoxCC4piMultiPi(){}
    

    // positive  TPC pion candidates 
    AnaTrackB* PositivePionTPCtracks[100];
    int nPositivePionTPCtracks;

    // negative TPC pion candidates 
    AnaTrackB* NegativePionTPCtracks[100];
    int nNegativePionTPCtracks;

    //  Michel Electron candidates 
    AnaFgdTimeBinB* MichelElectrons[100];
    int nMichelElectrons;

    // IsoFGD Pion candidates 
    AnaTrackB* IsoFGDPiontracks[100];
    int nIsoFGDPiontracks;

    // Electrons and positrons from Pio in the TPC
    AnaTrackB* PosPi0TPCtracks[100];
    AnaTrackB* ElPi0TPCtracks[100];
    int nPosPi0TPCtracks;
    int nElPi0TPCtracks;

    // IsoFGD electron and positron candidates 
    AnaTrackB* IsoFGDElPi0tracks[100];
    int nIsoFGDElPi0tracks;

    int nPosPions;
    int nOtherPions;

    // isoECal track
    std::vector<AnaTrackB*> ISOEcal;
    // Most energetic object in ECal
    AnaTrackB* HObject;
    // Candidate Pi0 from Ecal veto
    AnaTrackB* Pi0Ecaltrack;

    int nPi0Ecaltracks;
};

inline ToyBoxB* numuCC4piMultiPiSelection::MakeToyBox() {return new ToyBoxCC4piMultiPi();}

class FillSummaryAction_numuCC4piMultiPi: public StepBase{
public:
    using StepBase::Apply;
    bool Apply(AnaEventC& event, ToyBoxB& box) const;
    StepBase* MakeClone(){return new FillSummaryAction_numuCC4piMultiPi();}
};

namespace numuCC4piMultiPi{

        class FindPionsAction: public StepBase{
            public:
                using StepBase::Apply;
                FindPionsAction(){
                    useTPCPions = (bool)ND::params().GetParameterI("psycheSelections.numuCC4piMultiPi.UseTPCPions");
                    useME = (bool)ND::params().GetParameterI("psycheSelections.numuCC4piMultiPi.UseME");
                    useFGDPions = (bool)ND::params().GetParameterI("psycheSelections.numuCC4piMultiPi.UseFGDPions");
                    useOldSecondaryPID = (bool) ND::params().GetParameterI("psycheSelections.numuCC4piMultiPi.OldSecondaryPID");
                    prod5Cut = (bool)ND::params().GetParameterI("psycheSelections.numuCC4piMultiPi.Prod5Cuts");
                }
                bool useTPCPions;
                bool useME;
                bool useFGDPions;
                bool useOldSecondaryPID;
                bool prod5Cut;

                bool Apply(AnaEventC& event, ToyBoxB& box) const;  
                StepBase* MakeClone(){return new FindPionsAction();}
        };

        class  FindEcalPhotonsAction: public StepBase{
            public:
                using StepBase::Apply;
                FindEcalPhotonsAction(){
                    MostUpstreamLayerHitCut = (bool)ND::params().GetParameterI("psycheSelections.numuCC4piMultiPi.MostUpstreamLayerHitCut");
                    prod5Cut = (bool)ND::params().GetParameterI("psycheSelections.numuCC4piMultiPi.Prod5Cuts");
                }
                int MostUpstreamLayerHitCut;
                bool prod5Cut;

                bool Apply(AnaEventC& event, ToyBoxB& box) const;  
                StepBase* MakeClone(){return new  FindEcalPhotonsAction();}
        };

        class  EcalPi0VetoCut: public StepBase{
            public:
                using StepBase::Apply;
                bool Apply(AnaEventC& event, ToyBoxB& box) const;  
                StepBase* MakeClone(){return new  EcalPi0VetoCut();}
        };


        class NoPionCut: public StepBase{  
            public:
                using StepBase::Apply;
                bool Apply(AnaEventC& event, ToyBoxB& box) const;
                StepBase* MakeClone(){return new NoPionCut();}
        };

        class OnePionCut: public StepBase{  
            public:
                using StepBase::Apply;
                bool Apply(AnaEventC& event, ToyBoxB& box) const;
                StepBase* MakeClone(){return new OnePionCut();}
        };

        class OthersCut: public StepBase{  
            public:
                using StepBase::Apply;
                bool Apply(AnaEventC& event, ToyBoxB& box) const;
                StepBase* MakeClone(){return new OthersCut();}
        };
}
namespace numuCC4piMultiPiUtils{
  
    //From NumuCCBANFF Standard Actions.
    void FindTPCPions(AnaEventC& event, ToyBoxB& box, SubDetId::SubDetEnum det, bool useOldSecondaryPID);
    void FindIsoFGDPions(AnaEventC& event, ToyBoxB& box, SubDetId::SubDetEnum det);
    void FindIsoFGDElPi0(AnaEventC& event, ToyBoxB& box, SubDetId::SubDetEnum det);
    void FindMEPions(AnaEventC& event, SubDetId::SubDetEnum det, bool prod5Cut = 0);

    //From NumuCCBANFF Analysis Utils.
    bool TPCpionSelection(AnaTrackB *track);
    bool TPCElPi0Selection(AnaTrackB *track);
    bool FGDpionSelection(AnaTrackB *track, SubDetId::SubDetEnum det);
    bool FGDElPi0Selection(AnaEventC& event, ToyBoxB& box, AnaTrackB *track, SubDetId::SubDetEnum det);

    // For ECal Pi0 veto
    bool ECALPi0Selection(AnaEventC& event, ToyBoxB& box, AnaTrackB *HO, int MostUpstreamLayerHitCut, SubDetId::SubDetEnum det);
}


#endif