updating
[u/mrichter/AliRoot.git] / PHOS / AnaESD.C
index 165561e..382c7ae 100644 (file)
 #include "AliEMCALRecParticle.h"
 #include "AliPHOSRecParticle.h"
 #include "AliKalmanTrack.h"
+#include "AliPHOSGridFile.h"
 #endif
 
-Double_t Match(TParticle * pp, AliESDtrack * cp) ; 
-TH1D * heta = new TH1D("heta", "Eta correlation", 100, -2., 2.) ; 
-TH1D * hphi = new TH1D("hphi", "Phi correlation", 360, 0., 360.) ; 
-void Ana() 
+void Match(TParticle * pp, AliESDtrack * cp, Double_t * dist) ; 
+TH1D * heta = new TH1D("heta", "Eta correlation", 100, 0., 360.) ; 
+TH1D * hphi = new TH1D("hphi", "Phi correlation", 100, 0., 360.) ; 
+// Data Challenge identification
+const TString kYear("2004") ; 
+const TString kProd("02") ; 
+const TString kVers("V4.01.Rev.00") ; 
+
+Bool_t Ana(const TString type = "per5", const Int_t run = 1, const Int_t nOfEvt = 1) 
 { 
-  AliPHOSGetter * gime = AliPHOSGetter::Instance("galice.root") ; 
-  Int_t nEvent = gime->MaxEvent() ;  
-  Int_t event ; 
-  AliESD * esd = 0 ;
-  for (event = 0 ; event < nEvent; event++) {
-    esd = gime->ESD(event) ; 
-    //esd->Print();  
-    Int_t caloindex ;
-    // Calorimeter tracks 
-    AliESDCaloTrack * ct ; 
-    for (caloindex = 0 ; caloindex < esd->GetNumberOfCaloTracks() ; caloindex++) {
-      // get the calorimeter type of particles (PHOS or EMCAL)
-      ct = esd->GetCaloTrack(caloindex) ;
-      TParticle * part = ct->GetRecParticle() ; 
+  Double_t dist[3] ; 
+  // get the LFN file name in the Grid catalogue ; 
+  AliPHOSGridFile lfn ; 
+  if (!lfn.IsConnected()) 
+    return kFALSE ; 
+  lfn.SetPath(kYear, kProd, kVers, type) ;  
+  lfn.SetRun(run) ; 
+
+  //loop over the events 
+  Int_t nevt, evt = 0 ; 
+  for (nevt = 0 ; nevt < nOfEvt ; nevt++) { 
+    evt++ ; 
+    lfn.SetEvt(evt) ;
+    TString fileName = lfn.GetLFN() ; 
+    
+    if (fileName.IsNull()) {
+      nevt-- ; 
+      continue ; 
+    }
 
-      AliESDtrack * cp ; 
-      Int_t cpindex ; 
-      for (cpindex = 0 ; cpindex < esd->GetNumberOfTracks() ; cpindex++) {
-       // get the charged tracks from central tracking
-       cp = esd->GetTrack(cpindex) ;
-       Double_t dist = Match(part, cp) ; 
+    printf(">>>>>>>>>>>> Processing %s-%s/%s/%s : run # %d event # %d \n", 
+          kYear.Data(), kProd.Data(), kVers.Data(), type.Data(), run, evt) ;  
+    AliPHOSGetter * gime = AliPHOSGetter::Instance(fileName) ; 
+    
+    Int_t nEvent = gime->MaxEvent() ;  
+    Int_t event ; 
+    AliESD * esd = 0 ;
+    for (event = 0 ; event < nEvent; event++) {
+      esd = gime->ESD(event) ; 
+      if (!esd) 
+       return kFALSE ; 
+      
+      //esd->Print();  
+      Int_t caloindex ;
+      // Calorimeter tracks 
+      AliESDCaloTrack * ct ; 
+      for (caloindex = 0 ; caloindex < esd->GetNumberOfCaloTracks() ; caloindex++) {
+       // get the calorimeter type of particles (PHOS or EMCAL)
+       ct = esd->GetCaloTrack(caloindex) ;
+       TParticle * part = ct->GetRecParticle() ; 
        
-       if (dist < 99999.) 
-         cout << "================ Distance = " << dist << endl ; 
+       AliESDtrack * cp ; 
+       Int_t cpindex ; 
+       for (cpindex = 0 ; cpindex < esd->GetNumberOfTracks() ; cpindex++) {
+         // get the charged tracks from central tracking
+         cp = esd->GetTrack(cpindex) ;
+         Match(part, cp, dist) ; 
+       }
+       heta->Fill( dist[1] ) ; 
+       hphi->Fill( dist[2] ) ; 
       }
     }
   }
-  //  heta->Draw() ; 
-  hphi->Draw() ; 
+  heta->Draw() ; 
+  //hphi->Draw() ; 
+  return kTRUE ; 
 }
-Double_t Match(TParticle * part, AliESDtrack * cp) 
+
+void Match(TParticle * part, AliESDtrack * cp, Double_t * dist) 
 {
   // Calculates the distance (x,z) between  the particle detected by PHOS and 
   // the charged particle reconstructed by the global tracking 
 
-  Double_t dist = 99999. ;
    
   AliPHOSRecParticle * pp  = dynamic_cast<AliPHOSRecParticle*>(part) ;  
   AliEMCALRecParticle * ep = dynamic_cast<AliEMCALRecParticle*>(part) ;  
@@ -87,9 +119,11 @@ Double_t Match(TParticle * part, AliESDtrack * cp)
   Double_t phZ, phX ; 
   
   if (pp) { // it is a PHOS particle 
-    
- //    cout << "PHOS particle # " << " pos (" 
-//      << pp->GetPos().X() << ", " << pp->GetPos().Y() << ", " << pp->GetPos().Z() << ")" << endl ;
+    Double_t cpTheta,  cpPhi ;  
+    Double_t phTheta,  phPhi ; 
+    cpTheta = cpPhi = phTheta = phPhi = 0. ; 
+   //    cout << "PHOS particle # " << " pos (" 
+    //          << pp->GetPos().X() << ", " << pp->GetPos().Y() << ", " << pp->GetPos().Z() << ")" << endl ;
     
     AliPHOSGetter * gime = AliPHOSGetter::Instance() ; 
     gime->PHOSGeometry()->ImpactOnEmc(*pp, phN, phZ, phX) ; 
@@ -102,33 +136,31 @@ Double_t Match(TParticle * part, AliESDtrack * cp)
       //       do not continue;
       //       if ((status & AliESDtrack::kTRDStop)!=0) 
       //       do not continue;  
-//       cout << "Charged particle # " << " pos (" 
-//        << xyzAtPHOS[0] << ", " << xyzAtPHOS[1] << ", " << xyzAtPHOS[2] << ")" <<  endl ;     
+      //       cout << "Charged particle # " << " pos (" 
+      //          << xyzAtPHOS[0] << ", " << xyzAtPHOS[1] << ", " << xyzAtPHOS[2] << ")" <<  endl ;     
       TVector3 poscp(xyzAtPHOS[0], xyzAtPHOS[1], xyzAtPHOS[2]) ;
       Int_t cpN ;
       Double_t cpZ,cpX ; 
       gime->PHOSGeometry()->ImpactOnEmc(poscp, cpN, cpZ, cpX) ; 
       if (cpN) {// we are inside the PHOS acceptance 
-//     cout << "Charged Matching 1: " << cpN << " " << cpZ << " " << cpX << endl ; 
-//     cout << "Charged Matching 2: " << phN << " " << phZ << " " << phX << endl ; 
-       dist = TMath::Sqrt( (cpZ-phZ)*(cpZ-phZ) + (cpX-phX)*(cpX-phX)) ;  
+       //      cout << "Charged Matching 1: " << cpN << " " << cpZ << " " << cpX << endl ; 
+       //      cout << "Charged Matching 2: " << phN << " " << phZ << " " << phX << endl ; 
+       dist[0] = TMath::Sqrt( (cpZ-phZ)*(cpZ-phZ) + (cpX-phX)*(cpX-phX)) ;  
       } 
-      Double_t phTheta = pp->Theta() ; 
-      Double_t phPhi   = pp->Phi() ;
+      phTheta = pp->Theta() ; 
+      phPhi   = pp->Phi() ;
       TParticle tempo ; 
       tempo.SetMomentum(xyzAtPHOS[0], xyzAtPHOS[1], xyzAtPHOS[2], 0.) ;  
-      Double_t cpTheta = tempo.Theta() ; 
-      Double_t cpPhi   = tempo.Phi() ;
+      cpTheta = tempo.Theta() ; 
+      cpPhi   = tempo.Phi() ;
       //cout << phTheta << " " << phPhi << " " << endl 
       //cout <<         cpTheta << " " << cpPhi-phPhi << " " << endl ; 
-      heta->Fill( (phTheta - cpTheta)*TMath::RadToDeg() ) ; 
-      hphi->Fill( (phPhi - cpPhi)*TMath::RadToDeg() ) ; 
     }
+    dist[1] = (phTheta - cpTheta)*TMath::RadToDeg() ; 
+    dist[2] = (phPhi - cpPhi)*TMath::RadToDeg() ; 
   }
   
   if (ep) {
     //cout << "EMCAL particle # " << endl ; 
   }
-  return dist ; 
 }