PndRecoKalmanFit2 Class Reference

#include <PndRecoKalmanFit2.h>

Inheritance diagram for PndRecoKalmanFit2:

Public Member Functions
	PndRecoKalmanFit2 ()
	~PndRecoKalmanFit2 ()
void	SetGeane (Bool_t opt=kTRUE)
void	SetPropagateToIP (Bool_t opt=kTRUE)
void	SetPropagateDistance (Float_t opt=-1.f)
void	SetPerpPlane (Bool_t opt=kTRUE)
void	SetNumIterations (Int_t num)
void	SetVerbose (Int_t verb)
void	SetMvdBranchName (const TString &name)
void	SetCentralTrackerBranchName (const TString &name)
Bool_t	Init ()
PndTrack *	Fit (PndTrack *tBefore, Int_t PDG)
genfit::MeasurementFactory < genfit::AbsMeasurement > *	GetRecoHitFactory ()

Private Member Functions
	ClassDef (PndRecoKalmanFit2, 0)
	Verbose level. More...

Private Attributes
genfit::MeasurementFactory < genfit::AbsMeasurement > *	fTheRecoHitFactory
genfit::KalmanFitter	fGenFitter
FairGeanePro *	fPro
TString	fMvdBranchName
	Geane Propagator. More...
TString	fCentralTrackerBranchName
	Name of the TCA for MVD. More...
Bool_t	fUseGeane
	Name of the TCA for central tracker. More...
Bool_t	fPropagateToIP
	Flag to use Geane. More...
Float_t	fPropagateDistance
	Flag to propagate to the interaction point. More...
Bool_t	fPerpPlane
	Distance in [cm] to back-propagate the parameters, negative number means no backpropagation. More...
Int_t	fNumIt
	Flag to use as initial plane the one perpendicular to the track. More...
Int_t	fVerbose
	Number of iterations. More...

Detailed Description

Definition at line 31 of file PndRecoKalmanFit2.h.

Constructor & Destructor Documentation

PndRecoKalmanFit2::PndRecoKalmanFit2

(

)

Definition at line 70 of file PndRecoKalmanFit2.cxx.

References PndGeoHandling::Instance().

                                    : TNamed("Genfit", "Fit Tracks"),
                                      fMvdBranchName(""), fCentralTrackerBranchName(""),
                                      fUseGeane(kTRUE), fPropagateToIP(kTRUE), fPropagateDistance(-1.f), fPerpPlane(kFALSE), fNumIt(1), fVerbose(0)
{
  PndGeoHandling::Instance();
}

PndRecoKalmanFit2::~PndRecoKalmanFit2

(

)

Definition at line 206 of file PndRecoKalmanFit2.cxx.

{ }

Member Function Documentation

PndRecoKalmanFit2::ClassDef ( PndRecoKalmanFit2  , 0    )

private

Verbose level.

PndTrack * PndRecoKalmanFit2::Fit	(	PndTrack *	tBefore,
		Int_t	PDG
	)

Definition at line 208 of file PndRecoKalmanFit2.cxx.

References Bool_t, f, fabs(), fGenFitter, fPerpPlane, fPropagateDistance, fPropagateToIP, fTheRecoHitFactory, fVerbose, Genfit2Track2PndTrack(), genfit::Track::getFitStatus(), PndTrack::GetParamFirst(), PndTrack::GetTrackCand(), par, PndTrackCand2Genfit2TrackCand(), genfit::FitStatus::Print(), genfit::AbsFitter::processTrack(), genfit::TrackCand::setCovSeed(), PndTrack::SetFlag(), genfit::TrackCand::setPosMomSeedAndPdgCode(), Track, trk, and genfit::Exception::what().

