feat: Add index file loading to Sketch constructor

waveygang · Oct 12, 2024 · f11f911 · f11f911
1 parent 6fe2be0
commit f11f911
Showing 1 changed file with 23 additions and 2 deletions.
diff --git a/src/map/include/winSketch.hpp b/src/map/include/winSketch.hpp
@@ -135,13 +135,34 @@ namespace skch
        */
       Sketch(skch::Parameters p,
              SequenceIdManager& idMgr,
-             const std::vector<std::string>& targets = {})
+             const std::vector<std::string>& targets = {},
+             const std::string& indexFilename = "")
         : param(std::move(p)),
           idManager(idMgr)
       {
-        initialize(targets);
+        if (!indexFilename.empty()) {
+          loadIndex(indexFilename);
+        } else {
+          initialize(targets);
+        }
+      }
+
+      void loadIndex(const std::string& indexFilename) {
+        std::ifstream inStream(indexFilename, std::ios::binary);
+        if (!inStream) {
+          std::cerr << "Error: Unable to open index file: " << indexFilename << std::endl;
+          exit(1);
+        }
+        readParameters(inStream);
+        readSketchBinary(inStream);
+        readPosListBinary(inStream);
+        readFreqKmersBinary(inStream);
+        inStream.close();
+        isInitialized = true;
+        std::cerr << "[mashmap::skch::Sketch] Sketch loaded from index file: " << indexFilename << std::endl;
       }
 
+    public:
       void initialize(const std::vector<std::string>& targets = {}) {
         std::cerr << "[mashmap::skch::Sketch] Initializing Sketch..." << std::endl;