EventSelectionManager.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef EVENTSELECTIONMANAGER_H_
#define EVENTSELECTIONMANAGER_H_
 
#include <vector>
#include <memory>
 
#include "TopConfiguration/SelectionConfigurationData.h"
#include "TopAnalysis/EventSelection.h"
 
class TFile;
 
namespace EL {
  class Worker;
}
 
namespace xAOD {
  class SystematicEvent;
}
 
namespace top {
  class Event;
  class TopConfig;
  class ParticleLevelEvent;
}
 
namespace top {
  class EventSelectionManager {
  public:
    explicit EventSelectionManager(const std::vector<SelectionConfigurationData>& selectionConfigData,
                                   TFile* outputFile, const std::string& toolLoaderNames,
                                   const std::shared_ptr<top::TopConfig>& config, EL::Worker* wk = nullptr);
 
    virtual ~EventSelectionManager() {}
 
    EventSelectionManager(EventSelectionManager&& other);
 
    EventSelectionManager() = delete;
    EventSelectionManager(const EventSelectionManager& rhs) = delete;
    EventSelectionManager& operator = (const EventSelectionManager& rhs) = delete;
 
    virtual void countInitial(const float mcEventWeight, const float pileupWeight);
    virtual void countGRL(const float mcEventWeight, const float pileupWeight);
    virtual void countGoodCalo(const float mcEventWeight, const float pileupWeight);
    virtual void countPrimaryVertex(const float mcEventWeight, const float pileupWeight);
 
    virtual bool apply(top::Event& event, const xAOD::SystematicEvent& currentSystematic);
 
    virtual bool applyParticleLevel(const top::ParticleLevelEvent& plEvent);
 
    virtual void finalise();
 
    virtual void addExtraBranches(std::vector<std::string>& extraBranchList);
 
    std::vector<std::string> GetFakesMMConfigs(const std::string& selection) const;
  private:
    std::vector<top::EventSelection> m_selections;
  };
}
 
#endif