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/SiStripCluster/interface/SiStripCluster.h"
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.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);
   
    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) const;
    private:
      ClusterHitRecord() {}
      uint32_t detid_;
      TrackingRecHitCollection *hits_;
      size_t   index_;
      ClusterRefType ref_;
    };
    
    typedef ClusterHitRecord<SiPixelRecHit::ClusterRef>   PixelClusterHitRecord;
    typedef ClusterHitRecord<SiStripRecHit2D::ClusterRef> StripClusterHitRecord;

    //------------------------------------------------------------------
    //------------------------------------------------------------------
    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_;
  };

}

#endif