HGCalTriggerNtupleManager.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "L1Trigger/L1THGCalUtilities/interface/HGCalTriggerNtupleBase.h"

class HGCalTriggerNtupleManager : public edm::EDAnalyzer {
public:
  typedef std::unique_ptr<HGCalTriggerNtupleBase> ntuple_ptr;

public:
  explicit HGCalTriggerNtupleManager(const edm::ParameterSet& conf);
  ~HGCalTriggerNtupleManager() override{};
  void beginRun(const edm::Run&, const edm::EventSetup&) override{};
  void analyze(const edm::Event&, const edm::EventSetup&) override;

private:
  edm::Service<TFileService> file_service_;
  std::vector<ntuple_ptr> hgc_ntuples_;
  TTree* tree_;
};

DEFINE_FWK_MODULE(HGCalTriggerNtupleManager);

HGCalTriggerNtupleManager::HGCalTriggerNtupleManager(const edm::ParameterSet& conf) {
  tree_ = file_service_->make<TTree>("HGCalTriggerNtuple", "HGCalTriggerNtuple");
  const std::vector<edm::ParameterSet>& ntuple_cfgs = conf.getParameterSetVector("Ntuples");
  for (const auto& ntuple_cfg : ntuple_cfgs) {
    const std::string& ntuple_name = ntuple_cfg.getParameter<std::string>("NtupleName");
    hgc_ntuples_.emplace_back(HGCalTriggerNtupleFactory::get()->create(ntuple_name, ntuple_cfg));
    hgc_ntuples_.back()->initialize(*tree_, ntuple_cfg, consumesCollector());
  }
}

void HGCalTriggerNtupleManager::analyze(const edm::Event& e, const edm::EventSetup& es) {
  for (auto& hgc_ntuple : hgc_ntuples_) {
    hgc_ntuple->fill(e, es);
  }
  tree_->Fill();
}