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