Missed a log message on last commit.

[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis / AliFMDAnalysisTaskSE.cxx

diff --git a/PWG2/FORWARD/analysis/AliFMDAnalysisTaskSE.cxx b/PWG2/FORWARD/analysis/AliFMDAnalysisTaskSE.cxx
index 9cdae7ec7a80c95ef12804040fa9931794e4e016..dbdb02720c3810acb0a45a619b56cd8a341e49d9 100644 (file)
--- a/PWG2/FORWARD/analysis/AliFMDAnalysisTaskSE.cxx
+++ b/PWG2/FORWARD/analysis/AliFMDAnalysisTaskSE.cxx
@@ -8,8 +8,16 @@
 #include "AliLog.h"
 #include "AliFMDDndeta.h"
 #include "TDirectory.h"
+#include "TProfile2D.h"
 ClassImp(AliFMDAnalysisTaskSE)
-
+//
+// This task controls the running of the FMD analysis. The current version is made for 
+// dN/deta analysis but multiplicity and correlation tasks will be added here as well.
+//
+// To get the dN/detadphi per selected event as a TH2F* object connect to the 
+// output list of this task in the analysis framework and do
+//
+// TH2F* hFMDdNdetadphi = (TH2F*)list->FindObject("dNdetadphiHistogramTrVtx");
 //_____________________________________________________________________
 AliFMDAnalysisTaskSE::AliFMDAnalysisTaskSE():
 AliAnalysisTaskSE(),
@@ -19,7 +27,10 @@ AliAnalysisTaskSE(),
   fBackground("BackgroundCorrected",kFALSE),
   fDndeta("dNdeta",kFALSE), 
   fBFCorrelation("BFCorrelation",kFALSE), 
-  fParams(0)
+  fParams(0),
+  fFirstEvent(kTRUE),
+  fCentralityLow(0),
+  fCentralityHigh(100)
 {
   // Default constructor
 }
@@ -32,16 +43,22 @@ AliFMDAnalysisTaskSE::AliFMDAnalysisTaskSE(const char* name):
   fBackground("BackgroundCorrected",kFALSE),
   fDndeta("dNdeta",kFALSE), 
   fBFCorrelation("BFCorrelation",kFALSE), 
-  fParams(0)
+  fParams(0),
+  fFirstEvent(kTRUE),
+  fCentralityLow(0),
+  fCentralityHigh(100)
 {
   SetParams(AliFMDAnaParameters::Instance());
   DefineOutput(1, TList::Class());
+  // DefineOutput(2, TH2F::Class());
 }
 //_____________________________________________________________________
 void AliFMDAnalysisTaskSE::UserCreateOutputObjects()
 {
 // Create the output containers
 //
+  
+  
   fListOfHistos = new TList();
   
   AliESDFMD* fmd = new AliESDFMD();
@@ -50,7 +67,7 @@ void AliFMDAnalysisTaskSE::UserCreateOutputObjects()
   TList* densitylist = new TList();
   
   TList* bgcorlist = new TList();
-  
+    
   fSharing.SetFMDData(fmd);
   fSharing.SetVertex(vertex);
   fSharing.SetOutputList(fListOfHistos);
@@ -74,7 +91,9 @@ void AliFMDAnalysisTaskSE::UserCreateOutputObjects()
   fBackground.CreateOutputObjects();
   fDndeta.CreateOutputObjects();
   fBFCorrelation.CreateOutputObjects();
+ 
   
+  PostData(1, fListOfHistos);
   
 }
 //_____________________________________________________________________
@@ -87,12 +106,23 @@ void AliFMDAnalysisTaskSE::UserExec(Option_t */*option*/)
 {
   // Execute analysis for current event
   //
+  AliESDEvent* fESD = (AliESDEvent*)InputEvent();
   
-  //AliFMDAnaParameters* pars = AliFMDAnaParameters::Instance();
+  AliFMDAnaParameters* pars = AliFMDAnaParameters::Instance();
   
+  // Centrality selection - work in progress
+  Float_t centrality = 1;
+  if( centrality < fCentralityLow || centrality > fCentralityHigh )  return;
   
-  AliESDEvent* fESD = (AliESDEvent*)InputEvent();
-  //std::cout<<fESD->GetBeamEnergy()<<"   "<<fESD->GetBeamType()<<"    "<<fESD->GetCurrentL3()<<std::endl;
+  //End of centrality selection
+  
+  if(fFirstEvent) {
+    pars->SetParametersFromESD(fESD);
+    pars->PrintStatus();
+    fFirstEvent = kFALSE;
+  }
+  
+  pars->SetTriggerStatus(fESD);
   fSharing.SetInputESD(fESD);
   
   fSharing.Exec("");
@@ -100,64 +130,91 @@ void AliFMDAnalysisTaskSE::UserExec(Option_t */*option*/)
     fDensity.Exec("");
     if(fDensity.GetEventStatus()) {
       fBackground.Exec("");  
-      fDndeta.Exec("");
-      fBFCorrelation.Exec("");
+      if(pars->GetRunDndeta())        fDndeta.Exec("");
+      if(pars->GetRunBFCorrelation()) fBFCorrelation.Exec("");
     }
     else return;
   }
   else return;
   
-  PostData(1, fListOfHistos);
-  
-  //fListOfHistos = fBackground.GetOutputList();
-  
  
+  PostData(1, fListOfHistos);
+     
 }
 //_____________________________________________________________________
 void AliFMDAnalysisTaskSE::Terminate(Option_t */*option*/)
 {
   
   TList* outputList = (TList*)GetOutputData(1);
+  AliFMDAnaParameters* pars = AliFMDAnaParameters::Instance();
+  
+  if(outputList) {
+    fSharing.SetOutputList(outputList);
+    fBackground.SetHitList(outputList);
+    fDndeta.SetOutputList(outputList); 
+    //fBFCorrelation.SetOutputList(outputList); 
+    fSharing.Terminate("");
+    fBackground.Terminate("");
+    if(fSharing.GetVtxEfficiencyFromData() > 0)
+      fDndeta.SetVtxEfficiency(fSharing.GetVtxEfficiencyFromData());
+    else
+      fDndeta.SetVtxEfficiency(pars->GetVtxSelectionEffFromMC());
+    
+    AliInfo(Form("Vertex efficiencies:  NSD_data=%f, INEL_data=%f, INEL_mc=%f", 
+                fSharing.GetNSDVtxEfficiencyFromData(),
+                fSharing.GetVtxEfficiencyFromData(),
+                pars->GetVtxSelectionEffFromMC()));
+    
+    if(fSharing.GetNSDVtxEfficiencyFromData() > 0)
+      fDndeta.SetVtxEfficiencyNSD(fSharing.GetNSDVtxEfficiencyFromData());
+    else
+      fDndeta.SetVtxEfficiencyNSD(pars->GetVtxSelectionEffFromMC());
+    
+    fDndeta.Terminate("");
+    //fBFCorrelation.Terminate("");
+    
+    AliFMDDndeta t;
+    t.SetNbinsToCut(2);
+    t.Init(outputList);
+    t.GenerateMult(AliFMDDndeta::kMult);
+    
+    TList* dNdetalist = t.GetMultList(AliFMDDndeta::kMult);
+    TList* cloneList = (TList*)dNdetalist->Clone("dNdeta");
+    cloneList->SetName("dNdeta");
+    outputList->Add(cloneList);
+    
+    t.GenerateMult(AliFMDDndeta::kMultTrVtx);
+    TList* dNdetalist2 = t.GetMultList(AliFMDDndeta::kMultTrVtx);
+    TList* cloneList2 = (TList*)dNdetalist2->Clone("dNdetaTrVtx");
+    cloneList2->SetName("dNdetaTrVtx");
+    outputList->Add(cloneList2);
+  
+    t.GenerateMult(AliFMDDndeta::kHits);
+    TList* dNdetalist3 = t.GetMultList(AliFMDDndeta::kHits);
+    TList* cloneList3 = (TList*)dNdetalist3->Clone("Hits");
+    cloneList3->SetName("Hits");
+    outputList->Add(cloneList3);
+    
+    t.GenerateMult(AliFMDDndeta::kHitsTrVtx);
+    TList* dNdetalist4 = t.GetMultList(AliFMDDndeta::kHits);
+    TList* cloneList4 = (TList*)dNdetalist4->Clone("HitsTrVtx");
+    cloneList4->SetName("HitsTrVtx");
+    outputList->Add(cloneList4);
+    
+    t.GenerateMult(AliFMDDndeta::kMultNSD);
+    TList* dNdetalist5 = t.GetMultList(AliFMDDndeta::kMultNSD);
+    TList* cloneList5 = (TList*)dNdetalist5->Clone("MultNSD");
+    cloneList5->SetName("MultNSD");
+    outputList->Add(cloneList5);
+    
+    // TFile file("fmd_ana_histos_tmp.root","RECREATE");
+    //  fListOfHistos->Write();
+    // file.Close();
+  }
+  else
+    AliWarning("no merged output from manager");
   
   
-  fSharing.SetOutputList(outputList);
-  fBackground.SetHitList(outputList);
-  fDndeta.SetOutputList(outputList); 
-  fBFCorrelation.SetOutputList(outputList); 
-  fSharing.Terminate("");
-  fBackground.Terminate("");
-  fDndeta.Terminate("");
-  fBFCorrelation.Terminate("");
-  
-  AliFMDDndeta t;
-  t.SetNbinsToCut(2);
-  t.Init(outputList);
-  t.GenerateMult(AliFMDDndeta::kMult);
-  
-  TList* dNdetalist = t.GetMultList(AliFMDDndeta::kMult);
-  TList* cloneList = (TList*)dNdetalist->Clone("dNdeta");
-  cloneList->SetName("dNdeta");
-  outputList->Add(cloneList);
-  
-  t.GenerateMult(AliFMDDndeta::kMultTrVtx);
-  TList* dNdetalist2 = t.GetMultList(AliFMDDndeta::kMultTrVtx);
-  TList* cloneList2 = (TList*)dNdetalist2->Clone("dNdetaTrVtx");
-  cloneList2->SetName("dNdetaTrVtx");
-  outputList->Add(cloneList2);
-  
-  t.GenerateMult(AliFMDDndeta::kHits);
-  TList* dNdetalist3 = t.GetMultList(AliFMDDndeta::kHits);
-  TList* cloneList3 = (TList*)dNdetalist3->Clone("Hits");
-  cloneList3->SetName("Hits");
-  outputList->Add(cloneList3);
-  t.GenerateMult(AliFMDDndeta::kHitsTrVtx);
-  TList* dNdetalist4 = t.GetMultList(AliFMDDndeta::kHits);
-  TList* cloneList4 = (TList*)dNdetalist3->Clone("HitsTrVtx");
-  cloneList4->SetName("HitsTrVtx");
-  outputList->Add(cloneList4);
-  // TFile file("fmd_ana_histos_tmp.root","RECREATE");
-  //  fListOfHistos->Write();
-  // file.Close();
 }
 
 //_____________________________________________________________________
@@ -182,3 +239,4 @@ void AliFMDAnalysisTaskSE::Print(Option_t* option) const
 //
 // EOF
 //
+// EOF