from bento.commands import hooks

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

    def builder(extension):
        return default_builder(extension,
                               use="superlu_src FLAPACK")
    context.register_builder("_superlu", builder)

    def st_builder(extension):
        return default_builder(extension,
                               features="c bento cstlib",
                               defines=["USE_VENDOR_BLAS=2"])
    context.register_compiled_library_builder("superlu_src", st_builder)
