#include <PndHypIdealTrackFinderTask.h>

Inheritance diagram for PndHypIdealTrackFinderTask:

Public Member Functions
	PndHypIdealTrackFinderTask ()

virtual	~PndHypIdealTrackFinderTask ()

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)

Double_t	GetTrackDip (PndMCTrack *myTrack)

Double_t	GetTrackCurvature (PndMCTrack *myTrack)

void	Register ()

void	Reset ()

void	ProduceHits ()

	ClassDef (PndHypIdealTrackFinderTask, 2)

Private Attributes
TString	fHitBranchStrip

TString	fMcBranch

TString	fTrackBranch

TClonesArray *	fStripHitArray

TClonesArray *	fMcArray

TClonesArray *	fTrackArray

TClonesArray *	fTrackCandArray

std::map< Int_t, GFTrackCand * >	fTrackCandMap

Int_t	fVerbose

Detailed Description

Definition at line 26 of file PndHypIdealTrackFinderTask.h.

Constructor & Destructor Documentation

PndHypIdealTrackFinderTask::PndHypIdealTrackFinderTask ( )

Default constructor

Definition at line 21 of file PndHypIdealTrackFinderTask.cxx.

References fHitBranchStrip, fMcBranch, and fTrackBranch.

                                                        :
   FairTask("HYP Ideal Track Finding Task")
 {
   fHitBranchStrip = "HypHit";
    //fHitBranchPixel = "MVDHitsPixel";
   /* fClusterBranchStrip = "MVDStripClusterCand";
   //fClusterBranchPixel = "MVDClusterCand";
   fDigiBranchStrip = "MVDStripDigis"; 
   //fDigiBranchPixel = "MVDPixelDigis";*/
   
   fMcBranch = "HypPoint";
   fTrackBranch = "MCTrack";
 }

PndHypIdealTrackFinderTask::~PndHypIdealTrackFinderTask ( )

virtual

Destructor

Definition at line 40 of file PndHypIdealTrackFinderTask.cxx.

41 {

42 }

Member Function Documentation

void PndHypIdealTrackFinderTask::AddAndExpand	(	Int_t	trackID,
		Int_t	detnum,
		Int_t	iHit
	)

private

Definition at line 167 of file PndHypIdealTrackFinderTask.cxx.

References fTrackArray, fTrackCandMap, GetTrackCurvature(), GetTrackDip(), GFTrackCand::setCurv(), GFTrackCand::setDip(), and GFTrackCand::setInverted().

Referenced by Exec().

                                                                                     {
   if (fTrackCandMap[trackID] == 0){
     GFTrackCand *myTCand = new GFTrackCand();
     PndMCTrack* myMCTrack = (PndMCTrack*)fTrackArray->At(trackID);
     myTCand->setCurv(GetTrackCurvature(myMCTrack));
     myTCand->setDip(GetTrackDip(myMCTrack));
     myTCand->setInverted(false);
     //myTCand->setMCID(trackID);
     fTrackCandMap[trackID] = myTCand;
   }
   fTrackCandMap[trackID]->addHit(detnum,iHit);
 }

PndHypIdealTrackFinderTask::ClassDef	(	PndHypIdealTrackFinderTask	,
		2
	)

private

void PndHypIdealTrackFinderTask::ClearTrackCandMap ( )

private

Definition at line 212 of file PndHypIdealTrackFinderTask.cxx.

References fTrackCandMap.

Referenced by Exec().

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

void PndHypIdealTrackFinderTask::Exec ( Option_t * opt )

virtual

Virtual method Exec

Definition at line 130 of file PndHypIdealTrackFinderTask.cxx.

References AddAndExpand(), ClearTrackCandMap(), fMcArray, fStripHitArray, fTrackArray, fTrackCandArray, fTrackCandMap, fVerbose, i, and PrintResult().

 {
 // Reset output array
   if ( ! fTrackCandArray )
     Fatal("Exec", "No trackCandArray");
   fTrackCandArray->Delete();
 
   Int_t 
     nStripHits = fStripHitArray->GetEntriesFast();
 
   for (Int_t iHit = 0; iHit < nStripHits; iHit++){
   PndHypHit* myHit = (PndHypHit*)(fStripHitArray->At(iHit));
  
   /*PndHypCluster* myCluster =(PndHypCluster*)(fStripClusterArray->At(myHit->GetRefIndex()));
   PndHypDigiStrip* astripdigi = (PndHypDigiStrip*)fStripDigiArray->At(myCluster->GetDigiIndex(0));*/
 
   PndHypPoint* myPoint = (PndHypPoint*)(fMcArray->At(myHit->GetRefIndex()));
   
   AddAndExpand(myPoint->GetTrackID(),2,iHit);
   }
   
   
   if(fVerbose>0) PrintResult();
 
   Int_t i = 0;
   for (std::map<Int_t,GFTrackCand*>::const_iterator ci=fTrackCandMap.begin();
        ci != fTrackCandMap.end(); ci++){
     new((*fTrackCandArray)[i]) GFTrackCand(*(ci->second));
     i++;
   }
   ClearTrackCandMap();
   fMcArray->Delete();
   fTrackArray->Delete();
   
 }

