PndGenfitAdapters2.cxx File Reference

#include "PndGenfitAdapters2.h"
#include <iostream>
#include "PndTrack.h"
#include "PndTrackCand.h"
#include "FairTrackParP.h"
#include "AbsTrackRep.h"
#include "DetPlane.h"
#include "Exception.h"
#include "RKTrackRep.h"
#include "SharedPlanePtr.h"
#include "Track.h"
#include "TrackCand.h"
#include <TMatrixD.h>
#include <TMatrixDSym.h>
#include <cmath>

Go to the source code of this file.

Functions
PndTrackCand *	Genfit2TrackCand2PndTrackCand (const genfit::TrackCand *cand)
genfit::TrackCand *	PndTrackCand2Genfit2TrackCand (PndTrackCand *cand)
PndTrack *	Genfit2Track2PndTrack (const genfit::Track *tr)

Function Documentation

PndTrack* Genfit2Track2PndTrack

(

const genfit::Track *

tr

)

Definition at line 60 of file PndGenfitAdapters2.cxx.

References genfit::Track::constructTrackCand(), count, Genfit2TrackCand2PndTrackCand(), genfit::StateOnPlane::getAuxInfo(), genfit::Track::getCardinalRep(), genfit::FitStatus::getChi2(), genfit::MeasuredStateOnPlane::getCov(), genfit::Track::getFitStatus(), genfit::Track::getFittedState(), genfit::FitStatus::getNdf(), genfit::StateOnPlane::getPlane(), genfit::StateOnPlane::getState(), i, genfit::FitStatus::isFitConverged(), PndTrack::SetChi2(), PndTrack::SetFlag(), and PndTrack::SetNDF().

Referenced by PndRecoDafFit2::Fit(), and PndRecoKalmanFit2::Fit().

                                                      {

  if (dynamic_cast<genfit::RKTrackRep*>(tr->getCardinalRep())==NULL) {
    std::cerr << " GenfitAbsTrackRep2PndTrack() can currently only handle RKTrackRep" << std::endl;
    throw;
  }

  const TVectorD& firstState = tr->getFittedState().getState();
  const TVectorD& lastState = tr->getFittedState(-1).getState();
  const TMatrixDSym& firstCov = tr->getFittedState().getCov();
  const TMatrixDSym& lastCov = tr->getFittedState(-1).getCov();
  genfit::SharedPlanePtr firstPlane = tr->getFittedState().getPlane();
  genfit::SharedPlanePtr lastPlane = tr->getFittedState(-1).getPlane();

  //make the FairTrackParP for first and last hit
  double firstCova[15];
  int count=0;
  for(int i=0; i<5;++i){
     for(int j=i;j<5;++j){
           firstCova[count++]=firstCov(i,j);
     }
  }
  double lastCova[15];
  count=0;
  for(int i=0; i<5;++i){
    for(int j=i;j<5;++j){
          lastCova[count++]=lastCov(i,j);
    }
  }

  double first_spu = (tr->getFittedState().getAuxInfo())(0);
  double last_spu = (tr->getFittedState(-1).getAuxInfo())(0);
    
  FairTrackParP first(firstState[3],firstState[4],firstState[1],firstState[2],firstState[0],firstCova,firstPlane->getO(),firstPlane->getU(),firstPlane->getV(),first_spu);
  FairTrackParP last(lastState[3],lastState[4],lastState[1],lastState[2],lastState[0],lastCova,lastPlane->getO(),lastPlane->getU(),lastPlane->getV(),last_spu);
    
  //copy the trackCand
  genfit::TrackCand* genfitCand = tr->constructTrackCand();
  PndTrackCand* pndCand = Genfit2TrackCand2PndTrackCand(genfitCand);
  PndTrack* retVal =  new PndTrack(first,last,*pndCand);
  retVal->SetChi2(tr->getFitStatus()->getChi2());
  retVal->SetNDF(tr->getFitStatus()->getNdf());
  if (tr->getFitStatus()->isFitConverged()) {
    retVal->SetFlag(1);
  }
  else {
    retVal->SetFlag(-1);
  }
  delete genfitCand;
  delete pndCand;
  return retVal;
  
}

PndTrackCand* Genfit2TrackCand2PndTrackCand

(

const genfit::TrackCand *

cand

)

Definition at line 24 of file PndGenfitAdapters2.cxx.

References PndTrackCand::AddHit(), genfit::TrackCand::getHit(), genfit::TrackCand::getMcTrackId(), genfit::TrackCand::getNHits(), i, nhits, and PndTrackCand::setMcTrackId().

Referenced by Genfit2Track2PndTrack().

{
  // Utility to convert from genfit TrackCand to PndTrackCand.
  // Since PndTrackCand does not store any seed value, those componets
  // will not be present in the final object. 

  PndTrackCand* retVal = new PndTrackCand();
  const unsigned nhits = cand->getNHits();
  int detId,hitId;
  double rho;
  for(unsigned int i=0;i<nhits;++i){
    cand->getHit(i,detId,hitId,rho);//
    retVal->AddHit(detId,hitId,rho);//
  }
  
  retVal->setMcTrackId(cand->getMcTrackId());
  return retVal;
}

genfit::TrackCand* PndTrackCand2Genfit2TrackCand

(

PndTrackCand *

cand

)

Definition at line 43 of file PndGenfitAdapters2.cxx.

References genfit::TrackCand::addHit(), PndTrackCandHit::GetDetId(), PndTrackCandHit::GetHitId(), PndTrackCand::getMcTrackId(), PndTrackCand::GetNHits(), PndTrackCandHit::GetRho(), PndTrackCand::GetSortedHit(), i, nhits, and genfit::TrackCand::setMcTrackId().

Referenced by PndRecoDafFit2::Fit(), and PndRecoKalmanFit2::Fit().

{
  // Utility to convert from PndTrackCand to genfit TrackCand.
  // PndTrackCand does not store any Seed, then to create a proper TrackCand you need
  // to fill the seed afterwards taking the values from the PndTrack 
 
  genfit::TrackCand* retVal = new genfit::TrackCand();
  unsigned int nhits = cand->GetNHits();
  for(unsigned int i=0;i<nhits;++i){
    PndTrackCandHit candHit = cand->GetSortedHit(i);
    retVal->addHit(candHit.GetDetId(),candHit.GetHitId(),i,candHit.GetRho());
  }
  
  retVal->setMcTrackId(cand->getMcTrackId());
  return retVal;
}