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