JetMETCorrections/FFTJetModules/plugins/FFTJetCorrectorDBReader.cc - master - githubmirror/cmssw

JetMETCorrections/FFTJetModules/plugins/FFTJetCorrectorDBReader.cc
124 строки4 KB
Ivan Razumov
Replace CPP11_auto_ptr with std::unique_ptr (CPP11_auto_ptr was #defined as std::unique_ptr)
30 ноя 2023, 13:37
30 ноя 2023, 13:3706ade60
// -*- C++ -*-
//
// Package:    JetMETCorrections/FFTJetModules
// Class:      FFTJetCorrectorDBReader
//
/**\class FFTJetCorrectorDBReader FFTJetCorrectorDBReader.cc JetMETCorrections/FFTJetModules/plugins/FFTJetCorrectorDBReader.cc

 Description: writes a blob from a file into a database

 Implementation:
     [Notes on implementation]
*/
//
// Original Author:  Igor Volobouev
//         Created:  Wed Aug  1 20:59:12 CDT 2012
//
//

#include <iostream>
#include <sstream>
#include <fstream>

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

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"

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

#include "CondFormats/JetMETObjects/interface/FFTJetCorrectorParameters.h"
#include "CondFormats/DataRecord/interface/FFTJetCorrectorParametersRcdTypes.h"

#include "JetMETCorrections/FFTJetObjects/interface/FFTJetCorrectorParametersLoader.h"

#define init_param(type, varname) varname(ps.getParameter<type>(#varname))

//
// class declaration
//
class FFTJetCorrectorDBReader : public edm::stream::EDAnalyzer<> {
public:
  explicit FFTJetCorrectorDBReader(const edm::ParameterSet&);
  FFTJetCorrectorDBReader() = delete;
  FFTJetCorrectorDBReader(const FFTJetCorrectorDBReader&) = delete;
  FFTJetCorrectorDBReader& operator=(const FFTJetCorrectorDBReader&) = delete;
  ~FFTJetCorrectorDBReader() override {}

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

  std::string record;
  std::string outputFile;
  bool printAsString;
  bool readArchive;
  bool isArchiveCompressed;

  FFTJetCorrectorParametersLoader esLoader;
};

FFTJetCorrectorDBReader::FFTJetCorrectorDBReader(const edm::ParameterSet& ps)
    : init_param(std::string, record),
      init_param(std::string, outputFile),
      init_param(bool, printAsString),
      init_param(bool, readArchive),
      init_param(bool, isArchiveCompressed) {
  esLoader.acquireToken(record, consumesCollector());
}

void FFTJetCorrectorDBReader::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
  edm::ESHandle<FFTJetCorrectorParameters> JetCorParams = esLoader.load(record, iSetup);

  if (printAsString || readArchive)
    std::cout << "++++ FFTJetCorrectorDBReader: info for record \"" << record << '"' << std::endl;

  if (printAsString)
    std::cout << "++++ String rep: \"" << JetCorParams->str() << '"' << std::endl;
  else if (readArchive) {
    std::unique_ptr<gs::StringArchive> par;

    {
      std::istringstream is(JetCorParams->str());
      if (isArchiveCompressed)
        par = gs::read_compressed_item<gs::StringArchive>(is);
      else
        par = gs::read_item<gs::StringArchive>(is);
    }

    const unsigned long long idSmall = par->smallestId();
    if (!idSmall)
      std::cout << "++++ No valid records in the archive" << std::endl;
    else {
      std::cout << "++++ Archive metadata begins" << std::endl;
      const unsigned long long idLarge = par->largestId();
      unsigned long long count = 0;
      for (unsigned long long id = idSmall; id <= idLarge; ++id)
        if (par->itemExists(id)) {
          std::shared_ptr<const gs::CatalogEntry> e = par->catalogEntry(id);
          std::cout << '\n';
          e->humanReadable(std::cout);
          ++count;
        }
      std::cout << "\n++++ Archive metadata ends, " << count << " items total" << std::endl;
    }
  }

  if (!outputFile.empty()) {
    std::ofstream of(outputFile.c_str(), std::ios_base::binary);
    if (!of.is_open())
      throw cms::Exception("InvalidArgument") << "Failed to open file \"" << outputFile << '"' << std::endl;
    if (!JetCorParams->empty()) {
      of.write(JetCorParams->getBuffer(), JetCorParams->length());
      if (of.fail())
        throw cms::Exception("SystemError")
            << "Output stream failure while writing file \"" << outputFile << '"' << std::endl;
    }
  }
}

DEFINE_FWK_MODULE(FFTJetCorrectorDBReader);