TPC/AliTPCFindTracksMI.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 "AliTPCParam.h"
   8   #include "AliTPCtracker.h"
   9   #include "AliTPCtrackerMI.h"
  10   #include "AliRun.h"
  11   #include "AliMagF.h"
  12   #include "AliRunLoader.h"
  13   #include "AliTPCLoader.h"
  14   #include "AliESD.h"
  15   #include "TFile.h"
  16   #include "TStopwatch.h"
  17 #endif
  18
  19 extern AliRun *gAlice;
  20
  21
  22 Int_t AliTPCFindTracksMI(Int_t N=-1) {
  23
  24    cerr<<"Looking for tracks...\n";
  25
  26    if (gAlice)
  27     {
  28      delete gAlice->GetRunLoader();
  29      delete gAlice;
  30      gAlice = 0x0;
  31     }
  32
  33    AliRunLoader *rl = AliRunLoader::Open("galice.root");
  34    if (rl == 0x0)
  35     {
  36       cerr<<"Can not open session"<<endl;
  37       return 1;
  38     }
  39    AliTPCLoader *tpcl = (AliTPCLoader*)rl->GetLoader("TPCLoader");
  40    if (tpcl == 0x0)
  41     {
  42       cerr<<"Can not get TPC Loader"<<endl;
  43       return 1;
  44     }
  45
  46    if (rl->LoadgAlice())
  47     {
  48       cerr<<"Error occured while l"<<endl;
  49       return 1;
  50     }
  51    AliKalmanTrack::SetConvConst(1000/0.299792458/rl->GetAliRun()->Field()->SolenoidField());
  52
  53    rl->CdGAFile();
  54
  55    AliTPCParam *param=(AliTPCParam *)gDirectory->Get("75x40_100x60_150x60");
  56    if (!param)
  57     {
  58      param=(AliTPCParam *)gDirectory->Get("75x40_100x60");
  59      if (!param)
  60       {
  61         cerr<<"TPC parameters have not been found !\n";
  62         return 1;
  63       }
  64      else
  65       {
  66         cout<<"TPC 75x40_100x60 geometry found"<<endl;
  67       }
  68     }
  69    else
  70     {
  71       cout<<"TPC 75x40_100x60_150x60  geometry found"<<endl;
  72     }
  73
  74
  75    tpcl->LoadTracks("recreate");
  76
  77    Int_t eventn;
  78    if (N<=0)
  79     {
  80      eventn = rl->GetNumberOfEvents();
  81      rl->UnloadHeader();
  82     }
  83    else
  84     eventn = N;
  85
  86    TStopwatch timer;
  87    Int_t rc=0;
  88    for (Int_t i=0;i<eventn;i++)
  89     {
  90       rl->GetEvent(i);
  91       TTree * input = tpcl->TreeR();
  92       if (input == 0x0)
  93        {
  94          tpcl->LoadRecPoints("read");
  95          input = tpcl->TreeR();
  96          if (input == 0x0)
  97           {
  98             cerr << "Problems with input tree (TreeR) for event " << i <<endl;
  99             continue;
 100           }
 101        }
 102       TTree * output = tpcl->TreeT();
 103       if (output == 0x0)
 104        {
 105          tpcl->MakeTree("T");
 106          output = tpcl->TreeT();
 107          if (output == 0x0)
 108           {
 109             cerr << "Problems with output tree (TreeT) for event " << i <<endl;
 110             continue;
 111           }
 112        }
 113
 114       printf("Processing event %d\n",i);
 115       AliTPCtrackerMI *tracker = new AliTPCtrackerMI(param);
 116       tracker->SetIO();
 117       tracker->LoadClusters();
 118       rc=tracker->Clusters2Tracks();
 119       tracker->WriteTracks(output);
 120       tracker->UnloadClusters();
 121       tpcl->WriteTracks("OVERWRITE");
 122       //output->GetDirectory()->cd();
 123       //output->Write();
 124       delete tracker;
 125     }
 126    timer.Stop(); timer.Print();
 127    rl->UnloadgAlice();
 128
 129    delete param;
 130    delete rl;
 131    return rc;
 132 }