[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEpriVtx.h

/**************************************************************************
 * Copyright(c) 1998-1999, 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.                  *
 **************************************************************************/

/**************************************************************************
 *                                                                        *
 * QA class of primary vertex study for Heavy Flavor electrons            * 
 *                                                                        *
 **************************************************************************/

#ifndef _ALIHFEPRIVTX_H_
#define _ALIHFEPRIVTX_H_

#ifndef ROOT_TObject
#include <TObject.h>
#endif

class TH1F;
class TH1I;
class TH2F;
class TString;
class AliESDEvent;
class AliESDtrack;
class AliStack;

//________________________________________________________________
class AliHFEpriVtx : public TObject {

        public: 
                AliHFEpriVtx();
                AliHFEpriVtx(const AliHFEpriVtx &p); // copy constructor
                AliHFEpriVtx &operator=(const AliHFEpriVtx &); // assignment operator
                virtual ~AliHFEpriVtx();

                void CreateHistograms(TString hnopt=""); // create histograms
                void Init();
                void SetEvent(AliESDEvent* ESD){fESD1=ESD;}; // set ESD pointer
                void SetStack(AliStack* stack){fStack=stack;} // set stack pointer
                void CountNtracks(Int_t sourcePart, Int_t recpid, Double_t recprob); // count number of tracks passed certain cut
                void FillNtracks(); // fill counted number of tracks
                void CountPriVxtElecContributor(AliESDtrack *ESDelectron, Int_t sourcePart, Int_t recpid, Double_t recprob); 
                void GetNPriVxtContributor();
                void FillNprimVtxContributor();

                Int_t GetMCPID(AliESDtrack *track); // return mc pid

        private:

                AliESDEvent* fESD1; // ESD event 
                AliStack* fStack; // MC Stack

                TString fkSourceLabel[10]; // storing source label

                enum kSources {kAll, kDirectCharm, kDirectBeauty, kBeautyCharm, kGamma, kPi0, kElse, kBeautyGamma, kBeautyPi0, kBeautyElse};

                struct hists{
                        TH1F *fNtracks; // histogram to fill number of counted tracks for different sources
                        TH1F *fNprimVtxContributor; // histogram to fill number of tracks contributing primary vertex 
                        TH1F *fPtElec; // histogram to fill pt of electron tracks
                        TH1F *fPtElecContributor; // histogram to fill pt of electron tracks contributing primary vertex
                        Int_t fNtrackCount; // number of counted track
                        Int_t fNprimVtxContributorCount; // number of tracks contributing primary vertex
                };

                hists fPrimVtx[10]; //

                Int_t fNtrackswoPid; //  number of track counted
                TH1F *fHNtrackswoPid; // histogram to fill number of track counted
                TH1I *fNESDprimVtxContributor; // histogram to fill number of primary vertex contributor for given event 
                TH1I *fNESDprimVtxIndices; // histogram to fill number of primary vertex indices for given event
                TH2F *fDiffDCAvsPt; // histogram to fill DCA difference as a function of pT
                TH2F *fDiffDCAvsNt; // histogram to fill DCA difference as a function of pT


        ClassDef(AliHFEpriVtx,0);
};

#endif
Commit	Line	Data
259c3296	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	/**************************************************************************
	17	* *
	18	* QA class of primary vertex study for Heavy Flavor electrons *
	19	* *
	20	**************************************************************************/
	21
	22	#ifndef _ALIHFEPRIVTX_H_
	23	#define _ALIHFEPRIVTX_H_
	24
	25	#ifndef ROOT_TObject
	26	#include <TObject.h>
	27	#endif
	28
	29	class TH1F;
	30	class TH1I;
	31	class TH2F;
	32	class TString;
	33	class AliESDEvent;
	34	class AliESDtrack;
	35	class AliStack;
	36
	37	//________________________________________________________________
	38	class AliHFEpriVtx : public TObject {
	39
	40	public:
	41	AliHFEpriVtx();
	42	AliHFEpriVtx(const AliHFEpriVtx &p); // copy constructor
	43	AliHFEpriVtx &operator=(const AliHFEpriVtx &); // assignment operator
	44	virtual ~AliHFEpriVtx();
	45
259c3296	46	void CreateHistograms(TString hnopt=""); // create histograms
259c3296	47	void Init();
	48	void SetEvent(AliESDEvent* ESD){fESD1=ESD;}; // set ESD pointer
	49	void SetStack(AliStack* stack){fStack=stack;} // set stack pointer
	50	void CountNtracks(Int_t sourcePart, Int_t recpid, Double_t recprob); // count number of tracks passed certain cut
	51	void FillNtracks(); // fill counted number of tracks
	52	void CountPriVxtElecContributor(AliESDtrack *ESDelectron, Int_t sourcePart, Int_t recpid, Double_t recprob);
	53	void GetNPriVxtContributor();
	54	void FillNprimVtxContributor();
	55
	56	Int_t GetMCPID(AliESDtrack *track); // return mc pid
	57
	58	private:
	59
	60	AliESDEvent* fESD1; // ESD event
	61	AliStack* fStack; // MC Stack
	62
	63	TString fkSourceLabel[10]; // storing source label
	64
	65	enum kSources {kAll, kDirectCharm, kDirectBeauty, kBeautyCharm, kGamma, kPi0, kElse, kBeautyGamma, kBeautyPi0, kBeautyElse};
	66
	67	struct hists{
	68	TH1F *fNtracks; // histogram to fill number of counted tracks for different sources
	69	TH1F *fNprimVtxContributor; // histogram to fill number of tracks contributing primary vertex
	70	TH1F *fPtElec; // histogram to fill pt of electron tracks
	71	TH1F *fPtElecContributor; // histogram to fill pt of electron tracks contributing primary vertex
	72	Int_t fNtrackCount; // number of counted track
	73	Int_t fNprimVtxContributorCount; // number of tracks contributing primary vertex
	74	};
	75
	76	hists fPrimVtx[10]; //
	77
	78	Int_t fNtrackswoPid; // number of track counted
	79	TH1F *fHNtrackswoPid; // histogram to fill number of track counted
	80	TH1I *fNESDprimVtxContributor; // histogram to fill number of primary vertex contributor for given event
	81	TH1I *fNESDprimVtxIndices; // histogram to fill number of primary vertex indices for given event
	82	TH2F *fDiffDCAvsPt; // histogram to fill DCA difference as a function of pT
	83	TH2F *fDiffDCAvsNt; // histogram to fill DCA difference as a function of pT
	84
	85
dbe3abbe	86	ClassDef(AliHFEpriVtx,0);
259c3296	87	};
	88
	89	#endif