BeamSpot.h

Go to the documentation of this file.

#ifndef BeamSpot_BeamSpot_h
#define BeamSpot_BeamSpot_h

#include <Rtypes.h>
#include "DataFormats/Math/interface/Error.h"
#include "DataFormats/Math/interface/Point3D.h"
#include <string>
#include <sstream>

namespace reco {

  class BeamSpot {
  public:
    enum BeamType { Unknown = -1, Fake = 0, LHC = 1, Tracker = 2 };

    typedef math::XYZPoint Point;
    enum { dimension = 7 };
    typedef math::Error<dimension>::type CovarianceMatrix;
    enum { dim3 = 3 };
    typedef math::Error<dim3>::type Covariance3DMatrix;
    enum { resdim = 2 };
    typedef math::Error<resdim>::type ResCovMatrix;

    BeamSpot();

    BeamSpot(const Point& point,
             double sigmaZ,
             double dxdz,
             double dydz,
             double BeamWidthX,
             const CovarianceMatrix& error,
             BeamType type = Unknown) {
      position_ = point;
      sigmaZ_ = sigmaZ;
      dxdz_ = dxdz;
      dydz_ = dydz;
      BeamWidthX_ = BeamWidthX;
      BeamWidthY_ = BeamWidthX;
      error_ = error;
      type_ = type;
      emittanceX_ = emittanceY_ = 0;
      betaStar_ = 0;
    };

    const Point& position() const { return position_; }
    double x0() const { return position_.X(); }
    double y0() const { return position_.Y(); }
    double z0() const { return position_.Z(); }

    double x(const double z) const { return x0() + dxdz() * (z - z0()); }
    double y(const double z) const { return y0() + dydz() * (z - z0()); }
    const Point position(const double z) const;
    //    const Point position(const double z) const {Point pos(x(z),y(z),z);    return pos;}

    double sigmaZ() const { return sigmaZ_; }
    double dxdz() const { return dxdz_; }
    double dydz() const { return dydz_; }
    double BeamWidthX() const { return BeamWidthX_; }
    double BeamWidthY() const { return BeamWidthY_; }
    double x0Error() const { return sqrt(error_(0, 0)); }
    double y0Error() const { return sqrt(error_(1, 1)); }
    double z0Error() const { return sqrt(error_(2, 2)); }
    double sigmaZ0Error() const { return sqrt(error_(3, 3)); }
    double dxdzError() const { return sqrt(error_(4, 4)); }
    double dydzError() const { return sqrt(error_(5, 5)); }

    double BeamWidthXError() const { return sqrt(error_(6, 6)); }
    double BeamWidthYError() const { return sqrt(error_(6, 6)); }

    void setBeamWidthX(double v) { BeamWidthX_ = v; }
    void setBeamWidthY(double v) { BeamWidthY_ = v; }

    double covariance(int i, int j) const { return error_(i, j); }
    CovarianceMatrix covariance() const { return error_; }
    Covariance3DMatrix covariance3D() const {
      Covariance3DMatrix matrix;
      for (int j = 0; j < 3; j++) {
        for (int k = j; k < 3; k++) {
          matrix(j, k) = error_(j, k);
        }
      }
      return matrix;
    };
    BeamType type() const { return type_; }
    void setType(BeamType type) { type_ = type; }
    Covariance3DMatrix rotatedCovariance3D() const;

    double emittanceX() const { return emittanceX_; }
    double emittanceY() const { return emittanceY_; }
    double betaStar() const { return betaStar_; }
    double beamWidthFromBeta(double z, double e) const {
      return sqrt(e * betaStar_ * (1 + pow((z - position_.Z()) / betaStar_, 2)));
    }
    void setEmittanceX(double v) { emittanceX_ = v; }
    void setEmittanceY(double v) { emittanceY_ = v; }
    void setbetaStar(double v) { betaStar_ = v; }

    void print(std::stringstream& ss) const;

  private:
    Point position_;
    CovarianceMatrix error_;

    Double32_t sigmaZ_;
    Double32_t BeamWidthX_;
    Double32_t BeamWidthY_;
    Double32_t dxdz_;
    Double32_t dydz_;
    Double32_t emittanceX_;
    Double32_t emittanceY_;
    Double32_t betaStar_;

    BeamType type_;
  };
  std::ostream& operator<<(std::ostream&, BeamSpot beam);

}  // namespace reco

#endif