PWG2/AliAnalysisTaskProtons.h

   1 #ifndef AliAnalysisTaskProtons_cxx
   2 #define AliAnalysisTaskProtons_cxx
   3
   4 // Analysis task creating a the 2d y-p_t spectrum of p and antip
   5 // Author: Panos Cristakoglou
   6
   7 class TString;
   8 class TList;
   9 class AliESDEvent;
  10 class AliAODEvent;
  11 #include "TF1.h"
  12
  13 #include "PWG2spectra/SPECTRA/AliProtonAnalysis.h"
  14 #include "AliAnalysisTask.h"
  15
  16 class AliAnalysisTaskProtons : public AliAnalysisTask {
  17  public:
  18   AliAnalysisTaskProtons(const char *name = "AliAnalysisTaskProtons");
  19   virtual ~AliAnalysisTaskProtons() {}
  20
  21   virtual void   ConnectInputData(Option_t *);
  22   virtual void   CreateOutputObjects();
  23   virtual void   Exec(Option_t *option);
  24   virtual void   Terminate(Option_t *);
  25
  26   void SetType(const char* type) {fAnalysisType = type;}
  27   void SetPriorProbabilityFunctions(TF1 *felectrons,
  28                                     TF1 *fmuons,
  29                                     TF1 *fpions,
  30                                     TF1 *fkaons,
  31                                     TF1 *fprotons) {
  32     fFunctionUsed = kTRUE;
  33     fElectronFunction = felectrons;
  34     fMuonFunction = fmuons;
  35     fPionFunction = fpions;
  36     fKaonFunction = fkaons;
  37     fProtonFunction = fprotons;
  38   }
  39
  40  private:
  41   AliESDEvent *fESD;    //ESD object
  42   AliAODEvent *fAOD;    //AOD object
  43   TString fAnalysisType;//"ESD" or "AOD"
  44   TList  *fList; //TList output object
  45   AliProtonAnalysis *fAnalysis; //analysis object
  46   TF1 *fElectronFunction; //TF1 for e
  47   TF1 *fMuonFunction; //TF1 for mu
  48   TF1 *fPionFunction; //TF1 for pi
  49   TF1 *fKaonFunction; //TF1 for K
  50   TF1 *fProtonFunction; //TF1 for p
  51
  52   Bool_t fFunctionUsed; //kTRUE if Functions are used
  53
  54   AliAnalysisTaskProtons(const AliAnalysisTaskProtons&); // not implemented
  55   AliAnalysisTaskProtons& operator=(const AliAnalysisTaskProtons&); // not implemented
  56
  57   ClassDef(AliAnalysisTaskProtons, 1); // example of analysis
  58 };
  59
  60 #endif