DataMixingHcalWorker.h

Go to the documentation of this file.

#ifndef SimDataMixingHcalWorker_h
#define SimDataMixingHcalWorker_h

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventPrincipal.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/Provenance/interface/ProductID.h"

#include <map>
#include <string>
#include <vector>

namespace edm {
  class ModuleCallingContext;

  class DataMixingHcalWorker {
  public:
    DataMixingHcalWorker();

    explicit DataMixingHcalWorker(const edm::ParameterSet &ps, edm::ConsumesCollector &&iC);

    virtual ~DataMixingHcalWorker();

    void putHcal(edm::Event &e);
    void addHcalSignals(const edm::Event &e);
    void addHcalPileups(const int bcr, const edm::EventPrincipal *, unsigned int EventId, ModuleCallingContext const *);

  private:
    // data specifiers

    // Hcal
    edm::InputTag HBHErechitCollectionSig_;  // secondary name given to collection
                                             // of EB rechits
    edm::InputTag HOrechitCollectionSig_;    // secondary name given to collection of
                                             // EB rechits
    edm::InputTag HFrechitCollectionSig_;    // secondary name given to collection of
                                             // EB rechits
    edm::InputTag ZDCrechitCollectionSig_;   // secondary name given to collection
                                             // of EB rechits

    edm::InputTag HBHEPileRecHitInputTag_;  // InputTag for HB RecHits for Pileup
    edm::InputTag HOPileRecHitInputTag_;    // InputTag for HO RecHits for Pileup
    edm::InputTag HFPileRecHitInputTag_;    // InputTag for HF RecHits for Pileup
    edm::InputTag ZDCPileRecHitInputTag_;   // InputTag for ZDC RecHits for Pileup

    edm::EDGetTokenT<HBHERecHitCollection> HBHERecHitToken_;  // Token to retrieve information
    edm::EDGetTokenT<HORecHitCollection> HORecHitToken_;      // Token to retrieve information
    edm::EDGetTokenT<HFRecHitCollection> HFRecHitToken_;      // Token to retrieve information
    edm::EDGetTokenT<ZDCRecHitCollection> ZDCRecHitToken_;    // Token to retrieve information

    edm::EDGetTokenT<HBHERecHitCollection> HBHERecHitPToken_;  // Token to retrieve information
    edm::EDGetTokenT<HORecHitCollection> HORecHitPToken_;      // Token to retrieve information
    edm::EDGetTokenT<HFRecHitCollection> HFRecHitPToken_;      // Token to retrieve information
    edm::EDGetTokenT<ZDCRecHitCollection> ZDCRecHitPToken_;    // Token to retrieve information

    std::string HBHERecHitCollectionDM_;  // secondary name to be given to EB
                                          // collection of hits
    std::string HORecHitCollectionDM_;    // secondary name to be given to EB
                                          // collection of hits
    std::string HFRecHitCollectionDM_;    // secondary name to be given to EB
                                          // collection of hits
    std::string ZDCRecHitCollectionDM_;   // secondary name to be given to EB
                                          // collection of hits

    typedef std::multimap<DetId, HBHERecHit> HBHERecHitMap;
    typedef std::multimap<DetId, HFRecHit> HFRecHitMap;
    typedef std::multimap<DetId, HORecHit> HORecHitMap;
    typedef std::multimap<DetId, ZDCRecHit> ZDCRecHitMap;

    HBHERecHitMap HBHERecHitStorage_;
    HFRecHitMap HFRecHitStorage_;
    HORecHitMap HORecHitStorage_;
    ZDCRecHitMap ZDCRecHitStorage_;

    //      unsigned int eventId_; //=0 for signal, from 1-n for pileup events

    std::string label_;
  };
}  // namespace edm

#endif  // SimDataMixingHcalWorker_h