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