AliAnalysisTaskChargedParticlesMCTriggerMimic.h

Go to the documentation of this file.

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

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

class TClonesArray;
class THistManager;

class AliAnalysisUtils;
class AliGenPythiaEventHeader;
class AliMCEvent;
class AliVParticle;

class AliEmcalTrackSelection;

namespace EMCalTriggerPtAnalysis {

class AliEMCalTriggerWeightHandler;

class AliAnalysisTaskChargedParticlesMCTriggerMimic : public AliAnalysisTaskEmcal {
public:

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

  enum PatchType_t {
    kEMCEGA = 0,    
    kEMCEJE = 1,    
    kUndef = -1     
  };

  enum Observable_t {
    kTracks = 0,    
    kClusters = 1,  
    kEGAPatches = 2,
    kEJEPatches = 3 
  };

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

  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                        InitializeTrackCuts(TString cutname, bool isAOD);
  void                        SetWeightHandler(const AliEMCalTriggerWeightHandler * wh) { fWeightHandler = wh; }
  void                        SetNameClusters(const TString &nameClusters) { fNameClusters = nameClusters; }

  void                        SetEMCALTrigger(PatchType_t patchtype, Double_t threshold) { fPatchType = patchtype; fEnergyThreshold = threshold; }
  void                        SetObservable(Observable_t observable) { fObservables.SetBitNumber(observable); }
  Bool_t                      HasObservable(Observable_t observable) { return fObservables.TestBitNumber(observable); }

protected:

  virtual void                UserCreateOutputObjects();
  virtual Bool_t              IsEventSelected();
  virtual Bool_t              Run();

  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, const char *pid);
  void                        FillClusterHistos(double weight, double energy, double transversenergy, double eta, double phi);
  void                        FillPatchHistos(const char *patchname, double weight, double energy, double transverseenergy, double eta, double phi, int col, int row);
  Bool_t                      IsPhysicalPrimary(const AliVParticle* const part, AliMCEvent* const mcevent);

  Bool_t                      SelectEmcalTrigger(const TClonesArray * triggerpatches);

private:

  class PtBinning : public TCustomBinning{
  public:
    PtBinning();
    virtual ~PtBinning() {}
  };

  AliAnalysisTaskChargedParticlesMCTriggerMimic(const AliAnalysisTaskChargedParticlesMCTriggerMimic &);
  AliAnalysisTaskChargedParticlesMCTriggerMimic &operator=(const AliAnalysisTaskChargedParticlesMCTriggerMimic &);


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

  // Kinematical selection
  Double_t                              fYshift;                    
  Double_t                              fEtaSign;                   
  AliCutValueRange<double>              fEtaLabCut;                 
  AliCutValueRange<double>              fEtaCmsCut;                 
  AliCutValueRange<double>              fPhiCut;                    

  // Trigger specific information
  PatchType_t                           fPatchType;                 
  Double_t                              fEnergyThreshold;           

  // Settings
  TBits                                 fObservables;               
  TString                               fNameClusters;              

  ClassDef(AliAnalysisTaskChargedParticlesMCTriggerMimic, 1);
};

} /* namespace EMCalTriggerPtAnalysis */

#endif /* ALIANALYSISTASKCHARGEDPARTICLEMCTRIGGERMIMIC_H */