PFDisplacedVertexCandidate.h

Go to the documentation of this file.

#ifndef DataFormat_ParticleFlowReco_PFDisplacedVertexCandidate_h
#define DataFormat_ParticleFlowReco_PFDisplacedVertexCandidate_h 

#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include <vector>
#include <map>
#include <iostream>

namespace reco {

  

  class PFDisplacedVertexCandidate {


  public:

    struct VertexLink {
      VertexLink()   : distance_(-1), dcaPoint_(0,0,0), test_(0) {}
      VertexLink(float d, GlobalPoint p, char t) : distance_(d), dcaPoint_(p), test_(t) {}
      float distance_;
      GlobalPoint dcaPoint_;
      char test_;
    };

    enum VertexLinkTest {
      LINKTEST_DCA,
      LINKTEST_DUMMY,
      LINKTEST_ALL
    };

    
    typedef std::map< unsigned int, VertexLink >  VertexLinkData;

    typedef std::map< float, std::pair<int,int> > DistMap;
    typedef std::vector< float > DistVector;

    PFDisplacedVertexCandidate();


    void addElement(const TrackBaseRef);

    void setLink(unsigned i1, 
         unsigned i2, 
         const float dist, 
         const GlobalPoint& dcaPoint,
         const VertexLinkTest test=LINKTEST_DCA );

    
    void associatedElements( const unsigned i,
                             const VertexLinkData& vertexLinkData, 
                             std::multimap<float, unsigned>& sortedAssociates,
                 const VertexLinkTest test=LINKTEST_DCA ) const; 


    DistMap r2Map() const;

    DistVector r2Vector() const;

    DistVector distVector() const;
    
    const GlobalPoint dcaPoint( unsigned ie1, unsigned ie2) const;


    bool isValid() const 
      {return elements_.size()>1;}    


    const TrackBaseRef& tref(unsigned ie) const 
      {return elements_[ie];}

    const std::vector < TrackBaseRef >& elements() const 
      {return elements_;}

    unsigned nTracks() const 
      {return elements_.size();}
 

    const VertexLinkData& vertexLinkData() const 
      {return vertexLinkData_;}

    void Dump(std::ostream& out = std::cout) const;

  private:


    const float dist( unsigned ie1, unsigned ie2) const;

    bool testLink (unsigned ie1, unsigned ie2) const;

    friend std::ostream& operator<<( std::ostream&, const PFDisplacedVertexCandidate& );
    
     

    unsigned vertexLinkDataSize() const;

    bool matrix2vector(unsigned i, unsigned j, unsigned& index) const;



    std::vector < TrackBaseRef >          elements_;

    VertexLinkData                        vertexLinkData_;


  };
}

#endif