AliAnalysisTaskEmcalJetMass.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKEMCALJETMASS_H
#define ALIANALYSISTASKEMCALJETMASS_H

class TH1;
class TH2;
class TH3;
class TH3F;
class THnSparse;
class TClonesArray;
class TArrayI;
class AliAnalysisManager;
class AliJetContainer;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskEmcalJetMass : public AliAnalysisTaskEmcalJet {
 public:
  enum JetMassType {
    kRaw   = 0,  //mass form anti-kt 4-vector
    kDeriv = 1   //area based subtracted jet mass
  };

  AliAnalysisTaskEmcalJetMass();
  AliAnalysisTaskEmcalJetMass(const char *name);
  virtual ~AliAnalysisTaskEmcalJetMass();

  void                                UserCreateOutputObjects();
  void                                Terminate(Option_t *option);

  //Setters
  void SetJetContainerBase(Int_t c)                                  { fContainerBase     = c   ; }
  void SetJetContainerUnsub(Int_t c)                                 { fContainerUnsub    = c   ; }
  void SetMinFractionShared(Double_t f, Bool_t useUnsubJet = kFALSE) { fMinFractionShared = f   ; SetUseUnsubJet(useUnsubJet); }
  void SetUseUnsubJet(Bool_t b)                                      { fUseUnsubJet       = b   ; }
  void SetJetMassType(JetMassType t)                                 { fJetMassType       = t   ; }
  void SetUseSumw2(Bool_t b)                                         { fUseSumw2          = b   ; }

 protected:
  Bool_t                              RetrieveEventObjects();
  Bool_t                              Run();
  Bool_t                              FillHistograms();

  Double_t                            GetJetMass(AliEmcalJet *jet);
 
 private:
  Int_t                               fContainerBase;              // jets to be analyzed
  Int_t                               fContainerUnsub;             // unsubtracted jets
  Double_t                            fMinFractionShared;          // only fill histos for jets if shared fraction larger than X
  Bool_t                              fUseUnsubJet;                // calc fraction of unsubtracted jet (relevant for constituent subtraction)
  JetMassType                         fJetMassType;                // jet mass type to be used
  Bool_t                              fUseSumw2;                   // activate sumw2 for output histograms

  TH3F            **fh3PtJet1VsMassVsLeadPtAllSel;         
  TH3F            **fh3PtJet1VsMassVsLeadPtTagged;         
  TH3F            **fh3PtJet1VsMassVsLeadPtTaggedMatch;    
  TProfile        **fpPtVsMassJet1All;                     
  TProfile        **fpPtVsMassJet1Tagged;                  
  TProfile        **fpPtVsMassJet1TaggedMatch;             
  TH2F            **fh2MassVsAreaJet1All;                  
  TH2F            **fh2MassVsAreaJet1Tagged;               
  TH2F            **fh2MassVsAreaJet1TaggedMatch;          
  TH2F            **fh2MassVsNConstJet1All;                
  TH2F            **fh2MassVsNConstJet1Tagged;             
  TH2F            **fh2MassVsNConstJet1TaggedMatch;        

  TH3F            *fh3PtJet1VsMassVsCentAllSel;            
  TH3F            *fh3PtJet1VsMassVsCentTagged;            
  TH3F            *fh3PtJet1VsMassVsCentTaggedMatch;       

  TH3F            **fh3PtJet1VsRatVsLeadPtAllSel;          
  TH3F            **fh3PtJet1VsRatVsLeadPtTagged;          
  TH3F            **fh3PtJet1VsRatVsLeadPtTaggedMatch;     
  TProfile        **fpPtVsRatJet1All;                      
  TProfile        **fpPtVsRatJet1Tagged;                   
  TProfile        **fpPtVsRatJet1TaggedMatch;              
  TH2F            **fh2RatVsAreaJet1All;                   
  TH2F            **fh2RatVsAreaJet1Tagged;                
  TH2F            **fh2RatVsAreaJet1TaggedMatch;           
  TH2F            **fh2RatVsNConstJet1All;                 
  TH2F            **fh2RatVsNConstJet1Tagged;              
  TH2F            **fh2RatVsNConstJet1TaggedMatch;         

  TH3F            **fh3JetPtVsMassVsEPRelAllSel;           
  TH3F            **fh3JetPtVsMassVsEPRelTagged;           
  TH3F            **fh3JetPtVsMassVsEPRelTaggedMatch;      
  THnSparse       *fhnJetPtMassAndBkg;                     
  
  AliAnalysisTaskEmcalJetMass(const AliAnalysisTaskEmcalJetMass&);            // not implemented
  AliAnalysisTaskEmcalJetMass &operator=(const AliAnalysisTaskEmcalJetMass&); // not implemented

  ClassDef(AliAnalysisTaskEmcalJetMass, 11)
};
#endif