BeamSpotObjects.h

Go to the documentation of this file.

#ifndef BEAMSPOTOBJECTS_H
#define BEAMSPOTOBJECTS_H

#include "CondFormats/Serialization/interface/Serializable.h"

#include <cmath>
#include <sstream>
#include <cstring>

class BeamSpotObjects {
public:
  BeamSpotObjects() : sigmaZ_(0), beamwidthX_(0), beamwidthY_(0), dxdz_(0), dydz_(0), type_(-1) {
    beamwidthXError_ = 0;
    beamwidthYError_ = 0;
    emittanceX_ = 0;
    emittanceY_ = 0;
    betaStar_ = 0;
    std::memset(position_, 0, sizeof position_);
    std::memset(covariance_, 0, sizeof covariance_);
  }

  virtual ~BeamSpotObjects() {}

  void setPosition(double x, double y, double z) {
    position_[0] = x;
    position_[1] = y;
    position_[2] = z;
  };
  void setSigmaZ(double val) { sigmaZ_ = val; }
  void setdxdz(double val) { dxdz_ = val; }
  void setdydz(double val) { dydz_ = val; }
  void setBeamWidthX(double val) { beamwidthX_ = val; }
  void setBeamWidthY(double val) { beamwidthY_ = val; }
  void setBeamWidthXError(double val) { beamwidthXError_ = val; }
  void setBeamWidthYError(double val) { beamwidthYError_ = val; }
  void setCovariance(int i, int j, double val) { covariance_[i][j] = val; }
  void setType(int type) { type_ = type; }
  void setEmittanceX(double val) { emittanceX_ = val; }
  void setEmittanceY(double val) { emittanceY_ = val; }
  void setBetaStar(double val) { betaStar_ = val; }

  double x() const { return position_[0]; }
  double y() const { return position_[1]; }
  double z() const { return position_[2]; }
  double sigmaZ() const { return sigmaZ_; }
  double beamWidthX() const { return beamwidthX_; }
  double beamWidthY() const { return beamwidthY_; }
  double dxdz() const { return dxdz_; }
  double dydz() const { return dydz_; }
  double covariance(int i, int j) const { return covariance_[i][j]; }
  double xError() const { return sqrt(covariance_[0][0]); }
  double yError() const { return sqrt(covariance_[1][1]); }
  double zError() const { return sqrt(covariance_[2][2]); }
  double sigmaZError() const { return sqrt(covariance_[3][3]); }
  double beamWidthXError() const { return sqrt(covariance_[6][6]); }
  double beamWidthYError() const { return sqrt(covariance_[6][6]); }
  double dxdzError() const { return sqrt(covariance_[4][4]); }
  double dydzError() const { return sqrt(covariance_[5][5]); }
  int beamType() const { return type_; }
  double emittanceX() const { return emittanceX_; }
  double emittanceY() const { return emittanceY_; }
  double betaStar() const { return betaStar_; }

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

protected:
  double position_[3];
  double sigmaZ_;
  double beamwidthX_;
  double beamwidthY_;
  double beamwidthXError_;
  double beamwidthYError_;
  double dxdz_;
  double dydz_;
  double covariance_[7][7];
  int type_;
  double emittanceX_;
  double emittanceY_;
  double betaStar_;

  COND_SERIALIZABLE;
};

std::ostream& operator<<(std::ostream&, BeamSpotObjects beam);

#endif