BeamSpot.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Eugenio Paoloni                                          *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <TMatrixDSym.h>
#include <TVector3.h>
 
namespace Belle2 {
  class BeamSpot: public TObject {
  public:
 
    BeamSpot():  m_IPPosition(),
      m_IPPositionCovMatrix(3),
      m_sizeCovMatrix(3)
    { }
 
    ~BeamSpot() {}
 
 
    bool operator==(const BeamSpot& b) const
    {
      return
        b.m_IPPosition == m_IPPosition &&
        b.m_IPPositionCovMatrix == m_IPPositionCovMatrix &&
        b.m_sizeCovMatrix == m_sizeCovMatrix;
    }
 
    void setSizeCovMatrix(const TMatrixDSym& size)
    {
      m_sizeCovMatrix = size;
    }
 
    void setIP(const TVector3& ipPosition, const TMatrixDSym& covariance)
    {
      m_IPPosition = ipPosition;
      m_IPPositionCovMatrix = covariance;
    }
 
    const TVector3& getIPPosition() const
    {
      return m_IPPosition;
    }
 
    const TMatrixDSym& getIPPositionCovMatrix() const
    {
      return m_IPPositionCovMatrix;
    }
 
    const TMatrixDSym& getSizeCovMatrix() const
    {
      return m_sizeCovMatrix;
    }
 
    TMatrixDSym getCovVertex() const
    {
      return  getSizeCovMatrix() + getIPPositionCovMatrix();
    }
 
    static unsigned short getGlobalUniqueID() { return 1; }
 
  private:
 
    TVector3 m_IPPosition;
 
    TMatrixDSym m_IPPositionCovMatrix;
 
    TMatrixDSym m_sizeCovMatrix;
 
    ClassDef(BeamSpot, 2); 
  };
 
} //Belle2 namespace