AliAnalysisTaskJetCoreEmcal.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKJETCOREEMCAL_H
#define ALIANALYSISTASKJETCOREEMCAL_H

//
// Implementation of jet core task in Emcal framework
// Basic functionality copied from AliAnalysisTaskEmcalJetSample
//

class TH1F;
class TH1I;
class TH2F;
class TH3F;
class THnSparse;
class TRandom3;

#include "AliEventCuts.h"
#include "AliAnalysisTaskEmcalJet.h"
#include "THistManager.h"


class AliAnalysisTaskJetCoreEmcal : public AliAnalysisTaskEmcalJet {
 public:

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

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


  // Setters
  virtual void     SetJetShapeType(Int_t type){fJetShapeType=type;}
  virtual void     SetCentrality(Float_t cmin, Float_t cmax){fCentMin=cmin; fCentMax=cmax;}
  virtual void     SetTTLowRef(Float_t ttlow){fTTLowRef=ttlow;}
  virtual void     SetTTUpRef(Float_t ttup){fTTUpRef=ttup;}
  virtual void     SetTTLowSig(Float_t ttlows){fTTLowSig=ttlows;}
  virtual void     SetTTUpSig(Float_t ttups){fTTUpSig=ttups;}
  virtual void     SetNRPBins(Float_t nrpb){fNRPBins=nrpb;}
  virtual void     SetSignalFraction(Float_t sfrac){fFrac=sfrac;}
  virtual void     SetJetEtaMin(Float_t eta){fJetEtaMin=eta;}
  virtual void     SetJetEtaMax(Float_t eta){fJetEtaMax=eta;}
  virtual void     SetJetHadronDeltaPhi(Float_t delta){fJetHadronDeltaPhi=delta;}
  virtual void     SetMinFractionSharedPt(Float_t min){fMinFractionSharedPt=min;}
  virtual void     SetJetContName(TString cont){fJetContName=cont;}
  virtual void     SetJetContTrueName(TString cont){fJetContTrueName=cont;}
  virtual void     SetJetContPartName(TString cont){fJetContPartName=cont;}
  virtual void     SetFillTrackHistograms(Bool_t b){fFillTrackHistograms=b;}
  virtual void     SetFillJetHistograms(Bool_t b){fFillJetHistograms=b;}
  virtual void     SetFillInclusiveTree(Bool_t b){fFillInclusiveTree=b;}
  virtual void     SetFillRecoilTree(Bool_t b){fFillRecoilTree=b;}
  virtual void     SetPtHardBin(Int_t bin){fPtHardBin=bin;}

//  static AliAnalysisTaskJetCoreEmcal* AddTaskJetCoreEmcal(
//      const char *ntracks            = "usedefault",
//      const char *nclusters          = "usedefault",
//      const char* ncells             = "usedefault",
//      const char *suffix             = "");

  enum JetShapeType {
    kMCTrue = 0,   // generated jets only
    kTrueDet =1,  // detector and generated jets  
    kData   = 2,  // raw data 
    kDetEmbPart = 3,
//    kDetEmb = 3,  //detector embedded jets
//    kPythiaDef = 5,
//    kDetEmbPartPythia=6,
//    kGenOnTheFly = 7
  };

 protected:
  void                        ExecOnce()                                        ;
  Bool_t                      FillHistograms()                                  ;
  Bool_t                      Run()                                             ;

  void                        AllocateJetHistograms()                           ;
  void                        AllocateTrackHistograms()                         ;
  void                        AllocateClusterHistograms()                       ;
  void                        AllocateCellHistograms()                          ;
  void                        AllocateJetCoreHistograms()                          ;

  void                        DoJetLoop()                                       ;
  void                        DoTrackLoop()                                     ;
  void                        DoClusterLoop()                                   ;
  void                        DoCellLoop()                                      ;
  void                        DoJetCoreLoop()                                      ;
  void                        DoMatchingLoop()
    ;

  Int_t                       SelectTrigger(TList *list,Double_t minT,Double_t maxT,Int_t &number);
  Double_t                    RelativePhi(Double_t mphi,Double_t vphi);
  Int_t                       GetPhiBin(Double_t phi);

  THistManager                fHistManager;

  // flags and selection
  AliEventCuts fEventCuts; 
  Float_t fJetShapeType; 
  Float_t fCentMin; 
  Float_t fCentMax; 
  Float_t fTTLowRef; 
  Float_t fTTUpRef; 
  Float_t fTTLowSig; 
  Float_t fTTUpSig; 
  Int_t fNRPBins;  
  Float_t fFrac; 
  Float_t fJetEtaMin; 
  Float_t fJetEtaMax; 
  Float_t fJetHadronDeltaPhi; 
  Float_t fMinFractionSharedPt; 
  TString fJetContName; 
  TString fJetContTrueName; 
  TString fJetContPartName; 
  Bool_t fFillTrackHistograms; 
  Bool_t fFillJetHistograms; 
  Bool_t fFillInclusiveTree; 
  Bool_t fFillRecoilTree; 
  Int_t fPtHardBin; 
  //
  TRandom3 *fRandom; 
  Float_t fTreeVarsInclusive[8]; 
  Float_t fTreeVarsRecoil[11]; 
  //histograms to fill
  TH1I *fHistEvtSelection; 
  // recoil jet info contained in THnSparse
  THnSparse *fHJetSpec;  
  // recoil histograms
  TH1D *fh1TrigRef; 
  TH1D *fh1TrigSig; 
  TH2F *fh2Ntriggers; 
  TH2F *fh2RPJetsC10; 
  TH2F *fh2RPJetsC20; 
  TH2F *fh2RPTC10; 
  TH2F *fh2RPTC20; 
  TH2F *fhDphiPtSig; 
  TH2F *fhDphiPtRef; 
  // embedding histograms
  // inclusive jets
  TH2F *fhPtDetPart; 
  TH2F *fhPtHybrDet; 
  TH2F *fhPtHybrPart; 
  TH2F *fhPtHybrPartCor; 
  TH2F *fhPhiHybrPartCor; 
  TH1F *fhPtDet; 
  TH1F *fhPtDetMatchedToPart; 
  TH1F *fhResidual; 
  TH2F *fhPtResidual; 
  TH1F *fhPhiResidual; 
  TH2F *fhPhiPhiResidual; 
  //recoil jets
  TH2F *fhPtDetPartRecoil; 
  TH2F *fhPtHybrDetRecoil; 
  TH2F *fhPtHybrPartRecoil; 
  TH2F *fhPtHybrPartCorRecoil; 
  TH2F *fhPhiHybrPartCorRecoil; 
  TH1F *fhPtDetRecoil; 
  TH1F *fhPtDetMatchedToPartRecoil; 
  TH1F *fhResidualRecoil; 
  TH2F *fhPtResidualRecoil; 
  TH1F *fhDphiResidualRecoil; 
  TH2F *fhDphiphiResidualRecoil; 
  TH2F *fhTTPtDetMatchedToPart; 
  TH2F *fhTTPhiDetMatchedToPart; 
  TH2F *fhDPhiHybrPartCorRecoil; 
  // embedding trees
  TTree *fTreeEmbInclusive; 
  TTree *fTreeEmbRecoil; 

 private:
  AliAnalysisTaskJetCoreEmcal(const AliAnalysisTaskJetCoreEmcal&)           ; // not implemented
  AliAnalysisTaskJetCoreEmcal &operator=(const AliAnalysisTaskJetCoreEmcal&); // not implemented

  ClassDef(AliAnalysisTaskJetCoreEmcal, 5);
};
#endif