mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
7a289d5418
Summary: This supersedes https://github.com/facebookresearch/faiss/issues/1390 and https://github.com/facebookresearch/faiss/issues/1141. Pull Request resolved: https://github.com/facebookresearch/faiss/pull/1678 Reviewed By: mdouze Differential Revision: D26428986 Pulled By: beauby fbshipit-source-id: 1e2718f457c1aca18c19edc2436d6baf9b627a60
111 lines
4.7 KiB
C++
111 lines
4.7 KiB
C++
/**
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
// -*- c++ -*-
|
|
|
|
/// Utility macros for the C wrapper implementation.
|
|
|
|
#ifndef MACROS_IMPL_H
|
|
#define MACROS_IMPL_H
|
|
|
|
#include <iostream>
|
|
#include <stdexcept>
|
|
#include <faiss/impl/FaissException.h>
|
|
#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))); \
|
|
}
|
|
|
|
#endif
|