1 #if !defined(__CINT__) || defined(__MAKECINT__)
6 #include "../ITS/UPGRADE/AliITSUClusterPix.h"
7 #include "../ITS/UPGRADE/AliITSURecoLayer.h"
8 #include "../ITS/UPGRADE/AliITSURecoDet.h"
9 #include "../ITS/UPGRADE/AliITSUHit.h"
10 #include "../ITS/UPGRADE/AliITSUGeomTGeo.h"
11 #include "AliITSsegmentation.h"
12 #include "AliGeomManager.h"
14 #include "AliLoader.h"
15 #include "AliCDBManager.h"
19 #include "TGeoMatrix.h"
20 #include "TParticle.h"
22 #include "TPaveStats.h"
23 #include "TClonesArray.h"
28 enum {kNPixAll=0,kNPixSPL=1,kDR=0,kDTXodd,kDTXeven,kDTZ, kDTXoddSPL,kDTXevenSPL,kDTZSPL};
30 TPaveStats* GetStPad(TH1* hst);
31 TPaveStats* SetStPadPos(TH1* hst,float x1,float x2,float y1,float y2, Int_t stl=-1,Int_t col=-1);
32 TCanvas* DrawNP(int np, TObjArray* harr=0, TCanvas* cnv=0);
33 TH1* GetHistoClSize(int npix,int id,TObjArray* harr=0);
34 void DrawReport(const char* psname, TObjArray* harr=0);
58 void compClusHits(int nev=-1)
60 const int kSplit=0x1<<22;
61 const int kSplCheck=0x1<<23;
63 gSystem->Load("libITSUpgradeBase");
64 gSystem->Load("libITSUpgradeSim");
65 gSystem->Load("libITSUpgradeRec");
66 gROOT->SetStyle("Plain");
68 AliCDBManager* man = AliCDBManager::Instance();
69 man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
70 man->SetSpecificStorage("GRP/GRP/Data",
71 Form("local://%s",gSystem->pwd()));
72 man->SetSpecificStorage("ITS/Align/Data",
73 Form("local://%s",gSystem->pwd()));
74 man->SetSpecificStorage("ITS/Calib/RecoParam",
75 Form("local://%s",gSystem->pwd()));
80 AliRunLoader* runLoader = AliRunLoader::Open("galice.root");
81 runLoader->LoadgAlice();
83 gAlice = runLoader->GetAliRun();
85 runLoader->LoadHeader();
86 runLoader->LoadKinematics();
87 runLoader->LoadRecPoints();
88 runLoader->LoadSDigits();
89 runLoader->LoadHits();
91 AliLoader *dl = runLoader->GetDetectorLoader("ITS");
93 AliGeomManager::LoadGeometry("geometry.root");
95 AliGeomManager::LoadAlignObjsFromCDBSingleDet("ITS",algITS);
96 AliGeomManager::ApplyAlignObjsToGeom(algITS);
98 AliITSUGeomTGeo* gm = new AliITSUGeomTGeo(kTRUE);
99 AliITSUClusterPix::SetGeom(gm);
101 AliITSURecoDet *its = new AliITSURecoDet(gm, "ITSinterface");
102 its->CreateClusterArrays();
104 Double_t xg1,yg1,zg1=0.,xg0,yg0,zg0=0.,tg0;
106 TTree * cluTree = 0x0;
107 TTree *hitTree = 0x0;
108 TClonesArray *hitList=new TClonesArray("AliITSUHit");
110 TObjArray arrMCTracks; // array of hit arrays for each particle
112 Float_t xyzClGloF[3];
113 Double_t xyzClGlo[3],xyzClTr[3];
116 int nlr=its->GetNLayersActive();
117 int ntotev = (Int_t)runLoader->GetNumberOfEvents();
118 printf("N Events : %i \n",ntotev);
119 if (nev>0) ntotev = TMath::Min(nev,ntotev);
122 TFile* flOut = TFile::Open("clInfo.root","recreate");
123 TTree* trOut = new TTree("clitsu","clitsu");
125 trOut->Branch("evID", &cSum.evID ,"evID/I");
126 trOut->Branch("volID",&cSum.volID,"volID/I");
127 trOut->Branch("lrID", &cSum.lrID ,"lrID/I");
128 trOut->Branch("clID", &cSum.clID ,"clID/I");
129 trOut->Branch("nPix", &cSum.nPix ,"nPix/I");
130 trOut->Branch("nX" , &cSum.nX ,"nX/I");
131 trOut->Branch("nZ" , &cSum.nZ ,"nZ/I");
132 trOut->Branch("q" , &cSum.q ,"q/I");
133 trOut->Branch("pt" , &cSum.pt ,"pt/F");
134 trOut->Branch("eta" ,&cSum.eta ,"eta/F");
135 trOut->Branch("phi" , &cSum.phi ,"phi/F");
136 trOut->Branch("xyz", cSum.xyz, "xyz[3]/F");
137 trOut->Branch("dX" , &cSum.dX ,"dX/F");
138 trOut->Branch("dZ" , &cSum.dZ ,"dZ/F");
139 trOut->Branch("split",&cSum.split,"split/O");
140 trOut->Branch("prim", &cSum.prim, "prim/O");
141 trOut->Branch("pdg", &cSum.pdg, "pdg/I");
142 trOut->Branch("ntr", &cSum.ntr, "ntr/I");
144 for (Int_t iEvent = 0; iEvent < ntotev; iEvent++) {
145 printf("\n Event %i \n",iEvent);
146 runLoader->GetEvent(iEvent);
147 AliStack *stack = runLoader->Stack();
150 hitTree->SetBranchAddress("ITS",&hitList);
153 for (int ilr=nlr;ilr--;) {
154 TBranch* br = cluTree->GetBranch(Form("ITSRecPoints%d",ilr));
155 if (!br) {printf("Did not find cluster branch for lr %d\n",ilr); exit(1);}
156 br->SetAddress(its->GetLayerActive(ilr)->GetClustersAddress());
158 cluTree->GetEntry(0);
159 its->ProcessClusters();
162 for(Int_t iEnt=0;iEnt<hitTree->GetEntries();iEnt++){//entries loop degli hits
163 hitTree->GetEntry(iEnt);
164 int nh = hitList->GetEntries();
165 for(Int_t iHit=0; iHit<nh;iHit++){
166 AliITSUHit *pHit = (AliITSUHit*)hitList->At(iHit);
167 int mcID = pHit->GetTrack();
168 TClonesArray* harr = arrMCTracks.GetEntriesFast()>mcID ? (TClonesArray*)arrMCTracks.At(mcID) : 0;
170 harr = new TClonesArray("AliITSUHit"); // 1st encounter of the MC track
171 arrMCTracks.AddAtAndExpand(harr,mcID);
174 new ( (*harr)[harr->GetEntriesFast()] ) AliITSUHit(*pHit);
178 // compare clusters and hits
180 printf(" tree entries: %lld\n",cluTree->GetEntries());
182 for (int ilr=0;ilr<nlr;ilr++) {
183 AliITSURecoLayer* lr = its->GetLayerActive(ilr);
184 TClonesArray* clr = lr->GetClusters();
185 int nClu = clr->GetEntries();
186 printf("Layer %d : %d clusters\n",ilr,nClu);
188 for (int icl=0;icl<nClu;icl++) {
189 AliITSUClusterPix *cl = (AliITSUClusterPix*)clr->At(icl);
190 int modID = cl->GetVolumeId();
192 //------------ check if this is a split cluster
193 int sInL = modID - gm->GetFirstChipIndex(ilr);
194 if (!cl->TestBit(kSplCheck)) {
195 cl->SetBit(kSplCheck);
196 // check if there is no other cluster with same label on this module
197 AliITSURecoSens* sens = lr->GetSensor(sInL);
198 int nclSn = sens->GetNClusters();
199 int offs = sens->GetFirstClusterId();
200 // printf("To check for %d (mod:%d) N=%d from %d\n",icl,modID,nclSn,offs);
201 for (int ics=0;ics<nclSn;ics++) {
202 AliITSUClusterPix* clusT = (AliITSUClusterPix*)lr->GetCluster(offs+ics); // access to clusters
203 if (clusT==cl) continue;
204 for (int ilb0=0;ilb0<3;ilb0++) {
205 int lb0 = cl->GetLabel(ilb0); if (lb0<=-1) break;
206 for (int ilb1=0;ilb1<3;ilb1++) {
207 int lb1 = clusT->GetLabel(ilb1); if (lb1<=-1) break;
210 clusT->SetBit(kSplit);
212 printf("Discard clusters of module %d:\n",modID);
223 const AliITSsegmentation* segm = gm->GetSegmentation(ilr);
225 cl->GetGlobalXYZ(xyzClGloF);
226 int clsize = cl->GetNPix();
227 for (int i=3;i--;) xyzClGlo[i] = xyzClGloF[i];
228 const TGeoHMatrix* mat = gm->GetMatrixSens(modID);
229 if (!mat) {printf("failed to get matrix for module %d\n",cl->GetVolumeId());}
230 mat->MasterToLocal(xyzClGlo,xyzClTr);
233 segm->LocalToDet(xyzClTr[0],xyzClTr[2],row,col); // effective col/row
235 for (int il=0;il<3;il++) {
236 if (cl->GetLabel(il)>=0) labels[nLab++] = cl->GetLabel(il);
240 for (int il=0;il<nLab;il++) {
241 TClonesArray* htArr = (TClonesArray*)arrMCTracks.At(labels[il]);
242 if (!htArr) {printf("did not find MChits for label %d ",labels[il]); cl->Print(); continue;}
244 int nh = htArr->GetEntriesFast();
246 for (int ih=nh;ih--;) {
247 AliITSUHit* tHit = (AliITSUHit*)htArr->At(ih);
248 if (tHit->GetChip()!=modID) continue;
253 printf("did not find MChit for label %d on module %d ",il,modID);
259 pHit->GetPositionG(xg1,yg1,zg1);
260 pHit->GetPositionG0(xg0,yg0,zg0,tg0);
262 double txyzH[3],gxyzH[3] = { (xg1+xg0)/2, (yg1+yg0)/2, (zg1+zg0)/2 };
263 mat->MasterToLocal(gxyzH,txyzH);
264 double rcl = TMath::Sqrt(xyzClTr[0]*xyzClTr[0]+xyzClTr[1]*xyzClTr[1]);
265 double rht = TMath::Sqrt(txyzH[0]*txyzH[0]+txyzH[1]*txyzH[1]);
267 GetHistoClSize(clsize,kDR,&histoArr)->Fill((rht-rcl)*1e4);
268 if (cl->TestBit(kSplit)) {
269 if (col%2) GetHistoClSize(clsize,kDTXoddSPL,&histoArr)->Fill((txyzH[0]-xyzClTr[0])*1e4);
270 else GetHistoClSize(clsize,kDTXevenSPL,&histoArr)->Fill((txyzH[0]-xyzClTr[0])*1e4);
271 GetHistoClSize(clsize,kDTZSPL,&histoArr)->Fill((txyzH[2]-xyzClTr[2])*1e4);
272 GetHistoClSize(0,kNPixSPL,&histoArr)->Fill(clsize);
274 if (col%2) GetHistoClSize(clsize,kDTXodd,&histoArr)->Fill((txyzH[0]-xyzClTr[0])*1e4);
275 else GetHistoClSize(clsize,kDTXeven,&histoArr)->Fill((txyzH[0]-xyzClTr[0])*1e4);
276 GetHistoClSize(clsize,kDTZ,&histoArr)->Fill((txyzH[2]-xyzClTr[2])*1e4);
277 GetHistoClSize(0,kNPixAll,&histoArr)->Fill(clsize);
280 cSum.volID = cl->GetVolumeId();
283 cSum.nPix = cl->GetNPix();
284 cSum.nX = cl->GetNx();
285 cSum.nZ = cl->GetNz();
287 cSum.split = cl->TestBit(kSplit);
288 cSum.dX = (txyzH[0]-xyzClTr[0])*1e4;
289 cSum.dZ = (txyzH[2]-xyzClTr[2])*1e4;
290 int label = cl->GetLabel(0);
292 if (label>=0 && (part=stack->Particle(label)) ) {
293 cSum.pdg = part->GetPdgCode();
294 cSum.eta = part->Eta();
295 cSum.pt = part->Pt();
296 cSum.phi = part->Phi();
297 cSum.prim = stack->IsPhysicalPrimary(label);
300 for (int ilb=0;ilb<3;ilb++) if (cl->GetLabel(ilb)>=0) cSum.ntr++;
301 for (int i=0;i<3;i++) cSum.xyz[i] = xyzClGloF[i];
306 printf("\nL%d(%c) Mod%d, Cl:%d | %+5.1f %+5.1f (%d/%d)|H:%e %e %e | C:%e %e %e\n",ilr,cl->TestBit(kSplit) ? 'S':'N',
307 modID,icl,(txyzH[0]-xyzClTr[0])*1e4,(txyzH[2]-xyzClTr[2])*1e4, row,col,
308 gxyzH[0],gxyzH[1],gxyzH[2],xyzClGlo[0],xyzClGlo[1],xyzClGlo[2]);
312 double a0,b0,c0,a1,b1,c1,e0;
313 pHit->GetPositionL0(a0,b0,c0,e0);
314 pHit->GetPositionL(a1,b1,c1);
316 cl->GetLocalXYZ(cloc);
317 printf("LocH: %e %e %e | %e %e %e\n",a0,b0,c0,a1,b1,c1);
318 printf("LocC: %e %e %e | %e %e %e\n",cloc[0],cloc[1],cloc[2],xyzClTr[0],xyzClTr[1],xyzClTr[2]);
326 // layerClus.Clear();
328 arrMCTracks.Delete();
336 DrawReport("clinfo.ps",&histoArr);
340 void DrawReport(const char* psname, TObjArray* harr)
342 gStyle->SetOptFit(1);
343 if (!harr) harr = &histoArr;
344 TCanvas* cnv = new TCanvas("cl","cl",900,600);
346 TString psnm1 = psname;
347 if (psnm1.IsNull()) psnm1 = "clusters.ps";
348 TString psnm0 = psnm1.Data();
350 TString psnm2 = psnm1.Data();
352 cnv->Print(psnm0.Data());
354 TH1* clall = GetHistoClSize(0,kNPixAll,harr);
355 clall->SetLineColor(kRed);
357 TH1* clSpl = GetHistoClSize(0,kNPixSPL,harr);
358 clSpl->SetLineColor(kBlue);
359 clSpl->Draw("sames");
362 SetStPadPos(clall,0.75,0.97,0.8,1.,-1,clall->GetLineColor());
363 SetStPadPos(clSpl,0.75,0.97,0.6,0.8,-1,clSpl->GetLineColor());
369 cnv->Print(psnm1.Data());
371 // plot cluster sized from 1 to 10
372 for (int i=1;i<=10;i++) {
373 if (clall->GetBinContent(clall->FindBin(i))<100) continue;
375 cnv->Print(psnm1.Data());
377 cnv->Print(psnm2.Data());
380 //------------------------------------------
381 TH1* GetHistoClSize(int npix,int id,TObjArray* harr)
385 if (!harr) harr = &histoArr;
388 if (harr->GetEntriesFast()>=id && (h=(TH1*)harr->At(id))) return h;
389 h = new TH1F("npixAll","npixAll",150,0.5,54.5);
391 h->SetLineColor(kRed);
392 harr->AddAtAndExpand(h, kNPixAll);
394 h = new TH1F("npixSpl","npixSpl",150,0.5,54.5);
395 h->SetLineColor(kBlue);
397 harr->AddAtAndExpand(h, kNPixSPL);
399 h = (TH1*)harr->At(id);
400 if (!h) {printf("Unknown histo id=%d\n",id); exit(1);}
404 int idh = npix*10+id;
405 if (harr->GetEntriesFast()>=idh && (h=(TH1*)harr->At(idh))) return h;
408 const double kdiff=80;
409 // need to create set of histos
411 h = new TH1F(Form("dxy_npix%d",npix),Form("dr_npix%d",npix),nbin,-kdiff,kdiff);
413 harr->AddAtAndExpand(h, npix*10 + kDR);
415 h = new TH1F(Form("dtxODD_npix%d",npix),Form("dtxODD_npix%d",npix),nbin,-kdiff,kdiff);
417 h->SetLineColor(kRed);
418 harr->AddAtAndExpand(h, npix*10 + kDTXodd);
419 h = new TH1F(Form("dtxEVN_npix%d",npix),Form("dtxEVN_npix%d",npix),nbin,-kdiff,kdiff);
421 h->SetLineColor(kBlue);
422 harr->AddAtAndExpand(h, npix*10 + kDTXeven);
424 h = new TH1F(Form("dtz_npix%d",npix),Form("dtz_npix%d",npix),nbin,-kdiff,kdiff);
425 h->SetLineColor(kGreen);
427 harr->AddAtAndExpand(h, npix*10 + kDTZ);
430 h = new TH1F(Form("SPL_dtxODD_npix%d",npix),Form("SPL_dtxODD_npix%d",npix),nbin,-kdiff,kdiff);
431 h->SetLineColor(kMagenta);
432 h->SetFillColor(kMagenta);
433 h->SetFillStyle(3001);
437 harr->AddAtAndExpand(h, npix*10 + kDTXoddSPL);
438 h = new TH1F(Form("SPL_dtxEVN_npix%d",npix),Form("SPL_dtxEVN_npix%d",npix),nbin,-kdiff,kdiff);
439 h->SetLineColor(kCyan);
440 h->SetFillColor(kCyan);
441 h->SetFillStyle(3006);
444 harr->AddAtAndExpand(h, npix*10 + kDTXevenSPL);
446 h = new TH1F(Form("SPL_dtz_npix%d",npix),Form("SPLdtz_npix%d",npix),nbin,-kdiff,kdiff);
447 harr->AddAtAndExpand(h, npix*10 + kDTZSPL);
450 h->SetLineColor(kGreen+2);
451 h->SetFillColor(kGreen+2);
453 h->SetFillStyle(3001);
454 h = (TH1*)harr->At(idh);
455 if (!h) {printf("Unknown histo id=%d\n",idh); exit(1);}
461 TCanvas* DrawNP(int np, TObjArray* harr, TCanvas* cnv)
463 if (!harr) harr = &histoArr;
464 if (!cnv) cnv = new TCanvas(Form("cnv%d",np),Form("cnv%d",np),900,700);
469 TH1* dxodd = (TH1*)harr->At(np*10+kDTXodd);
470 TH1* dxevn = (TH1*)harr->At(np*10+kDTXeven);
471 TH1* dxoddS =(TH1*)harr->At(np*10+kDTXoddSPL);
472 TH1* dxevnS =(TH1*)harr->At(np*10+kDTXevenSPL);
473 double max = TMath::Max(dxodd->GetMaximum(),dxevn->GetMaximum());
474 dxodd->SetMaximum(1.1*max);
475 dxodd->GetXaxis()->SetTitle("#DeltaX, #mum");
476 dxodd->SetTitle(Form("#DeltaX for clSize=%d",np));
477 dxodd->Fit("gaus","","");
478 dxevn->Fit("gaus","","sames");
480 dxoddS->Draw("sames");
481 dxevnS->Draw("sames");
485 SetStPadPos(dxodd,0.75,0.97,0.8,1., -1,dxodd->GetLineColor());
486 SetStPadPos(dxevn,0.75,0.97,0.6,0.8, -1,dxevn->GetLineColor());
487 SetStPadPos(dxoddS,0.75,0.97,0.4,0.6, -1,dxoddS->GetLineColor());
488 SetStPadPos(dxevnS,0.75,0.97,0.2,0.4, -1,dxevnS->GetLineColor());
491 TH1* dz = (TH1*)harr->At(np*10+kDTZ);
492 dz->SetTitle(Form("#DeltaZ for clSize=%d",np));
493 dz->GetXaxis()->SetTitle("#DeltaZ, #mum");
495 TH1* dzS = (TH1*)harr->At(np*10+kDTZSPL);
499 SetStPadPos(dz,0.75,0.97,0.8,1., -1, dz->GetLineColor());
500 SetStPadPos(dzS,0.75,0.97,0.5,0.7, -1, dzS->GetLineColor());
509 TPaveStats* GetStPad(TH1* hst)
511 TList *lst = hst->GetListOfFunctions();
513 int nf = lst->GetSize();
514 for (int i=0;i<nf;i++) {
515 TPaveStats *fnc = (TPaveStats*) lst->At(i);
516 if (fnc->InheritsFrom("TPaveStats")) return fnc;
523 TPaveStats* SetStPadPos(TH1* hst,float x1,float x2,float y1,float y2, Int_t stl, Int_t col)
525 TPaveStats* pad = GetStPad(hst);
531 if (stl>=0) pad->SetFillStyle(stl);
532 if (col>=0) pad->SetTextColor(col);
533 pad->SetFillColor(0);