Standard cuts for (di-)muon analysis (Diego)

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

diff --git a/ANALYSIS/AliAnalysisTaskMCParticleFilter.cxx b/ANALYSIS/AliAnalysisTaskMCParticleFilter.cxx
index 18f01c2be40bcf2a23a00fbaf3b7602cd7e30fc9..be12fa9adcf3cfa9e714bd04f78f806d9bbd34d0 100644 (file)
--- a/ANALYSIS/AliAnalysisTaskMCParticleFilter.cxx
+++ b/ANALYSIS/AliAnalysisTaskMCParticleFilter.cxx
@@ -55,8 +55,10 @@ ClassImp(AliAnalysisTaskMCParticleFilter)
 
 //____________________________________________________________________
 AliAnalysisTaskMCParticleFilter::AliAnalysisTaskMCParticleFilter():
-AliAnalysisTaskSE(),
+  AliAnalysisTaskSE(),
   fTrackFilterMother(0x0),
+  fAODMcHeader(0x0),
+  fAODMcParticles(0x0),
   fHistList(0x0)
 {
   // Default constructor
@@ -99,7 +101,7 @@ Bool_t AliAnalysisTaskMCParticleFilter::Notify()
     }
     TTree *xtree = (TTree*)fxsec->Get("Xsection");
     if(!xtree){
-      Printf("%s:%d tree not found in the pyxsec.root",(char*)__FILE__,__LINE__);
+      AliWarning(Form("%s:%d tree not found in the pyxsec.root",(char*)__FILE__,__LINE__));
       return kTRUE;
     }
     xtree->SetBranchAddress("xsection",&xsection);
@@ -116,10 +118,12 @@ Bool_t AliAnalysisTaskMCParticleFilter::Notify()
 AliAnalysisTaskMCParticleFilter::AliAnalysisTaskMCParticleFilter(const char* name):
     AliAnalysisTaskSE(name),
     fTrackFilterMother(0x0),
