TrackingRecHitProduct.h

Go to the documentation of this file.

#ifndef FastSimulation_TrackingRecHitProducer_TrackingRecHitProduct_H
#define FastSimulation_TrackingRecHitProducer_TrackingRecHitProduct_H
 
#include "SimDataFormats/TrackingHit/interface/PSimHit.h"
 
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/TrackerRecHit2D/interface/FastSingleTrackerRecHit.h"
 
#include <memory>
#include <vector>
#include <unordered_map>
 
class TrackingRecHitProduct {
public:
  typedef std::pair<unsigned int, const PSimHit*> SimHitIdPair;
  typedef std::pair<FastSingleTrackerRecHit, std::vector<SimHitIdPair>> RecHitToSimHitIdPairs;
 
protected:
  const DetId& _detId;
 
  std::vector<SimHitIdPair> _simHitsIdPairList;
 
  std::vector<RecHitToSimHitIdPairs> _recHits;
 
public:
  TrackingRecHitProduct(const DetId& detId, std::vector<SimHitIdPair>& simHitsIdPairList)
      : _detId(detId), _simHitsIdPairList(simHitsIdPairList) {}
 
  inline const DetId& getDetId() const { return _detId; }
 
  virtual std::vector<SimHitIdPair>& getSimHitIdPairs() { return _simHitsIdPairList; }
 
  virtual void addRecHit(const FastSingleTrackerRecHit& recHit, std::vector<SimHitIdPair> simHitIdPairs = {}) {
    _recHits.push_back(std::make_pair(recHit, simHitIdPairs));
    for (unsigned int isimhit = 0; isimhit < simHitIdPairs.size(); ++isimhit) {
      _recHits.back().first.addSimTrackId(simHitIdPairs[isimhit].second->trackId());
    }
  }
 
  virtual const std::vector<RecHitToSimHitIdPairs>& getRecHitToSimHitIdPairs() const { return _recHits; }
 
  virtual unsigned int numberOfRecHits() const { return _recHits.size(); }
 
  virtual ~TrackingRecHitProduct() {}
};
 
typedef std::shared_ptr<TrackingRecHitProduct> TrackingRecHitProductPtr;
 
#endif