AliAnalysisTaskEmcalJetHF.h

Go to the documentation of this file.

#ifndef AliAnalysisTaskEmcalJetHF_h
#define AliAnalysisTaskEmcalJetHF_h

//ROOT
class TClonesArray;
class TH1;
class TH2;
class TH3;
class TH1F;
class TH2F;
class TH3F;
class THnSparse;
class TLorentzVector;
class TGraph;
class TClonesArray;
class TArrayI;
class TProfile;
//ALIROOT
class AliEMCALTrack;
class AliMagF;
class AliESDEvent;
class AliAODEvent;
class AliEMCALGeometry;
class AliEMCALRecoUtils;
class AliESDtrack;
class AliESDtrackCuts;
class AliAnalysisEtCuts;
class AliDetectorPID;
class AliESDCaloCluster;
class AliJetContainer;
class AliParticleContainer;
class AliClusterContainer;
class AliEMCALTriggerPatchInfo;
class AliVCaloTrigger;
//INCLUDES
#include <TRef.h>
#include <TBits.h>
#include <TMath.h>
#include <AliEmcalJet.h>
#include <AliVEvent.h>
#include <AliVTrack.h>
#include <AliVCluster.h>
#include <TClonesArray.h>
#include <TMath.h>
#include <TRandom3.h>
#include <AliLog.h>
#include "AliAnalysisTaskEmcalJet.h"
#include "AliEMCALPIDResponse.h"
#include <AliESDCaloCluster.h>
#include <AliESDtrackCuts.h>
#include "AliAODTrack.h"
#include "AliESDtrack.h"
#include "AliPID.h"
#include "AliTPCdEdxInfo.h"
#include "AliEMCALTriggerPatchInfo.h"
#include "AliAnalysisTaskEmcal.h"
#include "AliEmcalTriggerSetupInfo.h"
#include "AliPIDResponse.h"
#include "AliTPCPIDResponse.h"
#include "AliESDpid.h"
#include "AliAnalysisFilter.h"

class AliAnalysisTaskEmcalJetHF : public AliAnalysisTaskEmcalJet {
  public:
    
  enum MainPatchType {
    kManual = 0,    //just select highest energy patch in array
    kEmcalJet = 1   //use functionality of AliAnalysisTaskEmcal
  };
    
  AliAnalysisTaskEmcalJetHF();
  AliAnalysisTaskEmcalJetHF(const char *name);
  virtual                        ~AliAnalysisTaskEmcalJetHF();
  virtual void                   UserCreateOutputObjects();
  void                           SetMainPatchType(MainPatchType t)    { fMainPatchType      = t;}
  void                           SetMainTriggerTypeCat(TriggerCategory cat, Bool_t b) {fMainTrigCat = cat; fMainTrigSimple = b;}

  AliAnalysisEtCuts * GetCuts()  const { return fCuts; }
  virtual void                   SetCuts(const AliAnalysisEtCuts *cuts){ fCuts = (AliAnalysisEtCuts *) cuts; }
  void                           SetTPCITSTrackCuts(const AliESDtrackCuts *cuts){ fEsdtrackCutsITSTPC = (AliESDtrackCuts *) cuts;}
  void                           SetTPCOnlyTrackCuts(const AliESDtrackCuts *cuts){ fEsdtrackCutsTPC = (AliESDtrackCuts *) cuts;}
  void                           SetITSTrackCuts(const AliESDtrackCuts *cuts){ fEsdtrackCutsITS = (AliESDtrackCuts *) cuts;}
  //PID Sparse
  virtual THnSparse*              NewTHnSparseDHF(const char* name, UInt_t entries);
  virtual void                    GetDimParamsHF(Int_t iEntry,TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);
  //JetTrigger Sparse
  virtual THnSparse*              NewTHnSparseDJetTrigger(const char* name, UInt_t entries);
  virtual void                    GetDimParamsJetTrigger(Int_t iEntry,TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);//
  // setters
  virtual void                    SetGlobalQA(Bool_t PID)                { fGlobalQA = PID; }
  void                            SetJetPt(Double_t jpt)                 { fJetHIpt = jpt; }
  void                            SetTrackPtCut(Double_t trpt)           { fTrackPtCut = trpt; }
  virtual void                    SetTrackEta(Double_t e)                { fTrackEta   = e; }
  virtual void                    SetTrackQACut(Double_t trkQAcut)       { fTrkQAcut = trkQAcut; }
  void                            SetTrackCuts(AliESDtrackCuts *cuts)    { fesdTrackCuts = cuts; }
  virtual void                    SetFillHistograms(Bool_t fillhists)    { fFillHists = fillhists; }
  virtual void                    SetFillJetPID(Bool_t jetPID)           { fJetPID = jetPID; }
  // eta and phi limits of jets - setters
  virtual void                    SetJetEta(Double_t emin, Double_t emax)  { fEtamin = emin; fEtamax = emax; }
  virtual void                    SetJetPhi(Double_t pmin, Double_t pmax)  { fPhimin = pmin; fPhimax = pmax; }
    
