[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAnalysisTaskSEDStarSpectra.h

#ifndef ALIANALYSISTASKSEDSTARSPECTRA_H
#define ALIANALYSISTASKSEDSTARSPECTRA_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 "AliAnalysisTaskSE.h"

class AliRDHFCutsDStartoKpipi;
class AliNormalizationCounter;

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

  // Implementation of interface methods  
  virtual void UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);
 

 //Background simulation
  void     SideBandBackground(AliAODRecoCascadeHF *part, AliRDHFCutsDStartoKpipi *cuts, Int_t isSel, TList *listout);
  void     WrongSignForDStar(AliAODRecoCascadeHF *part, AliRDHFCutsDStartoKpipi *cuts, TList *listout);
    // histos
  void   FillSpectrum(AliAODRecoCascadeHF *part, Int_t isDStar, AliRDHFCutsDStartoKpipi *cuts, Int_t isSel, TList *listout);
  void     DefineHistograms();
  Int_t CheckOrigin(TClonesArray* arrayMC, const AliAODMCParticle *mcPartCandidate) const;
  void CreateImpactParameterHistos();

  // set analysis type
  void     SetAnalysisType(Int_t anaType) {fAnalysis = anaType;}
  void     PrintAnalysisType() {printf("Analysis type: %d\n(0: Heidelberg\t1: Utrecht)",fAnalysis);}
 // set MC usage
  void     SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
  Bool_t   GetMC() const {return fUseMCInfo;}
 // set rare mesons
  void     SetRareSearch(Bool_t theRareOn) {fDoSearch = theRareOn;}
  Bool_t   GetRareSearch() const {return fDoSearch;}
  //impact par study
  void SetDoImpactParameterHistos(Bool_t doImp=kTRUE){fDoImpParDstar=doImp;}
  Bool_t GetDoImpactParameterHistos() const {return fDoImpParDstar;}

  Float_t GetTrueImpactParameterD0(const AliAODMCHeader *mcHeader, TClonesArray* arrayMC, const AliAODMCParticle *partDp) const;

  void SetDoDStarVsY(Bool_t theDStarVsY) {fDoDStarVsY = theDStarVsY;}

 private:
  
  AliAnalysisTaskSEDStarSpectra(const AliAnalysisTaskSEDStarSpectra &source);
  AliAnalysisTaskSEDStarSpectra& operator=(const AliAnalysisTaskSEDStarSpectra& source); 
  
  Int_t  fEvents;                //  n. of events
  Int_t  fAnalysis;		 //  0: HD;	1: UU;
  Double_t fD0Window;		 //  select width on D0Mass
  Double_t fPeakWindow;          //  select width on DstarMass
  Bool_t fUseMCInfo;             //  Use MC info
  Bool_t fDoSearch;              //  Rare mesons
  TList *fOutput;                //!  User output
  TList *fOutputAll;             //!  User output2
  TList *fOutputPID;             //!  User output3
  Int_t  fNSigma;                //  n sigma for kaon PID
  AliRDHFCutsDStartoKpipi *fCuts; // Cuts - sent to output slot 3
  // define the histograms
  TH1F *fCEvents;             //!
  TH2F *fTrueDiff2;           //!
  TH1F *fDeltaMassD1;         //! 
  AliNormalizationCounter *fCounter;//!Counter for normalization slot 4
  Bool_t fDoImpParDstar;  // imppar studies
  Int_t  fNImpParBins;   // nunber of bins in impact parameter histos
  Float_t fLowerImpPar;  // lower limit in impact parameter (um)
  Float_t fHigherImpPar; // higher limit in impact parameter (um)
  Bool_t  fDoDStarVsY;   // flag to enable D* vs y 

  THnSparseF *fHistMassPtImpParTCDs[5];//! histograms for impact paramter studies

  ClassDef(AliAnalysisTaskSEDStarSpectra,9); // class for D* spectra
};

#endif
Commit	Line	Data
63147a75	1	#ifndef ALIANALYSISTASKSEDSTARSPECTRA_H
	2	#define ALIANALYSISTASKSEDSTARSPECTRA_H
	3	/**************************************************************************
	4	* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	5	* *
	6	* Author: The ALICE Off-line Project. *
	7	* Contributors are mentioned in the code where appropriate. *
	8	* *
	9	* Permission to use, copy, modify and distribute this software and its *
	10	* documentation strictly for non-commercial purposes is hereby granted *
	11	* without fee, provided that the above copyright notice appears in all *
	12	* copies and that both the copyright notice and this permission notice *
	13	* appear in the supporting documentation. The authors make no claims *
	14	* about the suitability of this software for any purpose. It is *
	15	* provided "as is" without express or implied warranty. *
	16	**************************************************************************/
	17
	18	/* $Id$ */
	19
	20	#include <TH2F.h>
