PathLengthToggledApplier.icc.h

/**************************************************************************
 * 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.                  *
 **************************************************************************/
#pragma once
 
#include <tracking/vxdHoughTracking/findlets/PathLengthToggledApplier.dcl.h>
#include <tracking/vxdHoughTracking/findlets/OnHitApplier.icc.h>
#include <tracking/trackFindingCDC/utilities/StringManipulation.h>
 
namespace Belle2 {
  namespace vxdHoughTracking {
 
    template <class AHit, class AFindlet>
    PathLengthToggledApplier<AHit, AFindlet>::PathLengthToggledApplier() : Super()
    {
      Super::addProcessingSignalListener(&m_twoHitFilterFindlet);
      Super::addProcessingSignalListener(&m_threeHitFilterFindlet);
      Super::addProcessingSignalListener(&m_fourHitFilterFindlet);
      Super::addProcessingSignalListener(&m_fiveHitFilterFindlet);
    }
 
    template <class AHit, class AFindlet>
    void PathLengthToggledApplier<AHit, AFindlet>::exposeParameters(ModuleParamList* moduleParamList,
        const std::string& prefix)
    {
      m_twoHitFilterFindlet.exposeParameters(moduleParamList, TrackFindingCDC::prefixed(prefix, "twoHit"));
      m_threeHitFilterFindlet.exposeParameters(moduleParamList, TrackFindingCDC::prefixed(prefix, "threeHit"));
      m_fourHitFilterFindlet.exposeParameters(moduleParamList, TrackFindingCDC::prefixed(prefix, "fourHit"));
      m_fiveHitFilterFindlet.exposeParameters(moduleParamList, TrackFindingCDC::prefixed(prefix, "fiveHit"));
    }
 
    template <class AHit, class AFindlet>
    void PathLengthToggledApplier<AHit, AFindlet>::apply(const std::vector<TrackFindingCDC::WithWeight<const AHit*>>& currentPath,
                                                         std::vector<TrackFindingCDC::WithWeight<AHit*>>& childHits)
    {
      // if currentPath.size() == 1, including the single child hits makes the total path length 2, so use the twoHitFilter
      // if currentPath.size() == 2, including the single child hits makes the total path length 3, so use the threeHitFilter
      // if currentPath.size() == 3, including the single child hits makes the total path length 4, so use the fourHitFilter
      // if currentPath.size() == 4, including the single child hits makes the total path length 5, so use the fiveHitFilter
      // in the rare case that currentPath already contains five or more hits, it's very likely already that it's a valid track
      // candidate, so no filter is applied here, but only the best N candidates sorted by quality are used further.
      if (currentPath.size() == 1) {
        m_twoHitFilterFindlet.apply(currentPath, childHits);
      } else if (currentPath.size() == 2) {
        m_threeHitFilterFindlet.apply(currentPath, childHits);
      } else if (currentPath.size() == 3) {
        m_fourHitFilterFindlet.apply(currentPath, childHits);
      } else if (currentPath.size() == 4) {
        m_fiveHitFilterFindlet.apply(currentPath, childHits);
      }
    }
 
  }
}