AliAnalysisTaskPWGJEQA.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKPWGJEQA_H
#define ALIANALYSISTASKPWGJEQA_H

/* Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

class TH1;
class TH2;
class TH3;
class THnSparse;
class AliPHOSGeometry;

#include "THistManager.h"
#include "AliTLorentzVector.h"
//#include "AliAnalysisTaskEmcalJetLight.h"
#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskPWGJEQA : public AliAnalysisTaskEmcalJet {
  
public:
  
  struct EventQA_t {
    EventQA_t() : fCent(0), fNTracks(0), fMaxTrack() { fNClusters[0] = 0; fNClusters[1] = 0; fNClusters[2] = 0;}
    Float_t fCent;
    Int_t fNTracks;
    Int_t fNClusters[3];
    AliTLorentzVector fMaxTrack;
    AliTLorentzVector fMaxCluster[3];
  };
  
  enum ClusterType {
    kNA       = -1,
    kEMCal    = 0, 
    kDCal     = 1, 
    kPHOS     = 2  
  };

  AliAnalysisTaskPWGJEQA();
  AliAnalysisTaskPWGJEQA(const char *name);
  virtual ~AliAnalysisTaskPWGJEQA();
  
  void                        UserCreateOutputObjects();

  void                        SetCellEnergyCut(Float_t cut)                        { fCellEnergyCut      = cut        ; }
  void                        SetSeparateEMCalDCal(Bool_t b)                       { fSeparateEMCalDCal = b           ; }
  void                        SetPtBin(Float_t w, Float_t max)                     { fPtBinWidth        = w; fMaxPt = max ; }
  void                        SetGeneratorLevelName(const char* name)              { fGeneratorLevelName = name       ; }
  void                        SetDetectorLevelName(const char* name)               { fDetectorLevelName = name        ; }
  
  void                        SetDoTrackQA(Bool_t b) { fDoTrackQA = b; }
  void                        SetDoEmcalQA(Bool_t b) { fDoEmcalQA = b; }
  void                        SetDoJetQA(Bool_t b)   { fDoJetQA   = b; }
  void                        SetDoEventQA(Bool_t b) { fDoEventQA = b; }
  void                        SetRejectOutlierEvents(Bool_t b) {fRejectOutlierEvents = b; }
  void                        SetIsPtHard(Bool_t b)            {fIsPtHard = b; }
  
protected:
  
  void                        ExecOnce()                                                    ;
  Bool_t                      FillHistograms()                                              ;
  Bool_t                      RetrieveEventObjects()                                        ;
  Bool_t                      UserNotify()                                                  ;
  
  void                        AllocateTrackHistograms()                                     ;
  void                        AllocateCellHistograms()                                      ;
  void                        AllocateClusterHistograms()                                   ;
  void                        AllocateJetHistograms()                                       ;
  void                        AllocateEventQAHistograms()                                   ;
  
  void                        FillTrackHistograms()                                         ;
  void                        FillCellHistograms()                                          ;
  void                        FillClusterHistograms()                                       ;
  void                        FillJetHistograms()                                           ;
  void                        FillEventQAHistograms()                                       ;
  
  void                        GenerateHistoBins()                                           ;
  void                        AllocateDetectorLevelTHnSparse()                              ;
  void                        AllocateGeneratorLevelTHnSparse()                             ;
  void                        AllocateMatchedParticlesTHnSparse()                           ;
  void                        FillDetectorLevelTHnSparse(Double_t cent, Double_t trackEta, Double_t trackPhi, Double_t trackPt,
                                                         Double_t sigma1OverPt, Byte_t trackType);
  void                        FillGeneratorLevelTHnSparse(Double_t cent, Double_t partEta, Double_t partPhi, Double_t partPt, Byte_t findable);
  void                        FillMatchedParticlesTHnSparse(Double_t cent, Double_t partEta, Double_t partPhi, Double_t partPt,
                                                            Double_t trackEta, Double_t trackPhi, Double_t trackPt, Byte_t trackType);
  
  Float_t                     fCellEnergyCut;            
  Float_t                     fPtBinWidth;               
  Float_t                     fMaxPt;                    
  Bool_t                      fSeparateEMCalDCal;        
  Int_t                       fNTotClusters[3];          
  AliTLorentzVector           fLeadingCluster[3];        
  Int_t                       fNTotTracks;               
  AliTLorentzVector           fLeadingTrack;             
  Bool_t                      fDoTrackQA;                
  Bool_t                      fDoEmcalQA;                
  Bool_t                      fDoJetQA;                  
  Bool_t                      fDoEventQA;                
  TString                     fGeneratorLevelName;       
  TString                     fDetectorLevelName;        
  Bool_t                      fRejectOutlierEvents;      
  Bool_t                      fIsPtHard;                 
  
  // Service fields (non-streamed)
  AliMCParticleContainer* fGeneratorLevel      ; 
  AliTrackContainer*    fDetectorLevel         ; 
  Int_t                 fNPtHistBins           ; 
  Double_t*             fPtHistBins            ; 
  Int_t                 fNEtaHistBins          ; 
  Double_t*             fEtaHistBins           ; 
  Int_t                 fNPhiHistBins          ; 
  Double_t*             fPhiHistBins           ; 
  Int_t                 fNCentHistBins         ; 
  Double_t*             fCentHistBins          ; 
  Int_t                 fNPtRelDiffHistBins    ; 
  Double_t*             fPtRelDiffHistBins     ; 
  Int_t                 fNPtResHistBins        ; 
  Double_t*             fPtResHistBins         ; 
  Double_t*             f1OverPtResHistBins    ; 
  Int_t                 fN1OverPtResHistBins   ; 
  Int_t                 fNIntegerHistBins      ; 
  Double_t*             fIntegerHistBins       ; 
  AliPHOSGeometry*      fPHOSGeo;              ; 
  
  // Histograms
  THnSparse*            fTracks                ; 
  THnSparse*            fParticlesPhysPrim     ; 
  THnSparse*            fParticlesMatched      ; 
  
  THistManager                fHistManager;              
  
private:
  AliAnalysisTaskPWGJEQA(const AliAnalysisTaskPWGJEQA&);            // not implemented
  AliAnalysisTaskPWGJEQA &operator=(const AliAnalysisTaskPWGJEQA&); // not implemented
  
  ClassDef(AliAnalysisTaskPWGJEQA, 2);
};
#endif