PndRecoKalmanFit Class Reference

#include <PndRecoKalmanFit.h>

Inheritance diagram for PndRecoKalmanFit:

Public Member Functions
	PndRecoKalmanFit ()
	~PndRecoKalmanFit ()
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	SetTrackRep (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)
GFRecoHitFactory *	GetRecoHitFactory ()

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

Private Attributes
GFRecoHitFactory *	fTheRecoHitFactory
GFKalman	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...
Short_t	fTrackRep
	Number of iterations. More...
Int_t	fVerbose
	(0) GeaneTrackRep, 1 RKTrackRep More...

Detailed Description

Definition at line 31 of file PndRecoKalmanFit.h.

Constructor & Destructor Documentation

PndRecoKalmanFit::PndRecoKalmanFit

(

)

Definition at line 70 of file PndRecoKalmanFit.cxx.

References PndGeoHandling::Instance().

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

PndRecoKalmanFit::~PndRecoKalmanFit

(

)

Definition at line 205 of file PndRecoKalmanFit.cxx.

References fPro.

                                    {
        delete(fPro);
}

Member Function Documentation

PndRecoKalmanFit::ClassDef ( PndRecoKalmanFit  , 0    )

private

Verbose level.

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

Definition at line 209 of file PndRecoKalmanFit.cxx.

References GFTrack::addHitVector(), Bool_t, GFRecoHitFactory::createMany(), f, fabs(), fGenFitter, fPerpPlane, fPro, fPropagateDistance, fPropagateToIP, fTheRecoHitFactory, fTrackRep, fVerbose, GenfitTrack2PndTrack(), GFTrack::getCand(), GFTrack::getNumHits(), PndTrack::GetParamFirst(), PndTrack::GetTrackCand(), par, PndTrackCand2GenfitTrackCand(), GFKalman::processTrack(), GFTrack::setCandidate(), PndTrack::SetFlag(), GeaneTrackRep::setPropDir(), trk, and GFException::what().

Referenced by PndRecoMultiKalmanTask::Exec(), and PndRecoKalmanTask::Exec().

{
  PndTrack* tAfter = NULL;
  if (fVerbose>0) std::cout<<"PndRecoKalmanFit::Fit"<<std::endl;
  if (fabs(tBefore->GetParamFirst().GetPz())<1e-9) 
    {
      tAfter = tBefore;
      tAfter->SetFlag(-10);
      return tAfter; // flag -10 : pz==0
    }
  
  Int_t  fCharge= tBefore->GetParamFirst().GetQ();
  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());
  TVector3 StartPosErr(tBefore->GetParamFirst().GetDX(),tBefore->GetParamFirst().GetDY(),tBefore->GetParamFirst().GetDZ()); 
  TVector3 StartMomErr(tBefore->GetParamFirst().GetDPx(),tBefore->GetParamFirst().GetDPy(),tBefore->GetParamFirst().GetDPz());
  
  GFAbsTrackRep* rep = 0;

  if (fPropagateToIP)
    { 
      // Calculating params at PCA to Origin
      FairTrackParP par = tBefore->GetParamFirst();
      Int_t ierr = 0;
      FairTrackParH helix(&par, ierr);
      FairGeanePro fPro0;
      if (fVerbose==0) fPro0.SetPrintErrors(kFALSE);
      FairTrackParH fRes;
      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());
          StartPosErr.SetXYZ(fRes.GetDX(), fRes.GetDY(), fRes.GetDZ());
          StartMomErr.SetXYZ(fRes.GetDPx(), fRes.GetDPy(), 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(&par, ierr);
      FairGeanePro fPro0;
      if (fVerbose==0) fPro0.SetPrintErrors(kFALSE);
      FairTrackParH fRes;
      fPro0.PropagateToLength(-fPropagateDistance);
      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());
          StartPosErr.SetXYZ(fRes.GetDX(), fRes.GetDY(), fRes.GetDZ());
          StartMomErr.SetXYZ(fRes.GetDPx(), fRes.GetDPy(), fRes.GetDPz());
        }
    }
  
  TVector3 plane_v1, plane_v2;
  if (fPerpPlane)
    {
      plane_v1 = StartMom.Orthogonal();
      plane_v2 = StartMom.Cross(plane_v1);
    }
  else
    {
      plane_v1.SetXYZ(1.,0.,0.);
      plane_v2.SetXYZ(0.,1.,0.);
    }
  GFDetPlane start_pl(StartPos, plane_v1, plane_v2);
  GFTrack* trk;
  if (fTrackRep==0)
    {
      GeaneTrackRep *grep = new GeaneTrackRep(fPro,
                                           start_pl,StartMom,
                                           StartPosErr,StartMomErr,
                                           fCharge,PDGCode);
      grep->setPropDir(1);
      rep = grep;
    }
  else if (fTrackRep==1)
    {
      RKTrackRep *grep = new RKTrackRep(StartPos, StartMom,
                                           StartPosErr, StartMomErr,
                                           PDGCode);
      rep = grep;
    }
  else
    {
      std::cout << "*** PndRecoKalmanFit::Fit" << "\t" << "Not existing Track Representation " << fTrackRep << std::endl;
      return NULL; // any smarted ideas?
    }

  trk= new GFTrack(rep);

  PndTrackCand trackCand = tBefore->GetTrackCand();
  trk->setCandidate(*PndTrackCand2GenfitTrackCand(&trackCand));
  
  // Load RecoHits
  try 
    {
      trk->addHitVector(fTheRecoHitFactory->createMany(trk->getCand()));
      if (fVerbose>0) std::cout<<trk->getNumHits()<<" hits in track " << std::endl;
    }
  catch(GFException& e)
    {
      std::cout << "*** PndRecoKalmanFit::Fit" << "\t" << "Genfit Exception: trk->addHitVector " << e.what() << std::endl;
      //throw e;
    }
  // Start Fitter
  try
    {
      fGenFitter.processTrack(trk);
    }
  catch (GFException e)
    {
      std::cout<<"*** PndRecoKalmanFit::Fit" << "\t" << "FITTER EXCEPTION ***"<<std::endl;
      std::cout<<e.what()<<std::endl;
    }
  if (fVerbose>0) std::cout<<"*** PndRecoKalmanFit::Fit" << "\t" << "SUCCESSFULL FIT!"<<std::endl;
  
  try
    { 
      tAfter = (PndTrack*)GenfitTrack2PndTrack(trk);
    }
  catch (GFException e)
    {
      std::cout<<"*** PndRecoKalmanFit::Fit" << "\t" << "CONVERSION EXCEPTION ***"<<std::endl;
      std::cout<<e.what()<<std::endl;
      tAfter = tBefore;
      tAfter->SetFlag(-2); // flag -2: conversion failed
    } 

  if (fVerbose>0) std::cout<<"Fitting done"<<std::endl;

  delete(trk);

  return tAfter;
}

