]> git.uio.no Git - u/mrichter/AliRoot.git/blob - ITS/UPGRADE/readDigits.C
The segmentations are made data member of the AliITSUGeomTGeo, will be loaded
[u/mrichter/AliRoot.git] / ITS / UPGRADE / readDigits.C
1 void readDigits(){
2
3   gSystem->Load("libITSUpgradeBase");
4   gSystem->Load("libITSUpgradeSim");
5   gROOT->SetStyle("Plain");
6
7
8   gAlice=NULL;
9   AliRunLoader* runLoader = AliRunLoader::Open("galice.root");
10   runLoader->LoadgAlice();
11
12   gAlice = runLoader->GetAliRun();
13
14   runLoader->LoadHeader();
15   runLoader->LoadKinematics();
16   runLoader->LoadSDigits();
17   runLoader->LoadDigits();
18
19   AliGeomManager::LoadGeometry("geometry.root");
20   AliITSUGeomTGeo* gm = new AliITSUGeomTGeo(kTRUE,kTRUE);
21   //
22   Int_t nLayers = gm->GetNLayers();
23   Int_t nModules = gm->GetNModules();
24   Int_t nbins=100;
25   Int_t xmin=0;
26   Int_t xmax=50000;//00*1e-09;
27
28
29   TH1D **hNelSDig = new TH1D*[nLayers], **hNelDig = new TH1D*[nLayers];
30   //
31   for(Int_t i=0; i< nLayers; i++ ) {
32     hNelSDig[i] = new TH1D(Form("hNelSDig%i",i),Form("electron distribution in SDigits [ Layer %i] ",i),nbins,xmin,xmax);
33     hNelSDig[i]->SetXTitle("N electrons");
34     hNelDig[i] = new TH1D(Form("hNelDig%i",i),Form("electron distribution in Digits [ Layer %i] ",i),nbins,xmin,xmax);
35     hNelDig[i]->SetXTitle("N electrons");
36   }
37   
38   AliLoader *dl = runLoader->GetDetectorLoader("ITS");
39
40
41   //SDIGITS INIT
42   TTree * sDigTree = 0x0;
43   TClonesArray *sDigArr= new TClonesArray("AliITSUSDigit");
44
45   //DIGITS INIT
46   TTree * digTree = 0x0;
47   TClonesArray *digArr= new TClonesArray("AliITSUDigitPix");
48
49   printf("N Events : %i \n",(Int_t)runLoader->GetNumberOfEvents());
50
51   for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
52     printf("\n Event %i \n",iEvent);
53     runLoader->GetEvent(iEvent);
54     AliStack *stack = runLoader->Stack();
55     sDigTree=dl->TreeS();
56     digTree=dl->TreeD();
57     //
58     sDigTree->SetBranchAddress("ITS",&sDigArr);
59     digTree->SetBranchAddress("ITSDigitsPix",&digArr);
60
61     for (int imod=0;imod<nModules;imod++) {
62       sDigTree->GetEntry(imod);
63       digTree->GetEntry(imod);      
64       int detType = gm->GetModuleDetTypeID(imod);
65       AliITSUSegmentationPix* segm = (AliITSUSegmentationPix*)gm->GetSegmentationByID(detType);
66       int lay,lad,det;
67       int nsdig = sDigArr->GetEntries();
68       int ndig  = digArr->GetEntries();
69       if (nsdig<1) continue;
70       gm->GetModuleId(imod, lay,lad,det);
71       printf("\nModule %3d: (det %2d in ladder %2d of Layer %d) |NSDigits: %4d NDigits: %4d\n",imod,det,lad,lay,nsdig,ndig);
72       //
73       for (int isdig=0;isdig<nsdig;isdig++) {
74         AliITSUSDigit *pSdig = (AliITSUSDigit*)sDigArr->At(isdig);
75         int sdinfo = pSdig->GetUniqueID();
76         printf("#%3d Sdigit col:%4d/row:%4d generated by track %5d (%s)\t",isdig, sdinfo/segm->Npx(),sdinfo%segm->Npx(),
77                pSdig->GetTrack(0),stack->Particle(pSdig->GetTrack(0))->GetName());
78         pSdig->Print();
79         hNelSDig[lay]->Fill(pSdig->GetSignal()); 
80       }
81       //
82       for (int idig=0;idig<ndig;idig++) {
83         AliITSUDigitPix *pDig = (AliITSUDigitPix*)digArr->At(idig);
84         printf("#%3d digit, col:%4d/row:%4d signal: %.2e,  generated by tracks ",idig,pDig->GetCoord1(),pDig->GetCoord2(),pDig->GetSignalPix()); 
85         for (int itr=0;itr<pDig->GetNTracks();itr++) if (pDig->GetTrack(itr)>=0) printf(" %5d",pDig->GetTrack(itr)); printf("\n");
86         //
87         hNelDig[lay]->Fill(pDig->GetSignalPix()); 
88       }
89       //
90     }
91
92   }//event loop
93
94   TCanvas *cSd = new TCanvas("cSd","Summable Digits",1000,800);
95   cSd->Divide(nLayers&0x1 ?  (nLayers/2+1) : nLayers/2,2); 
96   TCanvas *cD = new TCanvas("cD","Digits",1000,800);
97   cD->Divide(nLayers&0x1 ?  (nLayers/2+1) : nLayers/2,2); 
98
99   for(Int_t ip =1; ip<=nLayers; ip++){
100     cSd->cd(ip);
101     hNelSDig[ip-1]->Draw();
102     cD->cd(ip);
103     hNelDig[ip-1]->Draw();
104   }
105
106
107 }