new macros compliant with multiple events/files for TreeS,D,R

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

diff --git a/ITS/AliITSDigits2RecPoints.C b/ITS/AliITSDigits2RecPoints.C
index 962cccc716279f85c2cfc6af0f69b190e273e13f..e0dc51d2acddbfd913bd47296f49175ddc39cec4 100644 (file)
--- a/ITS/AliITSDigits2RecPoints.C
+++ b/ITS/AliITSDigits2RecPoints.C
@@ -1,15 +1,55 @@
-void AliITSDigits2RecPoints(Int_t evNumber1=0,Int_t evNumber2=0, const char *filename="galice.root"){
-    TStopwatch timer;
-
-    if(gAlice){
-       delete gAlice;
-       gAlice = 0;
-    } // end if gAlice
-    cout << "Creating reconstructed points from digits for the ITS..." << endl;
-    AliITSreconstruction *itsr = new AliITSreconstruction(filename);
-    timer.Start();
-    itsr->Init();
-    itsr->Exec(); 
-    timer.Stop(); timer.Print();    
-    delete itsr;
+TFile* AccessFile(TString inFile="galice.root", TString acctype="R");
+
+void AliITSDigits2RecPoints(TString inFile="galice.root", TString outFile="galice.root"){
+
+  TFile *file;
+  if(outFile.Data() == inFile.Data()){
+    file = AccessFile(inFile,"U");
+  }
+  else {
+    file = AccessFile(inFile);
+  }
+  
+  TStopwatch timer;
+
+  cout << "Creating reconstructed points from digits for the ITS..." << endl;
+  const char *nulptr=0;
+  AliITSreconstruction *itsr = new AliITSreconstruction(nulptr);
+  if(outFile.Data() != inFile.Data())itsr->SetOutputFile(outFile);
+  timer.Start();
+  itsr->Init();
+  itsr->Exec(); 
+  timer.Stop(); 
+  timer.Print();    
+  delete itsr;
+}
+
+//-------------------------------------------------------------------
+TFile * AccessFile(TString FileName, TString acctype){
+
+  // Function used to open the input file and fetch the AliRun object
+
+  TFile *retfil = 0;
+  TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject(FileName);
+  if (file) {file->Close(); delete file; file = 0;}
+  if(acctype.Contains("U")){
+    file = new TFile(FileName,"update");
+  }
+  if(acctype.Contains("N") && !file){
+    file = new TFile(FileName,"recreate");
+  }
+  if(!file) file = new TFile(FileName);   // default readonly
+  if (!file->IsOpen()) {
+       cerr<<"Can't open "<<FileName<<" !" << endl;
+       return retfil;
+  } 
+
+  // Get AliRun object from file or return if not on file
+  if (gAlice) {delete gAlice; gAlice = 0;}
+  gAlice = (AliRun*)file->Get("gAlice");
+  if (!gAlice) {
+       cerr << "AliRun object not found on file"<< endl;
+       return retfil;
+  } 
+  return file;
 }