macros_impl.h

/**
 * Copyright (c) 2015-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD+Patents 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 "faiss_c.h"
#include "FaissException.h"
#include "error_impl.h"
#include <stdexcept>
#include <iostream>

#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