p0dNumuCCSelection.cxx

#include "baseSelection.hxx"
#include "p0dNumuCCSelection.hxx"
#include "CutUtils.hxx"
#include "trackerSelectionUtils.hxx"
#include "EventBoxUtils.hxx"


//********************************************************************
p0dNumuCCSelection::p0dNumuCCSelection(bool forceBreak): SelectionBase(forceBreak, EventBoxId::kEventBoxTracker) {
//********************************************************************

}

//********************************************************************
void p0dNumuCCSelection::DefineSteps(){
//********************************************************************

    // Cuts must be added in the right order
    // last "true" means the step sequence is broken if cut is not passed (default is "false")
    AddStep(StepBase::kCut,    "event quality",      new EventQualityCut(),           true);
    AddStep(StepBase::kCut,    "> 0 tracks ",        new TotalMultiplicityCut(),      true);  
    AddStep(StepBase::kAction, "find leading tracks",new FindP0DLeadingTracksAction());  
    AddStep(StepBase::kAction, "find vertex",        new FindVertexAction());  
    AddStep(StepBase::kCut,    "quality+fiducial",   new TrackQualityFiducialCut(),   true);  
    AddStep(StepBase::kAction, "find veto track", new FindP0DVetoAction());
    AddStep(StepBase::kCut,    "veto", new P0DSelectionVetoCut(),true);
    SetBranchAlias(0,"trunk");
}

//**************************************************
void p0dNumuCCSelection::InitializeEvent(AnaEventC& eventC){
//**************************************************

  AnaEventB& event = *static_cast<AnaEventB*>(&eventC); 

  // Create the appropriate EventBox if it does not exist yet
  if (!event.EventBoxes[EventBoxId::kEventBoxTracker])
    event.EventBoxes[EventBoxId::kEventBoxTracker] = new EventBoxTracker();

  boxUtils::FillTracksWithTPC(event,             static_cast<SubDetId::SubDetEnum>(GetDetectorFV()));
  boxUtils::FillTracksWithP0D(event);
  boxUtils::FillTrajsChargedInTPC(event);
  boxUtils::FillTrajsChargedInP0D(event);
  
}

//********************************************************************
void p0dNumuCCSelection::DefineDetectorFV(){
//********************************************************************

  SetDetectorFV(SubDetId::kP0D);
}


//**************************************************
bool FindP0DLeadingTracksAction::Apply(AnaEventC& event, ToyBoxB& boxB) const{
//**************************************************

  // Cast the ToyBox to the appropriate type
  ToyBoxTracker& box = *static_cast<ToyBoxTracker*>(&boxB); 

  trackerSelUtils::FindLeadingTracks(event, box);

  // For this selection the main track is the HMN track
  box.MainTrack = box.HMNtrack;
  return true;
}

//************************************************** 
bool FindP0DVetoAction::Apply(AnaEventC& event, ToyBoxB& boxB) const{ 
//************************************************** 
  ToyBoxTracker* box = static_cast<ToyBoxTracker*> (&boxB); 
  box->VetoTrack = cutUtils::FindP0DVetoTrack(event); 

  return true; 
} 

//************************************************** 
bool P0DSelectionVetoCut::Apply(AnaEventC& event, ToyBoxB& boxB) const{ 
//************************************************** 

  (void)event;
  
  ToyBoxTracker* box = static_cast<ToyBoxTracker*> (&boxB); 
  if (box->VetoTrack)
    return false; 

  return true;

}