AliAnalysisTaskChargedParticlesRef.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKCHARGEDPARTICLESREF_H
#define ALIANALYSISTASKCHARGEDPARTICLESREF_H
/* Copyright(c) 1998-2015, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include "AliAnalysisTaskSE.h"
#include <string>
#include <TCustomBinning.h>
#include <TString.h>

class TArrayD;
class THistManager;
class TObjArray;

class AliAnalysisUtils;
class AliEmcalTrackSelection;
class AliEMCALGeometry;
class AliOADBContainer;

namespace EMCalTriggerPtAnalysis {

class AliEmcalTriggerOfflineSelection;

class AliAnalysisTaskChargedParticlesRef : public AliAnalysisTaskSE {
public:
  enum BeamDirection_t{
    kpPb = 1,
    kPbp = -1
  };
  AliAnalysisTaskChargedParticlesRef();
  AliAnalysisTaskChargedParticlesRef(const char *name);
  virtual ~AliAnalysisTaskChargedParticlesRef();

  void UserCreateOutputObjects();
  void UserExec(Option_t *);
  void Terminate(Option_t *) {}

  void SetRapidityShift(Double_t yshift) { fYshift = yshift; }
  void SetBeamDirection(BeamDirection_t beamdir) { fEtaSign = static_cast<Double_t>(beamdir); }
  void UseTriggerPatches(Bool_t doUse) { fTriggerStringFromPatches = doUse; }

  void SetKineCorrelation(Bool_t useKineCorrelation) { fKineCorrelation = useKineCorrelation; }
  void SetPlotPID(Bool_t plotPID) { fStudyPID = plotPID; }
  void SetEMCALTrackSelection(AliEmcalTrackSelection *sel) { fTrackCuts = sel; }
  void InitializeTrackCuts(TString cutname, bool isAOD);
  void SetOfflineTriggerSelection(AliEmcalTriggerOfflineSelection *sel) { fTriggerSelection = sel; }
  void SetAnalysisUtil(AliAnalysisUtils *util) { fAnalysisUtil = util; }
  void SetEtaLabCut(double etamin, double etamax) { fEtaLabCut[0] = etamin; fEtaLabCut[1] = etamax; }
  void SetEtaCMSCut(double etamin, double etamax) { fEtaCmsCut[0] = etamin; fEtaCmsCut[1] = etamax; }
  void SetDownscaleOADB(TString oadbname) { fNameDownscaleOADB = oadbname; }

protected:
  virtual void ExecOnce();
  virtual void RunChanged(Int_t runnuber);

  Double_t GetTriggerWeight(const std::string &triggerclass) const;
  void FillEventCounterHists(const std::string &triggerclass, double vtxz, bool isSelected);
  void FillTrackHistos(const std::string &eventclass, Double_t pt, Double_t eta, Double_t etacent, Double_t phi, Bool_t etacut, Bool_t inEmcal, Bool_t hasTRD);
  void FillPIDHistos(const std::string &eventclass, const AliVTrack &track);
  TString GetFiredTriggerClassesFromPatches(const TClonesArray* triggerpatches) const;

  AliEmcalTrackSelection          *fTrackCuts;                
  AliAnalysisUtils                *fAnalysisUtil;             
  AliEmcalTriggerOfflineSelection *fTriggerSelection;         
  THistManager                    *fHistos;                   
  AliEMCALGeometry                *fGeometry;                 
  TClonesArray                    *fTriggerPatches;           

  Bool_t                          fTriggerStringFromPatches;  
  Double_t                        fYshift;                    
  Double_t                        fEtaSign;                   

  Double_t                        fEtaLabCut[2];              
  Double_t                        fEtaCmsCut[2];              

  Bool_t                          fKineCorrelation;           
  Bool_t                          fStudyPID;                  

  TString                         fNameDownscaleOADB;         
  AliOADBContainer                *fDownscaleOADB;            
  TObjArray                       *fDownscaleFactors;         

  Int_t                           fCurrentRun;                
  Bool_t                          fInitialized;               

private:

  class NewPtBinning : public TCustomBinning{
  public:
    NewPtBinning();
    virtual ~NewPtBinning() {}
  };

  AliAnalysisTaskChargedParticlesRef(const AliAnalysisTaskChargedParticlesRef &);
  AliAnalysisTaskChargedParticlesRef &operator=(const AliAnalysisTaskChargedParticlesRef &);

  ClassDef(AliAnalysisTaskChargedParticlesRef, 1);
};

} /* namespace EMCalTriggerPtAnalysis */

#endif /* ALIANALYSISTASKCHARGEDPARTICLESREF_H */