from bento.commands import hooks

@hooks.pre_build
def pre_build(context):
    default_builder = context.default_builder

    context.tweak_extension("flapack", features="c fc pyext bento cshlib f2py",
                          use="LAPACK")

    def clapack_builder(extension):
        kw = {"features": "c fc pyext bento cshlib f2py"}
        if context.waf_context.env.HAS_CLAPACK:
            kw["use"] = "CLAPACK"
        else:
            kw["source"] = ["dummy_clapack.pyf"]
        return default_builder(extension, **kw)
    context.register_builder("clapack", clapack_builder)

    context.tweak_extension("calc_lwork",
                          features="c fc pyext bento cshlib f2py f2py_fortran",
                          use="LAPACK CLIB")
