v0fitter.cc

/**************************************************************************
 * 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.                  *
 **************************************************************************/
#include <tracking/v0Finding/fitter/V0Fitter.h>
 
#include <framework/gearbox/Const.h>
 
#include <genfit/FieldManager.h>
#include <genfit/MaterialEffects.h>
#include <genfit/ConstField.h>
#include <genfit/TGeoMaterialInterface.h>
 
#include <utility>
 
#include <gtest/gtest.h>
 
using namespace std;
 
namespace Belle2 {
  class V0FitterTest : public ::testing::Test {
  };
 
  TEST_F(V0FitterTest, GetTrackHypotheses)
  {
    genfit::MaterialEffects::getInstance()->init(new genfit::TGeoMaterialInterface());
    genfit::FieldManager::getInstance()->init(new genfit::ConstField(0., 0., 1.5));
    V0Fitter v0Fitter;
    const auto kShortTracks = v0Fitter.getTrackHypotheses(Const::Kshort);
    const auto photonTracks = v0Fitter.getTrackHypotheses(Const::photon);
    const auto lambdaTracks = v0Fitter.getTrackHypotheses(Const::Lambda);
    const auto antiLambdaTracks = v0Fitter.getTrackHypotheses(Const::antiLambda);
 
    EXPECT_EQ(Const::pion, kShortTracks.first);
    EXPECT_EQ(Const::pion, kShortTracks.second);
 
    EXPECT_EQ(Const::electron, photonTracks.first);
    EXPECT_EQ(Const::electron, photonTracks.second);
 
    EXPECT_EQ(Const::proton, lambdaTracks.first);
    EXPECT_EQ(Const::pion, lambdaTracks.second);
 
    EXPECT_EQ(Const::pion, antiLambdaTracks.first);
    EXPECT_EQ(Const::proton, antiLambdaTracks.second);
  }
 
  TEST_F(V0FitterTest, InitializeCuts)
  {
    genfit::MaterialEffects::getInstance()->init(new genfit::TGeoMaterialInterface());
    genfit::FieldManager::getInstance()->init(new genfit::ConstField(0., 0., 1.5));
    V0Fitter v0Fitter;
    v0Fitter.initializeCuts(1.0, 10000., {0.425, 0.575}, {1.09, 1.14}, {0, 0.1});
    EXPECT_EQ(1.0, v0Fitter.m_beamPipeRadius);
    EXPECT_EQ(10000., v0Fitter.m_vertexChi2CutOutside);
    EXPECT_EQ(std::make_tuple(0.425, 0.575), v0Fitter.m_invMassRangeKshort);
    EXPECT_EQ(std::make_tuple(1.09, 1.14),   v0Fitter.m_invMassRangeLambda);
    EXPECT_EQ(std::make_tuple(0, 0.1),       v0Fitter.m_invMassRangePhoton);
  }
 
}  // namespace