AliAnalysisTaskSEPicoV0MakerMC.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKSEPICOV0MAKERMC_H
#define ALIANALYSISTASKSEPICOV0MAKERMC_H

//*************************************************************************
// Class AliAnalysisTaskSEPicoV0MakerMC
// AliAnalysisTaskSE for V0s (K0 short, Lambda... ) filtering
// lite version only for MC
// Author: X-M. Zhang, xmzhang@lbl.gov
//*************************************************************************

#include "AliAnalysisTaskSE.h"

class TClonesArray;

class AliAODv0;
class AliESDv0;
class AliAODEvent;
class AliESDEvent;
class AliCentrality;
class AliPIDResponse;
class AliAnalysisUtils;

class AliAnalysisTaskSEPicoV0MakerMC : public AliAnalysisTaskSE {

 public :

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

  virtual void Init();
  virtual void LocalInit() { Init(); }
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t *opt);
  virtual void Terminate(Option_t *opt);
  virtual void NotifyRun();
//=============================================================================

  void SetTriggerMask(UInt_t w)       { fTriggerMask   = w; }
  void SetCollitionType(UInt_t w)     { fCollisionType = w; }
  void SetVertexContributorN(Int_t i) { fCutMinEventVtxContr = i; }
  void SetCentralityEstimator(TString s) { fCentEst = s; }

  void SetRefitV0ESD()   { fIsRefitV0sESD  = kTRUE; }
  void SetSkipFastOnly() { fIsSkipFastOnly = kTRUE; }
  void SetDMPjetMC()     { fIsDPMjetMC     = kTRUE; }

  void SetV0PtRange(Double_t  dMin, Double_t dMax) { fCutMinV0Pt  = dMin; fCutMaxV0Pt  = dMax; }
  void SetV0RapRange(Double_t dMin, Double_t dMax) { fCutMinV0Rap = dMin; fCutMaxV0Rap = dMax; }

  void SetDauPtRange(Double_t d) { fCutMinDauPt = d; }
  void SetDauEtaRange(Double_t dMin, Double_t dMax) { fCutMinDauEta = dMin; fCutMaxDauEta = dMax; }

  void SetV0Cuts(Double_t d[14]);
  void SetKaCutMaxKaArmFrac(Double_t d) { fCutMaxKshortArmFrac = d; }
  void SetLaCutMaxLaArmFrac(Double_t d) { fCutMaxLambdaArmFrac = d; }
//=============================================================================

 private :

  AliAnalysisTaskSEPicoV0MakerMC(const AliAnalysisTaskSEPicoV0MakerMC &);
  AliAnalysisTaskSEPicoV0MakerMC& operator=(const AliAnalysisTaskSEPicoV0MakerMC &);
//=============================================================================

  void FillPicoV0s();

  AliPicoV0MC* SelectV0Candidate(AliAODv0 const *pV0);
  AliPicoV0MC* SelectV0Candidate(AliESDv0 const *pV0);

  Bool_t IsEventNotAcpt();
  Bool_t IsEventNotINEL();
  Bool_t IsEventNotMBsa();

  void FillHistograms();
  void CreateHistograms();

  void InitAnalysis();
//=============================================================================

  AliAODEvent      *fEventAOD;  
  AliESDEvent      *fEventESD;  
  AliCentrality    *fCentInfo;  
  AliPIDResponse   *fRespoPID;  
  AliAnalysisUtils *fAnaUtils;  

  UInt_t fEventAcptMask;    //
  Double_t fPrimaryVtx[3];  
//=============================================================================

  UInt_t fTriggerMask;    //
  UInt_t fCollisionType;  //

  TString fCentEst;  //

  Bool_t fIsRefitV0sESD;   //
  Bool_t fIsSkipFastOnly;  //
  Bool_t fIsDPMjetMC;      //
//=============================================================================

  Int_t   fCutMinEventVtxContr;  //
  Float_t fCutMaxEventVzAbs;     //

  Double_t fCutMinV0Pt;   //
  Double_t fCutMaxV0Pt;   //
  Double_t fCutMinV0Rap;  //
  Double_t fCutMaxV0Rap;  //

  Double_t fCutMinDauPt;   //
  Double_t fCutMinDauEta;  //
  Double_t fCutMaxDauEta;  //

  Double_t fCutMaxV0Chi2;    //
  Double_t fCutMinV0Radius;  //
  Double_t fCutMaxV0Radius;  //

  Double_t fCutMaxDausDCA;      //
  Double_t fCutMinDauDCAtoPV;   //
  Float_t  fCutMinDauXrowsTPC;  //
  Double_t fCutMinDauXrowsOverFindableClusTPC;  //

  Float_t  fCutMaxKshortSigmaTPC;  //
  Double_t fCutMinKshortCosPA;     //
  Double_t fCutMaxKshortCtau;      //
  Double_t fCutMaxKshortArmFrac;   //
  Double_t fCutMinKshortDeltaM;    //

  Float_t  fCutMaxLambdaSigmaTPC;  //
  Double_t fCutMinLambdaCosPA;     //
  Double_t fCutMaxLambdaCtau;      //
  Double_t fCutMaxLambdaArmFrac;   //
  Double_t fCutMinLambdaDeletaM;   //
//=============================================================================

  static const Double_t fgkMassPion;    //
  static const Double_t fgkMassKshort;  //
  static const Double_t fgkMassLambda;  //
  static const Double_t fgkMassProton;  //
//=============================================================================

  TClonesArray *fPicoV0sClArr;  

  TList *fListUserOutputs;  

  ClassDef(AliAnalysisTaskSEPicoV0MakerMC, 1)
};

#endif