#include <PndHypGeIdealTrackFinderTask.h>

Inheritance diagram for PndHypGeIdealTrackFinderTask:

Public Member Functions
	PndHypGeIdealTrackFinderTask ()

virtual	~PndHypGeIdealTrackFinderTask ()

virtual void	SetParContainers ()

virtual InitStatus	Init ()

virtual InitStatus	ReInit ()

virtual void	Exec (Option_t *opt)

void	WriteHistograms ()

Private Member Functions
	ClassDef (PndHypGeIdealTrackFinderTask, 1)

Private Attributes
TString	fMcBranch

TString	fTrackBranch

TClonesArray *	fMcArray

TClonesArray *	fTrackArray

TH1F *	hismom

int	fEventNr

std::map< int, std::vector< int > >	mcHitMap

Int_t	fVerbose

Detailed Description

Definition at line 26 of file PndHypGeIdealTrackFinderTask.h.

Constructor & Destructor Documentation

PndHypGeIdealTrackFinderTask::PndHypGeIdealTrackFinderTask ( )

Default constructor

Definition at line 22 of file PndHypGeIdealTrackFinderTask.cxx.

References fMcBranch, and fTrackBranch.

                                                            :
   FairTask("HYP Ideal Track Finding Task"), fEventNr(0)
 {
  
 
   fMcBranch = "HypGePoint";
   fTrackBranch = "MCTrack";
 }

PndHypGeIdealTrackFinderTask::~PndHypGeIdealTrackFinderTask ( )

virtual

Destructor

Definition at line 36 of file PndHypGeIdealTrackFinderTask.cxx.

References hismom.

 { 
   delete(hismom);
 }

Member Function Documentation

PndHypGeIdealTrackFinderTask::ClassDef	(	PndHypGeIdealTrackFinderTask	,
		1
	)

private

void PndHypGeIdealTrackFinderTask::Exec ( Option_t * opt )

virtual

Virtual method Exec

Definition at line 109 of file PndHypGeIdealTrackFinderTask.cxx.

References Double_t, fEventNr, fMcArray, fTrackArray, PndMCTrack::GetMotherID(), PndMCTrack::GetPdgCode(), PndHypGePoint::GetTrackID(), hismom, i, map, mcHitMap, mom, and PndHypGePoint::Momentum().

 {
   // Reset output array
   // if ( ! fTrackCandArray )
   //   Fatal("Exec", "No trackCandArray");
   // fTrackCandArray->Delete();
   
   
   
   //mcHitMap = AssignHitsToTracks();
    for (int i = 0; i < fMcArray->GetEntriesFast(); i++){                                                                                        //get all MC Hits
     PndHypGePoint* myPoint = (PndHypGePoint*)(fMcArray->At(i));                                                                 //sort MCHits with Tracks
     
     TVector3 mom;
     myPoint->Momentum(mom);
     
     
     PndMCTrack* myTrack = (PndMCTrack*)(fTrackArray->At(myPoint->GetTrackID()));
     mcHitMap[myPoint->GetTrackID()].push_back(i);
     
     
   }
   
   std::cout << "------------Event " << fEventNr << "-------------" << std::endl;
   fEventNr++;
   std::cout<<" size map "<<mcHitMap.size()<<std::endl;
   
   
   
   for (std::map<int, std::vector<int> >::const_iterator it = mcHitMap.begin(); 
        it!= mcHitMap.end(); it++) {  //go through all tracks
     
     PndMCTrack* myTrack = (PndMCTrack*)(fTrackArray->At(it->first));
     if(myTrack==0)continue;
     
     std::vector<int> MChits = it->second;
     
     
     if (MChits.size() >0){
       
       PndHypGePoint* startPoint = (PndHypGePoint*)(fMcArray->At(MChits[0]));
       if(startPoint==0)continue;
       
       TVector3 mome,momst;
       
       startPoint->Momentum(momst);
       
       
       Int_t MothId, Mothpdg;
       MothId= myTrack->GetMotherID();
       
       if  (MothId==-1){
         Mothpdg = myTrack->GetPdgCode();
         if(Mothpdg==2112){
           /*else {
             cout<<"mother id "<<MothId<<endl;
             PndMCTrack *mother =(PndMCTrack*)fMcTr->At(MothId);
             Mothpdg = mother->GetPdgCode();
             }
           */
           
           Double_t Theta;
           hismom->Fill(180/3.1416*momst.Theta());
         }
       }
     }
   }
   
   
   fMcArray->Delete();
   fTrackArray->Delete();
   
 }

InitStatus PndHypGeIdealTrackFinderTask::Init ( )

virtual

Definition at line 69 of file PndHypGeIdealTrackFinderTask.cxx.

References fMcArray, fMcBranch, fTrackArray, fTrackBranch, and hismom.

 {
   
   FairRootManager* ioman = FairRootManager::Instance();
 
   if ( ! ioman ) 
     {
       std::cout << "-E- PndHypGeIdealTrackFinderTask::Init: "
      << "RootManager not instantiated!" << std::endl;
       return kFATAL;
     }
     
 
 // Get input array
  
   fMcArray = (TClonesArray*) ioman->GetObject(fMcBranch);
   if ( !fMcArray){
     std::cout << "-W- PndHypGeIdealTrackFinderTask::Init: " << "No mcArray!" << std::endl;
     return kERROR;
   }
 
   fTrackArray = (TClonesArray*) ioman->GetObject(fTrackBranch);
   if ( !fTrackArray){
     std::cout << "-W- PndHypGeIdealTrackFinderTask::Init: " << "No trackArray!" << std::endl;
     return kERROR;
   }
   
   hismom = new TH1F("hismom","HYPGe MC Points, momentum",100,0.,180.);
   hismom->SetTitle("Kinetic energy GeV");
   
  
   
   std::cout << "-I- PndHypGeIdealTrackFinderTask: Initialisation successfull" << std::endl;
   return kSUCCESS;
 }

InitStatus PndHypGeIdealTrackFinderTask::ReInit ( )

virtual

Definition at line 53 of file PndHypGeIdealTrackFinderTask.cxx.

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

void PndHypGeIdealTrackFinderTask::SetParContainers ( )

virtual

Virtual method Init

Definition at line 43 of file PndHypGeIdealTrackFinderTask.cxx.

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

void PndHypGeIdealTrackFinderTask::WriteHistograms ( )

Definition at line 205 of file PndHypGeIdealTrackFinderTask.cxx.

References file, and hismom.

                                                   {
           TFile* file = FairRootManager::Instance()->GetOutFile();
           file->cd();
           file->mkdir("neutron");
           file->cd("neutron");
         
           hismom->Write();
           delete hismom;
           hismom=NULL; 
 }