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 "AliEventCuts.h"
#include "THistManager.h"
#include "AliTLorentzVector.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();

  static AliAnalysisTaskPWGJEQA* AddTaskPWGJEQA(
                                         const char* ntracks            = "usedefault",
                                         const char* nclusters          = "usedefault",
                                         const char* ncells             = "usedefault",
                                         const char *nGenLev            = "mcparticles",
                                         Bool_t      doTrackQA          = kTRUE,
                                         Bool_t      doCaloQA           = kTRUE,
                                         Bool_t      doJetQA            = kTRUE,
                                         Bool_t      doEventQA          = kTRUE,
                                         Double_t    trackPtCut         = 0.15,
                                         Double_t    clusECut           = 0.30,
                                         const char* suffix             = ""
                                               );
  
  void                        UserCreateOutputObjects();

  void                        SetUseAliEventCuts(Bool_t b)                         { fUseAliEventCuts    = b          ; }
  void                        SetUseManualEvtCuts(Bool_t input)                    { fUseManualEventCuts = input      ; }
  void                        SetCellEnergyCut(Float_t cut)                        { fCellEnergyCut      = cut        ; }
  void                        SetGeneratorLevelName(const char* name)              { fGeneratorLevelName = name       ; }
  void                        SetDetectorLevelName(const char* name)               { fDetectorLevelName  = name       ; }
  
  void                        SetDoTrackQA(Bool_t b) { fDoTrackQA = b; }
  void                        SetDoCaloQA(Bool_t b)  { fDoCaloQA  = 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                      IsEventSelected()                                             ;
  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);
  // Event selection
  Bool_t                      fUseAliEventCuts;          
  AliEventCuts                fEventCuts;                
  TList                      *fEventCutList;             
  Bool_t                      fUseManualEventCuts;       

  Float_t                     fCellEnergyCut;            
  Float_t                     fMaxPt;                    
  Int_t                       fNTotClusters[3];          
  AliTLorentzVector           fLeadingCluster[3];        
  Int_t                       fNTotTracks;               
  AliTLorentzVector           fLeadingTrack;             
  Bool_t                      fDoTrackQA;                
  Bool_t                      fDoCaloQA;                 
  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         ; 
  Int_t                 fNIntegerHistBins      ; 
  Double_t*             fIntegerHistBins       ; 
  AliPHOSGeometry*      fPHOSGeo               ; 
  
  THistManager          fHistManager           ; 
  
private:
  AliAnalysisTaskPWGJEQA(const AliAnalysisTaskPWGJEQA&);            // not implemented
  AliAnalysisTaskPWGJEQA &operator=(const AliAnalysisTaskPWGJEQA&); // not implemented
  
  ClassDef(AliAnalysisTaskPWGJEQA, 5);
};
#endif