- AliMUONAlignmentTask and AddTaskMuonAlignment: New AliAnalysisTask to run the muon

[u/mrichter/AliRoot.git] / MUON / AliMUONReAlignTask.h

diff --git a/MUON/AliMUONReAlignTask.h b/MUON/AliMUONReAlignTask.h
new file mode 100644 (file)
index 0000000..481d530
--- /dev/null
+++ b/MUON/AliMUONReAlignTask.h
@@ -0,0 +1,65 @@
+#ifndef ALIMUONREALIGNTASK_H
+#define ALIMUONREALIGNTASK_H
+
+/// \ingroup ""
+/// \class AliMUONReAlignTask
+/// \brief Task to refit ESD tracks with relaigned geometry
+///
+//  Author Javier Castillo, CEA/Saclay - Irfu/SPhN
+
+class TTree;
+class TString;
+class AliESDEvent;
+class AliMUONClusterInfo;
+class AliMUONESDInterface;
+class AliMUONRefitter;
+class AliMUONRecoParam;
+class AliMUONGeoemetryTransformer;
+class AliMUONVStore;
+
+#include "AliAnalysisTask.h"
+
+class AliMUONReAlignTask : public AliAnalysisTask {
+ public:
+  AliMUONReAlignTask(const char *name = "AliMUONReAlignTask", const char *geofilename = "geometry.root", const char *defaultocdb = "local://$ALICE_ROOT/OCDB", const char *misalignocdb = "local://ReAlignOCDB");
+  AliMUONReAlignTask(const AliMUONReAlignTask& obj);
+  AliMUONReAlignTask& operator=(const AliMUONReAlignTask& other); 
+  virtual ~AliMUONReAlignTask();
+  
+  virtual void   LocalInit();
+  virtual void   ConnectInputData(Option_t *);
+  virtual void   CreateOutputObjects();
+  virtual void   Exec(Option_t *option);
+  virtual void   Terminate(const Option_t*);
+
+  /// Set geoemetry file name
+  void SetGeoFilename(const char* geoFilename) {fGeoFilename = geoFilename;}
+  /// Set mis align ocdb
+  void SetMisAlignOCDB(const char* misalignOCDB) {fMisAlignOCDB = misalignOCDB;}
+  /// Set default ocdb
+  void SetDefaultOCDB(const char* defaultOCDB) {fDefaultOCDB = defaultOCDB;}
+  void Prepare(const char* geoFilename, const char* defaultOCDB, const char* misalignOCDB);
+  UInt_t BuildClusterMap(AliMUONTrack &track);
+
+ private:
+  AliESDEvent *fESD;                  ///< ESD object
+  TTree *fClusterInfoTree;            ///< ClusterInfo tree 
+  AliMUONClusterInfo *fClusterInfo;   ///< ClusterInfo object
+  AliMUONESDInterface *fESDInterface; //!< MUONESDInterface
+  AliMUONRefitter  *fRefitter;        //!< The refitter class 
+  AliMUONRecoParam *fRecoParam;       //!< Parameters for reconstruction
+  TString fGeoFilename;               ///< Geometry file name
+  TString fMisAlignOCDB;               ///< OCDB with misalignment file
+  TString fDefaultOCDB;               ///< Default OCDB
+  AliMUONGeometryTransformer *fGeoTransformer;    //!< Original geometry
+  AliMUONGeometryTransformer *fNewGeoTransformer; //!< Aligned geometry
+  AliMUONVStore *fGainStore; ///< Store for gains
+  AliMUONVStore *fPedStore;  ///< Store for pedestals
+  Int_t fPrintLevel;         //!< Print information
+  Int_t fLastRun;            //!< Last run number
+
+  ClassDef(AliMUONReAlignTask, 1); // example of analysis
+};
+
+#endif
+