PndGFRecoFit.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#pragma once

#include "PndTrack.h"
#include "PndGFRecoHitFactoryCreator.h"

#include "AbsKalmanFitter.h"
#include "MeasurementFactory.h"

#include "eventDisplay/include/EventDisplay.h"

#include "FairGeanePro.h"

class PndGFRecoFit {
 public:
  PndGFRecoFit();
  virtual ~PndGFRecoFit();

  void Init();

  PndTrack *Fit(PndTrack *prefit, int pdgcode);

  void SetFitter(genfit::AbsKalmanFitter *fitter) { fFitter = fitter; }

  void SetPropagateToIP(bool val = true) { fPropagateToIP = val; }

  void SetPropagateDistance(double val) { fPropagateDistance = val; }

  void UseEventDisplay(){ fGFdisplay = genfit::EventDisplay::getInstance(); }

 protected:
  void SetDefaultFitter();

 private:
  genfit::MeasurementFactory<genfit::AbsMeasurement> *fHitFactory = nullptr;
  genfit::AbsKalmanFitter *fFitter = nullptr;
  bool fPropagateToIP = false;
  float fPropagateDistance = 2.0;
  //    bool fPerpPlane = false;
  FairGeanePro *fPro0 = nullptr; // todo: replace Geane propagator with genfit propagator
  genfit::EventDisplay* fGFdisplay = nullptr;

  int fVerbose = 0;
};