Basic2DGenericPFlowClusterizer.h

Go to the documentation of this file.

#ifndef __Basic2DGenericPFlowClusterizer_H__
#define __Basic2DGenericPFlowClusterizer_H__

#include "RecoParticleFlow/PFClusterProducer/interface/PFClusterBuilderBase.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHitFraction.h"

#include <unordered_map>

class Basic2DGenericPFlowClusterizer : public PFClusterBuilderBase {
  typedef Basic2DGenericPFlowClusterizer B2DGPF;
 public:
  Basic2DGenericPFlowClusterizer(const edm::ParameterSet& conf);
    
  ~Basic2DGenericPFlowClusterizer() override = default;
  Basic2DGenericPFlowClusterizer(const B2DGPF&) = delete;
  B2DGPF& operator=(const B2DGPF&) = delete;

  void update(const edm::EventSetup& es) override { 
    _positionCalc->update(es); 
    if( _allCellsPosCalc ) _allCellsPosCalc->update(es);
    if( _convergencePosCalc ) _convergencePosCalc->update(es);
  }

  void buildClusters(const reco::PFClusterCollection&,
             const std::vector<bool>&,
             reco::PFClusterCollection& outclus) override;

 private:  
  const unsigned _maxIterations;
  const double _stoppingTolerance;
  const double _showerSigma2;
  const bool _excludeOtherSeeds;
  const double _minFracTot;
  const std::unordered_map<std::string,int> _layerMap;

  std::unordered_map<int,std::pair<std::vector<int>,std::vector<double> > > _recHitEnergyNorms;
  std::unique_ptr<PFCPositionCalculatorBase> _allCellsPosCalc;
  std::unique_ptr<PFCPositionCalculatorBase> _convergencePosCalc;
  
  void seedPFClustersFromTopo(const reco::PFCluster&,
                  const std::vector<bool>&,
                  reco::PFClusterCollection&) const;

  void growPFClusters(const reco::PFCluster&,
              const std::vector<bool>&,
              const unsigned toleranceScaling,
              const unsigned iter,
              double dist,
              reco::PFClusterCollection&) const;
  
  void prunePFClusters(reco::PFClusterCollection&) const;
};

DEFINE_EDM_PLUGIN(PFClusterBuilderFactory,
          Basic2DGenericPFlowClusterizer,
          "Basic2DGenericPFlowClusterizer");

#endif