EventSelection.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef EVENTSELECTION_H_
#define EVENTSELECTION_H_
 
#include "TopEventSelectionTools/EventSelectorBase.h"
#include "TopEventSelectionTools/ToolLoaderBase.h"
#include "TopCorrections/ScaleFactorRetriever.h"
 
#include <string>
#include <vector>
#include <memory>
 
class TH1D;
class TFile;
 
namespace EL {
  class Worker;
}
 
namespace top {
  class Event;
  class TopConfig;
  class ParticleLevelEvent;
}
 
namespace top {
  class EventSelection final {
  public:
    EventSelection(const std::string& name, const std::vector<std::string>& cutNames, TFile* outputFile,
                   const std::vector<std::unique_ptr<top::ToolLoaderBase> >& toolLoaders,
                   const std::shared_ptr<top::TopConfig>& config, EL::Worker* wk);
 
    virtual ~EventSelection() {}
 
    EventSelection(EventSelection&& other);
 
    EventSelection() = delete;
 
    EventSelection(const EventSelection&) = delete;
 
    EventSelection& operator = (const EventSelection&) = delete;
 
    virtual void countInitial(const float mcEventWeight, const float pileupWeight) const;
    virtual void countGRL(const float mcEventWeight, const float pileupWeight) const;
    virtual void countGoodCalo(const float mcEventWeight, const float pileupWeight) const;
    virtual void countPrimaryVertex(const float mcEventWeight, const float pileupWeight) const;
 
    virtual bool apply(const top::Event& event) const;
 
    virtual bool applyParticleLevel(const top::ParticleLevelEvent& plEvent) const;
 
    virtual void finalise() const;
 
    const std::string name() const;
 
    std::vector<std::string> GetFakesMMConfigs() const;
 
    bool ToBeSaved() const {return m_toBeSaved;};
  private:
    virtual void printCuts();
 
    std::vector<std::unique_ptr<EventSelectorBase> > m_allCuts;
 
    mutable TH1D* m_cutflow;
    mutable TH1D* m_cutflow_Loose;
 
    mutable TH1D* m_cutflowMCWeights;
    mutable TH1D* m_cutflowMCWeights_Loose;
 
    mutable TH1D* m_cutflowPUWeights;
    mutable TH1D* m_cutflowPUWeights_Loose;
 
    mutable TH1D* m_cutflowMCPUWeights;
    mutable TH1D* m_cutflowMCPUWeights_Loose;
 
    mutable TH1D* m_cutflowScaleFactors;
    mutable TH1D* m_cutflowScaleFactors_Loose;
 
    mutable TH1D* m_cutflowParticleLevel;
    mutable TH1D* m_cutflowParticleLevelMCWeights;
 
    std::string m_name;
 
    mutable bool m_isMC;
 
    std::shared_ptr<TopConfig> m_config;
 
    ScaleFactorRetriever* m_sfRetriever;
    void initialiseTopScaleFactorRetriever(const std::shared_ptr<TopConfig>& config);
 
    // Nominal hash value
    std::size_t m_nominalHashValue;
 
    // Does this selection contain initial and GRL?
    // If so, what are their positions in the cut flow?
    bool m_containsInitial;
    bool m_containsGRL;
    bool m_containsGoodCalo;
    bool m_containsPrimaryVertex;
    unsigned int m_positionInitial;
    unsigned int m_positionGRL;
    unsigned int m_positionGoodCalo;
    unsigned int m_positionPrimaryVertex;
 
    // If true, event will be saved if it passes this selection
    // If false, event will only be saved if it passes another selection
    // If an event is saved, the flags of all selections (those passed and those not passed) are saved
    bool m_toBeSaved;
  };
}
 
#endif