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 TString;
class AliAnalysisUtils;
class AliAODTrack;
class AliEMCALGeometry;
class AliESDtrackCuts;
class AliESDtrack;

namespace EMCalTriggerPtAnalysis {

class AliEMCalHistoContainer;

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

  Bool_t TrackSelectionESD(AliESDtrack *track);
  Bool_t TrackSelectionAOD(AliAODTrack *track);

  AliESDtrackCuts                 *fTrackCuts;                
  AliAnalysisUtils                *fAnalysisUtil;             
  AliEMCalHistoContainer          *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 */