AliAnalysisTaskSEB0toDStarPi.h

Go to the documentation of this file.

#ifndef AliAnalysisTaskSEB0toDStarPi_H
#define AliAnalysisTaskSEB0toDStarPi_H
/**************************************************************************
 * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */


#include <TH2F.h>
#include <TH3F.h>
#include <THnSparse.h>
#include <vector>
#include "AliAODRecoCascadeHF.h"
#include "AliAODVertex.h"

#include "AliAnalysisTaskSE.h"

class AliRDHFCutsB0toDStarPi;
class AliNormalizationCounter;

class AliAnalysisTaskSEB0toDStarPi : public AliAnalysisTaskSE 
{
  
 public:
  
  AliAnalysisTaskSEB0toDStarPi();
  AliAnalysisTaskSEB0toDStarPi(const Char_t* name,AliRDHFCutsB0toDStarPi* cuts);
  virtual ~AliAnalysisTaskSEB0toDStarPi();

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

  //selection and reconstruction
  void     B0toDStarPiSignalTracksInMC(TClonesArray * mcTrackArray,AliAODEvent*  aodevent,TMatrix * B0toDStarPiLabelMatrix, TList *listout);
  void     D0PionSelection(AliAODEvent* aodEvent,TClonesArray * mcTrackArray, TMatrix * B0toDStarPiLabelMatrix);
  void     D0KaonSelection(AliAODEvent* aodEvent,TClonesArray * mcTrackArray, TMatrix * B0toDStarPiLabelMatrix);
  void     DStarPionSelection(AliAODEvent* aodEvent,TClonesArray * mcTrackArray, TMatrix * B0toDStarPiLabelMatrix);
  void     B0PionSelection(AliAODEvent* aodEvent,TClonesArray * mcTrackArray, TMatrix * B0toDStarPiLabelMatrix);  
  void     D0Selection(AliAODEvent* aodEvent, AliAODVertex *primaryVertex, Double_t bz,TClonesArray * mcTrackArray,TMatrix *  B0toDStarPiLabelMatrix);
  void     DStarAndB0Selection(AliAODEvent* aodEvent, AliAODVertex *primaryVertex, Double_t bz, TClonesArray * mcTrackArray, TMatrix * B0toDStarPiLabelMatrix);

  AliAODVertex* RecalculateVertex(const AliVVertex *primary,TObjArray *tracks,Double_t bField, Int_t finderAlgorithm = 1);
  void     FillFinalTrackHistograms(AliAODRecoCascadeHF * motherCascadeHF, Bool_t isDesiredCandidate,TClonesArray * mcTrackArray);

  void     FillD0Histograms(AliAODRecoDecayHF2Prong * selectedMother, AliAODVertex *primaryVertex, Double_t bz, Int_t motherType, Int_t histType);
  void     FillCascadeMotherHistograms(AliAODRecoCascadeHF * selectedMother, AliAODVertex *primaryVertex, Double_t bz, Int_t motherType, Int_t histType);

  // set MC usage
  void     SetMC(Bool_t bUseMCInfo) {fUseMCInfo = bUseMCInfo;}
  Bool_t   GetMC() const {return fUseMCInfo;}

  Double_t DeltaInvMassDStarKpipi(AliAODRecoCascadeHF * DStar) const;
  Double_t DeltaInvMassB0Kpipipi(AliAODRecoCascadeHF * B0) const;

  void     SetQuickSignalAnalysis(Bool_t value){fQuickSignalAnalysis = value;}
  void     SetGetCutInfo(Bool_t value){fGetCutInfo = value;}

 private:
  
  AliAnalysisTaskSEB0toDStarPi(const AliAnalysisTaskSEB0toDStarPi &source);
  AliAnalysisTaskSEB0toDStarPi& operator=(const AliAnalysisTaskSEB0toDStarPi& source); 
  
  Int_t  fEvents;                
  Bool_t fUseMCInfo;             

  TList *fOutput;                
  TList *fOutputD0Pion;
  TList *fOutputD0Kaon;
  TList *fOutputDStarPion;
  TList *fOutputB0Pion;
  TList *fOutputD0;
  TList *fOutputDStar;
  TList *fOutputB0;
  TList *fOutputD0_D0Pt;
  TList *fOutputD0_DStarPt;
  TList *fOutputDStar_DStarPt;
  TList *fOutputB0MC;             


  AliRDHFCutsB0toDStarPi *fCuts; 

  TH1F *fCEvents;             

  AliNormalizationCounter *fCounter;

  std::vector<Int_t> * fD0PionTracks;
  std::vector<Int_t> * fD0KaonTracks;
  std::vector<Int_t> * fDStarPionTracks;
  std::vector<Int_t> * fB0PionTracks;
  TClonesArray *fD0Tracks;
  TClonesArray *fDStarTracks;
  TClonesArray *fB0Tracks;

    
  Int_t fnPtBins;
  Int_t fnPtBinLimits;
  Float_t * fPtBinLimits;                    //[fnPtBinLimits]
  Int_t fnPtBinsD0forD0ptbin;
  Int_t fnPtBinsD0forD0ptbinLimits;
  Float_t * fPtBinLimitsD0forD0ptbin;        //[fnPtBinsD0forD0ptbinLimits]
  Int_t fnPtBinsD0forDStarptbin;
  Int_t fnPtBinsD0forDStarptbinLimits;
  Float_t * fPtBinLimitsD0forDStarptbin;     //[fnPtBinsD0forDStarptbinLimits]
  Int_t fnPtBinsDStarforDStarptbin;
  Int_t fnPtBinsDStarforDStarptbinLimits;
  Float_t * fPtBinLimitsDStarforDStarptbin;  //[fnPtBinsDStarforDStarptbinLimits]

  Int_t* fDaughterHistogramArray[4][5][15];
  Int_t* fDaughterHistogramArray2D[4][6];
  Int_t* fMotherHistogramArray[6][50][46];
  Int_t* fMotherHistogramArray2D[6][7];

  Bool_t fQuickSignalAnalysis;
  Bool_t fGetCutInfo;

  ClassDef(AliAnalysisTaskSEB0toDStarPi,1); 
};

#endif