859bb281	21	#include <TH3F.h>
63147a75	22	#include <THnSparse.h>
	23
	24	#include "AliAnalysisTaskSE.h"
5d836c02	25
	26	class AliRDHFCutsDStartoKpipi;
	27	class AliNormalizationCounter;
63147a75	28
	29	class AliAnalysisTaskSEDStarSpectra : public AliAnalysisTaskSE
	30	{
	31
	32	public:
	33
	34	AliAnalysisTaskSEDStarSpectra();
	35	AliAnalysisTaskSEDStarSpectra(const Char_t* name,AliRDHFCutsDStartoKpipi* cuts);
	36	virtual ~AliAnalysisTaskSEDStarSpectra();
	37
	38	// Implementation of interface methods
	39	virtual void UserCreateOutputObjects();
	40	virtual void Init();
	41	virtual void LocalInit() {Init();}
	42	virtual void UserExec(Option_t *option);
	43	virtual void Terminate(Option_t *option);
	44
	45
	46	//Background simulation
	47	void SideBandBackground(AliAODRecoCascadeHF part, AliRDHFCutsDStartoKpipi cuts, Int_t isSel, TList *listout);
	48	void WrongSignForDStar(AliAODRecoCascadeHF part, AliRDHFCutsDStartoKpipi cuts, TList *listout);
	49	// histos
	50	void FillSpectrum(AliAODRecoCascadeHF part, Int_t isDStar, AliRDHFCutsDStartoKpipi cuts, Int_t isSel, TList *listout);
	51	void DefineHistograms();
5d836c02	52	Int_t CheckOrigin(TClonesArray* arrayMC, const AliAODMCParticle *mcPartCandidate) const;
63147a75	53	void CreateImpactParameterHistos();
	54
	55	// set analysis type
	56	void SetAnalysisType(Int_t anaType) {fAnalysis = anaType;}
	57	void PrintAnalysisType() {printf("Analysis type: %d\n(0: Heidelberg\t1: Utrecht)",fAnalysis);}
	58	// set MC usage
	59	void SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
	60	Bool_t GetMC() const {return fUseMCInfo;}
	61	// set rare mesons
	62	void SetRareSearch(Bool_t theRareOn) {fDoSearch = theRareOn;}
	63	Bool_t GetRareSearch() const {return fDoSearch;}
	64	//impact par study
	65	void SetDoImpactParameterHistos(Bool_t doImp=kTRUE){fDoImpParDstar=doImp;}
5d836c02	66	Bool_t GetDoImpactParameterHistos() const {return fDoImpParDstar;}
63147a75	67
5d836c02	68	Float_t GetTrueImpactParameterD0(const AliAODMCHeader mcHeader, TClonesArray arrayMC, const AliAODMCParticle *partDp) const;
63147a75	69
859bb281	70	void SetDoDStarVsY(Bool_t theDStarVsY) {fDoDStarVsY = theDStarVsY;}
859bb281	71
63147a75	72	private:
	73
	74	AliAnalysisTaskSEDStarSpectra(const AliAnalysisTaskSEDStarSpectra &source);
	75	AliAnalysisTaskSEDStarSpectra& operator=(const AliAnalysisTaskSEDStarSpectra& source);
	76
	77	Int_t fEvents; // n. of events
	78	Int_t fAnalysis; // 0: HD; 1: UU;
	79	Double_t fD0Window; // select width on D0Mass
	80	Double_t fPeakWindow; // select width on DstarMass
	81	Bool_t fUseMCInfo; // Use MC info
	82	Bool_t fDoSearch; // Rare mesons
	83	TList *fOutput; //! User output
	84	TList *fOutputAll; //! User output2
	85	TList *fOutputPID; //! User output3
	86	Int_t fNSigma; // n sigma for kaon PID
	87	AliRDHFCutsDStartoKpipi *fCuts; // Cuts - sent to output slot 3
	88	// define the histograms
	89	TH1F *fCEvents; //!
	90	TH2F *fTrueDiff2; //!
	91	TH1F *fDeltaMassD1; //!
	92	AliNormalizationCounter *fCounter;//!Counter for normalization slot 4
	93	Bool_t fDoImpParDstar; // imppar studies
	94	Int_t fNImpParBins; // nunber of bins in impact parameter histos
	95	Float_t fLowerImpPar; // lower limit in impact parameter (um)
	96	Float_t fHigherImpPar; // higher limit in impact parameter (um)
859bb281	97	Bool_t fDoDStarVsY; // flag to enable D* vs y
63147a75	98
	99	THnSparseF *fHistMassPtImpParTCDs[5];//! histograms for impact paramter studies
	100
	101	ClassDef(AliAnalysisTaskSEDStarSpectra,9); // class for D* spectra
	102	};
	103
	104	#endif
	105