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 O2_MCH_TRACKFITTER_H_
#define O2_MCH_TRACKFITTER_H_
 
#include "DataFormatsMCH/Cluster.h"
#include "MCHTracking/Track.h"
#include "MCHTracking/TrackParam.h"
 
namespace o2
{
namespace mch
{
 
class TrackFitter
{
 public:
  TrackFitter() = default;
  ~TrackFitter() = default;
 
  TrackFitter(const TrackFitter&) = delete;
  TrackFitter& operator=(const TrackFitter&) = delete;
  TrackFitter(TrackFitter&&) = delete;
  TrackFitter& operator=(TrackFitter&&) = delete;
 
  void initField(float l3Current, float dipoleCurrent);
 
  void setBendingVertexDispersion(double ey) { mBendingVertexDispersion2 = ey * ey; }
 
  void smoothTracks(bool smooth) { mSmooth = smooth; }
  bool isSmootherEnabled() { return mSmooth; }
 
  void useClusterResolution() { mUseChamberResolution = false; }
  void useChamberResolution() { mUseChamberResolution = true; }
  // Set the chamber resolution in x and y directions
  void setChamberResolution(double ex, double ey);
 
  void fit(Track& track, bool smooth = true, bool finalize = true,
           std::list<TrackParam>::reverse_iterator* itStartingParam = nullptr,
           bool skipLocalChi2Calculation = false);
 
  void runKalmanFilter(TrackParam& trackParam);
 
  static constexpr double getMaxChi2() { return SMaxChi2; }
 
 private:
  void initTrack(const Cluster& cl1, const Cluster& cl2, TrackParam& param);
  void addCluster(const TrackParam& startingParam, const Cluster& cl, TrackParam& param);
  void smoothTrack(Track& track, bool finalize, bool skipLocalChi2Calculation);
  void runSmoother(const TrackParam& previousParam, TrackParam& param, bool skipLocalChi2Calculation);
 
  static constexpr double SMaxChi2 = 2.e10; 
  static constexpr double SDefaultChamberZ[10] = {-526.16, -545.24, -676.4, -695.4, -967.5,
                                                  -998.5, -1276.5, -1307.5, -1406.6, -1437.6};
  static constexpr double SChamberThicknessInX0[10] = {0.065, 0.065, 0.075, 0.075, 0.035,
                                                       0.035, 0.035, 0.035, 0.035, 0.035};
 
  double mBendingVertexDispersion2 = 4900.; 
 
  bool mSmooth = false; 
 
  bool mUseChamberResolution = false; 
  double mChamberResolutionX2 = 0.04; 
  double mChamberResolutionY2 = 0.04; 
};
 
//_________________________________________________________________________________________________
inline void TrackFitter::setChamberResolution(double ex, double ey)
{
  mChamberResolutionX2 = ex * ex;
  mChamberResolutionY2 = ey * ey;
}
 
} // namespace mch
} // namespace o2
 
#endif // O2_MCH_TRACKFITTER_H_