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 GetX() const { return position_[0]; }
  double GetY() const { return position_[1]; }
  double GetZ() const { return position_[2]; }
  double GetSigmaZ() const { return sigmaZ_; }
  double GetBeamWidthX() const { return beamwidthX_; }
  double GetBeamWidthY() const { return beamwidthY_; }
  double Getdxdz() const { return dxdz_; }
  double Getdydz() const { return dydz_; }
  double GetCovariance(int i, int j) const { return covariance_[i][j]; }
  double GetXError() const { return sqrt(covariance_[0][0]); }
  double GetYError() const { return sqrt(covariance_[1][1]); }
  double GetZError() const { return sqrt(covariance_[2][2]); }
  double GetSigmaZError() const { return sqrt(covariance_[3][3]); }
  double GetBeamWidthXError() const { return sqrt(covariance_[6][6]); }
  double GetBeamWidthYError() const { return sqrt(covariance_[6][6]); }
  double GetdxdzError() const { return sqrt(covariance_[4][4]); }
  double GetdydzError() const { return sqrt(covariance_[5][5]); }
  int GetBeamType() const { return type_; }
  double GetEmittanceX() const { return emittanceX_; }
  double GetEmittanceY() const { return emittanceY_; }
  double GetBetaStar() 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