PSimHitMap.h

Go to the documentation of this file.

#ifndef MCTruth_PSimHitMap_h
#define MCTruth_PSimHitMap_h
 
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "SimDataFormats/CrossingFrame/interface/CrossingFrame.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include <map>
 
class PSimHitMap {
public:
  PSimHitMap(const edm::InputTag &iT, edm::ConsumesCollector &&iC) : theMap(), theEmptyContainer() {
    sh_token = iC.consumes<CrossingFrame<PSimHit>>(iT);
  }
 
  void fill(const edm::Event &e);
 
  const edm::PSimHitContainer &hits(int detId) const;
 
  std::vector<int> detsWithHits() const;
 
private:
  std::map<int, edm::PSimHitContainer> theMap;
  edm::PSimHitContainer theEmptyContainer;
  //  edm::InputTag simHitsTag;
  edm::EDGetTokenT<CrossingFrame<PSimHit>> sh_token;
};
 
#endif