PFDisplacedVertexSeed.h

Go to the documentation of this file.

#ifndef DataFormat_ParticleFlowReco_PFDisplacedVertexSeed_h
#define DataFormat_ParticleFlowReco_PFDisplacedVertexSeed_h

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

#include <vector>
#include <iostream>

namespace reco {


  class PFDisplacedVertexSeed {
  public:
    PFDisplacedVertexSeed();

    void addElement(TrackBaseRef);

    void reserveElements(size_t);

    void updateSeedPoint(const GlobalPoint& dcaPoint, const TrackBaseRef, const TrackBaseRef, double weight = 1);

    void mergeWith(const PFDisplacedVertexSeed& displacedVertex);

    bool isEmpty() const { return (elements_.empty()); }

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

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

    const GlobalPoint& seedPoint() const { return seedPoint_; }

    const double totalWeight() const { return totalWeight_; }

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

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


    std::vector<TrackBaseRef> elements_;
    GlobalPoint seedPoint_;
    float totalWeight_;
  };
}  // namespace reco

#endif