cleaning before reading new data added in ReadTreeR

[u/mrichter/AliRoot.git] / ITS / AliITSFindTracksV2.C

diff --git a/ITS/AliITSFindTracksV2.C b/ITS/AliITSFindTracksV2.C
index fa490266e8d83df3ff7cb41f9702d49cd857075c..ab96d5f81867914e1ae0b2127bb697e847ba8f8e 100644 (file)
--- a/ITS/AliITSFindTracksV2.C
+++ b/ITS/AliITSFindTracksV2.C
@@ -1,12 +1,13 @@
 #ifndef __CINT__
-  #include <iostream.h>
+  #include <Riostream.h>
+  #include "AliITSgeom.h"
   #include "AliITStrackerV2.h"
 
   #include "TFile.h"
   #include "TStopwatch.h"
 #endif
 
-Int_t AliITSFindTracksV2() {
+Int_t AliITSFindTracksV2(Int_t nev=1) {  //number of events to process
    cerr<<"Looking for tracks...\n";
 
    TFile *out=TFile::Open("AliITStracksV2.root","new");
@@ -19,12 +20,24 @@ Int_t AliITSFindTracksV2() {
    if (!file->IsOpen()) {cerr<<"Can't open AliITSclustersV2.root !\n";return 3;}
 
    AliITSgeom *geom=(AliITSgeom*)file->Get("AliITSgeom");
+   if (!geom) {cerr<<"Can't get AliITSgeom !\n"; return 4;}
 
+   Int_t rc=0;
    TStopwatch timer;
    AliITStrackerV2 tracker(geom);
-   Int_t rc=tracker.Clusters2Tracks(in,out);
+   for (Int_t i=0; i<nev; i++) {
+     cerr<<"Processing event number : "<<i<<endl;
+     tracker.SetEventNumber(i);
+     //Double_t xyz[]={0.,0.,0.}; tracker.SetVertex(xyz);  //primary vertex
+     //Int_t flag[]={1};                                   //some default flags
+     //flag[0]= 0; tracker.SetupFirstPass(flag);           //no constraint
+     //flag[0]=-1; tracker.SetupSecondPass(flag);          //skip second pass
+     rc=tracker.Clusters2Tracks(in,out);
+   }
    timer.Stop(); timer.Print();
 
+   delete geom; //Thanks to Mariana Bondila
+
    file->Close();
    in->Close();
    out->Close();