L1TriggerScouting/Utilities/plugins/DumpScObjects.cc - master - githubmirror/cmssw

L1TriggerScouting/Utilities/plugins/DumpScObjects.cc
235 строк8 KB
Shahzad Malik Muzaffar
[DAQ] Changes suggested by new llvm18 clang-format
05 сен 2024, 17:13
05 сен 2024, 17:13eeb9ce1
#include "FWCore/Framework/interface/MakerMacros.h"

#include <fstream>
#include <iomanip>
#include <memory>
#include <string>
#include <cmath>

#include "FWCore/Framework/interface/stream/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/MessageLogger/interface/MessageDrop.h"

#include "DataFormats/L1Scouting/interface/OrbitCollection.h"
#include "DataFormats/L1Scouting/interface/L1ScoutingMuon.h"
#include "DataFormats/L1Scouting/interface/L1ScoutingCalo.h"
#include "L1TriggerScouting/Utilities/interface/printScObjects.h"
#include "L1TriggerScouting/Utilities/interface/convertToL1TFormat.h"

using namespace l1ScoutingRun3;

// ----------------------------- CLASS DECLARATION  ----------------------------
class DumpScObjects : public edm::stream::EDAnalyzer<> {
public:
  // constructor and destructor
  explicit DumpScObjects(const edm::ParameterSet&);
  ~DumpScObjects() override {}

  // method for analyzing the events
  void analyze(const edm::Event&, const edm::EventSetup&) override;

private:
  // dump contenct of BX
  void printBx(unsigned bx);

  // the tokens to access the data
  edm::EDGetTokenT<MuonOrbitCollection> gmtMuonsToken_;
  edm::EDGetTokenT<JetOrbitCollection> caloJetsToken_;
  edm::EDGetTokenT<EGammaOrbitCollection> caloEGammasToken_;
  edm::EDGetTokenT<TauOrbitCollection> caloTausToken_;
  edm::EDGetTokenT<BxSumsOrbitCollection> caloEtSumsToken_;

  edm::Handle<MuonOrbitCollection> muonHandle_;
  edm::Handle<JetOrbitCollection> jetHandle_;
  edm::Handle<EGammaOrbitCollection> eGammaHandle_;
  edm::Handle<TauOrbitCollection> tauHandle_;
  edm::Handle<BxSumsOrbitCollection> etSumHandle_;

  // the min and max BX to be analyzed
  unsigned minBx_;
  unsigned maxBx_;

  // select collection to be printed
  bool checkMuons_;
  bool checkJets_;
  bool checkEGammas_;
  bool checkTaus_;
  bool checkEtSums_;

  // dump a specific (ORBIT, BX RANGE)
  bool searchEvent_;
  unsigned orbitNum_;
  unsigned searchStartBx_;
  unsigned searchStopBx_;

  // utils
  bool skipEmptyBx_;
};
// -----------------------------------------------------------------------------

// -------------------------------- constructor  -------------------------------

DumpScObjects::DumpScObjects(const edm::ParameterSet& iConfig)
    : minBx_(iConfig.getUntrackedParameter<unsigned>("minBx", 0)),
      maxBx_(iConfig.getUntrackedParameter<unsigned>("maxBx", 3564)),

      checkMuons_(iConfig.getUntrackedParameter<bool>("checkMuons", true)),
      checkJets_(iConfig.getUntrackedParameter<bool>("checkJets", true)),
      checkEGammas_(iConfig.getUntrackedParameter<bool>("checkEGammas", true)),
      checkTaus_(iConfig.getUntrackedParameter<bool>("checkTaus", true)),
      checkEtSums_(iConfig.getUntrackedParameter<bool>("checkEtSums", true)),

      searchEvent_(iConfig.getUntrackedParameter<bool>("searchEvent", false)),
      orbitNum_(iConfig.getUntrackedParameter<unsigned>("orbitNumber", 0)),
      searchStartBx_(iConfig.getUntrackedParameter<unsigned>("searchStartBx", 0)),
      searchStopBx_(iConfig.getUntrackedParameter<unsigned>("searchStopBx", 0)),

      skipEmptyBx_(iConfig.getUntrackedParameter<bool>("skipEmptyBx", true)) {
  if (checkMuons_)
    gmtMuonsToken_ = consumes<MuonOrbitCollection>(iConfig.getParameter<edm::InputTag>("gmtMuonsTag"));
  if (checkJets_)
    caloJetsToken_ = consumes<JetOrbitCollection>(iConfig.getParameter<edm::InputTag>("caloJetsTag"));
  if (checkEGammas_)
    caloEGammasToken_ = consumes<EGammaOrbitCollection>(iConfig.getParameter<edm::InputTag>("caloEGammasTag"));
  if (checkTaus_)
    caloTausToken_ = consumes<TauOrbitCollection>(iConfig.getParameter<edm::InputTag>("caloTausTag"));
  if (checkEtSums_)
    caloEtSumsToken_ = consumes<BxSumsOrbitCollection>(iConfig.getParameter<edm::InputTag>("caloEtSumsTag"));
}
// -----------------------------------------------------------------------------

