#include <PndMvdIdealTrackFinderTask.h>

Inheritance diagram for PndMvdIdealTrackFinderTask:

Public Member Functions
	PndMvdIdealTrackFinderTask ()

virtual	~PndMvdIdealTrackFinderTask ()

	PndMvdIdealTrackFinderTask (const PndMvdIdealTrackFinderTask &)=delete

PndMvdIdealTrackFinderTask &	operator= (const PndMvdIdealTrackFinderTask &)=delete

virtual void	SetParContainers ()

virtual InitStatus	Init ()

virtual InitStatus	ReInit ()

virtual void	Exec (Option_t *opt)

void	PrintResult ()

void	SetVerbose (Int_t verbose)

Private Member Functions
void	ClearTrackCandMap ()

void	AddAndExpand (Int_t trackID, Int_t detnum, Int_t iHit, PndSdsHit *theHit)

Double_t	GetTrackDip (PndMCTrack *myTrack)

Double_t	GetTrackCurvature (PndMCTrack *myTrack)

void	Register ()

void	Reset ()

void	ProduceHits ()

	ClassDef (PndMvdIdealTrackFinderTask, 2)

Private Attributes
TString	fHitBranchStrip

TString	fHitBranchPixel

TString	fClusterBranchStrip

TString	fClusterBranchPixel

TString	fDigiBranchStrip

TString	fDigiBranchPixel

TString	fMcBranch

TString	fTrackBranch

TClonesArray *	fStripHitArray

TClonesArray *	fPixelHitArray

TClonesArray *	fStripClusterArray

TClonesArray *	fPixelClusterArray

TClonesArray *	fStripDigiArray

TClonesArray *	fPixelDigiArray

TClonesArray *	fMcArray

TClonesArray *	fTrackArray

TClonesArray *	fTrackCandArray

std::map< Int_t, PndTrackCand * >	fTrackCandMap

Detailed Description

Definition at line 27 of file PndMvdIdealTrackFinderTask.h.

Constructor & Destructor Documentation

PndMvdIdealTrackFinderTask::PndMvdIdealTrackFinderTask ( )

Default constructor

Definition at line 22 of file PndMvdIdealTrackFinderTask.cxx.

                                                        :
   FairTask("MVD Ideal Track Finding Task"),
       fHitBranchStrip("MVDHitsStrip"),
       fHitBranchPixel("MVDHitsPixel"),
       fClusterBranchStrip("MVDStripClusterCand"),
       fClusterBranchPixel("MVDPixelClusterCand"),
       fDigiBranchStrip("MVDStripDigis"),
       fDigiBranchPixel("MVDPixelDigis"),
       fMcBranch("MVDPoint"),
       fTrackBranch("MCTrack"),
       fStripHitArray(NULL),
       fPixelHitArray(NULL),
       fStripClusterArray(NULL),
       fPixelClusterArray(NULL),
       fStripDigiArray(NULL),
       fPixelDigiArray(NULL),
       fMcArray(NULL),
       fTrackArray(NULL),
       fTrackCandArray(NULL),
       fTrackCandMap()
 {
 }

PndMvdIdealTrackFinderTask::~PndMvdIdealTrackFinderTask ( )

virtual

Destructor

Definition at line 50 of file PndMvdIdealTrackFinderTask.cxx.

51 {

52 }

PndMvdIdealTrackFinderTask::PndMvdIdealTrackFinderTask ( const PndMvdIdealTrackFinderTask & )

delete

Member Function Documentation

void PndMvdIdealTrackFinderTask::AddAndExpand	(	Int_t	trackID,
		Int_t	detnum,
		Int_t	iHit,
		PndSdsHit *	theHit
	)

private

Definition at line 198 of file PndMvdIdealTrackFinderTask.cxx.

References fTrackCandMap, PndSdsHit::GetPosition(), and PndTrackCand::setMcTrackId().