Referenced by PndRecoMultiKalmanTask2::Exec(), and PndRecoKalmanTask2::Exec().

                                                             {
        PndTrack* tAfter = NULL;
        if (fVerbose > 0)
                std::cout << "PndRecoKalmanFit2::Fit" << std::endl;
        if (fabs(tBefore->GetParamFirst().GetPz()) < 1e-9) {
                tAfter = tBefore;
                tAfter->SetFlag(-10);
                std::cout << "tAfter = tBefore " << tBefore->GetParamFirst().GetPz()
                                << std::endl;
                return tAfter; // flag -10 : pz==0
        }

        //Int_t fCharge = tBefore->GetParamFirst().GetQ(); //[R.K. 01/2017] unused variable?
        Int_t PDGCode = PDG;
        TVector3 StartPos(tBefore->GetParamFirst().GetX(),
                        tBefore->GetParamFirst().GetY(), tBefore->GetParamFirst().GetZ());
        TVector3 StartMom(tBefore->GetParamFirst().GetPx(),
                        tBefore->GetParamFirst().GetPy(), tBefore->GetParamFirst().GetPz());

        TMatrixDSym covSeed(6);
        covSeed(0, 0) = tBefore->GetParamFirst().GetDX()
                        * tBefore->GetParamFirst().GetDX();
        covSeed(1, 1) = tBefore->GetParamFirst().GetDY()
                        * tBefore->GetParamFirst().GetDY();
        covSeed(2, 2) = tBefore->GetParamFirst().GetDZ()
                        * tBefore->GetParamFirst().GetDZ();

        covSeed(3, 3) = tBefore->GetParamFirst().GetDPx()
                        * tBefore->GetParamFirst().GetDPx();
        covSeed(4, 4) = tBefore->GetParamFirst().GetDPy()
                        * tBefore->GetParamFirst().GetDPy();
        covSeed(5, 5) = tBefore->GetParamFirst().GetDPz()
                        * tBefore->GetParamFirst().GetDPz();

        FairTrackParP par = tBefore->GetParamFirst();
        Int_t ierr = 0;
        FairTrackParH helix(&par, ierr);
        FairGeanePro fPro0;
        if (fVerbose == 0)
                fPro0.SetPrintErrors(kFALSE);
        FairTrackParH fRes;



        if (fPropagateToIP) {
                // Calculating params at PCA to Origin

                fPro0.SetPoint(TVector3(0, 0, 0));
                fPro0.PropagateToPCA(1, -1);
                Bool_t rc = fPro0.Propagate(&helix, &fRes, PDGCode);
                if (rc) {
                        StartPos.SetXYZ(fRes.GetX(), fRes.GetY(), fRes.GetZ());
                        StartMom.SetXYZ(fRes.GetPx(), fRes.GetPy(), fRes.GetPz());

                        covSeed(0, 0) = fRes.GetDX() * fRes.GetDX();
                        covSeed(1, 1) = fRes.GetDY() * fRes.GetDY();
                        covSeed(2, 2) = fRes.GetDZ() * fRes.GetDZ();

                        covSeed(3, 3) = fRes.GetDPx() * fRes.GetDPx();
                        covSeed(4, 4) = fRes.GetDPy() * fRes.GetDPy();
                        covSeed(5, 5) = fRes.GetDPz() * fRes.GetDPz();
                }
        } else if (fPropagateDistance > 0.f) {
                // Calculating params at fPropagateDistance cm before the first hit
//              FairTrackParP par = tBefore->GetParamFirst();
//              Int_t ierr = 0;
//              FairTrackParH *helix = new FairTrackParH(&par, ierr);
//              FairGeanePro *fPro0 = new FairGeanePro();
//              if (fVerbose == 0)
//                      fPro0->SetPrintErrors(kFALSE);
//              FairTrackParH *fRes = new FairTrackParH();
                fPro0.PropagateToLength(-fPropagateDistance);
                std::cout << "PndRekoKalmanFit2::Fit helix: ";
                helix.GetMomentum().Print();
                helix.GetPosition().Print();
                std::cout << std::endl;
                Bool_t rc = fPro0.Propagate(&helix, &fRes, PDGCode);
                if (rc) {
                        StartPos.SetXYZ(fRes.GetX(), fRes.GetY(), fRes.GetZ());
                        StartMom.SetXYZ(fRes.GetPx(), fRes.GetPy(), fRes.GetPz());

                        covSeed(0, 0) = fRes.GetDX() * fRes.GetDX();
                        covSeed(1, 1) = fRes.GetDY() * fRes.GetDY();
                        covSeed(2, 2) = fRes.GetDZ() * fRes.GetDZ();

                        covSeed(3, 3) = fRes.GetDPx() * fRes.GetDPx();
                        covSeed(4, 4) = fRes.GetDPy() * fRes.GetDPy();
                        covSeed(5, 5) = fRes.GetDPz() * fRes.GetDPz();
                }
        }

        TVector3 plane_v1, plane_v2;
        if (fPerpPlane) {
                plane_v1 = StartMom.Orthogonal();
                plane_v2 = StartPos.Cross(plane_v1);
        } else {
                plane_v1.SetXYZ(1., 0., 0.);
                plane_v2.SetXYZ(0., 1., 0.);
        }

        if (StartMom.Mag2() == 0) {
                std::cout
                                << "*** PndRecoKalmanFit2::Fit\tMomentum seed is ZERO. Cannot fit. ***"
                                << std::endl;
                return tBefore;
        }

        PndTrackCand trackCand = tBefore->GetTrackCand();

        genfit::AbsTrackRep *rep = new genfit::RKTrackRep(PDGCode);
        // PndTrackCand does not store seed, then PndTrackCand2Genfit2TrackCand cannot convert the seed.
        // You need to set the seed afterwards, taking it from PndTrack (setCovSeed/setPosMomSeedAndPdgCode)
        genfit::TrackCand* gfCand = PndTrackCand2Genfit2TrackCand(&trackCand); // TODO: link TrackCand to track
        gfCand->setCovSeed(covSeed);
        gfCand->setPosMomSeedAndPdgCode(StartPos, StartMom, PDGCode);

        genfit::Track* trk = new genfit::Track(*gfCand, *fTheRecoHitFactory, rep);
        delete (gfCand);

        // Start Fitter
        try {
                fGenFitter.processTrack(trk);
        } catch (genfit::Exception& e) {
//              delete(trk);
                std::cout << "*** PndRecoKalmanFit2::Fit" << "\t"
                                << "FITTER EXCEPTION ***" << std::endl;
                std::cerr << e.what();
        }
        if (fVerbose > 0) {
                std::cout << "*** PndRecoKalmanFit2::Fit" << "\t" << "SUCCESSFULL FIT!"
                                << std::endl;
                if (fVerbose > 2)
                        trk->getFitStatus()->Print();
        }

        try {
                tAfter = (PndTrack*) Genfit2Track2PndTrack(trk);
        } catch (genfit::Exception& e) {
                std::cout << "*** PndRecoKalmanFit2::Fit" << "\t"
                                << "CONVERSION EXCEPTION ***" << std::endl;
                std::cerr << e.what();
                tAfter = tBefore;
                tAfter->SetFlag(-2); // flag -2: conversion failed
//              delete(trk);
        }
//      tAfter = tBefore;
        delete(trk);

        if (fVerbose > 0)
                std::cout << "*** PndRecoKalmanFit2::Fit" << "\t" << "Fitting done"
                                << std::endl;

        return tAfter;
}

