Tests and example macros working with ESD (Yu.Belikov)

[u/mrichter/AliRoot.git] / TPC / AliTPCFindTracks.C

diff --git a/TPC/AliTPCFindTracks.C b/TPC/AliTPCFindTracks.C
index a9ea84d0d951846011c218762b7d9ff3cd05fc52..7bc53ee6d3badebf11e98ff3f5aaa8e6c902fbb2 100644 (file)
--- a/TPC/AliTPCFindTracks.C
+++ b/TPC/AliTPCFindTracks.C
@@ -1,40 +1,108 @@
-#ifndef __CINT__
+/****************************************************************************
+ *           Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch                 *
+ ****************************************************************************/
+
+#if !defined(__CINT__) || defined(__MAKECINT__)
   #include <iostream.h>
   #include "AliTPCParam.h"
   #include "AliTPCtracker.h"
+  #include "AliRun.h"
+  #include "AliMagF.h"
+  #include "AliRunLoader.h"
+  #include "AliTPCLoader.h"
+  #include "AliESD.h"
 
   #include "TFile.h"
   #include "TStopwatch.h"
 #endif
 
-Int_t AliTPCFindTracks(Int_t eventn=1) { 
+extern AliRun *gAlice;
+
+Int_t AliTPCFindTracks(Int_t nev=5) {
+
    cerr<<"Looking for tracks...\n";
 
-   TFile *out=TFile::Open("AliTPCtracks.root","new");
-   if (!out->IsOpen()) {cerr<<"Delete old AliTPCtracks.root !\n"; return 1;}
+   if (gAlice) {
+     delete gAlice->GetRunLoader();
+     delete gAlice;
+     gAlice = 0x0;
+   }
 
-   TFile *in=TFile::Open("AliTPCclusters.root");
-   if (!in->IsOpen()) {cerr<<"Can't open AliTPCclusters.root !\n"; return 2;}
+   AliRunLoader *rl = AliRunLoader::Open("galice.root");
+   if (rl == 0x0) {
+      cerr<<"Can not open session"<<endl;
+      return 1;
+   }
 
-   AliTPCParam *par=(AliTPCParam*)in->Get("75x40_100x60");
-   if (!par) {cerr<<"Can't get TPC parameters !\n"; return 3;}
- 
-   TStopwatch timer;
+   AliTPCLoader *tpcl = (AliTPCLoader*)rl->GetLoader("TPCLoader");
+   if (tpcl == 0x0) {
+      cerr<<"Can not get TPC Loader"<<endl;
+      return 1;
+   }
+   
+   if (rl->LoadgAlice()) {
+      cerr<<"Error occured while loading gAlice"<<endl;
+      return 1;
+   }
 
+   AliKalmanTrack::SetConvConst(
+     1000/0.299792458/rl->GetAliRun()->Field()->SolenoidField()
+   );
+
+   rl->CdGAFile();
+   AliTPCParam *dig=(AliTPCParam *)gDirectory->Get("75x40_100x60_150x60");
+   if (!dig) { 
+        cerr<<"TPC parameters have not been found !\n";
+        return 1;
+   }
+
+   //rl->UnloadgAlice();   
+
+   tpcl->LoadRecPoints("read");
+
+   if (nev>rl->GetNumberOfEvents()) nev=rl->GetNumberOfEvents();
+    
+   TFile *ef=TFile::Open("AliESDtpc.root","RECREATE");
+   if ((!ef)||(!ef->IsOpen())) {
+      cerr<<"Can't AliESDtpc.root !\n"; return 1;
+   }
+   TStopwatch timer;
    Int_t rc=0;
-   for (Int_t i=0;i<eventn;i++){
+   AliTPCtracker tracker(dig);
+   for (Int_t i=0;i<nev;i++){
      printf("Processing event %d\n",i);
-     AliTPCtracker *tracker = new AliTPCtracker(par,i);
-     //Double_t xyz[]={0.,0.,0.}; tracker->SetVertex(xyz); //primary vertex
-     rc=tracker->Clusters2Tracks(0,out);
-     delete tracker;
+     AliESD *event=new AliESD(); 
+     rl->GetEvent(i);
+
+     TTree *in=tpcl->TreeR();
+     if (!in) {
+        cerr<<"Can't get clusters tree !\n";
+        return 4;
+     }
+
+     tracker.LoadClusters(in);
+     rc=tracker.Clusters2Tracks(event);
+     tracker.UnloadClusters();
+
+    if (rc==0) {
+        Char_t ename[100]; 
+        sprintf(ename,"%d",i);
+        ef->cd();
+        if (!event->Write(ename)) rc++;
+     } 
+     if (rc) {
+        cerr<<"Something bad happened...\n";
+     }
+     delete event;
    }
+
    timer.Stop(); timer.Print();
  
-   delete par; //Thanks to Mariana Bondila
+   ef->Close();
+
+   delete dig; //Thanks to Mariana Bondila
 
-   in->Close();
-   out->Close();
+   delete rl;
 
    return rc;
 }