  protected:
  Bool_t                          Run();
  virtual void                    Terminate(Option_t *);
  virtual Int_t                   AcceptMyJet(AliEmcalJet *jet);
  virtual Int_t                   AcceptJetforTag(AliVCluster *clusMatch, AliVTrack *AcceptedTrack);
  virtual Int_t                   GetCentBin(Double_t cent) const;
  void                        ExecOnce();
  void                            CheckClusTrackMatchingQA();
  void                            ExtractMainPatch();
  // Trigger bit
  Bool_t                          IsEventEMCALL1Gamma1()                 const { return fEventTrigEMCALL1Gamma1  ; }
  Bool_t                          IsEventEMCALL1Gamma2()                 const { return fEventTrigEMCALL1Gamma2  ; }
  Bool_t                          IsEventEMCALL1Gamma()                  const { return (fEventTrigEMCALL1Gamma1 || fEventTrigEMCALL1Gamma2) ; }
  Bool_t                          fEventTrigEMCALL1Gamma1 ;
  Bool_t                          fEventTrigEMCALL1Gamma2 ;
  // data type switch
  AliVEvent                      *fInputEvent;
  AliAnalysisEtCuts              *fCuts;
  // cuts
  Double_t                       fPhimin;
  Double_t                       fPhimax;
  Double_t                       fEtamin;
  Double_t                       fEtamax;
  Double_t                       fAreacut;
  Double_t                       fJetHIpt;
  Double_t                       fTrackPtCut;
  Double_t                       fTrackEta;
  Double_t                       fTrkQAcut;
  Double_t                       fM02max;
  Double_t                       fM02min;
  AliESDtrackCuts                *fesdTrackCuts;
  //General Task Switches
  Bool_t                         fGlobalQA;
  Bool_t                         fFillHists;
  Bool_t                         fJetPID;
  // event no.
  Int_t event;
  // PID                                                                                                                                    
  AliPIDResponse                 *fPIDResponse;
  AliTPCPIDResponse              *fTPCResponse;
  AliESDtrackCuts*               fEsdtrackCutsITSTPC;
  AliESDtrackCuts*               fEsdtrackCutsTPC;
  AliESDtrackCuts*               fEsdtrackCutsITS;
  //Containers
  AliJetContainer                *fJetsCont;
  AliParticleContainer           *fTracksCont;
  AliClusterContainer            *fCaloClustersCont;
  AliParticleContainer           *fTracksJetCont;
  AliClusterContainer            *fCaloClustersJetCont;

  private:                                                               //Declare it private to avoid compilation warning
  AliESDEvent                   *fESD;
  AliAODEvent                   *fAOD;
  AliEMCALTriggerPatchInfo      *fMaxPatch;
  MainPatchType                 fMainPatchType;
  TriggerCategory               fMainTrigCat; 
  Bool_t                        fMainTrigSimple;
  //Histos
  TH1F                          *fHistMatchedClusJet;
  TH1F                          *fHistTriggerBitInfo;
  TH1F                          *fHistMaxTriggerBitInfo;
  TH1F                          *fHistEventSelection;
  TH2F                          *fHistRecalcGASize;
  TH1F                          *fHistRecalcGAEnergy;
  TH2F                          *fHistCorrJetEvsPatchE;
  TH2F                          *fHistClusEvPatchE;
  TH2F                          *fHistdEtaPatchvdPhiPatch;
  TH2F                          *fHistRawJetEvPatchE;
  //Sparse
  THnSparse                     *fhnTriggerInfo;
  THnSparse                     *fhnTrackClusterQA;
  THnSparse                     *fhnPIDHFTtoC;
  THnSparse                     *fhnJetTrigger;
  
  AliAnalysisTaskEmcalJetHF(const AliAnalysisTaskEmcalJetHF & g) ;// cpy ctor
  AliAnalysisTaskEmcalJetHF& operator=(const AliAnalysisTaskEmcalJetHF&);// not implemented
  ClassDef(AliAnalysisTaskEmcalJetHF, 4);// Emcal jet Heavy Flavor
};
#endif