AliAnalysisTaskJetShapeBase.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKJETSHAPEBASE_H
#define ALIANALYSISTASKJETSHAPEBASE_H

class TH1;
class TH2;
class TH3F;
class THnSparse;
class TF1;
class TArrayI;
class TTree;
class TLorentzVector;
class AliAnalysisManager;
class AliVParticle;
class AliJetContainer;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskJetShapeBase : public AliAnalysisTaskEmcalJet {
 public:
  enum JetMassVarType {
    kMass   = 0,  //jet mass
    kRatMPt = 1    //ratio mass/pt jet
  };
  enum ResponseReference {
    kDet    = 0,   //detector level
    kPart   = 1    //particle level
  };

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

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

  //Setters
  void SetCreateTree(Bool_t b)                                  { fCreateTree        = b   ; }

  void SetJetContainerBase(Int_t c)                             { fContainerBase     = c   ; }
  void SetJetContainerNoEmb(Int_t c)                            { fContainerNoEmb    = c   ; }
  void SetJetContainerOverlap(Int_t c)                          { fContainerOverlap  = c   ; }
  void SetMinFractionShared(Double_t f)                         { fMinFractionShared = f   ; }
  void SetSingleTrackEmbedding(Bool_t b, Int_t min = 99999, Int_t max = 999999) { fSingleTrackEmb = b; fMinLabelEmb = min; fMaxLabelEmb = max; }
  void SetRemoveOverlapTrackJet(Bool_t b, Double_t r)                       { fOverlap = b; fRadius = r; if(!fSingleTrackEmb) Printf("No effect, since fSingleTrackEmb is false");  }
  void SetJetMassVarType(JetMassVarType t)                      { fJetMassVarType    = t   ; }
  void SetResponseReference(ResponseReference r)                { fResponseReference = r   ; }
  void SetUseSumw2(Bool_t b)                                    { fUseSumw2          = b   ; }
  void SetSmallSystRanges(Bool_t small = kTRUE)                 { fSmallSyst = small; }
  void SetNamesForTree(TString path, TString treename, TString branchnameD, TString branchnameP) { fPathTreeinputFile = path; fTreeinputName = treename; fBranchJDetName = branchnameD; fBranchJParName = branchnameP;}
  
 protected:
  Bool_t                              RetrieveEventObjects();
  Bool_t                              Run();
  //Bool_t                              FillHistograms();

  AliVParticle*                       GetEmbeddedConstituent(AliEmcalJet *jet);
  void           SetTree(TTree *tree);
  void           SetTreeFromFile(TString filenameM, TString treename);
  Int_t          MatchEmbeddedConstituentWithParticleLevel();
  
  Int_t                               fContainerBase;              
  Int_t                               fContainerSub;               

  Int_t                               fContainerNoEmb;             
  Int_t                               fContainerOverlap;           
  Double_t                            fMinFractionShared;          
  Bool_t                              fSingleTrackEmb;             
  Bool_t                              fCreateTree;                 
  JetMassVarType                      fJetMassVarType;             
  ResponseReference                   fResponseReference;          
  Bool_t                              fUseSumw2;                   
  Bool_t                              fOverlap;                    
  Double_t                               fRadius;                    
                                                                    
  TTree           *fTreeJetBkg;                                    
  TLorentzVector  *fJet1Vec;                                       
  TLorentzVector  *fJet2Vec;                                       
  TLorentzVector  *fJetSubVec;                                     
  Float_t         fArea;                                           
  Float_t         fAreaPhi;                                        
  Float_t         fAreaEta;                                        
  Float_t         fRho;                                            
  Float_t         fRhoM;                                           
  Int_t           fNConst;                                         
  Int_t           fMatch;                                          
  Int_t           fMinLabelEmb;                                    
  Int_t           fMaxLabelEmb;                                    
  Bool_t          fSmallSyst;                                      
  TH2F          **fh2MSubMatch;                                    
  TH2F          **fh2MSubPtRawAll;                                 
  TH3F          **fh3MSubPtRawDRMatch;                             
  TH3F          **fh3MSubPtTrueLeadPt;                             
  TH3F          **fh3MTruePtTrueLeadPt;                            
  TH3F          **fh3PtTrueDeltaMLeadPt;                           
  TH3F          **fh3PtTrueDeltaMRelLeadPt;                        
  THnSparse     **fhnMassResponse;                                 
  //THnSparse     **fhnDeltaMass;                                    //!<! deltaM vs deltapT
  THnSparse      *fhnDeltaMassAndBkgInfo;                          
  THnSparse      *fhnResolution;                                    
  TH1F          *fhNJetsSelEv;                                      
  TH2F          *fhRjetTrvspTj;                                     
  TH2F          *fhJetEtaPhiOvl;                                       
  TH1F          *fhpTTracksJet1;                                   
  TH1F          *fhpTTracksJetO;                                   
  TH1F          *fhpTTracksCont;                                   
  TH1F          *fhptjetSMinusSingleTrack;                         
  TH3F          *fhJetSubMatchEtaPhiPt;                            
  TTree         *fTreeEmb;                                         
  Bool_t        fFromTree;                                         
  TString       fPathTreeinputFile;                                
  TString       fTreeinputName;                                    
  TString       fBranchJDetName;                                   
  TString       fBranchJParName;                                   
  Int_t         fThisEntry;                                        
  Int_t         fMaxTreeEntries;                                   
  TLorentzVector *fVecD;                                           
  TLorentzVector *fVecP;                                           
 
  private:
     
     AliAnalysisTaskJetShapeBase(const AliAnalysisTaskJetShapeBase&);            // not implemented
     AliAnalysisTaskJetShapeBase &operator=(const AliAnalysisTaskJetShapeBase&); // not implemented
  
  ClassDef(AliAnalysisTaskJetShapeBase, 5);
};
#endif