Initial commit

packages/leann-backend-hnsw/third_party/faiss/c_api/macros_impl.h

@@ -0,0 +1,107 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+// -*- c++ -*-
+
+/// Utility macros for the C wrapper implementation.
+
+#ifndef MACROS_IMPL_H
+#define MACROS_IMPL_H
+
+#include <faiss/impl/FaissException.h>
+#include <iostream>
+#include <stdexcept>
+#include "error_impl.h"
+#include "faiss_c.h"
+
+#ifdef NDEBUG
+#define CATCH_AND_HANDLE                                                      \
+    catch (faiss::FaissException & e) {                                       \
+        faiss_last_exception = std::make_exception_ptr(e);                    \
+        return -2;                                                            \
+    }                                                                         \
+    catch (std::exception & e) {                                              \
+        faiss_last_exception = std::make_exception_ptr(e);                    \
+        return -4;                                                            \
+    }                                                                         \
+    catch (...) {                                                             \
+        faiss_last_exception =                                                \
+                std::make_exception_ptr(std::runtime_error("Unknown error")); \
+        return -1;                                                            \
+    }                                                                         \
+    return 0;
+#else
+#define CATCH_AND_HANDLE                                                      \
+    catch (faiss::FaissException & e) {                                       \
+        std::cerr << e.what() << '\n';                                        \
+        faiss_last_exception = std::make_exception_ptr(e);                    \
+        return -2;                                                            \
+    }                                                                         \
+    catch (std::exception & e) {                                              \
+        std::cerr << e.what() << '\n';                                        \
+        faiss_last_exception = std::make_exception_ptr(e);                    \
+        return -4;                                                            \
+    }                                                                         \
+    catch (...) {                                                             \
+        std::cerr << "Unrecognized exception!\n";                             \
+        faiss_last_exception =                                                \
+                std::make_exception_ptr(std::runtime_error("Unknown error")); \
+        return -1;                                                            \
+    }                                                                         \
+    return 0;
+#endif
+
+#define DEFINE_GETTER(clazz, ty, name)                             \
+    ty faiss_##clazz##_##name(const Faiss##clazz* obj) {           \
+        return static_cast<ty>(                                    \
+                reinterpret_cast<const faiss::clazz*>(obj)->name); \
+    }
+
+#define DEFINE_GETTER_SUBCLASS(clazz, parent, ty, name)                    \
+    ty faiss_##clazz##_##name(const Faiss##clazz* obj) {                   \
+        return static_cast<ty>(                                            \
+                reinterpret_cast<const faiss::parent::clazz*>(obj)->name); \
+    }
+
+#define DEFINE_GETTER_PERMISSIVE(clazz, ty, name)                      \
+    ty faiss_##clazz##_##name(const Faiss##clazz* obj) {               \
+        return (ty)(reinterpret_cast<const faiss::clazz*>(obj)->name); \
+    }
+
+#define DEFINE_GETTER_SUBCLASS_PERMISSIVE(clazz, parent, ty, name)             \
+    ty faiss_##clazz##_##name(const Faiss##clazz* obj) {                       \
+        return (ty)(reinterpret_cast<const faiss::parent::clazz*>(obj)->name); \
+    }
+
+#define DEFINE_SETTER(clazz, ty, name)                           \
+    void faiss_##clazz##_set_##name(Faiss##clazz* obj, ty val) { \
+        reinterpret_cast<faiss::clazz*>(obj)->name = val;        \
+    }
+
+#define DEFINE_SETTER_STATIC(clazz, ty_to, ty_from, name)                     \
+    void faiss_##clazz##_set_##name(Faiss##clazz* obj, ty_from val) {         \
+        reinterpret_cast<faiss::clazz*>(obj)->name = static_cast<ty_to>(val); \
+    }
+
+#define DEFINE_DESTRUCTOR(clazz)                     \
+    void faiss_##clazz##_free(Faiss##clazz* obj) {   \
+        delete reinterpret_cast<faiss::clazz*>(obj); \
+    }
+
+#define DEFINE_INDEX_DOWNCAST(clazz)                                        \
+    Faiss##clazz* faiss_##clazz##_cast(FaissIndex* index) {                 \
+        return reinterpret_cast<Faiss##clazz*>(dynamic_cast<faiss::clazz*>( \
+                reinterpret_cast<faiss::Index*>(index)));                   \
+    }
+
+#define DEFINE_SEARCH_PARAMETERS_DOWNCAST(clazz)                            \
+    Faiss##clazz* faiss_##clazz##_cast(FaissSearchParameters* sp) {         \
+        return reinterpret_cast<Faiss##clazz*>(dynamic_cast<faiss::clazz*>( \
+                reinterpret_cast<faiss::SearchParameters*>(sp)));           \
+    }
+
+#endif