Function make_ufuncs

numpy/_core/code_generators/generate_umath.py:1540–1631

(funcdict)

    return "\n".join(code1list), "\n".join(code2list)

def make_ufuncs(funcdict):
    code3list = []
    names = sorted(funcdict.keys())
    for name in names:
        uf = funcdict[name]
        mlist = []
        if uf.signature is None:
            sig = "NULL"
        else:
            sig = f'"{uf.signature}"'
        fmt = textwrap.dedent("""\
            identity = {identity_expr};
            if ({has_identity} && identity == NULL) {{
                return -1;
            }}
            f = PyUFunc_FromFuncAndDataAndSignatureAndIdentity(
                {funcs}, {data}, {signatures}, {nloops},
                {nin}, {nout}, {identity}, "{name}",
                {doc}, 0, {sig}, identity
            );
            if ({has_identity}) {{
                Py_DECREF(identity);
            }}
            if (f == NULL) {{
                return -1;
            }}
        """)
        args = {
            "name": name,
            "funcs": f"{name}_functions" if not uf.empty else "NULL",
            "data": f"{name}_data" if not uf.empty else "NULL",
            "signatures": f"{name}_signatures" if not uf.empty else "NULL",
            "nloops": len(uf.type_descriptions),
            "nin": uf.nin, "nout": uf.nout,
            "has_identity": '0' if uf.identity is None_ else '1',
            "identity": 'PyUFunc_IdentityValue',
            "identity_expr": uf.identity,
            "doc": uf.docstring,
            "sig": sig,
        }

        # Only PyUFunc_None means don't reorder - we pass this using the old
        # argument
        if uf.identity is None_:
            args['identity'] = 'PyUFunc_None'
            args['identity_expr'] = 'NULL'

        mlist.append(fmt.format(**args))
        if uf.typereso is not None:
            mlist.append(rf"((PyUFuncObject *)f)->type_resolver = &{uf.typereso};")
        for c in uf.indexed:
            # Handle indexed loops by getting the underlying ArrayMethodObject
            # from the dict in f._loops and setting its field appropriately
            fmt = textwrap.dedent("""
            {{
                PyArray_DTypeMeta *dtype = PyArray_DTypeFromTypeNum({typenum});
                PyObject *info = get_info_no_cast((PyUFuncObject *)f,
                                                   dtype, {count});