Right Local Reference System in theta Cerenkov reconstruction
[u/mrichter/AliRoot.git] / HMPID / Hdisp.C
1 TCanvas *pAll=0;
2 AliRunLoader *gAL=0; AliLoader *gHL=0; AliESD *gEsd=0; TTree *gEsdTr=0; AliHMPID *gH=0;
3 Int_t gEvt=0; Int_t gMaxEvt=0;
4 TObjArray *pNmean;
5 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
6 void Hdisp()
7 {//display events from files if any in current directory or simulated events
8   pAll=new TCanvas("all","",1300,900); pAll->Divide(3,3,0,0);
9 //  pAll->ToggleEditor();
10   pAll->Connect("ProcessedEvent(Int_t,Int_t,Int_t,TObject*)",0,"","DoZoom(Int_t,Int_t,Int_t,TObject*)");
11
12   OpenCalib();  
13   if(gSystem->IsFileInIncludePath("galice.root")){// tries to open session
14     if(gAlice) delete gAlice;                                               //in case we execute this in aliroot delete default AliRun object 
15     gAL=AliRunLoader::Open();                                                                    //try to open galice.root from current dir 
16     gAL->LoadgAlice();                                                                           //take new AliRun object from galice.root   
17     gHL=gAL->GetDetectorLoader("HMPID");  gH=(AliHMPID*)gAL->GetAliRun()->GetDetector("HMPID");  //get HMPID object from galice.root
18     gMaxEvt=gAL->GetNumberOfEvents()-1;
19     gHL->LoadHits(); gHL->LoadSDigits(); gHL->LoadDigits(); gHL->LoadRecPoints();
20
21     TFile *pEsdFl=TFile::Open("AliESDs.root"); gEsdTr=(TTree*) pEsdFl->Get("esdTree"); gEsdTr->SetBranchAddress("ESD", &gEsd);
22     pAll->cd(7); TButton *pBtn=new TButton("Next","ReadEvt()",0,0,0.2,0.1);   pBtn->Draw();
23                  TButton *pHitBtn=new TButton("Print hits","PrintHits()",0,0.2,0.3,0.3);   pHitBtn->Draw();
24                  TButton *pSdiBtn=new TButton("Print sdis","PrintSdis()",0,0.4,0.3,0.5);   pSdiBtn->Draw();
25                  TButton *pDigBtn=new TButton("Print digs","PrintDigs()",0,0.6,0.3,0.7);   pDigBtn->Draw();
26                  TButton *pCluBtn=new TButton("Print clus","PrintClus()",0,0.8,0.3,0.9);   pCluBtn->Draw();  
27     ReadEvt();
28   }else{
29     pAll->cd(7); TButton *pBtn=new TButton("Next","SimEvt()",0,0,0.2,0.1);   pBtn->Draw(); 
30     SimEvt();
31   }      
32 }
33 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
34 void ReadEvt()
35 {// Read curent event and display it assumes that session is alredy opened
36   TClonesArray hits("AliHMPIDHit");
37   if(gEvt>gMaxEvt) gEvt=0; if(gEvt<0) gEvt=gMaxEvt;
38   
39   gEsdTr->GetEntry(gEvt); gAL->GetEvent(gEvt); 
40   ReadHits(&hits); gHL->TreeS()->GetEntry(0); gHL->TreeD()->GetEntry(0); gHL->TreeR()->GetEntry(0);
41   
42   
43   DrawEvt(&hits,gH->DigLst(),gH->CluLst(),gEsd);
44   gEvt++;
45 }
46 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
47 void SimEvt()
48 {
49   TClonesArray hits("AliHMPIDHit"); 
50   TClonesArray sdig("AliHMPIDDigit"); 
51   TObjArray    digs(7); for(Int_t i=0;i<7;i++) digs.AddAt(new TClonesArray("AliHMPIDDigit"),i);
52   TObjArray    clus(7); for(Int_t i=0;i<7;i++) clus.AddAt(new TClonesArray("AliHMPIDCluster"),i);
53   AliESD esd;
54   AliHMPIDDigit::fSigmas=4;
55   AliHMPIDDigitizer::DoNoise(kFALSE);
56   SimEsd(&esd);
57   SimHits(&esd,&hits);
58              AliHMPIDv1::Hit2Sdi(&hits,&sdig);                               
59       AliHMPIDDigitizer::Sdi2Dig(&sdig,&digs);     
60       AliHMPIDReconstructor::Dig2Clu(&digs,&clus);
61       AliHMPIDTracker::Recon(&esd,&clus,pNmean);
62   
63   pAll->cd(3);  gPad->Clear(); TLatex txt;txt.DrawLatex(0.2,0.2,Form("Simulated event %i",gEvt));
64   DrawEvt(&hits,&digs,&clus,&esd);  
65   gEvt++;
66 }//SimEvt()
67 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
68 void SimEsd(AliESD *pEsd)
69 {
70   TParticle part; TLorentzVector mom;
71   for(Int_t iTrk=0;iTrk<100;iTrk++){//stack loop
72     part.SetPdgCode(kProton);
73     part.SetProductionVertex(0,0,0,0);
74     Double_t eta= -0.2+gRandom->Rndm()*0.4; //rapidity is random [-0.2,+0.2]
75     Double_t phi= gRandom->Rndm()*60.*TMath::DegToRad();   //phi is random      [ 0  , 60 ] degrees    
76     mom.SetPtEtaPhiM(5,eta,phi,part.GetMass());
77     part.SetMomentum(mom);
78     AliESDtrack trk(&part);
79     pEsd->AddTrack(&trk);
80   }//stack loop  
81 }//EsdFromStack()
82 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
83 void SimHits(AliESD *pEsd, TClonesArray *pHits)
84 {
85   const Int_t kCerenkov=50000050;
86   const Int_t kFeedback=50000051;
87   
88   AliHMPIDRecon rec;
89   
90   Int_t hc=0; 
91   for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){//tracks loop
92     AliESDtrack *pTrk=pEsd->GetTrack(iTrk);
93     Float_t xRa,yRa;
94     Int_t ch=AliHMPIDTracker::IntTrkCha(pTrk,xRa,yRa);
95     if(ch<0) continue; //this track does not hit HMPID
96     Float_t beta = pTrk->GetP()/(TMath::Sqrt(pTrk->GetP()*pTrk->GetP()+0.938*0.938));
97     Float_t ckov=TMath::ACos(1./(beta*1.292));
98
99     Float_t theta,phi,xPc,yPc,; 
100     pTrk->GetHMPIDtrk(xPc,yPc,theta,phi); 
101     rec.SetTrack(xRa,yRa,theta,phi); 
102     
103     if(!AliHMPIDDigit::IsInDead(xPc,yPc)) new((*pHits)[hc++]) AliHMPIDHit(ch,200e-9,kProton  ,iTrk,xPc,yPc);                 //mip hit
104     Int_t nPhots = (Int_t)(20.*TMath::Power(TMath::Sin(ckov),2)/TMath::Power(TMath::Sin(TMath::ACos(1./1.292)),2));
105     for(int i=0;i<nPhots;i++){
106       TVector2 pos;
107       pos=rec.TracePhot(ckov,gRandom->Rndm()*TMath::TwoPi());
108       if(!AliHMPIDDigit::IsInDead(pos.X(),pos.Y())) new((*pHits)[hc++]) AliHMPIDHit(ch,7.5e-9,kCerenkov,iTrk,pos.X(),pos.Y());
109     }                      //photon hits  
110   }//tracks loop    
111 }//SimHits()
112 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
113 void ReadHits(TClonesArray *pHitLst)
114 {
115   pHitLst->Delete();  Int_t cnt=0;
116   for(Int_t iEnt=0;iEnt<gHL->TreeH()->GetEntries();iEnt++){       //TreeH loop
117     gHL->TreeH()->GetEntry(iEnt);                                 //get current entry (prim)                
118     for(Int_t iHit=0;iHit<gH->Hits()->GetEntries();iHit++){       //hits loop
119       AliHMPIDHit *pHit = (AliHMPIDHit*)gH->Hits()->At(iHit);     //get current hit        
120       new((*pHitLst)[cnt++]) AliHMPIDHit(*pHit);
121     }//hits loop for this entry
122   }//tree entries loop
123   
124 }//ReadHits()
125 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
126 void DrawCh(Int_t iCh) 
127
128   gPad->Range(-10,-10,AliHMPIDDigit::SizeAllX()+5,AliHMPIDDigit::SizeAllY()+5); 
129   if(iCh>=0){TLatex txt; txt.SetTextSize(0.1); txt.DrawLatex(-5,-5,Form("%i",iCh));}
130   
131   for(Int_t iPc=AliHMPIDDigit::kMinPc;iPc<=AliHMPIDDigit::kMaxPc;iPc++){
132     TBox *pBox=new TBox(AliHMPIDDigit::fMinPcX[iPc],AliHMPIDDigit::fMinPcY[iPc],
133                         AliHMPIDDigit::fMaxPcX[iPc],AliHMPIDDigit::fMaxPcY[iPc]);
134     
135     pBox->SetFillStyle(0);
136     pBox->Draw();
137     
138   }//PC loop      
139 }//DrawCh()
140 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
141 void DrawEvt(TClonesArray *pHitLst,TObjArray *pDigLst,TObjArray *pCluLst,AliESD *pEsd)
142 {//draws all the objects of current event in given canvas
143   AliHMPIDRecon rec;  
144   TPolyMarker *pTxC[7], *pRin[7]; TMarker *pMip,*pCko,*pFee,*pDig,*pClu;
145   pMip=new TMarker; pMip->SetMarkerColor(kRed);  pMip->SetMarkerStyle(kOpenTriangleUp);
146   pCko=new TMarker; pCko->SetMarkerColor(kRed);  pCko->SetMarkerStyle(kOpenCircle);
147   pFee=new TMarker; pFee->SetMarkerColor(kRed);  pFee->SetMarkerStyle(kOpenDiamond);
148   pDig=new TMarker; pDig->SetMarkerColor(kGreen);pDig->SetMarkerStyle(kOpenSquare);
149   pClu=new TMarker; pClu->SetMarkerColor(kBlue); pClu->SetMarkerStyle(kStar);
150   for(Int_t ch=0;ch<7;ch++){
151     pTxC[ch]=new TPolyMarker; pTxC[ch]->SetMarkerStyle(2); pTxC[ch]->SetMarkerColor(kRed); pTxC[ch]->SetMarkerSize(3);
152     pRin[ch]=new TPolyMarker; pRin[ch]->SetMarkerStyle(6); pRin[ch]->SetMarkerColor(kMagenta);
153   }
154   
155   for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){//tracks loop to collect cerenkov rings and intersection points
156     AliESDtrack *pTrk=pEsd->GetTrack(iTrk);
157     Int_t ch=pTrk->GetHMPIDcluIdx();
158     if(ch<0) continue; //this track does not hit HMPID
159     ch/=1000000; 
160     Float_t th,ph,xRad,yRad; pTrk->GetHMPIDtrk(xRad,yRad,th,ph);//get info on current track
161 //    pTxC[ch]->SetNextPoint(xPc,yPc);                          //add new intersection point  TEMPORARLY DISABLED...no more available in ESD!
162     
163     Float_t ckov=pTrk->GetHMPIDsignal();  Float_t err=TMath::Sqrt(pTrk->GetHMPIDchi2());
164     if(ckov>0){
165       Printf("theta %f phi %f ckov %f",th*TMath::RadToDeg(),ph*TMath::RadToDeg(),ckov);
166       rec.SetTrack(xRad,yRad,th,ph);
167       for(Int_t j=0;j<100;j++){ 
168         TVector2 pos;
169         pos=rec.TracePhot(ckov,j*0.0628);
170        if(!AliHMPIDDigit::IsInDead(pos.X(),pos.Y())) pRin[ch]->SetNextPoint(pos.X(),pos.Y());
171       }      
172     }
173   }//tracks loop
174       
175   Int_t totHit=pHitLst->GetEntriesFast(),totDig=0,totClu=0,totTxC=0;
176   Int_t totCkov=0, totFeed=0,totMip=0;
177   for(Int_t iCh=0;iCh<7;iCh++){//chambers loop    
178     totTxC+=pTxC[iCh]->GetN();
179     totDig+=((TClonesArray*)pDigLst->At(iCh))->GetEntriesFast();
180     totClu+=((TClonesArray*)pCluLst->At(iCh))->GetEntriesFast();
181     
182     switch(iCh){
183       case 6: pAll->cd(1); break; case 5: pAll->cd(2); break;
184       case 4: pAll->cd(4); break; case 3: pAll->cd(5); break; case 2: pAll->cd(6); break;
185                                   case 1: pAll->cd(8); break; case 0: pAll->cd(9); break;
186     }
187     gPad->SetEditable(kTRUE); gPad->Clear(); 
188     DrawCh(iCh);
189                            
190     ((TClonesArray*)pDigLst->At(iCh))->Draw();               //draw digits
191     
192     totCkov=0;totFeed=0;totMip=0;
193     for(Int_t iHit=0;iHit<pHitLst->GetEntriesFast();iHit++){ // Draw hits
194       AliHMPIDHit *pHit=(AliHMPIDHit*)pHitLst->At(iHit);
195       if(pHit->Ch()==iCh) pHit->Draw();
196       if(pHit->Pid()==50000050) totCkov++;
197       else if(pHit->Pid()==50000051) totFeed++;
198       else totMip++;
199     }    
200            
201     ((TClonesArray*)pCluLst->At(iCh))->Draw();              //draw clusters
202                             pTxC[iCh]->Draw();              //draw intersections
203                             pRin[iCh]->Draw("CLP");         //draw rings
204     gPad->SetEditable(kFALSE);
205   }//chambers loop
206   
207   pAll->cd(3);  gPad->Clear(); TLegend *pLeg=new TLegend(0.2,0.2,0.8,0.8);
208                                         pLeg->SetHeader(Form("Event %i Total %i",gEvt,gMaxEvt+1));
209                                         pLeg->AddEntry(pTxC[0],Form("TRKxPC %i"     ,totTxC),"p");
210                                         pLeg->AddEntry(pMip   ,Form("Mip hits %i"   ,totMip),"p");    
211                                         pLeg->AddEntry(pCko   ,Form("Ckov hits %i"  ,totCkov),"p");    
212                                         pLeg->AddEntry(pFee   ,Form("Feed hits %i"  ,totFeed),"p");    
213                                         pLeg->AddEntry(pDig   ,Form("Digs %i"       ,totDig),"p");    
214                                         pLeg->AddEntry(pClu   ,Form("Clus %i"       ,totClu),"p");    
215                                         pLeg->Draw();
216 }//DrawEvt()
217 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
218 void DoZoom(Int_t evt, Int_t px, Int_t py, TObject *obj)
219 {
220   if(evt!=5 && evt!=6) return; //5- zoom in 6-zoom out
221   const Int_t minZoom=64;
222   const Int_t maxZoom=2;
223   static Int_t zoom=minZoom; //zoom level
224   if(evt==5&&zoom==maxZoom) return; 
225   if(evt==6&&zoom==minZoom) return; 
226   
227  // if(!obj->IsA()->InheritsFrom("TPad")) return;  //current object is not pad
228   TVirtualPad *pPad=gPad->GetSelectedPad();
229   if(pPad->GetNumber()==3 || pPad->GetNumber()==7) return; //current pad is wrong
230
231  // Printf("evt=%i (%i,%i) %s",evt,px,py,obj->GetName());
232     
233   Float_t x=pPad->AbsPixeltoX(px); Float_t y=pPad->AbsPixeltoY(py); 
234  
235   if(evt==5){ zoom=zoom/2;     pPad->Range(x-zoom*2,y-zoom*2,x+zoom*2,y+zoom*2);} //zoom in
236   else      { zoom=zoom*2;     pPad->Range(x-zoom*2,y-zoom*2,x+zoom*2,y+zoom*2);} //zoom out 
237   if(zoom==minZoom) pPad->Range(-10,-10,AliHMPIDDigit::SizeAllX()+5,AliHMPIDDigit::SizeAllY()+5);
238   ((TCanvas *)gTQSender)->SetTitle(Form("zoom x%i",minZoom/zoom));
239   pPad->Modified();
240   pPad->Update();                                              
241 }
242 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
243 void PrintHits()
244 {
245 //Prints a list of HMPID hits for a given event. Default is event number 0.
246   Printf("List of HMPID hits for event %i",gEvt);
247   
248   Int_t iTotHits=0;
249   for(Int_t iPrim=0;iPrim<gHL->TreeH()->GetEntries();iPrim++){//prims loop
250     gHL->TreeH()->GetEntry(iPrim);      
251     gH->Hits()->Print();
252     iTotHits+=gH->Hits()->GetEntries();
253   }
254   Printf("totally %i hits for event %i",iTotHits,gEvt);
255 }
256 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
257 void PrintSdis()
258 {
259 //prints a list of HMPID sdigits  for a given event
260   Printf("List of HMPID sdigits for event %i",gEvt);
261   gH->SdiLst()->Print();
262   Printf("totally %i sdigits for event %i",gH->SdiLst()->GetEntries(),gEvt);
263 }
264 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
265 void PrintDigs()
266 {
267 //prints a list of HMPID digits
268   Printf("List of HMPID digits for event %i",gEvt);  
269   gH->DigLst()->Print();
270   Int_t totDigs=0;  for(Int_t i=0;i<7;i++) {totDigs+=gH->DigLst(i)->GetEntries();}
271   Printf("totally %i digits for event %i",totDigs,gEvt);
272 }
273 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
274 void PrintClus()
275 {//prints a list of HMPID clusters  for a given event
276   Printf("List of HMPID clusters for event %i",gEvt);
277   gH->CluLst()->Print();
278   
279   Int_t iCluCnt=0; for(Int_t iCh=0;iCh<7;iCh++) iCluCnt+=gH->CluLst(iCh)->GetEntries();
280   
281   Printf("totally %i clusters for event %i",iCluCnt,gEvt);
282 }
283 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
284 void OpenCalib()
285 {
286   AliCDBManager* pCDB = AliCDBManager::Instance();
287   pCDB->SetDefaultStorage("local://$HOME");
288   AliCDBEntry *pQthreEnt=pCDB->Get("HMPID/Calib/Qthre",0);
289   AliCDBEntry *pNmeanEnt=pCDB->Get("HMPID/Calib/Nmean",0);
290   
291   if(!pQthreEnt || ! pNmeanEnt) return;
292   
293   pNmean=(TObjArray*)pNmeanEnt->GetObject(); 
294 }