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 "AliAnalysisTaskSE.h"
#include <TString.h>
#include <TCustomBinning.h>

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 AliAnalysisTaskSE {
public:
  enum BeamDirection_t{
    kpPb = 1,
    kPbp = -1
  };

  AliAnalysisTaskChargedParticlesRefMC();
  AliAnalysisTaskChargedParticlesRefMC(const char *name);
  virtual ~AliAnalysisTaskChargedParticlesRefMC();

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

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

  void                        SetAnalysisUtil(AliAnalysisUtils *util) { fAnalysisUtil = util; }
  void                        SetTrackSelection(AliEmcalTrackSelection * sel) { fTrackCuts = sel; }
  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                        SetOutlierCut(double fracpthard = 1.2) { fFracPtHard = fracpthard; }
  void                        SetOfflineTriggerSelection(AliEmcalTriggerOfflineSelection *sel) { fTriggerSelection = sel; }
  void                        SetUsePythiaHard(Bool_t usePythiaHard) { fUsePythiaHard = usePythiaHard; }
  void                        InitializeTrackCuts(TString cutname, bool isAOD);
  void                        SetWeightHandler(const AliEMCalTriggerWeightHandler * wh) { fWeightHandler = wh; }

protected:
  void                        FillTrackHistos(const char *eventclass, Double_t weight, Double_t pt, Double_t eta, Double_t etacent, Double_t phi, Bool_t etacut, Bool_t inEmcal, Bool_t hasTRD, const char *pid);
  void                        FillTriggerJetHistograms(Bool_t aftercut, AliGenPythiaEventHeader *const header);

  TString                     GetFiredTriggerClasses(const TClonesArray * triggerpatches);
  Bool_t                      PythiaInfoFromFile(const char* currFile, Float_t &fXsec, Float_t &fTrials, Int_t &pthard) const;
  AliGenPythiaEventHeader    *GetPythiaHeader() const;
  Bool_t                      IsPhysicalPrimary(const AliVParticle *const part, AliMCEvent *const mcevent);
  Bool_t                      IsOutlier(AliGenPythiaEventHeader * const header) const;


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

  // Monte-Carlo specific information
  Bool_t                          fUsePythiaHard;             
  Double_t                        fPtHard;                    
  Int_t                           fPtHardBin;                 
  Int_t                           fNTrials;                   
  Float_t                         fXsection;                  

  Double_t                        fYshift;                    
  Double_t                        fEtaSign;                   
  Double_t                        fEtaLabCut[2];              
  Double_t                        fEtaCmsCut[2];              
  Double_t                        fFracPtHard;                

private:

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

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

  ClassDef(AliAnalysisTaskChargedParticlesRefMC, 1);
};

} /* namespace EMCalTriggerPtAnalysis */

#endif /* ALIANALYSISTASKCHARGEDPARTICLESREFMC_H */