Referenced by Exec().

                                                                                                       {
   if (fTrackCandMap[trackID] == 0){
     PndTrackCand *myTCand = new PndTrackCand();
     //PndMCTrack* myMCTrack = (PndMCTrack*)fTrackArray->At(trackID); //[R.K.03/2017] unused variable
 //    myTCand->setCurv(GetTrackCurvature(myMCTrack));
 //    myTCand->setDip(GetTrackDip(myMCTrack));
 //    myTCand->setInverted(false);
     //int pdg = myMCTrack->GetPdgCode(); //[R.K.02/2017] Unused variable?
     //double charge; //[R.K.02/2017] Unused variable?
     //if(pdg<100000000){ //[R.K.02/2017] Unused variable?
       //charge =  TDatabasePDG::Instance()->GetParticle(pdg)->Charge()/3.; //[R.K.02/2017] Unused variable?
     //} //[R.K.02/2017] Unused variable?
     //else{ //[R.K.02/2017] Unused variable?
       //charge = 0.; //[R.K.02/2017] Unused variable?
     //} //[R.K.02/2017] Unused variable?
 //    myTCand->setTrackSeed(
 //                        myMCTrack->GetStartVertex(),
 //                        myMCTrack->GetMomentum(),
 //                        charge/myMCTrack->GetMomentum().Mag()
 //                        );
     myTCand->setMcTrackId(trackID);
     fTrackCandMap[trackID] = myTCand;
   }
 
   fTrackCandMap[trackID]->AddHit(detnum,iHit,theHit->GetPosition().Mag());
 }

PndMvdIdealTrackFinderTask::ClassDef	(	PndMvdIdealTrackFinderTask	,
		2
	)

private

void PndMvdIdealTrackFinderTask::ClearTrackCandMap ( )

private

Definition at line 258 of file PndMvdIdealTrackFinderTask.cxx.

References fTrackCandMap.

Referenced by Exec().

 {
   for (std::map<Int_t,PndTrackCand*>::const_iterator kIt=fTrackCandMap.begin();
           kIt != fTrackCandMap.end(); kIt++){
     delete(kIt->second);
   }
   fTrackCandMap.clear();
 }

void PndMvdIdealTrackFinderTask::Exec ( Option_t * opt )

virtual

Virtual method Exec

Definition at line 154 of file PndMvdIdealTrackFinderTask.cxx.

References AddAndExpand(), ClearTrackCandMap(), Double_t, fMcArray, fPixelClusterArray, fPixelDigiArray, fPixelHitArray, fStripClusterArray, fStripDigiArray, fStripHitArray, fTrackCandArray, fTrackCandMap, fVerbose, PndSdsHit::GetClusterIndex(), PndSdsDigi::GetDetID(), PndSdsCluster::GetDigiIndex(), PndSdsDigi::GetIndex(), i, and PrintResult().

 {
 
   // Reset output array
   if ( ! fTrackCandArray )
     Fatal("Exec", "No trackCandArray");
   fTrackCandArray->Delete();
 
   Int_t nStripHits = fStripHitArray->GetEntriesFast();
   Int_t nPixelHits = fPixelHitArray->GetEntriesFast();
 //TODO: Thi becomes easier now, with the FairHit::GetRefIndex()
   //pixel part
   for (Int_t iHit = 0; iHit < nPixelHits; iHit++){
     PndSdsHit* myHit = (PndSdsHit*)(fPixelHitArray->At(iHit));
     PndSdsCluster* myCluster = (PndSdsCluster*)(fPixelClusterArray->At(myHit->GetClusterIndex()));
     PndSdsDigiPixel* apixeldigi = (PndSdsDigiPixel*)fPixelDigiArray->At(myCluster->GetDigiIndex(0));
     if (apixeldigi->GetIndex(0) == -1) continue; // sort out noise
     PndSdsMCPoint* myPoint = (PndSdsMCPoint*)(fMcArray->At(apixeldigi->GetIndex(0)));
 
     AddAndExpand(myPoint->GetTrackID(),apixeldigi->GetDetID(),iHit, myHit);
 
   }
   //strip part
   for (Int_t iHit = 0; iHit < nStripHits; iHit++){
     PndSdsHit* myHit = (PndSdsHit*)(fStripHitArray->At(iHit));
     PndSdsCluster* myCluster =  (PndSdsCluster*)(fStripClusterArray->At(myHit->GetClusterIndex()));
     PndSdsDigiStrip* astripdigi = (PndSdsDigiStrip*)fStripDigiArray->At(myCluster->GetDigiIndex(0));
     if (astripdigi->GetIndex(0) == -1) continue; // sort out noise
     PndSdsMCPoint* myPoint = (PndSdsMCPoint*)(fMcArray->At(astripdigi->GetIndex(0)));
 
     AddAndExpand(myPoint->GetTrackID(),astripdigi->GetDetID(),iHit, myHit);
   }
 
   if(fVerbose>0) PrintResult();
 
   Int_t i = 0;
   for (std::map<Int_t,PndTrackCand*>::const_iterator kIt=fTrackCandMap.begin(); kIt != fTrackCandMap.end(); kIt++){
           Double_t eventTime = FairRootManager::Instance()->GetEventTime();
           kIt->second->SetTimeStamp(eventTime);
           new((*fTrackCandArray)[i]) PndTrackCand(*(kIt->second));
   }
   ClearTrackCandMap();
 }