genfit::MeasurementFactory<genfit::AbsMeasurement>* PndRecoKalmanFit2::GetRecoHitFactory ( )

inline

Definition at line 52 of file PndRecoKalmanFit2.h.

References fTheRecoHitFactory.

{ return fTheRecoHitFactory;};

Bool_t PndRecoKalmanFit2::Init

(

)

Definition at line 77 of file PndRecoKalmanFit2.cxx.

References genfit::MeasurementFactory< measurement_T >::addProducer(), fCentralTrackerBranchName, fGenFitter, PndFtsMapCreator::FillTubeArray(), PndSttMapCreator::FillTubeArray(), fMvdBranchName, fNumIt, fPro, fTheRecoHitFactory, fUseGeane, fVerbose, PndGeoSttPar::GetGeometryType(), PndGeoFtsPar::GetGeometryType(), genfit::MaterialEffects::getInstance(), genfit::FieldManager::getInstance(), genfit::MaterialEffects::init(), genfit::FieldManager::init(), genfit::Exception::quiet(), rtdb, and genfit::AbsKalmanFitter::setMaxIterations().

Referenced by PndRecoMultiKalmanTask2::Init(), and PndRecoKalmanTask2::Init().

{
  //Get ROOT Manager
  FairRootManager* ioman= FairRootManager::Instance();
  if(ioman==0)
    {
      Error("PndRecoKalmanFit2::Init","RootManager not instantiated!");
      return kFALSE;
    }

  // STT map loading
  FairRuntimeDb* rtdb = FairRunAna::Instance()->GetRuntimeDb();
  PndGeoSttPar *sttParameters = (PndGeoSttPar*) rtdb->getContainer("PndGeoSttPar");
  TClonesArray *tubeArray = NULL;
  if(sttParameters->GetGeometryType() == 1) {
    PndSttMapCreator *mapper = new PndSttMapCreator(sttParameters);
    tubeArray = mapper->FillTubeArray();
  }

  // FTS map loading
  PndGeoFtsPar *ftsParameters = (PndGeoFtsPar*) rtdb->getContainer("PndGeoFtsPar");
  TClonesArray *ftsTubeArray = NULL;
  if(ftsParameters->GetGeometryType() != -1) {
    PndFtsMapCreator *ftsMapper = new PndFtsMapCreator(ftsParameters);
    ftsTubeArray = ftsMapper->FillTubeArray(); 
  }

  // Build hit factory -----------------------------
  fTheRecoHitFactory = new genfit::MeasurementFactory<genfit::AbsMeasurement>();
  if (fVerbose<2) genfit::Exception::quiet(true);
  
  TClonesArray* stripar;  TClonesArray* pixelar;
  if (fMvdBranchName == "Mix")
    {
      stripar=(TClonesArray*) ioman->GetObject("MVDHitsStripMix");
      if(stripar!=0)
        {
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MVDHitsStripMix"),new genfit::MeasurementProducer<PndSdsHit,PndSdsRecoHit2>(stripar));
          std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "MVDHitsStripMix array  found" << std::endl;
        }
      
      pixelar=(TClonesArray*) ioman->GetObject("MVDHitsPixelMix");
      if(pixelar!=0)
        { 
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MVDHitsPixelMix"),new genfit::MeasurementProducer<PndSdsHit,PndSdsRecoHit2>(pixelar));
          std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "MVDHitsPixelMix array  found" << std::endl;
        }
    }
  else
    {
      stripar=(TClonesArray*) ioman->GetObject("MVDHitsStrip");
      if(stripar!=0)
        {
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MVDHitsStrip"),new genfit::MeasurementProducer<PndSdsHit,PndSdsRecoHit2>(stripar));
          std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "MVDHitsStrip array  found" << std::endl;
        }
      
      pixelar=(TClonesArray*) ioman->GetObject("MVDHitsPixel");
      if(pixelar!=0)
        { 
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MVDHitsPixel"),new genfit::MeasurementProducer<PndSdsHit,PndSdsRecoHit2>(pixelar));
          std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "MVDHitsPixel array  found" << std::endl;
        }
    }
  
  TClonesArray* sthit; //TClonesArray *sttr;   //[R.K. 01/2017] unused variable?

  if (fCentralTrackerBranchName == "Mix")
    {
      sthit=(TClonesArray*) ioman->GetObject("STTHitMix");
      if(sthit!=0)
        {
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("STTHitMix"),new PndSttRecoHitProducer2<PndSttHit,PndSttRecoHit2>(sthit, tubeArray));
          std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "STTHitMix array  found" << std::endl;
        }
    }
  else
    {
      sthit=(TClonesArray*) ioman->GetObject("STTHit");
      if(sthit!=0)
        {
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("STTHit"),new PndSttRecoHitProducer2<PndSttHit,PndSttRecoHit2>(sthit, tubeArray));
          std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "SttHit array  found" << std::endl;
        }
    }
  
  TClonesArray* gemar=(TClonesArray*) ioman->GetObject("GEMHit");
  if(gemar!=0)
    {
      fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("GEMHit"),new genfit::MeasurementProducer<PndGemHit,PndGemRecoHit2>(gemar));
      std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "GEMHit array  found" << std::endl;
    }
  
  TClonesArray* mdtar=(TClonesArray*) ioman->GetObject("MdtHit");
  if(mdtar!=0)
    {
      fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MdtHit"),new genfit::MeasurementProducer<PndMdtHit,PndMdtRecoHit2>(mdtar));
      std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "MdtHit array  found" << std::endl;
    }

  TClonesArray* ftsar=(TClonesArray*) ioman->GetObject("FTSHit");
  if(ftsar!=0)
    {
      fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("FTSHit"),new PndFtsRecoHitProducer2<PndFtsHit,PndFtsRecoHit2>(ftsar, ftsTubeArray));
      std::cout << "*** PndRecoKalmanFit2::Init" << "\t" << "FtsHit array  found" << std::endl;
    }
 
  if (fUseGeane)
    {
      fPro = new FairGeanePro();
      if (fVerbose==0) fPro->SetPrintErrors(kFALSE);
    }
  else
    {
      Error("PndRecoKalmanFit2::Init","Only GEANE Propagation available!!!");
      return kFALSE;
    }
  
  fGenFitter.setMaxIterations(fNumIt);

  genfit::FieldManager::getInstance()->init(new PndGenfitField2());
  genfit::MaterialEffects::getInstance()->init(new genfit::TGeoMaterialInterface());

  std::cout << "===PndRecoKalmanFit2::Init() finished ===================================================" << std::endl;

  return kTRUE;
}

