AxialTrackUtil.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2014 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Viktor Trusov, Nils Braun                                *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/topology/ISuperLayer.h>
#include <tracking/trackFindingCDC/numerics/ESign.h>
 
#include <array>
#include <vector>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class CDCTrack;
    class CDCRecoHit3D;
    class CDCTrajectory2D;
    class CDCWireHit;
    class Vector2D;
 
    struct AxialTrackUtil {
 
    public:
      static void addCandidateFromHits(const std::vector<const CDCWireHit*>& foundAxialWireHits,
                                       const std::vector<const CDCWireHit*>& allAxialWireHits,
                                       std::vector<CDCTrack>& axialTracks,
                                       bool withPostprocessing = true);
 
      static bool postprocessTrack(CDCTrack& track, const std::vector<const CDCWireHit*>& allAxialWireHits);
 
    private:
      static bool checkTrackQuality(const CDCTrack& track);
 
    public:
      static void normalizeTrack(CDCTrack& track);
 
      static void updateRecoHit3D(const CDCTrajectory2D& trajectory2D, CDCRecoHit3D& hit);
 
      static void deleteHitsFarAwayFromTrajectory(CDCTrack& track, double maximumDistance = 0.2);
 
      static void assignNewHitsToTrack(CDCTrack& track,
                                       const std::vector<const CDCWireHit*>& allAxialWireHits,
                                       double minimalDistance = 0.2);
 
      static std::vector<CDCRecoHit3D> splitBack2BackTrack(CDCTrack& track);
 
      static bool isBack2BackTrack(CDCTrack& track);
 
      static ESign getMajorArmSign(const CDCTrack& track, const Vector2D& center);
 
      static int getArmSignVote(const CDCTrack& track, const Vector2D& center);
 
 
      static void removeHitsAfterSuperLayerBreak(CDCTrack& track);
 
      static ESign getArmSign(const CDCRecoHit3D& hit, const Vector2D& center);
 
    public:
      static void deleteTracksWithLowFitProbability(std::vector<CDCTrack>& axialTracks,
                                                    double minimal_probability_for_good_fit = 0.4);
 
      static void deleteShortTracks(std::vector<CDCTrack>& axialTracks, double minimal_size = 5);
 
    private:
      static std::vector<ISuperLayer> getSLayerHoles(const std::array<int, ISuperLayerUtil::c_N>& nHitsBySLayer);  // return 0;
 
      static ISuperLayer getFirstOccupiedISuperLayer(const std::array<int, ISuperLayerUtil::c_N>& nHitsBySLayer);
 
      static ISuperLayer getLastOccupiedISuperLayer(const std::array<int, ISuperLayerUtil::c_N>& nHitsBySLayer);
    };
  }
}