Double_t PndMvdIdealTrackFinderTask::GetTrackCurvature ( PndMCTrack * myTrack )

private

Definition at line 225 of file PndMvdIdealTrackFinderTask.cxx.

References PndMCTrack::GetMomentum(), p, and CAMath::Sqrt().

 {
   TVector3 p = myTrack->GetMomentum();
   return (2/TMath::Sqrt(p.Px()*p.Px() + p.Py()*p.Py()));
 }

Double_t PndMvdIdealTrackFinderTask::GetTrackDip ( PndMCTrack * myTrack )

private

Definition at line 231 of file PndMvdIdealTrackFinderTask.cxx.

References PndMCTrack::GetMomentum(), p, and CAMath::Sqrt().

 {
   TVector3 p= myTrack->GetMomentum();
   return (p.Mag()/TMath::Sqrt(p.Px()*p.Px() + p.Py()*p.Py()));
 }

InitStatus PndMvdIdealTrackFinderTask::Init ( )

virtual

Definition at line 82 of file PndMvdIdealTrackFinderTask.cxx.

References fClusterBranchPixel, fClusterBranchStrip, fDigiBranchPixel, fDigiBranchStrip, fHitBranchPixel, fHitBranchStrip, fMcArray, fMcBranch, fPixelClusterArray, fPixelDigiArray, fPixelHitArray, fStripClusterArray, fStripDigiArray, fStripHitArray, fTrackArray, fTrackBranch, and fTrackCandArray.

 {
 
   FairRootManager* ioman = FairRootManager::Instance();
 
   if ( ! ioman )
     {
       std::cout << "-E- PndMvdIdealTrackFinderTask::Init: "
      << "RootManager not instantiated!" << std::endl;
       return kFATAL;
     }
 
   // Get input array
   fStripHitArray = (TClonesArray*) ioman->GetObject(fHitBranchStrip);
   if ( !fStripHitArray){
     std::cout << "-W- PndMvdIdealTrackFinderTask::Init: " << "No fStripHitArray!" << std::endl;
     return kERROR;
   }
 
   fPixelHitArray = (TClonesArray*) ioman->GetObject(fHitBranchPixel);
   if ( !fPixelHitArray){
     std::cout << "-W- PndMvdIdealTrackFinderTask::Init: " << "No fPixelHitArray!" << std::endl;
     return kERROR;
   }
 
   fStripClusterArray = (TClonesArray*) ioman->GetObject(fClusterBranchStrip);
   if ( !fStripClusterArray){
     std::cout << "-W- PndMvdIdealTrackFinderTask::Init: " << "No StripclusterArray!" << std::endl;
     return kERROR;
   }
   fPixelClusterArray = (TClonesArray*) ioman->GetObject(fClusterBranchPixel);
   if ( !fPixelClusterArray){
     std::cout << "-W- PndMvdIdealTrackFinderTask::Init: " << "No PixelclusterArray!" << std::endl;
     return kERROR;
   }
 
 
   fStripDigiArray = (TClonesArray*) ioman->GetObject(fDigiBranchStrip);
   if ( !fStripDigiArray){
     std::cout << "-W- PndMvdIdealTrackFinderTask::Init: " << "No StripdigiArray!" << std::endl;
     return kERROR;
   }
 
   fPixelDigiArray = (TClonesArray*) ioman->GetObject(fDigiBranchPixel);
   if ( !fPixelDigiArray){
     std::cout << "-W- PndMvdIdealTrackFinderTask::Init: " << "No PixeldigiArray!" << std::endl;
     return kERROR;
   }
 
   fMcArray = (TClonesArray*) ioman->GetObject(fMcBranch);
   if ( !fMcArray){
     std::cout << "-W- PndMvdIdealTrackFinderTask::Init: " << "No mcArray!" << std::endl;
     return kERROR;
   }
 
   fTrackArray = (TClonesArray*) ioman->GetObject(fTrackBranch);
   if ( !fTrackArray){
     std::cout << "-W- PndMvdIdealTrackFinderTask::Init: " << "No trackArray!" << std::endl;
     return kERROR;
   }
 
   fTrackCandArray = new TClonesArray("PndTrackCand");
   ioman->Register("MVDIdealTrackCand", "MVD", fTrackCandArray, kTRUE);
 
   std::cout << "-I- PndMvdIdealTrackFinderTask: Initialisation successfull" << std::endl;
   return kSUCCESS;
 }

