#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