PileUpEventPrincipal.h

Go to the documentation of this file.

#ifndef SimGeneral_MixingModule_PileUpEventPrincipal_h
#define SimGeneral_MixingModule_PileUpEventPrincipal_h

#include <set>
#include <string>

#include "FWCore/Framework/interface/EventPrincipal.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Utilities/interface/ProductKindOfType.h"
#include "FWCore/Utilities/interface/TypeID.h"
#include "DataFormats/Common/interface/BasicHandle.h"
#include "DataFormats/Common/interface/ConvertHandle.h"
#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"

namespace edm {
  class ModuleCallingContext;
}

class PileUpEventPrincipal {
public:

  PileUpEventPrincipal(edm::EventPrincipal const& ep, edm::ModuleCallingContext const* mcc, int bcr) :
    principal_(ep), mcc_(mcc), bunchCrossing_(bcr) {}

  edm::EventPrincipal const& principal() {
    return principal_;
  }

  edm::EventPrincipal const& principal() const {
    return principal_;
  }

  edm::ModuleCallingContext const *moduleCallingContext() const {
    return mcc_;
  }

  int bunchCrossing() const {
    return bunchCrossing_;
  }


  template<typename T>
  bool
  getByLabel(edm::InputTag const& tag, edm::Handle<T>& result) const {
    edm::BasicHandle bh = principal_.getByLabel(edm::PRODUCT_TYPE, edm::TypeID(typeid(T)), tag, nullptr, nullptr, mcc_);
    result = edm::convert_handle<T>(std::move(bh));
    return result.isValid();
  }

private: 
  edm::EventPrincipal const& principal_;
  edm::ModuleCallingContext const* mcc_;
  int bunchCrossing_;
};

#endif