AliAnalysisTaskChargedParticlesRefMC.h

Go to the documentation of this file.

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

#include "AliAnalysisTaskEmcal.h"
#include "AliCutValueRange.h"
#include <TString.h>
#include <TCustomBinning.h>

#include <vector>

class TArrayD;
class TClonesArray;
class THistManager;

class AliAnalysisUtils;
class AliAODTrack;
class AliEMCALGeometry;
class AliESDtrack;
class AliEmcalTrackSelection;
class AliGenPythiaEventHeader;
class AliVParticle;
class AliMCEvent;

namespace EMCalTriggerPtAnalysis {

class AliEMCalTriggerWeightHandler;

class AliAnalysisTaskChargedParticlesRefMC: public AliAnalysisTaskEmcal {
public:
  enum BeamDirection_t{
    kpPb = 1,       
    kPbp = -1       
  };

  AliAnalysisTaskChargedParticlesRefMC();

  AliAnalysisTaskChargedParticlesRefMC(const char *name);

  virtual ~AliAnalysisTaskChargedParticlesRefMC();

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

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

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

  void  SetAnalysisUtil(AliAnalysisUtils *util) { fAliAnalysisUtils = util; }

  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); }

  void SetMinPtTracks(Double_t minpt) { fMinPt = minpt; }

  void SetOfflineTriggerSelection(AliEmcalTriggerOfflineSelection *sel) { fTriggerSelection = sel; }

  void SetRequireTOFBunchCrossing(Bool_t doRequire) { fRequireTOFBunchCrossing = doRequire; }

  AliEmcalTriggerOfflineSelection *GetOfflineTriggerSelection() const { return fTriggerSelection; }

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

  void InitializeTrackCuts(TString cutname, bool isAOD);

  void SetWeightHandler(const AliEMCalTriggerWeightHandler * wh) { fWeightHandler = wh; }

  void SetTriggerAcceptanceOADB(const TString &name) { fNameAcceptanceOADB = name; }

  void SetStudyEMCALgeo(Bool_t doStudy) { fStudyEMCALgeo = doStudy; }

  static AliAnalysisTaskChargedParticlesRefMC *AddTaskChargedParticlesRefMC(const TString &suffix);

  static AliAnalysisTaskChargedParticlesRefMC *AddTaskChargedParticlesRefMCDefault(const TString &cutname = "standard");

protected:

  virtual void UserCreateOutputObjects();

  virtual bool  Run();

  virtual bool IsEventSelected();

  virtual void ExecOnce();

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

  TString GetFiredTriggerClasses(const TClonesArray * triggerpatches);

  Bool_t IsPhysicalPrimary(const AliVParticle *const part, AliMCEvent *const mcevent);

private:

  class PtBinning : public TCustomBinning{
  public:

    PtBinning();

    virtual ~PtBinning() {}
  };

  AliAnalysisTaskChargedParticlesRefMC(const AliAnalysisTaskChargedParticlesRefMC &);
  AliAnalysisTaskChargedParticlesRefMC &operator=(const AliAnalysisTaskChargedParticlesRefMC &);

  AliEmcalTrackSelection                *fTrackCuts;                
  AliEmcalTriggerOfflineSelection       *fTriggerSelection;         
  THistManager                          *fHistos;                   
  const AliEMCalTriggerWeightHandler    *fWeightHandler;            

  std::vector<TString>                  fEventTriggers;             
  Double_t                              fEventWeight;               

  Double_t                              fYshift;                    
  Double_t                              fEtaSign;                   
  Double_t                              fMinPt;                     
  AliCutValueRange<double>              fEtaLabCut;                 
  AliCutValueRange<double>              fEtaCmsCut;                 
  AliCutValueRange<double>              fPhiCut;                    
  Double_t                              fFracPtHard;                
  Bool_t                                fEnableSumw2;               
  Bool_t                                fStudyPID;                  
  Bool_t                                fStudyEMCALgeo;             
  Bool_t                                fRequireTOFBunchCrossing;   

  TString                               fNameAcceptanceOADB;        

  ClassDef(AliAnalysisTaskChargedParticlesRefMC, 1);
};

} /* namespace EMCalTriggerPtAnalysis */

#endif /* ALIANALYSISTASKCHARGEDPARTICLESREFMC_H */