PCLMetadataWriter.cc

Go to the documentation of this file.

/*
 *  See header file for a description of this class.
 *
 *  \author G. Cerminara - CERN
 */

#include "Calibration/TkAlCaRecoProducers/plugins/PCLMetadataWriter.h"

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

#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "CondFormats/Common/interface/DropBoxMetadata.h"
#include "CondFormats/DataRecord/interface/DropBoxMetadataRcd.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/JobReport.h"
#include "FWCore/ServiceRegistry/interface/Service.h"

#include <iostream>

using namespace std;
using namespace edm;

PCLMetadataWriter::PCLMetadataWriter(const edm::ParameterSet &pSet) {
  readFromDB = pSet.getParameter<bool>("readFromDB");

  vector<ParameterSet> recordsToMap = pSet.getParameter<vector<ParameterSet>>("recordsToMap");
  for (vector<ParameterSet>::const_iterator recordPset = recordsToMap.begin(); recordPset != recordsToMap.end();
       ++recordPset) {
    // record is the key which identifies one set of metadata in
    // DropBoxMetadataRcd (not necessarily a record in the strict framework
    // sense)
    string record = (*recordPset).getUntrackedParameter<string>("record");

    map<string, string> jrInfo;
    if (!readFromDB) {
      vector<string> paramKeys = (*recordPset).getParameterNames();
      for (vector<string>::const_iterator key = paramKeys.begin(); key != paramKeys.end(); ++key) {
        jrInfo["Source"] = "AlcaHarvesting";
        jrInfo["FileClass"] = "ALCA";
        if (*key != "record") {
          jrInfo[*key] = (*recordPset).getUntrackedParameter<string>(*key);
        }
      }
    }
    recordMap[record] = jrInfo;
  }
}

PCLMetadataWriter::~PCLMetadataWriter() {}

void PCLMetadataWriter::analyze(const edm::Event &event, const edm::EventSetup &eSetup) {}

void PCLMetadataWriter::beginRun(const edm::Run &run, const edm::EventSetup &eSetup) {}

void PCLMetadataWriter::endRun(const edm::Run &run, const edm::EventSetup &eSetup) {
  const DropBoxMetadata *metadata = nullptr;

  if (readFromDB) {
    // Read the objects
    edm::ESHandle<DropBoxMetadata> mdPayload;
    eSetup.get<DropBoxMetadataRcd>().get(mdPayload);

    metadata = mdPayload.product();
  }

  // get the PoolDBOutputService
  Service<cond::service::PoolDBOutputService> poolDbService;
  if (poolDbService.isAvailable()) {
    edm::Service<edm::JobReport> jr;
    if (jr.isAvailable()) {
      // the filename is unique for all records
      string filename = poolDbService->session().connectionString();

      // loop over all records
      for (map<string, map<string, string>>::const_iterator recordAndMap = recordMap.begin();
           recordAndMap != recordMap.end();
           ++recordAndMap) {
        string record = (*recordAndMap).first;

        // this is the map of metadata that we write in the JR
        map<string, string> jrInfo = (*recordAndMap).second;
        if (readFromDB) {
          if (metadata->knowsRecord(record)) {
            jrInfo = metadata->getRecordParameters(record).getParameterMap();
          }
        }

        // name of the the input tag in the metadata for the condUploader
        // metadata needs to be the same as the tag written out by the
        // harvesting step
        jrInfo["inputtag"] = poolDbService->tag(record);

        // actually write in the job report
        jr->reportAnalysisFile(filename, jrInfo);
      }
    }
  }
}

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(PCLMetadataWriter);