AxialTrackCreatorHitLegendre.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Bastian Kronenbitter, Thomas Hauth, Viktor Trusov,       *
 *               Nils Braun, Oliver Frost, Dmitrii Neverov                *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/findlets/base/Findlet.h>
 
#include <tracking/trackFindingCDC/legendre/quadtree/AxialHitQuadTreeProcessor.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
    class CDCTrack;
    class CDCWireHit;
 
    class AxialTrackCreatorHitLegendre : public Findlet<const CDCWireHit* const, CDCTrack> {
 
    private:
      using Super = Findlet<const CDCWireHit* const, CDCTrack>;
 
    public:
      enum class EPass {
        NonCurlers,
        NonCurlersWithIncreasingThreshold,
        FullRange,
        Straight
      };
 
    public:
      AxialTrackCreatorHitLegendre();
 
      explicit AxialTrackCreatorHitLegendre(EPass pass);
 
      std::string getDescription() final;
 
      void exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix) final;
 
      void initialize() final;
 
      void apply(const std::vector<const CDCWireHit*>& axialWireHits,
                 std::vector<CDCTrack>& tracks) final;
 
    private:
      std::unique_ptr<AxialHitQuadTreeProcessor> constructQTProcessor(EPass pass);
 
      using CandidateReceiver = AxialHitQuadTreeProcessor::CandidateReceiver;
      void executeRelaxation(const CandidateReceiver& candidateReceiver,
                             AxialHitQuadTreeProcessor& qtProcessor);
 
    private: // Parameters
      EPass m_pass = EPass::NonCurlers;
 
      const double m_param_stepScale = 0.75;
 
      int m_param_minNHits = 10;
 
      double m_param_precision = 0.00000001;
    };
  }
}