[u/mrichter/AliRoot.git] / PWGHF / hfe / AliSelectNonHFE.h

#ifndef ALISELECTNONHFE_H
#define ALISELECTNONHFE_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

////////////////////////////////////////////////////////////////////////
//                                                                    //
//  	Task for the Selection of Non-Heavy-Flavour-Electrons 	      //
//                                                                    //
//  		Author: Elienos Pereira de Oliveira Filho 				  //
//					(University of São Paulo)						  //	
//                                                                    //
////////////////////////////////////////////////////////////////////////

#ifndef ROOT_TNamed
#include <TNamed.h>
#endif

class TH1F;
class TH2F;
class AliESDEvent;
class AliESDtrack;
class AliESDtrackCuts;
class AliPIDResponse;


class AliSelectNonHFE : public TNamed {
 public:
  AliSelectNonHFE();
  AliSelectNonHFE(const char *name, const Char_t *title);
  virtual ~AliSelectNonHFE();
  Int_t GetNLS() const {return fNLS;};
  Int_t GetNULS() const {return fNULS;};
  Int_t* GetPartnersLS() const {return fLSPartner;};
  Int_t* GetPartnersULS() const {return fULSPartner;};
  Bool_t IsLS() const {return fIsLS;};
  Bool_t IsULS() const {return fIsULS;};
  void FindNonHFE(Int_t iTrack, AliESDtrack *track1, AliESDEvent *fESD);
  void SetAlgorithm(TString Algorithm) {fAlgorithm = Algorithm;};
  void SetChi2OverNDFCut(Double_t Chi2OverNDFCut) {fChi2OverNDFCut = Chi2OverNDFCut;};
  void SetDCACut(Double_t dcaCut) {fdcaCut = dcaCut;};
  void SetHistAngleBack(TH1F *HistAngleBack) {fHistAngleBack = HistAngleBack;};
  void SetHistAngle(TH1F *HistAngle) {fHistAngle = HistAngle;};
  void SetHistDCABack(TH1F *HistDCABack) {fHistDCABack = HistDCABack;};
  void SetHistDCA(TH1F *HistDCA) {fHistDCA = HistDCA;};
  void SetHistMassBack(TH1F *HistMassBack) {fHistMassBack = HistMassBack;};
  void SetHistMass(TH1F *HistMass) {fHistMass = HistMass;};
  void SetInvariantMassCut(Double_t MassCut) {fMassCut = MassCut;};
  void SetOpeningAngleCut(Double_t AngleCut) {fAngleCut = AngleCut;};
  void SetPIDresponse(AliPIDResponse* PIDResponse) {fPIDResponse = PIDResponse;};
	
	void SetTrackCuts(Double_t TPCnSigmaMin, Double_t TPCnSigmaMax, Double_t TOFnSigmaMin, Double_t TOFnSigmaMax) 
	{
		fTPCnSigmaMin = TPCnSigmaMin; 
		fTPCnSigmaMax = TPCnSigmaMax; 
		fTOFnSigmaMin = TOFnSigmaMin; 
		fTOFnSigmaMax = TOFnSigmaMax;
	};
	
	void SetTrackCuts(Double_t TPCnSigmaMin, Double_t TPCnSigmaMax, Double_t TOFnSigmaMin, Double_t TOFnSigmaMax, AliESDtrackCuts* TrackCuts) 
	{
		fTPCnSigmaMin = TPCnSigmaMin; 
		fTPCnSigmaMax = TPCnSigmaMax; 
		fTOFnSigmaMin = TOFnSigmaMin; 
		fTOFnSigmaMax = TOFnSigmaMax; 
		fTrackCuts = TrackCuts;
	};
  
