L1uGTTreeProducer.cc

Go to the documentation of this file.

// system include files
#include <memory>
#include <boost/format.hpp>

// framework
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

// data formats
#include "DataFormats/L1TGlobal/interface/GlobalAlgBlk.h"
#include "CondFormats/DataRecord/interface/L1TUtmTriggerMenuRcd.h"
#include "CondFormats/L1TObjects/interface/L1TUtmTriggerMenu.h"

// ROOT output stuff
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "TTree.h"

//
// class declaration
//

class L1uGTTreeProducer : public edm::EDAnalyzer {
public:
  explicit L1uGTTreeProducer(edm::ParameterSet const &);
  ~L1uGTTreeProducer() override;

private:
  void beginJob() override;
  void analyze(edm::Event const &, edm::EventSetup const &) override;
  void endJob() override;

private:
  // output file
  edm::Service<TFileService> fs_;

  // pointers to the objects that will be stored as branches within the tree
  GlobalAlgBlk const *results_;

  // tree
  TTree *tree_;

  // EDM input tokens
  const edm::EDGetTokenT<GlobalAlgBlkBxCollection> ugt_token_;

  // L1 uGT menu
  unsigned long long cache_id_;
};

L1uGTTreeProducer::L1uGTTreeProducer(edm::ParameterSet const &config)
    : results_(nullptr),
      tree_(nullptr),
      ugt_token_(consumes<GlobalAlgBlkBxCollection>(config.getParameter<edm::InputTag>("ugtToken"))),
      cache_id_(0) {
  // set up the TTree and its branches
  tree_ = fs_->make<TTree>("L1uGTTree", "L1uGTTree");
  tree_->Branch("L1uGT", "GlobalAlgBlk", &results_, 32000, 3);
}

L1uGTTreeProducer::~L1uGTTreeProducer() {
  //if (tree_) { delete tree_; tree_ = NULL; }
  //if (results_) { delete results_; results_ = NULL; }  // It seems TTree owns this pointer...
}

//
// member functions
//

// ------------ method called to for each event  ------------
void L1uGTTreeProducer::analyze(edm::Event const &event, edm::EventSetup const &setup) {
  unsigned long long id = setup.get<L1TUtmTriggerMenuRcd>().cacheIdentifier();
  if (id != cache_id_) {
    cache_id_ = id;
    edm::ESHandle<L1TUtmTriggerMenu> menu;
    setup.get<L1TUtmTriggerMenuRcd>().get(menu);

    for (auto const &keyval : menu->getAlgorithmMap()) {
      std::string const &name = keyval.second.getName();
      unsigned int index = keyval.second.getIndex();
      //std::cerr << (boost::format("bit %4d: %s") % index % name).str() << std::endl;
      tree_->SetAlias(name.c_str(), (boost::format("L1uGT.m_algoDecisionInitial[%d]") % index).str().c_str());
    }
  }

  edm::Handle<GlobalAlgBlkBxCollection> ugt;

  event.getByToken(ugt_token_, ugt);

  if (ugt.isValid()) {
    results_ = &ugt->at(0, 0);
  }

  tree_->Fill();
}

// ------------ method called once each job just before starting event loop  ------------
void L1uGTTreeProducer::beginJob(void) {}

// ------------ method called once each job just after ending the event loop  ------------
void L1uGTTreeProducer::endJob() {}

//define this as a plug-in
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(L1uGTTreeProducer);