BeamParameters.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/dataobjects/MCInitialParticles.h>
#include <TMatrixDSym.h>
 
namespace Belle2 {
  class BeamParameters: public MCInitialParticles {
  public:
    using MCInitialParticles::setLER;
    using MCInitialParticles::setHER;
    using MCInitialParticles::setVertex;
 
    BeamParameters(): MCInitialParticles(), m_covHER{0}, m_covLER{0}, m_covVertex{0} {}
    BeamParameters(const BeamParameters& b): MCInitialParticles(b)
    {
      std::copy_n(b.m_covHER, 6, m_covHER);
      std::copy_n(b.m_covLER, 6, m_covLER);
      std::copy_n(b.m_covVertex, 6, m_covVertex);
    }
    BeamParameters& operator=(const BeamParameters& b)
    {
      MCInitialParticles::operator=(b);
      std::copy_n(b.m_covHER, 6, m_covHER);
      std::copy_n(b.m_covLER, 6, m_covLER);
      std::copy_n(b.m_covVertex, 6, m_covVertex);
      return *this;
    }
    bool operator==(const BeamParameters& b) const
    {
      // since we only save the covariance matrices with float precision we
      // need to also do the comparison with float precision.
      auto floatcmp = [](double dbl_a, double dbl_b) { return (float)dbl_a == (float)dbl_b; };
      return MCInitialParticles::operator==(b) &&
             std::equal(m_covHER, m_covHER + 6, b.m_covHER, floatcmp) &&
             std::equal(m_covLER, m_covLER + 6, b.m_covLER, floatcmp) &&
             std::equal(m_covVertex, m_covVertex + 6, b.m_covVertex, floatcmp);
    }
 
    void setCovHER(const TMatrixDSym& cov) { setCovMatrix(m_covHER, cov); }
    void setCovLER(const TMatrixDSym& cov) { setCovMatrix(m_covLER, cov); }
    void setCovVertex(const TMatrixDSym& cov) { setCovMatrix(m_covVertex, cov); }
 
    void setHER(double energy, double angleX, double angleY, const std::vector<double>& cov);
 
    void setLER(double energy, double angleX, double angleY, const std::vector<double>& cov);
 
    void setVertex(const TVector3& vertex, const std::vector<double>& cov);
 
    TMatrixDSym getCovHER() const { return getCovMatrix(m_covHER); }
    TMatrixDSym getCovLER() const { return getCovMatrix(m_covLER); }
    TMatrixDSym getCovVertex() const { return getCovMatrix(m_covVertex); }
 
    //double getEnergySmearingLER() const;
    //double getEnergySmearingHER() const;
    //double getEnergySmearingCMS() const;
 
  private:
    static TLorentzVector getFourVector(double energy, double angleX, double angleY);
    static void setCovMatrix(Double32_t* member, const std::vector<double>& cov, bool common);
    static void setCovMatrix(Double32_t* member, const TMatrixDSym& cov);
    static TMatrixDSym getCovMatrix(const Double32_t* member);
    Double32_t m_covHER[6];
    Double32_t m_covLER[6];
    Double32_t m_covVertex[6];
 
    ClassDef(BeamParameters, 2); 
  };
 
} //Belle2 namespace