fourHitFilters.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 <gtest/gtest.h>
 
#include <tracking/spacePointCreation/SpacePoint.h>
#include <tracking/trackFindingVXD/filterMap/threeHitVariables/CircleCenterXY.h>
#include <tracking/trackFindingVXD/filterMap/fourHitVariables/DeltaPt.h>
#include <tracking/trackFindingVXD/filterMap/fourHitVariables/DeltaCircleRadius.h>
#include <tracking/trackFindingVXD/filterMap/fourHitVariables/DeltaDistCircleCenter.h>
#include <tracking/trackFindingVXD/filterTools/SelectionVariableHelper.h>
 
#include <tracking/trackFindingVXD/filterMap/filterFramework/Shortcuts.h>
 
#include <framework/geometry/B2Vector3.h>
 
#include <vxd/geometry/SensorInfoBase.h>
 
#include <math.h>
 
 
using namespace std;
using namespace Belle2;
 
namespace VXDTFfourHitFilterTest {
 
  class FourHitFilterTest : public ::testing::Test {
  protected:
  };
 
 
 
  VXD::SensorInfoBase createSensorInfo(VxdID aVxdID, double globalX = 0., double globalY = 0., double globalZ = -0.)
  {
    // (SensorType type, VxdID id, double width, double length, double thickness, int uCells, int vCells, double width2=-1, double splitLength=-1, int vCells2=0)
    VXD::SensorInfoBase sensorInfoBase(VXD::SensorInfoBase::PXD, aVxdID, 2.3, 4.2, 0.3, 2, 4, -1);
 
    TGeoRotation r1;
    r1.SetAngles(45, 20, 30);      // rotation defined by Euler angles
    TGeoTranslation t1(globalX, globalY, globalZ);
    TGeoCombiTrans c1(t1, r1);
    TGeoHMatrix transform = c1;
    sensorInfoBase.setTransformation(transform);
    // also need to set the reco-transform
    sensorInfoBase.setTransformation(transform, true);
 
    return sensorInfoBase;
  }
 
 
 
  SpacePoint provideSpacePointDummy(double X, double Y, double Z)
  {
    VxdID aVxdID = VxdID(1, 1, 1);
    VXD::SensorInfoBase sensorInfoBase = createSensorInfo(aVxdID, X, Y, Z);
 
    PXDCluster aCluster = PXDCluster(aVxdID, 0., 0., 0.1, 0.1, 0, 0, 1, 1, 1, 1, 1, 1);
 
    return SpacePoint(&aCluster, &sensorInfoBase);
  }
 
 
 
  double lastResult = 0.;
 
 
 
  class ResultsObserver : public VoidObserver {
  public:
    template<class Var, typename ... otherTypes>
    static void notify(const Var& filterType,
                       typename Var::variableType fResult,
                       otherTypes ...)
    {
      B2INFO("ResultsObserver: Filter " << filterType.name() << " got result of " << fResult);
      lastResult = fResult;
    }
 
  };
 
 
  TEST_F(FourHitFilterTest, SpacePointCreation)
  {
    SpacePoint testSP = provideSpacePointDummy(1.2, 2.3, 4.2);
    EXPECT_FLOAT_EQ(1.2, testSP.getPosition().X()) ;
    EXPECT_FLOAT_EQ(2.3, testSP.getPosition().Y()) ;
    EXPECT_FLOAT_EQ(4.2, testSP.getPosition().Z()) ;
  }
 
 
  TEST_F(FourHitFilterTest, SelectionVariableName)
  {
    auto dDCircleCenter = DeltaDistCircleCenter<SpacePoint>();
    EXPECT_EQ("DeltaDistCircleCenter", dDCircleCenter.name());
    auto dPt = DeltaPt<SpacePoint>();
    EXPECT_EQ("DeltaPt", dPt.name());
  }
 
 
  TEST_F(FourHitFilterTest, TestDeltaPtAndDeltaDistCircleCenter)
  {
    SpacePoint outerSP = provideSpacePointDummy(0, 0, 0.);
    SpacePoint outerCenterSP = provideSpacePointDummy(-2, 0, 0.);
    SpacePoint innerCenterSP = provideSpacePointDummy(0, 2, 0.);
    SpacePoint innerSP = provideSpacePointDummy(2, 0, 0.);
 
    B2Vector3D centerO_OC_IC = CircleCenterXY<SpacePoint>::value(outerSP, outerCenterSP, innerCenterSP);
    B2Vector3D centerOC_IC_I = CircleCenterXY<SpacePoint>::value(outerCenterSP, innerCenterSP, innerSP);
    EXPECT_FLOAT_EQ(-1., centerO_OC_IC[0]);
    EXPECT_FLOAT_EQ(1., centerO_OC_IC[1]);
    EXPECT_FLOAT_EQ(0., centerOC_IC_I[0]);
    EXPECT_FLOAT_EQ(0., centerOC_IC_I[1]);
 
    Filter< DeltaDistCircleCenter<SpacePoint>, Range<double, double>, ResultsObserver > filterDeltaDistCircleCenter(
      Range<double, double>(1.41, 1.42));
    EXPECT_TRUE(filterDeltaDistCircleCenter.accept(outerSP, outerCenterSP, innerCenterSP, innerSP));
    EXPECT_FLOAT_EQ(sqrt(2), lastResult);
 
    Filter< DeltaPt<SpacePoint>, Range<double, double>, ResultsObserver > filteDeltaPt(Range<double, double>(0.002, 0.003));
    EXPECT_TRUE(filteDeltaPt.accept(outerSP, outerCenterSP, innerCenterSP, innerSP));
    EXPECT_FLOAT_EQ(0.00263349, lastResult);
 
    Filter< DeltaCircleRadius<SpacePoint>, Range<double, double>, ResultsObserver > filteDeltaCircleRadius(Range<double, double>(-0.59,
        -0.58));
    EXPECT_TRUE(filteDeltaCircleRadius.accept(outerSP, outerCenterSP, innerCenterSP, innerSP));
    EXPECT_FLOAT_EQ(-0.58578646, lastResult); // outerRad = 1.41..., innerRad = 2 -> outer - inner
  }
 
}