ITS/AliITSVertexerTracksTest.C

   1 #if !defined(__CINT__) || defined(__MAKECINT__)
   2 //-- --- standard headers-------------
   3 #include <Riostream.h>
   4 //--------Root headers ---------------
   5 #include <TSystem.h>
   6 #include <TFile.h>
   7 #include <TStopwatch.h>
   8 #include <TObject.h>
   9 #include <TTree.h>
  10 #include <TStopwatch.h>
  11 #include <TKey.h>
  12 //----- AliRoot headers ---------------
  13 #include "alles.h"
  14 #include "AliRun.h"
  15 #include "AliRunLoader.h"
  16 #include "AliMagF.h"
  17 #include "AliKalmanTrack.h"
  18 #include "AliESDVertex.h"
  19 #include "AliITSVertexer.h"
  20 #include "AliITSVertexerTracks.h"
  21 //-------------------------------------
  22 #endif
  23 void AliITSVertexerTracksTest(Int_t evFirst=0,Int_t evLast=0,Bool_t esd=kTRUE,
  24                               const Char_t *galiceName="galice.root",
  25                               const Char_t *inName="AliESDs.root",
  26                               const Char_t *outName="AliESDs.root") {
  27   /*******************************************************************
  28    *                                                                 *
  29    * Test macro for vertexing in pp using tracks.                    *
  30    * If(esd) {                                                       *
  31    *   Input file must contain ESD events with tracks reco in ITS    *
  32    *   if output file = input file, vertices will be stored in the   *
  33    *   ESD events and these rewritten to the same file; otherwise    *
  34    *   they will be written to a new file                            *
  35    * } else {                                                        *
  36    *   input file must contain a tree with AliITStracksV2 and        *
  37    *   can be written to the same file (output file = input file)    *
  38    *   or to a new one                                               *
  39    * }                                                               *
  40    * If file galice.root is present field will be read from it       *
  41    * otherwise it can be set here "by hand".                         *
  42    *                                                                 *
  43    * Origin: A.Dainese, Padova  andrea.dainese@pd.infn.it            *
  44    *******************************************************************/
  45
  46   TStopwatch timer;
  47   timer.Start();
  48
  49   // Look for field value in galice.root
  50   Double_t field = 0.4;
  51   if(!gSystem->AccessPathName(galiceName,kFileExists)) {
  52     AliRunLoader *rl = AliRunLoader::Open(galiceName);
  53     rl->LoadgAlice();
  54     AliMagF *fiel = (AliMagF*)gAlice->Field();
  55     field=(Double_t)fiel->SolenoidField()/10.;
  56     AliKalmanTrack::SetConvConst(100/0.299792458/field);
  57     printf(" B = %3.1f read from gAlice and set\n",field);
  58   } else {
  59     printf(" File galice.root not found: default 0.4 T being used!\n");
  60   }
  61
  62   // Open input and output files
  63   TFile *inFile  = TFile::Open(inName);
  64   TFile *outFile = TFile::Open(outName,"update");
  65
  66   // Create vertexer
  67   AliITSVertexerTracks *vertexer =
  68     new AliITSVertexerTracks(inFile,outFile,field,evFirst,evLast);
  69   // Find vertices
  70   if(esd) {
  71     vertexer->FindVerticesESD();
  72   } else {
  73     vertexer->FindVertices();
  74   }
  75
  76   timer.Stop(); timer.Print();
  77
  78   delete vertexer;
  79
  80   inFile->Close();
  81   outFile->Close();
  82   delete inFile;
  83   delete outFile;
  84
  85   return;
  86 }
  87 //----------------------------------------------------------------------------
  88 void VertexRecoInESDChain() {
  89
  90   TStopwatch timer;
  91   timer.Start();
  92
  93   // Look for field value in galice.root
  94   Double_t field = 0.4;
  95   if(!gSystem->AccessPathName("galice.root",kFileExists)) {
  96     AliRunLoader *rl = AliRunLoader::Open("galice.root");
  97     rl->LoadgAlice();
  98     AliMagF *fiel = (AliMagF*)gAlice->Field();
  99     field=(Double_t)fiel->SolenoidField()/10.;
 100     AliKalmanTrack::SetConvConst(100/0.299792458/field);
 101     printf(" B = %3.1f read from gAlice and set\n",field);
 102   } else {
 103     printf(" File galice.root not found: default 0.4 T being used!\n");
 104   }
 105
 106   // open file with ESD events (this would not be necessary in the chain...)
 107   if(gSystem->AccessPathName("AliESDs.root",kFileExists))
 108     { printf(" ESD file not found!\n"); return; }
 109   TFile *esdFile = new TFile("AliESDs.root","update");
 110
 111   AliITSVertexerTracks *vertexer = new AliITSVertexerTracks();
 112   vertexer->SetField(field);
 113
 114   Int_t n=0;
 115   TKey *key=0;
 116   TIter next(esdFile->GetListOfKeys());
 117
 118   //******* The loop over events
 119   while ((key=(TKey*)next())!=0) {
 120     printf("--- Processing event number : %d",n++);
 121
 122     AliESD *event=(AliESD*)key->ReadObj();
 123
 124     // find vertex and store it in the ESD (only pos and cov matrix)
 125     vertexer->FindVertexForCurrentEvent(event);
 126
 127     // write event on file
 128     event->Write();
 129
 130   }
 131
 132   timer.Stop(); timer.Print();
 133
 134   delete vertexer;
 135   esdFile->Close();
 136   delete esdFile;
 137
 138   return;
 139 }
 140
 141
 142
 143
 144
 145
 146