AliAnalysisTaskEmcalJetQA.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKEMCALJETQA_H
#define ALIANALYSISTASKEMCALJETQA_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 AliVVZERO;

#include "THistManager.h"
#include "AliTLorentzVector.h"
#include "AliAnalysisTaskEmcalLight.h"

class AliAnalysisTaskEmcalJetQA : public AliAnalysisTaskEmcalLight {

public:

  struct EventQA_t {
    EventQA_t() : fCent(0), fCent2(0), fCent3(0), fV0A(0), fV0C(0), fEP(0), fNTracks(0), fNCells(0), fMaxTrack() { fNClusters[0] = 0; fNClusters[1] = 0; }

    Float_t fCent;
    Float_t fCent2;
    Float_t fCent3;
    Float_t fV0A;
    Float_t fV0C;
    Float_t fEP;

    Int_t fNTracks;
    Int_t fNClusters[3];
    Int_t fNCells;

    AliTLorentzVector fMaxTrack;
    AliTLorentzVector fMaxCluster[3];
  };

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

  void                        UserCreateOutputObjects();

  void                        SetCellEnergyCut(Float_t cut)                        { fCellEnergyCut      = cut        ; }
  void                        SetParticleLevel(Bool_t s)                           { fParticleLevel      = s          ; }
  void                        SetMC(Bool_t m)                                      { fIsMC               = m          ; }
  void                        SetAdditionalCentEst(const char* meth2, const char* meth3="") { fCentMethod2 = meth2; fCentMethod3 = meth3; }
  void                        SetDoV0QA(Int_t b)                                   { fDoV0QA             = b          ; }
  void                        SetDoEPQA(Int_t b)                                   { fDoEPQA             = b          ; }
  void                        SetMaxCellsInCluster(Int_t b)                        { fMaxCellsInCluster  = b          ; }
  void                        SetDoLeadingObjectPosition(Int_t b)                  { fDoLeadingObjectPosition = b     ; }
  void                        SetSeparateEMCalDCal(Bool_t b)                       { fSeparateEMCalDCal = b           ; }
  void                        SetPtBin(Float_t w, Float_t max)                     { fPtBinWidth        = w; fMaxPt = max ; }
  void                        SetIsEmbedded(Bool_t i)                              { fIsEmbedded        = i           ; }

  static AliAnalysisTaskEmcalJetQA* AddTaskEmcalJetQA(TString ntracks = "usedefault", TString nclusters = "usedefault", TString ncells = "usedefault", TString subdir = "", TString suffix = "");
  static void AddTaskEmcalJetQA_QAtrain(Int_t runnumber);
protected:

  void                        ExecOnce()                                                    ;
  Bool_t                      FillHistograms()                                              ;
  void                        FillEventQAHisto(const EventQA_t& eventQA);
  Bool_t                      RetrieveEventObjects()                                        ;
  Int_t                       DoCellLoop()                                      ;
  void                        DoTrackLoop()                                                 ;
  void                        DoClusterLoop()                                               ;
  Double_t                    GetFcross(AliVCluster *cluster, AliVCaloCells *cells)         ;

  Float_t                     fCellEnergyCut;            
  Bool_t                      fParticleLevel;            
  Bool_t                      fIsMC;                     
  TString                     fCentMethod2;              
  TString                     fCentMethod3;              
  Int_t                       fDoV0QA;                   
  Int_t                       fDoEPQA;                   
  Int_t                       fDoLeadingObjectPosition;  
  Int_t                       fMaxCellsInCluster;        
  Float_t                     fPtBinWidth;               
  Float_t                     fMaxPt;                    
  Bool_t                      fSeparateEMCalDCal;        
  Bool_t                      fIsEmbedded;               
  THistManager                fHistManager;              

  Double_t                    fCent2;                    
  Double_t                    fCent3;                    
  AliVVZERO                  *fVZERO;                    
  Double_t                    fV0ATotMult;               
  Double_t                    fV0CTotMult;               
  Int_t                       fNTotClusters[3];          
  AliTLorentzVector           fLeadingCluster[3];        
  Int_t                       fNTotTracks;               
  AliTLorentzVector           fLeadingTrack;             

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

  ClassDef(AliAnalysisTaskEmcalJetQA, 4)
};
#endif