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; TVector2 pos;
  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     /*
 105     for(int i=0;i<4;i++) {
 106       Float_t x=gRandom->Rndm()*130;Float_t y=gRandom->Rndm()*126;
 107       if(!AliHMPIDDigit::IsInDead(x,y)) new((*pHits)[hc++]) AliHMPIDHit(ch,7.5e-9,kFeedback,iTrk,x,y); //bkg hits 4 per track
 108     }
 109     */
 110     Int_t nPhots = (Int_t)(20.*TMath::Power(TMath::Sin(ckov),2)/TMath::Power(TMath::Sin(TMath::ACos(1./1.292)),2));
 111     for(int i=0;i<nPhots;i++){
 112       rec.TracePhot(ckov,gRandom->Rndm()*TMath::TwoPi(),pos);
 113       if(!AliHMPIDDigit::IsInDead(pos.X(),pos.Y())) new((*pHits)[hc++]) AliHMPIDHit(ch,7.5e-9,kCerenkov,iTrk,pos.X(),pos.Y());
 114     }                      //photon hits
 115   }//tracks loop
 116 }//SimHits()
 117 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 118 void ReadHits(TClonesArray *pHitLst)
 119 {
 120   pHitLst->Delete();  Int_t cnt=0;
 121   for(Int_t iEnt=0;iEnt<gHL->TreeH()->GetEntries();iEnt++){       //TreeH loop
 122     gHL->TreeH()->GetEntry(iEnt);                                 //get current entry (prim)
 123     for(Int_t iHit=0;iHit<gH->Hits()->GetEntries();iHit++){       //hits loop
 124       AliHMPIDHit *pHit = (AliHMPIDHit*)gH->Hits()->At(iHit);     //get current hit
 125       new((*pHitLst)[cnt++]) AliHMPIDHit(*pHit);
 126     }//hits loop for this entry
 127   }//tree entries loop
 128
 129 }//ReadHits()
 130 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 131 void DrawCh(Int_t iCh)
 132 {
 133   gPad->Range(-10,-10,AliHMPIDDigit::SizeAllX()+5,AliHMPIDDigit::SizeAllY()+5);
 134   if(iCh>=0){TLatex txt; txt.SetTextSize(0.1); txt.DrawLatex(-5,-5,Form("%i",iCh));}
 135
 136   for(Int_t iPc=AliHMPIDDigit::kMinPc;iPc<=AliHMPIDDigit::kMaxPc;iPc++){
 137     TBox *pBox=new TBox(AliHMPIDDigit::fMinPcX[iPc],AliHMPIDDigit::fMinPcY[iPc],
 138                         AliHMPIDDigit::fMaxPcX[iPc],AliHMPIDDigit::fMaxPcY[iPc]);
 139
 140     pBox->SetFillStyle(0);
 141     pBox->Draw();
 142
 143   }//PC loop
 144 }//DrawCh()
 145 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 146 void DrawEvt(TClonesArray *pHitLst,TObjArray *pDigLst,TObjArray *pCluLst,AliESD *pEsd)
 147 {//draws all the objects of current event in given canvas
 148
 149   AliHMPIDRecon rec;
 150   TPolyMarker *pTxC[7], *pRin[7]; TMarker *pMip,*pCko,*pFee,*pDig,*pClu;
 151   pMip=new TMarker; pMip->SetMarkerColor(kRed);  pMip->SetMarkerStyle(kOpenTriangleUp);
 152   pCko=new TMarker; pCko->SetMarkerColor(kRed);  pCko->SetMarkerStyle(kOpenCircle);
 153   pFee=new TMarker; pFee->SetMarkerColor(kRed);  pFee->SetMarkerStyle(kOpenDiamond);
 154   pDig=new TMarker; pDig->SetMarkerColor(kGreen);pDig->SetMarkerStyle(kOpenSquare);
 155   pClu=new TMarker; pClu->SetMarkerColor(kBlue); pClu->SetMarkerStyle(kStar);
 156   for(Int_t ch=0;ch<7;ch++){
 157     pTxC[ch]=new TPolyMarker; pTxC[ch]->SetMarkerStyle(2); pTxC[ch]->SetMarkerColor(kRed); pTxC[ch]->SetMarkerSize(3);
 158     pRin[ch]=new TPolyMarker; pRin[ch]->SetMarkerStyle(6); pRin[ch]->SetMarkerColor(kMagenta);
 159   }
 160
 161   for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){//tracks loop to collect cerenkov rings and intersection points
 162     AliESDtrack *pTrk=pEsd->GetTrack(iTrk);
 163     Int_t ch=pTrk->GetHMPIDcluIdx();
 164     if(ch<0) continue; //this track does not hit HMPID
 165     ch/=1000000;
 166     Float_t th,ph,xPc,yPc; pTrk->GetHMPIDtrk(xPc,yPc,th,ph);  //get info on current track
 167     pTxC[ch]->SetNextPoint(xPc,yPc);                          //add new intersection point
 168
 169     Float_t ckov=pTrk->GetHMPIDsignal();  Float_t err=TMath::Sqrt(pTrk->GetHMPIDchi2());
 170     if(ckov>0){
 171       rec.SetTrack(xPc,yPc,th,ph);
 172       TVector2 pos;
 173       for(int j=0;j<100;j++){
 174        rec.TracePhot(ckov,j*0.0628,pos);
 175        if(!AliHMPIDDigit::IsInDead(pos.X(),pos.Y())) pRin[ch]->SetNextPoint(pos.X(),pos.Y());
 176       }
 177     }
 178   }//tracks loop
 179
 180   Int_t totHit=pHitLst->GetEntriesFast(),totDig=0,totClu=0,totTxC=0;
 181   Int_t totCkov=0, totFeed=0,totMip=0;
 182   for(Int_t iCh=0;iCh<7;iCh++){//chambers loop
 183     totTxC+=pTxC[iCh]->GetN();
 184     totDig+=((TClonesArray*)pDigLst->At(iCh))->GetEntriesFast();
 185     totClu+=((TClonesArray*)pCluLst->At(iCh))->GetEntriesFast();
 186
 187     switch(iCh){
 188       case 6: pAll->cd(1); break; case 5: pAll->cd(2); break;
 189       case 4: pAll->cd(4); break; case 3: pAll->cd(5); break; case 2: pAll->cd(6); break;
 190                                   case 1: pAll->cd(8); break; case 0: pAll->cd(9); break;
 191     }
 192     gPad->SetEditable(kTRUE); gPad->Clear();
 193     DrawCh(iCh);
 194
 195     ((TClonesArray*)pDigLst->At(iCh))->Draw();               //draw digits
 196
 197     totCkov=0;totFeed=0;totMip=0;
 198     for(Int_t iHit=0;iHit<pHitLst->GetEntriesFast();iHit++){ // Draw hits
 199       AliHMPIDHit *pHit=(AliHMPIDHit*)pHitLst->At(iHit);
 200       if(pHit->Ch()==iCh) pHit->Draw();
 201       if(pHit->Pid()==50000050) totCkov++;
 202       else if(pHit->Pid()==50000051) totFeed++;
 203       else totMip++;
 204     }
 205
 206     ((TClonesArray*)pCluLst->At(iCh))->Draw();              //draw clusters
 207                             pTxC[iCh]->Draw();              //draw intersections
 208                             pRin[iCh]->Draw("CLP");         //draw rings
 209     gPad->SetEditable(kFALSE);
 210   }//chambers loop
 211
 212   pAll->cd(3);  gPad->Clear(); TLegend *pLeg=new TLegend(0.2,0.2,0.8,0.8);
 213                                         pLeg->SetHeader(Form("Event %i Total %i",gEvt,gMaxEvt+1));
 214                                         pLeg->AddEntry(pTxC[0],Form("TRKxPC %i"     ,totTxC),"p");
 215                                         pLeg->AddEntry(pMip   ,Form("Mip hits %i"   ,totMip),"p");
 216                                         pLeg->AddEntry(pCko   ,Form("Ckov hits %i"  ,totCkov),"p");
 217                                         pLeg->AddEntry(pFee   ,Form("Feed hits %i"  ,totFeed),"p");
 218                                         pLeg->AddEntry(pDig   ,Form("Digs %i"       ,totDig),"p");
 219                                         pLeg->AddEntry(pClu   ,Form("Clus %i"       ,totClu),"p");
 220                                         pLeg->Draw();
 221 }//DrawEvt()
 222 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 223 void DoZoom(Int_t evt, Int_t px, Int_t py, TObject *obj)
 224 {
 225   if(evt!=5 && evt!=6) return; //5- zoom in 6-zoom out
 226   const Int_t minZoom=64;
 227   const Int_t maxZoom=2;
 228   static Int_t zoom=minZoom; //zoom level
 229   if(evt==5&&zoom==maxZoom) return;
 230   if(evt==6&&zoom==minZoom) return;
 231
 232  // if(!obj->IsA()->InheritsFrom("TPad")) return;  //current object is not pad
 233   TVirtualPad *pPad=gPad->GetSelectedPad();
 234   if(pPad->GetNumber()==3 || pPad->GetNumber()==7) return; //current pad is wrong
 235
 236  // Printf("evt=%i (%i,%i) %s",evt,px,py,obj->GetName());
 237
 238   Float_t x=pPad->AbsPixeltoX(px); Float_t y=pPad->AbsPixeltoY(py);
 239
 240   if(evt==5){ zoom=zoom/2;     pPad->Range(x-zoom*2,y-zoom*2,x+zoom*2,y+zoom*2);} //zoom in
 241   else      { zoom=zoom*2;     pPad->Range(x-zoom*2,y-zoom*2,x+zoom*2,y+zoom*2);} //zoom out
 242   if(zoom==minZoom) pPad->Range(-10,-10,AliHMPIDDigit::SizeAllX()+5,AliHMPIDDigit::SizeAllY()+5);
 243   ((TCanvas *)gTQSender)->SetTitle(Form("zoom x%i",minZoom/zoom));
 244   pPad->Modified();
 245   pPad->Update();
 246 }
 247 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 248 void PrintHits()
 249 {
 250 //Prints a list of HMPID hits for a given event. Default is event number 0.
 251   Printf("List of HMPID hits for event %i",gEvt);
 252
 253   Int_t iTotHits=0;
 254   for(Int_t iPrim=0;iPrim<gHL->TreeH()->GetEntries();iPrim++){//prims loop
 255     gHL->TreeH()->GetEntry(iPrim);
 256     gH->Hits()->Print();
 257     iTotHits+=gH->Hits()->GetEntries();
 258   }
 259   Printf("totally %i hits for event %i",iTotHits,gEvt);
 260 }
 261 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 262 void PrintSdis()
 263 {
 264 //prints a list of HMPID sdigits  for a given event
 265   Printf("List of HMPID sdigits for event %i",gEvt);
 266   gH->SdiLst()->Print();
 267   Printf("totally %i sdigits for event %i",gH->SdiLst()->GetEntries(),gEvt);
 268 }
 269 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 270 void PrintDigs()
 271 {
 272 //prints a list of HMPID digits
 273   Printf("List of HMPID digits for event %i",gEvt);
 274   gH->DigLst()->Print();
 275   Int_t totDigs=0;  for(Int_t i=0;i<7;i++) {totDigs+=gH->DigLst(i)->GetEntries();}
 276   Printf("totally %i digits for event %i",totDigs,gEvt);
 277 }
 278 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 279 void PrintClus()
 280 {//prints a list of HMPID clusters  for a given event
 281   Printf("List of HMPID clusters for event %i",gEvt);
 282   gH->CluLst()->Print();
 283
 284   Int_t iCluCnt=0; for(Int_t iCh=0;iCh<7;iCh++) iCluCnt+=gH->CluLst(iCh)->GetEntries();
 285
 286   Printf("totally %i clusters for event %i",iCluCnt,gEvt);
 287 }
 288 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 289 void OpenCalib()
 290 {
 291   AliCDBManager* pCDB = AliCDBManager::Instance();
 292   pCDB->SetDefaultStorage("local://$HOME");
 293   AliCDBEntry *pQthreEnt=pCDB->Get("HMPID/Calib/Qthre",0);
 294   AliCDBEntry *pNmeanEnt=pCDB->Get("HMPID/Calib/Nmean",0);
 295
 296   if(!pQthreEnt || ! pNmeanEnt) return;
 297
 298   pNmean=(TObjArray*)pNmeanEnt->GetObject();
 299 }