PWGCF/Correlations/JCORRAN/AliJJetTask.h

   1 #ifndef ALIJJETTASK_H
   2 #define ALIJJETTASK_H
   3 class TH1;
   4 class TH1D;
   5 class TH2;
   6 class TH3;
   7 class TGraphErrors;
   8 class TProfile;
   9 class AliJetContainer;
  10 class AliParticleContainer;
  11 class AliClusterContainer;
  12
  13
  14 #include "AliAnalysisTaskEmcalJet.h"
  15 #include <TClonesArray.h>
  16 #include <TString.h>
  17
  18
  19 class AliJJetTask : public AliAnalysisTaskEmcalJet {
  20  public:
  21
  22   AliJJetTask();
  23   AliJJetTask(const char *name, const int NJetFinder);
  24   //AliJJetTask(const char * name, int njetfinder );
  25   virtual ~AliJJetTask();
  26
  27
  28   void                        UserCreateOutputObjects();
  29   void SetTrackArrayName( char *c ) { fTrackArrayName = c; }
  30   vector<TClonesArray>* GetAliJJetCollection() {return &fJJets;}
  31   TObjArray*           GetAliJJetList(int i) {return &fJJets[i]; }
  32   TClonesArray*        GetJTracks()  {return &fJTracks;}
  33   int GetNumberOfJetCollections() {return fNJetFinder;}
  34   //void SetNumberOfJetCollections(const int n) {
  35   //    fJetsCont.resize(n);
  36   //    fJJets.resize(n);
  37   //    fTracksCont.resize(n);
  38   //    fCaloClustersCont.resize(n);
  39   //    fNJetFinder=n;
  40   //}
  41
  42   int GetTaskEntry()        {return fTaskEntry;}
  43   void                        Terminate(Option_t *option);
  44
  45   void SetDebug(int n) {debug = n; }
  46   int  GetDebug(){ return debug; }
  47   vector<TString> &GetJetFinderString() { return fJetFinderString;}
  48
  49
  50  protected:
  51   void                        ExecOnce();
  52   Bool_t                      FillHistograms()   ;
  53   Bool_t                      Run()              ;
  54   void                        CheckClusTrackMatching();
  55
  56
  57
  58
  59
  60
  61   vector<AliJetContainer*>            fJetsCont;                   //!Jets
  62   //AliJetContainer            **fJetsConts;              //!Jets
  63   //Int_t                    nJetsConts;
  64   vector<AliParticleContainer*>       fTracksCont;                 //!Tracks
  65   vector<AliClusterContainer*>        fCaloClustersCont;           //!Clusters
  66
  67  private:
  68   AliJJetTask(const AliJJetTask& ap);            // not implemented
  69   AliJJetTask &operator=(const AliJJetTask&); // not implemented
  70
  71   //TVector *EtaGapThresholds;
  72   //TVector *RGapThresholds;
  73   //TVector *KlongBorders;
  74   //TVector *XeBorders;
  75
  76   TClonesArray              fJTracks;  //! tracks array
  77   vector<TClonesArray>      fJJets;  //! tracks array
  78   UInt_t                    fTaskEntry;
  79
  80   vector<TString> fJetFinderString;
  81
  82
  83
  84   TString    fTrackArrayName; // track constituents array name
  85
  86
  87   Int_t fNJetFinder;
  88   Int_t debug;
  89
  90   ClassDef(AliJJetTask, 4) // jet sample analysis task
  91
  92
  93
  94 };
  95
  96
  97 #endif