PndMvdIdealTrackFinderTask& PndMvdIdealTrackFinderTask::operator= ( const PndMvdIdealTrackFinderTask & )

delete

void PndMvdIdealTrackFinderTask::PrintResult ( )

Definition at line 237 of file PndMvdIdealTrackFinderTask.cxx.

References fPixelHitArray, fStripHitArray, fTrackCandMap, PndTrackCandHit::GetDetId(), PndTrackCandHit::GetHitId(), PndTrackCand::GetNHits(), PndTrackCand::GetSortedHit(), and i.

Referenced by Exec().

 {
   std::cout << "**** TrackFinding *****" << std::endl;
   Int_t nStripHits = fStripHitArray->GetEntriesFast();
    for (std::map<Int_t, PndTrackCand*>::const_iterator kIt=fTrackCandMap.begin();
         kIt != fTrackCandMap.end(); kIt++){
      std::cout << "TrackID: " << kIt->first << std::endl;
      PndTrackCand* trackCand = kIt->second;
      unsigned int detId, hitId;
      for (unsigned int i = 0; i < trackCand->GetNHits(); i++){
        detId=trackCand->GetSortedHit(i).GetDetId();
        hitId=trackCand->GetSortedHit(i).GetHitId();
       PndSdsHit* myHit;
       if(hitId<(UInt_t)nStripHits) myHit = (PndSdsHit*)(fStripHitArray->At(hitId));
       else myHit = (PndSdsHit*)(fPixelHitArray->At(hitId - nStripHits));
       std::cout << "Detector no. " << detId <<": "<< *myHit;
      }
    }
    std::cout << std::endl;
 }

void PndMvdIdealTrackFinderTask::ProduceHits ( )

private

void PndMvdIdealTrackFinderTask::Register ( )

private

InitStatus PndMvdIdealTrackFinderTask::ReInit ( )

virtual

Definition at line 66 of file PndMvdIdealTrackFinderTask.cxx.

 {
 
   InitStatus stat=kSUCCESS; 
   return stat;
 
   /*
   FairRun* ana = FairRun::Instance();
   FairRuntimeDb* rtdb=ana->GetRuntimeDb();
   fGeoPar=(PndMvdGeoPar*)(rtdb->getContainer("PndMvdGeoPar"));
 
   return kSUCCESS;
   */
 }

void PndMvdIdealTrackFinderTask::Reset ( )

private

void PndMvdIdealTrackFinderTask::SetParContainers ( )

virtual

Virtual method Init

Definition at line 56 of file PndMvdIdealTrackFinderTask.cxx.

 {
   // Get Base Container
 /*
   FairRun* ana = FairRun::Instance();
   FairRuntimeDb* rtdb=ana->GetRuntimeDb();
   fGeoPar = (PndMvdGeoPar*)(rtdb->getContainer("PndMvdGeoPar"));
 */
 }