[u/mrichter/AliRoot.git] / PWG3 / 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.                  *
 **************************************************************************/

//-----------------------------------------------------------------------
// Author : A. Grelli, UTRECHT
//-----------------------------------------------------------------------

#include <TH2F.h>
#include "TROOT.h"
#include "TSystem.h"
#include "AliAnalysisTaskSE.h"
#include "AliAnalysisVertexingHF.h"
#include "AliAODEvent.h"

class TH2F;
class TH1I;
class TParticle;
class TFile;
class TClonesArray;
class AliCFManager;
class AliAODRecoDecay;
class AliAODRecoDecayHF2Prong;
class AliAODMCParticle;

class AliAnalysisTaskSEDStarSpectra : public AliAnalysisTaskSE {
  
 public:
  
  AliAnalysisTaskSEDStarSpectra();
  AliAnalysisTaskSEDStarSpectra(const Char_t* name);
  AliAnalysisTaskSEDStarSpectra& operator= (const AliAnalysisTaskSEDStarSpectra& c);
  AliAnalysisTaskSEDStarSpectra(const AliAnalysisTaskSEDStarSpectra& c);
  virtual ~AliAnalysisTaskSEDStarSpectra();
  
  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(Double_t finvM, Double_t finvMDStar, Double_t pt, Int_t okD0, Int_t okD0bar);
  void     WrongSignForDStar(Double_t finvM, Double_t finvMDStar, Double_t pt, Int_t okD0, Int_t okD0bar);
  //cuts
  Bool_t   SetUtrechtSelections(Double_t ptD0, AliAnalysisVertexingHF *fVHF);  
  Bool_t   SelectPID(AliAODTrack *track, Double_t nsig);
  // histos
  Bool_t   DefineHistoFroAnalysis(); 
  // set minimum ITS clusters for the analysis
  void     SetMinITSClusters(Int_t minITSClusters) {fMinITSClusters = minITSClusters;}
  Int_t    GetMinITSClusters() const {return fMinITSClusters;}
  // set minimum for soft pion pt
  void     SetMinITSClustersSoft(Int_t minITSClustersSoft) {fMinITSClustersSoft = minITSClustersSoft;}
  Int_t    GetMinITSClustersSoft() const {return fMinITSClustersSoft;}
  // kaon PID
  void     SetPID(Bool_t usePIDforKaons) {fPID = usePIDforKaons;}
  Int_t    GetPID() const {return fPID;}
  // Set N sigmas for PID
  void     SetNSigmasPID(Int_t numberOfSigmasPID) {fNSigma = numberOfSigmasPID;}
  Int_t    GetNSigmasPID() const {return fNSigma;}
  // set MC usage
  void     SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
  Bool_t   GetMC() const {return fUseMCInfo;}
  
 protected:
  
  Int_t  fEvents;                //  n. of events
  AliAnalysisVertexingHF *fVHF;  //  Set the cuts
  Int_t  fMinITSClusters;        //  min n. of ITS clusters for RecoDecay
  Int_t  fMinITSClustersSoft;    //  min n. of ITS clusters for RecoDecay soft pions
  Bool_t fUseMCInfo;             //  Use MC info
  TList *fOutput;                //!  User output
  Int_t  fNSigma;                //  n sigma for kaon PID
  Bool_t fPID;                   //  PID flag
  AliAODTrack* fAODTrack;        //!
  
  // define the histograms
  TH1F *fMCDStarPt;           //!    
  TH1F *fCEvents;             //!
  TH1F *fDStarMass;           //!
  TH1F *fTrueDiff;            //!
  TH2F *fTrueDiff2;           //!
  TH1F *fInvMass;             //!
  TH1F *fInvMass1;            //!
  TH1F *fInvMass2;            //!
  TH1F *fInvMass3;            //!
  TH1F *fInvMass4;            //!
  TH1F *fInvMass5;            //!
  TH1F *fPtDStar;             //!
  TH1F *fDStar;               //!
  TH1F *fDiff;                //!
  TH1F *fDiff1;               //!
  TH1F *fDiff2;               //!
  TH1F *fDiff3;               //!
  TH1F *fDiff4;               //!
  TH1F *fDiff5;               //!
  TH1F *fDiffSideBand;        //!
  TH1F *fDiffSideBand1;       //!
  TH1F *fDiffSideBand2;       //!
  TH1F *fDiffSideBand3;       //!
  TH1F *fDiffSideBand4;       //!
  TH1F *fDiffSideBand5;       //!
  TH1F *fDiffWrongSign;       //!
  TH1F *fDiffWrongSign1;      //!
  TH1F *fDiffWrongSign2;      //!
  TH1F *fDiffWrongSign3;      //!
  TH1F *fDiffWrongSign4;      //!
  TH1F *fDiffWrongSign5;      //!
 
  ClassDef(AliAnalysisTaskSEDStarSpectra,1); // class for D* spectra
};

