X-Git-Url: http://git.nguyen.vg/gitweb/?a=blobdiff_plain;f=incbwt%2Frlcsa_builder.h;fp=incbwt%2Frlcsa_builder.h;h=0b0b6b277459642592238c8726447f5e367469ea;hb=40ddf9aca842bdc081b6350a4ebfe36b066c94c9;hp=0000000000000000000000000000000000000000;hpb=af8938dbee21244687184dd0502a84ce1af70c50;p=SXSI%2FTextCollection.git

diff --git a/incbwt/rlcsa_builder.h b/incbwt/rlcsa_builder.h
new file mode 100644
index 0000000..0b0b6b2
--- /dev/null
+++ b/incbwt/rlcsa_builder.h
@@ -0,0 +1,58 @@
+#ifndef RLCSA_BUILDER_H
+#define RLCSA_BUILDER_H
+
+#include <cstdlib>
+#include "rlcsa.h"
+
+
+namespace CSA
+{
+
+
+class RLCSABuilder
+{
+  public:
+    RLCSABuilder(usint _block_size, usint _sample_rate, usint _buffer_size);
+    ~RLCSABuilder();
+
+    void insertSequence(char* sequence, usint length, bool delete_sequence);
+
+    // User must free the index. Builder no longer contains it.
+    RLCSA* getRLCSA();
+
+    // User must free the BWT. length becomes the length of BWT.
+    char* getBWT(usint& length);
+
+    bool isOk();
+
+    // These times are not reset with the rest of the builder.
+    double getBuildTime();
+    double getSearchTime();
+    double getMergeTime();
+
+  private:
+    RLCSA* index;
+
+    usint block_size;
+    usint sample_rate;
+    usint buffer_size;
+
+    uchar* buffer;
+    usint chars;
+
+    bool ok;
+
+    clock_t build_time;
+    clock_t search_time;
+    clock_t merge_time;
+
+    void flush();
+    void addRLCSA(RLCSA* increment, uchar* sequence, usint length, bool delete_sequence);
+    void reset();
+};
+
+
+} // namespace CSA
+
+
+#endif // RLCSA_BUILDER_H