+    fAODMcHeader(0x0),
+    fAODMcParticles(0x0),
     fHistList(0x0)
 {
   // Default constructor
-  DefineOutput(1,TList::Class());  
+  DefineOutput(1, TList::Class());  
 }
 
 /*
@@ -134,7 +138,14 @@ AliAnalysisTaskMCParticleFilter::AliAnalysisTaskMCParticleFilter(const AliAnalys
 //____________________________________________________________________
 AliAnalysisTaskMCParticleFilter::~AliAnalysisTaskMCParticleFilter()
 {
-  //  if( fTrackFilterMother ) delete fTrackFilterMother;
+
+  if(fAODMcHeader){
+    delete fAODMcHeader;
+  }
+  if(fAODMcParticles){
+    fAODMcParticles->Delete();
+    delete fAODMcParticles;
+  }
 }
 
 /*
@@ -153,6 +164,8 @@ AliAnalysisTaskMCParticleFilter& AliAnalysisTaskMCParticleFilter::operator=(cons
 void AliAnalysisTaskMCParticleFilter::UserCreateOutputObjects()
 {
   // Create the output container
+  PostData(1,fHistList);
+
     if (OutputTree()&&fTrackFilterMother) 
        OutputTree()->GetUserInfo()->Add(fTrackFilterMother);
 
@@ -166,16 +179,14 @@ void AliAnalysisTaskMCParticleFilter::UserCreateOutputObjects()
 
 
     // mcparticles
-    TClonesArray *tca = new TClonesArray("AliAODMCParticle", 0);
-    tca->SetName(AliAODMCParticle::StdBranchName());
-    AddAODBranch("TClonesArray",&tca);
+    fAODMcParticles = new TClonesArray("AliAODMCParticle", 0);
+    fAODMcParticles->SetName(AliAODMCParticle::StdBranchName());
+    AddAODBranch("TClonesArray",&fAODMcParticles);
 
     // MC header...
-    AliAODMCHeader *mcHeader = new AliAODMCHeader();
-    mcHeader->SetName(AliAODMCHeader::StdBranchName());
-    AddAODBranch("AliAODMCHeader",&mcHeader);    
-
-    
+    fAODMcHeader = new AliAODMCHeader();
+    fAODMcHeader->SetName(AliAODMCHeader::StdBranchName());
+    AddAODBranch("AliAODMCHeader",&fAODMcHeader);    
 
     AliMCEventHandler *mcH = (AliMCEventHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler()); 
     AliAODHandler *aodH = dynamic_cast<AliAODHandler*> ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
@@ -251,24 +262,23 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
   Int_t nprim = mcE->GetNumberOfPrimaries();
   // TODO ADD MC VERTEX
 
-  AliAODMCHeader *aodMCHo = (AliAODMCHeader *) aod->FindListObject("mcHeader");
   // Get the proper MC Collision Geometry
   AliGenEventHeader* mcEH = mcE->GenEventHeader();
 
-  AliGenPythiaEventHeader *pyH = dynamic_cast<AliGenPythiaEventHeader*>(mcEH);
-  AliGenHijingEventHeader *hiH = 0;
-  AliCollisionGeometry *colG = 0;
+  AliGenPythiaEventHeader *pyH  = dynamic_cast<AliGenPythiaEventHeader*>(mcEH);
+  AliGenHijingEventHeader *hiH  = 0;
+  AliCollisionGeometry    *colG = 0;
   AliGenDPMjetEventHeader *dpmH = 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){
-      colG = dynamic_cast<AliCollisionGeometry *>(mcEH);
-      if(!colG)dpmH = dynamic_cast<AliGenDPMjetEventHeader*>(mcEH);
-    }
+      hiH = dynamic_cast<AliGenHijingEventHeader*>(mcEH);
+      if(!hiH){
+         dpmH = dynamic_cast<AliGenDPMjetEventHeader*>(mcEH);
+      }
   }
-
+  
+  if (hiH || dpmH) colG = dynamic_cast<AliCollisionGeometry*>(mcEH);
 
   // fetch the trials on a event by event basis, not from pyxsec.root otherwise 
   // we will get a problem when running on proof since Notify may be called 
@@ -280,10 +290,10 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
     if (ccEH) {
       TList *genHeaders = ccEH->GetHeaders();
       for (int imch=0; imch<genHeaders->GetEntries(); imch++) {
-       if(!pyH)dynamic_cast<AliGenPythiaEventHeader*>(genHeaders->At(imch));
-       if(!hiH)dynamic_cast<AliGenHijingEventHeader*>(genHeaders->At(imch));
+       if(!pyH)pyH = dynamic_cast<AliGenPythiaEventHeader*>(genHeaders->At(imch));
+       if(!hiH)hiH = dynamic_cast<AliGenHijingEventHeader*>(genHeaders->At(imch));
        if(!colG)colG = dynamic_cast<AliCollisionGeometry *>(genHeaders->At(imch));
-       if(!dpmH)dynamic_cast<AliGenDPMjetEventHeader*>(genHeaders->At(imch));
+       if(!dpmH)dpmH = dynamic_cast<AliGenDPMjetEventHeader*>(genHeaders->At(imch));
       }
     }
   }
@@ -298,8 +308,8 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
 
 
   if (colG) {
-    aodMCHo->SetReactionPlaneAngle(colG->ReactionPlaneAngle());
-    printf("Found Collision Geometry. Got Reaction Plane %lf\n", colG->ReactionPlaneAngle());
+    fAODMcHeader->SetReactionPlaneAngle(colG->ReactionPlaneAngle());
+    AliInfo(Form("Found Collision Geometry. Got Reaction Plane %lf\n", colG->ReactionPlaneAngle()));
   }
 
 
@@ -324,18 +334,18 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
     if (ip < nprim) {
       // Select the primary event
       write = kTRUE;
-    } else if (part->GetUniqueID() == 4) {
+    } else if (part->GetUniqueID() == kPDecay) {
       // Particles from decay
       // Check that the decay chain ends at a primary particle
       AliMCParticle* mother = mcpart;
       Int_t imo = mcpart->GetMother();
-      while((imo >= nprim) && (mother->GetUniqueID() == 4)) {
+      while((imo >= nprim) && (mother->GetUniqueID() == kPDecay)) {
        mother =  (AliMCParticle*) mcE->GetTrack(imo);
        imo =  mother->GetMother();
       }
       // Select according to pseudorapidity and production point of primary ancestor
       if (imo < nprim && Select(((AliMCParticle*) mcE->GetTrack(imo))->Particle(), rv, zv))write = kTRUE;         
-    } else if (part->GetUniqueID() == 5) {
+    } else if (part->GetUniqueID() == kPPair) {
       // Now look for pair production
       Int_t imo = mcpart->GetMother();
       if (imo < nprim) {
@@ -345,7 +355,7 @@ void AliAnalysisTaskMCParticleFilter::UserExec(Option_t */*option*/)
        // Check if the gamma comes from the decay chain of a primary particle
        AliMCParticle* mother =  (AliMCParticle*) mcE->GetTrack(imo);
        imo = mother->GetMother();
-       while((imo >= nprim) && (mother->GetUniqueID() == 4)) {
+       while((imo >= nprim) && (mother->GetUniqueID() == kPDecay)) {
          mother =   (AliMCParticle*) mcE->GetTrack(imo);
          imo =  mother->GetMother();
        }