AliEmcalMCTrackSelector.h

Go to the documentation of this file.

#ifndef ALIEMCALMCTRAKCSELECTOR_H
#define ALIEMCALMCTRAKCSELECTOR_H

class TClonesArray;
class TString;
class AliVEvent;
class AliMCEvent;
class AliNamedArrayI;
class AliAODMCParticle;

#include "AliAnalysisTaskSE.h"

class AliEmcalMCTrackSelector : public AliAnalysisTaskSE {
 public:
  AliEmcalMCTrackSelector();
  AliEmcalMCTrackSelector(const char *name);
  virtual ~AliEmcalMCTrackSelector();

  void UserCreateOutputObjects();
  void UserExec(Option_t *option);

  void SetOnlyPhysPrim(Bool_t s)                        { fOnlyPhysPrim     = s    ; }  
  void SetChargedMC(Bool_t c = kTRUE)                   { fChargedMC        = c    ; }
  void SetEtaMax(Double_t e)                            { fEtaMax           = e    ; }
  void SetRejectNK(Bool_t r = kTRUE)                    { fRejectNK         = r    ; }
  void SetOnlyHIJING(Bool_t s)                          { fOnlyHIJING       = s    ; }
  void SetParticlesOutName(const char *name)            { fParticlesOutName = name ; }

  void                      ConvertMCParticles(AliMCEvent* mcEvent, TClonesArray* partOut, AliNamedArrayI* partMap=0);    // for ESD analysis
  void                      CopyMCParticles(TClonesArray* partIn, TClonesArray* partOut, AliNamedArrayI* partMap=0); // for AOD analysis
  
  static AliEmcalMCTrackSelector* AddTaskMCTrackSelector(TString outname = "mcparticles", Bool_t nk = kFALSE, Bool_t ch = kFALSE, Double_t etamax = 1, Bool_t physPrim = kTRUE);
 protected:
  virtual Bool_t            AcceptParticle(AliAODMCParticle* part) const;
  
  TString                   fParticlesOutName;     // name of output particle array
  Bool_t                    fOnlyPhysPrim;         // true = only physical primary particles
  Bool_t                    fRejectNK;             // true = reject K_0^L and neutrons
  Bool_t                    fChargedMC;            // true = only charged particles
  Bool_t                    fOnlyHIJING;           // true = only HIJING particles
  Double_t                  fEtaMax;               // maximum eta to accept particles
  TString                   fParticlesMapName;     
  Bool_t                    fInit;                 
  TClonesArray             *fParticlesIn;          
  TClonesArray             *fParticlesOut;         
  AliNamedArrayI           *fParticlesMap;         
  AliVEvent                *fEvent;                
  AliMCEvent               *fMC;                   
  Bool_t                    fIsESD;                
  Bool_t                    fDisabled;             

 private:
  AliEmcalMCTrackSelector(const AliEmcalMCTrackSelector&);            // not implemented
  AliEmcalMCTrackSelector &operator=(const AliEmcalMCTrackSelector&); // not implemented

  ClassDef(AliEmcalMCTrackSelector, 5); // Task to select particle in MC events
};
#endif