PndGFRecoFit.h

Go to the documentation of this file.

#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;
};