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