FitSegmentPairRelationVarSet.cc

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Oliver Frost                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#include <tracking/trackFindingCDC/filters/segmentPairRelation/FitSegmentPairRelationVarSet.h>
 
#include <tracking/trackFindingCDC/eventdata/tracks/CDCSegmentPair.h>
 
using namespace Belle2;
using namespace TrackFindingCDC;
 
bool FitSegmentPairRelationVarSet::extract(const Relation<const CDCSegmentPair>* ptrSegmentPairRelation)
{
  if (not ptrSegmentPairRelation) return false;
 
  const CDCSegmentPair* fromSegmentPair = ptrSegmentPairRelation->first;
  const CDCSegmentPair* toSegmentPair   = ptrSegmentPairRelation->second;
 
  CDCTrajectory3D fromFit = fromSegmentPair->getTrajectory3D();
  CDCTrajectory3D toFit   = toSegmentPair->getTrajectory3D();
 
  Vector3D commonOrigin = fromFit.getSupport();
 
  fromFit.setLocalOrigin(commonOrigin);
  toFit.setLocalOrigin(commonOrigin);
 
  const UncertainHelix& fromHelix = fromFit.getLocalHelix();
  const UncertainHelix& toHelix = toFit.getLocalHelix();
  UncertainHelix commonHelix = UncertainHelix::average(fromHelix, toHelix);
 
  finitevar<named("is_fitted")>() = not commonHelix->isInvalid();
  finitevar<named("curv")>() = commonHelix->curvatureXY();
  finitevar<named("tanl")>() = commonHelix->tanLambda();
  finitevar<named("z0")>() = commonOrigin.z();
 
  using namespace NHelixParameterIndices;
  finitevar<named("curv_var")>() = commonHelix.variance(c_Curv);
  finitevar<named("tanl_var")>() = commonHelix.variance(c_TanL);
  finitevar<named("z0_var")>() = commonHelix.variance(c_Z0);
 
  finitevar<named("chi2")>() = std::fabs(commonHelix.chi2());
  finitevar<named("ndf")>() = commonHelix.ndf();
  finitevar<named("chi2_per_ndf")>() = std::fabs(commonHelix.chi2()) / commonHelix.ndf();
  finitevar<named("p_value")>() = TMath::Prob(commonHelix.chi2(), commonHelix.ndf());
  return true;
}