X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=HMPID%2FHESDfromKin.C;h=f0694adff2a6daadddb9c413991de50ec647a715;hb=307f55b2f3b75279165dce06731be7b0ad11753d;hp=d5ee5ab672ae1ef10aba3f4897a6b1d5c039b2e6;hpb=c0d7adf8ec1430dc8fdbe05aad513e0a2ddeb617;p=u%2Fmrichter%2FAliRoot.git diff --git a/HMPID/HESDfromKin.C b/HMPID/HESDfromKin.C index d5ee5ab672a..f0694adff2a 100644 --- a/HMPID/HESDfromKin.C +++ b/HMPID/HESDfromKin.C @@ -1,9 +1,9 @@ AliRunLoader *gAL=0; Int_t gEvt=0; Int_t gMaxEvt=0; -TObjArray *pNmean; +TObjArray *pNmean,*pQthre; TTree *gEsdTr; //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ -void HESDfromKin(const char *name) +void HESDfromKin(const char *name="default") {//simulate ESD from kinematics if(gSystem->IsFileInIncludePath("galice.root")){// tries to open session @@ -15,107 +15,121 @@ void HESDfromKin(const char *name) AliLoader *pHL=gAL->GetDetectorLoader("HMPID"); pHL->LoadRecPoints(); - AliESD *pEsd = new AliESD(); + AliESDEvent *pEsd = new AliESDEvent(); TFile *pEsdFl=TFile::Open("AliESDs.root","recreate"); gEsdTr=new TTree("esdTree","Sim ESD from kinematics"); - gEsdTr->Branch("ESD", &pEsd); - + pEsd->CreateStdContent(); pEsd->WriteToTree(gEsdTr); //clm: new ESD write schema: see Task Force meeting 20th June, 2007 + gEsdTr->GetUserInfo()->Add(pEsd); //clm: TList has to be created for ReadFromTree method -- this was not needed by the old ESD + + } else return; - OpenCalib(); + if(!OpenCalib()) {Printf("Problems in OpenCalib!Bye.");return;} TString ttl=name; Bool_t htaCheck=ttl.Contains("HTA"); - if(!htaCheck) SimEsd(pHL,pEsd); else SimEsdHidden(pHL,pEsd); +// if(!htaCheck) SimEsd(pHL,pEsd); else SimEsdHidden(pHL,pEsd); + SimEsd(pHL,pEsd,htaCheck); + + pEsdFl->cd(); pEsdFl->Write();pEsdFl->Close(); } //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ -void SimEsd(AliLoader *pHL,AliESD *pEsd) +void SimEsd(AliLoader *pHL,AliESDEvent *pEsd,Bool_t htaCheck) { - Printf("-----------------------------------------------"); - Printf("| SimESD: Utility to embed ESD from kinematics|"); - Printf("-----------------------------------------------"); - AliHMPIDTracker::SetFieldMap(gAL->GetAliRun()->Field(),kTRUE); - AliHMPID *pH=(AliHMPID*)gAL->GetAliRun()->GetDetector("HMPID"); - Int_t mtid=-1; - Int_t iNevt=gAL->GetNumberOfEvents(); - Printf("Number of events to process: %i",iNevt); - for(Int_t iEvt=0;iEvtGetEvent(iEvt); - pHL->TreeR()->GetEntry(0); - AliStack *pStack=gAL->Stack(); - for(Int_t i=0;iGetNtrack();i++){ - TParticle *pTrack=pStack->Particle(i); - mtid=pTrack->GetFirstMother(); - if(mtid>=0) continue; // only primaries - AliESDtrack trk(pTrack); - pEsd->AddTrack(&trk); - AliHMPIDTracker::Recon(pEsd,pH->CluLst(),pNmean); - }// track loop - pEsd->SetMagneticField(AliHMPIDTracker::GetBz()); - gEsdTr->Fill(); - pEsd->Reset(); - }// event loop - Printf("Events processed %i",iEvt); - gAL->UnloadHeader(); gAL->UnloadKinematics(); -}//Esd() -//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ -void SimEsdHidden(AliLoader *pHL,AliESD *pEsd) -{ - Double_t rd=TMath::RadToDeg(); - Printf("----------------------------------------------"); - Printf("| SimHTA:Utility to embed ESD from kinematics|"); - Printf("| with Hidden Track Algorithm (HTA) |"); - Printf("----------------------------------------------"); - AliHMPIDTracker::SetFieldMap(gAL->GetAliRun()->Field(),kTRUE); + if(htaCheck) { + TFile *fout = new TFile("HTA.root","recreate"); + TH1F *hdC = new TH1F("dC" ,";delta Cerenkov (rad)",100,-0.2,0.2); + TH1F *hCer = new TH1F("Cer" ,"Theta Cerenkov (rad)",250,0.,0.75); + TH2F *htvsp = new TH2F("tvsp",";momentum (GeV/c);theta Cerenkov (rad)",100,0.,5.,1000,0.,0.75); + TH1F *hdth = new TH1F("dth" ,";Delta theta Trk (mrad)",100,-250,250); + TH1F *hdph = new TH1F("dph" ,";Delta phi Trk (mrad)",100,-500,500); + Double_t rd=TMath::RadToDeg(); + Printf("----------------------------------------------"); + Printf("| SimHTA:Utility to embed ESD from kinematics|"); + Printf("| with Hidden Track Algorithm (HTA) |"); + Printf("----------------------------------------------"); + } else { + Printf("-----------------------------------------------"); + Printf("| SimESD: Utility to embed ESD from kinematics|"); + Printf("-----------------------------------------------"); +} + + AliGRPManager *grpMan = new AliGRPManager(); + + grpMan->ReadGRPEntry(); + grpMan->SetMagField(); + + AliHMPIDTracker pTracker; AliHMPID *pH=(AliHMPID*)gAL->GetAliRun()->GetDetector("HMPID"); - Int_t mtid=-1; Int_t iNevt=gAL->GetNumberOfEvents(); - Printf("Number of events to process: %i",iNevt); + pEsd->SetMagneticField(AliHMPIDTracker::GetBz()); for(Int_t iEvt=0;iEvtGetEvent(iEvt); pHL->TreeR()->GetEntry(0); AliStack *pStack=gAL->Stack(); + Int_t nTrkHMPID=0; + for(Int_t i=0;iGetNtrack();i++){ + if(!pStack->IsPhysicalPrimary(i)) continue; TParticle *pTrack=pStack->Particle(i); - mtid=pTrack->GetFirstMother(); - if(mtid>=0) continue; // only primaries - //find the chamber that intersects HMPID - AliESDtrack trk(pTrack); - Float_t xPc,yPc; - Int_t iCh=AliHMPIDTracker::IntTrkCha(&trk,xPc,yPc); //get chamber intersected by this track - if(iCh<0) continue; //no intersection at all, go after next track - Float_t radX,radY,thetaTrk,phiTrk; - trk.GetHMPIDtrk(radX,radY,thetaTrk,phiTrk); - Printf("simulated track theta %f phi %f",thetaTrk*rd,phiTrk*rd); - TObjArray *pClus = pH->CluLst(); - if(AliHMPIDTracker::ReconHiddenTrk(iCh,&trk,(TClonesArray *)pClus->At(iCh),pNmean)!=0) continue; - trk.GetHMPIDtrk(radX,radY,thetaTrk,phiTrk); - Printf("reconstr. track theta %f phi %f",thetaTrk*rd,phiTrk*rd); + if(pTrack->GetPDG()->Charge()==0) continue; + Printf("track n. %i",i); + AliESDtrack trk(pTrack); + Float_t xPc,yPc,xRa,yRa,thRa,phRa; + Int_t iCh=pTracker.IntTrkCha(&trk,xPc,yPc,xRa,yRa,thRa,phRa); //get chamber intersected by this track + if(iCh<0) { + trk.SetHMPIDtrk(0,0,0,0); //no intersection found + trk.SetHMPIDcluIdx (99,99999); //chamber not found, mip not yet considered + trk.SetHMPIDsignal(AliHMPIDRecon::kNotPerformed); //ring reconstruction not yet performed + continue; //no intersection at all, go after next track + } + nTrkHMPID++; + trk.SetHMPIDcluIdx (iCh,99999); //chamber not found, mip not yet considered + + if(phRa<0) phRa += TMath::TwoPi(); // to be verified + + trk.SetHMPIDtrk(xPc,yPc,thRa,phRa); //store initial infos pEsd->AddTrack(&trk); - Double_t thetaCerSim = TMath::ACos(pTrack->Energy()/(1.292*pTrack->P())); - Printf(" theta Cerenkov simulated %f",thetaCerSim); - Printf(" theta Cerenkov reconstructed %f",trk.GetHMPIDsignal()); + + Int_t status; + if(!htaCheck) status = pTracker.Recon (pEsd,pH->CluLst(),pNmean,pQthre); + else status = pTracker.ReconHiddenTrk(pEsd,pH->CluLst(),pNmean,pQthre); + +// Printf("status %i",status); + if(status !=0) continue; + + }// track loop - pEsd->SetMagneticField(AliHMPIDTracker::GetBz()); + + if(!(iEvt%50)) Printf("Number of events processed: %i with tracks %i in HMPID",iEvt,nTrkHMPID); +// Printf("Number of events processed: %i with tracks %i in HMPID",iEvt,nTrkHMPID); + gEsdTr->Fill(); pEsd->Reset(); }// event loop Printf("Events processed %i",iEvt); + if(htaCheck) { + fout->Write(); + fout->Close(); + delete fout; + } gAL->UnloadHeader(); gAL->UnloadKinematics(); -}//EsdHidden() +}//SimEsd() //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ -void OpenCalib() +Bool_t OpenCalib() { AliCDBManager* pCDB = AliCDBManager::Instance(); - pCDB->SetDefaultStorage("local://$HOME"); + pCDB->SetDefaultStorage("local://$ALICE_ROOT/OCDB"); + pCDB->SetRun(0); AliCDBEntry *pQthreEnt=pCDB->Get("HMPID/Calib/Qthre",0); AliCDBEntry *pNmeanEnt=pCDB->Get("HMPID/Calib/Nmean",0); - if(!pQthreEnt || ! pNmeanEnt) return; + if(!pQthreEnt || !pNmeanEnt) return kFALSE; pNmean=(TObjArray*)pNmeanEnt->GetObject(); + pQthre=(TObjArray*)pQthreEnt->GetObject(); + + if(!pQthre || !pNmean) return kFALSE; + return kTRUE; } -//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++