PFBlockAlgo.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_PFProducer_PFBlockAlgo_h
#define RecoParticleFlow_PFProducer_PFBlockAlgo_h
 
#include "DataFormats/ParticleFlowReco/interface/PFBlock.h"
#include "DataFormats/ParticleFlowReco/interface/PFBlockElement.h"
#include "DataFormats/ParticleFlowReco/interface/PFBlockFwd.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "RecoParticleFlow/PFProducer/interface/BlockElementImporterBase.h"
#include "RecoParticleFlow/PFProducer/interface/BlockElementLinkerBase.h"
#include "RecoParticleFlow/PFProducer/interface/KDTreeLinkerBase.h"
#include "DataFormats/Common/interface/OwnVector.h"
 
#include <memory>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace std {
  template <>
  struct hash<std::pair<unsigned int, unsigned int>> {
    typedef std::pair<unsigned int, unsigned int> arg_type;
    typedef unsigned int value_type;
    value_type operator()(const arg_type& arg) const { return arg.first ^ (arg.second << 1); }
  };
}  // namespace std
 
 
class PFBlockAlgo {
public:
  // the element list should **always** be a list of (smart) pointers
  typedef std::vector<std::unique_ptr<reco::PFBlockElement>> ElementList;
  //for skipping ranges
  typedef std::array<std::pair<unsigned int, unsigned int>, reco::PFBlockElement::kNBETypes> ElementRanges;
 
  PFBlockAlgo();
 
  ~PFBlockAlgo();
 
  void setLinkers(const std::vector<edm::ParameterSet>&);
 
  void setImporters(const std::vector<edm::ParameterSet>&, edm::ConsumesCollector&);
 
  // update event setup info of all linkers
  void updateEventSetup(const edm::EventSetup&);
 
  // run all of the importers and build KDtrees
  void buildElements(const edm::Event&);
 
  reco::PFBlockCollection findBlocks();
 
  void setDebug(bool debug) { debug_ = debug; }
 
private:
  void packLinks(reco::PFBlock& block,
                 const std::unordered_map<std::pair<unsigned int, unsigned int>, double>& links) const;
 
  inline void link(const reco::PFBlockElement* el1, const reco::PFBlockElement* el2, double& dist) const;
 
  // the test elements will be transferred to the blocks
  ElementList elements_;
  ElementRanges ranges_;
 
  bool debug_;
 
  friend std::ostream& operator<<(std::ostream&, const PFBlockAlgo&);
  bool useHO_;
 
  std::vector<std::unique_ptr<BlockElementImporterBase>> importers_;
 
  const std::unordered_map<std::string, reco::PFBlockElement::Type> elementTypes_;
  std::vector<std::unique_ptr<BlockElementLinkerBase>> linkTests_;
  unsigned int linkTestSquare_[reco::PFBlockElement::kNBETypes][reco::PFBlockElement::kNBETypes];
 
  std::vector<std::unique_ptr<KDTreeLinkerBase>> kdtrees_;
};
 
#endif