[u/mrichter/AliRoot.git] / STEER / AliAlignmentTracks.h

#ifndef ALIALIGNMENTTRACKS_H
#define ALIALIGNMENTTRACKS_H

//*************************************************************************
// AliAlignmentTracks: main steering class which deals with the alignment *
// procedures based on reconstructed tracks.                              *
// More comments will come with the development of the interfaces and     *
// functionalities of the class.                                          *
//*************************************************************************

#include <TObject.h>

#include "AliAlignObj.h"

class TChain;
class AliTrackPointArray;
class AliAlignObj;
class AliTrackFitter;
class AliTrackResiduals;

class AliAlignmentTracks : public TObject {

 public:

  AliAlignmentTracks();
  AliAlignmentTracks(TChain *esdchain);
  AliAlignmentTracks(const char *esdfilename, const char *esdtreename = "esdTree");
  AliAlignmentTracks(const AliAlignmentTracks & alignment);
  AliAlignmentTracks& operator= (const AliAlignmentTracks& alignment);
  virtual ~AliAlignmentTracks();

  void AddESD(TChain *esdchain);
  void AddESD(const char *esdfilename, const char *esdtreename = "esdTree");

  void SetPointsFilename(const char *pointsfilename = "AliTrackPoints.root") { fPointsFilename = pointsfilename; }

  void ProcessESD(TSelector *selector);
  void ProcessESD();

  void BuildIndex();
/*   void BuildIndexLayer(AliAlignObj::ELayerID layer); */
/*   void BuildIndexVolume(UShort_t volid); */

  Bool_t ReadAlignObjs(const char *alignobjfilename = "AlignObjs.root");

  void SetTrackFitter(AliTrackFitter *fitter) { fTrackFitter = fitter; }
  void SetMinimizer(AliTrackResiduals *minimizer) { fMinimizer = minimizer; }

  void Align(Int_t iterations = 100);
  void AlignLayer(AliAlignObj::ELayerID layer,
		  AliAlignObj::ELayerID layerRangeMin = AliAlignObj::kFirstLayer,
		  AliAlignObj::ELayerID layerRangeMax = AliAlignObj::kLastLayer,
		  Int_t iterations = 1);
  void AlignVolume(UShort_t volid,
		   AliAlignObj::ELayerID layerRangeMin = AliAlignObj::kFirstLayer,
		   AliAlignObj::ELayerID layerRangeMax = AliAlignObj::kLastLayer);

 protected:

  void InitIndex();
  void ResetIndex();
  void DeleteIndex();

  void InitAlignObjs();
  void ResetAlignObjs();
  void DeleteAlignObjs();

  Int_t LoadPoints(UShort_t volid, AliTrackPointArray** &points);
  void  UnloadPoints(Int_t n, AliTrackPointArray **points);

  AliTrackFitter *CreateFitter();
  AliTrackResiduals *CreateMinimizer();

  TChain           *fESDChain;       //! Chain with ESDs
  TString           fPointsFilename; //  Name of the file containing the track point arrays
  TFile            *fPointsFile;     //  File containing the track point arrays
  TTree            *fPointsTree;     //  Tree with the track point arrays
  Int_t           **fLastIndex;      //! Last filled index in volume arrays
  TArrayI        ***fArrayIndex;     //! Volume arrays which contains the tree index
  Bool_t            fIsIndexBuilt;   //  Is points tree index built
  AliAlignObj    ***fAlignObjs;      //  Array with alignment objects
  AliTrackFitter   *fTrackFitter;    //  Pointer to the track fitter
  AliTrackResiduals*fMinimizer;      //  Pointer to track residuals minimizer

  ClassDef(AliAlignmentTracks,1)

};

#endif
Commit	Line	Data
98937d93	1	#ifndef ALIALIGNMENTTRACKS_H
	2	#define ALIALIGNMENTTRACKS_H
	3
	4	//*************************************************************************
	5	// AliAlignmentTracks: main steering class which deals with the alignment *
	6	// procedures based on reconstructed tracks. *
	7	// More comments will come with the development of the interfaces and *
	8	// functionalities of the class. *
	9	//*************************************************************************
	10
	11	#include <TObject.h>
	12
	13	#include "AliAlignObj.h"
	14
	15	class TChain;
	16	class AliTrackPointArray;
	17	class AliAlignObj;
	18	class AliTrackFitter;
	19	class AliTrackResiduals;
	20
	21	class AliAlignmentTracks : public TObject {
	22
	23	public:
	24
	25	AliAlignmentTracks();
	26	AliAlignmentTracks(TChain *esdchain);
	27	AliAlignmentTracks(const char esdfilename, const char esdtreename = "esdTree");
	28	AliAlignmentTracks(const AliAlignmentTracks & alignment);
	29	AliAlignmentTracks& operator= (const AliAlignmentTracks& alignment);
	30	virtual ~AliAlignmentTracks();
	31
	32	void AddESD(TChain *esdchain);
	33	void AddESD(const char esdfilename, const char esdtreename = "esdTree");
	34
	35	void SetPointsFilename(const char *pointsfilename = "AliTrackPoints.root") { fPointsFilename = pointsfilename; }
	36
	37	void ProcessESD(TSelector *selector);
	38	void ProcessESD();
	39
	40	void BuildIndex();
	41	/* void BuildIndexLayer(AliAlignObj::ELayerID layer); */
	42	/* void BuildIndexVolume(UShort_t volid); */
	43
	44	Bool_t ReadAlignObjs(const char *alignobjfilename = "AlignObjs.root");
	45
	46	void SetTrackFitter(AliTrackFitter *fitter) { fTrackFitter = fitter; }
	47	void SetMinimizer(AliTrackResiduals *minimizer) { fMinimizer = minimizer; }
	48
	49	void Align(Int_t iterations = 100);
	50	void AlignLayer(AliAlignObj::ELayerID layer,
	51	AliAlignObj::ELayerID layerRangeMin = AliAlignObj::kFirstLayer,
	52	AliAlignObj::ELayerID layerRangeMax = AliAlignObj::kLastLayer,
	53	Int_t iterations = 1);
	54	void AlignVolume(UShort_t volid,
	55	AliAlignObj::ELayerID layerRangeMin = AliAlignObj::kFirstLayer,
	56	AliAlignObj::ELayerID layerRangeMax = AliAlignObj::kLastLayer);
	57
	58	protected:
	59
	60	void InitIndex();
	61	void ResetIndex();
	62	void DeleteIndex();
	63
	64	void InitAlignObjs();
65	void ResetAlignObjs();
66	void DeleteAlignObjs();
67
68	Int_t LoadPoints(UShort_t volid, AliTrackPointArray** &points);
69	void UnloadPoints(Int_t n, AliTrackPointArray **points);
70
71	AliTrackFitter *CreateFitter();
72	AliTrackResiduals *CreateMinimizer();
73
74	TChain *fESDChain; //! Chain with ESDs
75	TString fPointsFilename; // Name of the file containing the track point arrays
76	TFile *fPointsFile; // File containing the track point arrays
77	TTree *fPointsTree; // Tree with the track point arrays
78	Int_t **fLastIndex; //! Last filled index in volume arrays
79	TArrayI ***fArrayIndex; //! Volume arrays which contains the tree index
80	Bool_t fIsIndexBuilt; // Is points tree index built
81	AliAlignObj ***fAlignObjs; // Array with alignment objects
82	AliTrackFitter *fTrackFitter; // Pointer to the track fitter
83	AliTrackResiduals*fMinimizer; // Pointer to track residuals minimizer
84
85	ClassDef(AliAlignmentTracks,1)
86
87	};
88
89	#endif