numuCCMultiPiSelection.hxx

#ifndef numuCCMultiPiSelection_h
#define numuCCMultiPiSelection_h

#include "numuCCSelection.hxx"
#include "Parameters.hxx"

class numuCCMultiPiSelection: public SelectionBase{

public:
  numuCCMultiPiSelection(bool forceBreak=true);
  virtual ~numuCCMultiPiSelection(){}
  
  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 IsRelevantTrueObjectForSystematic(const AnaEventC& event, AnaTrueObjectC* trueObj, SystId_h systId, Int_t branch) const;
  bool IsRelevantRecObjectForSystematicInToy(const AnaEventC& event, const ToyBoxB& box, AnaRecObjectC* recObj, SystId_h systId, Int_t branch) const;
  bool IsRelevantTrueObjectForSystematicInToy(const AnaEventC& event, const ToyBoxB& box, AnaTrueObjectC* trueTrack, SystId_h systId, Int_t branch) 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);
  
protected:
  numuCCSelection _numuCCSelection;

  Int_t _MuonPIDCutIndex;
  Int_t _FindPionsStepIndex;   
};

class ToyBoxCCMultiPi: public ToyBoxTracker{

    public:
        ToyBoxCCMultiPi(){
            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 ~ToyBoxCCMultiPi(){}
  
        void Reset(){
            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;
        }

        // 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* numuCCMultiPiSelection::MakeToyBox(){return new ToyBoxCCMultiPi();}

namespace numuCCMultiPiUtils{
        //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);
}


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();}
};

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

class FindPionsAction: public StepBase{
    public:
        using StepBase::Apply;
        FindPionsAction(){
            useTPCPions = (bool)ND::params().GetParameterI("psycheSelections.numuCCMultiPi.UseTPCPions");
            useME = (bool)ND::params().GetParameterI("psycheSelections.numuCCMultiPi.UseME");
            useFGDPions = (bool)ND::params().GetParameterI("psycheSelections.numuCCMultiPi.UseFGDPions");
            useOldSecondaryPID = (bool) ND::params().GetParameterI("psycheSelections.numuCCMultiPi.OldSecondaryPID");
            prod5Cut = (bool)ND::params().GetParameterI("psycheSelections.numuCCMultiPi.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.numuCCMultiPi.MostUpstreamLayerHitCut");
            prod5Cut = (bool)ND::params().GetParameterI("psycheSelections.numuCCMultiPi.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();}
};

#endif