structinfo_config.py

"""
Configuration file to structinfo_generator.
"""
from distutils.sysconfig import get_python_lib

site_packages = get_python_lib()
lib_dirs = [f"{site_packages}/{i}" for i in ["ndtypes", "gumath", "xnd"]]
include_dir = site_packages[: site_packages.find("lib")] + "include"

output_filename = "xnd_structinfo.c"
modulename = "xnd_structinfo"

include_dirs = lib_dirs + [include_dir]
library_dirs = [site_packages[: site_packages.find("/python")]]
includes = ["pyndtypes.h", "pyxnd.h", "pygumath.h"]
libraries = []

# Additional C code to be added to output
c_source = """
extern PyObject* ndt_from_type(ndt_t* val) { return Ndt_FromType(val); }
extern PyObject* xnd_from_type_xnd(PyTypeObject* t, xnd_t* val) { return Xnd_FromXnd(t, val); };
extern PyObject* xnd_view_move_ndt(const PyObject *v, ndt_t *t) { return Xnd_ViewMoveNdt(v, t); };
extern PyObject* xnd_from_xnd_view(xnd_view_t *v) {return Xnd_FromXndView(v); };
//extern PyObject* xnd_from_xndonly(xnd_t *val) { return Xnd_FromXndOnly(val); };
extern void print_bytes(const void *object, size_t size)
{
    // This is for C++; in C just drop the static_cast<>() and assign.
    const unsigned char *const bytes = object;
    size_t i;
    printf("Bytes: [ ");
    for (i = 0; i < size; i++)
    {
        printf("%lld ", (int64_t)bytes[i]);
    }
    printf("]\\n");
    const int64_t *bytes2 = object;
    size_t i2;
    printf("Ints: [ ");
    for (i2 = 0; i2 < (size / 8); i2++)
    {
        printf("%lld ", bytes2[i2]);
    }
    printf("]\\n\\n");
}
"""

# Additional Python C/API code to be added to output
pycapi_source = """
"""

# Additional lines to be included in PyMethodDef struct initialization
PyMethodDef_items = """
"""

# Additional C code to be inserted to module initialization function
PyInit_source = """
  import_ndtypes();
  import_xnd();
  import_gumath();
"""