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

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

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
// class for running the reconstruction                                      //
// Clusters and tracks are created for all detectors and all events by       //
// typing:                                                                   //
//                                                                           //
//   AliReconstruction rec;                                                  //
//   rec.Run();                                                              //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////


#include <TNamed.h>
#include <TString.h>
#include <TObjArray.h>

class AliReconstructor;
class AliRunLoader;
class AliRawReader;
class AliLoader;
class AliVertexer;
class AliTracker;
class AliESD;
class TFile;


class AliReconstruction: public TNamed {
public:
  AliReconstruction(const char* gAliceFilename = "galice.root",
		    const char* name = "AliReconstruction", 
		    const char* title = "reconstruction");
  AliReconstruction(const AliReconstruction& rec);
  AliReconstruction& operator = (const AliReconstruction& rec);
  virtual ~AliReconstruction();

  void           SetGAliceFile(const char* fileName);
  void           SetInput(const char* input) {fInput = input;};
  void           SetOption(const char* detector, const char* option);

  void           SetRunLocalReconstruction(const char* detectors) {
    fRunLocalReconstruction = detectors;};
  void           SetRunVertexFinder(Bool_t run) {fRunVertexFinder = run;};
  void           SetRunTracking(const char* detectors) {
    fRunTracking = detectors;};
  void           SetFillESD(const char* detectors) {fFillESD = detectors;};
  void           SetRunReconstruction(const char* detectors) {
    SetRunLocalReconstruction(detectors); 
    SetRunTracking(detectors);
    SetFillESD(detectors);};

  void           SetStopOnError(Bool_t stopOnError) 
    {fStopOnError = stopOnError;}
  void           SetCheckPointLevel(Int_t checkPointLevel)
    {fCheckPointLevel = checkPointLevel;}

  virtual Bool_t Run(const char* input = NULL);

private:
  Bool_t         RunLocalReconstruction(const TString& detectors);
  Bool_t         RunVertexFinder(AliESD*& esd);
  Bool_t         RunTracking(AliESD*& esd);
  Bool_t         FillESD(AliESD*& esd, const TString& detectors);

  Bool_t         IsSelected(TString detName, TString& detectors) const;
  AliReconstructor* GetReconstructor(Int_t iDet);
  Bool_t         CreateVertexer();
  Bool_t         CreateTrackers(const TString& detectors);
  void           CleanUp(TFile* file = NULL);

  Bool_t         ReadESD(AliESD*& esd, const char* recStep) const;
  void           WriteESD(AliESD* esd, const char* recStep) const;

  TString        fRunLocalReconstruction; // run the local reconstruction for these detectors
  Bool_t         fRunVertexFinder;    // run the vertex finder
  TString        fRunTracking;        // run the tracking for these detectors
  TString        fFillESD;            // fill ESD for these detectors
  TString        fGAliceFileName;     // name of the galice file
  TString        fInput;              // name of input file or directory
  Bool_t         fStopOnError;        // stop or continue on errors
  Int_t          fCheckPointLevel;    // level of ESD check points
  TObjArray      fOptions;            // options for reconstructor objects

  AliRunLoader*  fRunLoader;          //! current run loader object
  AliRawReader*  fRawReader;          //! current raw data reader

  static const Int_t fgkNDetectors = 15;   //! number of detectors
  static const char* fgkDetectorName[fgkNDetectors]; //! names of detectors
  AliReconstructor*  fReconstructor[fgkNDetectors];  //! array of reconstructor objects
  AliLoader*     fLoader[fgkNDetectors];   //! detector loaders
  AliVertexer*   fVertexer;                //! vertexer for ITS
  AliTracker*    fTracker[fgkNDetectors];  //! trackers

  ClassDef(AliReconstruction, 3)      // class for running the reconstruction
};

#endif
Commit	Line	Data
596a855f	1	#ifndef ALIRECONSTRUCTION_H
	2	#define ALIRECONSTRUCTION_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
af7ba10c	8	///////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// class for running the reconstruction //
	11	// Clusters and tracks are created for all detectors and all events by //
	12	// typing: //
	13	// //
	14	// AliReconstruction rec; //
	15	// rec.Run(); //
	16	// //
	17	///////////////////////////////////////////////////////////////////////////////
	18
	19
