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_;
  };
}  // namespace reco
 
#endif