Cleaning up

[SXSI/TextCollection.git] / TCImplementation.h

diff --git a/TCImplementation.h b/TCImplementation.h
index 75f1d40..e8e8c4b 100644 (file)
--- a/TCImplementation.h
+++ b/TCImplementation.h
@@ -20,6 +20,9 @@
 
 #ifndef _TCImplementation_H_
 #define _TCImplementation_H_
+
+#include "incbwt/bits/deltavector.h"
+
 #include "BitRank.h"
 #include "TextCollection.h"
 #include "BlockArray.h"
@@ -41,8 +44,11 @@
 #undef bitset
 #undef bitget
 
+
 #include "TextStorage.h"
+#include "ArrayDoc.h"
 #include <set>
+#include <string>
 
 namespace SXSI 
 {
@@ -54,7 +60,8 @@ namespace SXSI
  */
 class TCImplementation : public SXSI::TextCollection {
 public:
-    TCImplementation(uchar *, ulong, unsigned, unsigned, ulong, ulong, char);
+    TCImplementation(uchar *, ulong, unsigned, unsigned, ulong, ulong, 
+                     CSA::DeltaVector &, const std::string &, char);
     ~TCImplementation();
 
     bool EmptyText(DocId) const;
@@ -157,7 +164,8 @@ private:
     ulong maxTextLength;
 
     // Array of document id's in the order of end-markers in BWT
-    static_sequence *Doc;
+//    static_sequence *Doc;
+    ArrayDoc *Doc;
 
     // Text storage for fast extraction
     TextStorage * textStorage;
@@ -165,7 +173,7 @@ private:
     // Following methods are not part of the public API
     uchar * BWT(uchar *);
     void makewavelet(uchar *);
-    void maketables(ulong, char);
+    void maketables(ulong, char, CSA::DeltaVector &, const string &);
     DocId DocIdAtTextPos(BlockArray*, TextPosition) const;
     ulong Search(uchar const *, TextPosition, TextPosition *, TextPosition *) const;
     ulong Search(uchar const *, TextPosition, TextPosition *, TextPosition *, DocId, DocId) const;