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