 private:
  AliESDtrackCuts	*fTrackCuts;		//! Track quality
  TString		fAlgorithm;       	//Algorithm choice: "MA" (Manual Algorithm) or "KF" (Kalman Filter Algorithm)
  Double_t 		fAngleCut; 		//Maximum opening angle between the tracks
  Double_t 		fdcaCut;   		//Maximum dca between the tracks
  Double_t 		fTPCnSigmaMin; 		//Minimum partner TPCnSigma value
  Double_t 		fTPCnSigmaMax;  		//Maximum partner TPCnSigma value
  Double_t 		fTOFnSigmaMin; 		//Minimum partner TOFnSigma value
  Double_t 		fTOFnSigmaMax;  		//Maximum partner TOFnSigma value
  Double_t 		fMassCut;		//Maximum Invariant Mass Value for Non-HF-Electrons
  Double_t		fChi2OverNDFCut;        //Maximum value of Chi2 over NDF in the case of KF Algorithm
  Bool_t		fIsLS;			//Unlike signal pairs Flag
  Bool_t		fIsULS;			//like signal pairs Flag
  Int_t			fNLS;			//Number of Unlike signal pairs
  Int_t			fNULS;			//Number of like signal pairs
  Int_t			*fLSPartner;	        //! Pointer for the LS partners index
  Int_t			*fULSPartner;	        //! Pointer for the ULS partners index
  TH1F			*fHistMass;		//! Invariant mass histogram for Unlike sign pairs
  TH1F			*fHistMassBack;         //! Invariant mass histogram for like sign pairs
  TH1F			*fHistDCA;		//! DCA histogram for Unlike sign pairs
  TH1F			*fHistDCABack;	        //! DCA histogram for like sign pairs
  TH1F			*fHistAngle;	        //! Opening Angle histogram for Unlike sign pairs
  TH1F			*fHistAngleBack;        //! Opening Angle histogram for like sign pairs
  AliPIDResponse *fPIDResponse;     //! PID response object
  
  AliSelectNonHFE(const AliSelectNonHFE&); // not implemented
  AliSelectNonHFE& operator=(const AliSelectNonHFE&); // not implemented
  
  ClassDef(AliSelectNonHFE, 1); //!example of analysis
};

#endif
Commit	Line	Data
dfcc2025	1	#ifndef ALISELECTNONHFE_H
	2	#define ALISELECTNONHFE_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	////////////////////////////////////////////////////////////////////////
	8	// //
	9	// Task for the Selection of Non-Heavy-Flavour-Electrons //
	10	// //
	11	// Author: Elienos Pereira de Oliveira Filho //
	12	// (University of São Paulo) //
	13	// //
	14	////////////////////////////////////////////////////////////////////////
	15
	16	#ifndef ROOT_TNamed
	17	#include <TNamed.h>
	18	#endif
	19
	20	class TH1F;
	21	class TH2F;
	22	class AliESDEvent;
	23	class AliESDtrack;
	24	class AliESDtrackCuts;
