antiNumuCCMultiPiSelection.hxx

#ifndef antiNumuCCMultiPiSelection_h
#define antiNumuCCMultiPiSelection_h

#include "SelectionBase.hxx"
#include "antiNumuCCSelection.hxx"

class antiNumuCCMultiPiSelection: public SelectionBase{
public:
  antiNumuCCMultiPiSelection(bool forceBreak=true);
  virtual ~antiNumuCCMultiPiSelection(){}

  //---- These are mandatory functions
  void DefineSteps();
  void DefineDetectorFV();
  ToyBoxB* MakeToyBox();
  bool FillEventSummary(AnaEventC& event, Int_t allCutsPassed[]);
  SampleId::SampleEnum GetSampleEnum(){return SampleId::kFGD1AntiNuMuCC1Track;}
  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 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:
  antiNumuCCSelection _antiNumuCCSelection;

  Int_t _MuonPIDCutIndex;
  Int_t _FindPionsStepIndex; 

};


class ToyBoxAntiCCMultiPi: public ToyBoxTracker{

public:
  ToyBoxAntiCCMultiPi(){
    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 ~ToyBoxAntiCCMultiPi(){}

  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;
    nNegPions = 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 nNegPions;
  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* antiNumuCCMultiPiSelection::MakeToyBox(){return new ToyBoxAntiCCMultiPi();}

namespace antinumuCCMultiPiUtils{
  //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 FillSummaryAction_antinuCCMultiPi: public StepBase{
public:
  using StepBase::Apply;
  bool Apply(AnaEventC& event, ToyBoxB& box) const;  
  StepBase* MakeClone(){return new FillSummaryAction_antinuCCMultiPi();}
};

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

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

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

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

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

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

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

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



#endif