#endif
Commit	Line	Data
645e004b	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	//-----------------------------------------------------------------------
	19	// Author : A. Grelli, UTRECHT
	20	//-----------------------------------------------------------------------
	21
	22	#include <TH2F.h>
	23	#include "TROOT.h"
	24	#include "TSystem.h"
	25	#include "AliAnalysisTaskSE.h"
	26	#include "AliAnalysisVertexingHF.h"
	27	#include "AliAODEvent.h"
	28
	29	class TH2F;
	30	class TH1I;
	31	class TParticle;
	32	class TFile;
	33	class TClonesArray;
	34	class AliCFManager;
	35	class AliAODRecoDecay;
	36	class AliAODRecoDecayHF2Prong;
	37	class AliAODMCParticle;
	38
	39	class AliAnalysisTaskSEDStarSpectra : public AliAnalysisTaskSE {
	40
	41	public:
	42
	43	AliAnalysisTaskSEDStarSpectra();
	44	AliAnalysisTaskSEDStarSpectra(const Char_t* name);
	45	AliAnalysisTaskSEDStarSpectra& operator= (const AliAnalysisTaskSEDStarSpectra& c);
	46	AliAnalysisTaskSEDStarSpectra(const AliAnalysisTaskSEDStarSpectra& c);
	47	virtual ~AliAnalysisTaskSEDStarSpectra();
	48
	49	virtual void UserCreateOutputObjects();
	50	virtual void Init();
	51	virtual void LocalInit() {Init();}
	52	virtual void UserExec(Option_t *option);
	53	virtual void Terminate(Option_t *option);
	54
	55	//Background simulation
	56	void SideBandBackground(Double_t finvM, Double_t finvMDStar, Double_t pt, Int_t okD0, Int_t okD0bar);
	57	void WrongSignForDStar(Double_t finvM, Double_t finvMDStar, Double_t pt, Int_t okD0, Int_t okD0bar);
	58	//cuts
	59	Bool_t SetUtrechtSelections(Double_t ptD0, AliAnalysisVertexingHF *fVHF);
	60	Bool_t SelectPID(AliAODTrack *track, Double_t nsig);
	61	// histos
	62	Bool_t DefineHistoFroAnalysis();
	63	// set minimum ITS clusters for the analysis
	64	void SetMinITSClusters(Int_t minITSClusters) {fMinITSClusters = minITSClusters;}
65	Int_t GetMinITSClusters() const {return fMinITSClusters;}
66	// set minimum for soft pion pt
67	void SetMinITSClustersSoft(Int_t minITSClustersSoft) {fMinITSClustersSoft = minITSClustersSoft;}
68	Int_t GetMinITSClustersSoft() const {return fMinITSClustersSoft;}
69	// kaon PID
70	void SetPID(Bool_t usePIDforKaons) {fPID = usePIDforKaons;}
71	Int_t GetPID() const {return fPID;}
72	// Set N sigmas for PID
73	void SetNSigmasPID(Int_t numberOfSigmasPID) {fNSigma = numberOfSigmasPID;}
74	Int_t GetNSigmasPID() const {return fNSigma;}
75	// set MC usage
76	void SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
77	Bool_t GetMC() const {return fUseMCInfo;}
78
79	protected:
80
81	Int_t fEvents; // n. of events
82	AliAnalysisVertexingHF *fVHF; // Set the cuts
83	Int_t fMinITSClusters; // min n. of ITS clusters for RecoDecay
84	Int_t fMinITSClustersSoft; // min n. of ITS clusters for RecoDecay soft pions
85	Bool_t fUseMCInfo; // Use MC info
86	TList *fOutput; //! User output
87	Int_t fNSigma; // n sigma for kaon PID
88	Bool_t fPID; // PID flag
89	AliAODTrack* fAODTrack; //!
90
91	// define the histograms
92	TH1F *fMCDStarPt; //!
93	TH1F *fCEvents; //!
94	TH1F *fDStarMass; //!
95	TH1F *fTrueDiff; //!
96	TH2F *fTrueDiff2; //!
97	TH1F *fInvMass; //!
98	TH1F *fInvMass1; //!
99	TH1F *fInvMass2; //!
100	TH1F *fInvMass3; //!
101	TH1F *fInvMass4; //!
102	TH1F *fInvMass5; //!
103	TH1F *fPtDStar; //!
104	TH1F *fDStar; //!
105	TH1F *fDiff; //!
106	TH1F *fDiff1; //!
107	TH1F *fDiff2; //!
108	TH1F *fDiff3; //!
109	TH1F *fDiff4; //!
110	TH1F *fDiff5; //!
111	TH1F *fDiffSideBand; //!
112	TH1F *fDiffSideBand1; //!
113	TH1F *fDiffSideBand2; //!
114	TH1F *fDiffSideBand3; //!
115	TH1F *fDiffSideBand4; //!
116	TH1F *fDiffSideBand5; //!
117	TH1F *fDiffWrongSign; //!
118	TH1F *fDiffWrongSign1; //!
119	TH1F *fDiffWrongSign2; //!
120	TH1F *fDiffWrongSign3; //!
121	TH1F *fDiffWrongSign4; //!
122	TH1F *fDiffWrongSign5; //!
123
124	ClassDef(AliAnalysisTaskSEDStarSpectra,1); // class for D* spectra
125	};
126
127	#endif