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);
    }
  };
}


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