Initial commit

packages/leann-backend-diskann/third_party/DiskANN/include/memory_mapper.h

@@ -0,0 +1,43 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT license.
+
+#pragma once
+
+#ifndef _WINDOWS
+#include <fcntl.h>
+#include <sys/mman.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <unistd.h>
+
+#else
+#include <Windows.h>
+#endif
+#include <string>
+
+namespace diskann
+{
+class MemoryMapper
+{
+  private:
+#ifndef _WINDOWS
+    int _fd;
+#else
+    HANDLE _bareFile;
+    HANDLE _fd;
+
+#endif
+    char *_buf;
+    size_t _fileSize;
+    const char *_fileName;
+
+  public:
+    MemoryMapper(const char *filename);
+    MemoryMapper(const std::string &filename);
+
+    char *getBuf();
+    size_t getFileSize();
+
+    ~MemoryMapper();
+};
+} // namespace diskann