Double_t PndHypIdealTrackFinderTask::GetTrackCurvature ( PndMCTrack * myTrack )

private

Definition at line 181 of file PndHypIdealTrackFinderTask.cxx.

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

Referenced by AddAndExpand().

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

Double_t PndHypIdealTrackFinderTask::GetTrackDip ( PndMCTrack * myTrack )

private

Definition at line 187 of file PndHypIdealTrackFinderTask.cxx.

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

Referenced by AddAndExpand().

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

InitStatus PndHypIdealTrackFinderTask::Init ( )

virtual

Definition at line 72 of file PndHypIdealTrackFinderTask.cxx.

References fHitBranchStrip, fMcArray, fMcBranch, fStripHitArray, fTrackArray, fTrackBranch, and fTrackCandArray.

 {
   
   FairRootManager* ioman = FairRootManager::Instance();
 
   if ( ! ioman ) 
     {
       std::cout << "-E- PndHypIdealTrackFinderTask::Init: "
      << "RootManager not instantiated!" << std::endl;
       return kFATAL;
     }
     
 
 // Get input array
   fStripHitArray = (TClonesArray*) ioman->GetObject(fHitBranchStrip);
   if ( !fStripHitArray){
     std::cout << "-W- PndHypIdealTrackFinderTask::Init: " << "No hitArray!" << std::endl;
     return kERROR;
   }
   
  /* fStripClusterArray = (TClonesArray*) ioman->GetObject(fClusterBranchStrip);
   if ( !fStripClusterArray){
     std::cout << "-W- PndHypIdealTrackFinderTask::Init: " << "No clusterArray!" << std::endl;
     return kERROR;
   }
 
   fStripDigiArray = (TClonesArray*) ioman->GetObject(fDigiBranchStrip);
   if ( !fStripDigiArray){
     std::cout << "-W- PndHypIdealTrackFinderTask::Init: " << "No digiArray!" << std::endl;
     return kERROR;
   }
 
 */
   fMcArray = (TClonesArray*) ioman->GetObject(fMcBranch);
   if ( !fMcArray){
     std::cout << "-W- PndHypIdealTrackFinderTask::Init: " << "No mcArray!" << std::endl;
     return kERROR;
   }
 
   fTrackArray = (TClonesArray*) ioman->GetObject(fTrackBranch);
   if ( !fTrackArray){
     std::cout << "-W- PndHypIdealTrackFinderTask::Init: " << "No trackArray!" << std::endl;
     return kERROR;
   }
   
 
   
   fTrackCandArray = new TClonesArray("GFTrackCand");
   ioman->Register("HypTrackCand", "HYP", fTrackCandArray, kTRUE);
   
   std::cout << "-I- PndHypIdealTrackFinderTask: Initialisation successfull" << std::endl;
   return kSUCCESS;
 }

void PndHypIdealTrackFinderTask::PrintResult ( )

Definition at line 193 of file PndHypIdealTrackFinderTask.cxx.

References fStripHitArray, fTrackCandMap, GFTrackCand::getHit(), GFTrackCand::getNHits(), and i.

Referenced by Exec().

 {
   std::cout << "**** TrackFinding *****" << std::endl; 
   Int_t nStripHits = fStripHitArray->GetEntriesFast();
    for (std::map<Int_t, GFTrackCand*>::const_iterator ci=fTrackCandMap.begin();
         ci != fTrackCandMap.end(); ci++){
      std::cout << "TrackID: " << ci->first << std::endl;
      GFTrackCand* trackCand = ci->second;
      for (unsigned int i = 0; i < trackCand->getNHits(); i++){
        unsigned int detId, hitId;
        trackCand->getHit(i, detId, hitId);
        PndHypHit* myHit; 
        myHit = (PndHypHit*)fStripHitArray->At(hitId);
        std::cout << "Detector no. " << detId <<": "<< *myHit;
      } 
    }
    std::cout << std::endl;
 }

void PndHypIdealTrackFinderTask::ProduceHits ( )

private

void PndHypIdealTrackFinderTask::Register ( )

private

InitStatus PndHypIdealTrackFinderTask::ReInit ( )

virtual

Definition at line 56 of file PndHypIdealTrackFinderTask.cxx.

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

void PndHypIdealTrackFinderTask::Reset ( )

private

void PndHypIdealTrackFinderTask::SetParContainers ( )

virtual

Virtual method Init

Definition at line 46 of file PndHypIdealTrackFinderTask.cxx.

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