BeamParameters.cc

#include <framework/utilities/TestHelpers.h>
#include <framework/dbobjects/BeamParameters.h>
 
#include <gtest/gtest.h>
 
using namespace std;
using namespace Belle2;
 
namespace {
  TEST(BeamParameters, BoostUpsilon4SToLab)
  {
    BeamParameters beamparams;
    //some values from Belle
    beamparams.setLER(3.49841, M_PI, std::vector<double>());
    beamparams.setHER(7.99638, 0.022, std::vector<double>());
 
    const TLorentzVector upsVec = beamparams.getHER() + beamparams.getLER();
    TLorentzVector upsVecCMS = upsVec;
    upsVecCMS.Transform(beamparams.getLabToCMS());
 
    const double mUpsilon = 10.5794;
    EXPECT_TRUE(fabs(upsVecCMS.E() - mUpsilon) < 1e-2);
    EXPECT_TRUE(fabs(upsVecCMS.X()) < 1e-15);
    EXPECT_TRUE(fabs(upsVecCMS.Y()) < 1e-15);
    EXPECT_TRUE(fabs(upsVecCMS.Z()) < 2e-15) << fabs(upsVecCMS.Z());
 
    EXPECT_TRUE(fabs(beamparams.getMass() - mUpsilon) < 1e-2);
  }
 
  TEST(BeamParameters, BoostIntoCMSAndBack)
  {
    BeamParameters beamparams;
    //some values from Belle
    beamparams.setLER(3.49841, M_PI, std::vector<double>());
    beamparams.setHER(7.99638, 0.022, std::vector<double>());
 
    auto backAndForth = beamparams.getCMSToLab().MatrixMultiplication(beamparams.getLabToCMS());
    TLorentzVector vec(1, 1, 1, 1);
    vec.Transform(backAndForth);
    EXPECT_TRUE(fabs(vec.X() - 1) < 1e-15);
    EXPECT_TRUE(fabs(vec.Y() - 1) < 1e-15);
    EXPECT_TRUE(fabs(vec.Z() - 1) < 1e-15);
  }
 
  TEST(BeamParameters, CovFromMatrix)
  {
    BeamParameters beamparams;
    TMatrixDSym upper(3), sym(3);
    int n(0);
    for (int i = 0; i < 3; i++) {
      for (int j = 0; j < 3; j++) {
        upper(i, j) = (j >= i) ? ++n : -1;
        if (j >= i) sym(i, j) = sym(j, i) = upper(i, j);
      }
    }
    beamparams.setCovHER(upper);
    EXPECT_EQ(sym, beamparams.getCovHER());
  }
 
  TEST(BeamParameters, ZeroElement)
  {
    BeamParameters beamparams;
    std::vector<double> cov;
    TMatrixDSym matrix(3);
    beamparams.setHER(0, 0, cov);
    beamparams.setVertex(TVector3(0, 0, 0), cov);
    EXPECT_EQ(matrix, beamparams.getCovHER());
    EXPECT_EQ(matrix, beamparams.getCovVertex());
  }
 
  TEST(BeamParameters, OneElement)
  {
    BeamParameters beamparams;
    for (int i = 0; i < 10; ++i) {
      std::vector<double> cov{1.*i};
      TMatrixDSym beam(3), vertex(3);
      beam(0, 0) = i;
      vertex(0, 0) = i;
      vertex(1, 1) = i;
      vertex(2, 2) = i;
      beamparams.setHER(0, 0, cov);
      beamparams.setVertex(TVector3(0, 0, 0), cov);
      EXPECT_EQ(beam, beamparams.getCovHER());
      EXPECT_EQ(vertex, beamparams.getCovVertex());
    }
  }
 
  TEST(BeamParameters, ThreeElements)
  {
    BeamParameters beamparams;
    std::vector<double> cov{1., 2., 3.};
    TMatrixDSym matrix(3);
    matrix(0, 0) = 1;
    matrix(1, 1) = 2;
    matrix(2, 2) = 3;
    beamparams.setHER(0, 0, cov);
    beamparams.setVertex(TVector3(0, 0, 0), cov);
    EXPECT_EQ(matrix, beamparams.getCovHER());
    EXPECT_EQ(matrix, beamparams.getCovVertex());
  }
 
  TEST(BeamParameters, SixElements)
  {
    BeamParameters beamparams;
    std::vector<double> cov{1., 2., 3., 4., 5., 6.};
    TMatrixDSym matrix(3);
    matrix(0, 0) = 1;
    matrix(0, 1) = matrix(1, 0) = 2;
    matrix(0, 2) = matrix(2, 0) = 3;
    matrix(1, 1) = 4;
    matrix(1, 2) = matrix(2, 1) = 5;
    matrix(2, 2) = 6;
    beamparams.setHER(0, 0, cov);
    beamparams.setVertex(TVector3(0, 0, 0), cov);
    EXPECT_EQ(matrix, beamparams.getCovHER());
    EXPECT_EQ(matrix, beamparams.getCovVertex());
  }
 
  TEST(BeamParameters, NineElements)
  {
    BeamParameters beamparams;
    std::vector<double> cov{1., 2., 3., 4., 5., 6., 7., 8., 9.};
    TMatrixDSym matrix(3);
    matrix(0, 0) = 1;
    matrix(0, 1) = matrix(1, 0) = 2;
    matrix(0, 2) = matrix(2, 0) = 3;
    matrix(1, 1) = 5;
    matrix(1, 2) = matrix(2, 1) = 6;
    matrix(2, 2) = 9;
    beamparams.setHER(0, 0, cov);
    beamparams.setVertex(TVector3(0, 0, 0), cov);
    EXPECT_EQ(matrix, beamparams.getCovHER());
    EXPECT_EQ(matrix, beamparams.getCovVertex());
  }
 
}  // namespace