[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSUTrackHyp.h

#ifndef ALIITSUTRACKHYP_H
#define ALIITSUTRACKHYP_H

#include <TObject.h>
#include <TObjArray.h>
#include "AliKalmanTrack.h"
#include "AliITSUSeed.h"
class AliESDtrack;
class AliCluster;

// Container for track hypotheses

class AliITSUTrackHyp: public AliKalmanTrack
{
 public:
  AliITSUTrackHyp(Int_t nlr=0);
  AliITSUTrackHyp(const AliITSUTrackHyp& src);
  AliITSUTrackHyp &operator=(const AliITSUTrackHyp &src);
  virtual ~AliITSUTrackHyp();
  //
  Int_t              GetNLayers()        const {return fNLayers;}
  Int_t              GetNSeeds(Int_t lr) const {return fLayerSeeds[lr].GetEntriesFast();}
  AliITSUSeed*       GetSeed(Int_t lr, Int_t id) const {return (AliITSUSeed*)fLayerSeeds[lr].UncheckedAt(id);}
  AliITSUSeed*       GetWinner()         const;
  AliESDtrack*       GetESDTrack()       const {return fESDTrack;}
  void               DefineWinner(Int_t lr, Int_t id);
  const TObjArray*   GetLayerSeeds(Int_t lr) const {return lr<fNLayers ? &fLayerSeeds[lr] : 0;}
  void               AddSeed(AliITSUSeed* seed, Int_t lr) {fLayerSeeds[lr].AddLast(seed);}
  void               SetESDTrack(AliESDtrack* esdtr) {fESDTrack = esdtr;}
  void               UpdateESD();
  //
  void               SetChi2(Double_t chi2) {fChi2 = chi2;}
  //
  virtual Double_t   GetPredictedChi2(const AliCluster *c) const;
  virtual Bool_t     PropagateTo(Double_t xr, Double_t x0, Double_t rho);
  virtual Bool_t     Update(const AliCluster* c, Double_t chi2, Int_t index);
  virtual Int_t      GetClusterIndex(Int_t)  const { return -1;}
  virtual Int_t      GetNumberOfTracklets()  const { return 0;}
  virtual Int_t      GetTrackletIndex(Int_t) const { return -1;}
  virtual Double_t   GetPIDsignal()          const { return 0;}
  //
  virtual void       Print(Option_t* option = "") const;
  //
 protected:
  UChar_t          fNLayers;               // number of layers
  AliESDtrack*     fESDTrack;              // reference esd track
  TObjArray*       fLayerSeeds;            // seeds of given layer
  //
  ClassDef(AliITSUTrackHyp,1)
};


#endif
Commit	Line	Data
716ccba7	1	#ifndef ALIITSUTRACKHYP_H
	2	#define ALIITSUTRACKHYP_H
	3
	4	#include <TObject.h>
	5	#include <TObjArray.h>
3e4e3c23	6	#include "AliKalmanTrack.h"
716ccba7	7	#include "AliITSUSeed.h"
3e4e3c23	8	class AliESDtrack;
3e4e3c23	9	class AliCluster;
716ccba7	10
	11	// Container for track hypotheses
	12
3e4e3c23	13	class AliITSUTrackHyp: public AliKalmanTrack
716ccba7	14	{
	15	public:
	16	AliITSUTrackHyp(Int_t nlr=0);
	17	AliITSUTrackHyp(const AliITSUTrackHyp& src);
	18	AliITSUTrackHyp &operator=(const AliITSUTrackHyp &src);
	19	virtual ~AliITSUTrackHyp();
	20	//
	21	Int_t GetNLayers() const {return fNLayers;}
3e4e3c23	22	Int_t GetNSeeds(Int_t lr) const {return fLayerSeeds[lr].GetEntriesFast();}
	23	AliITSUSeed* GetSeed(Int_t lr, Int_t id) const {return (AliITSUSeed*)fLayerSeeds[lr].UncheckedAt(id);}
	24	AliITSUSeed* GetWinner() const;
	25	AliESDtrack* GetESDTrack() const {return fESDTrack;}
	26	void DefineWinner(Int_t lr, Int_t id);
716ccba7	27	const TObjArray* GetLayerSeeds(Int_t lr) const {return lr<fNLayers ? &fLayerSeeds[lr] : 0;}
3e4e3c23	28	void AddSeed(AliITSUSeed* seed, Int_t lr) {fLayerSeeds[lr].AddLast(seed);}
	29	void SetESDTrack(AliESDtrack* esdtr) {fESDTrack = esdtr;}
	30	void UpdateESD();
	31	//
	32	void SetChi2(Double_t chi2) {fChi2 = chi2;}
716ccba7	33	//
3e4e3c23	34	virtual Double_t GetPredictedChi2(const AliCluster *c) const;
	35	virtual Bool_t PropagateTo(Double_t xr, Double_t x0, Double_t rho);
	36	virtual Bool_t Update(const AliCluster* c, Double_t chi2, Int_t index);
	37	virtual Int_t GetClusterIndex(Int_t) const { return -1;}
	38	virtual Int_t GetNumberOfTracklets() const { return 0;}
	39	virtual Int_t GetTrackletIndex(Int_t) const { return -1;}
	40	virtual Double_t GetPIDsignal() const { return 0;}
716ccba7	41	//
	42	virtual void Print(Option_t* option = "") const;
	43	//
	44	protected:
	45	UChar_t fNLayers; // number of layers
3e4e3c23	46	AliESDtrack* fESDTrack; // reference esd track
716ccba7	47	TObjArray* fLayerSeeds; // seeds of given layer
	48	//
	49	ClassDef(AliITSUTrackHyp,1)
	50	};
	51
716ccba7	52
716ccba7	53	#endif