PFMultiDepthClusterizer.h

Go to the documentation of this file.

#ifndef __PFMultiDepthClusterizer_H__
#define __PFMultiDepthClusterizer_H__
 
#include "RecoParticleFlow/PFClusterProducer/interface/PFClusterBuilderBase.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHitFraction.h"
 
#include <unordered_map>
 
class PFMultiDepthClusterizer final : public PFClusterBuilderBase {
  typedef PFMultiDepthClusterizer B2DGPF;
 
public:
  PFMultiDepthClusterizer(const edm::ParameterSet& conf);
 
  ~PFMultiDepthClusterizer() override = default;
  PFMultiDepthClusterizer(const B2DGPF&) = delete;
  B2DGPF& operator=(const B2DGPF&) = delete;
 
  void update(const edm::EventSetup& es) override { _allCellsPosCalc->update(es); }
 
  void buildClusters(const reco::PFClusterCollection&,
                     const std::vector<bool>&,
                     reco::PFClusterCollection& outclus) override;
 
private:
  std::unique_ptr<PFCPositionCalculatorBase> _allCellsPosCalc;
  double nSigmaEta_;
  double nSigmaPhi_;
 
  class ClusterLink {
  public:
    ClusterLink(unsigned int i, unsigned int j, double DR, double DZ, double energy) {
      from_ = i;
      to_ = j;
      linkDR_ = DR;
      linkDZ_ = DZ;
      linkE_ = energy;
    }
 
    ~ClusterLink() = default;
 
    unsigned int from() const { return from_; }
    unsigned int to() const { return to_; }
    double dR() const { return linkDR_; }
    double dZ() const { return linkDZ_; }
    double energy() const { return linkE_; }
 
  private:
    unsigned int from_;
    unsigned int to_;
    double linkDR_;
    double linkDZ_;
    double linkE_;
  };
 
  void calculateShowerShapes(const reco::PFClusterCollection&, std::vector<double>&, std::vector<double>&);
  std::vector<ClusterLink> link(const reco::PFClusterCollection&,
                                const std::vector<double>&,
                                const std::vector<double>&);
  std::vector<ClusterLink> prune(std::vector<ClusterLink>&, std::vector<bool>& linkedClusters);
 
  void expandCluster(reco::PFCluster&,
                     unsigned int point,
                     std::vector<bool>& mask,
                     const reco::PFClusterCollection&,
                     const std::vector<ClusterLink>& links);
 
  void absorbCluster(reco::PFCluster&, const reco::PFCluster&);
};
 
DEFINE_EDM_PLUGIN(PFClusterBuilderFactory, PFMultiDepthClusterizer, "PFMultiDepthClusterizer");
 
#endif