void PndRecoKalmanFit2::SetCentralTrackerBranchName ( const TString &  name )

inline

Definition at line 47 of file PndRecoKalmanFit2.h.

References fCentralTrackerBranchName, and name.

Referenced by PndRecoMultiKalmanTask2::Init(), and PndRecoKalmanTask2::Init().

{ fCentralTrackerBranchName = name; }

void PndRecoKalmanFit2::SetGeane ( Bool_t  opt = kTRUE )

inline

Definition at line 40 of file PndRecoKalmanFit2.h.

References fUseGeane.

Referenced by PndRecoKalmanTask2::Init().

{ fUseGeane = opt;     }

void PndRecoKalmanFit2::SetMvdBranchName ( const TString &  name )

inline

Definition at line 46 of file PndRecoKalmanFit2.h.

References fMvdBranchName, and name.

Referenced by PndRecoMultiKalmanTask2::Init(), and PndRecoKalmanTask2::Init().

{ fMvdBranchName = name; }

void PndRecoKalmanFit2::SetNumIterations ( Int_t  num )

inline

Definition at line 44 of file PndRecoKalmanFit2.h.

References fNumIt, and num.

Referenced by PndRecoMultiKalmanTask2::Init(), and PndRecoKalmanTask2::Init().

{ fNumIt    = num;     } 

