#ifndef CONVOLVE_INCLUDE
#define CONVOLVE_INCLUDE

#include <stdbool.h>
#include <stddef.h>

// Forcibly disable OpenMP support at the src level
#undef _OPENMP

#if defined(_MSC_VER)

#define FORCE_INLINE __forceinline
#define NEVER_INLINE __declspec(noinline)

// Other compilers (including GCC & Clang)
#else

#define FORCE_INLINE inline __attribute__((always_inline))
#define NEVER_INLINE __attribute__((noinline))

#endif

// MSVC implements OpenMP 2.0 which mandates signed integers for its parallel loops
#if defined(_MSC_VER)
typedef signed omp_iter_var;
#else
typedef size_t omp_iter_var;
#endif

// MSVC exports
#if defined(_MSC_VER)
#define LIB_CONVOLVE_EXPORT __declspec(dllexport)
#else
#define LIB_CONVOLVE_EXPORT // nothing
#endif

#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
#define NO_IMPORT_ARRAY
#include "numpy/ndarrayobject.h"
#define DTYPE npy_float64


LIB_CONVOLVE_EXPORT void convolveNd_c(
    DTYPE *const result,
    const DTYPE *const f,
    const unsigned n_dim,
    const size_t *const image_shape,
    const DTYPE *const g,
    const size_t *const kernel_shape,
    const bool nan_interpolate,
    const bool embed_result_within_padded_region,
    const unsigned n_threads
);

// 1D
void convolve1d_c(
    DTYPE *const result,
    const DTYPE *const f,
    const size_t nx,
    const DTYPE *const g,
    const size_t nkx,
    const bool nan_interpolate,
    const bool embed_result_within_padded_region,
    const unsigned n_threads
);
FORCE_INLINE void convolve1d(
    DTYPE *const result,
    const DTYPE *const f,
    const size_t nx,
    const DTYPE *const g,
    const size_t nkx,
    const bool nan_interpolate,
    const bool embed_result_within_padded_region,
    const unsigned n_threads
);

// 2D
void convolve2d_c(
    DTYPE *const result,
    const DTYPE *const f,
    const size_t nx,
    const size_t ny,
    const DTYPE *const g,
    const size_t nkx,
    const size_t nky,
    const bool nan_interpolate,
    const bool embed_result_within_padded_region,
    const unsigned n_threads
);
FORCE_INLINE void convolve2d(
    DTYPE *const result,
    const DTYPE *const f,
    const size_t nx,
    const size_t ny,
    const DTYPE *const g,
    const size_t nkx,
    const size_t nky,
    const bool nan_interpolate,
    const bool embed_result_within_padded_region,
    const unsigned n_threads
);

// 3D
void convolve3d_c(
    DTYPE *const result,
    const DTYPE *const f,
    const size_t nx,
    const size_t ny,
    const size_t nz,
    const DTYPE *const g,
    const size_t nkx,
    const size_t nky,
    const size_t nkz,
    const bool nan_interpolate,
    const bool embed_result_within_padded_region,
    const unsigned n_threads
);
FORCE_INLINE void convolve3d(
    DTYPE *const result,
    const DTYPE *const f,
    const size_t nx,
    const size_t ny,
    const size_t nz,
    const DTYPE *const g,
    const size_t nkx,
    const size_t nky,
    const size_t nkz,
    const bool nan_interpolate,
    const bool embed_result_within_padded_region,
    const unsigned n_threads
);


#endif
