*jUdZddlmZddlZddlZddlmZddlZddl m Z ddl m Z erddl mZdad ed <dd Zddd Zddd ZddZeZ ddZy)z%Common utilities for Numba operations) annotationsN) TYPE_CHECKING)import_optional_dependency)NumbaUtilError)CallableFboolGLOBAL_USE_NUMBAc(|dk(xs |duxrtS)z%Signal whether to use numba routines.numbaN)r )engines 9C:\FundamentalWorkbench\vendor\pandas\core\util\numba_.pymaybe_use_numbars W  E4!D4DEc"|r td|ay)Nr )rr )enables r set_use_numbars "7+rc|i}|jdd}|jdd}|jdd}|||dS)aP Return arguments to pass to numba.JIT, falling back on pandas default JIT settings. Parameters ---------- engine_kwargs : dict, default None user passed keyword arguments for numba.JIT Returns ------- dict[str, bool] nopython, nogil, parallel Raises ------ NumbaUtilError nopythonTnogilFparallel)rrr)get) engine_kwargsrrrs r get_jit_argumentsr sR$   T2H   gu -E  U3H 5h GGrctrddl}n td}|jj |r|}|St t |jd|ust|tjr|}|S|jj|}|S)a If user function is not jitted already, mark the user's function as jitable. Parameters ---------- func : function user defined function Returns ------- function Numba JITed function, or function marked as JITable by numba rNr F) rr r extending is_jittedgetattrnp__name__ isinstancetypesBuiltinFunctionTyperegister_jitable)funcr numba_funcs r jit_user_functionr&;s*73   &   T]]E *d 2j e''7   __55d; rc|s||fStj|}|jgtg|z|i|}|j |j }|j }|r td||d}||fS)a Prepare arguments for jitted function. As numba functions do not support kwargs, we try to move kwargs into args if possible. Parameters ---------- func : function User defined function args : tuple User input positional arguments kwargs : dict User input keyword arguments num_required_args : int The number of leading positional arguments we will pass to udf. These are not supplied by the user. e.g. for groupby we require "values", "index" as the first two arguments: `numba_func(group, group_index, *args)`, in this case num_required_args=2. See :func:`pandas.core.groupby.numba_.generate_numba_agg_func` Returns ------- tuple[tuple, dict] args, kwargs znumba does not support keyword-only argumentshttps://github.com/numba/numba/issues/2916, https://github.com/numba/numba/issues/6846N)inspect signaturebind _sentinelapply_defaultsargskwargsr)r$r-r.num_required_argsr) argumentss r prepare_function_argumentsr1as8 V|!!$'I Q .? ?Q$Q&QI  >>D   F  9  !" #D <r)r z str | Nonereturnr)F)rrr2None)N)rzdict[str, bool] | Noner2zdict[str, bool])r$rr2r) r$rr-tupler.dictr/intr2ztuple[tuple, dict])__doc__ __future__rr(r!typingrnumpyrpandas.compat._optionalr pandas.errorsrcollections.abcrr __annotations__rrrr&objectr+r1rr rAs{+" >(($F H6 F H 5 55)-5EH55r