framework updates, including Track rotation method (Jens); updated filtering macro

[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectronMC.h

diff --git a/PWG3/dielectron/AliDielectronMC.h b/PWG3/dielectron/AliDielectronMC.h
index ab774156dc0eceadb674cb802ca8adf8729dd576..6be9d961ffe6c7371ed25cea6f940a3cdc80b328 100644 (file)
--- a/PWG3/dielectron/AliDielectronMC.h
+++ b/PWG3/dielectron/AliDielectronMC.h
@@ -3,6 +3,8 @@
 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
+/* $Id$ */ 
+
 //#####################################################
 //#                                                   # 
 //#              Class AliDielectronMC                #
@@ -20,6 +22,7 @@ class AliHFEpid;
 class AliStack;
 class AliMCEvent;
 class AliESDtrack;
+class AliAODTrack;
 class TParticle;
 class AliMCParticle;
 class AliAODMCParticle;
@@ -42,9 +45,13 @@ public:
   void Initialize();                              // initialization
   Int_t GetNMCTracks();                                     // return number of generated tracks
   Int_t GetNMCTracksFromStack();                            // return number of generated tracks from stack
+  Int_t GetNPrimary();                                            // return number of primary tracks
+  Int_t GetNPrimaryFromStack();                                   // return number of primary tracks from stack
   Int_t GetMCPID(const AliESDtrack* _track);                      // return MC PID
+  Int_t GetMCPID(const AliAODTrack* _track);                      // return MC PID for AODtrack
   Int_t GetMCPIDFromStack(const AliESDtrack* _track);             // return MC PID
   Int_t GetMotherPDG(const AliESDtrack* _track);                  // return mother PID from the MC stack
+  Int_t GetMotherPDG(const AliAODTrack* _track);                  // return mother PID from the MC stack
   Int_t GetMotherPDGFromStack(const AliESDtrack* _track);         // return mother PID from the MC stack
   Int_t GetMCProcess(const AliESDtrack* _track);                  // return process number
   Int_t GetMCProcessFromStack(const AliESDtrack* _track);         // return process number
@@ -64,20 +71,25 @@ public:
   Int_t GetLabelMotherWithPdg(const AliVParticle *particle1, const AliVParticle *particle2, Int_t pdgMother);
   
   AliVParticle* GetMCTrackFromMCEvent(AliVParticle *track);   // return MC track directly from MC event
-  AliVParticle* GetMCTrackFromMCEvent(Int_t _itrk);           // return MC track directly from MC event
+  AliVParticle* GetMCTrackFromMCEvent(Int_t itrk);           // return MC track directly from MC event
   TParticle* GetMCTrackFromStack(const AliESDtrack* _track);        // return MC track from stack
   AliMCParticle* GetMCTrack(const AliESDtrack* _track);             // return MC track associated with reco track
+  AliAODMCParticle* GetMCTrack( const AliAODTrack* _track);          // return MC track associated with reco AOD track
   
   TParticle* GetMCTrackMotherFromStack(const AliESDtrack* _track);  // return MC mother track from stack
   AliMCParticle* GetMCTrackMother(const AliESDtrack* _track);       // return MC mother track from stack
+  AliAODMCParticle* GetMCTrackMother(const AliAODTrack* _track);       // return MC mother fot track AODTrack
   AliMCParticle* GetMCTrackMother(const AliMCParticle* _particle);       // return MC mother track from stack
   AliAODMCParticle* GetMCTrackMother(const AliAODMCParticle* _particle);       // return MC mother track from stack
 
   Int_t NumberOfDaughters(const AliESDtrack* track);                 // return number of daughters
+  Int_t NumberOfDaughters(const AliAODTrack* track);                 // return number of daughters for AOD track
   Int_t NumberOfDaughters(const AliMCParticle* particle);                 // return number of daughters
   Int_t NumberOfDaughters(const AliAODMCParticle* particle);                 // return number of daughters
 
   void GetDaughters(const TObject *mother, AliVParticle* &d1, AliVParticle* &d2);
+  Int_t IsJpsiPrimary(const AliDielectronPair * pair);
+  Int_t IsJpsiPrimary(const AliVParticle * pair);
   
 private:
   AliMCEvent    *fMCEvent;  // MC event object
@@ -87,7 +99,9 @@ private:
   Bool_t fHasMC;            // Do we have an MC handler?
   
   static AliDielectronMC* fgInstance; //! singleton pointer
-  
+  TClonesArray* fMcArray; //mcArray for AOD MC particles 
+
+ 
   AliDielectronMC(const AliDielectronMC &c);
   AliDielectronMC &operator=(const AliDielectronMC &c);
 
@@ -116,4 +130,5 @@ inline Int_t AliDielectronMC::GetLabelMotherWithPdg(const AliDielectronPair* pai
   return GetLabelMotherWithPdg(pair->GetFirstDaughter(),pair->GetSecondDaughter(),pdgMother);
 }
 
-#endif 
+#endif
+