AliAnalysisTaskSEDplus.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKSEDPLUS_H
#define ALIANALYSISTASKSEDPLUS_H

/* Copyright(c) 1998-2008, 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 <TH1F.h>
#include <TH2F.h>
#include <TH3F.h>
#include <THnSparse.h>
#include <TArrayD.h>

#include "AliRDHFCutsDplustoKpipi.h"
#include "AliAnalysisTaskSE.h"
#include "AliAnalysisVertexingHF.h"
#include "AliNormalizationCounter.h"
#include "AliAODMCHeader.h"
#include "AliAODMCParticle.h"

class AliAnalysisTaskSEDplus : public AliAnalysisTaskSE
{
 public:

  AliAnalysisTaskSEDplus();
  AliAnalysisTaskSEDplus(const char *name, AliRDHFCutsDplustoKpipi* analysiscuts,Int_t fillNtuple=0);
  virtual ~AliAnalysisTaskSEDplus();

  void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
  void SetDoLikeSign(Int_t dols=0){fDoLS=dols;}
  void SetSystem(Int_t system=0){fSystem=system;}
  void SetCutsDistr(Bool_t cutsDistr=kTRUE){fCutsDistr=cutsDistr;}
  void SetDoImpactParameterHistos(Bool_t doImp=kTRUE){fDoImpPar=doImp;}
  void SetDoCutVarsSparses(Bool_t doSparse=kTRUE){fDoSparse=doSparse;}
  void SetDoTrackVarHistos(Bool_t doTrackHist=kTRUE){fDoTrackVarHist=doTrackHist;}
  void SetDoMCAcceptanceHistos(Bool_t doMCAcc=kTRUE){fStepMCAcc=doMCAcc;}
  void SetImpactParameterBinning(Int_t nbins, Float_t dmin, Float_t dmax){
    fNImpParBins=nbins;
    fLowerImpPar=dmin;
    fHigherImpPar=dmax;
  }
  void SetUseStrangeness(Bool_t uses=kTRUE){fUseStrangeness=uses;}
  void SetMassLimits(Float_t range);
  void SetMassLimits(Float_t lowlimit, Float_t uplimit);
  void SetBinWidth(Float_t w);
  void SetUseBit(Bool_t dols=kTRUE){fUseBit=dols;}
  void SetAODMismatchProtection(Int_t opt=1) {fAODProtection=opt;}

  void SetUseOnlyPositiveEta(){fEtaSelection=1;}
  void SetUseOnlyNegativeEta(){fEtaSelection=-1;}
  void SetUseFullEta(){fEtaSelection=0;}
  void SetUseQuarkLevelTag(Bool_t opt){fUseQuarkTagInKine=opt;}
  
  void SetCutOnNtracklets(Bool_t applycut=kTRUE, Int_t Ntrckmin=0, Int_t Ntrckmax=100) {
    fCutOnTrckl=applycut;
    fNtrcklMin=Ntrckmin;
    fNtrcklMax=Ntrckmax;
  }
  
  Float_t GetUpperMassLimit(){return fUpmasslimit;}
  Float_t GetLowerMassLimit(){return fLowmasslimit;}
  Int_t GetNBinsPt(){return fNPtBins;}
  Float_t GetBinWidth(){return fBinWidth;}
  Int_t GetNBinsHistos();

  void LSAnalysis(TClonesArray *arrayOppositeSign,TClonesArray *arrayLikeSign,AliAODEvent *aod,AliAODVertex *vtx1, Int_t nDplusOS);

  void CreateLikeSignHistos();
  void CreateImpactParameterHistos();
  void CreateCutVarsSparses();
  void CreateTrackVarHistos();
  void CreateMCAcceptanceHistos();

  Bool_t CheckAcc(TClonesArray* arrayMC,Int_t nProng, Int_t *labDau);
  void FillMCAcceptanceHistos(TClonesArray *arrayMC, AliAODMCHeader *mcHeader, Int_t tracklets);

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

 private:

  AliAnalysisTaskSEDplus(const AliAnalysisTaskSEDplus &source);
  AliAnalysisTaskSEDplus& operator=(const AliAnalysisTaskSEDplus& source);
  Int_t GetHistoIndex(Int_t iPtBin) const { return iPtBin*3;}
  Int_t GetSignalHistoIndex(Int_t iPtBin) const { return iPtBin*3+1;}
  Int_t GetBackgroundHistoIndex(Int_t iPtBin) const { return iPtBin*3+2;}
  Int_t GetLSHistoIndex(Int_t iPtBin)const { return iPtBin*5;}
  Float_t GetTrueImpactParameter(const AliAODMCHeader *mcHeader, TClonesArray* arrayMC, const AliAODMCParticle *partDp) const;
  Float_t GetStrangenessWeights(const AliAODRecoDecayHF3Prong* d, TClonesArray* arrayMC, Float_t factor[3]) const;

  enum {kVarForSparse=13,kVarForSparseFD=14,kVarForTrackSparse=7,kVarForImpPar=3};

  TList* fOutput;            
  TH1F* fHistNEvents;        
  TH1F* fHistNCandidates;    
  TH1F** fMassHist;          
  TH1F** fMassHistPlus;      
  TH1F** fMassHistMinus;     
  TH1F** fMassHistNoPid;     
  TH1F** fCosPHist;          
  TH1F** fDLenHist;          
  TH1F** fSumd02Hist;        
  TH1F** fSigVertHist;       
  TH1F** fPtMaxHist;         
  TH1F** fPtKHist;           
  TH1F** fPtpi1Hist;         
  TH1F** fPtpi2Hist;         
  TH1F** fDCAHist;           
  TH1F** fDLxy;              
  TH1F** fCosxy;             
  TH2F *fCorreld0Kd0pi[3];   
  TH2F *fHistCentrality[3];  
  THnSparseF *fHistMassPtImpPar[5];  
  THnSparseF *fSparseCutVars[3];     
  THnSparseF *fHistTrackVar;         
  THnSparseF *fMCAccPrompt;          
  THnSparseF *fMCAccBFeed;           
  TH2F *fPtVsMassNoPid;      
  TH2F *fPtVsMass;           
  TH2F *fPtVsMassPlus;       
  TH2F *fPtVsMassMinus;      
  TH2F *fPtVsMassBadDaus;    
  TH2F *fPtVsMassGoodDaus;   
  TH3F *fYVsPtNoPid;         
  TH3F *fYVsPt;              
  TH2F *fYVsPtSigNoPid;      
  TH2F *fYVsPtSig;           
  TH2F *fPhiEtaCand;         
  TH2F *fPhiEtaCandSigReg;   
  TH1F *fSPDMult;            
  TH1F* fDaughterClass;      
  TH1F* fDeltaID;            
  TH2F* fIDDauVsIDTra;       

  TH1F** fMassHistLS;        
  TH1F** fCosPHistLS;        
  TH1F** fDLenHistLS;        
  TH1F** fSumd02HistLS;      
  TH1F** fSigVertHistLS;     
  TH1F** fPtMaxHistLS;       
  TH1F** fDCAHistLS;         

  TNtuple *fNtupleDplus; 
  Float_t fUpmasslimit;  
  Float_t fLowmasslimit; 
  Int_t fNPtBins; 
  Float_t fBinWidth;
  TList *fListCuts; 
  AliRDHFCutsDplustoKpipi *fRDCutsAnalysis; 
  AliNormalizationCounter *fCounter;
  Int_t fFillNtuple;   
  Int_t fAODProtection;  
  Bool_t fReadMC;    
  Bool_t fUseStrangeness;
  Bool_t fUseBit;      
  Bool_t fCutsDistr;    
  Bool_t fDoImpPar;    
  Bool_t fDoSparse;    
  Bool_t fDoTrackVarHist; 
  Bool_t fStepMCAcc;   
  Bool_t fUseQuarkTagInKine; 
  Int_t  fNImpParBins;   
  Float_t fLowerImpPar;  
  Float_t fHigherImpPar; 
  Int_t  fDoLS;        
  Int_t fEtaSelection; 
  Int_t fSystem;   
  Int_t fNtrcklMin;   
  Int_t fNtrcklMax;   
  Bool_t fCutOnTrckl;  
  
  ClassDef(AliAnalysisTaskSEDplus,31); 
};

#endif