]>
Commit | Line | Data |
---|---|---|
d8770a0f | 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 | ||
3b49956b | 27 | // SimEsd(pHL,pEsd); |
28 | SimEsdHidden(pHL,pEsd); | |
d8770a0f | 29 | |
30 | pEsdFl->Write();pEsdFl->Close(); | |
31 | } | |
32 | //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ | |
33 | void SimEsd(AliLoader *pHL,AliESD *pEsd) | |
34 | { | |
34058145 | 35 | Printf("---------------------------------------"); |
36 | Printf("| Utility to embed ESD from kinematics|"); | |
37 | Printf("---------------------------------------"); | |
d8770a0f | 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(); | |
3b49956b | 62 | }//Esd() |
d8770a0f | 63 | //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ |
64 | void SimEsdHidden(AliLoader *pHL,AliESD *pEsd) | |
3b49956b | 65 | { |
66 | Double_t rd=TMath::RadToDeg(); | |
67 | Printf("---------------------------------------"); | |
68 | Printf("| Utility to embed ESD from kinematics|"); | |
69 | Printf("---------------------------------------"); | |
d8770a0f | 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); | |
3b49956b | 77 | gAL->GetEvent(iEvt); |
d8770a0f | 78 | pHL->TreeR()->GetEntry(0); |
3b49956b | 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 | |
d8770a0f | 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(); | |
3b49956b | 107 | }//EsdHidden() |
d8770a0f | 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 | //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ |