#ifndef ALIITSPID_H
#define ALIITSPID_H
#include <TObject.h>
#include <TClonesArray.h>
#include <TVector.h>
#include "../TPC/AliTPCtrack.h"
#include "AliITSIOTrack.h"
#include "AliITStrackV2.h"
#include <assert.h>
//#include <stl.h>
//___________________________________________________________________________
class  AliITSPid :
  public TObject {

public:
		AliITSPid(Int_t ntrs=1000);
		virtual ~AliITSPid(){}
	void	SetEdep(Int_t track,Float_t Edep);
	void	SetPmom(Int_t track,Float_t Pmom);
	void	SetPcod(Int_t track,Int_t Pcod);
	void	Print(Int_t track);
	void	Tab(void);
	void    Reset(void);
	void	SetVec(Int_t track,TVector info);
	TVector* GetVec(Int_t track);
	Int_t	GetPcode(TClonesArray*,Float_t);
	Int_t	GetPcode(Float_t,Float_t);
	Int_t   GetPcode(AliTPCtrack*);  // For PID TPC
//        Int_t   GetPcode(AliITSIOTrack*); // For PID ITS tracking V1
        Int_t   GetPcode(AliITStrackV2*); // For PID ITS tracking V2
	void	SetCut(Int_t,Float_t,Float_t,Float_t,
			    Float_t,Float_t,Float_t,Float_t);
	Float_t GetWpi(){return fWpi;}
	Float_t GetWk(){return fWk;}
	Float_t GetWp(){return fWp;}
	Int_t	GetPid(){return fPcode;};
protected:
public:
	Float_t cut[13][7];
	Int_t       mxtrs;
	TClonesArray *trs;
	Float_t qtot;
	Float_t fWpi,fWk,fWp;
	Float_t fRpik,fRppi,fRpka,fRp; 
	Int_t 	fPcode;
//private:
public:
	Float_t qcorr(Float_t);
	int	qcomp(Float_t* qa,Float_t* qb){return qa[0]>qb[0]?1:0;}
	Float_t qtrm(Int_t track);
	Float_t qtrm(Float_t qarr[6],Int_t narr);
	Int_t	wpik(Int_t,Float_t);
	Int_t	wpikp(Int_t,Float_t);
	Int_t	pion(){return fWpi=1.,fPcode=211;}
	Int_t	kaon(){return fWk=1.,fPcode=321;}
	Int_t	proton(){return fWp=1.,fPcode=2212;}
public:
  ClassDef(AliITSPid,1) // Class for ITS PID
};

#endif