KinkFitter.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 <framework/datastore/StoreArray.h>
#include <framework/gearbox/Const.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <mdst/dataobjects/Kink.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/trackFitting/fitter/base/TrackFitter.h>
 
#include <utility>
 
namespace Belle2 {
  class KinkFitter {
 
  public:
    KinkFitter(const std::string& trackFitResultsName = "", const std::string& kinksName = "",
               const std::string& recoTracksName = "",
               const std::string& copiedRecoTracksName = "RecoTracksKinkTmp");
 
    void initializeCuts(const double vertexDistanceCut, const double vertexChi2Cut, const double precutDistance);
 
    void setFitterMode(const unsigned char fitterMode);
 
    bool fitAndStore(const Track* trackMother, const Track* trackDaughter, short filterFlag);
 
  private:
 
    RecoTrack* copyRecoTrackAndSplit(const RecoTrack* splitRecoTrack, const bool motherFlag, const unsigned int delta);
 
    bool splitRecoTrack(const RecoTrack* recoTrackSplit, short& recoTrackIndexMother, short& recoTrackIndexDaughter);
 
    unsigned int combineTracksAndFit(const Track* trackMother, const Track* trackDaughter);
 
    int findHitPositionForReassignment(const RecoTrack* recoTrack,
                                       ROOT::Math::XYZVector& vertexPos,
                                       int direction);
 
    RecoTrack* copyRecoTrackAndReassignCDCHits(RecoTrack* motherRecoTrack, RecoTrack* daughterRecoTrack,
                                               const bool motherFlag, const int delta);
 
    bool refitRecoTrackAfterReassign(RecoTrack* recoTrackMotherRefit, RecoTrack* recoTrackDaughterRefit,
                                     const RecoTrack* recoTrackMother, const RecoTrack* recoTrackDaughter);
 
    RecoTrack* copyRecoTrackForFlipAndRefit(const RecoTrack* recoTrack,
                                            const ROOT::Math::XYZVector& momentumSeed,
                                            const ROOT::Math::XYZVector& positionSeed,
                                            const double& timeSeed);
 
    RecoTrack* copyRecoTrackForRefit(const RecoTrack* recoTrack,
                                     const ROOT::Math::XYZVector& momentumSeed,
                                     const ROOT::Math::XYZVector& positionSeed,
                                     const double& timeSeed,
                                     const bool blockInnerStereoHits = false, const bool useAnotherFitter = false);
 
    bool isRefitImproveFilter6(const RecoTrack* recoTrackDaughterRefit, const ROOT::Math::XYZVector& motherPosLast);
 
    bool vertexFitWithRecoTracks(RecoTrack* recoTrackMother, RecoTrack* recoTrackDaughter,
                                 unsigned int& reassignHitStatus,
                                 ROOT::Math::XYZVector& vertexPos, double& distance,
                                 ROOT::Math::XYZVector vertexPosSeed = ROOT::Math::XYZVector(0, 0, 0));
 
    bool extrapolateToVertex(genfit::MeasuredStateOnPlane& stMother, genfit::MeasuredStateOnPlane& stDaughter,
                             const ROOT::Math::XYZVector& vertexPosition, unsigned int& reassignHitStatus);
 
    TrackFitResult* buildTrackFitResult(RecoTrack* recoTrack, const genfit::MeasuredStateOnPlane& msop,
                                        const double Bz, const Const::ParticleType trackHypothesis);
 
    void errMatrixForKFit(ROOT::Math::PxPyPzEVector& fourMomentum, TMatrixDSym& covMatrix6,
                          TMatrixDSym& errMatrix7);
 
 
  private:
 
    // variables used for input
    std::string m_recoTracksName;   
    StoreArray <RecoTrack> m_recoTracks; 
 
    // variables used for output
    StoreArray <TrackFitResult> m_trackFitResults;  
    StoreArray <Kink> m_kinks;  
 
    // cut variables
    double m_vertexDistanceCut;  
    double m_vertexChi2Cut;  
    double m_precutDistance;  
 
    // fitter working mode variables
    unsigned char m_kinkFitterMode;  
    bool m_kinkFitterModeHitsReassignment; 
    bool m_kinkFitterModeFlipAndRefit; 
    bool m_kinkFitterModeCombineAndFit; 
    bool m_kinkFitterModeSplitTrack; 
 
    // helper variables
    StoreArray <RecoTrack> m_copiedRecoTracks; 
    genfit::MeasuredStateOnPlane m_stMotherBuffer; 
    genfit::MeasuredStateOnPlane m_stDaughterBuffer; 
    RecoTrack* m_motherKinkRecoTrackCache; 
    RecoTrack* m_daughterKinkRecoTrackCache; 
  };
 
}