// ----------------------- method called for each orbit  -----------------------
void DumpScObjects::analyze(const edm::Event& iEvent, const edm::EventSetup& evSetup) {
  if (checkMuons_)
    iEvent.getByToken(gmtMuonsToken_, muonHandle_);
  if (checkJets_)
    iEvent.getByToken(caloJetsToken_, jetHandle_);
  if (checkEGammas_)
    iEvent.getByToken(caloEGammasToken_, eGammaHandle_);
  if (checkTaus_)
    iEvent.getByToken(caloTausToken_, tauHandle_);
  if (checkEtSums_)
    iEvent.getByToken(caloEtSumsToken_, etSumHandle_);

  // get the orbit number
  unsigned currOrbit = iEvent.id().event();

  // if we are looking for a specific orbit
  if (searchEvent_) {
    if (currOrbit != orbitNum_)
      return;

    // found the orbit
    for (unsigned bx = searchStartBx_; bx <= searchStopBx_; bx++) {
      printBx(bx);
    }
  } else {
    if (skipEmptyBx_) {
      // create a set of non empty BXs
      std::set<unsigned> uniqueBx;

      if (checkMuons_) {
        for (const unsigned& bx : muonHandle_->getFilledBxs()) {
          if ((bx >= minBx_) || (bx <= maxBx_))
            uniqueBx.insert(bx);
        }
      }
      if (checkJets_) {
        for (const unsigned& bx : jetHandle_->getFilledBxs()) {
          if ((bx >= minBx_) || (bx <= maxBx_))
            uniqueBx.insert(bx);
        }
      }
      if (checkEGammas_) {
        for (const unsigned& bx : eGammaHandle_->getFilledBxs()) {
          if ((bx >= minBx_) || (bx <= maxBx_))
            uniqueBx.insert(bx);
        }
      }
      if (checkTaus_) {
        for (const unsigned& bx : tauHandle_->getFilledBxs()) {
          if ((bx >= minBx_) || (bx <= maxBx_))
            uniqueBx.insert(bx);
        }
      }
      if (checkEtSums_) {
        for (const unsigned& bx : etSumHandle_->getFilledBxs()) {
          if ((bx >= minBx_) || (bx <= maxBx_))
            uniqueBx.insert(bx);
        }
      }

      // process bx
      for (const unsigned& bx : uniqueBx) {
        printBx(bx);
      }

    } else {
      // dump all objects
      for (unsigned bx = minBx_; bx <= maxBx_; bx++) {
        printBx(bx);
      }
    }
  }
}
// -----------------------------------------------------------------------------

void DumpScObjects::printBx(unsigned bx) {
  std::cout << "BX = " << bx << " ****" << std::endl;

  if (checkMuons_ && muonHandle_.isValid()) {
    int i = 0;
    const auto& muons = muonHandle_->bxIterator(bx);
    for (const auto& muon : muons) {
      std::cout << "--- Muon " << i << " ---\n";
      printMuon(muon);
      i++;
    }
  }

  if (checkJets_ && jetHandle_.isValid()) {
    int i = 0;
    const auto& jets = jetHandle_->bxIterator(bx);
    for (const auto& jet : jets) {
      std::cout << "--- Jet " << i << " ---\n";
      printJet(jet);
      i++;
    }
  }

  if (checkEGammas_ && jetHandle_.isValid()) {
    int i = 0;
    const auto& eGammas = eGammaHandle_->bxIterator(bx);
    for (const auto& egamma : eGammas) {
      std::cout << "--- E/Gamma " << i << " ---\n";
      printEGamma(egamma);
      i++;
    }
  }

  if (checkTaus_ && tauHandle_.isValid()) {
    int i = 0;
    const auto& taus = tauHandle_->bxIterator(bx);
    for (const auto& tau : taus) {
      std::cout << "--- Tau " << i << " ---\n";
      printTau(tau);
      i++;
    }
  }

  if (checkEtSums_ && etSumHandle_.isValid()) {
    const auto& sums = etSumHandle_->bxIterator(bx);
    for (const auto& sum : sums) {
      std::cout << "--- Calo Sums ---\n";
      printBxSums(sum);
    }
  }
}

DEFINE_FWK_MODULE(DumpScObjects);