Updates from Prabhat

diff --git a/PWGCF/Correlations/DPhi/AliDptDptInMC.cxx b/PWGCF/Correlations/DPhi/AliDptDptInMC.cxx
index 9ec0e0c9afbe3d89275a010f5600515187e08e04..eb29352e885716cddc93ffe1c4f59db728fa3f20 100644 (file)
--- a/PWGCF/Correlations/DPhi/AliDptDptInMC.cxx
+++ b/PWGCF/Correlations/DPhi/AliDptDptInMC.cxx
@@ -1251,20 +1251,9 @@ void  AliDptDptInMC::UserExec(Option_t */*option*/)
       if(fAnalysisType == "MCAOD")
         { 
 
-         fArrayMC = dynamic_cast<TClonesArray*>(fAODEvent->FindListObject(AliAODMCParticle::StdBranchName()));
-          if (!fArrayMC) {
-            Printf("Error: MC particles branch not found!\n");
-            return;
-          }
-         AliAODMCHeader *mcHdr=0;
-          mcHdr=(AliAODMCHeader*)fAODEvent->GetList()->FindObject(AliAODMCHeader::StdBranchName());
-          if(!mcHdr) {
-            Printf("MC header branch not found!\n");
-            return;
-          }
-
          AliMCEvent* mcEvent = MCEvent();
-          _nTracks = mcEvent->GetNumberOfTracks();
+
+         _nTracks = mcEvent->GetNumberOfTracks();
           _mult3    = _nTracks;
           //loop over tracks starts here                                                                                                       
           for (int iTrack=0; iTrack< _nTracks; iTrack++)
@@ -1278,7 +1267,7 @@ void  AliDptDptInMC::UserExec(Option_t */*option*/)
                 }
          
 
-             //if(!aodTrack->IsPhysicalPrimary()) continue;
+             if(!aodTrack->IsPhysicalPrimary()) continue;
 
              q      = aodTrack->Charge();
               charge = int(q);
@@ -1515,8 +1504,8 @@ void  AliDptDptInMC::UserExec(Option_t */*option*/)
                 nsigmakaon      > fNSigmaCut &&
                 nsigmaproton    > fNSigmaCut ) continue;
 
-             //Float_t dcaXY = t->DCA();     
-             //Float_t dcaZ  = t->ZAtDCA();  
+             Float_t dcaXY = t->DCA();     
+             Float_t dcaZ  = t->ZAtDCA();  
              
              // Kinematics cuts 
              if( pt < 0.2 || pt > 2.0)  continue;
@@ -1526,7 +1515,7 @@ void  AliDptDptInMC::UserExec(Option_t */*option*/)
              AliAODMCParticle *AODmcTrack = (AliAODMCParticle*) fArrayMC->At(label);
 
              //W/Wo Secondaries
-             //if (!AODmcTrack->IsPhysicalPrimary()) continue;
+             if (!AODmcTrack->IsPhysicalPrimary()) continue;
              
              if (AODmcTrack)
                {
@@ -1535,7 +1524,8 @@ void  AliDptDptInMC::UserExec(Option_t */*option*/)
           
              _etadis->Fill(eta);
              _phidis->Fill(phi);
-
+             _dcaz->Fill(dcaZ);
+             _dcaxy->Fill(dcaXY);
              
              //Particle 1                                                                                                                  
              if (t->Charge() > 0)