GFRecoHitFactory* PndRecoKalmanFit::GetRecoHitFactory ( )

inline

Definition at line 53 of file PndRecoKalmanFit.h.

References fTheRecoHitFactory.

{ return fTheRecoHitFactory;};

Bool_t PndRecoKalmanFit::Init

(

)

Definition at line 77 of file PndRecoKalmanFit.cxx.

References GFRecoHitFactory::addProducer(), fCentralTrackerBranchName, fGenFitter, PndFtsMapCreator::FillTubeArray(), PndSttMapCreator::FillTubeArray(), fMvdBranchName, fNumIt, fPro, fTheRecoHitFactory, fTrackRep, fUseGeane, fVerbose, PndGeoSttPar::GetGeometryType(), PndGeoFtsPar::GetGeometryType(), GFFieldManager::getInstance(), GFFieldManager::init(), GFException::quiet(), rtdb, and GFKalman::setNumIterations().

Referenced by PndRecoMultiKalmanTask::Init(), PndRecoKalmanTask::Init(), and PndPidCorrelator::Init().

{
  //Get ROOT Manager
  FairRootManager* ioman= FairRootManager::Instance();
  if(ioman==0)
    {
      Error("PndRecoKalmanFit::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(sttParameters);
    tubeArray = mapper.FillTubeArray();
  }
  // FTS map loading
  PndGeoFtsPar *ftsParameters = (PndGeoFtsPar*) rtdb->getContainer("PndGeoFtsPar");
  TClonesArray *ftsTubeArray = NULL;
  if(ftsParameters->GetGeometryType() != -1) {
    PndFtsMapCreator ftsMapper(ftsParameters);
    ftsTubeArray = ftsMapper.FillTubeArray();
  }


  // Build hit factory -----------------------------
  fTheRecoHitFactory = new GFRecoHitFactory();
  if (fVerbose<2) GFException::quiet(true);
  
  TClonesArray* stripar;  TClonesArray* pixelar;
  if (fMvdBranchName == "Mix")
    {
      stripar=(TClonesArray*) ioman->GetObject("MVDHitsStripMix");
      if(stripar!=0)
        {
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MVDHitsStripMix"),new GFRecoHitProducer<PndSdsHit,PndSdsRecoHit>(stripar));
          std::cout << "*** PndRecoKalmanFit::Init" << "\t" << "MVDHitsStripMix array  found" << std::endl;
        }
      
      pixelar=(TClonesArray*) ioman->GetObject("MVDHitsPixelMix");
      if(pixelar!=0)
        { 
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MVDHitsPixelMix"),new GFRecoHitProducer<PndSdsHit,PndSdsRecoHit>(pixelar));
          std::cout << "*** PndRecoKalmanFit::Init" << "\t" << "MVDHitsPixelMix array  found" << std::endl;
        }
    }
  else
    {
      stripar=(TClonesArray*) ioman->GetObject("MVDHitsStrip");
      if(stripar!=0)
        {
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MVDHitsStrip"),new GFRecoHitProducer<PndSdsHit,PndSdsRecoHit>(stripar));
          std::cout << "*** PndRecoKalmanFit::Init" << "\t" << "MVDHitsStrip array  found" << std::endl;
        }
      
      pixelar=(TClonesArray*) ioman->GetObject("MVDHitsPixel");
      if(pixelar!=0)
        { 
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MVDHitsPixel"),new GFRecoHitProducer<PndSdsHit,PndSdsRecoHit>(pixelar));
          std::cout << "*** PndRecoKalmanFit::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 PndSttRecoHitProducer<PndSttHit,PndSttRecoHit>(sthit, tubeArray));
          std::cout << "*** PndRecoKalmanFit::Init" << "\t" << "STTHitMix array  found" << std::endl;
        }
    }
  else
    {
      sthit=(TClonesArray*) ioman->GetObject("STTHit");
      if(sthit!=0)
        {
          fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("STTHit"),new PndSttRecoHitProducer<PndSttHit,PndSttRecoHit>(sthit, tubeArray));
          std::cout << "*** PndRecoKalmanFit::Init" << "\t" << "SttHit array  found" << std::endl;
        }
    }
  
  TClonesArray* gemar=(TClonesArray*) ioman->GetObject("GEMHit");
  if(gemar!=0)
    {
      fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("GEMHit"),new GFRecoHitProducer<PndGemHit,PndGemRecoHit>(gemar));
      std::cout << "*** PndRecoKalmanFit::Init" << "\t" << "GEMHit array  found" << std::endl;
    }
  
  TClonesArray* mdtar=(TClonesArray*) ioman->GetObject("MdtHit");
  if(mdtar!=0)
    {
      fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("MdtHit"),new GFRecoHitProducer<PndMdtHit,PndMdtRecoHit>(mdtar));
      std::cout << "*** PndRecoKalmanFit::Init" << "\t" << "MdtHit array  found" << std::endl;
    }

  TClonesArray* ftsar=(TClonesArray*) ioman->GetObject("FTSHit");
  if(ftsar!=0)
    {
      fTheRecoHitFactory->addProducer(FairRootManager::Instance()->GetBranchId("FTSHit"),new PndFtsRecoHitProducer<PndFtsHit,PndFtsRecoHit>(ftsar, ftsTubeArray));
      std::cout << "*** PndRecoKalmanFit::Init" << "\t" << "FtsHit array  found" << std::endl;
    }
 
  if (fUseGeane)
    {
      fPro = new FairGeanePro();
      if (fVerbose==0) fPro->SetPrintErrors(kFALSE);
    }
  else
    {
      Error("PndRecoKalmanFit::Init","Only GEANE Propagatio available!!!");
      return kFALSE;
    }
  
  fGenFitter.setNumIterations(fNumIt);
  
  if (fTrackRep==1) GFFieldManager::getInstance()->init(new PndGenfitField());

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

  return kTRUE;
}

