TrackFitter.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_MFT_TrackFitter_H_
#define ALICEO2_MFT_TrackFitter_H_
 
#include "MFTTracking/Cluster.h"
#include "MFTTracking/TrackCA.h"
#include "MFTTracking/MFTTrackingParam.h"
#include "DataFormatsMFT/TrackMFT.h"
#include <TLinearFitter.h>
#include <list>
 
namespace o2
{
namespace mft
{
 
template <typename T>
class TrackFitter
{
 
  using SMatrix55Sym = ROOT::Math::SMatrix<double, 5, 5, ROOT::Math::MatRepSym<double, 5>>;
  using SMatrix5 = ROOT::Math::SVector<Double_t, 5>;
 
 public:
  TrackFitter() = default;
  ~TrackFitter() = default;
 
  TrackFitter(const TrackFitter&) = delete;
  TrackFitter& operator=(const TrackFitter&) = delete;
  TrackFitter(TrackFitter&&) = delete;
  TrackFitter& operator=(TrackFitter&&) = delete;
 
  void setBz(float bZ) { mBZField = bZ; }
  void setMFTRadLength(float MFT_x2X0) { mMFTDiskThicknessInX0 = MFT_x2X0 / 5.0; }
  void setVerbosity(bool v) { mVerbose = v; }
  void setTrackModel(Int_t m) { mTrackModel = m; }
  auto& getTrackModel() const { return mTrackModel; }
  void setAlignResiduals(Float_t res) { mAlignResidual = res; }
 
  bool initTrack(T& track, bool outward = false);
  bool fit(T& track, bool outward = false);
 
  static constexpr double getMaxChi2() { return SMaxChi2; }
 
 private:
  bool propagateToZ(T& track, double z);
  bool propagateToNextClusterWithMCS(T& track, double z, int& startingLayerID, const int& newLayerID);
  bool computeCluster(T& track, int cluster, int& startingLayerID);
 
  bool mFieldON = true;
  Float_t mBZField; // kiloGauss.
  Float_t mMFTDiskThicknessInX0 = 0.042 / 5;
  Int_t mTrackModel = MFTTrackModel::Optimized;
  Float_t mAlignResidual = 0.f;
 
  static constexpr double SMaxChi2 = 2.e10; 
  bool mVerbose = false;
};
 
// Functions to estimate momentum and charge from track curvature
template <typename T>
Double_t invQPtFromFCF(const T& track, Double_t bFieldZ, Double_t& chi2);
Bool_t LinearRegression(Int_t nVal, std::vector<double>& xVal, std::vector<double>& yVal, std::vector<double>& yErr, Double_t& a, Double_t& ae, Double_t& b, Double_t& be);
 
} // namespace mft
} // namespace o2
 
#endif // ALICEO2_MFT_TrackFitter_H_