#ifndef ALIISOLATIONCUT_H
#define ALIISOLATIONCUT_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */
/* $Id:  */

//_________________________________________________________________________
// Class containing methods for the isolation cut. 
// An AOD candidate (AliAODPWG4ParticleCorrelation type)
// is passed. Look in a cone around the candidate and study
// the hadronic activity inside to decide if the candidate is isolated
//
//
// 
// -- Author: Gustavo Conesa (INFN-LNF)

//-Yaxian Mao (add the possibility for different IC method with different pt range, 01/10/2010)

// --- ROOT system --- 
#include <TObject.h>
class TObjArray ;

// --- ANALYSIS system ---
class AliAODPWG4ParticleCorrelation ;
class AliCaloTrackReader ;

class AliIsolationCut : public TObject {
  
 public: 
  AliIsolationCut() ; // default ctor
  virtual ~AliIsolationCut() {;} //virtual dtalr
 private:
  AliIsolationCut(const AliIsolationCut & g) ; // cpy ctor
  AliIsolationCut & operator = (const AliIsolationCut & g) ;//cpy assignment

 public: 
 
  enum type {kPtThresIC, kSumPtIC, kPtFracIC, kSumPtFracIC};
  enum partInCone {kNeutralAndCharged=0, kOnlyNeutral=1, kOnlyCharged=2};
	
  Float_t    GetConeSize()           const {return fConeSize   ;}
  Float_t    GetPtThreshold()        const {return fPtThreshold;}
  Float_t    GetSumPtThreshold()     const {return fSumPtThreshold;}
  Float_t    GetPtFraction()         const {return fPtFraction ;}
  Int_t      GetICMethod()           const {return fICMethod   ;}
  Int_t      GetParticleTypeInCone() const {return fPartInCone ;}
	
  TString    GetICParametersList() ; 

  void MakeIsolationCut(TObjArray * const plCTS, TObjArray * const plNe, AliCaloTrackReader * const reader, 
			const Bool_t bFillAOD, AliAODPWG4ParticleCorrelation  * pCandidate, const TString &aodObjArrayName,
			Int_t &n, Int_t & nfrac, Float_t &ptsum, Bool_t & isolated) const ;  
  
  void Print(const Option_t * opt)const;
  
  void SetConeSize(Float_t r)        {fConeSize = r ; }
  void SetPtThreshold(Float_t pt)    {fPtThreshold = pt; }
  void SetSumPtThreshold(Float_t sumpt)    {fSumPtThreshold = sumpt; }
  void SetPtFraction(Float_t pt)     {fPtFraction = pt; }
  void SetICMethod(Int_t i )         {fICMethod = i ; }
  void SetParticleTypeInCone(Int_t i){fPartInCone = i;}
	
  void InitParameters();
  
  
 private:
  
  Float_t      fConeSize ;    //Size of the isolation cone 
  Float_t      fPtThreshold ; //Mimium pt of the particles in the cone or sum in cone (UE pt mean in the forward region cone)
  Float_t      fSumPtThreshold ; //Minium of sum pt of the particles in the cone (UE sum in the forward region cone)
  Float_t      fPtFraction ;  //Fraction of the momentum of particles in cone or sum in cone
  Int_t        fICMethod ;    //Isolation cut method to be used
                              // kPtIC: Pt threshold method
                              // kSumPtIC: Cone pt sum method
                              // kPtFracIC:   Pt threshold, fraction of candidate pt, method
                              // kSumPtFracIC:   Cone pt sum , fraction of cone sum, method
  Int_t        fPartInCone;   // Type of particles inside cone:
							  // kNeutralAndCharged, kOnlyNeutral, kOnlyCharged
	                         
	
	
  ClassDef(AliIsolationCut,3)
} ;


#endif //ALIISOLATIONCUT_H