]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalJetHMEC.h
Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskEmcalJetHMEC.h
1 #ifndef AliAnalysisTaskEmcalJetHMEC_H
2 #define AliAnalysisTaskEmcalJetHMEC_H
3
4 // $Id$
5
6 class TClonesArray;
7 class TList;
8 class TH1;
9 class TH2;
10 class THnSparse;
11 class AliEmcalJet;
12 class AliESDEvent;
13 class AliAODEvent;
14 class AliEventPoolManager;
15
16 #include "AliAnalysisTaskEmcalJet.h"
17
18 class AliAnalysisTaskEmcalJetHMEC : public AliAnalysisTaskEmcalJet {
19  public:
20   AliAnalysisTaskEmcalJetHMEC();
21   AliAnalysisTaskEmcalJetHMEC(const char *name);
22   virtual ~AliAnalysisTaskEmcalJetHMEC() {}
23   
24   virtual void            UserCreateOutputObjects();
25   virtual Double_t        RelativePhi(Double_t mphi, Double_t vphi);
26 //  virtual void            UserExec(Option_t *option);
27   virtual void            Terminate(Option_t *);
28   virtual Int_t           AcceptthisJet(AliEmcalJet *jet);
29   virtual THnSparse*      NewTHnSparseF(const char* name, UInt_t entries);
30   virtual void            GetDimParams(Int_t iEntry,TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);
31
32   virtual void            SetTracksName(const char *n)             {fTracksName=n;}
33   virtual void            SetJetsName(const char *jn)              {fJetsName=jn;}
34
35   virtual void            SetAreaCut(Double_t a)                   { fAreacut    = a; }
36   virtual void            SetTrkBias(Double_t b)                   { fTrkBias    = b; }  //require a track with pt > b in jet
37   virtual void            SetClusBias(Double_t b)                  { fClusBias   = b; }  //require a cluster with pt > b in jet
38
39   virtual void            SetTrkEta(Double_t e)                    { fTrkEta   = e; }  //eta range of the associated tracks
40
41   virtual void            SetJetEta(Double_t emin, Double_t emax)  { fEtamin = emin; fEtamax = emax; }
42   virtual void            SetJetPhi(Double_t pmin, Double_t pmax)  { fPhimin = pmin; fPhimax = pmax; }
43   virtual void            SetEventMixing(Int_t yesno)              { fDoEventMixing=yesno;}
44   virtual void            SetMixingTracks(Int_t tracks)            { fMixingTracks = tracks; }
45
46
47  protected:
48   void                                   ExecOnce();
49   Bool_t                             Run();
50   virtual Int_t          GetCentBin(Double_t cent) const;
51   virtual Int_t          GetEtaBin(Double_t eta) const;
52   virtual Int_t          GetpTjetBin(Double_t pt) const;
53
54   TString                fTracksName;              // name of tracks collection
55   TString                fJetsName;                // name of Jet collection
56   Double_t               fPhimin;                  // phi min of jet
57   Double_t               fPhimax;                  // phi max of jet
58   Double_t               fEtamin;                  // eta min of jet
59   Double_t               fEtamax;                  // eta max of jet
60   Double_t               fAreacut;                 // area cut of jet
61   Double_t               fTrkBias;
62   Double_t               fClusBias;
63   Double_t               fTrkEta;                  // eta min/max of tracks
64   Int_t                  fDoEventMixing;           // flag to do evt mixing
65   Int_t                          fMixingTracks;                // size of track buffer for event mixing
66   TObjArray*             CloneAndReduceTrackList(TObjArray* tracks);
67
68   AliESDEvent           *fESD;    //! ESD object
69   AliAODEvent                   *fAOD;    //! AOD object
70   AliEventPoolManager   *fPoolMgr; //!
71   TH1                   *fHistTrackPt; //! Pt spectrum
72   TH1                   *fHistCentrality;//!
73   TH2                   *fHistJetEtaPhi;//!
74   TH2                   *fHistTrackEtaPhi[7];//!
75   TH2                   *fHistJetHEtaPhi;//!
76
77   TH1                   *fHistJetPt[6]; //!
78   TH1                   *fHistJetPtBias[6];//!
79   TH1                   *fHistLeadJetPt[6];//!
80   TH1                   *fHistLeadJetPtBias[6];//!
81   TH1                   *fHistJetPtTT[6];//!
82   TH2                   *fHistJetH[6][5][3];//!
83   TH2                   *fHistJetHBias[6][5][3];//!
84   TH2                   *fHistJetHTT[6][5][3];//!
85   THnSparse             *fhnMixedEvents;      //!mixed events matrix
86   THnSparse             *fhnJH;      //!Fg events matrix
87
88  private:
89    
90   AliAnalysisTaskEmcalJetHMEC(const AliAnalysisTaskEmcalJetHMEC&); // not implemented
91   AliAnalysisTaskEmcalJetHMEC& operator=(const AliAnalysisTaskEmcalJetHMEC&); // not implemented
92   
93   ClassDef(AliAnalysisTaskEmcalJetHMEC, 10); 
94 };
95 #endif