MkFitHitIndexMap.h

Go to the documentation of this file.

#ifndef RecoTracker_MkFit_MkFitHitIndexMap_h
#define RecoTracker_MkFit_MkFitHitIndexMap_h
 
#include "DataFormats/Provenance/interface/ProductID.h"
 
#include <vector>
 
class TrackingRecHit;
 
class MkFitHitIndexMap {
public:
  // This class holds the index and layer of a hit in the hit data
  // structure passed to mkFit
  class MkFitHit {
  public:
    MkFitHit() = default;
    explicit MkFitHit(int i, int l) : index_{i}, layer_{l} {}
 
    int index() const { return index_; }
    int layer() const { return layer_; }
 
  private:
    int index_ = -1;
    int layer_ = -1;
  };
 
  MkFitHitIndexMap() = default;
 
  void resizeByClusterIndex(edm::ProductID id, size_t clusterIndex);
 
  void increaseLayerSize(int layer, size_t additionalSize);
 
  void insert(edm::ProductID id, size_t clusterIndex, MkFitHit hit, const TrackingRecHit* hitPtr);
 
  const MkFitHit& mkFitHit(edm::ProductID id, size_t clusterIndex) const;
 
  const TrackingRecHit* hitPtr(MkFitHit hit) const { return mkFitToCMSSW_.at(hit.layer()).at(hit.index()).ptr; }
 
  size_t clusterIndex(MkFitHit hit) const { return mkFitToCMSSW_.at(hit.layer()).at(hit.index()).clusterIndex; }
 
private:
  // Helper struct to map (edm::ProductID, cluster index) to MkFitHit
  struct ClusterToMkFitHit {
    explicit ClusterToMkFitHit(edm::ProductID id) : productID(id) {}
    edm::ProductID productID;
    std::vector<MkFitHit> mkFitHits;  // indexed by cluster index
  };
 
  // Helper struct to map MkFitHit to (TrackingRecHit *, cluster index)
  struct CMSSWHit {
    CMSSWHit() = default;
    explicit CMSSWHit(const TrackingRecHit* p, size_t i) : ptr{p}, clusterIndex{i} {}
    const TrackingRecHit* ptr = nullptr;
    size_t clusterIndex = 0;
  };
 
  std::vector<ClusterToMkFitHit> cmsswToMkFit_;  // mapping from CMSSW(ProductID, cluster index) -> mkFit(index, layer)
  std::vector<std::vector<CMSSWHit> > mkFitToCMSSW_;  // reverse mapping, mkFit(layer, index) -> CMSSW hit
};
 
#endif