AliEmcalTrackingQATask.h

Go to the documentation of this file.

#ifndef ALIEMCALTRACKINGQATASK_H
#define ALIEMCALTRACKINGQATASK_H

#if !(defined(__CINT__) || defined(__MAKECINT__))
#include <tuple>
#endif

#include <vector>

#include "AliAnalysisTaskEmcalLight.h"

class THnSparse;

class AliEmcalTrackingQATask : public AliAnalysisTaskEmcalLight {

 public:
  AliEmcalTrackingQATask();
  AliEmcalTrackingQATask(const char *name); 
  virtual ~AliEmcalTrackingQATask();

  void                   UserCreateOutputObjects();
  void                   SetDoSigma1OverPt(Bool_t s)      {fDoSigma1OverPt     = s; }
  void                   SetDoSigmaPtOverPtGen(Bool_t s)  {fDoSigmaPtOverPtGen = s; }
  void                   SetDoSeparateTRDrefit(Bool_t s)  {fDoSeparateTRDrefit = s; }

  static AliEmcalTrackingQATask* AddTaskTrackingQA(Bool_t isMC, const char *suffix = "");

 protected:
  Bool_t                 FillHistograms()                               ;
  void                   ExecOnce();
  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, Int_t mcGen, Byte_t trackType);
  void                   FillGeneratorLevelTHnSparse(Double_t cent, Double_t partEta, Double_t partPhi, Double_t partPt, Int_t mcGen, 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);
#if !(defined(__CINT__) || defined(__MAKECINT__))
  THnSparse* GenerateTHnSparse(const char* name, const std::vector<std::tuple<std::string, std::vector<Double_t>::iterator, std::vector<Double_t>::iterator>>& axis);
#endif

  // Task configuration
  Bool_t                  fDoSigma1OverPt        ; 
  Bool_t                  fDoSigmaPtOverPtGen    ; 
  Bool_t                  fDoSeparateTRDrefit    ; 

  // Service fields (non-streamed)
  Bool_t                  fIsEsd                 ; 
  AliMCParticleContainer* fGeneratorLevel        ; 
  AliTrackContainer*      fDetectorLevel         ; 
  std::vector<Double_t>   fPtHistBins            ; 
  std::vector<Double_t>   fEtaHistBins           ; 
  std::vector<Double_t>   fPhiHistBins           ; 
  std::vector<Double_t>   fCentHistBins          ; 
  std::vector<Double_t>   fPtRelDiffHistBins     ; 
  std::vector<Double_t>   fPtResHistBins         ; 
  std::vector<Double_t>   f1OverPtResHistBins    ; 
  std::vector<Double_t>   fIntegerHistBins       ; 

  // Histograms
  THnSparse*              fTracks                ; 
  THnSparse*              fParticlesPhysPrim     ; 
  THnSparse*              fParticlesMatched      ; 
  
 private:
  AliEmcalTrackingQATask(const AliEmcalTrackingQATask&);            // not implemented
  AliEmcalTrackingQATask &operator=(const AliEmcalTrackingQATask&); // not implemented

  ClassDef(AliEmcalTrackingQATask, 4) // Track QA task (efficiency and pt resolution)
};
#endif