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 "AliAnalysisTaskEmcalTriggerBase.h"
#include "AliCutValueRange.h"
#include <vector>
#include <TCustomBinning.h>
#include <TString.h>

class THistManager;
class AliEmcalTrackSelection;

namespace EMCalTriggerPtAnalysis {

class AliAnalysisTaskChargedParticlesRef : public AliAnalysisTaskEmcalTriggerBase {
public:

  enum BeamDirection_t{
    kpPb = 1,       
    kPbp = -1       
  };

  AliAnalysisTaskChargedParticlesRef();

  AliAnalysisTaskChargedParticlesRef(const char *name);

  virtual ~AliAnalysisTaskChargedParticlesRef();

  void EnableSumw2(Bool_t doEnable) { fEnableSumw2 = doEnable; }

  void InitializeTrackCuts(TString cutname, bool isAOD);

  void SetRapidityShift(Double_t yshift) { fYshift = yshift; }

  void SetBeamDirection(BeamDirection_t beamdir) { fEtaSign = static_cast<Double_t>(beamdir); }

  void SetPlotPID(Bool_t plotPID) { fStudyPID = plotPID; }

  void SetEMCALTrackSelection(AliEmcalTrackSelection *sel) { fTrackCuts = sel; }

  void SetEtaLabCut(double etamin, double etamax) { fEtaLabCut.SetLimits(etamin, etamax); }

  void SetEtaCMSCut(double etamin, double etamax) { fEtaCmsCut.SetLimits(etamin, etamax); }

  void SetTrackPhiCut(double phimin, double phimax) { fPhiCut.SetLimits(phimin, phimax); }

  static AliAnalysisTaskChargedParticlesRef *AddTaskChargedParticlesRef(const TString &suffix);

  static AliAnalysisTaskChargedParticlesRef *AddTaskChargedParticlesRefDefault(const TString &cutname = "standard");

protected:

  virtual void CreateUserHistos();

  virtual void CreateUserObjects();

  virtual bool Run();

  virtual void UserFillHistosBeforeEventSelection();

  virtual void UserFillHistosAfterEventSelection();

  void FillTrackHistos(const TString &eventclass, Bool_t posCharge, Double_t pt, Double_t eta, Double_t etacent, Double_t phi, Bool_t inEmcal);

  void FillPIDHistos(const TString &eventclass, const AliVTrack &track);

  AliEmcalTrackSelection          *fTrackCuts;                

  Double_t                        fYshift;                    
  Double_t                        fEtaSign;                   

  AliCutValueRange<double>        fEtaLabCut;                 
  AliCutValueRange<double>        fEtaCmsCut;                 
  AliCutValueRange<double>        fPhiCut;                    

  Bool_t                          fStudyPID;                  
  Bool_t                          fEnableSumw2;               

private:

  class PtBinning : public TCustomBinning{
  public:

    PtBinning();

    virtual ~PtBinning() {}
  };

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

  ClassDef(AliAnalysisTaskChargedParticlesRef, 1);
};

} /* namespace EMCalTriggerPtAnalysis */

#endif /* ALIANALYSISTASKCHARGEDPARTICLESREF_H */