EventCleaningSelection.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef ANALYSISTOP_TOPOBJECTSELECTIONTOOLS_EVENTCLEANINGSELECTION_H
#define ANALYSISTOP_TOPOBJECTSELECTIONTOOLS_EVENTCLEANINGSELECTION_H
 
#include <string>
#include <unordered_set>
#include <vector>
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
 
// GRL include(s):
#include "AsgAnalysisInterfaces/IGoodRunsListSelectionTool.h"
 
// Trigger include(s):
#include "TrigConfInterfaces/ITrigConfigTool.h"
#include "TrigDecisionTool/TrigDecisionTool.h"
#include "TriggerMatchingTool/IMatchingTool.h"
#include "TrigTauMatching/ITrigTauMatching.h"
#include "TriggerAnalysisInterfaces/ITrigGlobalEfficiencyCorrectionTool.h"
 
// Top include(s):
#include "TopConfiguration/SelectionConfigurationData.h"
 
namespace top {
  class TopConfig;
 
  class EventCleaningSelection final: public asg::AsgTool {
  public:
    explicit EventCleaningSelection(const std::string& name);
    virtual ~EventCleaningSelection() {}
 
    StatusCode initialize();
 
    void setEventSelections(const std::vector<top::SelectionConfigurationData>& selections);
    bool applyGRL() const;
    bool applyTrigger();
    bool applyGoodCalo() const;
    bool applyPrimaryVertex() const;
 
    // Print trigger configuration - very verbose, but tells you which triggers are included for an event
    void printTriggers() const;
 
    // helper to allow us to added trigger decisions to our output files
    void addExtraBranches(std::vector<std::string>& extraBranchList);
 
    // Delete Standard constructors
    EventCleaningSelection(const EventCleaningSelection& rhs) = delete;
    EventCleaningSelection(EventCleaningSelection&& rhs) = delete;
    EventCleaningSelection& operator = (const EventCleaningSelection& rhs) = delete;
 
    bool isElectronTrigger(std::string const& trigger) const;
    bool isMuonTrigger(std::string const& trigger) const;
    bool isPhotonTrigger(std::string const& trigger) const;
    static std::vector<std::pair<std::string, int> > getIndividualFromGlobalTriggers(std::vector<std::pair<std::string, int> > const& triggers);
  private:
    void matchElectrons();
 
    void matchMuons();
 
    void matchTaus();
 
    void matchPhotons();
 
    std::shared_ptr<top::TopConfig> m_config;
 
    ToolHandle<IGoodRunsListSelectionTool> m_grlTool;
 
    ToolHandle<TrigConf::ITrigConfigTool> m_trigConfTool;
 
    ToolHandle<Trig::TrigDecisionTool> m_trigDecisionTool;
 
    // Trigger matching tool
    ToolHandle<Trig::IMatchingTool> m_trigMatchTool;
 
    //Tau trigger matching
    ToolHandle<Trig::ITrigTauMatchingTool> m_trigMatchTauTool;
 
    //Global trigger efficiency tool
    ToolHandle<ITrigGlobalEfficiencyCorrectionTool> m_globalTriggerSF;
    ToolHandle<ITrigGlobalEfficiencyCorrectionTool> m_globalTriggerSFLoose;
 
    std::vector<std::pair<std::string, int> > m_allTriggers_Tight;
    std::vector<std::pair<std::string, int> > m_electronTriggers_Tight;
    std::vector<std::pair<std::string, int> > m_muonTriggers_Tight;
    std::vector<std::pair<std::string, int> > m_tauTriggers_Tight;
    std::vector<std::pair<std::string, int> > m_photonTriggers_Tight;
    std::vector<std::pair<std::string, int> > m_allTriggers_Loose;
    std::vector<std::pair<std::string, int> > m_electronTriggers_Loose;
    std::vector<std::pair<std::string, int> > m_muonTriggers_Loose;
    std::vector<std::pair<std::string, int> > m_tauTriggers_Loose;
    std::vector<std::pair<std::string, int> > m_photonTriggers_Loose;
 
    bool m_vetoEventsTrigger;
 
    bool m_vetoEventsGRL;
 
    bool m_vetoEventsGoodCalo;
 
    bool m_vetoEventsPriVtx;
  };
}
#endif