IgProfModule.cc

Go to the documentation of this file.

#include "DataFormats/Provenance/interface/RunLumiEventNumber.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/LuminosityBlock.h"
#include "FWCore/Framework/interface/FileBlock.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Run.h"

#include "FWCore/ServiceRegistry/interface/Service.h"
#include <string>
#include <dlfcn.h>
#include <cstdio>
#include <cstring>

class IgProfModule : public edm::EDAnalyzer {
public:
  IgProfModule(const edm::ParameterSet &ps)
      : dump_(nullptr), prescale_(0), nrecord_(0), nevent_(0), nrun_(0), nlumi_(0), nfile_(0) {
    // Removing the __extension__ gives a warning which
    // is acknowledged as a language problem in the C++ Standard Core
    // Language Defect Report
    //
    // http://www.open-std.org/jtc1/sc22/wg21/docs/cwg_defects.html#195
    //
    // since the suggested decision seems to be that the syntax should
    // actually be "Conditionally-Supported Behavior" in some
    // future C++ standard I simply silence the warning.
    if (void *sym = dlsym(nullptr, "igprof_dump_now"))
      dump_ = __extension__(void (*)(const char *)) sym;
    else
      edm::LogWarning("IgProfModule") << "IgProfModule requested but application is not"
                                      << " currently being profiled with igprof\n";

    prescale_ = ps.getUntrackedParameter<int>("reportEventInterval", prescale_);
    atBeginJob_ = ps.getUntrackedParameter<std::string>("reportToFileAtBeginJob", atBeginJob_);
    atEndJob_ = ps.getUntrackedParameter<std::string>("reportToFileAtEndJob", atEndJob_);
    atBeginLumi_ = ps.getUntrackedParameter<std::string>("reportToFileAtBeginLumi", atBeginLumi_);
    atEndLumi_ = ps.getUntrackedParameter<std::string>("reportToFileAtEndLumi", atEndLumi_);
    atInputFile_ = ps.getUntrackedParameter<std::string>("reportToFileAtInputFile", atInputFile_);
    atEvent_ = ps.getUntrackedParameter<std::string>("reportToFileAtEvent", atEvent_);
  }

  void beginJob() override { makeDump(atBeginJob_); }

  void endJob(void) override { makeDump(atEndJob_); }

  void analyze(const edm::Event &e, const edm::EventSetup &) override {
    nevent_ = e.id().event();
    if (prescale_ > 0 && (++nrecord_ % prescale_) == 1)
      makeDump(atEvent_);
  }

  void beginRun(const edm::Run &r, const edm::EventSetup &) override {
    nrun_ = r.run();
    makeDump(atBeginRun_);
  }

  void endRun(const edm::Run &, const edm::EventSetup &) override { makeDump(atEndRun_); }

  void beginLuminosityBlock(const edm::LuminosityBlock &l, const edm::EventSetup &) override {
    nlumi_ = l.luminosityBlock();
    makeDump(atBeginLumi_);
  }

  void endLuminosityBlock(const edm::LuminosityBlock &l, const edm::EventSetup &) override { makeDump(atEndLumi_); }

  void respondToOpenInputFile(const edm::FileBlock &) override {
    ++nfile_;
    makeDump(atInputFile_);
  }

private:
  void makeDump(const std::string &format) {
    if (!dump_ || format.empty())
      return;

    std::string final(format);
    final = replace(final, "%I", nrecord_);
    final = replaceU64(final, "%E", nevent_);
    final = replaceU64(final, "%R", nrun_);
    final = replaceU64(final, "%L", nlumi_);
    final = replace(final, "%F", nfile_);
    dump_(final.c_str());
  }

  static std::string replace(const std::string &s, const char *pat, int val) {
    size_t pos = 0;
    size_t patlen = strlen(pat);
    std::string result = s;
    while ((pos = result.find(pat, pos)) != std::string::npos) {
      char buf[64];
      int n = sprintf(buf, "%d", val);
      result.replace(pos, patlen, buf);
      pos = pos - patlen + n;
    }

    return result;
  }

  static std::string replaceU64(const std::string &s, const char *pat, unsigned long long val) {
    size_t pos = 0;
    size_t patlen = strlen(pat);
    std::string result = s;
    while ((pos = result.find(pat, pos)) != std::string::npos) {
      char buf[64];
      int n = sprintf(buf, "%llu", val);
      result.replace(pos, patlen, buf);
      pos = pos - patlen + n;
    }

    return result;
  }

  void (*dump_)(const char *);
  std::string atBeginJob_;
  std::string atEndJob_;
  std::string atBeginRun_;
  std::string atEndRun_;
  std::string atBeginLumi_;
  std::string atEndLumi_;
  std::string atInputFile_;
  std::string atEvent_;
  int prescale_;
  int nrecord_;
  edm::EventNumber_t nevent_;
  edm::RunNumber_t nrun_;
  edm::LuminosityBlockNumber_t nlumi_;
  int nfile_;
};

DEFINE_FWK_MODULE(IgProfModule);