Moving the copy constructors and assignment operators to private: and removing the...
[u/mrichter/AliRoot.git] / HMPID / HESDfromKin.C
CommitLineData
d8770a0f 1AliRunLoader *gAL=0;
2Int_t gEvt=0; Int_t gMaxEvt=0;
3df18479 3TObjArray *pNmean,*pQthre;
d8770a0f 4TTree *gEsdTr;
5//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
e4290ede 6void HESDfromKin(const char *name="default")
d8770a0f 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();
6e06db1d 18 AliESDEvent *pEsd = new AliESDEvent();
d8770a0f 19 TFile *pEsdFl=TFile::Open("AliESDs.root","recreate");
20 gEsdTr=new TTree("esdTree","Sim ESD from kinematics");
e4290ede 21 pEsd->CreateStdContent(); pEsd->WriteToTree(gEsdTr); //clm: new ESD write schema: see Task Force meeting 20th June, 2007
22 gEsdTr->GetUserInfo()->Add(pEsd); //clm: TList has to be created for ReadFromTree method -- this was not needed by the old ESD
23
24
d8770a0f 25 } else return;
26
3df18479 27 if(!OpenCalib()) {Printf("Problems in OpenCalib!Bye.");return;}
d8770a0f 28
c0d7adf8 29 TString ttl=name;
30 Bool_t htaCheck=ttl.Contains("HTA");
9785d5fb 31// if(!htaCheck) SimEsd(pHL,pEsd); else SimEsdHidden(pHL,pEsd);
32 SimEsd(pHL,pEsd,htaCheck);
e4290ede 33
34 pEsdFl->cd();
d8770a0f 35 pEsdFl->Write();pEsdFl->Close();
36}
37//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
9785d5fb 38void SimEsd(AliLoader *pHL,AliESDEvent *pEsd,Bool_t htaCheck)
d8770a0f 39{
9785d5fb 40 if(htaCheck) {
41 TFile *fout = new TFile("HTA.root","recreate");
42 TH1F *hdC = new TH1F("dC" ,";delta Cerenkov (rad)",100,-0.2,0.2);
43 TH1F *hCer = new TH1F("Cer" ,"Theta Cerenkov (rad)",250,0.,0.75);
44 TH2F *htvsp = new TH2F("tvsp",";momentum (GeV/c);theta Cerenkov (rad)",100,0.,5.,1000,0.,0.75);
45 TH1F *hdth = new TH1F("dth" ,";Delta theta Trk (mrad)",100,-250,250);
46 TH1F *hdph = new TH1F("dph" ,";Delta phi Trk (mrad)",100,-500,500);
47 Double_t rd=TMath::RadToDeg();
48 Printf("----------------------------------------------");
49 Printf("| SimHTA:Utility to embed ESD from kinematics|");
50 Printf("| with Hidden Track Algorithm (HTA) |");
51 Printf("----------------------------------------------");
52 } else {
53 Printf("-----------------------------------------------");
54 Printf("| SimESD: Utility to embed ESD from kinematics|");
55 Printf("-----------------------------------------------");
56}
efc29373 57
106794e3 58 AliGRPManager *grpMan = new AliGRPManager();
59
60 grpMan->ReadGRPEntry();
61 grpMan->SetMagField();
efc29373 62
e56b695f 63 AliHMPIDTracker pTracker;
d8770a0f 64 AliHMPID *pH=(AliHMPID*)gAL->GetAliRun()->GetDetector("HMPID");
d8770a0f 65 Int_t iNevt=gAL->GetNumberOfEvents();
1b46ef29 66 pEsd->SetMagneticField(AliHMPIDTracker::GetBz());
d8770a0f 67 for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//events loop
d8770a0f 68 gAL->GetEvent(iEvt);
69 pHL->TreeR()->GetEntry(0);
70 AliStack *pStack=gAL->Stack();
9785d5fb 71 Int_t nTrkHMPID=0;
2b919fd5 72
d8770a0f 73 for(Int_t i=0;i<pStack->GetNtrack();i++){
9785d5fb 74 if(!pStack->IsPhysicalPrimary(i)) continue;
d8770a0f 75 TParticle *pTrack=pStack->Particle(i);
1b46ef29 76 if(pTrack->GetPDG()->Charge()==0) continue;
6b938944 77// Printf("track n. %i",i);
e4290ede 78 AliESDtrack trk(pTrack);
39cd22e6 79 Float_t xPc,yPc,xRa,yRa,thRa,phRa;
e56b695f 80 Int_t iCh=pTracker.IntTrkCha(&trk,xPc,yPc,xRa,yRa,thRa,phRa); //get chamber intersected by this track
39cd22e6 81 if(iCh<0) {
82 trk.SetHMPIDtrk(0,0,0,0); //no intersection found
83 trk.SetHMPIDcluIdx (99,99999); //chamber not found, mip not yet considered
84 trk.SetHMPIDsignal(AliHMPIDRecon::kNotPerformed); //ring reconstruction not yet performed
85 continue; //no intersection at all, go after next track
86 }
9785d5fb 87 nTrkHMPID++;
39cd22e6 88 trk.SetHMPIDcluIdx (iCh,99999); //chamber not found, mip not yet considered
e56b695f 89
90 if(phRa<0) phRa += TMath::TwoPi(); // to be verified
91
97a4d538 92 trk.SetHMPIDtrk(xPc,yPc,thRa,phRa); //store initial infos
6b938944 93 trk.SetLabel(i);
d8770a0f 94 pEsd->AddTrack(&trk);
9785d5fb 95
6574baa2 96 Int_t status;
97 if(!htaCheck) status = pTracker.Recon (pEsd,pH->CluLst(),pNmean,pQthre);
98 else status = pTracker.ReconHiddenTrk(pEsd,pH->CluLst(),pNmean,pQthre);
99
2b919fd5 100// Printf("status %i",status);
6574baa2 101 if(status !=0) continue;
9785d5fb 102
ac66a50d 103
6574baa2 104 }// track loop
2b919fd5 105
106 if(!(iEvt%50)) Printf("Number of events processed: %i with tracks %i in HMPID",iEvt,nTrkHMPID);
107// Printf("Number of events processed: %i with tracks %i in HMPID",iEvt,nTrkHMPID);
108
d8770a0f 109 gEsdTr->Fill();
110 pEsd->Reset();
111 }// event loop
112 Printf("Events processed %i",iEvt);
204299b8 113 if(htaCheck) {
114 fout->Write();
115 fout->Close();
116 delete fout;
117 }
d8770a0f 118 gAL->UnloadHeader(); gAL->UnloadKinematics();
e56b695f 119}//SimEsd()
d8770a0f 120//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
3df18479 121Bool_t OpenCalib()
d8770a0f 122{
123 AliCDBManager* pCDB = AliCDBManager::Instance();
efc29373 124 pCDB->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
125 pCDB->SetRun(0);
d8770a0f 126 AliCDBEntry *pQthreEnt=pCDB->Get("HMPID/Calib/Qthre",0);
127 AliCDBEntry *pNmeanEnt=pCDB->Get("HMPID/Calib/Nmean",0);
128
3df18479 129 if(!pQthreEnt || !pNmeanEnt) return kFALSE;
d8770a0f 130
131 pNmean=(TObjArray*)pNmeanEnt->GetObject();
3df18479 132 pQthre=(TObjArray*)pQthreEnt->GetObject();
133
134 if(!pQthre || !pNmean) return kFALSE;
135 return kTRUE;
d8770a0f 136}