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/TrackingHit/interface/PSimHit.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.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