CondCore/Utilities/plugins/EmptyIOVSource.cc - master - githubmirror/cmssw

CondCore/Utilities/plugins/EmptyIOVSource.cc
98 строк4 KB
Cms Build
Clang-Format
22 май 2019, 05:03
22 май 2019, 05:0363b6371
#include "FWCore/Sources/interface/ProducerSourceBase.h"
#include "CondCore/CondDB/interface/Time.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include <string>
namespace cond {
  class EmptyIOVSource : public edm::ProducerSourceBase {
  public:
    EmptyIOVSource(edm::ParameterSet const&, edm::InputSourceDescription const&);
    ~EmptyIOVSource() override;
    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

  private:
    void produce(edm::Event& e) override;
    bool setRunAndEventInfo(edm::EventID& id,
                            edm::TimeValue_t& time,
                            edm::EventAuxiliary::ExperimentType& eType) override;
    void initialize(edm::EventID& id, edm::TimeValue_t& time, edm::TimeValue_t& interval) override;

  private:
    TimeType m_timeType;
    Time_t m_firstValid;
    Time_t m_lastValid;
    Time_t m_interval;
    Time_t m_current;
  };
}  // namespace cond

#include "FWCore/Utilities/interface/EDMException.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/IOVSyncValue.h"
//#include "DataFormats/Provenance/interface/EventID.h"
//#include <iostream>
namespace cond {
  //allowed parameters: firstRun, firstTime, lastRun, lastTime,
  //common paras: timetype,interval
  EmptyIOVSource::EmptyIOVSource(edm::ParameterSet const& pset, edm::InputSourceDescription const& desc)
      : edm::ProducerSourceBase(pset, desc, true),
        m_timeType(time::timeTypeFromName(pset.getParameter<std::string>("timetype"))),
        m_firstValid(pset.getParameter<unsigned long long>("firstValue")),
        m_lastValid(pset.getParameter<unsigned long long>("lastValue")),
        m_interval(pset.getParameter<unsigned long long>("interval")) {
    m_current = m_firstValid;
  }
  EmptyIOVSource::~EmptyIOVSource() {}
  void EmptyIOVSource::produce(edm::Event&) {}
  bool EmptyIOVSource::setRunAndEventInfo(edm::EventID& id,
                                          edm::TimeValue_t& time,
                                          edm::EventAuxiliary::ExperimentType& eType) {
    if (m_current <= m_lastValid) {
      if (m_timeType == cond::runnumber) {
        id = edm::EventID(m_current, id.luminosityBlock(), 1);
      } else if (m_timeType == cond::timestamp) {
        time = m_current;
      } else if (m_timeType == cond::lumiid) {
        edm::LuminosityBlockID l(m_current);
        id = edm::EventID(l.run(), l.luminosityBlock(), 1);
        //std::cout<<"run "<<l.run()<<std::endl;
        //std::cout<<"luminosityBlock "<<l.luminosityBlock()<<std::endl;
      }
    }
    bool ok = !(m_lastValid < m_current);
    m_current += m_interval;
    return ok;
  }
  void EmptyIOVSource::initialize(edm::EventID& id, edm::TimeValue_t& time, edm::TimeValue_t& interval) {
    if (m_timeType == cond::runnumber) {
      id = edm::EventID(m_firstValid, id.luminosityBlock(), 1);
      interval = 0LL;
    } else if (m_timeType == cond::timestamp) {
      time = m_firstValid;
      interval = m_interval;
    } else if (m_timeType == cond::lumiid) {
      edm::LuminosityBlockID l(m_firstValid);
      id = edm::EventID(l.run(), l.luminosityBlock(), 1);
      interval = 0LL;
    }
  }

  void EmptyIOVSource::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;
    desc.setComment("Creates runs, lumis and events containing no products.");
    ProducerSourceBase::fillDescription(desc);

    desc.add<std::string>("timetype");
    desc.add<unsigned long long>("firstValue")->setComment("The first run number to use or the first time");
    desc.add<unsigned long long>("lastValue")->setComment("The last run number to use or the last time");
    desc.add<unsigned long long>("interval");

    descriptions.add("source", desc);
  }

}  // namespace cond

#include "FWCore/Framework/interface/InputSourceMacros.h"
using cond::EmptyIOVSource;

DEFINE_FWK_INPUT_SOURCE(EmptyIOVSource);