[u/mrichter/AliRoot.git] / TRD / AliTRDtrackerV1.h

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

/* $Id$ */

////////////////////////////////////////////////////////////////////////////
//                                                                        //
//  The TRD tracker                                                       //
//                                                                        //
//  Authors:                                                              //
//    Alex Bercuci <A.Bercuci@gsi.de>                                     //
//    Markus Fasel <M.Fasel@gsi.de>                                       //
//                                                                        //
/////////////////////////////////////////////////////////////////////////////

#ifndef ALITRDTRACKER_H
#include "AliTRDtracker.h"
#endif

#define DEBUG

/**************************************************************************
 * Class Status see source file                                           *
 **************************************************************************/
 
class TFile;
class TTreeSRedirector;
class TClonesArray;

class AliRieman;
class AliESDEvent;

class AliTRDseedV1;
class AliTRDstackLayer;
class AliTRDtrackerFitter;
class AliTRDrecoParam;
class AliTRDtrackV1;
class AliTrackPoint;
class AliTRDtrackerV1 : public AliTRDtracker
{

 public:
	enum{
		kNTimeBins = 35,
		kNPlanes = 6,
		kNSeedPlanes = 4,
		kMaxTracksStack = 100,
		kNConfigs = 15
	};
	AliTRDtrackerV1(AliTRDrecoParam *p = 0x0);
	AliTRDtrackerV1(const TFile *in, AliTRDrecoParam *p);
	virtual ~AliTRDtrackerV1();
  
	Int_t          Clusters2Tracks(AliESDEvent *esd);
	static void    GetExtrapolationConfig(Int_t iconfig, Int_t planes[2]);
	static void    GetSeedingConfig(Int_t iconfig, Int_t planes[4]);
	Int_t          FollowBackProlongation(AliTRDtrackV1 &t);
	Int_t          FollowProlongation(AliTRDtrackV1 &t);
	Int_t          PropagateBack(AliESDEvent *event);
	Int_t          RefitInward(AliESDEvent *event);
	void           SetRecoParam(AliTRDrecoParam *p){fRecoParam = p;}
	void           UnloadClusters();

 protected:
	Double_t       BuildSeedingConfigs(AliTRDstackLayer *layer, Int_t *configs);
	Int_t          Clusters2TracksSM( AliTRDtracker::AliTRDtrackingSector *sector, AliESDEvent *esd);
	Int_t          Clusters2TracksStack(AliTRDstackLayer *layer, TClonesArray *esdTrackList);
	void           CookLabel(AliKalmanTrack *pt, Float_t wrong) const;
	Int_t          GetSeedingLayers(AliTRDstackLayer *layers, Double_t *params);
	void           GetMeanCLStack(AliTRDstackLayer *layers, Int_t *planes, Double_t *params);
	AliTRDseedV1*  GetTracklet(AliTRDtrackV1 *trk, Int_t plane, Int_t &idx);
	virtual Bool_t GetTrackPoint(Int_t index, AliTrackPoint &p) const;
	AliTRDcluster *FindSeedingCluster(AliTRDstackLayer *layers, AliTRDseedV1/*AliRieman*/ *sfit);
	
	Double_t       MakeSeedingPlanes(AliTRDstackLayer *layer);
	AliTRDstackLayer *MakeSeedingLayer(AliTRDstackLayer *layers, Int_t Plane);
	Int_t          MakeSeeds(AliTRDstackLayer *layers, AliTRDseedV1 *sseed, Int_t *ipar);
	AliTRDtrackV1*   MakeTrack(AliTRDseedV1 *seeds, Double_t *params);
	Int_t          SetTracklet(AliTRDseedV1 *tracklet);

private:
	AliTRDtrackerV1(const AliTRDtrackerV1 &tracker);
	AliTRDtrackerV1 &operator=(const AliTRDtrackerV1 &tracker);
	Double_t       CookLikelihood(AliTRDseedV1 *cseed, Int_t planes[4], Double_t *chi2);
	void           ImproveSeedQuality(AliTRDstackLayer *layer, AliTRDseedV1 *cseed);

private:

	static Double_t      fgTopologicQA[kNConfigs];         //  Topologic quality
	Double_t             fTrackQuality[kMaxTracksStack];  //  Track quality 
	Int_t                fSeedLayer[kMaxTracksStack];     //  Seed layer
	Int_t                fSieveSeeding;                   //! Seeding iterator
	TClonesArray        *fTracklets;                      // List of tracklets for all sectors
	AliTRDrecoParam     *fRecoParam;                      //  Reconstruction parameters
	AliTRDtrackerFitter *fFitter;                         //! Fitter class of the tracker

