- Adding check and flagging for HG present
[u/mrichter/AliRoot.git] / MUON / AliMUONRefitter.h
1 #ifndef ALIMUONREFITTER_H
2 #define ALIMUONREFITTER_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice                               */
6
7 // $Id$
8
9 /// \ingroup rec
10 /// \class AliMUONRefitter
11 /// \brief class to refit the ESD clusters/tracks
12 /// 
13 //  Author Philippe Pillot
14
15 #include <TObject.h>
16
17 class AliMUONGeometryTransformer;
18 class AliMUONVClusterFinder;
19 class AliMUONVClusterServer;
20 class AliMUONVTrackReconstructor;
21 class AliMUONESDInterface;
22 class AliMUONVClusterStore;
23 class AliMUONVTrackStore;
24 class AliMUONTrack;
25 class AliMUONRecoParam;
26
27 class AliMUONRefitter : public TObject
28 {
29 public:
30   
31   AliMUONRefitter(const AliMUONRecoParam* recoParam);
32   virtual ~AliMUONRefitter();
33   
34   /// connect to the ESD interface containing MUON data to refit
35   void Connect(const AliMUONESDInterface* esdInterface) {fkESDInterface = esdInterface;}
36   
37   // re-reconstruct all tracks (clusters) in the ESD event
38   AliMUONVTrackStore* ReconstructFromDigits();
39   AliMUONVTrackStore* ReconstructFromClusters();
40   
41   // refit a particular track in the ESD event
42   AliMUONTrack* RetrackFromDigits(UInt_t trackId);
43   AliMUONTrack* RetrackFromClusters(UInt_t trackId);
44   
45   // re-clusterize a particular cluster in the ESD event
46   AliMUONVClusterStore* ReClusterize(UInt_t trackId, UInt_t clusterId);
47   AliMUONVClusterStore* ReClusterize(UInt_t clusterId);
48   
49   
50 protected:
51   
52   AliMUONRefitter (const AliMUONRefitter&); ///< copy constructor
53   AliMUONRefitter& operator=(const AliMUONRefitter&); ///< assignment operator
54   
55   
56 private:
57   
58   void CreateGeometryTransformer();
59   void CreateClusterServer(AliMUONGeometryTransformer& transformer);
60   
61   AliMUONTrack* RetrackFromDigits(const AliMUONTrack& track);
62   
63   Bool_t AddClusterToTracks(const AliMUONVClusterStore &localClusterStore, AliMUONVTrackStore &trackStore);
64   
65 private:
66     
67   const AliMUONRecoParam*     fkRecoParam;          ///< pointer to reco param (not owner)
68   const AliMUONESDInterface*  fkESDInterface;       ///< container of MUON tracks/clusters/digits (not owner)
69   AliMUONGeometryTransformer* fGeometryTransformer; ///< geometry transformer (owner)
70   AliMUONVClusterServer*      fClusterServer;       ///< clusterizer (owner)
71   AliMUONVTrackReconstructor* fTracker;             ///< tracker (owner)
72   
73   
74   ClassDef(AliMUONRefitter,0)
75 };
76
77 #endif
78