ITS/AliITSFindTracksV2.C

   1 /****************************************************************************
   2  *           Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch                 *
   3  ****************************************************************************/
   4
   5 #if !defined(__CINT__) || defined(__MAKECINT__)
   6   #include "Riostream.h"
   7   #include "TKey.h"
   8   #include "TStopwatch.h"
   9
  10   #include "AliRun.h"
  11   #include "AliMagF.h"
  12   #include "AliRunLoader.h"
  13   #include "AliTPCLoader.h"
  14   #include "AliITSLoader.h"
  15   #include "AliITS.h"
  16   #include "AliITSgeom.h"
  17   #include "AliITStrackerV2.h"
  18   #include "AliESD.h"
  19 #endif
  20
  21 extern AliRun *gAlice;
  22
  23 Int_t AliITSFindTracksV2(Int_t nev=5) {  //number of events to process
  24    cerr<<"Looking for tracks...\n";
  25
  26    if (gAlice) {
  27       delete gAlice->GetRunLoader();
  28       delete gAlice;
  29       gAlice=0;
  30    }
  31
  32    AliRunLoader* rl = AliRunLoader::Open("galice.root");
  33    if (rl == 0x0) {
  34       cerr<<"AliITSFindTracks.C : Can not open session RL=NULL"<< endl;
  35       return 3;
  36    }
  37
  38    AliITSLoader* itsl = (AliITSLoader*)rl->GetLoader("ITSLoader");
  39    if (itsl == 0x0) {
  40       cerr<<"AliITSFindTracksV2.C : Can not get ITS loader"<<endl;
  41       return 4;
  42    }
  43
  44    if (rl->LoadgAlice()) {
  45       cerr<<"AliITSFindTracksV2.C : LoadgAlice returned error"<<endl;
  46       delete rl;
  47       return 3;
  48    }
  49
  50    AliKalmanTrack::SetConvConst(
  51       1000/0.299792458/rl->GetAliRun()->Field()->SolenoidField()
  52    );
  53
  54    AliITS *dITS = (AliITS*)rl->GetAliRun()->GetDetector("ITS");
  55    if (!dITS) {
  56       cerr<<"AliITSFindClusters.C : Can not find the ITS detector !"<<endl;
  57       return 6;
  58    }
  59    AliITSgeom *geom = dITS->GetITSgeom();
  60
  61    AliITStrackerV2 tracker(geom);
  62
  63    itsl->LoadRecPoints("read");
  64
  65    if (nev>rl->GetNumberOfEvents()) nev=rl->GetNumberOfEvents();
  66    Int_t rc=0;
  67
  68    TFile *itsf=TFile::Open("AliESDits.root","RECREATE");
  69    if ((!itsf)||(!itsf->IsOpen())) {
  70       cerr<<"Can't AliESDits.root !\n"; return 1;
  71    }
  72    TFile *tpcf=TFile::Open("AliESDtpc.root");
  73    if ((!tpcf)||(!tpcf->IsOpen())) {
  74       cerr<<"Can't AliESDtpc.root !\n"; return 1;
  75    }
  76    TKey *key=0;
  77    TIter next(tpcf->GetListOfKeys());
  78    TStopwatch timer;
  79    for (Int_t i=0; i<nev; i++) {
  80        tpcf->cd();
  81        if ((key=(TKey*)next())==0) break;
  82        cerr<<"Processing event number: "<<i<<endl;
  83        AliESD *event=(AliESD*)key->ReadObj();
  84
  85        rl->GetEvent(i);
  86
  87        TTree *cTree=itsl->TreeR();
  88        if (!cTree) {
  89           cerr<<"AliITSFindTracksV2.C : Can't get the clusters tree !"<<endl;
  90           return 4;
  91        }
  92
  93        tracker.LoadClusters(cTree);
  94        rc=tracker.Clusters2Tracks(event);
  95        tracker.UnloadClusters();
  96
  97        if (rc==0) {
  98           Char_t ename[100];
  99           sprintf(ename,"%d",i);
 100           itsf->cd();
 101           if (!event->Write(ename)) rc++;
 102        }
 103        if (rc) {
 104           cerr<<"Something bad happened...\n";
 105        }
 106        delete event;
 107    }
 108    timer.Stop(); timer.Print();
 109
 110    tpcf->Close();
 111    itsf->Close();
 112
 113    delete rl;
 114
 115    return rc;
 116 }