AliAnalysisTaskSESignificance.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKSESIGNIFICANCE_H
#define ALIANALYSISTASKSESIGNIFICANCE_H

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

/* $Id$ */ 

//*************************************************************************
//*************************************************************************

#include "AliAnalysisTaskSE.h"
#include "AliAnalysisVertexingHF.h"

class TH1F;
class AliMultiDimVector;
class AliRDHFCuts;

class AliAnalysisTaskSESignificance : public AliAnalysisTaskSE
{
 public:

  enum FeedDownEnum {kBoth,kCharmOnly,kBeautyOnly};
  enum ChanDs {kAllReson,kPhi,kK0star};

  AliAnalysisTaskSESignificance();
  AliAnalysisTaskSESignificance(const char *name, TList *listMDV,AliRDHFCuts *RDCuts, Int_t decaychannel,Int_t selectionlevel=AliRDHFCuts::kAll);
 
  virtual ~AliAnalysisTaskSESignificance();

  Bool_t CheckConsistency();
  void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
  void SetBFeedDown(FeedDownEnum flagB);//see enum
  void SetDFromCharmOnly(){SetBFeedDown(kCharmOnly);}
  void SetDFromBeautyOnly(){SetBFeedDown(kBeautyOnly);}
  void SetMassLimits(Float_t range,Int_t pdg);
  void SetMassLimits(Float_t lowlimit, Float_t uplimit);
  void SetNBins(Int_t nbins){fNBins=nbins;}
  void SetFillWithPartAntiPartBoth(Int_t value){fPartOrAndAntiPart=value;}
  void SetDsChannel(Int_t chan){fDsChannel=chan;}
  void SetUseSelBit(Bool_t selBit=kTRUE){fUseSelBit=selBit;}
  void SetAODMismatchProtection(Int_t opt=1) {fAODProtection=opt;}

  //void SetMultiVector(const AliMultiDimVector *MultiDimVec){fMultiDimVec->CopyStructure(MultiDimVec);}
  Float_t GetUpperMassLimit()const {return fUpmasslimit;}
  Float_t GetLowerMassLimit()const {return fLowmasslimit;}
  Int_t GetNBins()const {return fNBins;}
  Int_t GetFillWithPartAntiPartBoth()const {return fPartOrAndAntiPart;}
  Int_t GetBFeedDown()const {return fBFeedDown;}
  Int_t GetDsChannel()const {return fDsChannel;}
  Bool_t GetUseSelBit()const {return fUseSelBit;}

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

  void SetPDGdaughterDstoKKpi(){
    fPDGdaughters[0]=321;//K
    fPDGdaughters[1]=321;//K
    fPDGdaughters[2]=211;//pi
    fPDGdaughters[3]=0; //empty
  }
  void SetPDGdaughterDstopiKK(){
    fPDGdaughters[0]=211;//pi
    fPDGdaughters[1]=321;//K
    fPDGdaughters[2]=321;//K
    fPDGdaughters[3]=0; //empty
  }

  AliAnalysisTaskSESignificance(const AliAnalysisTaskSESignificance &source);
  AliAnalysisTaskSESignificance& operator=(const AliAnalysisTaskSESignificance& source);
  void SetPDGCodes();
  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;}
  Int_t CheckOrigin(const AliAODMCParticle* mcPart, const TClonesArray* mcArray) const;

  void FillDplus(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Int_t isSel);
  void FillD02p(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index, Int_t isSel);
  void FillDs(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Int_t isSel,Int_t optDecay);
  void FillDstar(AliAODRecoCascadeHF* dstarD0pi,TClonesArray *arrayMC,Int_t index,Int_t isSel);
  void FillD04p(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Int_t isSel);
  void FillLambdac(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index, Int_t isSel);


  enum {kMaxPtBins=8};
  enum {kMaxCutVar=10};
  enum {kMaxSteps=10};
  enum {kMaxNHist=500000};
  enum {kDplustoKpipi,kD0toKpi,kDstartoKpipi,kDstoKKpi,kD0toKpipipi,kLambdactopKpi};

  TList   *fOutput; 
  TList *fCutList; 
  TH1F *fMassHist[kMaxNHist]; 
  TH1F *fSigHist[kMaxNHist]; 
  TH1F *fBkgHist[kMaxNHist]; 
  TH1F *fRflHist[kMaxNHist]; 
  TH1F* fHistNEvents; 
  Float_t fUpmasslimit;  
  Float_t fLowmasslimit; 
  AliRDHFCuts *fRDCuts;
  Int_t fNPtBins; 
  Int_t fAODProtection;  
  Bool_t fReadMC;    
  Bool_t fUseSelBit;    
  FeedDownEnum fBFeedDown; 
  Int_t fDecChannel; 
  Int_t fPDGmother;  
  Int_t fNProngs;         
  Int_t fPDGdaughters[4]; 
  TString fBranchName;    
  Int_t fSelectionlevel;  
  Int_t   fNVars;         
  Float_t fVars[kMaxCutVar];       
  Int_t fNBins;  
  Int_t fPartOrAndAntiPart;  
  Int_t fDsChannel;          
  Int_t fPDGDStarToD0pi[2]; 
  Int_t fPDGD0ToKpi[2];    

  ClassDef(AliAnalysisTaskSESignificance,6); 
};

#endif