ATO-98 - connect distortion trees - with custom description ()
[u/mrichter/AliRoot.git] / TPC / AliTPCHits2Digits.C
index 30a7f30..7957a98 100644 (file)
@@ -1,90 +1,69 @@
-Int_t AliTPCHits2Digits()
-{
+#if !defined(__CINT__) || defined(__MAKECINT__)
+  #include <Riostream.h>
 
-  // new version by J.Belikov
+  #include "AliRun.h"
+  #include "AliRunLoader.h"
+  #include "AliLoader.h"
+  #include "AliTPC.h"
 
+  #include "TStopwatch.h"
+#endif
+
+extern AliRun *gAlice;
+
+Int_t AliTPCHits2Digits(Int_t nev=5) {
   // Connect the Root Galice file containing Geometry, Kine and Hits
+  if (gAlice) { 
+     delete AliRunLoader::Instance();
+     delete gAlice;//if everything was OK here it is already NULL
+     gAlice = 0x0;
+  }
 
-  const char * inFile_old = "galice.root"; 
-  const char * inFile_new = "rfio:galice.root";
-  TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject(inFile_old);
-  if (file) {file->Close(); delete file;}
-  file = new TFile(inFile_new,"UPDATE");
-  if (!file->IsOpen()) {
-    cerr<<"Can't open "<<inFile_new<<" !\n";
+  AliRunLoader *rl = AliRunLoader::Open("galice.root","Event","update");
+  if (!rl) {
+    cerr<<"Can't load RunLoader from "<<endl;
     return 1;
   }
 
   // Get AliRun object from file or create it if not on file
-  if (gAlice) delete gAlice;
-  gAlice = (AliRun*)file->Get("gAlice");
+
+  rl->LoadgAlice();
+  gAlice = rl->GetAliRun();
   if (!gAlice) {
     cerr<<"AliTPCHits2Digits.C : AliRun object not found on file\n";
     return 2;
   }
 
-  gAlice->GetEvent(0);
   AliTPC *TPC = (AliTPC*)gAlice->GetDetector("TPC");      
-
-//Set response functions
-
-  AliTPCParamSR *param=(AliTPCParamSR*)gDirectory->Get("75x40_100x60");
-  AliTPCPRF2D    * prfinner   = new AliTPCPRF2D;
-  AliTPCPRF2D    * prfouter   = new AliTPCPRF2D;
-  AliTPCRF1D     * rf    = new AliTPCRF1D(kTRUE);
-  rf->SetGauss(param->GetZSigma(),param->GetZWidth(),1.);
-  rf->SetOffset(3*param->GetZSigma());
-  rf->Update();
-
-  TDirectory *savedir=gDirectory;
-  TFile *f=TFile::Open("$ALICE_ROOT/TPC/AliTPCprf2d.root");
-  if (!f->IsOpen()) { 
-     cerr<<"Can't open $ALICE_ROOT/TPC/AliTPCprf2d.root !\n"
-     return 3;
+  AliLoader * tpcl = rl->GetLoader("TPCLoader");
+  if ((TPC == 0x0) || (tpcl == 0x0)) {
+    cerr<<"AliTPCHits2Digits.C : Can not find TPC or TPCLoader\n";
+    delete rl;
+    return 3;
   }
-  prfinner->Read("prf_07504_Gati_056068_d02");
-  prfouter->Read("prf_10006_Gati_047051_d03");
-  f->Close();
-  savedir->cd();
+  tpcl->LoadHits("READ");
+  tpcl->LoadDigits("recreate");
 
-  param->SetInnerPRF(prfinner);
-  param->SetOuterPRF(prfouter); 
-  param->SetTimeRF(rf);
-  TPC->SetParam(param);
-   
-  cerr<<"Digitizing TPC...\n";
+  TStopwatch timer;
+  timer.Start();
 
-  //setup TPCDigitsArray 
-  AliTPCDigitsArray *arr = new AliTPCDigitsArray; 
-  arr->SetClass("AliSimDigits");
-  arr->Setup(param);
-  TPC->SetParam(param);
-  arr->MakeTree();
+ // uncomment below lines to set sectors active
+ // Int_t sec[10]={0,1,2,3,4,5,6,7,8,9};
+ // TPC->SetActiveSectors(sec,10);
 
-  TPC->SetDigitsArray(arr);
-  TPC->Hits2Digits();
-  /*  TPC->Hits2DigitsSector(1);             
-  TPC->Hits2DigitsSector(2);             
-  TPC->Hits2DigitsSector(3);             
-  TPC->Hits2DigitsSector(1+18);             
-  TPC->Hits2DigitsSector(2+18);             
-  TPC->Hits2DigitsSector(3+18);             
+  for (Int_t i=0; i<nev; i++){
+    printf("Processing event %d \n",i);
+    if(rl->GetEvent(i)) break;
+    TPC->SetActiveSectors(); // all sectors set active
+    TPC->Hits2Digits(i);
+  }
 
-  TPC->Hits2DigitsSector(36+1);             
-  TPC->Hits2DigitsSector(36+2);             
-  TPC->Hits2DigitsSector(36+3);             
-  TPC->Hits2DigitsSector(36+1+18);             
-  TPC->Hits2DigitsSector(36+2+18);             
-  TPC->Hits2DigitsSector(36+3+18); */
-            
-  //write results
+  delete rl;
 
-  char treeName[100];
-  sprintf(treeName,"TreeD_%s",param->GetTitle());
-  TPC->GetDigitsArray()->GetTree()->Write(treeName);
+  timer.Stop();
+  timer.Print();
 
-  delete gAlice; gAlice=0;
-  file->Close(); delete file;
   return 0;
-};
+}