[u/mrichter/AliRoot.git] / FMD / AliFMDReconstruction.cxx

// --- ROOT system ---
#include "TTask.h"
#include "TTree.h"
#include "TSystem.h"
#include "TFile.h"
// --- Standard library ---

// --- AliRoot header files ---

#include "AliFMDdigit.h"
#include "AliFMDReconstParticles.h"
#include "AliFMD.h"
#include "AliFMDv1.h"
#include "AliFMDReconstruction.h"
#include "AliRun.h"
#include "AliDetector.h"

#include "TROOT.h"
#include "TFolder.h"
#include <stdlib.h>
#include <iostream.h>
#include <fstream.h>

ClassImp(AliFMDReconstruction)

        
//____________________________________________________________________________ 

AliFMDReconstruction::AliFMDReconstruction():TTask("AliFMDReconstruction","") 
{
  fNevents = 0 ;  // Number of events to rreconnstraction, 0 means all events in current file
  // add Task to //root/Tasks folder
  TTask * roottasks = (TTask*)gROOT->GetRootFolder()->FindObject("Tasks") ; 
  roottasks->Add(this) ; 
}
//____________________________________________________________________________ 

AliFMDReconstruction::AliFMDReconstruction(char* HeaderFile, char *ReconstParticlesFile):TTask("AliFMDReconstruction","")
{
  fNevents = 0 ;    // Number of events to rreconnstraction, 0 means all events in current file
  fReconstParticlesFile=ReconstParticlesFile ;
  fHeadersFile=HeaderFile ;
  //add Task to //root/Tasks folder
  TTask * roottasks = (TTask*)gROOT->GetRootFolder()->FindObject("Tasks") ; 
  roottasks->Add(this) ;     
}

//____________________________________________________________________________ 

AliFMDReconstruction::~AliFMDReconstruction()
{
 
}

//____________________________________________________________________________

void AliFMDReconstruction::Exec(TClonesArray *fReconParticles,Option_t *option) 
{ 
 //Collects all digits in the same active volume into number of particles

  AliFMD * FMD = (AliFMD *) gAlice->GetDetector("FMD");
  if(fNevents == 0) fNevents=(Int_t)gAlice->TreeD()->GetEntries(); 
  for(Int_t ievent=0;ievent<fNevents;ievent++)
    { 
      gAlice->GetEvent(ievent) ;
      if(gAlice->TreeH()==0) return; 
      if(gAlice->TreeR()==0) gAlice->MakeTree("R");
      //Make branches
      FMD->MakeBranch("R");
      Int_t threshold[]={   0,   18,  37,  56,   76, 
			    96, 119, 138, 165,  172, 
			    218, 231, 238, 277,  304, 
			    330, 357, 423, 449,  476, 
			    522, 542, 555, 568,  581, 
			    614, 624, 657, 674,  687, 
			    700, 713, 720, 727,  733, 
			    740, 759, 778, 797,  816, 
			    834, 853, 872, 891,  910, 
			    929, 948, 967, 986,  1024};
                         
      
      int threshold_array_size=sizeof(threshold)/sizeof(threshold[0]);
      AliFMDdigit  *fmdDigit;
       
      if (FMD)
	{
	  gAlice->TreeD()->GetEvent(0); 
	  TClonesArray *FMDdigits=FMD->Digits();
	  Int_t nDigits=FMDdigits->GetEntries(); 
	   Int_t RecParticles[4];
	   Int_t nRecPart=0 ;
	   for (Int_t digit=0;digit<nDigits;digit++) 
	     {
	      fmdDigit=(AliFMDdigit*)FMDdigits->UncheckedAt(digit);    
	      RecParticles[0] = fmdDigit->Volume();
	      RecParticles[1] = fmdDigit->NumberOfSector();
	      RecParticles[2] = fmdDigit->NumberOfRing();
	      Int_t ADC=fmdDigit->ADCsignal(); 
	      RecParticles[3]=0; //case when fmdDigit->ADCsignal()==0
	      for (int i=0;i<threshold_array_size-1;i++)
		    {
		      if(ADC>threshold[i]&&ADC<=threshold[i+1])
			 RecParticles[3]=i;		 
		    }
	      new((*fReconParticles)[nRecPart++]) AliFMDReconstParticles(RecParticles); 
	     } //digit loop
	 }//if FMD
       gAlice->TreeR()->Reset();
       gAlice->TreeR()->Fill(); 
        gAlice->TreeR()->Write(0,TObject::kOverwrite);
    } //event loop
  cout<<"\nAliFMDReconstruction::Exec(TClonesArray *fReconParticles,Option_t *option) finished"<<endl;
}
//__________________________________________________________________

