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

    friend std::ostream& operator<<( std::ostream& out, const PFBlock& co );

  private:
    
    unsigned linkDataSize() const;
    
    LinkData                                        linkData_;
     
    edm::OwnVector< reco::PFBlockElement >          elements_;
        
  };
}

#endif