o TdO @s2dZddlZGdddejZddZddZdS) zR This module provides helper functions to find the first line of a function body. Ncs8eZdZdZddZfddZdejfddZZ S) FindDefFirstLinez Attributes ---------- first_stmt_line : int or None This stores the first statement line number if the definition is found. Or, ``None`` if the definition is not found. cCs|j|_|j|_d|_dS)zf Parameters ---------- code : The function's code object. N)co_name_co_nameco_firstlineno_co_firstlinenofirst_stmt_line)selfcoder m/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/misc/firstlinefinder.py__init__s zFindDefFirstLine.__init__cs t|D]}t|qdS)N)astiter_child_nodessupervisit)rnodechild __class__r r _visit_childrensz FindDefFirstLine._visit_childrenrcCs||j|jkr7t|jg}|jr|jd}||j|j|vr7|jr6|jd}t|r0|jd}|j|_ dS | |dS)Nr) namersetlinenodecorator_listaddrbody _is_docstringrr)rrpossible_start_lines first_decor first_stmtr r r visit_FunctionDef!s       z"FindDefFirstLine.visit_FunctionDef) __name__ __module__ __qualname____doc__r rr FunctionDefr! __classcell__r r rr r s  rcCs0t|tjrt|jtjrt|jjtrdSdS)NTF) isinstancer ExprvalueConstantstr)rr r r r?s  rc Csx|j}zt|j }|}Wdn1swYWn ttfy*YdSwt|}t|}| ||j S)a  Look up the first line of function body using the file in ``pyfunc.__code__.co_filename``. Returns ------- lineno : int; or None The first line number of the function body; or ``None`` if the first line cannot be determined. N) __code__open co_filenamereadFileNotFoundErrorOSErrorr parserrr)pyfunccofin file_contenttreefinderr r r get_func_body_first_linenoGs     r:)r%r NodeVisitorrrr:r r r r s 6