	ClassDef(AliTRDtrackerV1, 1)                          //  Stand alone tracker development class

};
#endif
Commit	Line	Data
e4f2f73d	1	#ifndef ALITRDTRACKERV1_H
	2	#define ALITRDTRACKERV1_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
	8	////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// The TRD tracker //
	11	// //
	12	// Authors: //
	13	// Alex Bercuci <A.Bercuci@gsi.de> //
	14	// Markus Fasel <M.Fasel@gsi.de> //
	15	// //
	16	/////////////////////////////////////////////////////////////////////////////
	17
	18	#ifndef ALITRDTRACKER_H
	19	#include "AliTRDtracker.h"
	20	#endif
	21
	22	#define DEBUG
	23
	24	/**************************************************************************
	25	* Class Status see source file *
	26	**************************************************************************/
	27
	28	class TFile;
	29	class TTreeSRedirector;
	30	class TClonesArray;
	31
	32	class AliRieman;
	33	class AliESDEvent;
	34
	35	class AliTRDseedV1;
	36	class AliTRDstackLayer;
	37	class AliTRDtrackerFitter;
	38	class AliTRDrecoParam;
d9950a5a	39	class AliTRDtrackV1;
d9950a5a	40	class AliTrackPoint;
e4f2f73d	41	class AliTRDtrackerV1 : public AliTRDtracker
	42	{
	43
	44	public:
	45	enum{
	46	kNTimeBins = 35,
	47	kNPlanes = 6,
	48	kNSeedPlanes = 4,
	49	kMaxTracksStack = 100,
	50	kNConfigs = 15
	51	};
	52	AliTRDtrackerV1(AliTRDrecoParam *p = 0x0);
	53	AliTRDtrackerV1(const TFile in, AliTRDrecoParam p);
d9950a5a	54	virtual ~AliTRDtrackerV1();
e4f2f73d	55
e4f2f73d	56	Int_t Clusters2Tracks(AliESDEvent *esd);
d9950a5a	57	static void GetExtrapolationConfig(Int_t iconfig, Int_t planes[2]);
	58	static void GetSeedingConfig(Int_t iconfig, Int_t planes[4]);
	59	Int_t FollowBackProlongation(AliTRDtrackV1 &t);
	60	Int_t FollowProlongation(AliTRDtrackV1 &t);
	61	Int_t PropagateBack(AliESDEvent *event);
	62	Int_t RefitInward(AliESDEvent *event);
e4f2f73d	63	void SetRecoParam(AliTRDrecoParam *p){fRecoParam = p;}
d9950a5a	64	void UnloadClusters();
e4f2f73d	65
d9950a5a	66	protected:
e4f2f73d	67	Double_t BuildSeedingConfigs(AliTRDstackLayer layer, Int_t configs);
d9950a5a	68	Int_t Clusters2TracksSM( AliTRDtracker::AliTRDtrackingSector sector, AliESDEvent esd);
e4f2f73d	69	Int_t Clusters2TracksStack(AliTRDstackLayer layer, TClonesArray esdTrackList);
d9950a5a	70	void CookLabel(AliKalmanTrack *pt, Float_t wrong) const;
e4f2f73d	71	Int_t GetSeedingLayers(AliTRDstackLayer layers, Double_t params);
e4f2f73d	72	void GetMeanCLStack(AliTRDstackLayer layers, Int_t planes, Double_t *params);
d9950a5a	73	AliTRDseedV1* GetTracklet(AliTRDtrackV1 *trk, Int_t plane, Int_t &idx);
d9950a5a	74	virtual Bool_t GetTrackPoint(Int_t index, AliTrackPoint &p) const;
e4f2f73d	75	AliTRDcluster FindSeedingCluster(AliTRDstackLayer layers, AliTRDseedV1/AliRieman/ *sfit);
d9950a5a	76
d9950a5a	77	Double_t MakeSeedingPlanes(AliTRDstackLayer *layer);
e4f2f73d	78	AliTRDstackLayer MakeSeedingLayer(AliTRDstackLayer layers, Int_t Plane);
d9950a5a	79	Int_t MakeSeeds(AliTRDstackLayer layers, AliTRDseedV1 sseed, Int_t *ipar);
	80	AliTRDtrackV1* MakeTrack(AliTRDseedV1 seeds, Double_t params);
	81	Int_t SetTracklet(AliTRDseedV1 *tracklet);
e4f2f73d	82
d9950a5a	83	private:
e4f2f73d	84	AliTRDtrackerV1(const AliTRDtrackerV1 &tracker);
e4f2f73d	85	AliTRDtrackerV1 &operator=(const AliTRDtrackerV1 &tracker);
d9950a5a	86	Double_t CookLikelihood(AliTRDseedV1 cseed, Int_t planes[4], Double_t chi2);
d9950a5a	87	void ImproveSeedQuality(AliTRDstackLayer layer, AliTRDseedV1 cseed);
e4f2f73d	88
d9950a5a	89	private:
e4f2f73d	90
d9950a5a	91	static Double_t fgTopologicQA[kNConfigs]; // Topologic quality
e4f2f73d	92	Double_t fTrackQuality[kMaxTracksStack]; // Track quality
	93	Int_t fSeedLayer[kMaxTracksStack]; // Seed layer
	94	Int_t fSieveSeeding; //! Seeding iterator
d9950a5a	95	TClonesArray *fTracklets; // List of tracklets for all sectors
e4f2f73d	96	AliTRDrecoParam *fRecoParam; // Reconstruction parameters
e4f2f73d	97	AliTRDtrackerFitter *fFitter; //! Fitter class of the tracker
e4f2f73d	98
	99	ClassDef(AliTRDtrackerV1, 1) // Stand alone tracker development class
	100
	101	};
	102	#endif