]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - JETAN/AliJetBkg.cxx
major dielectron update (included also the data and plotting macros for paper)
[u/mrichter/AliRoot.git] / JETAN / AliJetBkg.cxx
index 54ce715bcdd3782da5c7727fa54bedfd0448d40b..ad3c944aee10746b3f0ca753b3a151cdbd8e601c 100644 (file)
@@ -83,9 +83,215 @@ AliJetBkg& AliJetBkg::operator=(const AliJetBkg& source){
     return *this;
 
 }
+
+
 //___________________________________________________________________
-Float_t AliJetBkg::BkgFastJet(){
+  void AliJetBkg::BkgFastJetb(Double_t& rho,Double_t& sigma, 
+Double_t& meanarea){
+
+    AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader; 
+    Bool_t debug  = header->GetDebug();     // debug option 
+    if(debug)cout<<"===============  AliJetBkg::BkgFastJetb()  =========== "<<endl;
+  vector<fastjet::PseudoJet> inputParticles=fInputFJ->GetInputParticles();
+  
+  //cout<<"printing inputParticles for BKG "<<inputParticles.size()<<endl;
+  
+
+   
+
+  double rParamBkg = header->GetRparamBkg(); //Radius for background calculation
+
+  Double_t medianb,sigmab,meanareab;
+  CalcRhob(medianb,sigmab,meanareab,inputParticles,rParamBkg,"All");
+  rho=medianb;
+  sigma=sigmab;
+  meanarea=meanareab;
+}
+//_________________________________________________________________
+
+  void AliJetBkg::BkgFastJetWoHardest(Double_t& rho,Double_t& sigma, 
+Double_t& meanarea){
+    AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader; 
+    Bool_t debug  = header->GetDebug();     // debug option 
+    if(debug)cout<<"===============  AliJetBkg::BkgWoHardest()  =========== "<<endl;
+  vector<fastjet::PseudoJet> inputParticles=fInputFJ->GetInputParticles();
+  
+  //cout<<"printing inputParticles for BKG "<<inputParticles.size()<<endl;   
+  double rParamBkg = header->GetRparamBkg(); //Radius for background calculation  
+  Double_t medianb,sigmab,meanareab;
+  CalcRhoWoHardest(medianb,sigmab,meanareab,inputParticles,rParamBkg,"All");
+  rho=medianb;
+  sigma=sigmab;
+  meanarea=meanareab;
+
+  }
+
+//____________________________________________________________________
+
+ void AliJetBkg::CalcRhob(Double_t& median,Double_t& 
+                         sigma,Double_t& 
+                         meanarea,vector<fastjet::PseudoJet> inputParticles,Double_t 
+                         rParamBkg,TString method){
+  //calculate rho using the fastjet method
+
+  AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader;
+  Bool_t debug  = header->GetDebug();     // debug option
+
+  fastjet::Strategy strategy = header->GetStrategy();
+  fastjet::RecombinationScheme recombScheme = header->GetRecombScheme();
+  fastjet::JetAlgorithm algorithm = header->GetAlgorithm(); 
+  fastjet::JetDefinition jetDef(algorithm, rParamBkg, recombScheme, strategy);
+
+  // create an object that specifies how we to define the area
+  fastjet::AreaDefinition areaDef;
+  double ghostEtamax = header->GetGhostEtaMax(); 
+  double ghostArea   = header->GetGhostArea(); 
+  int    activeAreaRepeats = header->GetActiveAreaRepeats(); 
+
+  // now create the object that holds info about ghosts
+
+  
+
+  fastjet::GhostedAreaSpec ghost_spec(ghostEtamax, activeAreaRepeats, ghostArea);
+  // and from that get an area definition
+  fastjet::AreaType areaType = header->GetAreaType();
+  areaDef = fastjet::AreaDefinition(areaType,ghost_spec);
+  
+  //fastjet::ClusterSequenceArea clust_seq(inputParticles, jetDef);
+  fastjet::ClusterSequenceArea clust_seq(inputParticles, jetDef,areaDef);
+  TString comment = "Running FastJet algorithm for BKG calculation with the following setup. ";
+  comment+= "Jet definition: ";
+  comment+= TString(jetDef.description());
+  //  comment+= ". Area definition: ";
+  //comment+= TString(areaDef.description());
+  comment+= ". Strategy adopted by FastJet: ";
+  comment+= TString(clust_seq.strategy_string());
+  comment+= Form("Method: %s",method.Data());
+  header->SetComment(comment);
+  if(debug){
+    cout << "--------------------------------------------------------" << endl;
+    cout << comment << endl;
+    cout << "--------------------------------------------------------" << endl;
+  }
+
+  vector<fastjet::PseudoJet> inclusiveJets = clust_seq.inclusive_jets(0.);
+  vector<fastjet::PseudoJet> jets = sorted_by_pt(inclusiveJets); 
+
+  //cout<<"# of BKG jets = "<<jets.size()<<endl;
+      //  for (size_t j = 0; j < jets.size(); j++) { // loop for jets   
+      //printf("BKG Jet found %5d %9.5f %8.5f %10.3f %4.4f 
+      //\n",(Int_t)j,jets[j].rap(),jets[j].phi(),jets[j].perp(),
+      //clust_seq.area(jets[j]));
+      // }
+  
   
+  double phiMin = 0, phiMax = 0, rapMin = 0, rapMax = 0;
+
+    phiMin = 0;
+    phiMax = 2*TMath::Pi();
+  rapMax = ghostEtamax - rParamBkg;
+  rapMin = - ghostEtamax + rParamBkg;
+
+
+  fastjet::RangeDefinition range(rapMin, rapMax, phiMin, phiMax);
+
+
+  double medianb, sigmab, meanareab;
+  clust_seq.get_median_rho_and_sigma(inclusiveJets, range, false, medianb, sigmab, meanareab, false);
+  median=medianb;
+  sigma=sigmab;
+  meanarea=meanareab; 
+
+ }
+
+//____________________________________________________________________
+
+
+ void AliJetBkg::CalcRhoWoHardest(Double_t& median,Double_t& 
+sigma,Double_t& 
+meanarea,vector<fastjet::PseudoJet> inputParticles,Double_t 
+rParamBkg,TString method){
+  //calculate rho using the fastjet method
+
+  AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader;
+  Bool_t debug  = header->GetDebug();     // debug option
+
+  fastjet::Strategy strategy = header->GetStrategy();
+  fastjet::RecombinationScheme recombScheme = header->GetRecombScheme();
+  fastjet::JetAlgorithm algorithm = header->GetAlgorithm(); 
+  fastjet::JetDefinition jetDef(algorithm, rParamBkg, recombScheme, strategy);
+
+  // create an object that specifies how we to define the area
+  fastjet::AreaDefinition areaDef;
+  double ghostEtamax = header->GetGhostEtaMax(); 
+  double ghostArea   = header->GetGhostArea(); 
+  int    activeAreaRepeats = header->GetActiveAreaRepeats(); 
+
+  // now create the object that holds info about ghosts
+
+
+  fastjet::GhostedAreaSpec ghost_spec(ghostEtamax, activeAreaRepeats, ghostArea);
+  // and from that get an area definition
+  fastjet::AreaType areaType = header->GetAreaType();
+  areaDef = fastjet::AreaDefinition(areaType,ghost_spec);
+  //cout<<"rParamBkg="<<rParamBkg<<"  ghostEtamax="<<ghostEtamax<<"  ghostArea="<<ghostArea<<" areadef="<<TString(areaDef.description())<<endl;
+  //fastjet::ClusterSequenceArea clust_seq(inputParticles, jetDef);
+  fastjet::ClusterSequenceArea clust_seq(inputParticles,jetDef,areaDef);
+  TString comment = "Running FastJet algorithm for BKG calculation with the following setup. ";
+  comment+= "Jet definition: ";
+  comment+= TString(jetDef.description());
+  //  comment+= ". Area definition: ";
+  //comment+= TString(areaDef.description());
+  comment+= ". Strategy adopted by FastJet: ";
+  comment+= TString(clust_seq.strategy_string());
+  comment+= Form("Method: %s",method.Data());
+  header->SetComment(comment);
+  if(debug){
+    cout << "--------------------------------------------------------" << endl;
+    cout << comment << endl;
+    cout << "--------------------------------------------------------" << endl;
+  }
+
+  vector<fastjet::PseudoJet> inclusiveJets = clust_seq.inclusive_jets(0.);
+  vector<fastjet::PseudoJet> jets = sorted_by_pt(inclusiveJets); 
+  vector<fastjet::PseudoJet> jets2=sorted_by_pt(inclusiveJets);
+  if(jets2.size()>=2) jets2.erase(jets2.begin(),jets2.begin()+1);
+  
+   
+  
+  double phiMin = 0, phiMax = 0, rapMin = 0, rapMax = 0;
+
+    phiMin = 0;
+    phiMax = 2*TMath::Pi();
+  rapMax = ghostEtamax - rParamBkg;
+  rapMin = - ghostEtamax + rParamBkg;
+
+
+  fastjet::RangeDefinition range(rapMin, rapMax, phiMin, phiMax);
+
+
+  double medianb, sigmab, meanareab;
+  clust_seq.get_median_rho_and_sigma(jets2, range, false, medianb, sigmab, 
+meanareab, false);
+  median=medianb;
+  sigma=sigmab;
+  meanarea=meanareab; 
+
+  
+ }
+
+
+//___________________________________________________________________
+Float_t AliJetBkg::BkgFastJet(){
+// Return background  
   AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader;
   Bool_t debug  = header->GetDebug();     // debug option
 
@@ -107,7 +313,7 @@ Float_t AliJetBkg::BkgFastJet(){
 }
 //___________________________________________________________________
 Float_t  AliJetBkg::BkgChargedFastJet(){
-
+// Background for charged jets
   AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader;
   Bool_t debug  = header->GetDebug();     // debug option
 
@@ -126,7 +332,7 @@ Float_t  AliJetBkg::BkgChargedFastJet(){
 
   Double_t rho=CalcRho(inputParticlesCharged,rParam,"Charg");
 
-  cout<<"-------- rho (from CHARGED part)="<<rho<<endl; 
+  if(debug)cout<<"-------- rho (from CHARGED part)="<<rho<<endl; 
   return rho;
 }
 
@@ -149,160 +355,6 @@ Float_t AliJetBkg::BkgStat()
   return enTot/accEMCal;
 
 }
-/////////////////////////////////
-Float_t AliJetBkg::BkgRemoveJetLeading(TClonesArray* fAODJets)
-{
-  // Remove the particles of the
-  // two largest jets using the track references  stored in the AODJet from the estimation of new rho. 
-
-  AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader;
-  Bool_t debug  = header->GetDebug();     // debug option
-
-  if(debug)cout<<"==============AliJetBkg::BkgRemoveJetLeading()============="<<endl;
-
-  // check if we are reading AOD jets
-  TRefArray *refs = 0;
-  Bool_t fromAod = !strcmp(fReader->ClassName(),"AliJetESDReader");
-  if (fromAod) { refs = fReader->GetReferences(); }
-  
-  //Hard wired Calorimeter area (get it later from the AliJetReaderHeader.h)
-  Double_t accEMCal=2*0.7*110./180*TMath::Pi();//2.68 area of EMCal
-
-  Int_t nJ=fAODJets->GetEntries(); //this must be the # of jets... 
-  if(debug)cout<<"nJets:  "<<nJ<<endl;
-  
-
-  //begin unit array 
-  TClonesArray* fUnit = fReader->GetUnitArray();
-  if(fUnit == 0) { cout << "Could not get the momentum array" << endl; return -99; }
-
-  Int_t nIn = fUnit->GetEntries();
-  if(nIn == 0) { cout << "entries = 0 ; Event empty !!!" << endl ; return -99; }
-  
-  Float_t rhoback=0.0;
-  Float_t jetarea1=0.0,jetarea2=0.0;
-    
-  Int_t particlejet1=-99;
-  Int_t particlejet2=-99;
-  TRefArray *refarray1 = 0;
-  TRefArray *refarray2 = 0;
-  Int_t nJettracks1 = 0, nJettracks2 = 0;
-  Int_t acc=0,acc1=0;
-  AliAODJet *jet1;
-  AliAODJet *jet2;
-
-  if(nJ==1){
-    jet1 = dynamic_cast<AliAODJet*>(fAODJets->At(0));
-    jetarea1=jet1->EffectiveAreaCharged();
-    Float_t jetPhi=jet1->Phi();
-    Float_t jetEta=jet1->Eta();
-    if(jetPhi>1.396 && jetPhi<3.316 && jetEta>-0.7 && jetEta<0.7)acc=1;
-    refarray1=jet1->GetRefTracks();
-    nJettracks1=refarray1->GetEntries();
-    if(debug)cout<<"nJ = 1, acc="<<acc<<"  jetarea1="<<jetarea1<<endl;
-
-  }
-
-  if(nJ>=2){
-    jet1 = dynamic_cast<AliAODJet*>(fAODJets->At(0));
-    jetarea1=jet1->EffectiveAreaCharged();
-    Float_t jetPhi1=jet1->Phi();
-    Float_t jetEta1=jet1->Eta();
-    if(jetPhi1>1.396 && jetPhi1<3.316 && jetEta1>-0.7 && jetEta1<0.7)acc=1;
-    refarray1=jet1->GetRefTracks();
-    nJettracks1=refarray1->GetEntries();
-    if(debug)cout<<"npart = "<<nJettracks1<<endl;
-   
-    jet2 = dynamic_cast<AliAODJet*>(fAODJets->At(1));
-    jetarea2=jet2->EffectiveAreaCharged();
-    Float_t jetPhi2=jet2->Phi();
-    Float_t jetEta2=jet2->Eta();
-    if(jetPhi2>1.396 && jetPhi2<3.316 && jetEta2>-0.7 && jetEta2<0.7)acc1=1;
-    refarray2=jet2->GetRefTracks();
-    nJettracks2=refarray2->GetEntries();
-    if(debug)cout<<"nJ = "<<nJ<<", acc="<<acc<<"  acc1="<<acc1<<"  jetarea1="<<jetarea1<<"  jetarea2="<<jetarea2<<endl;
-  }
-
-
-  
-  // cout<<" nIn = "<<nIn<<endl;
-  Float_t sumPt=0;
-  Float_t eta,phi,pt;
-  Int_t ipart=0;
-
-  for(Int_t i=0; i<nIn; i++) 
-    { //Unit Array Loop
-      AliJetUnitArray *uArray = (AliJetUnitArray*)fUnit->At(i);
-
-      if(uArray->GetUnitEnergy()>0.){
-       eta   = uArray->GetUnitEta();
-       phi   = uArray->GetUnitPhi();
-       pt = uArray->GetUnitEnergy();
-       //      cout<<"ipart = "<<ipart<<" eta="<<eta<<"  phi="<<phi<<endl;
-       if(phi>1.396 && phi<3.316 && eta>-0.7 && eta<0.7){
-         //cout<<sumPt<<endl;
-           sumPt+=pt;
-
-           
-           if(nJ==1 && acc==1){
-             for(Int_t ii=0; ii<nJettracks1;ii++){    
-               
-               particlejet1  = ((AliJetUnitArray*)refarray1->At(ii))->GetUnitTrackID();
-        
-               if(ipart==particlejet1) {
-                 sumPt-=pt;
-               }
-             }
-           }
-         
-         
-           if(nJ>=2){
-             
-             //first jet
-             if(acc==1){
-               for(Int_t ii=0; ii<nJettracks1;ii++){    
-                 particlejet1  = ((AliJetUnitArray*)refarray1->At(ii))->GetUnitTrackID();
-                 
-               //cout<<"uArr loop = "<<i<<"  ipart in uArr (1/2)="<<ipart<<"  part in jet="<<ii<<"  partID="<<particlejet1<<" sumPt="<<sumPt<<endl; 
-                 if(ipart==particlejet1) {
-                   sumPt-=pt;
-                 }
-               }
-             }
-             if(acc1==1){
-               //second jet
-               for(Int_t ii=0; ii<nJettracks2;ii++){ 
-                 particlejet2  = ((AliJetUnitArray*)refarray2->At(ii))->GetUnitTrackID();
-                 //cout<<"uArr loop = "<<i<<"  ipart in uArr (2/2)="<<ipart<<"  part in jet="<<ii<<"  partID="<<particlejet2<<" sumPt="<<sumPt<<endl; 
-                 if(ipart==particlejet2) {
-                   sumPt-=pt;
-                 }
-               }
-             }
-             
-             
-           }
-       
-       }//if phi,eta
-       ipart++;
-      }//end if energy
-    }// end unit array loop          
-  
-
-  Float_t areasum=areasum=accEMCal-acc*jetarea1-acc1*jetarea2;
-  if(debug)cout<<"pt sum   "<<sumPt<<" area  "<<areasum<<endl;
-   
-  if(nJ>0) rhoback=sumPt/areasum;
-  else rhoback=0.;
-  if(debug)cout<<" rho from leading jet paricle array removed   "<<rhoback<<endl;
-  return rhoback;
-}
-
-
-
 
 ////////////////////////////////////////////////////////////////////////
 Float_t AliJetBkg::BkgFastJetCone(TClonesArray* fAODJets)
@@ -345,7 +397,7 @@ Float_t AliJetBkg::BkgFastJetCone(TClonesArray* fAODJets)
   Float_t rCone=0.4;
   
   if(nJ==1) { 
-    AliAODJet *jettmp = dynamic_cast<AliAODJet*>(fAODJets->At(0));
+    AliAODJet *jettmp = (AliAODJet*)(fAODJets->At(0));
     jeteta=jettmp->Eta();
     jetphi=jettmp->Phi();
     acc=EmcalAcceptance(jeteta,jetphi,rCone);
@@ -355,8 +407,8 @@ Float_t AliJetBkg::BkgFastJetCone(TClonesArray* fAODJets)
 
   
   if(nJ>=2) { 
-    AliAODJet *jettmp = dynamic_cast<AliAODJet*>(fAODJets->At(0));
-    AliAODJet *jettmp1 = dynamic_cast<AliAODJet*>(fAODJets->At(1));
+    AliAODJet *jettmp = (AliAODJet*)(fAODJets->At(0));
+    AliAODJet *jettmp1 = (AliAODJet*)(fAODJets->At(1));
     jeteta=jettmp->Eta();
     jetphi=jettmp->Phi();
     jeteta1=jettmp1->Eta();
@@ -541,8 +593,8 @@ Double_t  AliJetBkg::BkgFunction(Double_t */*x*/,Double_t */*par*/){
 }
 
 
-Bool_t AliJetBkg::EmcalAcceptance(const Float_t eta, const Float_t phi, const Float_t radius){
+Bool_t AliJetBkg::EmcalAcceptance(const Float_t eta, const Float_t phi, const Float_t radius) const{
+// Apply emcal acceptance cuts
   Float_t meanPhi=190./180.*TMath::Pi()-110./180.*TMath::Pi()/2;
   Float_t deltaphi=110./180.*TMath::Pi();
   Float_t phicut=deltaphi/2.-radius;