AxialHitQuadTreeProcessor.h

/**************************************************************************
* BASF2 (Belle Analysis Framework 2)                                     *
* Copyright(C) 2014 - Belle II Collaboration                             *
*                                                                        *
* Author: The Belle II Collaboration                                     *
* Contributors: Viktor Trusov, Thomas Hauth, Nils Braun,                 *
*               Dmitrii Neverov                                          *
*                                                                        *
* This software is provided "as is" without any warranty.                *
**************************************************************************/
#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;
    };
  }
}