AxialTrackUtil.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 <cdc/topology/ISuperLayer.h>
#include <tracking/trackingUtilities/numerics/ESign.h>
 
#include <Math/Vector2D.h>
 
#include <array>
#include <vector>
 
namespace Belle2 {
 
  namespace TrackingUtilities {
    class CDCTrack;
    class CDCTrajectory2D;
    class CDCWireHit;
    class CDCRecoHit3D;
  }
  namespace TrackFindingCDC {

    struct AxialTrackUtil {
 
    public:
      static void addCandidateFromHits(const std::vector<const TrackingUtilities::CDCWireHit*>& foundAxialWireHits,
                                       const std::vector<const TrackingUtilities::CDCWireHit*>& allAxialWireHits,
                                       std::vector<TrackingUtilities::CDCTrack>& axialTracks,
                                       bool withPostprocessing = true);

      static bool postprocessTrack(TrackingUtilities::CDCTrack& track,
                                   const std::vector<const TrackingUtilities::CDCWireHit*>& allAxialWireHits);
 
    private:
      static bool checkTrackQuality(const TrackingUtilities::CDCTrack& track);
 
    public:
      static void normalizeTrack(TrackingUtilities::CDCTrack& track);

      static void updateRecoHit3D(const TrackingUtilities::CDCTrajectory2D& trajectory2D, TrackingUtilities::CDCRecoHit3D& hit);

      static void deleteHitsFarAwayFromTrajectory(TrackingUtilities::CDCTrack& track, double maximumDistance = 0.2);

      static void assignNewHitsToTrack(TrackingUtilities::CDCTrack& track,
                                       const std::vector<const TrackingUtilities::CDCWireHit*>& allAxialWireHits,
                                       double minimalDistance = 0.2);

      static std::vector<TrackingUtilities::CDCRecoHit3D> splitBack2BackTrack(TrackingUtilities::CDCTrack& track);

      static bool isBack2BackTrack(TrackingUtilities::CDCTrack& track);

      static TrackingUtilities::ESign getMajorArmSign(const TrackingUtilities::CDCTrack& track,
                                                      const ROOT::Math::XYVector& center);

      static int getArmSignVote(const TrackingUtilities::CDCTrack& track, const ROOT::Math::XYVector& center);
 

      static void removeHitsAfterSuperLayerBreak(TrackingUtilities::CDCTrack& track);

      static TrackingUtilities::ESign getArmSign(const TrackingUtilities::CDCRecoHit3D& hit, const ROOT::Math::XYVector& center);
 
    public:
      static void deleteTracksWithLowFitProbability(std::vector<TrackingUtilities::CDCTrack>& axialTracks,
                                                    double minimal_probability_for_good_fit = 0.4);

      static void deleteShortTracks(std::vector<TrackingUtilities::CDCTrack>& axialTracks, double minimal_size = 5);
 
    private:
      static std::vector<CDC::ISuperLayer> getSLayerHoles(const std::array<int, CDC::ISuperLayerUtil::c_N>&
                                                          nHitsBySLayer);  // return 0;

      static CDC::ISuperLayer getFirstOccupiedISuperLayer(const std::array<int, CDC::ISuperLayerUtil::c_N>&
                                                          nHitsBySLayer);

      static CDC::ISuperLayer getLastOccupiedISuperLayer(const std::array<int, CDC::ISuperLayerUtil::c_N>&
                                                         nHitsBySLayer);
    };
  }
}