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 <set>
#include <iostream>



namespace reco {

  

  class PFDisplacedVertexSeed {

  public:


    typedef std::set< reco::TrackBaseRef >::iterator IEset;

    struct Compare{
      bool operator()(const TrackBaseRef& s1, const TrackBaseRef& s2) const
      {return s1.key() < s2.key();}
    };

    PFDisplacedVertexSeed();

    void addElement(TrackBaseRef);

    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::set < TrackBaseRef, Compare >& 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::set < TrackBaseRef , Compare >     elements_;
    GlobalPoint                   seedPoint_;
    float                         totalWeight_;

  };
}

#endif