TrackingParticleNumberOfLayers.h

Go to the documentation of this file.

#ifndef SimGeneral_TrackingAnalysis_TrackingParticleNumberOfLayers_h
#define SimGeneral_TrackingAnalysis_TrackingParticleNumberOfLayers_h

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

#include "DataFormats/Common/interface/ValueMap.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/TrackingHit/interface/PSimHit.h"

class TrackingParticleNumberOfLayers {
public:
  TrackingParticleNumberOfLayers(const edm::Event &iEvent,
                                 const std::vector<edm::EDGetTokenT<std::vector<PSimHit>>> &simHitTokens);

  enum { nTrackerLayers = 0, nPixelLayers = 1, nStripMonoAndStereoLayers = 2 };
  std::tuple<std::unique_ptr<edm::ValueMap<unsigned int>>,
             std::unique_ptr<edm::ValueMap<unsigned int>>,
             std::unique_ptr<edm::ValueMap<unsigned int>>>
  calculate(const edm::Handle<TrackingParticleCollection> &tps, const edm::EventSetup &iSetup) const;

private:
  // used as multimap, but faster
  std::vector<std::pair<unsigned int, const PSimHit *>> trackIdToHitPtr_;
};

#endif