PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalJetMass.h

   1 #ifndef ALIANALYSISTASKEMCALJETMASS_H
   2 #define ALIANALYSISTASKEMCALJETMASS_H
   3
   4 class TH1;
   5 class TH2;
   6 class TH3;
   7 class TH3F;
   8 class THnSparse;
   9 class TClonesArray;
  10 class TArrayI;
  11 class AliAnalysisManager;
  12 class AliJetContainer;
  13
  14 #include "AliAnalysisTaskEmcalJet.h"
  15
  16 class AliAnalysisTaskEmcalJetMass : public AliAnalysisTaskEmcalJet {
  17  public:
  18
  19   AliAnalysisTaskEmcalJetMass();
  20   AliAnalysisTaskEmcalJetMass(const char *name);
  21   virtual ~AliAnalysisTaskEmcalJetMass();
  22
  23   void                                UserCreateOutputObjects();
  24   void                                Terminate(Option_t *option);
  25
  26   //Setters
  27   void SetJetContainerBase(Int_t c)                             { fContainerBase     = c   ; }
  28   void SetMinFractionShared(Double_t f)                         { fMinFractionShared = f   ; }
  29   void SetJetMassAverage(Double_t m)                            { fJetMassAvg        = m   ; }
  30
  31  protected:
  32   Bool_t                              RetrieveEventObjects();
  33   Bool_t                              Run();
  34   Bool_t                              FillHistograms();
  35
  36   Double_t                            GetJetMass(AliEmcalJet *jet);
  37   TLorentzVector                      GetSubtractedVector(AliEmcalJet *jet);
  38   TLorentzVector                      GetBkgVector(AliEmcalJet *jet, AliJetContainer *cont);
  39
  40   Int_t                               fContainerBase;              // jets to be analyzed
  41   Double_t                            fMinFractionShared;          // only fill histos for jets if shared fraction larger than X
  42   Double_t                            fJetMassAvg;                 // average jet mass
  43
  44   TH2F            **fh2PtJet1VsLeadPtAllSel;         //!all jets after std selection vs leading track pt
  45   TH2F            **fh2PtJet1VsLeadPtTagged;         //!tagged jets vs leading track pt
  46   TH2F            **fh2PtJet1VsLeadPtTaggedMatch;    //!tagged jets vs leading track pt matched to MC
  47   TH2F            **fh2PtVsMassJet1All;              //!pT vs mass of all jets
  48   TH2F            **fh2PtVsMassJet1Tagged;           //!pT vs mass of tagged jets
  49   TH2F            **fh2PtVsMassJet1TaggedMatch;      //!pT vs mass of tagged jets matched to MC
  50   TProfile        **fpPtVsMassJet1All;               //!pT vs avg mass of all jets
  51   TProfile        **fpPtVsMassJet1Tagged;            //!pT vs avg mass of tagged jets
  52   TProfile        **fpPtVsMassJet1TaggedMatch;       //!pT vs avg mass of tagged jets matched to MC
  53   TH2F            **fh2MassVsAreaJet1All;            //!mass vs area of all jets
  54   TH2F            **fh2MassVsAreaJet1Tagged;         //!mass vs area of tagged jets
  55   TH2F            **fh2MassVsAreaJet1TaggedMatch;    //!mass vs area of tagged jets matched to MC
  56   TH2F            **fh2MassVsNConstJet1All;          //!mass vs number of constituents of all jets
  57   TH2F            **fh2MassVsNConstJet1Tagged;       //!mass vs number of constituents of tagged jets
  58   TH2F            **fh2MassVsNConstJet1TaggedMatch;  //!mass vs number of constituents of tagged jets matched to MC
  59   TH2F            **fh2EtMassOverEtRSq;           //!Et vs (M/Et*R)^2
  60
  61  private:
  62   AliAnalysisTaskEmcalJetMass(const AliAnalysisTaskEmcalJetMass&);            // not implemented
  63   AliAnalysisTaskEmcalJetMass &operator=(const AliAnalysisTaskEmcalJetMass&); // not implemented
  64
  65   ClassDef(AliAnalysisTaskEmcalJetMass, 4)
  66 };
  67 #endif
  68