c_build.py

#!/usr/bin/env python3

"""The rules of compilation for setuptools."""

import os

from setuptools import Extension
from setuptools.command.build_py import build_py as _build_py

import numpy as np


COMP_RULES = {
    "define_macros": [("NPY_NO_DEPRECATED_API", "NPY_1_7_API_VERSION")],  # for warning
    "extra_compile_args": [
        "-fopenmp",  # for threads
        "-fopenmp-simd",  # for single instruction multiple data
        "-lm",  # for math functions
        "-march=native",  # uses local processor instructions for optimization
        "-O3",  # hight optimization, -O3 include -ffast-math
        # "-Wall", "-Wextra",  # "-Wconversion",  # -Wtraditional,  # activate warnings,
        # "-pedantic", # ensurse all is standard and compilable anywhere
        "-std=c99",  # use iso c norm
    ],
    "include_dirs": [
        np.get_include(),  # requires for  #include <numpy/arrayobject.h>
        os.path.dirname(__file__),  # for  #include "laueimproc/c_check.h"
    ],
}


class Build(_build_py):
    """Builder to compile c files."""

    def run(self):
        self.run_command("build_ext")  # catch compilation failed ici!
        return super().run()

    def initialize_options(self):
        super().initialize_options()
        if self.distribution.ext_modules is None:
            self.distribution.ext_modules = []
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.geometry.c_metric",
                sources=["laueimproc/geometry/c_metric.c"],
                optional=True,
                **COMP_RULES,
            )
        )
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.ml.c_dist",
                sources=["laueimproc/ml/c_dist.c"],
                optional=True,
                **COMP_RULES,
            )
        )
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.improc.c_find_bboxes",
                sources=["laueimproc/improc/c_find_bboxes.c"],
                optional=True,
                **COMP_RULES,
            )
        )
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.gmm.c_fit",
                sources=["laueimproc/gmm/c_fit.c"],
                optional=True,
                **COMP_RULES,
            )
        )
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.gmm.c_gmm",
                sources=["laueimproc/gmm/c_gmm.c"],
                optional=True,
                **COMP_RULES,
            )
        )
        # self.distribution.ext_modules.append(
        #     Extension(
        #         "laueimproc.improc.c_morpho",
        #         sources=["laueimproc/improc/c_morpho.c"],
        #         optional=True,
        #         **COMP_RULES,
        #     )
        # )
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.improc.spot.c_basic",
                sources=["laueimproc/improc/spot/c_basic.c"],
                optional=True,
                **COMP_RULES,
            )
        )
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.improc.spot.c_extrema",
                sources=["laueimproc/improc/spot/c_extrema.c"],
                optional=True,
                **COMP_RULES,
            )
        )
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.improc.spot.c_pca",
                sources=["laueimproc/improc/spot/c_pca.c"],
                optional=True,
                **COMP_RULES,
            )
        )
        self.distribution.ext_modules.append(
            Extension(
                "laueimproc.opti.c_rois",
                sources=["laueimproc/opti/c_rois.c"],
                optional=True,
                **COMP_RULES,
            )
        )