Merge remote-tracking branch 'origin/flatdev' into mergeFlat2Master

[u/mrichter/AliRoot.git] / PWG / EMCAL / AliPicoTrack.h

diff --git a/PWG/EMCAL/AliPicoTrack.h b/PWG/EMCAL/AliPicoTrack.h
index 3110db2e951ea76d4833dc803e14d2a93be2d6f5..93ff68da84dd65071362ac6b5998c912da4aa9b7 100644 (file)
--- a/PWG/EMCAL/AliPicoTrack.h
+++ b/PWG/EMCAL/AliPicoTrack.h
@@ -1,10 +1,11 @@
 #ifndef AliPicoTrack_H
 #define AliPicoTrack_H
 
-// $Id$
+#include <TMath.h>
 
 #include "AliVTrack.h"
-#include <TMath.h>
+#include "AliAODMCParticle.h"
+
 class AliVCluster;
 
 class AliPicoTrack: public AliVTrack {
@@ -50,7 +51,7 @@ class AliPicoTrack: public AliVTrack {
   Bool_t          IsEMCAL()                   const { return fEmcal;  }
   ULong_t         GetStatus()                 const { return 0;       }
   Bool_t          GetXYZ(Double_t *v)         const { v[0]=0; v[1]=0; v[2]=0; return 0; }
-  using AliVVtrack::GetXYZ;
+  using AliVTrack::GetXYZ;
   Double_t        GetBz()                     const { return 0;       }
   void            GetBxByBz(Double_t b[3])    const { b[0]=0;b[1]=0;b[2]=0; }
   Bool_t          GetCovarianceXYZPxPyPz(Double_t /*cv*/[21]) const { return 0; }
@@ -64,25 +65,56 @@ class AliPicoTrack: public AliVTrack {
   void            Clear(Option_t * /*option*/ ="")  { fClusId = -1; fOrig = 0; } 
   Int_t           Compare(const TObject* obj) const;
   Bool_t          PropagateToDCA(const AliVVertex *, Double_t, Double_t, Double_t *, Double_t *) { return 0; }
+  void            SetFlag(UInt_t flag) {fFlag = flag;}
+  UInt_t          GetFlag() const {return fFlag;}
+  void SetPrimary(Bool_t b = kTRUE){
+    if(b)fFlag |= AliAODMCParticle::kPrimary;
+    else fFlag &= ~AliAODMCParticle::kPrimary;
+  }
+  Bool_t IsPrimary() const {return ((fFlag&AliAODMCParticle::kPrimary)==AliAODMCParticle::kPrimary);} 
+  
+  void SetPhysicalPrimary(Bool_t b = kTRUE){
+    if(b)fFlag |= AliAODMCParticle::kPhysicalPrim;
+    else fFlag &= ~AliAODMCParticle::kPhysicalPrim; 
+  } 
+  Bool_t IsPhysicalPrimary() const {return ((fFlag&AliAODMCParticle::kPhysicalPrim)==AliAODMCParticle::kPhysicalPrim);} 
+  
+  void SetSecondaryFromWeakDecay(Bool_t b = kTRUE){
+    if(b)fFlag |= AliAODMCParticle::kSecondaryFromWeakDecay;
+    else fFlag &= ~AliAODMCParticle::kSecondaryFromWeakDecay; 
+  } 
+  Bool_t IsSecondaryFromWeakDecay() const {return ((fFlag&AliAODMCParticle::kSecondaryFromWeakDecay)==AliAODMCParticle::kSecondaryFromWeakDecay);} 
+  
+  void SetSecondaryFromMaterial(Bool_t b = kTRUE){
+    if(b)fFlag |= AliAODMCParticle::kSecondaryFromMaterial;
+    else fFlag &= ~AliAODMCParticle::kSecondaryFromMaterial; 
+  } 
+  Bool_t IsSecondaryFromMaterial() const {return ((fFlag&AliAODMCParticle::kSecondaryFromMaterial)==AliAODMCParticle::kSecondaryFromMaterial);} 
+
+  Bool_t  IsFromHIJING() const { return (fGeneratorIndex == 0); }
+  void    SetGeneratorIndex(Short_t i) {fGeneratorIndex = i;}
+  Short_t GetGeneratorIndex() const {return fGeneratorIndex;}
 
   static void     GetEtaPhiDiff(const AliVTrack *t, const AliVCluster *v, Double_t &phidiff, Double_t &etadiff);
   static Byte_t   GetTrackType(const AliVTrack *t);
 
  protected:
-  Double32_t      fPt;       //[0,0,12]   pt at vertex
-  Double32_t      fEta;      //[-1,1,12]  eta at vertex
-  Double32_t      fPhi;      //[0,6.3,12] phi at vertex
-  Double32_t      fM;        //           mass
-  Byte_t          fQ;        //           charge
-  Int_t           fLabel;    //           label  
-  Byte_t          fTrackType;//           0=global track; 1=w/o SPD, w/ ITS refit; 2=w/o SPD, w/o ITS refit
-  Double32_t      fEtaEmc;   //[-1,1,12]  eta at emcal surface
-  Double32_t      fPhiEmc;   //[0,6.3,12] phi at emcal surface
-  Double32_t      fPtEmc;    //[0,0,12]   pt at emcal surface
-  Bool_t          fEmcal;    //           is true if track propagated to emcal
-  Short_t         fClusId;   //!          cluster id of matched cluster; -1 if not set
-  AliVTrack      *fOrig;     //!          ptr to original track
+  Double32_t      fPt;             //[0,0,12]   pt at vertex
+  Double32_t      fEta;            //[-1,1,12]  eta at vertex
+  Double32_t      fPhi;            //[0,6.3,12] phi at vertex
+  Double32_t      fM;              //           mass
+  Byte_t          fQ;              //           charge
+  Int_t           fLabel;          //           label  
+  Byte_t          fTrackType;      //           0=global track; 1=w/o SPD, w/ ITS refit; 2=w/o SPD, w/o ITS refit
+  Double32_t      fEtaEmc;         //[-1,1,12]  eta at emcal surface
+  Double32_t      fPhiEmc;         //[0,6.3,12] phi at emcal surface
+  Double32_t      fPtEmc;          //[0,0,12]   pt at emcal surface
+  Bool_t          fEmcal;          //           is true if track propagated to emcal
+  UInt_t          fFlag;           //!          Flag for indication of primary etc (MC)
+  Short_t         fGeneratorIndex; //!          Index of generator in cocktail
+  Short_t         fClusId;         //!          cluster id of matched cluster; -1 if not set
+  AliVTrack      *fOrig;           //!          ptr to original track
 
-  ClassDef(AliPicoTrack, 7) // Pico track class
+  ClassDef(AliPicoTrack, 8) // Pico track class
 };
 #endif