SimHitMatcher.h

Go to the documentation of this file.

#ifndef GEMValidation_SimHitMatcher_h
#define GEMValidation_SimHitMatcher_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "SimDataFormats/Vertex/interface/SimVertexContainer.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "Geometry/CommonTopologies/interface/StripTopology.h"
#include <vector>
#include <map>
#include <set>

class GEMGeometry;

class SimHitMatcher {
public:
  SimHitMatcher(const SimTrack& track,
                const edm::Event&,
                const GEMGeometry& geom,
                const edm::ParameterSet& cfg,
                edm::EDGetToken& simhitToken,
                edm::EDGetToken& simtrackToken,
                edm::EDGetToken& simvertexToken);

  ~SimHitMatcher();

  const edm::PSimHitContainer& simHitsGEM() const { return gem_hits_; }
  std::set<unsigned int> detIdsGEM() const;

  std::set<unsigned int> detIdsGEMCoincidences() const;

  std::set<unsigned int> chamberIdsGEM() const;

  std::set<unsigned int> superChamberIdsGEM() const;
  std::set<unsigned int> superChamberIdsGEMCoincidences() const;

  const edm::PSimHitContainer& hitsInDetId(unsigned int) const;
  const edm::PSimHitContainer& hitsInChamber(unsigned int) const;
  const edm::PSimHitContainer& hitsInSuperChamber(unsigned int) const;

  int nLayersWithHitsInSuperChamber(unsigned int) const;

  int nPadsWithHits() const;
  int nCoincidencePadsWithHits() const;

  GlobalPoint simHitsMeanPosition(const edm::PSimHitContainer& sim_hits) const;

  float simHitsMeanStrip(const edm::PSimHitContainer& sim_hits) const;

  std::set<int> hitStripsInDetId(unsigned int, int margin_n_strips = 0) const;  // GEM or CSC
  std::set<int> hitPadsInDetId(unsigned int) const;                             // GEM
  std::set<int> hitCoPadsInDetId(unsigned int) const;                           // GEM coincidence pads with hits

  // what unique partitions numbers were hit by this simtrack?
  std::set<int> hitPartitions() const;  // GEM

private:
  void init(const edm::Event&);
  const SimTrack& track_;
  std::vector<unsigned int> getIdsOfSimTrackShower(unsigned trk_id,
                                                   const edm::SimTrackContainer& simTracks,
                                                   const edm::SimVertexContainer& simVertices);

  void matchSimHitsToSimTrack(std::vector<unsigned int> track_ids, const edm::PSimHitContainer& gem_hits);

  bool simMuOnlyGEM_;
  bool discardEleHitsGEM_;
  bool verbose_;
  std::string simInputLabel_;

  edm::Handle<edm::PSimHitContainer> gem_hits;
  edm::Handle<edm::SimTrackContainer> sim_tracks;
  edm::Handle<edm::SimVertexContainer> sim_vertices;

  const GEMGeometry& gem_geo_;
  std::map<unsigned int, unsigned int> trkid_to_index_;

  edm::PSimHitContainer no_hits_;
  edm::PSimHitContainer gem_hits_;
  std::map<unsigned int, edm::PSimHitContainer> gem_detid_to_hits_;
  std::map<unsigned int, edm::PSimHitContainer> gem_chamber_to_hits_;
  std::map<unsigned int, edm::PSimHitContainer> gem_superchamber_to_hits_;

  // detids with hits in pads
  std::map<unsigned int, std::set<int> > gem_detids_to_pads_;
  // detids with hits in 2-layer pad coincidences
  std::map<unsigned int, std::set<int> > gem_detids_to_copads_;
};

#endif