PFBlock.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_PFAlgo_PFBlock_h
#define RecoParticleFlow_PFAlgo_PFBlock_h
 
#include <map>
#include <iostream>
 
/* #include "boost/graph/adjacency_matrix.hpp" */
 
// #include "DataFormats/ParticleFlowReco/interface/PFBlockLink.h"
#include "DataFormats/ParticleFlowReco/interface/PFBlockElement.h"
// #include "DataFormats/ParticleFlowReco/interface/PFBlockParticle.h"
#include "DataFormats/Common/interface/OwnVector.h"
 
namespace reco {
 
 
  class PFBlock {
  public:
    struct Link {
      Link() : distance(-1), test(0) {}
      Link(float d, char t) : distance(d), test(t) {}
      float distance;
      char test;
    };
 
    typedef edm::OwnVector<reco::PFBlockElement>::const_iterator IE;
    /*     typedef std::vector< reco::PFBlockLink >::const_iterator IL; */
 
    // typedef std::vector< std::vector<double> > LinkData;
    typedef std::map<unsigned int, Link> LinkData;
 
    enum LinkTest { LINKTEST_RECHIT, LINKTEST_NLINKTEST, LINKTEST_ALL };
 
    PFBlock() {}
    // PFBlock(const PFBlock& other);
 
    void addElement(reco::PFBlockElement* element);
 
    void bookLinkData();
 
    bool matrix2vector(unsigned i, unsigned j, unsigned& index) const;
 
    void setLink(unsigned i1, unsigned i2, double dist, LinkData& linkData, LinkTest test = LINKTEST_RECHIT) const;
 
 
    void associatedElements(unsigned i,
                            const LinkData& linkData,
                            std::multimap<double, unsigned>& sortedAssociates,
                            reco::PFBlockElement::Type type = PFBlockElement::NONE,
                            LinkTest test = LINKTEST_RECHIT) const;
 
    double dist(unsigned ie1, unsigned ie2, const LinkData& linkData, LinkTest test) const {
      return dist(ie1, ie2, linkData);
    }
 
    double dist(unsigned ie1, unsigned ie2, const LinkData& linkData) const;
 
    const edm::OwnVector<reco::PFBlockElement>& elements() const { return elements_; }
 
    const LinkData& linkData() const { return linkData_; }
 
    LinkData& linkData() { return linkData_; }
 
  private:
    unsigned linkDataSize() const;
 
    LinkData linkData_;
 
    edm::OwnVector<reco::PFBlockElement> elements_;
  };
 
  std::ostream& operator<<(std::ostream& out, const PFBlock& co);
 
}  // namespace reco
 
#endif