OneLumiPoolSource.cc

Go to the documentation of this file.

/*----------------------------------------------------------------------
----------------------------------------------------------------------*/
#include "PoolSource.h"
#include "FWCore/Framework/interface/InputSourceMacros.h"
#include "FWCore/Framework/interface/LuminosityBlockPrincipal.h"
#include "FWCore/Framework/interface/EventPrincipal.h"

#include "DataFormats/Provenance/interface/ProductRegistry.h"
#include "FWCore/Framework/interface/HistoryAppender.h"

namespace edm {
  
  class OneLumiPoolSource : public PoolSource {
  public:
    explicit OneLumiPoolSource(ParameterSet const& pset, InputSourceDescription const& desc);
    
  private:
    ItemType getNextItemType() override;
    std::shared_ptr<LuminosityBlockAuxiliary> readLuminosityBlockAuxiliary_() override;

    void readEvent_(EventPrincipal& eventPrincipal) override {
      PoolSource::readEvent_(eventPrincipal);
      eventPrincipal.setRunAndLumiNumber(eventPrincipal.run(),1);
    }

    bool seenFirstLumi_ = false;
  };

  OneLumiPoolSource::OneLumiPoolSource(ParameterSet const& pset, InputSourceDescription const& desc) :
    PoolSource(pset, desc)
  {
  }

  std::shared_ptr<LuminosityBlockAuxiliary>
  OneLumiPoolSource::readLuminosityBlockAuxiliary_() {
    auto ret = PoolSource::readLuminosityBlockAuxiliary_();
    auto hist = ret->processHistoryID();
    *ret = LuminosityBlockAuxiliary(ret->run(), 1, ret->beginTime(), ret->endTime());
    ret->setProcessHistoryID(hist);
    return ret;
  }

  InputSource::ItemType
  OneLumiPoolSource::getNextItemType() {
    auto type = PoolSource::getNextItemType();
    if(type==IsLumi) {
      if(seenFirstLumi_) {
        do {
          edm::HistoryAppender historyAppender;
          auto prodReg = std::make_shared<edm::ProductRegistry>();
          prodReg->setFrozen();
          edm::ProcessConfiguration procConfig;

          LuminosityBlockPrincipal temp(prodReg, procConfig, &historyAppender,0);
          readLuminosityBlock_(temp);
          type =PoolSource::getNextItemType();
        } while(type == IsLumi);
      } else {
        seenFirstLumi_ = true;
      }
    }
    return type;
  }
}
using namespace edm;

DEFINE_FWK_INPUT_SOURCE(OneLumiPoolSource);