PWGGA/GammaConv/AliAnalysisTaskMaterial.h

   1 #ifndef AliAnalysisTaskMaterial_cxx
   2 #define AliAnalysisTaskMaterial_cxx
   3
   4 #include "AliAnalysisTaskSE.h"
   5 #include "AliConversionPhotonBase.h"
   6 #include "TH1.h"
   7 #include "TH2.h"
   8 #include "TTreeStream.h"
   9 #include "AliLog.h"
  10 #include <vector>
  11 #include "AliV0ReaderV1.h"
  12 #include "AliConversionCuts.h"
  13 #include "TList.h"
  14 #include "AliStack.h"
  15 #include "TClonesArray.h"
  16
  17
  18 using namespace std;
  19
  20
  21 class AliAnalysisTaskMaterial : public AliAnalysisTaskSE{
  22
  23 public:
  24
  25     AliAnalysisTaskMaterial(const char *name);
  26     virtual ~AliAnalysisTaskMaterial();
  27
  28     virtual void   UserCreateOutputObjects();
  29     virtual void   UserExec(Option_t *option);
  30     virtual void   Terminate(Option_t *);
  31
  32     void SetV0Reader(AliV0ReaderV1 *v0Reader){fV0Reader=v0Reader;}
  33     void SetConversionCuts(AliConversionCuts* conversionCuts,Bool_t IsHeavyIon ){
  34        fConversionCuts=conversionCuts;
  35        fIsHeavyIon = IsHeavyIon;
  36     }
  37
  38 private:
  39
  40     void ProcessPhotons();
  41          void ProcessMCPhotons();
  42          Int_t CountESDTracks();
  43
  44     AliV0ReaderV1 *fV0Reader;
  45     TClonesArray *fConversionGammas; //Reconstructed Photons;
  46     AliConversionCuts *fConversionCuts; // Cuts used by the V0Reader
  47     TTreeSRedirector *fStreamMaterial;
  48          TTreeSRedirector *fStreamResolution;
  49     Bool_t fIsHeavyIon;
  50     TList *fOutputList;
  51          AliESDEvent *fESDEvent;
  52          AliMCEvent *fMCEvent;
  53
  54     ClassDef(AliAnalysisTaskMaterial, 0);
  55 };
  56
  57 #endif
  58