70448e3b	25	class AliPIDResponse;
dfcc2025	26
	27
	28	class AliSelectNonHFE : public TNamed {
	29	public:
	30	AliSelectNonHFE();
	31	AliSelectNonHFE(const char name, const Char_t title);
	32	virtual ~AliSelectNonHFE();
	33	Int_t GetNLS() const {return fNLS;};
	34	Int_t GetNULS() const {return fNULS;};
	35	Int_t* GetPartnersLS() const {return fLSPartner;};
	36	Int_t* GetPartnersULS() const {return fULSPartner;};
	37	Bool_t IsLS() const {return fIsLS;};
	38	Bool_t IsULS() const {return fIsULS;};
	39	void FindNonHFE(Int_t iTrack, AliESDtrack track1, AliESDEvent fESD);
	40	void SetAlgorithm(TString Algorithm) {fAlgorithm = Algorithm;};
	41	void SetChi2OverNDFCut(Double_t Chi2OverNDFCut) {fChi2OverNDFCut = Chi2OverNDFCut;};
	42	void SetDCACut(Double_t dcaCut) {fdcaCut = dcaCut;};
	43	void SetHistAngleBack(TH1F *HistAngleBack) {fHistAngleBack = HistAngleBack;};
	44	void SetHistAngle(TH1F *HistAngle) {fHistAngle = HistAngle;};
	45	void SetHistDCABack(TH1F *HistDCABack) {fHistDCABack = HistDCABack;};
	46	void SetHistDCA(TH1F *HistDCA) {fHistDCA = HistDCA;};
	47	void SetHistMassBack(TH1F *HistMassBack) {fHistMassBack = HistMassBack;};
	48	void SetHistMass(TH1F *HistMass) {fHistMass = HistMass;};
	49	void SetInvariantMassCut(Double_t MassCut) {fMassCut = MassCut;};
	50	void SetOpeningAngleCut(Double_t AngleCut) {fAngleCut = AngleCut;};
70448e3b	51	void SetPIDresponse(AliPIDResponse* PIDResponse) {fPIDResponse = PIDResponse;};
	52
	53	void SetTrackCuts(Double_t TPCnSigmaMin, Double_t TPCnSigmaMax, Double_t TOFnSigmaMin, Double_t TOFnSigmaMax)
	54	{
	55	fTPCnSigmaMin = TPCnSigmaMin;
	56	fTPCnSigmaMax = TPCnSigmaMax;
	57	fTOFnSigmaMin = TOFnSigmaMin;
	58	fTOFnSigmaMax = TOFnSigmaMax;
	59	};
	60
	61	void SetTrackCuts(Double_t TPCnSigmaMin, Double_t TPCnSigmaMax, Double_t TOFnSigmaMin, Double_t TOFnSigmaMax, AliESDtrackCuts* TrackCuts)
	62	{
	63	fTPCnSigmaMin = TPCnSigmaMin;
	64	fTPCnSigmaMax = TPCnSigmaMax;
	65	fTOFnSigmaMin = TOFnSigmaMin;
	66	fTOFnSigmaMax = TOFnSigmaMax;
	67	fTrackCuts = TrackCuts;
	68	};
dfcc2025	69
	70	private:
	71	AliESDtrackCuts *fTrackCuts; //! Track quality
	72	TString fAlgorithm; //Algorithm choice: "MA" (Manual Algorithm) or "KF" (Kalman Filter Algorithm)
	73	Double_t fAngleCut; //Maximum opening angle between the tracks
	74	Double_t fdcaCut; //Maximum dca between the tracks
70448e3b	75	Double_t fTPCnSigmaMin; //Minimum partner TPCnSigma value
	76	Double_t fTPCnSigmaMax; //Maximum partner TPCnSigma value
	77	Double_t fTOFnSigmaMin; //Minimum partner TOFnSigma value
	78	Double_t fTOFnSigmaMax; //Maximum partner TOFnSigma value
dfcc2025	79	Double_t fMassCut; //Maximum Invariant Mass Value for Non-HF-Electrons
	80	Double_t fChi2OverNDFCut; //Maximum value of Chi2 over NDF in the case of KF Algorithm
	81	Bool_t fIsLS; //Unlike signal pairs Flag
	82	Bool_t fIsULS; //like signal pairs Flag
	83	Int_t fNLS; //Number of Unlike signal pairs
	84	Int_t fNULS; //Number of like signal pairs
	85	Int_t *fLSPartner; //! Pointer for the LS partners index
	86	Int_t *fULSPartner; //! Pointer for the ULS partners index
	87	TH1F *fHistMass; //! Invariant mass histogram for Unlike sign pairs
	88	TH1F *fHistMassBack; //! Invariant mass histogram for like sign pairs
	89	TH1F *fHistDCA; //! DCA histogram for Unlike sign pairs
	90	TH1F *fHistDCABack; //! DCA histogram for like sign pairs
	91	TH1F *fHistAngle; //! Opening Angle histogram for Unlike sign pairs
	92	TH1F *fHistAngleBack; //! Opening Angle histogram for like sign pairs
70448e3b	93	AliPIDResponse *fPIDResponse; //! PID response object
dfcc2025	94
	95	AliSelectNonHFE(const AliSelectNonHFE&); // not implemented
	96	AliSelectNonHFE& operator=(const AliSelectNonHFE&); // not implemented
	97
	98	ClassDef(AliSelectNonHFE, 1); //!example of analysis
	99	};
	100
	101	#endif