FFTJetLookupTableESProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    JetMETCorrections/FFTJetModules
// Class:      FFTJetLookupTableESProducer
//
//
// Original Author:  Igor Volobouev
//         Created:  Thu Aug  2 22:34:02 CDT 2012
//
//

// system include files
#include <sstream>
#include <utility>

#include <memory>

#include "Alignment/Geners/interface/CompressedIO.hh"
#include "Alignment/Geners/interface/StringArchive.hh"
#include "Alignment/Geners/interface/Reference.hh"

// user include files
#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducer.h"
#include "FWCore/Framework/interface/ESProductHost.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/Utilities/interface/ReusableObjectHolder.h"

#include "CondFormats/JetMETObjects/interface/FFTJetCorrectorParameters.h"
#include "CondFormats/JetMETObjects/interface/FFTJetLUTTypes.h"

#include "JetMETCorrections/FFTJetObjects/interface/FFTJetLookupTableRcd.h"
#include "JetMETCorrections/FFTJetObjects/interface/FFTJetLookupTableSequence.h"

typedef std::shared_ptr<npstat::StorableMultivariateFunctor> StorableFunctorPtr;

static void insertLUTItem(FFTJetLookupTableSequence& seq,
                          StorableFunctorPtr fptr,
                          const std::string& name,
                          const std::string& category) {
  FFTJetLookupTableSequence::iterator it = seq.find(category);
  if (it == seq.end())
    it = seq.insert(std::make_pair(category, FFTJetDict<std::string, StorableFunctorPtr>())).first;
  it->second.insert(std::make_pair(name, fptr));
}

static void buildLookupTables(const FFTJetCorrectorParameters& tablePars,
                              const std::vector<edm::ParameterSet>& tableDefs,
                              const bool isArchiveCompressed,
                              const bool verbose,
                              FFTJetLookupTableSequence* ptr) {
  // Load the archive stored in the FFTJetCorrectorParameters object
  std::unique_ptr<gs::StringArchive> ar;
  {
    std::istringstream is(tablePars.str());
    if (isArchiveCompressed)
      ar = gs::read_compressed_item<gs::StringArchive>(is);
    else
      ar = gs::read_item<gs::StringArchive>(is);
  }

  ptr->clear();

  // Avoid loading the same item more than once
  std::set<unsigned long long> loadedSet;

  const unsigned nTables = tableDefs.size();
  for (unsigned itab = 0; itab < nTables; ++itab) {
    const edm::ParameterSet& ps(tableDefs[itab]);
    gs::SearchSpecifier nameSearch(ps.getParameter<std::string>("name"), ps.getParameter<bool>("nameIsRegex"));
    gs::SearchSpecifier categorySearch(ps.getParameter<std::string>("category"),
                                       ps.getParameter<bool>("categoryIsRegex"));
    gs::Reference<npstat::StorableMultivariateFunctor> ref(*ar, nameSearch, categorySearch);
    const unsigned long nItems = ref.size();
    for (unsigned long item = 0; item < nItems; ++item) {
      const unsigned long long id = ref.id(item);
      if (loadedSet.insert(id).second) {
        std::unique_ptr<npstat::StorableMultivariateFunctor> p(ref.get(item));
        StorableFunctorPtr fptr(p.release());
        std::shared_ptr<const gs::CatalogEntry> e = ar->catalogEntry(id);
        insertLUTItem(*ptr, fptr, e->name(), e->category());
        if (verbose)
          std::cout << "In buildLookupTables: loaded table with name \"" << e->name() << "\" and category \""
                    << e->category() << '"' << std::endl;
      }
    }
  }
}

//
// class declaration
//
template <typename CT>
class FFTJetLookupTableESProducer : public edm::ESProducer {
public:
  typedef std::shared_ptr<FFTJetLookupTableSequence> ReturnType;
  typedef FFTJetLookupTableRcd<CT> MyRecord;
  typedef FFTJetCorrectorParametersRcd<CT> ParentRecord;

  FFTJetLookupTableESProducer(const edm::ParameterSet&);
  ~FFTJetLookupTableESProducer() override {}

  ReturnType produce(const MyRecord&);

private:
  // Module parameters
  std::vector<edm::ParameterSet> tables;
  bool isArchiveCompressed;
  bool verbose;

  using HostType = edm::ESProductHost<FFTJetLookupTableSequence, ParentRecord>;
  edm::ReusableObjectHolder<HostType> holder_;

  edm::ESGetToken<FFTJetCorrectorParameters, ParentRecord> token_;
};

//
// constructors and destructor
//
template <typename CT>
FFTJetLookupTableESProducer<CT>::FFTJetLookupTableESProducer(const edm::ParameterSet& psIn)
    : tables(psIn.getParameter<std::vector<edm::ParameterSet> >("tables")),
      isArchiveCompressed(psIn.getParameter<bool>("isArchiveCompressed")),
      verbose(psIn.getUntrackedParameter<bool>("verbose")) {
  // The following line is needed to tell the framework what
  // data is being produced
  auto cc = setWhatProduced(this);
  token_ = cc.consumes();
}

// ------------ method called to produce the data  ------------
template <typename CT>
typename FFTJetLookupTableESProducer<CT>::ReturnType FFTJetLookupTableESProducer<CT>::produce(const MyRecord& iRecord) {
  auto host = holder_.makeOrGet([]() { return new HostType; });

  host->template ifRecordChanges<ParentRecord>(iRecord, [this, product = host.get()](auto const& rec) {
    edm::ESTransientHandle<FFTJetCorrectorParameters> parHandle = rec.getTransientHandle(token_);
    buildLookupTables(*parHandle, tables, isArchiveCompressed, verbose, product);
  });

  return host;
}

//
// define this as a plug-in
//
typedef FFTJetLookupTableESProducer<fftluttypes::EtaFlatteningFactors> FFTEtaFlatteningFactorsTableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::PileupRhoCalibration> FFTPileupRhoCalibrationTableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::PileupRhoEtaDependence> FFTPileupRhoEtaDependenceTableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT0> FFTLUT0TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT1> FFTLUT1TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT2> FFTLUT2TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT3> FFTLUT3TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT4> FFTLUT4TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT5> FFTLUT5TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT6> FFTLUT6TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT7> FFTLUT7TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT8> FFTLUT8TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT9> FFTLUT9TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT10> FFTLUT10TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT11> FFTLUT11TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT12> FFTLUT12TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT13> FFTLUT13TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT14> FFTLUT14TableESProducer;
typedef FFTJetLookupTableESProducer<fftluttypes::LUT15> FFTLUT15TableESProducer;

// =========================================================

DEFINE_FWK_EVENTSETUP_MODULE(FFTEtaFlatteningFactorsTableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTPileupRhoCalibrationTableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTPileupRhoEtaDependenceTableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT0TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT1TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT2TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT3TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT4TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT5TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT6TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT7TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT8TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT9TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT10TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT11TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT12TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT13TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT14TableESProducer);
DEFINE_FWK_EVENTSETUP_MODULE(FFTLUT15TableESProducer);