[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 * const ESD){fESD1=ESD;}; // set ESD pointer
                void SetStack(AliStack * const 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() const;

                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 AliHists{
                        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

			AliHists()
			: fNtracks()
			, fNprimVtxContributor()
			, fPtElec()
			, fPtElecContributor()
			, fNtrackCount(0)
			, fNprimVtxContributorCount(0)
			{
			  // default constructor
			}

			AliHists(const AliHists & p)
			: fNtracks(p.fNtracks)
			, fNprimVtxContributor(p.fNprimVtxContributor)
			, fPtElec(p.fPtElec)
			, fPtElecContributor(p.fPtElecContributor)
			, fNtrackCount(p.fNtrackCount)
			, fNprimVtxContributorCount(p.fNprimVtxContributorCount)
			{
			  // copy constructor
			}
			AliHists &operator=(const AliHists &)
			{
			  // assignment operator, not yet implemented
			  return *this;
			}
                };
                AliHists fPrimVtx[10]; // define structure of histograms

                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	**************************************************************************/
50685501	15	//
	16	// QA class of primary vertex study for Heavy Flavor electrons
	17	//
259c3296	18
75d81601	19	#ifndef ALIHFEPRIVTX_H
75d81601	20	#define ALIHFEPRIVTX_H
259c3296	21
259c3296	22	#ifndef ROOT_TObject
75d81601	23	//#include <TObject.h>
259c3296	24	#endif
	25
	26	class TH1F;
	27	class TH1I;
	28	class TH2F;
	29	class TString;
	30	class AliESDEvent;
	31	class AliESDtrack;
	32	class AliStack;
	33
	34	//________________________________________________________________
	35	class AliHFEpriVtx : public TObject {
	36
	37	public:
	38	AliHFEpriVtx();
	39	AliHFEpriVtx(const AliHFEpriVtx &p); // copy constructor
	40	AliHFEpriVtx &operator=(const AliHFEpriVtx &); // assignment operator
	41	virtual ~AliHFEpriVtx();
	42
259c3296	43	void CreateHistograms(TString hnopt=""); // create histograms
259c3296	44	void Init();
50685501	45	void SetEvent(AliESDEvent * const ESD){fESD1=ESD;}; // set ESD pointer
50685501	46	void SetStack(AliStack * const stack){fStack=stack;} // set stack pointer
259c3296	47	void CountNtracks(Int_t sourcePart, Int_t recpid, Double_t recprob); // count number of tracks passed certain cut
	48	void FillNtracks(); // fill counted number of tracks
	49	void CountPriVxtElecContributor(AliESDtrack *ESDelectron, Int_t sourcePart, Int_t recpid, Double_t recprob);
	50	void GetNPriVxtContributor();
75d81601	51	void FillNprimVtxContributor() const;
259c3296	52
	53	Int_t GetMCPID(AliESDtrack *track); // return mc pid
	54
	55	private:
	56
	57	AliESDEvent* fESD1; // ESD event
	58	AliStack* fStack; // MC Stack
	59
	60	TString fkSourceLabel[10]; // storing source label
	61
	62	enum kSources {kAll, kDirectCharm, kDirectBeauty, kBeautyCharm, kGamma, kPi0, kElse, kBeautyGamma, kBeautyPi0, kBeautyElse};
	63
75d81601	64	struct AliHists{
259c3296	65	TH1F *fNtracks; // histogram to fill number of counted tracks for different sources
	66	TH1F *fNprimVtxContributor; // histogram to fill number of tracks contributing primary vertex
	67	TH1F *fPtElec; // histogram to fill pt of electron tracks
	68	TH1F *fPtElecContributor; // histogram to fill pt of electron tracks contributing primary vertex
	69	Int_t fNtrackCount; // number of counted track
	70	Int_t fNprimVtxContributorCount; // number of tracks contributing primary vertex
259c3296	71
75d81601	72	AliHists()
	73	: fNtracks()
	74	, fNprimVtxContributor()
	75	, fPtElec()
	76	, fPtElecContributor()
	77	, fNtrackCount(0)
	78	, fNprimVtxContributorCount(0)
	79	{
	80	// default constructor
	81	}
	82
	83	AliHists(const AliHists & p)
	84	: fNtracks(p.fNtracks)
	85	, fNprimVtxContributor(p.fNprimVtxContributor)
	86	, fPtElec(p.fPtElec)
	87	, fPtElecContributor(p.fPtElecContributor)
	88	, fNtrackCount(p.fNtrackCount)
	89	, fNprimVtxContributorCount(p.fNprimVtxContributorCount)
	90	{
	91	// copy constructor
	92	}
	93	AliHists &operator=(const AliHists &)
	94	{
	95	// assignment operator, not yet implemented
	96	return *this;
	97	}
	98	};
	99	AliHists fPrimVtx[10]; // define structure of histograms
259c3296	100
	101	Int_t fNtrackswoPid; // number of track counted
	102	TH1F *fHNtrackswoPid; // histogram to fill number of track counted
	103	TH1I *fNESDprimVtxContributor; // histogram to fill number of primary vertex contributor for given event
	104	TH1I *fNESDprimVtxIndices; // histogram to fill number of primary vertex indices for given event
	105	TH2F *fDiffDCAvsPt; // histogram to fill DCA difference as a function of pT
	106	TH2F *fDiffDCAvsNt; // histogram to fill DCA difference as a function of pT
	107
	108
dbe3abbe	109	ClassDef(AliHFEpriVtx,0);
259c3296	110	};
	111
	112	#endif