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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/*
* 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++ -*-
// I/O code for indexes
#include "index_io_c.h"
#include <faiss/index_io.h>
#include "macros_impl.h"
using faiss::Index;
using faiss::IndexBinary;
using faiss::IOReader;
using faiss::IOWriter;
using faiss::VectorTransform;
int faiss_write_index(const FaissIndex* idx, FILE* f) {
try {
faiss::write_index(reinterpret_cast<const Index*>(idx), f);
}
CATCH_AND_HANDLE
}
int faiss_write_index_fname(const FaissIndex* idx, const char* fname) {
try {
faiss::write_index(reinterpret_cast<const Index*>(idx), fname);
}
CATCH_AND_HANDLE
}
int faiss_write_index_custom(
const FaissIndex* idx,
FaissIOWriter* io_writer,
int io_flags) {
try {
faiss::write_index(
reinterpret_cast<const Index*>(idx),
reinterpret_cast<IOWriter*>(io_writer),
io_flags);
}
CATCH_AND_HANDLE
}
int faiss_read_index(FILE* f, int io_flags, FaissIndex** p_out) {
try {
auto out = faiss::read_index(f, io_flags);
*p_out = reinterpret_cast<FaissIndex*>(out);
}
CATCH_AND_HANDLE
}
int faiss_read_index_fname(
const char* fname,
int io_flags,
FaissIndex** p_out) {
try {
auto out = faiss::read_index(fname, io_flags);
*p_out = reinterpret_cast<FaissIndex*>(out);
}
CATCH_AND_HANDLE
}
int faiss_read_index_custom(
FaissIOReader* io_reader,
int io_flags,
FaissIndex** p_out) {
try {
auto out = faiss::read_index(
reinterpret_cast<IOReader*>(io_reader), io_flags);
*p_out = reinterpret_cast<FaissIndex*>(out);
}
CATCH_AND_HANDLE
}
int faiss_write_index_binary(const FaissIndexBinary* idx, FILE* f) {
try {
faiss::write_index_binary(reinterpret_cast<const IndexBinary*>(idx), f);
}
CATCH_AND_HANDLE
}
int faiss_write_index_binary_fname(
const FaissIndexBinary* idx,
const char* fname) {
try {
faiss::write_index_binary(
reinterpret_cast<const IndexBinary*>(idx), fname);
}
CATCH_AND_HANDLE
}
int faiss_write_index_binary_custom(
const FaissIndexBinary* idx,
FaissIOWriter* io_writer) {
try {
faiss::write_index_binary(
reinterpret_cast<const IndexBinary*>(idx),
reinterpret_cast<IOWriter*>(io_writer));
}
CATCH_AND_HANDLE
}
int faiss_read_index_binary(FILE* f, int io_flags, FaissIndexBinary** p_out) {
try {
auto out = faiss::read_index_binary(f, io_flags);
*p_out = reinterpret_cast<FaissIndexBinary*>(out);
}
CATCH_AND_HANDLE
}
int faiss_read_index_binary_fname(
const char* fname,
int io_flags,
FaissIndexBinary** p_out) {
try {
auto out = faiss::read_index_binary(fname, io_flags);
*p_out = reinterpret_cast<FaissIndexBinary*>(out);
}
CATCH_AND_HANDLE
}
int faiss_read_index_binary_custom(
FaissIOReader* io_reader,
int io_flags,
FaissIndexBinary** p_out) {
try {
auto out = faiss::read_index_binary(
reinterpret_cast<IOReader*>(io_reader), io_flags);
*p_out = reinterpret_cast<FaissIndexBinary*>(out);
}
CATCH_AND_HANDLE
}
int faiss_read_VectorTransform_fname(
const char* fname,
FaissVectorTransform** p_out) {
try {
auto out = faiss::read_VectorTransform(fname);
*p_out = reinterpret_cast<FaissVectorTransform*>(out);
}
CATCH_AND_HANDLE
}
|