void AliFMDReconstruction::SetReconstParticlesFile(char * file )
{
   if (!fReconstParticlesFile.IsNull()) 
    cout<<"\nChanging reconstructed particles file from "<<
      (char *) fReconstParticlesFile.Data()<< " to "<<file<<endl;
    fReconstParticlesFile=file;
}
//__________________________________________________________________
void AliFMDReconstruction::Print(Option_t* option)const
{
  cout<<"------------------- "<<GetName() <<" -------------"<< endl ;
  if(fReconstParticlesFile.IsNull())
    cout<<"\nWriting reconstructed particles to file"<<endl ;
  else
    cout<<"\nWriting reconstructed particles to file  "<<
      (char*) fReconstParticlesFile.Data() << endl ;
}
Commit	Line	Data
dc8af42e	1	// --- ROOT system ---
	2	#include "TTask.h"
	3	#include "TTree.h"
	4	#include "TSystem.h"
	5	#include "TFile.h"
	6	// --- Standard library ---
	7
	8	// --- AliRoot header files ---
	9
	10	#include "AliFMDdigit.h"
	11	#include "AliFMDReconstParticles.h"
	12	#include "AliFMD.h"
	13	#include "AliFMDv1.h"
	14	#include "AliFMDReconstruction.h"
	15	#include "AliRun.h"
	16	#include "AliDetector.h"
	17
	18	#include "TROOT.h"
	19	#include "TFolder.h"
	20	#include <stdlib.h>
	21	#include <iostream.h>
	22	#include <fstream.h>
	23
	24	ClassImp(AliFMDReconstruction)
	25
	26
	27	//____________________________________________________________________________
	28
	29	AliFMDReconstruction::AliFMDReconstruction():TTask("AliFMDReconstruction","")
	30	{
	31	fNevents = 0 ; // Number of events to rreconnstraction, 0 means all events in current file
	32	// add Task to //root/Tasks folder
	33	TTask * roottasks = (TTask*)gROOT->GetRootFolder()->FindObject("Tasks") ;
	34	roottasks->Add(this) ;
	35	}
	36	//____________________________________________________________________________
	37
	38	AliFMDReconstruction::AliFMDReconstruction(char* HeaderFile, char *ReconstParticlesFile):TTask("AliFMDReconstruction","")
	39	{
	40	fNevents = 0 ; // Number of events to rreconnstraction, 0 means all events in current file
	41	fReconstParticlesFile=ReconstParticlesFile ;
	42	fHeadersFile=HeaderFile ;
	43	//add Task to //root/Tasks folder
	44	TTask * roottasks = (TTask*)gROOT->GetRootFolder()->FindObject("Tasks") ;
	45	roottasks->Add(this) ;
	46	}
	47
	48	//____________________________________________________________________________
	49
	50	AliFMDReconstruction::~AliFMDReconstruction()
	51	{
	52
	53	}
	54
	55	//____________________________________________________________________________
	56
	57	void AliFMDReconstruction::Exec(TClonesArray fReconParticles,Option_t option)
	58	{
	59	//Collects all digits in the same active volume into number of particles
	60
	61	AliFMD * FMD = (AliFMD *) gAlice->GetDetector("FMD");
	62	if(fNevents == 0) fNevents=(Int_t)gAlice->TreeD()->GetEntries();
	63	for(Int_t ievent=0;ievent<fNevents;ievent++)
	64	{
65	gAlice->GetEvent(ievent) ;
66	if(gAlice->TreeH()==0) return;
67	if(gAlice->TreeR()==0) gAlice->MakeTree("R");
68	//Make branches
69	FMD->MakeBranch("R");
70	Int_t threshold[]={ 0, 18, 37, 56, 76,
71	96, 119, 138, 165, 172,
72	218, 231, 238, 277, 304,
73	330, 357, 423, 449, 476,
74	522, 542, 555, 568, 581,
75	614, 624, 657, 674, 687,
76	700, 713, 720, 727, 733,
77	740, 759, 778, 797, 816,
78	834, 853, 872, 891, 910,
79	929, 948, 967, 986, 1024};
80
81
82	int threshold_array_size=sizeof(threshold)/sizeof(threshold[0]);
83	AliFMDdigit *fmdDigit;
84
85	if (FMD)
86	{
87	gAlice->TreeD()->GetEvent(0);
88	TClonesArray *FMDdigits=FMD->Digits();
89	Int_t nDigits=FMDdigits->GetEntries();
90	Int_t RecParticles[4];
91	Int_t nRecPart=0 ;
92	for (Int_t digit=0;digit<nDigits;digit++)
93	{
94	fmdDigit=(AliFMDdigit*)FMDdigits->UncheckedAt(digit);
95	RecParticles[0] = fmdDigit->Volume();
96	RecParticles[1] = fmdDigit->NumberOfSector();
97	RecParticles[2] = fmdDigit->NumberOfRing();
98	Int_t ADC=fmdDigit->ADCsignal();
99	RecParticles[3]=0; //case when fmdDigit->ADCsignal()==0
100	for (int i=0;i<threshold_array_size-1;i++)
101	{
102	if(ADC>threshold[i]&&ADC<=threshold[i+1])
103	RecParticles[3]=i;
104	}
105	new((*fReconParticles)[nRecPart++]) AliFMDReconstParticles(RecParticles);
106	} //digit loop
107	}//if FMD
108	gAlice->TreeR()->Reset();
109	gAlice->TreeR()->Fill();
110	gAlice->TreeR()->Write(0,TObject::kOverwrite);
111	} //event loop
112	cout<<"\nAliFMDReconstruction::Exec(TClonesArray fReconParticles,Option_t option) finished"<<endl;
113	}
114	//__________________________________________________________________
115
116	void AliFMDReconstruction::SetReconstParticlesFile(char * file )
117	{
118	if (!fReconstParticlesFile.IsNull())
119	cout<<"\nChanging reconstructed particles file from "<<
120	(char *) fReconstParticlesFile.Data()<< " to "<<file<<endl;
121	fReconstParticlesFile=file;
122	}
123	//__________________________________________________________________
124	void AliFMDReconstruction::Print(Option_t* option)const
125	{
126	cout<<"------------------- "<<GetName() <<" -------------"<< endl ;
127	if(fReconstParticlesFile.IsNull())
128	cout<<"\nWriting reconstructed particles to file"<<endl ;
129	else
130	cout<<"\nWriting reconstructed particles to file "<<
131	(char*) fReconstParticlesFile.Data() << endl ;
132	}
133
134
135
136
137