L1TZDCAnalyzer.cc

Go to the documentation of this file.

//Copied from L1TStage2CaloAnalyzer on 2023.08.04 as it exists in CMSSW_13_1_0_pre4
//Modified by Chris McGinn to instead work for just ZDC etSums
//Contact at christopher.mc.ginn@cern.ch or cfmcginn @ github for bugs

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

#include "CondFormats/L1TObjects/interface/CaloParams.h"
#include "CondFormats/DataRecord/interface/L1TCaloParamsRcd.h"

#include "DataFormats/L1TCalorimeter/interface/CaloTower.h"
#include "DataFormats/L1TCalorimeter/interface/CaloCluster.h"
#include "DataFormats/L1Trigger/interface/EGamma.h"
#include "DataFormats/L1Trigger/interface/Tau.h"
#include "DataFormats/L1Trigger/interface/Jet.h"
#include "DataFormats/L1Trigger/interface/EtSum.h"

//For the output
#include "TTree.h"
//string for some branch handling
#include <string>

//
// class declaration
//

namespace l1t {

  class L1TZDCAnalyzer : public edm::one::EDAnalyzer<edm::one::SharedResources> {
  public:
    explicit L1TZDCAnalyzer(const edm::ParameterSet&);
    ~L1TZDCAnalyzer() override = default;

    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

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

    //fileservice
    edm::Service<TFileService> fs_;
    //Declare a tree, member and pointer
    TTree* etSumZdcTree_;
    //Declare the etSum max and bpx max
    //MAX Here is effectively a hardcoded number - you just need to know it from the producer - realistically this is best handled by something passed but for now (2023.08.17) we hardcode
    static const int maxBPX_ = 5;
    float etSumZdcP_[maxBPX_];
    float etSumZdcM_[maxBPX_];

    // ----------member data ---------------------------
    edm::EDGetToken sumToken_;

    bool doHistos_;

    TFileDirectory evtDispDir_;
  };

  //
  // constants, enums and typedefs
  //

  //
  // static data member definitions
  //

  //
  // constructors and destructor
  //
  L1TZDCAnalyzer::L1TZDCAnalyzer(const edm::ParameterSet& iConfig)
      : doHistos_(iConfig.getUntrackedParameter<bool>("doHistos", true)) {
    usesResource(TFileService::kSharedResource);
    //now do what ever initialization is needed

    // register what you consume and keep token for later access:
    edm::InputTag nullTag("None");

    edm::InputTag sumTag = iConfig.getParameter<edm::InputTag>("etSumTag");
    sumToken_ = consumes<l1t::EtSumBxCollection>(sumTag);

    edm::LogInfo("L1TZDCAnalyzer") << "Processing " << sumTag.label() << std::endl;
  }

  //
  // member functions
  //

  // ------------ method called for each event  ------------
  void L1TZDCAnalyzer::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
    using namespace edm;

    //    Handle<EtSumBxCollection> sums;
    Handle<BXVector<l1t::EtSum> > sums;
    iEvent.getByToken(sumToken_, sums);

    int startBX = sums->getFirstBX();

    for (int ibx = startBX; ibx <= sums->getLastBX(); ++ibx) {
      for (auto itr = sums->begin(ibx); itr != sums->end(ibx); ++itr) {
        if (itr->getType() == l1t::EtSum::EtSumType::kZDCP)
          etSumZdcP_[ibx - startBX] = itr->hwPt();
        if (itr->getType() == l1t::EtSum::EtSumType::kZDCM)
          etSumZdcM_[ibx - startBX] = itr->hwPt();
      }
    }

    etSumZdcTree_->Fill();
  }

  // ------------ method called once each job just before starting event loop  ------------
  void L1TZDCAnalyzer::beginJob() {
    etSumZdcTree_ = fs_->make<TTree>("etSumZdcTree", "");
    etSumZdcTree_->Branch("etSumZdcP", etSumZdcP_, ("etSumZdcP[" + std::to_string(maxBPX_) + "]/F").c_str());
    etSumZdcTree_->Branch("etSumZdcM", etSumZdcM_, ("etSumZdcM[" + std::to_string(maxBPX_) + "]/F").c_str());
  }

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

  // ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
  void L1TZDCAnalyzer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;

    desc.add<edm::InputTag>("etSumTag", edm::InputTag("etSumZdcProducer", ""));
    descriptions.add("l1tZDCAnalyzer", desc);
  }

}  // namespace l1t

using namespace l1t;

//define this as a plug-in
DEFINE_FWK_MODULE(L1TZDCAnalyzer);