]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - HMPID/HESDfromKin.C
Too wide range for charge.
[u/mrichter/AliRoot.git] / HMPID / HESDfromKin.C
index df9b7ebf6ffbbd428b3916b320213ddd4134563e..92570b7aff554a30399c989f3c9b5802301b2f93 100644 (file)
@@ -1,9 +1,9 @@
 AliRunLoader *gAL=0; 
 Int_t gEvt=0; Int_t gMaxEvt=0;
-TObjArray *pNmean;
+TObjArray *pNmean,*pQthre;
 TTree *gEsdTr;
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
-void HESDfromKin()
+void HESDfromKin(const char *name="default")
 {//simulate ESD from kinematics
 
   if(gSystem->IsFileInIncludePath("galice.root")){// tries to open session
@@ -15,24 +15,30 @@ void HESDfromKin()
 
     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;}
     
-  SimEsd(pHL,pEsd);
-//  SimEsdHidden(pHL,pEsd);
+  TString ttl=name;
+  Bool_t htaCheck=ttl.Contains("HTA");
+  if(!htaCheck) SimEsd(pHL,pEsd); else SimEsdHidden(pHL,pEsd);
   
+  pEsdFl->cd();
   pEsdFl->Write();pEsdFl->Close();        
 }
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
-void SimEsd(AliLoader *pHL,AliESD *pEsd)
+void SimEsd(AliLoader *pHL,AliESDEvent *pEsd)
 {
-  Printf("---------okokokok");
+  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;
@@ -47,9 +53,9 @@ void SimEsd(AliLoader *pHL,AliESD *pEsd)
       TParticle *pTrack=pStack->Particle(i); 
       mtid=pTrack->GetFirstMother();
       if(mtid>=0) continue; // only primaries
-      AliESDtrack trk(pTrack);
+      AliESDtrack trk(pTrack); 
       pEsd->AddTrack(&trk);
-      AliHMPIDTracker::Recon(pEsd,pH->CluLst(),pNmean);
+      AliHMPIDTracker::Recon(pEsd,pH->CluLst(),pNmean,pQthre);
     }// track loop
     pEsd->SetMagneticField(AliHMPIDTracker::GetBz());
     gEsdTr->Fill();
@@ -57,10 +63,18 @@ void SimEsd(AliLoader *pHL,AliESD *pEsd)
   }// event loop
   Printf("Events processed %i",iEvt);
   gAL->UnloadHeader();  gAL->UnloadKinematics();
-}//EsdFromStack()
+}//Esd()
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
-void SimEsdHidden(AliLoader *pHL,AliESD *pEsd)
-{  
+void SimEsdHidden(AliLoader *pHL,AliESDEvent *pEsd)
+{
+  TH1F *hdC = new TH1F("dC","dC",100,-0.2,0.2);
+  TH1F *hCer = new TH1F("Cer","Cer",250,0.,0.75);
+  TH2F *htvsp = new TH2F("tvsp","tvsp",100,0.,5.,1000,0.,0.75);
+  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);
   AliHMPID *pH=(AliHMPID*)gAL->GetAliRun()->GetDetector("HMPID");
   Int_t mtid=-1;
@@ -68,35 +82,54 @@ void SimEsdHidden(AliLoader *pHL,AliESD *pEsd)
   Printf("Number of events to process: %i",iNevt);
   for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//events loop
     if(!(iEvt%50)) Printf("Events processed %i",iEvt);
-    gAL->GetEvent(iEvt);
+    gAL->GetEvent(iEvt);    
     pHL->TreeR()->GetEntry(0);
-    AliESDtrack trk;
-// Hidden Track
-    for(int iCh=AliHMPIDDigit::kMinCh;iCh<=AliHMPIDDigit::kMaxCh;iCh++){
-      AliESDtrack trk;
-       if(AliHMPIDTracker::ReconHiddenTrk(iCh,&trk,pH->CluLst(),pNmean);) continue;
-       pEsd->AddTrack(&trk);
-       Printf(" theta Cerenkov reconstructed %f",trk.GetHMPIDsignal());
-    }
-//    AliHMPIDTracker::Recon(pEsd,pH->CluLst(),pNmean);
-//      
+    AliStack *pStack=gAL->Stack();
+    for(Int_t i=0;i<pStack->GetNtrack();i++){
+      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,pQthre)!=0) continue;
+      trk.GetHMPIDtrk(radX,radY,thetaTrk,phiTrk);
+//      Printf("reconstr. track theta %f phi %f",thetaTrk*rd,phiTrk*rd);
+      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());
+      hdC->Fill(trk.GetHMPIDsignal()-thetaCerSim);
+      hCer->Fill(trk.GetHMPIDsignal());
+      htvsp->Fill(pTrack->P(),trk.GetHMPIDsignal());
+    }// track loop
     pEsd->SetMagneticField(AliHMPIDTracker::GetBz());
     gEsdTr->Fill();
     pEsd->Reset();
   }// event loop
   Printf("Events processed %i",iEvt);
   gAL->UnloadHeader();  gAL->UnloadKinematics();
-}//EsdFromStack()
+}//EsdHidden()
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
-void OpenCalib()
+Bool_t OpenCalib()
 {
   AliCDBManager* pCDB = AliCDBManager::Instance();
   pCDB->SetDefaultStorage("local://$HOME");
   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;
 }
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++