FWLiteESSource.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     CondLiteIO
// Class  :     FWLiteESSource
//
//
// Original Author:  Chris Jones
//         Created:  Thu Jun 17 15:47:35 CDT 2010
//

// system include files
#include <iostream>
#include <memory>
#include "TFile.h"
#include <mutex>

// user include files
#include "FWCore/Framework/interface/ESProductResolverProvider.h"
#include "FWCore/Framework/interface/ESSourceProductResolverNonConcurrentBase.h"
#include "FWCore/Framework/interface/EventSetupRecordIntervalFinder.h"
#include "DataFormats/FWLite/interface/EventSetup.h"
#include "DataFormats/FWLite/interface/Record.h"
#include "FWCore/Framework/interface/HCTypeTag.h"
#include "FWCore/Utilities/interface/TypeIDBase.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

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

#include "FWCore/Framework/interface/SourceFactory.h"

#include "FWCore/Concurrency/interface/SerialTaskQueue.h"
#include "FWCore/Concurrency/interface/WaitingTaskList.h"

// forward declarations
namespace edm {
  class EventSetupImpl;
}

namespace {
  struct TypeID : public edm::TypeIDBase {
    explicit TypeID(const std::type_info& iInfo) : edm::TypeIDBase(iInfo) {}
    TypeID() {}
    using TypeIDBase::typeInfo;
  };
  struct FWLiteESGenericHandle {
    FWLiteESGenericHandle(const TypeID& iType) : m_type(iType), m_data(nullptr), m_exception(nullptr) {}

    FWLiteESGenericHandle(const void* iData) : m_type(), m_data(iData), m_exception(nullptr) {}

    FWLiteESGenericHandle(cms::Exception* iException) : m_type(), m_data(nullptr), m_exception(iException) {}

    const std::type_info& typeInfo() const { return m_type.typeInfo(); }

    TypeID m_type;
    const void* m_data;
    cms::Exception* m_exception;
  };

  class FWLiteProductResolver : public edm::eventsetup::ESSourceProductResolverNonConcurrentBase {
  public:
    FWLiteProductResolver(const TypeID& iTypeID,
                          const fwlite::Record* iRecord,
                          edm::SerialTaskQueue* iQueue,
                          std::mutex* iMutex)
        : edm::eventsetup::ESSourceProductResolverNonConcurrentBase(iQueue, iMutex),
          m_type(iTypeID),
          m_record(iRecord),
          m_data{nullptr} {}

    void prefetch(const edm::eventsetup::DataKey& iKey, edm::EventSetupRecordDetails) final {
      assert(iKey.type() == m_type);

      FWLiteESGenericHandle h(m_type);
      m_record->get(h, iKey.name().value());
      m_data = h.m_data;

      if (nullptr != h.m_exception) {
        throw *(h.m_exception);
      }
    }

    void invalidateCache() override {
      edm::eventsetup::ESSourceProductResolverNonConcurrentBase::invalidateCache();
      m_data = nullptr;
    }

    void const* getAfterPrefetchImpl() const final { return m_data; }

  private:
    TypeID m_type;
    const fwlite::Record* m_record;
    void const* m_data;
  };
}  // namespace

class FWLiteESSource : public edm::eventsetup::ESProductResolverProvider, public edm::EventSetupRecordIntervalFinder {
public:
  FWLiteESSource(edm::ParameterSet const& iPS);
  FWLiteESSource(const FWLiteESSource&) = delete;
  const FWLiteESSource& operator=(const FWLiteESSource&) = delete;
  ~FWLiteESSource() override;

  using EventSetupRecordKey = edm::eventsetup::EventSetupRecordKey;

private:
  KeyedResolversVector registerResolvers(const EventSetupRecordKey&, unsigned int iovIndex) override;

  void setIntervalFor(const EventSetupRecordKey&, const edm::IOVSyncValue&, edm::ValidityInterval&) override;

  void delaySettingRecords() override;

  // ---------- member data --------------------------------
  std::unique_ptr<TFile> m_file;
  fwlite::EventSetup m_es;
  std::map<EventSetupRecordKey, fwlite::RecordID> m_keyToID;
  edm::SerialTaskQueue m_queue;
  std::mutex m_mutex;
};

FWLiteESSource::FWLiteESSource(edm::ParameterSet const& iPS)
    : m_file(TFile::Open(iPS.getParameter<std::string>("fileName").c_str())), m_es(m_file.get()) {}

FWLiteESSource::~FWLiteESSource() {}

edm::eventsetup::ESProductResolverProvider::KeyedResolversVector FWLiteESSource::registerResolvers(
    const EventSetupRecordKey& iRecordKey, unsigned int iovIndex) {
  KeyedResolversVector keyedProxiesVector;
  using edm::eventsetup::heterocontainer::HCTypeTag;

  fwlite::RecordID recID = m_keyToID[iRecordKey];
  const fwlite::Record& rec = m_es.get(recID);
  typedef std::vector<std::pair<std::string, std::string> > TypesAndLabels;
  TypesAndLabels typesAndLabels = rec.typeAndLabelOfAvailableData();

  std::cout << "Looking for data in record " << iRecordKey.name() << std::endl;
  for (TypesAndLabels::const_iterator it = typesAndLabels.begin(), itEnd = typesAndLabels.end(); it != itEnd; ++it) {
    std::cout << " need type " << it->first << std::endl;
    HCTypeTag tt = HCTypeTag::findType(it->first);
    if (tt != HCTypeTag()) {
      edm::eventsetup::DataKey dk(tt, edm::eventsetup::IdTags(it->second.c_str()));
      keyedProxiesVector.emplace_back(
          dk, std::make_shared<FWLiteProductResolver>(TypeID(tt.value()), &rec, &m_queue, &m_mutex));
    } else {
      LogDebug("UnknownESType") << "The type '" << it->first << "' is unknown in this job";
      std::cout << "    *****FAILED*****" << std::endl;
    }
  }
  return keyedProxiesVector;
}

void FWLiteESSource::setIntervalFor(const EventSetupRecordKey& iKey,
                                    const edm::IOVSyncValue& iSync,
                                    edm::ValidityInterval& oIOV) {
  m_es.syncTo(iSync.eventID(), iSync.time());

  const fwlite::Record& rec = m_es.get(m_keyToID[iKey]);
  edm::IOVSyncValue endSync(rec.endSyncValue().eventID(), rec.endSyncValue().time());
  if (rec.endSyncValue().eventID().run() == 0 && rec.endSyncValue().time().value() == 0ULL) {
    endSync = edm::IOVSyncValue::endOfTime();
  }
  oIOV = edm::ValidityInterval(edm::IOVSyncValue(rec.startSyncValue().eventID(), rec.startSyncValue().time()), endSync);
}

void FWLiteESSource::delaySettingRecords() {
  using edm::eventsetup::heterocontainer::HCTypeTag;
  std::vector<std::string> recordNames = m_es.namesOfAvailableRecords();

  for (std::vector<std::string>::const_iterator it = recordNames.begin(), itEnd = recordNames.end(); it != itEnd;
       ++it) {
    HCTypeTag t = HCTypeTag::findType(*it);
    if (t != HCTypeTag()) {
      EventSetupRecordKey key(t);
      findingRecordWithKey(key);
      usingRecordWithKey(key);
      m_keyToID[key] = m_es.recordID(it->c_str());
    }
  }
}

DEFINE_FWK_EVENTSETUP_SOURCE(FWLiteESSource);