end-of-line normalization

[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskMCParticleFilter.cxx

diff --git a/ANALYSIS/AliAnalysisTaskMCParticleFilter.cxx b/ANALYSIS/AliAnalysisTaskMCParticleFilter.cxx
index bc1a6edc1b7ec289083e0b61d5247ac0e74d8a54..7f6bae2200d94548dfdb49a254ff58e2f26d8fc5 100644 (file)
--- a/ANALYSIS/AliAnalysisTaskMCParticleFilter.cxx
+++ b/ANALYSIS/AliAnalysisTaskMCParticleFilter.cxx
@@ -35,6 +35,7 @@
 #include "AliStack.h"
 #include "AliMCEvent.h"
 #include "AliMCEventHandler.h"
+#include "AliESDInputHandler.h"
 #include "AliAODEvent.h"
 #include "AliAODHeader.h"
 #include "AliAODMCHeader.h"
@@ -46,6 +47,7 @@
 #include "AliGenHijingEventHeader.h"
 #include "AliGenPythiaEventHeader.h"
 #include "AliGenCocktailEventHeader.h"
+#include "AliGenEventHeaderTunedPbPb.h"
 
 #include "AliLog.h"
 
@@ -70,7 +72,8 @@ Bool_t AliAnalysisTaskMCParticleFilter::Notify()
   // Implemented Notify() to read the cross sections
   // from pyxsec.root
   // 
-  TTree *tree = AliAnalysisManager::GetAnalysisManager()->GetTree();
+  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
+  TTree *tree = mgr->GetTree();
   Double_t xsection = 0;
   UInt_t   ntrials  = 0;
   if(tree){
@@ -81,7 +84,11 @@ Bool_t AliAnalysisTaskMCParticleFilter::Notify()
     }
 
     TString fileName(curfile->GetName());
-    if(fileName.Contains("AliESDs.root")){
+    TString datafile = mgr->GetInputEventHandler()->GetInputFileName();
+    if (fileName.Contains(datafile)) {
+        fileName.ReplaceAll(datafile, "pyxsec.root");
+    }
+    else if(fileName.Contains("AliESDs.root")){
         fileName.ReplaceAll("AliESDs.root", "pyxsec.root");
     }
     else if(fileName.Contains("AliAOD.root")){
@@ -164,7 +171,7 @@ AliAnalysisTaskMCParticleFilter& AliAnalysisTaskMCParticleFilter::operator=(cons
 void AliAnalysisTaskMCParticleFilter::UserCreateOutputObjects()
 {
   // Create the output container
-  PostData(1,fHistList);
+  
 
     if (OutputTree()&&fTrackFilterMother) 
        OutputTree()->GetUserInfo()->Add(fTrackFilterMother);
@@ -212,17 +219,19 @@ void AliAnalysisTaskMCParticleFilter::UserCreateOutputObjects()
     h1Trials->GetXaxis()->SetBinLabel(1,"#sum{ntrials}");
     fHistList->Add(h1Trials);
 
+    PostData(1,fHistList);
     
 }
 
 //____________________________________________________________________
 void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
 {
-// Execute analysis for current event
-//
-
-// Fill AOD tracks from Kinematic stack
-    
+  // Execute analysis for current event
+  //
+  
+  // Fill AOD tracks from Kinematic stack
+  PostData(1,fHistList);
+  
   // get AliAOD Event 
   AliAODEvent* aod = AODEvent();
   if (!aod) {
@@ -269,12 +278,17 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
   AliGenHijingEventHeader *hiH  = 0;
   AliCollisionGeometry    *colG = 0;
   AliGenDPMjetEventHeader *dpmH = 0;
+  AliGenEventHeaderTunedPbPb *tunedH = 0;
+
   // it can be only one save some casts
   // assuming PYTHIA and HIJING are the most likely ones...
   if(!pyH){
       hiH = dynamic_cast<AliGenHijingEventHeader*>(mcEH);
       if(!hiH){
          dpmH = dynamic_cast<AliGenDPMjetEventHeader*>(mcEH);
+         if(!dpmH){
+           tunedH = dynamic_cast<AliGenEventHeaderTunedPbPb*>(mcEH);
+         }
       }
   }
   
@@ -311,13 +325,10 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
     fAODMcHeader->SetReactionPlaneAngle(colG->ReactionPlaneAngle());
     AliInfo(Form("Found Collision Geometry. Got Reaction Plane %lf\n", colG->ReactionPlaneAngle()));
   }
-
-
-
-  // check varibales for charm need all daughters
-  static int  iTaken = 0;
-  static int  iAll = 0;
-  static int  iCharm = 0;
+  else if(tunedH) {
+    fAODMcHeader->SetReactionPlaneAngle(tunedH->GetPsi2());
+    fAODMcHeader->SetCrossSection(tunedH->GetCentrality());
+  }
 
 
   Int_t j=0;
@@ -373,6 +384,12 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
   }
 
   /*
+
+  // check varibales for charm need all daughters
+  static int  iTaken = 0;
+  static int  iAll = 0;
+  static int  iCharm = 0;
+
   for (Int_t ip = 0; ip < np; ip++){
     AliMCParticle* mcpart = (AliMCParticle*) mcE->GetTrack(ip);
     TParticle* part = mcpart->Particle();
@@ -402,7 +419,6 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
 
 
   aodH->StoreMCParticles();
-  PostData(1,fHistList);
 
   return;
 }