HMPID/HESDfromKin.C

   1 AliRunLoader *gAL=0;
   2 Int_t gEvt=0; Int_t gMaxEvt=0;
   3 TObjArray *pNmean;
   4 TTree *gEsdTr;
   5 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   6 void HESDfromKin()
   7 {//simulate ESD from kinematics
   8
   9   if(gSystem->IsFileInIncludePath("galice.root")){// tries to open session
  10     if(gAlice) delete gAlice;                                               //in case we execute this in aliroot delete default AliRun object
  11     gAL=AliRunLoader::Open();                                                                    //try to open galice.root from current dir
  12     gAL->LoadgAlice();                                                                           //take new AliRun object from galice.root
  13     if(gAL->LoadHeader()) return;
  14     if(gAL->LoadKinematics()) return;
  15
  16     AliLoader *pHL=gAL->GetDetectorLoader("HMPID");
  17     pHL->LoadRecPoints();
  18     AliESD *pEsd = new AliESD();
  19     TFile *pEsdFl=TFile::Open("AliESDs.root","recreate");
  20     gEsdTr=new TTree("esdTree","Sim ESD from kinematics");
  21     gEsdTr->Branch("ESD", &pEsd);
  22
  23   }  else return;
  24
  25   OpenCalib();
  26
  27 //  SimEsd(pHL,pEsd);
  28   SimEsdHidden(pHL,pEsd);
  29
  30   pEsdFl->Write();pEsdFl->Close();
  31 }
  32 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
  33 void SimEsd(AliLoader *pHL,AliESD *pEsd)
  34 {
  35   Printf("---------------------------------------");
  36   Printf("| Utility to embed ESD from kinematics|");
  37   Printf("---------------------------------------");
  38   AliHMPIDTracker::SetFieldMap(gAL->GetAliRun()->Field(),kTRUE);
  39   AliHMPID *pH=(AliHMPID*)gAL->GetAliRun()->GetDetector("HMPID");
  40   Int_t mtid=-1;
  41   Int_t iNevt=gAL->GetNumberOfEvents();
  42   Printf("Number of events to process: %i",iNevt);
  43   for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//events loop
  44     if(!(iEvt%50)) Printf("Events processed %i",iEvt);
  45     gAL->GetEvent(iEvt);
  46     pHL->TreeR()->GetEntry(0);
  47     AliStack *pStack=gAL->Stack();
  48     for(Int_t i=0;i<pStack->GetNtrack();i++){
  49       TParticle *pTrack=pStack->Particle(i);
  50       mtid=pTrack->GetFirstMother();
  51       if(mtid>=0) continue; // only primaries
  52       AliESDtrack trk(pTrack);
  53       pEsd->AddTrack(&trk);
  54       AliHMPIDTracker::Recon(pEsd,pH->CluLst(),pNmean);
  55     }// track loop
  56     pEsd->SetMagneticField(AliHMPIDTracker::GetBz());
  57     gEsdTr->Fill();
  58     pEsd->Reset();
  59   }// event loop
  60   Printf("Events processed %i",iEvt);
  61   gAL->UnloadHeader();  gAL->UnloadKinematics();
  62 }//Esd()
  63 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
  64 void SimEsdHidden(AliLoader *pHL,AliESD *pEsd)
  65 {
  66   Double_t rd=TMath::RadToDeg();
  67   Printf("---------------------------------------");
  68   Printf("| Utility to embed ESD from kinematics|");
  69   Printf("---------------------------------------");
  70   AliHMPIDTracker::SetFieldMap(gAL->GetAliRun()->Field(),kTRUE);
  71   AliHMPID *pH=(AliHMPID*)gAL->GetAliRun()->GetDetector("HMPID");
  72   Int_t mtid=-1;
  73   Int_t iNevt=gAL->GetNumberOfEvents();
  74   Printf("Number of events to process: %i",iNevt);
  75   for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//events loop
  76     if(!(iEvt%50)) Printf("Events processed %i",iEvt);
  77     gAL->GetEvent(iEvt);
  78     pHL->TreeR()->GetEntry(0);
  79     AliStack *pStack=gAL->Stack();
  80     for(Int_t i=0;i<pStack->GetNtrack();i++){
  81       TParticle *pTrack=pStack->Particle(i);
  82       mtid=pTrack->GetFirstMother();
  83       if(mtid>=0) continue; // only primaries
  84       //find the chamber that intersects HMPID
  85       AliESDtrack trk(pTrack);
  86       Float_t xPc,yPc;
  87       Int_t iCh=AliHMPIDTracker::IntTrkCha(&trk,xPc,yPc);                           //get chamber intersected by this track
  88       if(iCh<0) continue;                                                           //no intersection at all, go after next track
  89       Float_t radX,radY,thetaTrk,phiTrk;
  90       trk.GetHMPIDtrk(radX,radY,thetaTrk,phiTrk);
  91       Printf("simulated track theta %f phi %f",thetaTrk*rd,phiTrk*rd);
  92       TObjArray *pClus = pH->CluLst();
  93       if(AliHMPIDTracker::ReconHiddenTrk(iCh,&trk,(TClonesArray *)pClus->At(iCh),pNmean)!=0) continue;
  94       trk.GetHMPIDtrk(radX,radY,thetaTrk,phiTrk);
  95       Printf("reconstr. track theta %f phi %f",thetaTrk*rd,phiTrk*rd);
  96       pEsd->AddTrack(&trk);
  97       Double_t thetaCerSim = TMath::ACos(pTrack->Energy()/(1.292*pTrack->P()));
  98       Printf(" theta Cerenkov simulated     %f",thetaCerSim);
  99       Printf(" theta Cerenkov reconstructed %f",trk.GetHMPIDsignal());
 100     }// track loop
 101     pEsd->SetMagneticField(AliHMPIDTracker::GetBz());
 102     gEsdTr->Fill();
 103     pEsd->Reset();
 104   }// event loop
 105   Printf("Events processed %i",iEvt);
 106   gAL->UnloadHeader();  gAL->UnloadKinematics();
 107 }//EsdHidden()
 108 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 109 void OpenCalib()
 110 {
 111   AliCDBManager* pCDB = AliCDBManager::Instance();
 112   pCDB->SetDefaultStorage("local://$HOME");
 113   AliCDBEntry *pQthreEnt=pCDB->Get("HMPID/Calib/Qthre",0);
 114   AliCDBEntry *pNmeanEnt=pCDB->Get("HMPID/Calib/Nmean",0);
 115
 116   if(!pQthreEnt || ! pNmeanEnt) return;
 117
 118   pNmean=(TObjArray*)pNmeanEnt->GetObject();
 119 }
 120 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++