VertexCompatibleWithBeam.h

Go to the documentation of this file.

#ifndef VertexCompatibleWithBeam_H
#define VertexCompatibleWithBeam_H

#include "DataFormats/VertexReco/interface/Vertex.h"
#include "RecoVertex/VertexPrimitives/interface/VertexState.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"

class VertexDistance;

class VertexCompatibleWithBeam {
public:
  VertexCompatibleWithBeam(const VertexDistance &dist, float cut);
  VertexCompatibleWithBeam(const VertexDistance &dist, float cut, const reco::BeamSpot &beamSpot);

  VertexCompatibleWithBeam(const VertexCompatibleWithBeam &other);
  VertexCompatibleWithBeam &operator=(const VertexCompatibleWithBeam &other);
  virtual ~VertexCompatibleWithBeam();

  void setBeamSpot(const reco::BeamSpot &beamSpot);
  virtual bool operator()(const reco::Vertex &) const;
  virtual bool operator()(const reco::Vertex &, const VertexState &) const;

  // return value of VertexDistance to beam
  float distanceToBeam(const reco::Vertex &) const;
  float distanceToBeam(const reco::Vertex &, const VertexState &) const;

private:
  VertexDistance *theDistance;
  float theCut;
  VertexState theBeam;
};

#endif