TrackFitter.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun                                               *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFitting/measurementCreator/adder/MeasurementAdder.h>
#include <framework/gearbox/Const.h>
 
#include <TError.h>
 
#include <string>
#include <memory>
 
namespace genfit {
  class AbsFitter;
  class AbsTrackRep;
}
 
namespace Belle2 {
  class RecoTrack;
 
  class TrackFitter {
  public:
    static constexpr double s_defaultDeltaPValue = 1.0;
    static constexpr double s_defaultProbCut = 0.001;
    static constexpr unsigned int s_defaultMaxFailedHits = 5;
 
    TrackFitter(const std::string& storeArrayNameOfPXDHits = "",
                const std::string& storeArrayNameOfSVDHits = "",
                const std::string& storeArrayNameOfCDCHits = "",
                const std::string& storeArrayNameOfBKLMHits = "",
                const std::string& storeArrayNameOfEKLMHits = "",
                const bool initializeCDCTranslators = true):
      m_measurementAdder(storeArrayNameOfPXDHits, storeArrayNameOfSVDHits, storeArrayNameOfCDCHits,
                         storeArrayNameOfBKLMHits, storeArrayNameOfEKLMHits, initializeCDCTranslators)
    {
      resetFitterToDefaultSettings();
    }
 
    static int createCorrectPDGCodeForChargedStable(const Const::ChargedStable& particleType, const RecoTrack& recoTrack);
 
    void resetFitter(const std::shared_ptr<genfit::AbsFitter>& fitter);
 
    void resetFitterToDefaultSettings();
 
    bool fit(RecoTrack& recoTrack, genfit::AbsTrackRep* trackRepresentation) const;
 
    bool fit(RecoTrack& recoTrack, const Const::ChargedStable& particleType) const;
 
    bool fit(RecoTrack& recoTrack, const int pdgCode) const;
 
    bool fit(RecoTrack& recoTrack) const;
 
    void resetMeasurementCreatorsToDefaultSettings()
    {
      m_measurementAdder.resetMeasurementCreatorsToDefaultSettings();
    }
 
    void resetMeasurementCreators(
      const std::vector<std::shared_ptr<PXDBaseMeasurementCreator>>& pxdMeasurementCreators,
      const std::vector<std::shared_ptr<SVDBaseMeasurementCreator>>& svdMeasurementCreators,
      const std::vector<std::shared_ptr<CDCBaseMeasurementCreator>>& cdcMeasurementCreators,
      const std::vector<std::shared_ptr<BKLMBaseMeasurementCreator>>& bklmMeasurementCreators,
      const std::vector<std::shared_ptr<EKLMBaseMeasurementCreator>>& eklmMeasurementCreators,
      const std::vector<std::shared_ptr<BaseMeasurementCreator>>& additionalMeasurementCreators)
    {
      m_measurementAdder.resetMeasurementCreators(pxdMeasurementCreators,
                                                  svdMeasurementCreators,
                                                  cdcMeasurementCreators,
                                                  bklmMeasurementCreators,
                                                  eklmMeasurementCreators,
                                                  additionalMeasurementCreators);
    }
 
    void resetMeasurementCreatorsUsingFactories(
      const std::map<std::string, std::map<std::string, std::string>>& pxdMeasurementCreators,
      const std::map<std::string, std::map<std::string, std::string>>& svdMeasurementCreators,
      const std::map<std::string, std::map<std::string, std::string>>& cdcMeasurementCreators,
      const std::map<std::string, std::map<std::string, std::string>>& bklmMeasurementCreators,
      const std::map<std::string, std::map<std::string, std::string>>& eklmMeasurementCreators,
      const std::map<std::string, std::map<std::string, std::string>>& additionalMeasurementCreators)
    {
      m_measurementAdder.resetMeasurementCreatorsUsingFactories(pxdMeasurementCreators,
                                                                svdMeasurementCreators,
                                                                cdcMeasurementCreators,
                                                                bklmMeasurementCreators,
                                                                eklmMeasurementCreators,
                                                                additionalMeasurementCreators);
    }
 
    void setgErrorIgnoreLevel(Int_t errorIgnoreLevel) { m_gErrorIgnoreLevel = errorIgnoreLevel; }
    Int_t getgErrorIgnoreLevel() { return m_gErrorIgnoreLevel; }
 
  private:
    std::shared_ptr<genfit::AbsFitter> m_fitter;
    bool m_skipDirtyCheck = false;
 
    MeasurementAdder m_measurementAdder;
 
    Int_t m_gErrorIgnoreLevel = kFatal;
 
    bool fitWithoutCheck(RecoTrack& recoTrack, const genfit::AbsTrackRep& trackRepresentation) const;
  };
}