FMD/AliFMDReconstruction.cxx

   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