void PndRecoKalmanFit::SetCentralTrackerBranchName ( const TString &  name )

inline

Definition at line 48 of file PndRecoKalmanFit.h.

References fCentralTrackerBranchName, and name.

Referenced by PndRecoMultiKalmanTask::Init(), and PndRecoKalmanTask::Init().

{ fCentralTrackerBranchName = name; }

void PndRecoKalmanFit::SetGeane ( Bool_t  opt = kTRUE )

inline

Definition at line 40 of file PndRecoKalmanFit.h.

References fUseGeane.

Referenced by PndRecoMultiKalmanTask::Init(), PndRecoKalmanTask::Init(), and PndPidCorrelator::Init().

{ fUseGeane = opt;     }

void PndRecoKalmanFit::SetMvdBranchName ( const TString &  name )

inline

Definition at line 47 of file PndRecoKalmanFit.h.

References fMvdBranchName, and name.

Referenced by PndRecoMultiKalmanTask::Init(), and PndRecoKalmanTask::Init().

{ fMvdBranchName = name; }

void PndRecoKalmanFit::SetNumIterations ( Int_t  num )

inline

Definition at line 44 of file PndRecoKalmanFit.h.

References fNumIt, and num.

Referenced by PndRecoMultiKalmanTask::Init(), PndRecoKalmanTask::Init(), and PndPidCorrelator::Init().

