PWG4/Base/AliAnalysisHelperJetTasks.h

   1 #ifndef  ALIANALYSISHELPERJETTASKS_H
   2 #define  ALIANALYSISHELPERJETTASKS_H
   3 #include "TObject.h"
   4
   5 class AliMCEvent;
   6 class AliAODJet;
   7 class AliVEvent;
   8 class TString;
   9 class TArrayI;
  10 class TArrayF;
  11 class AliGenPythiaEventHeader;
  12 class TVector3;
  13 class AliGenEventHeader;
  14
  15
  16 //
  17 // Helper Class that contains a lot of
  18 // usefull static functions jet matchin pythia access etc.
  19 //
  20
  21
  22 class AliAnalysisHelperJetTasks : public TObject {
  23  public:
  24   AliAnalysisHelperJetTasks() : TObject() {;}
  25   virtual ~AliAnalysisHelperJetTasks(){;}
  26
  27
  28   enum {kMaxJets = 6}; //  needed for array size not to fragemnt memory on the heap by many new/delete
  29
  30
  31   enum { kNone = 1<<0,
  32          kBunchBunch = 1<<1,
  33          kBunchEmpty = 1<<2,
  34          kEmptyEmpty= 1<<3,
  35          kV0A=1<<4,
  36          kV0C=1<<5,
  37          kNoV0BG=1<<6,
  38          kSPDFO=1<<7,
  39          kPhysicsSelection = 1<<8,
  40          kVertexIn = 1<<9,
  41          kIsCosmic = 1<<10,
  42          kIsPileUp = 1<<11,
  43          kIsMCND=1<<12,
  44          kIsMCDD=1<<13,
  45          kIsMCSD=1<<14,
  46          kTotalSelections = (1<<15) - 1};
  47
  48   enum Trigger {kAcceptAll = 0,kMB1,kMB2,kMB3,kSPDGFO,kTrigger}; //
  49   // same as in PWG0Helper
  50   enum MCProcessType { kInvalidProcess = -1, kND = 0x1, kDD = 0x2, kSD = 0x4, kOnePart = 0x8 };
  51
  52   static AliGenPythiaEventHeader*  GetPythiaEventHeader(const AliMCEvent *mcEvent);
  53   static void PrintStack(AliMCEvent *mcEvent,Int_t iFirst = 0,Int_t iLast = 0,Int_t iMaxPrint = 10);
  54   static void GetClosestJets(AliAODJet *genJets,
  55                              const Int_t &kGenJets,
  56                              const AliAODJet *recJets,
  57                              const Int_t &kRecJets,
  58                              Int_t *iGenIndex,
  59                              Int_t *iRecIndex,
  60                              Int_t iDebug = 0, Float_t maxDist = 0.3);
  61
  62   static void GetClosestJets(const TList *genJetsList,const Int_t &kGenJets,
  63                              const TList *recJetsList,const Int_t &kRecJets,
  64                              TArrayI &iGenIndex,TArrayI &iRecIndex,
  65                              Int_t iDebug = 0,Float_t maxDist = 0.3);
  66
  67   static void GetJetMatching(const TList *genJetsList, const Int_t &kGenJets,
  68                              const TList *recJetsList, const Int_t &kRecJets,
  69                              TArrayI &iMatchIndex, TArrayF &fPtFraction,
  70                              Int_t iDebug = 0, Float_t maxDist = 0.3, Int_t mode=1);
  71
  72   static Double_t GetFractionOfJet(const AliAODJet *recJet,const AliAODJet *genJet,Int_t mode=1);
  73
  74
  75   static void MergeOutputDirs(const char* cFiles,const char* cPattern,const char *cOutFile,Bool_t bUpdate = false); // merges all directories containing the pattern
  76
  77   static void MergeOutput(char* cFiles, char* cDir = "",char *cList = "",char* cOutFile ="allpt.root",Bool_t bUpdate = false); // Merges the files in the input text file  needs the two histograms fh1PtHard_Trials, fh1Xsec and the name of the input list
  78   static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root
  79   static Bool_t PrintDirectorySize(const char* currFile,Int_t iDetail = -1); // print the size of the output on a given file
  80   static Bool_t GetEventShapes(TVector3 &n01,const TVector3 * pTrack, Int_t nTracks, Double_t * eventShapes);
  81
  82   static MCProcessType GetPythiaEventProcessType(AliGenEventHeader* aHeader, Bool_t adebug = kFALSE);
  83   static MCProcessType GetDPMjetEventProcessType(AliGenEventHeader* aHeader, Bool_t adebug = kFALSE);
  84   static Int_t GetLastProcessType() { return fgLastProcessType; }
  85
  86   static Bool_t Selected(Bool_t bSet = kFALSE,Bool_t bNew = kTRUE); // static function to store the state of selection from service task
  87   static Double_t ReactionPlane(Bool_t bSet = kFALSE,Double_t fNew = 0);
  88   static Int_t GetPhiBin(Double_t phi,Int_t fNRPbins);
  89
  90   static Bool_t IsPileUp(); // Wrapper for SelectInfo with PileUp
  91   static Bool_t IsCosmic(); // Wrapper for SelectInfo with cosmic
  92   static Bool_t TestSelectInfo(UInt_t iMask); // Wrapper for testing the SelectInfo bitmask
  93   static Bool_t TestEventClass(Int_t iClass); // Wrapper for testing the SelectInfo bitmask
  94
  95   static UInt_t SelectInfo(Bool_t bSet = kFALSE,UInt_t iNew = 0); // static function to store the state bitmask of the selection from service task
  96   static Int_t  EventClass(Bool_t bSet = kFALSE,Int_t iNew = 0); // static function to store the event class of the selection from service task
  97
  98   // these methods have been essentially copied from PWG0/AliTriggerAnalysis and expanded to use with AOD
  99   static Bool_t IsTriggerFired(const AliVEvent* aEsd, Trigger trigger);
 100
 101   private:
 102
 103   static Int_t fgLastProcessType;    // stores the raw value of the last process type extracted
 104
 105   ClassDef(AliAnalysisHelperJetTasks, 7)
 106 };
 107
 108 #endif // ALIANALYSISHELPERJETTASKS_H