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"

class TArrayD;
class THistManager;
class TString;
class AliAnalysisUtils;
class AliEmcalTrackSelection;
class AliEMCALGeometry;

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 SetTrackSelection(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; }

protected:
  void CreateOldPtBinning(TArrayD &binning) const;
  void CreateNewPtBinning(TArrayD &binning) const;

  void FillEventCounterHists(const char *triggerclass, double vtxz, bool isSelected);
  void FillTrackHistos(const char *eventclass, Double_t pt, Double_t eta, Double_t etacent, Double_t phi, Bool_t etacut, Bool_t inEmcal, Bool_t hasTRD);
  TString GetFiredTriggerClassesFromPatches(const TClonesArray* triggerpatches) const;

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

  Bool_t                          fTriggerStringFromPatches;  
  Double_t                        fYshift;                    
  Double_t                        fEtaSign;                   

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

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

  ClassDef(AliAnalysisTaskChargedParticlesRef, 1);
};

} /* namespace EMCalTriggerPtAnalysis */

#endif /* ALIANALYSISTASKCHARGEDPARTICLESREF_H */