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 TList;
   8 class AliESDEvent;
   9 #include "TF1.h"
  10
  11 #include "PWG2spectra/SPECTRA/AliProtonAnalysis.h"
  12 #include "AliAnalysisTask.h"
  13
  14 class AliAnalysisTaskProtons : public AliAnalysisTask {
  15  public:
  16   AliAnalysisTaskProtons(const char *name = "AliAnalysisTaskProtons");
  17   virtual ~AliAnalysisTaskProtons() {}
  18
  19   virtual void   ConnectInputData(Option_t *);
  20   virtual void   CreateOutputObjects();
  21   virtual void   Exec(Option_t *option);
  22   virtual void   Terminate(Option_t *);
  23
  24   void SetPriorProbabilityFunctions(TF1 *felectrons,
  25                                     TF1 *fmuons,
  26                                     TF1 *fpions,
  27                                     TF1 *fkaons,
  28                                     TF1 *fprotons) {
  29     fFunctionUsed = kTRUE;
  30     fElectronFunction = felectrons;
  31     fMuonFunction = fmuons;
  32     fPionFunction = fpions;
  33     fKaonFunction = fkaons;
  34     fProtonFunction = fprotons;
  35   }
  36
  37  private:
  38   AliESDEvent *fESD;    //ESD object
  39   TList  *fList; //TList output object
  40   AliProtonAnalysis *fAnalysis; //analysis object
  41   TF1 *fElectronFunction; //TF1 for e
  42   TF1 *fMuonFunction; //TF1 for mu
  43   TF1 *fPionFunction; //TF1 for pi
  44   TF1 *fKaonFunction; //TF1 for K
  45   TF1 *fProtonFunction; //TF1 for p
  46
  47   Bool_t fFunctionUsed; //kTRUE if Functions are used
  48
  49   AliAnalysisTaskProtons(const AliAnalysisTaskProtons&); // not implemented
  50   AliAnalysisTaskProtons& operator=(const AliAnalysisTaskProtons&); // not implemented
  51
  52   ClassDef(AliAnalysisTaskProtons, 1); // example of analysis
  53 };
  54
  55 #endif