Clusters will be read via AliITSRecoDet array to allow access similar

[u/mrichter/AliRoot.git] / ITS / UPGRADE / readClusters.C

diff --git a/ITS/UPGRADE/readClusters.C b/ITS/UPGRADE/readClusters.C
index 53cc625e6ff51d806e67fae1cf0fa7ce633a85f6..c69213f9804e7db6fd952b8df10feee5b79a021a 100644 (file)
--- a/ITS/UPGRADE/readClusters.C
+++ b/ITS/UPGRADE/readClusters.C
@@ -1,4 +1,5 @@
-void readClusters(){
+void readClusters(int nev=-1,int evStart=0)
+{
 
   gSystem->Load("libITSUpgradeBase");
   gSystem->Load("libITSUpgradeRec");
@@ -20,20 +21,28 @@ void readClusters(){
   AliITSUGeomTGeo* gm = new AliITSUGeomTGeo(kTRUE);
   Int_t nLayers = gm->GetNLayers();
   AliITSUClusterPix::SetGeom(gm);
-
+  AliITSURecoDet *its = new AliITSURecoDet(gm, "ITSinterface");
+  its->CreateClusterArrays();
+  //
   TH2F *xyGlob = new TH2F("xyGlob"," X - Y Global coordinates ",500,-50,50,500,-50,50);
   xyGlob->SetXTitle("cm"); 
   xyGlob->SetMarkerStyle(7); 
   TH1F *zGlob  = new TH1F("zGlob", " Z Global coordinates ",200, -50,50 );
   zGlob->SetXTitle("cm"); 
-
+  //
+  TH1F* rGlob = new TH1F("rGlob","R global", 5000, 0,50.);
 
   TTree * cluTree = 0x0;
-  TObjArray layerClus;
-
-  printf("N Events : %i \n",(Int_t)runLoader->GetNumberOfEvents());
-
-  for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
+  
+  int nevTot = (Int_t)runLoader->GetNumberOfEvents();
+  printf("N Events : %i \n",nevTot);
+  evStart = evStart<nevTot ? evStart : nevTot-1;
+  if (evStart<0) evStart = 0;
+  //
+  int lastEv = nev<0 ? nevTot : evStart+nev;
+  if (lastEv > nevTot) lastEv = nevTot;
+  //
+  for (Int_t iEvent = evStart; iEvent < lastEv; iEvent++) {
     printf("\n Event %i \n",iEvent);
     runLoader->GetEvent(iEvent);
     //   AliStack *stack = runLoader->Stack();
@@ -42,34 +51,36 @@ void readClusters(){
     while(1) {
       TBranch* br = cluTree->GetBranch(Form("ITSRecPoints%d",nlr));
       if (!br) break;
-      TClonesArray* clr = 0;
-      br->SetAddress(&clr);
-      layerClus.AddLast(clr);
+      br->SetAddress(its->GetLayerActive(nlr)->GetClustersAddress());
       nlr++;
-    }
-
+    }    
     printf(" tree entries: %d\n",cluTree->GetEntries());
     cluTree->GetEntry(0);      
+    AliITSUClusterPix::SetSortMode( AliITSUClusterPix::SortModeIdTrkYZ());
+    for (int ilr=0;ilr<nlr;ilr++) {its->GetLayerActive(ilr)->GetClusters()->Sort();}
+    its->ProcessClusters();
     //
     for (int ilr=0;ilr<nlr;ilr++) {
-      TClonesArray* clr = (TClonesArray*)layerClus.At(ilr);
+      AliITSURecoLayer* lr = its->GetLayerActive(ilr);
+      TClonesArray* clr = lr->GetClusters();
+      
       int nClu = clr->GetEntries();
       printf("Layer %d : %d clusters\n",ilr,nClu);
       //
       for (int icl=0;icl<nClu;icl++) {
        AliITSUClusterPix *cl = (AliITSUClusterPix*)clr->At(icl);
-       cl->Print("glo");
-       Double_t loc[3]={cl->GetX(),cl->GetY(),cl->GetZ()}; 
-       Double_t glob[3]; 
-       gm->LocalToGlobal(cl->GetVolumeId(),loc,glob);
-       //      printf("%d: mod %d: loc(%.4lf,%.4lf,%.4lf); glob(%.4lf,%.4lf,%.4lf); \n",icl,cl->GetVolumeId(),
-       //             loc[0],loc[1],loc[2],glob[0],glob[1],glob[2]);
+       printf("#%4d | ",icl); cl->Print("glo");
+       Float_t loc[3];
+       cl->GetLocalXYZ(loc);
+       Float_t glob[3]; 
+       cl->GetGlobalXYZ(glob);
+       //printf("%d: mod %d: loc(%.4lf,%.4lf,%.4lf); glob(%.4lf,%.4lf,%.4lf); \n",icl,cl->GetVolumeId(),
+       //       loc[0],loc[1],loc[2],glob[0],glob[1],glob[2]);
        xyGlob->Fill(glob[0],glob[1]);
        zGlob->Fill(glob[2]);
- 
+       rGlob->Fill(TMath::Sqrt(glob[0]*glob[0]+glob[1]*glob[1]));
       }
     }
-    layerClus.Clear();
   }//event loop
 
   Int_t size = 400;
@@ -82,4 +93,8 @@ void readClusters(){
   zCanv->cd();
   zGlob->Draw();
 
+  TCanvas *rCanv =  new TCanvas("rCanvClus","RecPoint R Positions",size+20,10,size,size);
+  rCanv->cd();
+  rGlob->Draw();
+
 }