Adding checkpoints (T.Kuhr)
[u/mrichter/AliRoot.git] / STEER / AliReconstruction.h
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
8 #include <TNamed.h>
9 #include <TString.h>
10
11 class AliRunLoader;
12 class AliLoader;
13 class AliTracker;
14 class AliESD;
15 class TFile;
16
17
18 class AliReconstruction: public TNamed {
19 public:
20   AliReconstruction(const char* gAliceFilename = "galice.root",
21                     const char* name = "AliReconstruction", 
22                     const char* title = "reconstruction");
23   AliReconstruction(const AliReconstruction& rec);
24   AliReconstruction& operator = (const AliReconstruction& rec);
25   virtual ~AliReconstruction();
26
27   void           SetGAliceFile(const char* fileName);
28
29   void           SetRunReconstruction(const char* detectors) {
30     fRunReconstruction = detectors;};
31   void           SetRunTracking(Bool_t run) {fRunTracking = run;};
32   void           SetFillESD(const char* detectors) {fFillESD = detectors;};
33
34   void           SetStopOnError(Bool_t stopOnError) 
35     {fStopOnError = stopOnError;}
36   void           SetCheckPointLevel(Int_t checkPointLevel)
37     {fCheckPointLevel = checkPointLevel;}
38
39   virtual Bool_t Run();
40
41 private:
42   Bool_t         RunReconstruction(const TString& detectors);
43   Bool_t         RunTracking(AliESD*& esd);
44   Bool_t         FillESD(AliESD*& esd, const TString& detectors);
45
46   Bool_t         IsSelected(TString detName, TString& detectors) const;
47   Bool_t         CreateTrackers();
48   void           CleanUp(TFile* file = NULL);
49
50   Bool_t         ReadESD(AliESD*& esd, const char* recStep) const;
51   void           WriteESD(AliESD* esd, const char* recStep) const;
52
53   TString        fRunReconstruction;  // run the reconstr. for these detectors
54   Bool_t         fRunTracking;        // run the barrel tracking
55   TString        fFillESD;            // fill ESD for these detectors
56   TString        fGAliceFileName;     // name of the galice file
57   Bool_t         fStopOnError;        // stop or continue on errors
58   Int_t          fCheckPointLevel;    // level of ESD check points
59
60   AliRunLoader*  fRunLoader;          //! current run loader object
61   AliLoader*     fITSLoader;          //! loader for ITS
62   AliTracker*    fITSTracker;         //! tracker for ITS
63   AliLoader*     fTPCLoader;          //! loader for TPC
64   AliTracker*    fTPCTracker;         //! tracker for TPC
65   AliLoader*     fTRDLoader;          //! loader for TRD
66   AliTracker*    fTRDTracker;         //! tracker for TRD
67   AliLoader*     fTOFLoader;          //! loader for TOF
68   AliTracker*    fTOFTracker;         //! tracker for TOF
69
70   ClassDef(AliReconstruction, 1)      // class for running the reconstruction
71 };
72
73 #endif