1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*
* 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_INDEX_BINARY_DOWNCAST(clazz) \
Faiss##clazz* faiss_##clazz##_cast(FaissIndexBinary* index) { \
return reinterpret_cast<Faiss##clazz*>(dynamic_cast<faiss::clazz*>( \
reinterpret_cast<faiss::IndexBinary*>(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
|
