ClusterStorer.h

Go to the documentation of this file.

#ifndef RecoAlgos_ClusterStorer_h
#define RecoAlgos_ClusterStorer_h

#include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHit.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2D.h"
#include "DataFormats/TrackerRecHit2D/interface/Phase2TrackerRecHit1D.h"
#include "DataFormats/SiStripCluster/interface/SiStripCluster.h"
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"
#include "DataFormats/Phase2TrackerCluster/interface/Phase2TrackerCluster1D.h"
#include "DataFormats/Common/interface/DetSetVectorNew.h"

namespace helper {

  class ClusterStorer {
  public:
    ClusterStorer() {}
    void addCluster(TrackingRecHitCollection &hits, size_t index);
    void clear();
    //------------------------------------------------------------------
    //------------------------------------------------------------------
    void processAllClusters(edmNew::DetSetVector<SiPixelCluster> &pixelDsvToFill,
                            edm::RefProd<edmNew::DetSetVector<SiPixelCluster> > refPixelClusters,
                            edmNew::DetSetVector<SiStripCluster> &stripDsvToFill,
                            edm::RefProd<edmNew::DetSetVector<SiStripCluster> > refStripClusters,
                            edmNew::DetSetVector<Phase2TrackerCluster1D> &phase2OTDsvToFill,
                            edm::RefProd<edmNew::DetSetVector<Phase2TrackerCluster1D> > refPhase2OTClusters);

  private:
    template <typename ClusterRefType>
    class ClusterHitRecord {
    public:
      template <typename RecHitType>
      ClusterHitRecord(const RecHitType &hit, TrackingRecHitCollection &hits, size_t idx)
          : detid_(hit.geographicalId().rawId()), hits_(&hits), index_(idx), ref_(hit.cluster()) {}
      uint32_t detid() const { return detid_; }
      const ClusterRefType &clusterRef() const { return ref_; }
      bool operator<(const ClusterHitRecord<ClusterRefType> &other) const {
        return (detid_ != other.detid_) ? detid_ < other.detid_ : ref_ < other.ref_;
      }
      template <typename RecHitType>
      void rekey(const ClusterRefType &newRef);

    private:
      ClusterHitRecord() {}  
      uint32_t detid_;
      TrackingRecHitCollection *hits_;
      size_t index_;
      ClusterRefType ref_;
    };

    typedef ClusterHitRecord<SiPixelRecHit::ClusterRef> PixelClusterHitRecord;
    typedef ClusterHitRecord<SiStripRecHit2D::ClusterRef> StripClusterHitRecord;
    //FIXME:: this is just temporary solution for phase2,
    //probably is good to add a Phase2ClusterStorer?
    typedef ClusterHitRecord<Phase2TrackerRecHit1D::ClusterRef> Phase2OTClusterHitRecord;

    //------------------------------------------------------------------
    //------------------------------------------------------------------
    template <typename HitType, typename ClusterType>
    void processClusters(std::vector<ClusterHitRecord<typename HitType::ClusterRef> > &clusterRecords,
                         edmNew::DetSetVector<ClusterType> &dsvToFill,
                         edm::RefProd<edmNew::DetSetVector<ClusterType> > &refprod);

    //--- Information about the cloned clusters
    std::vector<PixelClusterHitRecord> pixelClusterRecords_;
    std::vector<StripClusterHitRecord> stripClusterRecords_;
    std::vector<Phase2OTClusterHitRecord> phase2OTClusterRecords_;
  };

}  // namespace helper

#endif