PWG2/FLOW/AliFlowCommon/AliFlowTrackSimple.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 ALIFLOWTRACKSIMPLE_H
   6 #define ALIFLOWTRACKSIMPLE_H
   7
   8 #include "TNamed.h"
   9 #include "TBits.h"
  10
  11 // AliFlowTrackSimple:
  12 // A simple track class to the the AliFlowEventSimple for flow analysis
  13 // author: N. van der Kolk (kolk@nikhef.nl)
  14
  15 class AliFlowTrackSimple: public TNamed {
  16
  17  public:
  18   AliFlowTrackSimple();
  19   AliFlowTrackSimple(const AliFlowTrackSimple& aTrack);
  20   AliFlowTrackSimple& operator=(const AliFlowTrackSimple& aTrack);
  21   virtual  ~AliFlowTrackSimple();
  22
  23   Bool_t  IsFolder() const {return kTRUE;};
  24   //  void Browse(TBrowser *b);
  25   //  void Print(Option_t* option = "") const;      //method to print stats
  26
  27
  28   Double_t Eta() const;
  29   Double_t Pt()  const;
  30   Double_t Phi() const;
  31   TBits    GetFlowBits() const;
  32   Bool_t   InRPSelection() const;
  33   Bool_t   InPOISelection() const;
  34   Bool_t   InSubevent(Int_t i) const;
  35
  36   void SetEta(Double_t eta);
  37   void SetPt(Double_t pt);
  38   void SetPhi(Double_t phi);
  39   void SetForRPSelection(Bool_t b);
  40   void SetForPOISelection(Bool_t b);
  41   void SetForSubevent(Int_t i);
  42
  43  private:
  44   Double_t fEta;    // eta
  45   Double_t fPt;     // pt
  46   Double_t fPhi;    // phi
  47   TBits fFlowBits;  // bits to set if track is selected
  48   TBits fSubEventBits;  // bits to set if track is selected for a subevent
  49
  50
  51   ClassDef(AliFlowTrackSimple,1)                 // macro for rootcint
  52
  53 };
  54
  55 inline Double_t AliFlowTrackSimple::Eta() const {
  56   return this->fEta; }
  57 inline Double_t AliFlowTrackSimple::Pt() const {
  58   //  cout << "Returned pt:" << fPt << endl;
  59   return this->fPt;}
  60 inline Double_t AliFlowTrackSimple::Phi() const {
  61   return this->fPhi; }
  62 //TBits
  63 inline TBits AliFlowTrackSimple::GetFlowBits() const {
  64   return this->fFlowBits; }
  65 inline Bool_t AliFlowTrackSimple::InRPSelection() const {
  66   return this->fFlowBits.TestBitNumber(0); }
  67 inline Bool_t AliFlowTrackSimple::InPOISelection() const {
  68   return this->fFlowBits.TestBitNumber(1); }
  69 inline Bool_t AliFlowTrackSimple::InSubevent(Int_t i) const {
  70   return this->fSubEventBits.TestBitNumber(i); }
  71
  72 inline void AliFlowTrackSimple::SetEta(Double_t val) {
  73   fEta = val; }
  74 inline void AliFlowTrackSimple::SetPt(Double_t val) {
  75   fPt = val; }
  76   //  cout << "pt set to:" << fPt << endl;}
  77 inline void AliFlowTrackSimple::SetPhi(Double_t val) {
  78   fPhi = val; }
  79 //TBits
  80 inline void AliFlowTrackSimple::SetForRPSelection(Bool_t val) {
  81   fFlowBits.SetBitNumber(0,val); }
  82 inline void AliFlowTrackSimple::SetForPOISelection(Bool_t val) {
  83   fFlowBits.SetBitNumber(1,val); }
  84 inline void AliFlowTrackSimple::SetForSubevent(Int_t i) {
  85   fSubEventBits.SetBitNumber(i,kTRUE); }
  86
  87 #endif
  88