CKFToSVDResult.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/ckf/svd/entities/CKFToSVDResult.h>
 
#include <tracking/trackingUtilities/utilities/ReversedRange.h>
 
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/spacePointCreation/SpacePoint.h>
#include <svd/dataobjects/SVDCluster.h>
 
using namespace Belle2;
 
CKFToSVDResult::CKFToSVDResult(const std::vector<TrackingUtilities::WithWeight<const CKFToSVDState*>>& path) :
  Super(path, path.back()->getMeasuredStateOnPlane())
{
  B2ASSERT("Path should not be empty", not path.empty());
 
  for (const TrackingUtilities::WithWeight<const CKFToSVDState*>& state : path) {
    const RecoTrack* relatedSVDTrack = state->getRelatedSVDTrack();
    if (m_relatedSVDRecoTrack) {
      B2ASSERT("There is a state with a different VXD track in it!", m_relatedSVDRecoTrack == relatedSVDTrack);
    } else {
      m_relatedSVDRecoTrack = relatedSVDTrack;
    }
  }
}
 
void CKFToSVDResult::addToRecoTrack(RecoTrack& recoTrack) const
{
  unsigned int sortingParameter = 0;
  for (const SpacePoint* spacePoint : TrackingUtilities::reversedRange(getHits())) {
    RelationVector<SVDCluster> relatedClusters = spacePoint->getRelationsTo<SVDCluster>();
    for (const SVDCluster& cluster : relatedClusters) {
      recoTrack.addSVDHit(&cluster, sortingParameter, Belle2::RecoHitInformation::c_CDCtoSVDCKF);
      sortingParameter++;
    }
  }
}
 
const RecoTrack* CKFToSVDResult::getRelatedSVDRecoTrack() const
{
  return m_relatedSVDRecoTrack;
}