void PndRecoKalmanFit2::SetPerpPlane ( Bool_t  opt = kTRUE )

inline

Definition at line 43 of file PndRecoKalmanFit2.h.

References fPerpPlane.

Referenced by PndRecoKalmanTask2::Init().

{ fPerpPlane = opt;    }

void PndRecoKalmanFit2::SetPropagateDistance ( Float_t  opt = -1.f )

inline

Definition at line 42 of file PndRecoKalmanFit2.h.

References fPropagateDistance.

Referenced by PndRecoKalmanTask2::Init().

{ fPropagateDistance = opt;}

void PndRecoKalmanFit2::SetPropagateToIP ( Bool_t  opt = kTRUE )

inline

Definition at line 41 of file PndRecoKalmanFit2.h.

References fPropagateToIP.

Referenced by PndRecoKalmanTask2::Init().

{ fPropagateToIP = opt;}

void PndRecoKalmanFit2::SetVerbose ( Int_t  verb )

inline

Definition at line 45 of file PndRecoKalmanFit2.h.

References fVerbose.

Referenced by PndRecoKalmanTask2::Init().

{ fVerbose  = verb;    }

Member Data Documentation

TString PndRecoKalmanFit2::fCentralTrackerBranchName

private

Name of the TCA for MVD.

Definition at line 65 of file PndRecoKalmanFit2.h.

Referenced by Init(), and SetCentralTrackerBranchName().

genfit::KalmanFitter PndRecoKalmanFit2::fGenFitter

private

Definition at line 60 of file PndRecoKalmanFit2.h.

Referenced by Fit(), and Init().

TString PndRecoKalmanFit2::fMvdBranchName

private

Geane Propagator.

Definition at line 64 of file PndRecoKalmanFit2.h.

Referenced by Init(), and SetMvdBranchName().

Int_t PndRecoKalmanFit2::fNumIt

private

Flag to use as initial plane the one perpendicular to the track.

Definition at line 71 of file PndRecoKalmanFit2.h.

Referenced by Init(), and SetNumIterations().

Bool_t PndRecoKalmanFit2::fPerpPlane

private

Distance in [cm] to back-propagate the parameters, negative number means no backpropagation.

Definition at line 70 of file PndRecoKalmanFit2.h.

Referenced by Fit(), and SetPerpPlane().

FairGeanePro* PndRecoKalmanFit2::fPro

private

Definition at line 62 of file PndRecoKalmanFit2.h.

Referenced by Init().

Float_t PndRecoKalmanFit2::fPropagateDistance

private

Flag to propagate to the interaction point.

Definition at line 69 of file PndRecoKalmanFit2.h.

Referenced by Fit(), and SetPropagateDistance().

Bool_t PndRecoKalmanFit2::fPropagateToIP

private

Flag to use Geane.

Definition at line 68 of file PndRecoKalmanFit2.h.

Referenced by Fit(), and SetPropagateToIP().

genfit::MeasurementFactory<genfit::AbsMeasurement>* PndRecoKalmanFit2::fTheRecoHitFactory

private

Definition at line 52 of file PndRecoKalmanFit2.h.

Referenced by Fit(), GetRecoHitFactory(), and Init().

Bool_t PndRecoKalmanFit2::fUseGeane

private

Name of the TCA for central tracker.

Definition at line 67 of file PndRecoKalmanFit2.h.

Referenced by Init(), and SetGeane().

Int_t PndRecoKalmanFit2::fVerbose

private

Number of iterations.

Definition at line 72 of file PndRecoKalmanFit2.h.

Referenced by Fit(), Init(), and SetVerbose().

---

The documentation for this class was generated from the following files:

• PndRecoKalmanFit2.h
• PndRecoKalmanFit2.cxx