596a855f	20	#include <TNamed.h>
596a855f	21	#include <TString.h>
59697224	22	#include <TObjArray.h>
596a855f	23
b8cd5251	24	class AliReconstructor;
596a855f	25	class AliRunLoader;
b649205a	26	class AliRawReader;
596a855f	27	class AliLoader;
2257f27e	28	class AliVertexer;
596a855f	29	class AliTracker;
596a855f	30	class AliESD;
e583c30d	31	class TFile;
596a855f	32
	33
	34	class AliReconstruction: public TNamed {
	35	public:
e583c30d	36	AliReconstruction(const char* gAliceFilename = "galice.root",
e583c30d	37	const char* name = "AliReconstruction",
596a855f	38	const char* title = "reconstruction");
	39	AliReconstruction(const AliReconstruction& rec);
	40	AliReconstruction& operator = (const AliReconstruction& rec);
	41	virtual ~AliReconstruction();
	42
	43	void SetGAliceFile(const char* fileName);
b649205a	44	void SetInput(const char* input) {fInput = input;};
efd2085e	45	void SetOption(const char* detector, const char* option);
596a855f	46
59697224	47	void SetRunLocalReconstruction(const char* detectors) {
59697224	48	fRunLocalReconstruction = detectors;};
2257f27e	49	void SetRunVertexFinder(Bool_t run) {fRunVertexFinder = run;};
b8cd5251	50	void SetRunTracking(const char* detectors) {
b8cd5251	51	fRunTracking = detectors;};
596a855f	52	void SetFillESD(const char* detectors) {fFillESD = detectors;};
b8cd5251	53	void SetRunReconstruction(const char* detectors) {
	54	SetRunLocalReconstruction(detectors);
	55	SetRunTracking(detectors);
	56	SetFillESD(detectors);};
596a855f	57
24f7a148	58	void SetStopOnError(Bool_t stopOnError)
	59	{fStopOnError = stopOnError;}
	60	void SetCheckPointLevel(Int_t checkPointLevel)
	61	{fCheckPointLevel = checkPointLevel;}
	62
b649205a	63	virtual Bool_t Run(const char* input = NULL);
596a855f	64
596a855f	65	private:
59697224	66	Bool_t RunLocalReconstruction(const TString& detectors);
2257f27e	67	Bool_t RunVertexFinder(AliESD*& esd);
24f7a148	68	Bool_t RunTracking(AliESD*& esd);
24f7a148	69	Bool_t FillESD(AliESD*& esd, const TString& detectors);
596a855f	70
e583c30d	71	Bool_t IsSelected(TString detName, TString& detectors) const;
b8cd5251	72	AliReconstructor* GetReconstructor(Int_t iDet);
2257f27e	73	Bool_t CreateVertexer();
b8cd5251	74	Bool_t CreateTrackers(const TString& detectors);
e583c30d	75	void CleanUp(TFile* file = NULL);
e583c30d	76
24f7a148	77	Bool_t ReadESD(AliESD& esd, const char recStep) const;
	78	void WriteESD(AliESD* esd, const char* recStep) const;
	79
59697224	80	TString fRunLocalReconstruction; // run the local reconstruction for these detectors
2257f27e	81	Bool_t fRunVertexFinder; // run the vertex finder
b8cd5251	82	TString fRunTracking; // run the tracking for these detectors
596a855f	83	TString fFillESD; // fill ESD for these detectors
596a855f	84	TString fGAliceFileName; // name of the galice file
b649205a	85	TString fInput; // name of input file or directory
e583c30d	86	Bool_t fStopOnError; // stop or continue on errors
24f7a148	87	Int_t fCheckPointLevel; // level of ESD check points
b8cd5251	88	TObjArray fOptions; // options for reconstructor objects
596a855f	89
596a855f	90	AliRunLoader* fRunLoader; //! current run loader object
b649205a	91	AliRawReader* fRawReader; //! current raw data reader
596a855f	92
482070f2	93	static const Int_t fgkNDetectors = 15; //! number of detectors
c757bafd	94	static const char* fgkDetectorName[fgkNDetectors]; //! names of detectors
b8cd5251	95	AliReconstructor* fReconstructor[fgkNDetectors]; //! array of reconstructor objects
	96	AliLoader* fLoader[fgkNDetectors]; //! detector loaders
	97	AliVertexer* fVertexer; //! vertexer for ITS
	98	AliTracker* fTracker[fgkNDetectors]; //! trackers
59697224	99
b8cd5251	100	ClassDef(AliReconstruction, 3) // class for running the reconstruction
596a855f	101	};
	102
	103	#endif