AliAnalysisTaskSED0MassNonPromptFraction.h

Go to the documentation of this file.

#ifndef AliAnalysisTaskSED0MassNonPromptFraction_H
#define AliAnalysisTaskSED0MassNonPromptFraction_H

/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */ 


#include <TROOT.h>
#include <TSystem.h>
#include <TNtuple.h>
#include <TTree.h>
#include <TH1F.h>
#include <THnSparse.h>
#include <TVector3.h>
#include <TVector2.h>
#include "AliAnalysisTaskSE.h"
#include "AliRDHFCutsD0toKpi.h"
#include "AliNormalizationCounter.h"

class AliAODEvent;

class AliAnalysisTaskSED0MassNonPromptFraction : public AliAnalysisTaskSE
{
 public:

  AliAnalysisTaskSED0MassNonPromptFraction();
  AliAnalysisTaskSED0MassNonPromptFraction(const char *name,AliRDHFCutsD0toKpi* cuts);
  virtual ~AliAnalysisTaskSED0MassNonPromptFraction();


  virtual void UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);

  void SetArray(Int_t type=AliAnalysisTaskSED0MassNonPromptFraction::kD0){fArray=type;}
  enum{kD0,kLS};

  void SetReadMC(Bool_t readMC=kFALSE){fReadMC=readMC;}
  void SetCutOnDistr(Bool_t cutondistr=kFALSE){fCutOnDistr=cutondistr;}
  void SetUsePid4Distr(Bool_t usepid=kTRUE){fUsePid4Distr=usepid;}
  void SetFillOnlyD0D0bar(Int_t flagfill){fFillOnlyD0D0bar=flagfill;}
  void SetFillVarHists(Bool_t flag) {fFillVarHists=flag;}
  void SetFillAfterCuts(Bool_t flag) {fFillAfterCuts=flag;}//new setter
  void SetFillPtHistos(Bool_t flag) {fFillPtHist=flag;}
  void SetFillYHistos(Bool_t flag) {fFillYHist=flag;}
  void SetFillImpactParameterHistos(Bool_t flag) {fFillImpParHist=flag;}
  void SetSystem(Int_t sys){fSys=sys; if(fSys==1) SetFillVarHists(kFALSE);}
  void SetRejectSDDClusters(Bool_t flag) { fIsRejectSDDClusters=flag; }
  void SetUseSelectionBit(Bool_t flag) { fUseSelectionBit=flag; }
  void SetWriteVariableTree(Bool_t flag) { fWriteVariableTree=flag; }
  void SetDrawDetSignal(Bool_t flag) { fDrawDetSignal=flag; }
  void SetPIDCheck(Bool_t flag) { fPIDCheck=flag; }
    void SetAODMismatchProtection(Int_t opt=1) {fAODProtection=opt;}
    void SetUseQuarkLevelTag(Bool_t opt){fUseQuarkTagInKine=opt;}



  Bool_t GetCutOnDistr() const {return fCutOnDistr;}
  Bool_t GetUsePid4Distr() const {return fUsePid4Distr;}
  Int_t  GetFillOnlyD0D0bar() const {return fFillOnlyD0D0bar;}
  Bool_t GetFillVarHists() const {return fFillVarHists;}
  Bool_t GetFillPtHistos() const {return fFillPtHist;}
  Bool_t GetFillYHistos() const {return fFillYHist;}
  Bool_t GetFillImpactParameterHistos() const {return fFillImpParHist;}
  Int_t  GetSystem() const {return fSys;}
  Bool_t GetRejectSDDClusters() const { return fIsRejectSDDClusters; }
  Bool_t GetUseSelectionBit() const { return fUseSelectionBit; }
  Bool_t GetWriteVariableTree() const {return fWriteVariableTree;}
     Bool_t GetFillAfterCuts() const {return fFillAfterCuts;}
  Bool_t GetDrawDetSignal() const {return fDrawDetSignal;}
  Bool_t GetPIDCheck() const {return fPIDCheck;}

 private:

  AliAnalysisTaskSED0MassNonPromptFraction(const AliAnalysisTaskSED0MassNonPromptFraction &source);
  AliAnalysisTaskSED0MassNonPromptFraction& operator=(const AliAnalysisTaskSED0MassNonPromptFraction& source); 
  void     DrawDetSignal(AliAODRecoDecayHF2Prong *part, TList *ListDetSignal);

  void     FillMassHists(AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliAODMCHeader *mcHeader, AliRDHFCutsD0toKpi *cuts, TList *listout);
    void  FillMCGenHists(AliAODEvent* aod, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout);
  void     FillVarHists(AliAODEvent *aodev,AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout);
  AliAODVertex* GetPrimaryVtxSkipped(AliAODEvent *aodev);
  void CreateImpactParameterHistos();
  Int_t CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
  Float_t GetTrueImpactParameter(AliAODMCHeader *mcHeader, TClonesArray* arrayMC, AliAODMCParticle *partD0) const ;

  TList    *fOutputMass;          
  TList    *fOutputMassPt;        
  TList    *fOutputMassY;        
  TList    *fDistr;               
  TH1F     *fNentries;            
  AliRDHFCutsD0toKpi *fCuts;      //  Cuts - sent to output slot 4
  THnSparseF *fHistMassPtImpParTC[5];   
  Int_t     fArray;               
  Bool_t    fReadMC;              
  Bool_t    fCutOnDistr;          
  Bool_t    fUsePid4Distr;        //  flag to use the particle identification to fill the signal histograms of distributions. It has effect only with fReadMC=kFALSE
  AliNormalizationCounter *fCounter;
  Int_t     fNPtBins;             
  Double_t  fLsNormalization;     
  Int_t     fFillOnlyD0D0bar;     
  TObjArray fDaughterTracks;      
  Int_t     fIsSelectedCandidate; 
     Bool_t     fFillAfterCuts; 
  Bool_t    fFillVarHists;        
    Bool_t    fFill;                //flag to enable filling variable histos for correlations
  Int_t     fSys;                 
  Bool_t    fIsRejectSDDClusters; 
  Bool_t    fFillPtHist;          
  Bool_t    fFillYHist;          
  Bool_t    fFillImpParHist;      
  Bool_t    fUseSelectionBit;     
    Int_t     fAODProtection;
    Bool_t fUseQuarkTagInKine;            // flag for quark/hadron level identification of prompt and feeddown

  Bool_t    fWriteVariableTree;       
  TTree    *fVariablesTree;           
  Double_t *fCandidateVariables;      
  Bool_t  fPIDCheck;      
  Bool_t    fDrawDetSignal;   
  TList    *fDetSignal;   

    TH2F *fhMultVZEROTPCoutTrackCorrNoCut;  
    TH2F *fhMultVZEROTPCoutTrackCorr;  
    Bool_t    fEnablePileupRejVZEROTPCout;
  ClassDef(AliAnalysisTaskSED0MassNonPromptFraction,19); 
};

#endif