{ fNumIt    = num;     } 

void PndRecoKalmanFit::SetPerpPlane ( Bool_t  opt = kTRUE )

inline

Definition at line 43 of file PndRecoKalmanFit.h.

References fPerpPlane.

Referenced by PndRecoMultiKalmanTask::Init(), and PndRecoKalmanTask::Init().

{ fPerpPlane = opt;    }

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

inline

Definition at line 42 of file PndRecoKalmanFit.h.

References fPropagateDistance.

Referenced by PndRecoMultiKalmanTask::Init(), and PndRecoKalmanTask::Init().

{ fPropagateDistance = opt;}

void PndRecoKalmanFit::SetPropagateToIP ( Bool_t  opt = kTRUE )

inline

Definition at line 41 of file PndRecoKalmanFit.h.

References fPropagateToIP.

Referenced by PndRecoMultiKalmanTask::Init(), and PndRecoKalmanTask::Init().

{ fPropagateToIP = opt;} 

void PndRecoKalmanFit::SetTrackRep ( Int_t  num )

inline

Definition at line 45 of file PndRecoKalmanFit.h.

References fTrackRep, and num.

Referenced by PndRecoMultiKalmanTask::Init(), and PndRecoKalmanTask::Init().

{ fTrackRep    = num;  } 

void PndRecoKalmanFit::SetVerbose ( Int_t  verb )

inline

Definition at line 46 of file PndRecoKalmanFit.h.

References fVerbose.

Referenced by PndRecoMultiKalmanTask::Init(), and PndRecoKalmanTask::Init().

{ fVerbose  = verb;    }

Member Data Documentation

TString PndRecoKalmanFit::fCentralTrackerBranchName

private

Name of the TCA for MVD.

Definition at line 64 of file PndRecoKalmanFit.h.

Referenced by Init(), and SetCentralTrackerBranchName().

GFKalman PndRecoKalmanFit::fGenFitter

private

Definition at line 59 of file PndRecoKalmanFit.h.

Referenced by Fit(), and Init().

TString PndRecoKalmanFit::fMvdBranchName

private

Geane Propagator.

Definition at line 63 of file PndRecoKalmanFit.h.

Referenced by Init(), and SetMvdBranchName().

Int_t PndRecoKalmanFit::fNumIt

private

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

Definition at line 70 of file PndRecoKalmanFit.h.

Referenced by Init(), and SetNumIterations().

Bool_t PndRecoKalmanFit::fPerpPlane

private

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

Definition at line 69 of file PndRecoKalmanFit.h.

Referenced by Fit(), and SetPerpPlane().

FairGeanePro* PndRecoKalmanFit::fPro

private

Definition at line 61 of file PndRecoKalmanFit.h.

Referenced by Fit(), Init(), and ~PndRecoKalmanFit().

Float_t PndRecoKalmanFit::fPropagateDistance

private

Flag to propagate to the interaction point.

Definition at line 68 of file PndRecoKalmanFit.h.

Referenced by Fit(), and SetPropagateDistance().

Bool_t PndRecoKalmanFit::fPropagateToIP

private

Flag to use Geane.

Definition at line 67 of file PndRecoKalmanFit.h.

Referenced by Fit(), and SetPropagateToIP().

GFRecoHitFactory* PndRecoKalmanFit::fTheRecoHitFactory

private

Definition at line 53 of file PndRecoKalmanFit.h.

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

Short_t PndRecoKalmanFit::fTrackRep

private

Number of iterations.

Definition at line 71 of file PndRecoKalmanFit.h.

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

Bool_t PndRecoKalmanFit::fUseGeane

private

Name of the TCA for central tracker.

Definition at line 66 of file PndRecoKalmanFit.h.

Referenced by Init(), and SetGeane().

Int_t PndRecoKalmanFit::fVerbose

private

(0) GeaneTrackRep, 1 RKTrackRep

Definition at line 72 of file PndRecoKalmanFit.h.

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

---

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

• PndRecoKalmanFit.h
• PndRecoKalmanFit.cxx