AxialHitQuadTreeProcessor.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/trackFindingCDC/legendre/quadtree/QuadTreeProcessor.h>
#include <tracking/trackFindingCDC/legendre/precisionFunctions/PrecisionUtil.h>
 
#include <tracking/trackFindingCDC/numerics/LookupTable.h>
#include <tracking/trackFindingCDC/geometry/Vector2D.h>
 
#include <vector>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class AxialHitQuadTreeProcessor : public QuadTreeProcessor<long, float, const CDCWireHit> {
 
    public:
      static const LookupTable<Vector2D>& getCosSinLookupTable();
      static std::vector<float> createCurvBound(YSpan curvSpan, int lastLevel);
 
    public:
      AxialHitQuadTreeProcessor(int lastLevel,
                                int seedLevel,
                                const XYSpans& ranges,
                                PrecisionUtil::PrecisionFunction precisionFunction);
 
      AxialHitQuadTreeProcessor(const Vector2D& localOrigin,
                                const YSpan& curvSpan,
                                const LookupTable<Vector2D>* cosSinLookupTable);
 
    protected: // Section of specialized functions
      bool isLeaf(QuadTree* node) const final;
 
      XYSpans createChild(QuadTree* node, int i, int j) const final;
 
      bool isInNode(QuadTree* node, const CDCWireHit* wireHit) const final;
 
    protected: // Implementation details
      bool checkDerivative(QuadTree* node, const CDCWireHit* wireHit) const;
 
      bool checkExtremum(QuadTree* node, const CDCWireHit* wireHit) const;
 
    public: // debug stuff
      void drawHits(std::vector<const CDCWireHit*> hits, unsigned int color = 46) const;
      void drawNode(QuadTree* node) const;
 
    private:
      const LookupTable<Vector2D>* m_cosSinLookupTable;
 
      Vector2D m_localOrigin;
 
      bool m_twoSidedPhaseSpace;
 
      const double c_curlCurv = 0.02;
 
      PrecisionUtil::PrecisionFunction m_precisionFunction;
    };
  }
}