Bug correction

[u/mrichter/AliRoot.git] / HBTAN / AliHBTReaderITSv2.h

diff --git a/HBTAN/AliHBTReaderITSv2.h b/HBTAN/AliHBTReaderITSv2.h
index f24b82043b5797f35a8aa0af87cce88e335d2888..e52a0e59ff08d3e13bdf43be21a1002919fafbbd 100644 (file)
--- a/HBTAN/AliHBTReaderITSv2.h
+++ b/HBTAN/AliHBTReaderITSv2.h
@@ -3,23 +3,44 @@
 
 #include "AliHBTReader.h"
 
-#include <TString.h>
 
+class AliLoader;
+class AliRunLoader;
+class TString;
 
-class AliHBTReaderITSv2: public AliHBTReaderTPC
+class AliHBTReaderITSv2: public AliHBTReader
 {
-  public:    
-    AliHBTReaderITSv2(const Char_t* trackfilename = "AliITStracksV2.root",
-                    const Char_t* clusterfilename = "AliITSclustersV2.root",
-       const Char_t* goodtracksfilename = "good_tracks_its",
-       const Char_t* galicefilename = "");
-       
+  public:
+    
+    AliHBTReaderITSv2();
+    AliHBTReaderITSv2(const Char_t* galicefilename);
+    AliHBTReaderITSv2(TObjArray* dirs, const Char_t* galicefilename = "galice.root");
+
     virtual ~AliHBTReaderITSv2();
     
-    Int_t Read(AliHBTRun* particles, AliHBTRun *tracks);//reads tracks and particles and puts them in runs
+    void          Rewind();
     
-  protected:    
-  private:
-  public:
-    ClassDef(AliHBTReaderITSv2,1)
+    Bool_t        ReadsTracks() const {return kTRUE;}
+    Bool_t        ReadsParticles() const {return kTRUE;}
+    
+    void          SetMagneticField(Float_t mf){fMagneticField=mf;}
+    void          UseMagneticFieldFromRun(Bool_t flag = kTRUE){fUseMagFFromRun=flag;}
+    
+  protected:
+    
+    Int_t         ReadNext();//reads tracks and particles and puts them in runs
+    Int_t         OpenNextFile();
+    void          DoOpenError( const char *va_(fmt), ...);
+        
+    TString       fFileName;//name of the file with galice.root
+    AliRunLoader* fRunLoader;//!Run Loader
+    AliLoader*    fITSLoader;//! ITS Loader
+        
+    Float_t       fMagneticField;//magnetic field value that was enforced while reading
+    Bool_t        fUseMagFFromRun;//flag indicating if using field specified in gAlice (kTRUE)
+                               // or enforece other defined by fMagneticField
+    
+    ClassDef(AliHBTReaderITSv2,2)
 };
+
+#endif