BeamSpotObjects.h

Go to the documentation of this file.

#ifndef BEAMSPOTOBJECTS_H
#define BEAMSPOTOBJECTS_H

#include <math.h>
#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;

  private:

    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_;
    
};

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

#endif