PWG/FLOW/Tasks/AliFlowTrack.h

   1 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   2 * See cxx source for full Copyright notice */
   3 /* $Id$ */
   4
   5 #ifndef ALIFLOWTRACK_H
   6 #define ALIFLOWTRACK_H
   7
   8 #include "AliFlowTrackSimple.h"
   9 class AliVParticle;
  10
  11 // AliFlowTrack:
  12 // A track class to the the AliFlowEvent for flow analysis
  13 // origin: Mikolaj Krzewicki (mikolaj.krzewicki@cern.ch)
  14
  15 class AliFlowTrack: public AliFlowTrackSimple {
  16
  17 public:
  18   enum trackSource { kFromESD=0,
  19                      kFromMC=1,
  20                      kFromAOD=2,
  21                      kFromTracklet=3,
  22                      kFromFMD=4,
  23                      kFromPMD=5,
  24                      kFromV0=6,
  25                      kFromMUON=7,// XZhang 20120604
  26                      kFromKink };
  27   AliFlowTrack();
  28   AliFlowTrack(const AliVParticle* p);
  29   AliFlowTrack& operator=(const AliFlowTrack& aTrack);
  30   //virtual AliFlowTrackSimple& operator=(const AliFlowTrackSimple& aTrack);
  31   AliFlowTrack(const AliFlowTrack& aTrack);
  32   virtual  ~AliFlowTrack();
  33   virtual AliFlowTrack* Clone(const char* option="") const;
  34
  35   void Set(const AliVParticle* p);
  36
  37   void SetSource( trackSource s )
  38                   { fTrackSourceBits.SetBitNumber(UInt_t(s),kTRUE); }
  39   Bool_t IsSource( trackSource s ) const
  40                  { return fTrackSourceBits.TestBitNumber(s); }
  41
  42   virtual void Clear(Option_t* o="") {AliFlowTrackSimple::Clear(o); fTrackSourceBits.Clear();}
  43
  44 private:
  45   TBits fTrackSourceBits; //where do i come from?
  46
  47   ClassDef(AliFlowTrack,1);
  48 };
  49
  50 #endif
  51