Your IP : 216.73.216.213
Z��Z~� � ' @ s� d d l m Z d d l Z e j d e d e d e d e d e d e d e d
e d e d e d
e d e d e d e d e d e d e d e d e � d d l Z d d l Z d d l Z d d l m Z d d l
m Z d d l m
Z
m Z m Z m Z d d l
m Z d d l
m Z d d l
m Z d d l m Z m Z d d l m Z m Z m Z m Z m Z m Z m Z d d l m Z d d l m! Z! d d! l
m" Z" d d" l
m# Z# d d# l
m$ Z$ d d$ l% m& Z& m' Z' m( Z( d% d& l) m* Z* e j+ d d' k re, Z- n e, e. f Z- d( d) � Z/ d* d+ � Z0 d d, d- � Z1 d. d/ � Z2 Gd0 d1 � d1 e3 � Z4 Gd2 d3 � d3 e3 � Z5 d4 d5 � Z6 e6 Gd6 d7 � d7 e � � Z7 Gd8 d9 � d9 e7 � Z8 Gd: d; � d; e � Z9 Gd< d= � d= e7 � Z: Gd> d? � d? e7 � Z; Gd@ dA � dA e; � Z< GdB dC � dC e7 � Z= GdD dE � dE e= � Z> GdF dG � dG e= � Z? GdH dI � dI e= � Z@ GdJ dK � dK e= � ZA GdL dM � dM e= � ZB GdN dO � dO e= � ZC GdP dQ � dQ e7 � ZD GdR dS � dS e7 � ZE GdT dU � dU e7 � ZF GdV dW � dW eE � ZG GdX dY � dY eE � ZH GdZ d[ � d[ eE � ZI Gd\ d] � d] eE � ZJ Gd^ d_ � d_ eE � ZK Gd` da � da eE � ZL Gdb dc � dc eE � ZM Gdd de � de eE � ZN Gdf dg � dg e; � ZO Gdh di � di e; � ZP Gdj dk � dk eP e9 � ZQ Gdl dm � dm e; � ZR Gdn do � do e; � ZS Gdp dq � dq e; � ZT Gdr ds � ds e; e9 � ZU Gdt du � du eU � ZV Gdv dw � dw e7 � ZW Gdx dy � dy e7 � ZX Gdz d{ � d{ eU � ZY Gd| d} � d} eU � ZZ Gd~ d � d eY � Z[ Gd� d� � d� e[ � Z\ Gd� d� � d� e\ � Z] Gd� d� � d� e\ � Z^ Gd� d� � d� eY � Z_ Gd� d� � d� e; � Z` Gd� d� � d� e; e9 � Za Gd� d� � d� ea � Zb Gd� d� � d� ea � Zc Gd� d� � d� e; � Zd Gd� d� � d� e; � Ze Gd� d� � d� e; � Zf Gd� d� � d� e; � Zg Gd� d� � d� e; � Zh Gd� d� � d� eh � Zi Gd� d� � d� eh � Zj Gd� d� � d� eh � Zk Gd� d� � d� eh � Zl Gd� d� � d� e; � Zm Gd� d� � d� e; � Zn Gd� d� � d� e; � Zo Gd� d� � d� e; � Zp Gd� d� � d� e: � Zq Gd� d� � d� e; � Zr Gd� d� � d� e; � Zs Gd� d� � d� e; � Zt Gd� d� � d� e; � Zu Gd� d� � d� e; � Zv Gd� d� � d� e; � Zw Gd� d� � d� e; � Zx Gd� d� � d� e7 � Zy Gd� d� � d� e; � Zz Gd� d� � d� e; � Z{ Gd� d� � d� e � Z| Gd� d� � d� e| e; � Z} Gd� d� � d� e7 � Z~ Gd� d� � d� e7 � Z d� d� � Z� Gd� d� � d� e| e; � Z� Gd� d� � d� e� � Z� Gd� d� � d� e� � Z� Gd� d� � d� e| e; � Z� Gd� d� � d� e; � Z� Gd� d� � d� eh � Z� Gd� d� � d� e; � Z� Gd� d� � d� e7 � Z� Gd� d� � d� e; � Z� Gd� d� � d� e� � Z� Gd� d� � d� e� � Z� Gd� d� � d� e; � Z� Gd� d� � d� e� � Z� d� d� � Z� d� d� d� � d� d� d� � d� e� i Z� d� d� d� d� f d� d� d� d� f i Z� Gd� d� � d� e; � Z� Gd� d� � d� e; � Z� Gd� d� � d� e; � Z� Gd� d� � d� e7 � Z� Gd� d� � d� e; e� � Z� Gd� d� � d� e� � Z� Gd� d � d e� � Z� Gdd� de; � Z� e# j� r( dZ� n dZ� e j� dd� Z� e j� dd� Z� e j� dd � Z� e j� d
d � Z� e j� d d � Z� e j� d d � Z� e j� d
d � Z� e j� dd � Z� e ddddde� g � Z� d S( � )�absolute_importN�sys�os�copy�Builtin�error�warning�Naming�
PyrexTypes�py_object_type� ModuleScope�
LocalScope� ClosureScope�StructOrUnionScope� PyClassScope�
CppClassScope� UtilityCode�
EncodedString�
error_type�
_py_int_types)�chain� )r )r r �
InternalError� CompileError)r )r
)� TypeSlots)r r )r r
r r r r �
TemplateScope)r )r )�Future)�Options)�
DebugFlags)�has_np_pythran� pythran_type�is_pythran_buffer� )�
add_metaclass� c C s | d j � | d f S)Nr r )Zget_filenametable_entry)�pos� r& �H/opt/alt/python35/lib64/python3.5/site-packages/Cython/Compiler/Nodes.py�relative_position) s r( c C s� t j s
| Sd t | � } | d k r3 t | � S| j } | d k rt y | j | � Wn t k
rs d } Yn X| s� t | � } n t | d | � } | | _ | S)Nz File: %s (starting at line %s)�
)r Zembed_pos_in_docstringr( r �encoding�encode�UnicodeEncodeError)r% Z docstringZpos_liner* �docr& r& r' �embed_position- s
r. c C s d } d } d } } | j r� t | j d � xb | j D]W \ } } | j sQ q9 | j d k ru d } | s� | } q9 | j d k r9 d } | } q9 W| r� | r� t | j d � | j | � } | j rl| j rl| j d k rl| d k r-| d k r-| j
r-| j | � }
|
r-|
j
r-d } d } | t j
t j t j f k r�| j d k rct j n t } n% | d k r�| j r�t | j d � | d k r�| r�| r�| j
r�t | j d
� t | j d | d d �} n) | r�t | j d � n t | j d � | | f S)NFzSDicts should no longer be used as type annotations. Use 'cython.int' etc. directly.�type� typeT�ctype� ctypez9Duplicate type declarations found in signature annotation�int�long�floatzWStrings should no longer be used for type declarations. Use 'cython.int' etc. directly.zOPython type declaration in signature annotation does not refer to a Python type�is_argz'Ambiguous types in annotation, ignoringz0Unknown type declaration in annotation, ignoring)r/ r0 )r1 r2 )r3 r4 r5 )�is_dict_literalr r% �key_value_pairsZis_string_literal�value�analyse_as_type�is_nameZcython_attribute�name� is_pyobjectZ
infer_typer
� c_long_type�
c_int_type� c_float_typeZ
c_double_typer �CAnalysedBaseTypeNode) �
annotation�env�assigned_value� base_typeZ is_ambiguousZexplicit_pytypeZexplicit_ctyper< r9 �arg_typeZ
assigned_typer& r& r' �analyse_type_annotationG sP
"" ! rG c s � � f d d � } | S)Nc s1 t | � d k r t | d � � r | d d � \ } } d d | j | j j � j | j d d � f } | j j j � } | j | � | j j j � } | j d 7_ � | | � } | j d 8_ | | j j j � k r� | j j j
| � n" | j d d d � } | j | � | S� | | � Sd S)Nr r" z( /* %s -> %s.%s %s */� � z->z<-) �len�
isinstanceZ
call_level� __class__�__name__r% �buffer�stream�tell�putln�truncate�replace)�args�kwds�node�code�markerZpristine�start�res)�codewriter_class�funcr& r' �fw s& %
zwrite_func_call.<locals>.fr& )r\ r[ r] r& )r[ r\ r' �write_func_callv s r^ c s" e Z d Z � f d d � Z � S)�VerboseCodeWriterc s� d d l m } d d l m } t | � } x< | j � D]. \ } } t | | � r9 t | | � | | <q9 Wt t | � j
| | | | � S)Nr )� FunctionTyper )� CCodeWriter) �typesr` �Codera �dict�itemsrK r^ �superr_ �__new__)�clsr< �bases�attrsr` ra �mname�m)rL r& r' rg � s zVerboseCodeWriter.__new__)rM �
__module__� __qualname__rg r& r& )rL r' r_ � s r_ c s= e Z d Z d Z e d d d g � Z � f d d � Z � S)�CheckAnalyserszCMetaclass to check that type analysis functions return a node.
�
analyse_types�analyse_expressions�analyse_target_typesc s� d d l m } d d � } t | � } xK | j � D]= \ } } t | | � r5 | | j k r5 | | | � | | <q5 Wt t | � j | | | | � S)Nr )r` c s � � f d d � } | S)Nc s6 � | | � } | d k r2 t d � | | f � | S)Nz%s %s %s)�print)rT �kwargs�retval)r\ r< r& r' �call� s z3CheckAnalysers.__new__.<locals>.check.<locals>.callr& )r< r\ rv r& )r\ r< r' �check� s z%CheckAnalysers.__new__.<locals>.check) rb r` rd re rK �methodsrf ro rg )rh r< ri rj r` rw rk rl )rL r& r' rg � s zCheckAnalysers.__new__)rM rm rn �__doc__�setrx rg r& r& )rL r' ro � s
ro c C s t j r t t � | � S| S)N)r Zdebug_trace_code_generationr# r_ )rh r& r& r' �_with_metaclass� s r{ c @ s e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z d Z d d � Z
d Z d Z d d d � Z d Z d d
� Z d d � Z d
d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d d d d d � Z d d d d � Z d S)!�Noder FNc K s | | _ | j j | � d S)N)r% �__dict__�update)�selfr% �kwr& r& r' �__init__� s z
Node.__init__Z Operationc C s t | j d | j � d S)Nz%s not allowed without gil)r r% � gil_message)r rC r& r& r' � gil_error� s zNode.gil_errorc C s | j � s | j � d S)N)Zis_cpp� cpp_error)r rC r& r& r' � cpp_check� s zNode.cpp_checkc C s t | j d | j � d S)Nz%s only allowed in c++)r r% � cpp_message)r r& r& r' r� � s zNode.cpp_errorc C sb t j | � } xL | j D]A } t | | � } t | t � r t | | d d � | D� � q W| S)ac Clone the node. This is defined as a shallow copy, except for member lists
amongst the child attributes (from get_child_accessors) which are also
copied. Lists containing child nodes are thus seen as a way for the node
to hold multiple children directly; the list is not treated as a separate
level in the tree.c S s g | ] } | � q Sr& r& )�.0�xr& r& r' �
<listcomp>� s z#Node.clone_node.<locals>.<listcomp>)r � child_attrs�getattrrK �list�setattr)r �result�attrnamer9 r& r& r' �
clone_node� s !zNode.clone_nodec C s d S)Nr& )r rC r& r& r' �analyse_declarations s zNode.analyse_declarationsc C s t d | j j � � d S)Nz*analyse_expressions not implemented for %s)r rL rM )r rC r& r& r' rq s zNode.analyse_expressionsc C s t d | j j � � d S)Nz$generate_code not implemented for %s)r rL rM )r rW r& r& r' �
generate_code s zNode.generate_codec C s# t | t � r | j j | � d S)N)rK � BlockNode�body�annotate)r rW r& r& r' r� s z
Node.annotatec C s� y | j SWn� t k
r� | j } | j s: | | _ | Sx| | j D]q } t | | � } | d k rh qD t | t � r� x; | D] } t | | j � � } q~ WqD t | | j � � } qD W| | _ | SYn Xd S)N) Z_end_pos�AttributeErrorr% r� r� rK r� �max�end_pos)r r% �attr�child�cr& r& r' r� # s"
z Node.end_posr% �d c
sD � d k r d S� d k r% t � � t | � � k rQ d | j j t | � f S� j t | � � � � � � f d d � � � f d d � | j j � D� } t | � d k r� d | j j t | � f Sd
| } d | j j t | � f } x8 | D]0 \ } } | d | | � | | d
� f 7} q� W| d | 7} | Sd S)zYDebug helper method that returns a recursive string representation of this node.
r z <...nesting level cutoff...>Nz <%s (0x%x) -- already output>c sm t | t � r) | j � � � d � � St | t � r_ d d j � � f d d � | D� � St | � Sd S)Nr z[%s]z, c s g | ] } � | � � � q Sr& r& )r� �item)�
dump_child�levelr& r' r� G s z1Node.dump.<locals>.dump_child.<locals>.<listcomp>)rK r| �dumpr� �join�repr)r� r� )�cutoffr� � encountered�
filter_out)r� r' r� C s
'z Node.dump.<locals>.dump_childc s. g | ]$ \ } } | � k r | | f � q Sr& r& )r� �keyr9 )r� r& r' r� K s z Node.dump.<locals>.<listcomp>z <%s (0x%x)>z z <%s (0x%x)
z %s %s: %s
r z%s>)rz �idrL rM �addr} re rJ )
r r� r� r� r� rj �indentrZ r� r9 r& )r� r� r� r� r' r� 8 s" "
(z Node.dumpz(#)c C s� | j s
d S| j \ } } } | j d d d d � } | t d | d � | � } | d
} | r� | d | � | | | d � } | j � d
| d <| | | | d � 7} d | j � | | d j | � f S)z[Debug helper method that returns the source code context of this node as a string.
� r* �ASCIIZerror_handling�ignorer r$ r Nz # <<<<<<<<<<<<<<
r" z"%s":%d:%d
%s
���r� )r% � get_linesr� �rstripZget_escaped_descriptionr� ) r Z mark_columnrX Z source_desc�line�col�contents�lines�currentr& r& r' �dump_posV s
"z
Node.dump_pos)r% ) rM rm rn r; �is_noneZ is_nonecheck�
is_literal�
is_terminator�
is_wrapper�tempsr� Zcf_stateZ
coercion_typer� r� � nogil_checkr� r� r� r� r� r� rq r� r� r� r� r� r& r& r& r' r| � s2 % r| c @ s[ e Z d Z d Z d g Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�CompilerDirectivesNodez9
Sets compiler directives for the children nodes
r� c C s2 | j } | j | _ | j j | � | | _ d S)N)�
directivesr� r� )r rC �oldr& r& r' r� p s z+CompilerDirectivesNode.analyse_declarationsc C s7 | j } | j | _ | j j | � | _ | | _ | S)N)r� r� rq )r rC r� r& r& r' rq v s
z*CompilerDirectivesNode.analyse_expressionsc C sP | j } | j j } | j | j _ | j j | | � | | _ | | j _ d S)N)r� � globalstater� � generate_function_definitions)r rC rW Zenv_oldZcode_oldr& r& r' r� } s z4CompilerDirectivesNode.generate_function_definitionsc C s; | j j } | j | j _ | j j | � | | j _ d S)N)r� r� r� �generate_execution_code)r rW r� r& r& r' r� � s z.CompilerDirectivesNode.generate_execution_codec C s; | j j } | j | j _ | j j | � | | j _ d S)N)r� r� r� r� )r rW r� r& r& r' r� � s z CompilerDirectivesNode.annotateN)
rM rm rn ry r� r� rq r� r� r� r& r& r& r' r� g s r� c @ s( e Z d Z d d � Z d d � Z d S)r� c C sA | j � j } x | D] } | j j | � q W| d d � =d S)N)� global_scopeZundeclared_cached_builtinsr� Zadd_cached_builtin_decl)r rC rW �entries�entryr& r& r' � generate_cached_builtins_decls� s
z(BlockNode.generate_cached_builtins_declsc C s( x! | j D] } | j | | � q
Wd S)N)Z lambda_defsr� )r rC rW rV r& r& r' �generate_lambda_definitions� s z%BlockNode.generate_lambda_definitionsN)rM rm rn r� r� r& r& r& r' r� � s r� c @ sg e Z d Z d g Z e d d � � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z
d S)� StatListNode�statsc O s t | | | � } | S)N)r� )r% rC rT r� rV r& r& r' �create_analysed� s z StatListNode.create_analysedc C s% x | j D] } | j | � q
Wd S)N)r� r� )r rC �statr& r& r' r� � s z!StatListNode.analyse_declarationsc s# � f d d � | j D� | _ | S)Nc s g | ] } | j � � � q Sr& )rq )r� r� )rC r& r' r� � s z4StatListNode.analyse_expressions.<locals>.<listcomp>)r� )r rC r& )rC r' rq � s z StatListNode.analyse_expressionsc C s( x! | j D] } | j | | � q
Wd S)N)r� r� )r rC rW r� r& r& r' r� � s z*StatListNode.generate_function_definitionsc C s5 x. | j D]# } | j | j � | j | � q
Wd S)N)r� �mark_posr% r� )r rW r� r& r& r' r� � s z$StatListNode.generate_execution_codec C s% x | j D] } | j | � q
Wd S)N)r� r� )r rW r� r& r& r' r� � s zStatListNode.annotateN) rM rm rn r� � staticmethodr� r� rq r� r� r� r& r& r& r' r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�StatNodec C s d S)Nr& )r rC rW r& r& r' r� � s z&StatNode.generate_function_definitionsc C s t d | j j � � d S)Nz.generate_execution_code not implemented for %s)r rL rM )r rW r& r& r' r� � s z StatNode.generate_execution_codeN)rM rm rn r� r� r& r& r& r' r� � s
r� c @ sI e Z d Z d g Z d d � Z d d � Z d d � Z d d � Z d
S) �CDefExternNoder� c C s� | j } d | _ | j j | � | | _ | j s= | j r� | j j } | j d s_ d } n( | sn d } n t d d � | D� � } | j | j | j | � d S)Nr Z preliminary_late_includes_cy28Fc s s | ] } t | t � Vq d S)N)rK � CVarDefNode)r� rV r& r& r' � <genexpr>� s z6CDefExternNode.analyse_declarations.<locals>.<genexpr>) � in_cincluder� r� Z include_fileZverbatim_includer� r� �all�add_include_file)r rC Zold_cinclude_flagr� Zlater& r& r' r� � s
z#CDefExternNode.analyse_declarationsc C s | S)Nr& )r rC r& r& r' rq � s z"CDefExternNode.analyse_expressionsc C s d S)Nr& )r rW r& r& r' r� � s z&CDefExternNode.generate_execution_codec C s | j j | � d S)N)r� r� )r rW r& r& r' r� � s zCDefExternNode.annotateN)rM rm rn r� r� rq r� r� r& r& r& r' r� � s
r� c @ s( e Z d Z g Z d Z d d � Z d S)�CDeclaratorNoder� c C s d S)Nr& )r r& r& r' �analyse_templates s z!CDeclaratorNode.analyse_templatesN)rM rm rn r� �calling_conventionr� r& r& r& r' r� � s
r� c @ s4 e Z d Z d g Z d Z d d d d d � Z d S)�CNameDeclaratorNode�defaultNr Fc C s� | r� | j d k r� | j s0 | j s0 | j rC t | j d � n@ | j r_ t | j d � n$ | j d d d d d �| _ t } | j r� | j
r� | j | j
� } | | _ | | f S)Nr� zMissing argument namezJUse spam() rather than spam(void) to declare a function with no arguments.Z for_displayr �pyrex)
r< �is_ptr�is_array� is_bufferr r% �is_void�declaration_coder �is_fused�fused_to_specific�
specializer/ )r rE rC �nonempty�
visibility�in_pxdr& r& r' �analyse s zCNameDeclaratorNode.analyse)rM rm rn r� r� r� r& r& r& r' r� s r� c @ s: e Z d Z d g Z d d � Z d d d d d � Z d S) �CPtrDeclaratorNode�basec C s
| j j � S)N)r� r� )r r& r& r' r� 1 s z$CPtrDeclaratorNode.analyse_templatesr NFc C sM | j r t | j d � t j | � } | j j | | d | d | d | �S)Nz+Pointer base type cannot be a Python objectr� r� r� )r= r r% r
�
c_ptr_typer� r� )r rE rC r� r� r� Zptr_typer& r& r' r� 4 s zCPtrDeclaratorNode.analyse)rM rm rn r� r� r� r& r& r& r' r� , s r� c @ s: e Z d Z d g Z d d � Z d d d d d � Z d S) �CReferenceDeclaratorNoder� c C s
| j j � S)N)r� r� )r r& r& r' r� @ s z*CReferenceDeclaratorNode.analyse_templatesr NFc C sM | j r t | j d � t j | � } | j j | | d | d | d | �S)Nz-Reference base type cannot be a Python objectr� r� r� )r= r r% r
Z
c_ref_typer� r� )r rE rC r� r� r� Zref_typer& r& r' r� C s z CReferenceDeclaratorNode.analyse)rM rm rn r� r� r� r& r& r& r' r� ; s r� c @ s1 e Z d Z d d g Z d d d d d � Z d S)�CArrayDeclaratorNoder� � dimensionr NFc s� | j r | j � s | j r� d d l m } t | j | � rO | j j } n | j f } � f d d � | D� } d | k r� | j d � } t | | j
d � t } n | j | j
| � } | j
j | � d | d | d | �S| j ri| j j � � | _ | j j j s&t | j j
d � | j j � }
|
d k roy t |
� }
Wqot k
reYqoXn d }
| j � s�t | j
d
| � | j r�t | j
d � | j r�t | j
d � t j | |
� } | j
j | � d | d | d | �S)
Nr )� TupleNodec s g | ] } | j � � � q Sr& )r: )r� �v)rC r& r' r� W s z0CArrayDeclaratorNode.analyse.<locals>.<listcomp>z Template parameter not a typer� r� r� zArray dimension not integerz%Array element type '%s' is incompletez'Array element cannot be a Python objectz"Array element cannot be a function)� is_cpp_class�is_template_type� is_cfunction� ExprNodesr� rK r� rT �indexr r% r �specialize_herer� r� �analyse_const_expressionr/ �is_int�get_constant_c_result_coder3 �
ValueError� is_completer= r
Z c_array_type) r rE rC r� r� r� r� rT �values�ix�sizeZ
array_typer& )rC r' r� P s> %
z CArrayDeclaratorNode.analyse)rM rm rn r� r� r& r& r& r' r� J s r� c @ sj e Z d Z d d d g Z d Z d Z d Z d Z d d � Z d d d d d d
� Z d d d � Z
d S)
�CFuncDeclaratorNoder� rT �exception_valuer Nc C s� t | j t � r� d d l m } m } | j j } t | | � rO | j } n/ t | | � rj | g } n t | j d � d Sg | _
xL | D]D } t | | � r� | j
j t j
| j � � q� t | j d � q� W| j j | _ | j
Sd Sd S)Nr )r� �NameNodez*Template arguments must be a list of names)rK r� r� r� r� r� r� rT r r% � templates�appendr
�TemplatePlaceholderTyper< )r r� r� �
template_nodeZtemplate_nodes�templater& r& r' r� � s"
z%CFuncDeclaratorNode.analyse_templatesFc s: | d k r i } | r" | d 8} g } x�t � j � D]�\ } } | j � d | d | d k ow � j ow d � j k �\ }
} |
j } | | k r| | }
|
j � � } | d k r� t |
j d � nK | t j
k r| j | � rt � j j d � t |
j d � n | } |
j
r4t � j d � | d k r[� j r[| j r[� j } | j rvt j | j � } | j r�t | j d
� | j t j | | | j � � | j r�� j d 7_ q8 � j r8 t � j d � q8 Wd } d } � j d k r5� j d
� � j d � � j d � � j d � | j rr� j sP� j rr� j d k rrt � j d � n�� j d k r�� j r�� j d k r�| j d k r�| d k r�| r�� j s�d d l m } | � j d | j d | �� _ � j r�� j j � � � _ � j d k r�� j j } | j rv| j rv| j! o_| j" j o_| j rvt � j j d � � j } n} � j j# | � � j � � � _ � j j$ � } | d k r�t% d � j j& j' � � | j( � j j � s�t � j j d � � j } | j! r!t � j d � t j) | | � j* d � j d | d | d � j j+ d � j, d � j- d � j. d! � j/ d" � j0 � } � j r�| j1 r�� � f d# d$ � } | | _2 n � j3 | � � � j d% } | r| j+ } | r| | k rt � j d& | | f � | | _+ � j j | � d' | d( | �S))Nr r� � is_self_argr r� z
Not a typez2Signature does not agree with previous declarationzPrevious declaration herez2Function argument cannot have C name specificationzJUse spam() rather than spam(void) to declare a function with no arguments.z-Non-default argument follows default argument�+Zios�newZ stdexceptZtypeinfozAException clause not allowed for function returning Python object�extern)� ConstNoder9 r/ zNException value must be a Python exception or cdef function with no arguments.z1get_constant_c_result_code not implemented for %sz6Exception value incompatible with function return typez!Function cannot return a function�optional_arg_countr� �exception_checkr� �nogil�with_gil�is_overridable�is_const_methodr� c s � j | � | � d S)N)�declare_optional_arg_struct)� func_type� fused_cname)rC r r& r' �declare_opt_arg_struct s z;CFuncDeclaratorNode.analyse.<locals>.declare_opt_arg_struct�callspecz2cannot have both '%s' and '%s' calling conventionsr� r� )4� enumeraterT r� �is_c_class_scoper� r< r: r r% r
r �same_asr� �cnameZis_unspecified� parent_typer� r� rE r� r � CFuncTypeArgr� r r
r� r= r� r� r r� r/ �is_errorr� � return_type� coerce_tor� r rL rM �assignable_from� CFuncType� has_varargsr� r r � overridabler r� r� r r )r r rC r� �directive_localsr� r� Zfunc_type_args�iZarg_node�name_declaratorr/ r< � type_node�
other_typeZexc_val� exc_checkr Z exc_val_typer r r r� r& )rC r r' r� � s�
-
'"
zCFuncDeclaratorNode.analysec
C s" t � } d t j } | j | t j | j � xS | j t | j � | j d � D]. } | j | j
| j | j d d d d �qS W| j t j
| j j
� } | d k r� t j | | � } | j � j d | d d d | d
d d | j d
| � } d | _ d | _ t j | j � | _ d S)a
Declares the optional argument struct (the struct used to hold the
values for optional arguments). For fused cdef functions, this is
deferred as analyse_declarations is called only once (on the fused
cdef function).
z%snNZallow_pyobjectTZallow_memoryviewr< �kind�struct�scope� typedef_flagr r% r r )r r � pyrex_prefix� declare_varr
r? r% rT rJ r r< r/ �mangleZopt_arg_prefixr� Zget_fused_cnamer� �declare_struct_or_union�defined_in_pxd�usedr� �
op_arg_struct) r r rC r r) Zarg_count_member�argZ struct_cnameZop_args_structr& r& r' r s$
*, z/CFuncDeclaratorNode.declare_optional_arg_struct) rM rm rn r� r r r r� r� r� r r& r& r& r' r� v s vr� c @ s. e Z d Z d g Z d d d d d � Z d S)�CConstDeclaratorNoder� r NFc C sM | j r t | j d � t j | � } | j j | | d | d | d | �S)Nz)Const base type cannot be a Python objectr� r� r� )r= r r% r
� c_const_typer� r� )r rE rC r� r� r� �constr& r& r' r� : s
z CConstDeclaratorNode.analyse)rM rm rn r� r� r& r& r& r' r3 5 s r3 c @ s� e Z d Z d d d d g Z d Z d Z d Z d Z d Z d Z d Z
d Z d Z d Z
d Z d d d d
� Z d d � Z d
d � Z d d � Z d d d d � Z d S)� CArgDeclNoderE �
declaratorr� rB r r NFc
C s
| r d | j _ | _ | j d k r�t | j t � r� | j j d k r� | r� | j j r� | j j | d d �} | j � } n | j j } t
| � | j _ d | j _ d | j _ d } n d } d | j _ | j j | d | �} t | j d � r | j j
r | j j
| j _ | j r�t | j t � r�t | j t � r�| j } x t | j t � ru| j } qWW| j j | _ | j } | j r�| r�| j d r�| j j d k r�| j | � } | d k r�| } | j j | | d | �S| j | j f Sd S)NTr� �
could_be_nameF�arg_name�annotation_typingr� )rE r r/ rK r7 r� r< �is_basic_c_typer� �empty_declaration_coder r6 �hasattrr9 r� �TemplatedTypeNoder� r� �array_declaratorrB r� � inject_type_from_annotationsr# )
r rC r� r r/ r9 r8 rE r7 rF r& r& r' r� ` s@ $
. zCArgDeclNode.analysec C sJ | j } | s d St | | d | j �\ } } | d k rF | | _ | S)NrD )rB rG r� rE )r rC rB rE rF r& r& r' r@ � s z)CArgDeclNode.inject_type_from_annotationsc C si | j d k rb | j rb | j j rM | j j | � | j j | j j � � S| j | j � | _ | j S)N)�
default_valuer� r� �generate_evaluation_coder/ � cast_coder� Zget_argument_default_const)r rW r& r& r' � calculate_default_value_code� s z)CArgDeclNode.calculate_default_value_codec C s | j r | j j | � d S)N)r� r� )r rW r& r& r' r� � s zCArgDeclNode.annotatec C s� | j } | d k s | j r" d S| d k r= | j | � } | j | � | j | � | ri | j � n | j | j � } | j d | | f � | j j r� | j
| j � � | j | � | j | � d S)Nz%s = %s;)
r� r� rD rB �make_owned_referencer� � result_asr/ rQ r= � put_giveref� generate_post_assignment_code�
free_temps)r rW �target�overloaded_assignmentr� r� r& r& r' �generate_assignment_code� s
$
z%CArgDeclNode.generate_assignment_code)rM rm rn r� r � is_type_arg�
is_generic�kw_only�not_none�or_noner/ r# rA rB �
is_dynamicr� r@ rD r� rL r& r& r& r' r6 B s" .
r6 c @ s e Z d Z d d � Z d S)�
CBaseTypeNodec C s
| j | � S)N)r� )r rC r& r& r' r: � s z CBaseTypeNode.analyse_as_typeN)rM rm rn r: r& r& r& r' rS � s rS c @ s% e Z d Z g Z d d d � Z d S)rA Fc C s | j S)N)r/ )r rC r8 r& r& r' r� � s z CAnalysedBaseTypeNode.analyseN)rM rm rn r� r� r& r& r& r' rA � s rA c @ s= e Z d Z g Z d Z g Z d Z d Z d d d � Z d S)�CSimpleBaseTypeNodeNFc C s� d } | j rF t j | j | j | j � } | st | j d � n�| j d k rh | j rh t } n�| j d k r� | j
r� | j r� | j } qt } nu| j r4| } xK | j D]@ } | j
| � } | d k r� | j r� | j j } q� d } Pq� W| d k r:| j | j | j � } | r:| j | _ n | } | r| j rU| j � } | j | j � } | d k rvn� | r�| j
r�| j r�| j } n t } t | j � | _ n^ | j r�| j | j k r�t | j d | j � t j | j � } n t | j d | j � | j r^| j s/| j r?t | j d � t j | � } | j | � n. | t j k r�t j } | j | � d | _ | r�| St j! Sd S)Nz&Unrecognised type modifier combination�objectz '%s' is not a type identifierz#can only complexify c numeric typesT)"r; r
Z
simple_c_type�signedZlongnessr< r r% � module_pathr r r r �lookupr� r/ r) �find_imported_moduler� r� Z lookup_typer r9 r� r �complex�
is_numeric�
is_complexZ CComplexTypeZ create_declaration_utility_coder Z complex_typeZc_double_complex_typer )r rC r8 r/ r) r� r� r& r& r' r� � sh
zCSimpleBaseTypeNode.analyse) rM rm rn r� r9 rW r; rZ r� r& r& r& r' rT � s
rT c @ s= e Z d Z d Z d d g Z d d d � Z d d � Z d S)
�MemoryViewSliceTypeNode�
memoryview�base_type_node�axesFc C s� | j j | � } | j r | Sd d l m } y | j | | j � } WnN t k
r� } z. t | j | j
� t j � | _
| j
SWYd d } ~ Xn X| j | j | � s� t | _
n5 t j | | � | _
| j
j | j � | j | � | j
S)Nr )�
MemoryView)r_ r� r r� ra Zget_axes_specsr` r r �positionZ message_onlyr
Z ErrorTyper/ Z
validate_axesr% r ZMemoryViewSliceTypeZvalidate_memslice_dtype�use_memview_utilities)r rC r8 rE ra Z
axes_specs�er& r& r' r� - s
z MemoryViewSliceTypeNode.analysec C s$ d d l m } | j | j � d S)Nr )ra )r� ra �use_utility_code�view_utility_code)r rC ra r& r& r' rc D s z-MemoryViewSliceTypeNode.use_memview_utilitiesN)rM rm rn r< r� r� rc r& r& r& r' r] ( s r] c @ s( e Z d Z d g Z d d d � Z d S)�CNestedBaseTypeNoderE Nc C s� | j j | � } | t j k r( t j S| j sL t | j d | � t j S| j j | j � } | sr | j
r� t | j d | | j f � t j S| j S)Nz '%s' is not a valid type scopez '%s.%s' is not a type identifier) rE r� r
r r� r r% r) � lookup_herer< �is_typer/ )r rC r8 rE �
type_entryr& r& r' r� Q s zCNestedBaseTypeNode.analyse)rM rm rn r� r� r& r& r& r' rg I s rg c @ s@ e Z d Z d d d d g Z d Z d Z d d d d � Z d S) r> r_ �positional_args� keyword_args�
dtype_nodeNFc C s� | d k r | j j | � } | j r+ | S| j r� | j � r� | j rt | j j rt t | j d � t j
| _ qag } xO | j D]D } | j
| � } | d k r� t | j d � t
} | j | � q� W| j | j | � | _ nz| j r�d d l m } | j | j | | j | j | j � } t j d d k rYt d d � | j � D� � } t j | | � | _ t | � rat | j � rat j t | j � | j � | _ n� t | j d d
d d �} t | j � d k s�| j j rt | j d � t j
| _ nY | j sd }
n
| j d }
t | j d
| d |
�| _! | j! j | | � d | _ | j j" r�| j# r�| j j$ | j# � | _ | j S)Nz+c++ templates cannot take keyword argumentsz!unknown type in template argumentr )�Bufferr r$ c S s+ g | ]! \ } } | j d � | f � q S)r� )r+ )r� r< r9 r& r& r' r� � s z-TemplatedTypeNode.analyse.<locals>.<listcomp>r< r� r zinvalid array declarationr� r� )%r_ r� r r� r� rl r8 r r% r
r r/ rk r: r r� r= r� rn �analyse_buffer_options�buffer_defaultsr � version_inford re Z
BufferTyper r! Z PythranExprr r� rJ r� r? r� r� r� ) r rC r8 rE �template_typesr r/ rn �optionsZempty_declaratorr� r& r& r' r� o sZ $!
zTemplatedTypeNode.analyse)rM rm rn r� rm r< r� r& r& r& r' r> _ s
r> c @ s+ e Z d Z d d g Z d d d � Z d S)�CComplexBaseTypeNoderE r7 Fc C s4 | j j | | � } | j j | | � \ } } | S)N)rE r� r7 )r rC r8 r� �_r/ r& r& r' r� � s z CComplexBaseTypeNode.analyseN)rM rm rn r� r� r& r& r& r' rt � s rt c @ s( e Z d Z d g Z d d d � Z d S)�CTupleBaseTypeNode�
componentsFc C sx g } xJ | j D]? } | j | � } | j rB t | j d � t S| j | � q W| j | j | � } d | _ | j S)Nz/Tuple types can't (yet) contain Python objects.T)
rw r� r= r r% r r Zdeclare_tuple_typer0 r/ )r rC r8 Zcomponent_typesr� r/ r� r& r& r' r� � s zCTupleBaseTypeNode.analyseN)rM rm rn r� r� r& r& r& r' rv � s rv c @ s7 e Z d Z d Z g Z d d � Z d d d � Z d S)�
FusedTypeNodea
Represents a fused type in a ctypedef statement:
ctypedef cython.fused_type(int, long, long long) integral
name str name of this fused type
types [CSimpleBaseTypeNode] is the list of types to be fused
c C s7 | j | � } | j | j | | j � } d | _ d S)NT)r� �declare_typedefr< r% r� )r rC r/ r� r& r& r' r� � s z"FusedTypeNode.analyse_declarationsFc C s� g } xe | j D]Z } | j | � } | s> t | j d � q | | k r] t | j d � q | j | � q Wt j | d | j �S)Nz
Not a typez Type specified multiple timesr< )rb r: r r% r r
� FusedTyper< )r rC r8 rb r$ r/ r& r& r' r� � s zFusedTypeNode.analyseN)rM rm rn ry r� r� r� r& r& r& r' rx � s rx c @ s( e Z d Z d g Z d d d � Z d S)�CConstTypeNoderE Fc C s; | j j | | � } | j r. t | j d � t j | � S)Nz)Const base type cannot be a Python object)rE r� r= r r% r
r4 )r rC r8 r� r& r& r' r� � s
zCConstTypeNode.analyseN)rM rm rn r� r� r& r& r& r' r{ � s r{ c @ s7 e Z d Z d d g Z d Z d Z d d d � Z d S)r� rE � declaratorsNc
C sA | j d k r i | _ | s$ | } | | _ | j rL | j d j � } n d } | d k r� | j d k r} t | j d � t | j � d k r� t | j d j d � t d | � } | j j
| _
x' | D] } | j | j | | j � q� W| j
j | � } | j r<| j r<| j s(| j r<t | j d � t Sd | _ | j } x�| j D]�} t | j � d k r�t | t � r�| j
d r�t | j d d � | j o�| j d k o�| j } | r�d
| _ t | t � r| j | | d | j d | d
| j �\ } }
n' | j | | d | d
| j �\ } }
|
j � s}| j d k r`|
j pf|
j s}t | j d |
� | j d k r�|
j r�t | j d � | j } | j } | d k r�t | j d � d S|
j r�| j d k r�t | j d � |
j r�d | j
k r d |
_! | j" | |
| j d | d | j d
| j d | j# d | j$ d | j �| _ | j d k r�t% j% | j � | j _ | r9| j j& j' | � d | j _( qX| j r�t | j d � | j) | |
| j d | d | d
| j d | j# d d �| _ t* j+ rXt, | j | j- � | j _- qXWd S)Nr r z Only extern functions allowedr z%Can't multiply declare template typesZ
func_templatez Fused types not allowed herezwarn.multiple_declaratorsz�Non-trivial type declarators in shared declaration (e.g. mix of pointers and values). Each pointer declaration should be on its own line.Fr! r� r� z Variable type '%s' is incompletez'Python object cannot be declared externr� z Missing name in declaration.z8C++ references cannot be declared; use a pointer insteadr� Tr �api� modifiersr z,Decorators can only be followed by functions�is_cdef).r! �
dest_scoper| r� r� r r% rJ r � outer_scoper� Z declare_typer< rE r� r� r� r �is_module_scoper r� rK r� r r r� r� r� �is_memoryviewslicer= r Z is_referencer� �is_static_method�declare_cfunctionr} r~ r r/ �create_to_py_utility_code�create_wrapperr, r �
docstringsr. r- )
r rC r� r� Ztemplate_paramrE r� r7 Zcreate_extern_wrapperr# r/ r< r r& r& r' r� s�
-! ! ! z CVarDefNode.analyse_declarations)rM rm rn r� �
decoratorsr! r� r& r& r& r' r� s r� c @ sL e Z d Z d g Z d d d � Z d d � Z d d � Z d d
� Z d S) �CStructOrUnionDefNode�
attributesNc
C sO | j | j | j | | j | j | j d | j d | j d | j �| _ d S)Nr� r} �packed)
r. r< r' r* r% r r� r} r� r� )r rC r) r& r& r' �declarez s z CStructOrUnionDefNode.declarec C s� d } | j d k r$ t | j � } | j | | � | j d k r� | j rb | j rb d | j _ x! | j D] } | j | | � ql W| j d k r� xU | j
D]J } | j } x | j r� | j
} q� W| | j j k r� t | j d � q� Wd S)Nr r z)Struct cannot contain itself as a member.)r� r r< r� r� r� r� r/ r� r� � var_entriesr/ r� rE r r% )r rC r) r� r/ r& r& r' r� � s
z*CStructOrUnionDefNode.analyse_declarationsc C s | S)Nr& )r rC r& r& r' rq � s z)CStructOrUnionDefNode.analyse_expressionsc C s d S)Nr& )r rW r& r& r' r� � s z-CStructOrUnionDefNode.generate_execution_code)rM rm rn r� r� r� rq r� r& r& r& r' r� l s
r� c @ s^ e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d S)� CppClassNodeNc C s� | j d k r d } np d d � | j D� } t d d � | j D� � } | r� t d d � | j d | � D� � r� t | j d � | j | j d | j | j d g d | j d | �| _ d S)
Nc S s) g | ] \ } } t j | | � � q Sr& )r
r )r� �
template_name�requiredr& r& r' r� � s z(CppClassNode.declare.<locals>.<listcomp>c s s | ] \ } } | Vq d S)Nr& )r� ru r� r& r& r' r� � s z'CppClassNode.declare.<locals>.<genexpr>c s s | ] \ } } | Vq d S)Nr& )r� ru r� r& r& r' r� � s zGRequired template parameters must precede optional template parameters.� base_classesr� r� )
r� �sumr� r r% �declare_cpp_classr< r r� r� )r rC rr Znum_optional_templatesr& r& r' r� � s
.zCppClassNode.declarec
s � j d k r d } } n, d d � � j D� } d d � � j D� } d � � j d k ru t � j � d | �� � f d d � } t | � � f d d � � j D� � } � j � j � � j � j | d � j d | �� _
� j
d k r� d Sd � j
_ � d k r � j
j � _ g } � f d
d � � � j d k r�� j
re� j red � j
_ xC � j D]8 } t | d d � } | r�| j � � | j � � qoWxI � � j � D]8 } | j | � � j d k r�d
d j | � | _ q�Wt � j d | �� _ � � _ d S)Nc S s g | ] \ } } | � q Sr& r& )r� r� ru r& r& r' r� � s z5CppClassNode.analyse_declarations.<locals>.<listcomp>c S s) g | ] \ } } t j | | � � q Sr& )r
r )r� r� r� r& r& r' r� � s r� c s. | j s | j r d St � j d | � d S)NTz&Base class '%s' not a struct or class.)r� � is_structr r% )Z
base_class)r r& r' �base_ok� s z2CppClassNode.analyse_declarations.<locals>.base_okc s% g | ] } | j � p � � � q Sr& )r� )r� �b)rC r) r& r' r� � s r� r c 3 s] xV | D]N } t | t � r$ | Vq t | t � r x � | j j � D] } | VqF Wq Wd S)N)rK � CFuncDefNoder� r� r� )r� r� Zsub_attr)�func_attributesr& r' r� � s
z:CppClassNode.analyse_declarations.<locals>.func_attributesr� ztemplate <typename %s>z , typename r� )r� r� r r< �filterr� r� r% r r� r� r� r/ r� r� r/ r� r� r� r r� �template_declarationr� r� r) )
r rC rr Ztemplate_namesr� Zbase_class_typesZ
defined_funcsr� r� r\ r& )rC r� r) r r' r� � sD
(!
z!CppClassNode.analyse_declarationsc C s" | j j | j j j � | _ | S)N)r� rq r� r/ r) )r rC r& r& r' rq � s z CppClassNode.analyse_expressionsc C s | j j | j j j | � d S)N)r� r� r� r/ r) )r rC rW r& r& r' r� � s z*CppClassNode.generate_function_definitionsc C s | j j | � d S)N)r� r� )r rW r& r& r' r� � s z$CppClassNode.generate_execution_codec C s | j j | � d S)N)r� r� )r rW r& r& r' r� � s zCppClassNode.annotate)
rM rm rn r� r� r� rq r� r� r� r& r& r& r' r� � s
/ r� c @ sI e Z d Z d g Z d d � Z d d � Z d d � Z d d � Z d
S) � CEnumDefNodere c
C sL | j | j | j d | j d | j d | j d | j d | j �| _ d S)Nr r* r� r} r� ) Z declare_enumr< r% r r* r� r} r� r� )r rC r& r& r' r� � s
zCEnumDefNode.declarec C sY | j d k rU | j r. | j r. d | j _ x$ | j D] } | j | | j � q8 Wd S)Nr )re r� r� r� r/ r� )r rC r� r& r& r' r� s
z!CEnumDefNode.analyse_declarationsc C s | S)Nr& )r rC r& r& r' rq s z CEnumDefNode.analyse_expressionsc C s� | j d k s | j r� | j | j � | j j t j d d �} x� | j j D]� } | j
d | | j | j | | j � f � | j
| � | j
d t j | j | | j | j � f � | j | t j � qP W| j j | � d S)N�public�
manage_refTz%s = PyInt_FromLong(%s); %sz.if (PyDict_SetItemString(%s, "%s", %s) < 0) %s)r� r} r� r% � funcstate�
allocate_tempr
r r� � enum_valuesrQ r �error_goto_if_null�
put_gotrefr Z
moddict_cnamer< �
error_goto�put_decref_clear� release_temp)r rW �tempr� r& r& r' r� s
z$CEnumDefNode.generate_execution_codeN)rM rm rn r� r� r� rq r� r& r& r& r' r� � s
r� c @ s% e Z d Z d g Z d d � Z d S)�CEnumDefItemNoder9 c C s� | j r] | j j | � | _ | j j j s] | j j t j | � | _ | j j | � | _ | j | j | j | j | j d | j
d | j d | j d | j
o� | j d k �} | j j | � | j r� | j j j | j � d S)Nr r� r} r� )r9 r� r/ r� r r
r? Z
declare_constr< r% r r� r} r� r� r r� )r rC Z
enum_entryr� r& r& r' r� ) s z%CEnumDefItemNode.analyse_declarationsN)rM rm rn r� r� r& r& r& r' r� " s r� c @ s@ e Z d Z d d g Z d d � Z d d � Z d d � Z d S)
� CTypeDefNoderE r7 c
C s� | j j | � } | j j | | d | j d | j �\ } } | j } | j } | j | | | j d | d | j d | j �} | j
r� d | _ | j r� | j r� d | _ d S)Nr� r� r r} Tr )
rE r� r7 r� r� r< r ry r% r} r� r� r/ )r rC r� r# r/ r< r r� r& r& r' r� B s $ z!CTypeDefNode.analyse_declarationsc C s | S)Nr& )r rC r& r& r' rq S s z CTypeDefNode.analyse_expressionsc C s d S)Nr& )r rW r& r& r' r� V s z$CTypeDefNode.generate_execution_codeN)rM rm rn r� r� rq r� r& r& r& r' r� 9 s r� c @ sB e Z d Z d Z d Z d Z d Z d Z d Z d Z g Z
d Z d Z d Z
d Z d Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z! d' d( � Z" d S))� FuncDefNodeNFc C s� d } x� | j D]� } | j r} d } | j ra | j j | � | _ | j j | j | � | _ q� t | j d � d | _ q | j r� d } q | r t | j d � q Wd S)Nr r z)This argument cannot have a default valuez/Non-default argument following default argument) rT r� rN rp r r/ r r% rO )r rC Z default_seenr2 r& r& r' �analyse_default_values} s z"FuncDefNode.analyse_default_valuesc C sF | d k r d S| j d s3 | j | � d k rB | j | � } | S)Nr: )r� r: rp )r rC rB r& r& r' �analyse_annotation� s
#z FuncDefNode.analyse_annotationc C s9 x2 | j D]' } | j r
| j | | j � | _ q
Wd S)N)rT rB r� )r rC r2 r& r& r' �analyse_annotations� s z FuncDefNode.analyse_annotationsc C s | j } | j } | j | k r@ | | j } | j | � } nT t | t � r� | j r� | j d r� | j } | j | � } | d k r� | Sn | S| d k r� t | j
d � nK | t k r� | j | � r� t | j
j
d � t | j
d � n | | _ | S)Nr: z
Not a typez2Signature does not agree with previous declarationzPrevious declaration here)r! r/ r< r: rK r6 rB r� r@ r r% r r rE )r rC r2 r! Z orig_typer$ r% r& r& r' �align_argument_type� s$
% z FuncDefNode.align_argument_typec C s d S)Nr r& )r �lenvr& r& r' �need_gil_acquisition� s z FuncDefNode.need_gil_acquisitionc C s� | } x | j s | j r' | j } q W| j ra t d | j j d | d | d | j j � } n! t d | j j d | d | � } | j | _ | j j
} | j r� | j o� | j
| _ | | _ | j | _ | S)Nr< r� Z parent_scopeZ
scope_name)�is_py_class_scoper r� �
needs_closurer r� r< r r
r r/ r� r r � local_scoper� )r rC �genvr� r/ r& r& r' �create_local_scope� s$
z FuncDefNode.create_local_scopec C s | j j | � d S)N)r� r� )r rC rW r& r& r' �generate_function_body� s z"FuncDefNode.generate_function_bodyc ' C s� d d l m } | j j r, d d l m } | j } | j ra | j ra d t j t j
f } n t j
} | j | � | j j
| | � | j | | � | j j d k o� | j j j } | j j d k o� | j j j } | p� | } | rd | j k r| j d g | _ | j � }
| j j d } | j j d } | sH| rd| j j t j d
d � � | j | � | j � | _ | j | j _ | j | j! � | j" | | � | j# d � |
r�| j# |
� | j$ | | � p�| j% }
| j& r| j& j' | d
|
d d �| j' | d
|
�| } x | j( s>| j rJ| j) } q,W| j* r�| j+ | j, j- j. t j � � | j# d � nj | j/ r�| j r�| j+ | j, j- j. t j � � | j# d � | j+ | j, j- j. t j
� � | j# d � | j0 | | � x0 | j1 D]% } | j2 p | j3 s| j4 | � qWd } | j j5 s�| j j6 rWd } n | j j rpd | j7 } | j# d | j j. t j8 � | f � | j9 � } | j: | � | j; } | j* p�| j/ } x$ | j< D] } | j- j6 r�d } Pq�Wd d � | j= D� } | j o"| j> o"| p"| } | j o;| j> o;| } | j pN| j> } | s]| rv| j? � d | j _ n | j r�| j> r�| j@ � | s�| r�| jA s�| jB � | jC r�| jC jD | � n d } | jE | � | r�| jF | � | r| jG � | jH | j j d | �| r.| jI | � | j* rHtJ jK d d � } tJ jL | j, j- j | � } | s}d | j, j- jM } | j# d t j | j, j- jN � | | j, j- jM t jO f � | j# d t j � | j# d t j | j, j- jP d � f � | jQ d tR � | j# | jS | j! � � | j# d � | jT t j � | j# d � | j/ r| jU r�| j# d | | j, j- jN � t jV f � n) | j# d! | | j, j- jN � t jV f � | j r�| j# d t j | f � n, | j* r| jQ | | j, j- � | jW | � | s
| rj| jA s^| jX r2| j j d" } n | j j } | jY | | j! d# | j j �d | j _Z | j[ | | � t\ | t] � } x� | j< D]� } | j- j6 r�| s�t^ | j_ � d k r | j2 r | j` | � q�| r�| j- j r�t^ | j_ � d k r�| ja | jb d$ | j j �q�Wx_ | j1 D]T } | j3 r+t^ | j_ � d k r+| j2 r+| jc rr| jd | � q+| j` | � q+WxC | j1 | j< D]1 } | j- je r�| jf jg jh r�| ji | | � q�W| jj | � x3 | j< D]( } | j- je r�| jk | | | j! � q�W| r( | jl � d% | j _ | jm | | � | j | j! d& d% �| j# d � | j# d' � | j jn s� | j j6 r� t j8 } | jo | | j � nO | j jp } | r� | j# d t j8 | f � n | j j5 s� | j# d( t j8 � | jq | jr k r� | j jn s
| js | jt � | ju | jq � x7 | j jv � D]& \ } }! | jw | |! d$ | j �q<
Wt^ | � d) k }" |" r�
| j j tx � | j# d* � | j# d+ � | j# d, � | j# d- � x | D] } | jy | | � q�
W| j# d. � | j j r | jz t j8 | � t j8 }# n | j{ � }# | j| � }$ |# d k sE |$ r� | j ro | j> ro | j# d/ � | j? � | j} | j j~ � | j r� | j> r� | jl � | j# d � n9 t | j j! d0 | j j~ d) � | j� | j j~ | j � | j jp }% |# d k r |% r |% }# |# d k r5 | j# d t j8 |# f � n | j j5 sU | j# d( t j8 � | rh | j� | � |" s� | s� | j j r� | js | j � | ju | jt � x | D] } | jy | | � q� W| r� | j� | � | j j rO
| j� | j j� t j8 � � }& | j# d1 |& � | j r"
| j? � | j# d2 � | j rB
| jl � | j# d � | ju | j � x� | j1 D]� } | jh si
| j2 r�
qi
| j- j r�
| j� | jb d$ | j �qi
| j- j6 ri
| j3 s�
t^ | j_ � d k ri
| jc r�
| j� | � qi
| j� | � qi
Wx� | j< D]� } | j- j6 rT| s:t^ | j_ � d k r�| j2 r�| j� | � q
| j- j r
| s|t^ | j_ � d k r
| j� | jb d$ | j �q
W| j* r�| j� t j | j, j- � | j s | j jp }% | j{ � }# |# d k r�|% r�|% }# | j j6 r | j� | j j� t j8 � � | j j� rX| j j d3 k rX| j# d4 t j8 t j8 f � | sd| r�d% | j _Z | jA s�| j j6 r�| j� t j8 d# | j j �n | j� d d# | j j �| j s�| j� � | s�| j r | j> r | jl � d% | j _ | j j5 s | j# d5 t j8 � | j# d � |
rD| j# d6 |
� | j� | j � | j� � | j& rz| j& j
| | � | j� | � d S)7Nr )rn )ra z%s->%s�
__getbuffer__�__releasebuffer__Z
cython_unused�profile� linetraceZProfilez Profile.cr� �with_pymethdef�
proto_onlyT�;z = NULLz = z%s%s;c S s g | ] } | j r | � q Sr& )r0 )r� r� r& r& r' r� 3 s z=FuncDefNode.generate_function_definitions.<locals>.<listcomp>� acquire_gilZtp_newrg z
%s->tp_newz%s = (%s)%s(%s, %s, NULL);zif (unlikely(!%s)) {z%s = %s;�Py_Nonez} else {�}z*%s = (%s) __Pyx_CyFunction_GetClosure(%s);z
%s = (%s) %s;z
(wrapper)r �have_gilF�tracez/* function exit code */z!__Pyx_pretend_to_initialize(&%s);r z0{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;�__Pyx_PyThreadState_declare�__Pyx_PyThreadState_assignz5__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);z5__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}�{z&Unraisable exception in function '%s'.z if (%s) {zOPyErr_SetString(PyExc_TypeError, "Memoryview return value is not initialized");�__hash__z5if (unlikely(%s == -1) && !PyErr_Occurred()) %s = -2;z
return %s;z#endif /*!(%s)*/)�r� rn r r� ra r� �is_closure_scopeZis_passthroughr �cur_scope_cname�outer_scope_cnameZmangle_closure_cnamesr� r� r� r� r< r) r r~ �get_preprocessor_guardr� r� re r � load_cached�enter_cfunc_scope� new_label� return_from_error_cleanup_labelr r� � gil_ownedr� r% r� rQ �needs_assignment_synthesis�pymethdef_required�py_func�generate_function_headerr� r� r� �put� scope_classr/ r� �needs_outer_scope� generate_argument_declarationsr� �
in_closurer6 �put_var_declarationr� r= Zmemslice_entry_init� retval_cname�insertion_point�generate_keyword_listr� � arg_entriesZbuffer_entries�has_with_gil_block�put_ensure_gil�declare_gilstate� is_generator�put_trace_declarations� code_object�calculate_result_code�put_trace_frame_init�getbuffer_check� put_declare_refcount_context�put_setup_refcount_context�getbuffer_initr ZConstructorSlotZget_slot_function�
typeptr_cnamer< � empty_tuplerC �
put_increfr r� r� �
is_cyfunction�
self_cnamerG r� �put_trace_call� can_trace� generate_argument_parsing_coderK r� rJ �cf_assignments�put_var_incref�put_incref_memoryviewslicer �xdecref_cleanupZput_var_xincrefr� Z
buffer_auxZbuflocal_nd_varr0 Z
put_init_vars� generate_argument_type_testsZput_acquire_arg_buffer�put_release_ensured_gilr� r� �put_init_to_py_nonerA � error_label� labels_used�put_goto� return_label� put_label�all_managed_temps� put_xdecref� restore_exception_utility_codeZput_release_buffer_codeZput_init_entry� error_value� caller_will_check_exceptions�put_add_traceback�qualified_namer Zput_unraisable�getbuffer_error_cleanup�getbuffer_normal_cleanup�unlikelyZerror_condition�put_xdecref_memoryviewsliceZput_var_xdecref�put_var_decref�
put_decref� put_xgiveref� as_pyobject�
is_special�put_trace_return�put_finish_refcount_context�put_temp_declarations�exit_cfunc_scope�generate_wrapper_functions)'r rC rW rn ra r� r� Zis_getbuffer_slotZis_releasebuffer_slotZis_buffer_slot�preprocessor_guardr� r� r� �cenvr� �init�tempvardecl_coder� Zhave_object_argsr2 Zused_buffer_entriesZ acquire_gil_for_var_decls_onlyZ acquire_gil_for_refnanny_onlyZ use_refnannyr� Ztp_slotZslot_func_cnameZ
trace_namer �lhs�valr r/ Zbuffers_present�err_valr& Zdefault_retval�condr& r& r' r� � sD
!
%' (
%
z)FuncDefNode.generate_function_definitionsc C s� | j j r t | j d � n@ | j j � r_ | j j pD | j j r_ t | j d | j � | j | j | j | j � } | j r� | j | _ | S)NzInvalid use of 'void'z Argument type '%s' is incomplete)
r/ r� r r% r� r� r� Z declare_argr< rB )r rC r2 r� r& r& r' �declare_argumentx s ) z FuncDefNode.declare_argumentc C s� | j j � r� | j j t j d d � � | j j } d | j j } | j d | | | j
| j | j j ow | j j
| j | j � f � n t | j d � d S)NZ ArgTypeTestzFunctionArguments.cz((PyObject *)%s)z:if (unlikely(!__Pyx_ArgTypeTest(%s, %s, %d, "%s", %s))) %szICannot test type of extern C class without type object name specification)r/ Ztypeobj_is_availabler� re r r� r� r� r rQ � accept_noner< �is_builtin_typeZ
require_exactr� r% r )r r2 rW r� �arg_coder& r& r' �generate_arg_type_test� s z"FuncDefNode.generate_arg_type_testc C s� | j j r d | j j } n | j j } | j d | � | j d t d t | j � � | j | j | j � f � | j d � d S)Nz
%s.memviewz,if (unlikely(((PyObject *)%s) == Py_None)) {zMPyErr_Format(PyExc_TypeError, "Argument '%%.%ds' must not be None", "%s"); %s�� r� )
r/ r� r� r rQ r� rJ r< r� r% )r r2 rW r r& r& r' �generate_arg_none_check� s z#FuncDefNode.generate_arg_none_checkc C s d S)Nr& )r rW r& r& r' r � s z&FuncDefNode.generate_wrapper_functionsc C s> | j | j � x' | j D] } | j s | j | � q Wd S)N)r� r% rT rR rL )r rW r2 r& r& r' r� � s z#FuncDefNode.generate_execution_codec C sX | j j d } y | j j j j d j } Wn t t f k
rM d } Yn X| | f S)Nr �obj)r� r� r/ rE r) r� r� �KeyError)r � py_buffer�obj_typer& r& r' �_get_py_buffer_info� s z FuncDefNode._get_py_buffer_infoc C sW | j � \ } } | j } | j d | � | j d � | j d � | j d � d S)Nzif (%s == NULL) {zZPyErr_SetString(PyExc_BufferError, "PyObject_GetBuffer: view==NULL argument is obsolete");z
return -1;r� )r& r rQ )r rW r$ ru �viewr& r& r' r� � s
zFuncDefNode.getbuffer_checkc C sg | j � \ } } | j } | rR | j rR | j d | | � | j d | � n | j d | � d S)Nz%s->objz%s->obj = NULL;)r& r r= r� rG rQ )r rW r$ r% r' r& r& r' r� � s zFuncDefNode.getbuffer_initc C s� | j � \ } } | j } | rp | j rp | j d | � | j d | � | j d | | � | j d � n | j d | � d S)Nzif (%s->obj != NULL) {z%s->objr� zPy_CLEAR(%s->obj);)r& r r= rQ r� r� )r rW r$ r% r' r& r& r' r � s z#FuncDefNode.getbuffer_error_cleanupc C sq | j � \ } } | j } | rm | j rm | j d | � | j d | � | j d | | � | j d � d S)Nzif (%s->obj == Py_None) {z%s->objr� )r& r r= rQ r� r� )r rW r$ r% r' r& r& r' r � s z$FuncDefNode.getbuffer_normal_cleanupc C s� | j j s d S| j j } t j j | � } | s8 d S| d k r^ | j j j d � r^ d S| d k r} | j j j r} d S| j � S)NZ__long__�__int__r� r� )r� r� )
r� r
r< r �method_name_to_slot�getr) rh r �preprocessor_guard_code)r r< �slotr& r& r' r� � s "z"FuncDefNode.get_preprocessor_guard)#rM rm rn r� r� r� r� r� �is_generator_bodyZ is_async_defr~ �has_fused_arguments�star_arg� starstar_argr� r� r� r� r� r� r� r� r� r� r r r! r r� r& r� r� r r r� r& r& r& r' r� Z sB � �
r� c sl e Z d Z d d d d g Z d Z d Z d Z d Z d Z d Z d Z
d Z d d � Z e
d d
� � Z d d � Z d
d � Z d d � Z d d d d � Z d d � Z d d � Z d d � Z d d � Z d d d � Z d d d d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z � f d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z! � S)3r� rE r7 r� � py_func_statFNc C s
| j j S)N)r� r< )r r& r& r' �unqualified_name s z CFuncDefNode.unqualified_namec C s | j r | j j Sd S)N)r� r� )r r& r& r' r� s zCFuncDefNode.code_objectc C s� | j | _ | j d k r$ i | _ | j j | j d � | j d k r� | j j | � } | d k r� t | j j d � t j
} n | j j | � } d | j k o� | j
d � | _ t | j t � r| j j | | d d | j d k d | j d | j �\ } } n7 | j j | | d d | j d k d | j �\ } } | j s`t | j d � | | _ | j | _ | j } x t | d � s�| j } q�W| | _ | j | _ | j j } | j d
k s�| j r�| r�t | j j d � | j d k r | j d
k r t | j j d � x t | j | j � D]� \ } } | j! | | � | j | _ | j" | _" | j# | _# | j$ | j | j | � | j j% r�d | _& | j j' r�d | j( k r�t | j d d � | j j' s�| j j) r6| j j* r t | j d � q6d | j( k r6t | j d d � q6W| j$ | j+ | j | � | j" } | j# }
| j, | _, | j | _ | j- | | | j d |
d | j d | j d | j d k d | j( d | j �| _. | j/ | j. _0 | j+ | _+ | j+ j1 r | j d
k r t | j d � | j+ j2 r.| j+ j3 | j d � | j r}| j4 r}| j r}t5 | j � d k st| j d j j6 r}d | _ | j7 | � | j8 | � d S) N�localsz
Not a typer� r� r" r! r� z*Suite attached to non-function declarationrT r� zBFunction with optional arguments may not be declared public or apir r z/Only extern functions can throw C++ exceptions.T�inlinez$Buffer unpacking not optimized away.r zUBuffer may not be acquired without the GIL. Consider using memoryview slices instead.r r} �definingr~ r z Function cannot return an arrayzused as a return valuer F)9r �is_c_class_methodr! r~ r� �directive_returnsr: r r% r
r rE r� rh r� rK r7 r� r� r� r� r/ r r
r= r� Zcfunc_declaratorrT r r} r
r �zipr� r< r �_validate_type_visibilityr� r. r� r~ �is_pythran_exprr r r r� r� �
inline_in_pxdZinline_func_in_pxdr� r� Zcheck_nullary_constructorr� rJ r= �declare_cpdef_wrapperr� ) r rC rE r# r/ r7 Z
opt_arg_count�
formal_arg�type_argr< r r& r& r' r� s� " .
"
)
z!CFuncDefNode.analyse_declarationsc C s� | j r�| j r" t | j d � | j j } | j d | j � } | j r� d d l m } t
| j d | | j d d ��g } | d j j | � n g } t
d | j d | j j d
| j d d d d d
| j d | d | d d � | _ | j | j _ | j j | � d | j j _ t | j d | j g �| _ t j | j _ | j j | j _ d | j _ | j j _ | j | j | <| j j r�| j s�t j r�t | j d | j �| _ t | j d | j | j g �| _ d S)Nz&static cpdef methods not yet supportedr� r )r� � decoratorr< r� r r% rT r/ r0 r- r� r� r� Tr� r� )!r r� r r% r� r< �call_self_noder� r� r� �
DecoratorNoder? rp �DefNoderT r- r� r� r
r� r1 r
r r/ � as_variabler0 r� �is_final_cmethodr �lookup_module_cpdef�OverrideCheckNode�overrider� )r rC r< Z py_func_bodyr� r� r& r& r' r< z s> '
z"CFuncDefNode.declare_cpdef_wrapperc C sq | j d k p | j } t | d d � } | rm | rm | j rm | j d k p] | j p] | j sm t | d � d S)zo
Ensure that types used in cdef functions are public or api, or
defined in a C header.
r� r� Nr z:Function declared public or api may not have private types)r� r )r� r} r� r� r� r )r r/ r% rC Z
public_or_apir� r& r& r' r9 � s z&CFuncDefNode._validate_type_visibilityr c
s� d d l m � � j j } | rB | d t | � � j j � } d d � | D� } | r� � j � j d � j j �} | } t
j } n� � j j r� � j j
j j } � j � j d | j �} | | _ � j � j d | d � j j �} d } n^ � j j d j j }
� j � j d |
j �} |
| _ � j � j d | d � j j �} | pet
j } � j � j d
| d � � f d d � | D� d
| �} t d � j d t j d | � S)Nr )r� c S s g | ] } | j � q Sr& )r< )r� r2 r& r& r' r� � s z/CFuncDefNode.call_self_node.<locals>.<listcomp>r< r" � attributeTr �functionrT c s( g | ] } � j � j d | �� q S)r< )r� r% )r� �n)r� r r& r' r� � s Z wrapper_callr% r r9 )r� r� r/ rT rJ r r� r% r� r< r rE r� r) r Z
AttributeNode�SimpleCallNode�ReturnStatNoder
r )
r Zomit_optional_argsr� rT � arg_names�cfuncZcall_arg_namesZ
skip_dispatchZ class_entryZ
class_noderj r> Zc_callr& )r� r r' r@ � s4 $ $ zCFuncDefNode.call_self_nodec C sD x= | j j D]/ } | j s, t | j d � | j | | � q
Wd S)NzMissing argument name)r/ rT r< r r% r )r rC r2 r& r& r' �declare_arguments� s z CFuncDefNode.declare_argumentsc C s
| j j S)N)r/ r )r r� r& r& r' r� � s z!CFuncDefNode.need_gil_acquisitionc C s | j } | j } | j r{ | r{ | j j r> t | j d � x: | j j D], } | j j rK | j rK t | j d � qK Wd S)Nz9Function with Python return type cannot be declared nogilz8Function declared nogil has Python locals or temporaries)
r/ r r r r= r r% r� r� Zin_with_gil_block)r rC r/ r r� r& r& r' r� � s zCFuncDefNode.nogil_checkc C s� | j | j _ | j d k r6 | j j | � | _ nA | j d k r] | j j | � | _ n | j | � | j | � | j | j � | _ | S)N) r� r� r1 rq r� r� r� r� r� )r rC r& r& r' rq � s
z CFuncDefNode.analyse_expressionsc C s d S)NFr& )r rC rW r& r& r' r� � s z'CFuncDefNode.needs_assignment_synthesisr c C s? | j } g } | j } xi | j d t | j � | j � D]D } | j � }
| j | j � } | j ss d |
}
| j |
� q< W| r� | j
r� t j j t
j � } | j r� | j | � n | j d | � | j r� | r� | j | j j t
j � � | j r| j d � | s"d g } | d k r:| j j } | j | d j | � � }
| j j d k r|d | k r|d } n d } d } | j | j j � } | j j |
d | �} | j } | j r�| r�| j j d
j | j � | j | j � | r!| j j d
j d | | | f � | j d | | | f � d S)
NzCYTHON_UNUSED %sz...�voidz, �privatez::zstatic r� � dll_linkageZmodule_declarationsz%s%s%s; /* proto*/z%s%s%s {) r� r/ rT rJ r r� rX r< �cf_usedr r r
r? r �skip_dispatch_cnamerG r1 �optional_args_cnamer r� �
func_cnameZfunction_header_coder� r� Zbuild_function_modifiersZfunc_modifiersr r6 r� r� �partsrQ )r rW r� �
with_opt_args�
with_dispatchr r) Z arg_declsr/ r2 Zarg_declr� Z dispatch_arg�entityZ
storage_classrR r~ �headerZ needs_protor& r& r' r� � sN *
z%CFuncDefNode.generate_function_headerc C s� | j } xs | j D]h } | j r | j | j � } | j sF | j r | j | � } | j d | j j
| j � | f � q Wd S)Nz%s = %s;) r� rT r� rX r< rG rS rD rQ r/ r� r )r rC rW r) r2 r� r� r& r& r' r�
s z+CFuncDefNode.generate_argument_declarationsc C s d S)Nr& )r rW r& r& r' r� %
s z"CFuncDefNode.generate_keyword_listc
su d } d } | j } | j j r2� j d t j � x� | j D]� � � j r? | j � j � } | j
sr | j r� � j d t j t j | f � � j
} x t | d � s� | j } q� W� j d � j t j | j j | j � f � | d 7} | d 7} q? Wx! t | � D] } � j d � qW� j d � � � f d d � } x' | j D] � | | j � j � � qQWd S)
Nr z if (%s) {zif (%s->%sn > %s) {r< z %s = %s->%s;r r� c sN | j rJ � j rJ � j d | j | j f � � j | � � j | � d S)Nz%s = %s;)r� r� rQ r �original_cnamer� �put_var_giveref)r� )r2 rW r& r' �put_into_closureB
s
zECFuncDefNode.generate_argument_parsing_code.<locals>.put_into_closure)r� r/ r rQ r rU rT r� rX r< rG rS r+ r7 r= r� r Z
opt_arg_cname�rangerh )
r rC rW r" r0 r) r� r7 ru r^ r& )r2 rW r' r� (
s4
z+CFuncDefNode.generate_argument_parsing_codec C s d S)Nr& )r rW r& r& r' �!generate_argument_conversion_codeK
s z.CFuncDefNode.generate_argument_conversion_codec C s] xV | j j D]H } | j r/ | j | | � q
| j j r
| j r
| j | | � q
Wd S)N)r/ rT �needs_type_testr r= r r! )r rW r2 r& r& r' r� N
s
z)CFuncDefNode.generate_argument_type_testsc s3 t t | � j | � | j r/ | j j | � d S)N)rf r� r� r1 )r rW )rL r& r' r� W
s z$CFuncDefNode.generate_execution_codec C s! | j j r d S| j j j Sd S)N�0)r r= r� r/ r� )r r& r& r' r \
s zCFuncDefNode.error_valuec C s
| j j j S)N)r� r/ r
)r r& r& r' r c
s z)CFuncDefNode.caller_will_check_exceptionsc C s� d } | j } | j } xo| j d k r�| d 7} | j } d | j j t j | f | _ | j � | j | d d | j j d | j j d | j �| j
j s� | j d � | j j
} d d � | d t | � | j j � D� } | j j r| j t j � n | j r | j d
� | j j r=| j t j � n | j rS| j d � | j d | j j d
j | � f � | j d � q Wd S)Nr r z %s%swrap_%srY rX r zreturn c S s g | ] } | j � q Sr& )r )r� r2 r& r& r' r� y
s z;CFuncDefNode.generate_wrapper_functions.<locals>.<listcomp>rb �NULLz%s(%s);z, r� )r� r/ Z
prev_entryrV r r+ rQ r� r
r r r� r� rT rJ r rT rU r� )r rW �kr� r rT Zarglistr& r& r' r f
s6
-
&z'CFuncDefNode.generate_wrapper_functions)"rM rm rn r� r; r� r! r7 rG r� r r1 r2 �propertyr� r� r< r9 r@ rO r� r� rq r� r� r� r� r� r` r� r� r r r r& r& )rL r' r� � s< ] $ -
# r� c @ s. e Z d Z g Z d Z d Z d d � Z d S)�
PyArgDeclNodeFc C s | j j | | � d S)N)r� r� )r rC rW r& r& r' r� �
s z+PyArgDeclNode.generate_function_definitionsN)rM rm rn r� r rM r� r& r& r& r' rf �
s rf c @ s e Z d Z d g Z d S)rA r? N)rM rm rn r� r& r& r& r' rA �
s rA c @ s� e Z d Z d d d d d d g Z d Z d Z d Z d Z d
Z d
Z d Z
d Z d Z d
Z
d
Z d Z d Z d Z d Z d Z d Z d Z d Z d Z d Z d d
� Z d d d d d d d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z! d d � Z" d d! � Z# d" d# � Z$ d d$ d% � Z% d& d' � Z& d( d) � Z' d* d+ � Z( d
d, d- � Z) d. d/ � Z* d0 d1 � Z+ d2 d3 � Z, d4 d5 � Z- d S)6rB rT r/ r0 r� r� �return_type_annotationFNrb r Tc K s� t j | | | � d } } } xJ | j D]? } | j rW | d 7} | j sW | d 7} | j s+ | d 7} q+ W| | _ | | _ | | _ d S)Nr r )r� r� rT rO r� �num_kwonly_args�num_required_kw_args�num_required_args)r r% rU rd Zrk�rr2 r& r& r' r� �
s
zDefNode.__init__c C sG | j r t | j j d � | j r8 t | j j d � | pA d% \ } } | d k rg } xf | j D][ }
|
j | d d �\ } } | j t j d | j d d d |
j
d t d
|
j � � qf Wt j d t d | d
d d d d | d d d d d | � }
t
| j d |
�} n� | d k r.| j } | j }
t | j � t |
j � k s^|
j r~t | j d � t | j d � x� t t | j |
j � � D]r \ } \ }
} |
j | d d d | d k o�| j �\ } } | d k s�| t j k r�| j |
_ | |
_ q�W| d k r\|
j d k r\d d l m } | | j d |
j d |
j �} t | j d t | j d | j d d �d | j d
d d |
j d | d |
j d |
j �} t | j d | p�g d t | j d |
j �d | d | j! d | j" d |
j# d |
d |
j d |
j d d! d" d d# t$ | d# i � d$ | �
S)&Nz'cdef function cannot have star argumentz+cdef function cannot have starstar argumentFr� r r< r rB r/ r% r rT r r� r
r r r
zwrong number of argumentszprevious declaration herer r )r r9 r� r~ rE r7 r� r- r r� rQ r} r! r7 )NF)%r/ r r% r0 rT r� r r
r r< rB r r r� r) r/ rJ r r r8 r r# r� r� r r r� r� r
r r r� rA r� r- r
r� )r rN r) r �returnsZ
except_valr~ r� r
Z
cfunc_argsr= r# r/ Z
cfunc_typer" r> r r7 r& r& r' � as_cfunction�
sz '.
zDefNode.as_cfunctionc C s' | j r
d S| j s | j r# d Sd S)z�Determines if the function's signature is compatible with a
cdef function. This can be used before calling
.as_cfunction() to see if that will be successful.
FT)r� r/ r0 )r r& r& r' �is_cdef_func_compatible s
z DefNode.is_cdef_func_compatiblec C s� | j r_ xS | j D]H } | j } | j r | j | j d k O_ | j | j d k O_ q W| j r� | j d � r� d | _ | j r� | j d � r� d | _ | j d k r� | j r� d | _ | j | � | j d k r� | j | � n
| j
| � | j | � | j j
j � | _ | j t k r| j r| j d r| j j rt | j | � \ } } | r| j r| | _ | j | � t | j d | d | j j d
| j d | j d | j d
| j �| _ | j j | � d S)N� classmethodr� Frg r z<lambda>r: rJ r< rT r/ r0 r ) r� r? r; �is_classmethodr< �is_staticmethodrh r� �analyse_argument_types�declare_lambda_function�declare_pyfunction�analyse_signaturer� � signaturer r rg r� r
rG r= r� �DefNodeWrapperr% rT r/ r0 �
py_wrapperr� )r rC r? r\ ru r r& r& r' r� ( sB
z DefNode.analyse_declarationsc C s} | j d | _ | j d } | j } d | _ x| j D]�} t | d � rW d } n� | j j | � } t | � r� | j r� t j
| | j g � } | j j | | � \ } } | j
| _
| | _ | j r� d | _ | j | | � | r| j rt | j d � | j j � | _ d | _ d | _ d | _ d | _ | j j sa| j j sa| j j r�| j rvd | _ q5| j r�d | _ q5| j j s�| j j! s�| j j s�| j j r�| j" r�| j" j# d k r�d | _ q�| | _ q5d | _ q9 d | _ | j r t | j d � | j r9 t | j d
� q9 W| | _ t | � rpd d � t$ | j � D� | _% n g | _% d S)
Nr3 � allow_none_for_extension_argsr< Tz9Python function argument cannot have C name specificationr r Fz.Only Python type arguments can have 'not None'z-Only Python type arguments can have 'or None'c S s( g | ] \ } } | j j r | � q Sr& )r/ Zis_numpy_buffer)r� r" �ar& r& r' r� � s z2DefNode.analyse_argument_types.<locals>.<listcomp>)&r� r! r� rT r= rE r� r r: r
rz Z
org_bufferr7 r< r/ r� r. r� r r r% Zas_argument_type�hdr_type�needs_conversionra rN r= r� r� rQ r rP �is_extension_typer r� �constant_resultr � np_args_idx)r rC ry Zf2sr2 r# rE r/ r& r& r' rr U sZ
$ "z DefNode.analyse_argument_typesc C sB | j j rV | j r% t | j d � t | j � d k oJ | j pI | j | j _ n2| j
d r�| j ps | j r�| j j t j
k rt | j � d k r� t j | j _ q�t | j � d k r�| j d j d k r�| j d j r�t j | j _ n� | j j t j k r�t | j � d k r=t j | j _ nK t | j � d k r�| j d j d k r�| j d j r�t j | j _ | j j } | j � } | t j k r�| d k r�t | j � d k r�| j r�t j
} | j _ d | _ d } | j rP| j rPd } d | _ t j | � | j _ } d | _ d | _ d | _ | j sb| j r�| j r�| j r�| j j d d � =x+t t! | t | j � � � D] } | j | } d | _" | j# | � r+| j r+| j rd | _$ t% j& | _' | _( n d | _# | j) | _' | _( d | _* nT | j+ | � | _' | j( j, | j' � s| j' j- rv| j( j- rvd | _. n d | _* | j* r�t/ j0 | j1 | _2 q�t/ j3 | j1 | _2 q�W| t | j � k r�| j4 � d S| t | j � k r>| j s | j4 � x; | j D]0 } | j" r
| j( j5 s1| j( j6 r
d | _. q
Wd S)Nz8special functions of cdef classes cannot have decoratorsr Zalways_allow_keywordsr r" T�*)7r� r
r� r r% rJ rT r/ r0 Ztrivial_signaturer� rv r Zpyfunction_signatureZpyfunction_noargsr� rO Zpyfunction_oneargZpymethod_signatureZ unaryfuncZ ibinaryfunc�num_fixed_args�self_in_starargrq r r Zfixed_arg_format�has_generic_argsrp r. Zdecorator_indirectionr� r_ �minrN r rM r Z type_typer{ r/ r r| Zfixed_arg_typer r= ra r Z
arg_prefixr< � hdr_cnameZ
var_prefix�
bad_signaturer} r )r rC �sigZnfixedr" r2 r& r& r' ru � sz 1!'' %
!zDefNode.analyse_signaturec C s� | j j } d | j � } | j r/ | d 7} | j } | j d � r_ | j d � r_ d } n d } t | j d | | j t | j
� | f � d S)Nz%dz or more�__zSpecial method�Methodz>%s %s has wrong number of arguments (%d declared, %s expected)) r� rv r� r� r< �
startswith�endswithr r% rJ rT )r r� Z expected_strr< �descr& r& r' r� � s
zDefNode.bad_signaturec C sn | j } | j | � } | rw | j rD | j j rD t | j d � | j j rw | j rw | j
rw t | j d d � | j | | j d | j
�} | | _
| j | j � } t j | | | j
_ t j rat | j | j � | _ t j | | | _ | j rj| j t j k s>| j s>| j d k rJ| j d rJd | _ qjt j | | | _ n d | _ d S)Nz:Only final types can have final Python (def/cpdef) methodsz'Overriding cdef method with def method.� Zallow_redefine� __getattr__Z fast_getattr) r< rh rD r �
is_final_typer r% r/ r� Zis_builtin_cmethodr� r rt r� �next_id� scope_prefixr Z
pyfunc_prefix� pyfunc_cnamer r� r. r- Zfuncdoc_prefix� doc_cnamer
r Z invisibler� �wrapperbase_cnameZwrapperbase_prefix)r rC r< r� �prefixr& r& r' rt � s( zDefNode.declare_pyfunctionc C s= | j | j | j � } d | _ | | _ | j | j _ d S)N)rs � lambda_namer% r- r� r r� )r rC r� r& r& r' rs s z DefNode.declare_lambda_functionc C s� x� | j D]� } | j s) t | j d � | j rn | j | j | j | j � | _ | j j r� d | j _ n | j
| | � | _ d | j _ d | j _ | j
| j _
q
W| j | | j � | j | | j � d S)NzMissing argument namerb r )rT r< r r% r| r, r/ r� r= r r r6 r0 r �declare_python_argr/ r0 )r rC r2 r& r& r' rO s ! zDefNode.declare_argumentsc C sw | rs | j d d k r% t j } n t } | j | j | | j � } d | _ d | _ d | _ d | _
| | _ d S)N� infer_typesFr rb ) r� r
Zunspecified_typer r, r< r% r6 r0 r r� r� )r rC r2 r/ r� r& r& r' r� s zDefNode.declare_python_argc C s� | j | j _ | j | � | j | � | j rJ | j | | j � | _ | j | � r� | j r� x3 | j d d d � D] } | j j | � | _ qz W| j
j | � | S)Nr r� ) r� r� r� r� rg r� r� r� r? rq rx �prepare_argument_coercion)r rC r? r& r& r' rq + s
zDefNode.analyse_expressionsc C s� | j r
d S| j s" | j j r& d S| j r3 d S| j j rC d S| j j rS d S| j se | j r� | d k r | j j
d S| j j
d S| j p� | j
S)NTFZbinding)rq �specialized_cpdefsr� Zis_fused_specialized�no_assignment_synthesisr
Z is_anonymousr� r r� r� r� r� r� )r rC rW r& r& r' r� 9 s z"DefNode.needs_assignment_synthesisc C s
| j j j S)N)r� rv r )r r& r& r' r K s zDefNode.error_valuec C s
| j j j S)N)r� rv r
)r r& r& r' r N s z$DefNode.caller_will_check_exceptionsc C sg | j r" | j j | j � | � | j rP | j j | j _ | j j | | � t j | | | � d S)N)�defaults_getterr� r� �py_wrapper_requiredr� rV rx r� )r rC rW r& r& r' r� Q s z%DefNode.generate_function_definitionsc C s� | r) | j r% | j j | | d � d Sg } | j j j rk d t j } | j s^ d | } | j | � d d � } x$ | j
D] } | j | | � � q� W| j r� | j | | j � � | j r� | j | | j � � | r� d j
| � } n d } | j j | j j � } | j d }
| j � } | r>|
j | � |
j d | | f � | rh|
j d
� | j d | | f � d S) NTz PyObject *%szCYTHON_UNUSED c S sP | j } | j r | j } n | j } | j j | � } | j sL d | } | S)NzCYTHON_UNUSED )r� r� r\ r r/ r� rS )r2 r� r Zdeclr& r& r' �
arg_decl_codej s
z7DefNode.generate_function_header.<locals>.arg_decl_codez, rP Zdeclszstatic %s(%s); /* proto */z#endifzstatic %s(%s) {)r� rx r� r� rv �
has_dummy_argr r� r� r rT r/ r0 r� r r� r� r� r� rQ ) r rW r� r� �
arg_code_list�self_argr� r2 r �dcZ
decls_coder r& r& r' r� ] s>
z DefNode.generate_function_headerc C s d S)Nr& )r rC rW r& r& r' r� � s z&DefNode.generate_argument_declarationsc C s d S)Nr& )r rW r& r& r' r� � s z DefNode.generate_keyword_listc sg � f d d � } x | j D] } | | j � q Wx- | j | j f D] } | rF | | j � qF Wd S)Nc sD | j r@ � j d | j | j f � � j | � � j | � d S)Nz%s = %s;)r� rQ r r\ r� r] )r� )rW r& r' r^ � s
z@DefNode.generate_argument_parsing_code.<locals>.put_into_closure)rT r� r/ r0 )r rC rW r^ r2 r& )rW r' r� � s z&DefNode.generate_argument_parsing_codec C s d S)Nr& )r rW r& r& r' r� � s z$DefNode.generate_argument_type_tests).rM rm rn r� rq rp r� Zreqd_kw_flags_cnamer� r� r� rg r� r� r� Z
py_cfunc_node�requires_classobj�defaults_structr- �
fused_py_funcr� rx r� rV r� r� rm rn r� rr ru r� rt rs rO r� rq r� r r r� r� r� r� r� r� r& r& r& r' rB �
sV B - @ M
.
rB c @ sK e Z d Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z
d d � Z d d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d S)5rw Nc O s\ t j | | | � | j j | _ | j j | _ | j j | _ | j j | _ d | _ d S)N)r� r� rJ rh ri rj r� rv )r rT rt r& r& r' r� � s zDefNodeWrapper.__init__c C sn | j j } | j } | j | j � } t j | | | _ t j | | | _ | j
| _
| j j | _ d S)N) rJ r� r< r� r� r Z
pywrap_prefixrV Zpymethdef_prefix�pymethdef_cnamerv r )r rC Z target_entryr< r� r& r& r' r� � s z#DefNodeWrapper.analyse_declarationsc C s� xZ | j D]O } | j j s1 | j j | � sY q
| j r
| j j r
| j j | � s
q
W| j r� | j j j r� | j j } d | _ x3 | j
D]( } | j r� | j j
r� d | j _ q� Wd S)Nr T)rT r/ r= �create_from_py_utility_coder{ r� r0 r� rS r� r� r6 r r; Z
cf_maybe_null)r rC r2 r� Zassr& r& r' r� � s z(DefNodeWrapper.prepare_argument_coercionc C sO t | j � } | d k sG | d k rK | j d j sG | j d j rK d Sd S)Nr r )rJ rT r rM )r Zargcountr& r& r' � signature_has_nongeneric_args� s z,DefNodeWrapper.signature_has_nongeneric_argsc C s
| j j S)N)rv r� )r r& r& r' �signature_has_generic_args� s z)DefNodeWrapper.signature_has_generic_argsc C s* g } | j j r" | j t j � xt | j D]i } | j r� | j j p\ | j j p\ | j j
r� | j | j j | j j
� � q, | j | j j
� q, W| j r� | j | j j j
� | j r� | j | j j j
� d j | � } | j j s| j d t j � | j d | j j j | f � d S)Nz, z%s = z%s(%s);)rv r� r r r� rT r{ r/ r� r� r\ rC r� r r/ r0 r� r r� r� r� rQ rJ r� )r rW rT r2 r& r& r' r� � s$
" z%DefNodeWrapper.generate_function_bodyc C s� | j j } | j | j � | j d � | j d � | j j � } | rX | j | � | j | � | j � | _ | j j | | � p� | j j
} | j | | � | j | | � | j
� } | j j r� d } n d } | j j s| j d | j j t j � | f � | j � | j d | j � | j | | � | j | � | j | � | j | j � | j | j � | j d � | j d � | j | j k r$| j | j � | j | j � x- | j j! � D] \ } } | j" | | � q�W| j# � }
|
d k r$| j d t j |
f � | j | j � x3 | j$ D]( } | j% r>| j& j r>| j' | � q>W| j( � | j j s�| j d t j � | j d � | j) � | r�| j d
| � d S) Nr� z/* Python wrapper */z = 0z%s%s;z %s (wrapper)z/* function exit code */z%s = %s;z
return %s;r� z#endif /*!(%s)*/)*rJ r� r� r% rQ r� r� r� r� r� r� r� r� r� r r= r� r� r r� r� r� r< r� r� r� r r� r� r� r� r� r� r� r� r r� r6 r/ r r r ) r rC rW r� r r� r Z retval_initr r/ r r� r& r& r' r� � sb
z,DefNodeWrapper.generate_function_definitionsr c
C s� g } | j } | j s! | j rN d t j } | j sA d | } | j | � x_ | j D]T } | j sX | j sy | j r� | j d | j
� qX | j | j j | j
� � qX W| j
j } | j r� | j � t j g k r� | j d � | j j r| j d k r| j d � | j r9| j d t j t j f � d j | � } d } | j d k ru| j j rud } d
} | j j | j � }
d | |
| f } | j d | � | r�| j
j r�| j
j j | | d
d �d St j r�| j! r�| j
j r�| j j" r�| j s | j# r�| j! } | j$ r=| j% � } | j d | j& | j' � f � | j r�| j d � | j d | j# � | j d � | s�| j
j r�| j( d | j) � | j* | j
j d d d
�| j d | � d S)Nz PyObject *%szCYTHON_UNUSED z CYTHON_UNUSED PyObject *unused�__ipow__zPyObject *%s, PyObject *%sz, r� r� r� Fzstatic %s%s(%s)z
%s; /*proto*/r� Tzstatic char %s[] = %s;z #if CYTHON_COMPILING_IN_CPYTHONzstruct wrapperbase %s;z#endifzstatic PyMethodDef %s = r� Z
allow_skipz%s {)r� r� )+rv r� r� r r� r rT rN r rM r� r{ r� rJ r� r
Z method_flagsr Z
method_noargsr) r r< r� �
args_cname�
kwds_cnamer� r rV rQ r� r� r r� r- Zis_property_scoper� Z
is_unicodeZas_utf8_stringr� Zas_c_string_literalr� r� Zput_pymethoddef)
r rW r� r� r� r� r r2 r� Zmfr� r[ Zdocstrr& r& r' r� 2
sr
"
z'DefNodeWrapper.generate_function_headerc C s{ xJ | j D]? } | j r
| j r9 | j d | j � q
| j | j � q
Wx' | j D] } | j rW | j | � qW Wd S)NzPyObject *%s = 0;) rT rN r| rQ r� r� r� r� r6 )r rC rW r2 r� r& r& r' r� }
s z-DefNodeWrapper.generate_argument_declarationsc C s� | j � } | j } | j d � } | j d k } | j d k pT | j d k pT | } x/ | j D]$ } | j j sa | j j | � sa qa W| j
� s� | r� t | j d � | j
| � n2 | j � s� | j | � n | j | j | | � | | _ | j | � r�| j | � s!| j | � | j | � | r�| j | j | � | j r�| j j j ru| j | j j � n | j | j j � | j | j j j � | j � | j d | j � � | j | � r�| j | � d S)NZargument_unpacking_doner z.This method cannot have * or keyword argumentsz
return %s;) �new_error_labelr� r� rh r/ r0 rT r/ r= r� r� r r% r` r� �generate_stararg_copy_code�'generate_tuple_and_keyword_parsing_code�
label_usedr� r� �generate_arg_decrefr� r� �put_var_xdecref_clear�put_var_decref_clearr rJ r r rQ r ) r rC rW �old_error_label�our_error_label� end_labelZhas_kwonly_argsZhas_star_or_kw_argsr2 r& r& r' r� �
sB
z-DefNodeWrapper.generate_argument_parsing_codec C s | r | j | j � d S)N)r� r� )r r2 rW r& r& r' �generate_arg_xdecref�
s z#DefNodeWrapper.generate_arg_xdecrefc C s | r | j | j � d S)N)r� r� )r r2 rW r& r& r' r� �
s z"DefNodeWrapper.generate_arg_decrefc C s� | j sl | j j t j d d � � | j d t j � | j d | j t j | j
� f � | j d � | j r� | j s� | j j j
r� d t j } q� d t j } n d t j t j f } | j j t j d d � � | j d
| t j | j t | j � | j
� f � | j rk| j j j
rkt d d � | j j j D� � r�| j d
| � | j d | j j j t j | j j j | j
� f � | j | j j j � | j d � | j d | j j j f � | j d � d | j j _ nt | j d | j j j t j t j f � | j d | j j j | j
� f � d | j j _ | j | j j j � | j rI| j j rI| j d | j j j t j | j j j f � | j r| j j j
r| j d � | j | j j j t � | j d | j
� � | j d � n | j d | j
� � | j | j j j � | j t j t � | j t j � | j d | j j j t j f � | j j t j d d �} | j d | | t j | f � | j d t j | f � | j d t � | j d � | j d | j j j | f � | j d � | j j! | � d | j j _ nN | j r�| j t j t � | j d | j j j t j f � d | j j _ d S) N�RaiseArgTupleInvalidzFunctionArguments.cz)if (unlikely(PyTuple_GET_SIZE(%s) > 0)) {zK__Pyx_RaiseArgtupleInvalid("%s", 1, 0, 0, PyTuple_GET_SIZE(%s)); return %s;r� z unlikely(%s)z%sz-unlikely(%s) && unlikely(PyDict_Size(%s) > 0)ZKeywordStringCheckzHif (%s && unlikely(!__Pyx_CheckKeywordStrings(%s, "%s", %d))) return %s;c s s | ] } | j j Vq d S)N)rV Z
allow_null)r� �refr& r& r' r� �
s z<DefNodeWrapper.generate_stararg_copy_code.<locals>.<genexpr>z if (%s) {z3%s = PyDict_Copy(%s); if (unlikely(!%s)) return %s;z} else {z
%s = NULL;r z,%s = (%s) ? PyDict_Copy(%s) : PyDict_New(); z if (unlikely(!%s)) return %s;r z=%s = PyTuple_New(PyTuple_GET_SIZE(%s)+1); if (unlikely(!%s)) r� z
return %s;z PyTuple_SET_ITEM(%s, 0, %s);r� Fz-for (%s=0; %s < PyTuple_GET_SIZE(%s); %s++) {z*PyObject* item = PyTuple_GET_ITEM(%s, %s);r� z!PyTuple_SET_ITEM(%s, %s+1, item);z%s = %s;)"r/ r� re r r� rQ r r� r� r< r r0 r� rS r� �boolr� Z
cf_referencesr r� r� r� rJ rq �put_xdecref_clearr r� r� rG r� r� r
�c_py_ssize_t_typer� )r rW Z kwarg_checkr� r& r& r' r� �
s�
"
z)DefNodeWrapper.generate_stararg_copy_codec s� � j d � } g } g } g } x� | D]� } | j r( | j r} | j r� | j r� | j rm | j | � q� | j | � q( | j r� | j | � q( | j r( | j r( | j | � q( W| | } | j | j }
t | � d k r | d j s| d j r |
d 8}
t | � } | j
o*|
| k } t | � }
| j sK| j
r[| j
| � � � j d � t | � t | � } � j d t j d j � f d d � | D� � f � | j | � � � j d | j d k r�d
p�d t j f � | j |
| | |
| | � � | j r'|
d k s3|
| k r�|
| k rR| j
rRd } n d
} � j d t j | |
f � � j | � | j r`| |
k r�| j
r�� j d t j | f � � j | � � j d � xLt | � D]w \ } } | j s�� j | j � } � j j t j d d � � � j d | j | f � � j � j | j! � � Pq�Wn�� j d � |
| k r�x�t | � D]) \ } } � j d | t j | f � q�Wnn� j d t j � | j
r�� j d � t" t | � � d d d! � } xw | D]o \ } } | |
d k rX| | d d k rC� j d � � j d | d � � j d | t j | f � qW|
d k r�� j d � � j d � � j d � | j
r�|
rx/ t# |
d d" d# � D] } � j d | � q�W� j | � n � j d � � j | � � j d � � j d � x1 t | � D]# \ } } | j$ | d | � � q>W� j d � � j% | � r�� j | � � j& | � � j j t j d d � � � j d | j | |
| t j f � � j � j | j! � � d S)$NZargtuple_errorr r r� z static PyObject **%s[] = {%s,0};�,c s& g | ] } d � j | j � � q S)z&%s)�intern_identifierr< )r� r2 )rW r& r' r� 9 s zJDefNodeWrapper.generate_tuple_and_keyword_parsing_code.<locals>.<listcomp>z
if (%s(%s)) {�likelyr z!=�<z(} else if (PyTuple_GET_SIZE(%s) %s %d) {z'} else if (PyTuple_GET_SIZE(%s) > %d) {z} else {�RaiseKeywordRequiredzFunctionArguments.cz&__Pyx_RaiseKeywordRequired("%s", %s); z&values[%d] = PyTuple_GET_ITEM(%s, %d);z switch (PyTuple_GET_SIZE(%s)) {zdefault:zCYTHON_FALLTHROUGH;z
case %2d: z case 0: zbreak;z case %2d:z default: r� z
values[%d]r� zD__Pyx_RaiseArgtupleInvalid("%s", %d, %d, %d, PyTuple_GET_SIZE(%s)); r� r� r� )'r� rN r� r rM rO r rj ri rJ r/ r� r0 �generate_stararg_init_coderQ �tupler �pykwdlist_cnamer� �#generate_argument_values_setup_coder� � generate_keyword_unpacking_coder� r� r r� r< r� re r r� r� r� r% r� r_ �generate_arg_assignmentr� r� )r rT Z
success_labelrW �argtuple_error_labelrk Zrequired_kw_only_argsZoptional_kw_only_argsr2 Z kw_only_args�min_positional_args�max_positional_args�has_fixed_positional_count�has_kw_only_args�all_argsZcomparer" �pystring_cnameZ
reversed_argsr& )rW r' r� s�
,
!
$
!
z6DefNodeWrapper.generate_tuple_and_keyword_parsing_codec C s3 | j j rV | j r0 t j | j t j | � } | j } | j d | j | f � n� | j j r| j
r| | j d | � | j | j j | | j j | j | � � | j
r/| j d � | j d | j j | j
| � f � | j j r| j | j j d d �| j d � n t | j d | j � d S)Nz%s = %s;z if (%s) {z} else {r� Tr� z2Cannot convert Python object argument to type '%s')r/ r= rN r
Ztypecastr r� rQ r �from_py_functionr� �from_py_call_coder% rD r� r� r )r r2 r� rW r� r& r& r' r� � s(
z&DefNodeWrapper.generate_arg_assignmentc C s� | j r` d | j j _ | j d | j j j | j j j | j � f � | j | j j j � | j r�d | j j _ | j d t j | f � | j d | j j j t j | t j f � | j d | j j j � | j r� | j
| j j j t � | j � | j d | j � � | j d � | j | j j j � | j d � | j
d | j j j t j f � | j t j t � | j d � d S)
Nr z0%s = PyDict_New(); if (unlikely(!%s)) return %s;z if (PyTuple_GET_SIZE(%s) > %d) {z4%s = PyTuple_GetSlice(%s, %d, PyTuple_GET_SIZE(%s));zif (unlikely(!%s)) {z
return %s;r� z} else {z %s = %s; )r0 r� r� rQ r r r� r/ r r� r� r r r� r� r� )r r� rW r& r& r' r� � s4
#z)DefNodeWrapper.generate_stararg_init_codec C s� t | � } | j d | d j d | � f � | j j rk | j d | j j t j | j j t j f � xa t | � D]S \ } } | j rx | j
j rx | j | � } | j d | | j
j
| � f � qx Wd S)Nz PyObject* values[%d] = {%s};r� rb z+%s *%s = __Pyx_CyFunction_Defaults(%s, %s);zvalues[%d] = %s;)rJ rQ r� rJ r� r Zdynamic_args_cnamer� r r� r/ r= rD r )r rT rW Zmax_argsr" r2 rA r& r& r' r� � s z2DefNodeWrapper.generate_argument_values_setup_codec
C s� | j d � | j d t j � | j d � | j rD | j d � x] t | d d! d" � D]E } | j d | d � | j d | t j | f � | j d � q[ W| j d � | j s� | j d
� | j | � | j d � | j d t j � | j s
| d
k r�d# } x) t | � D] \ } }
|
j
s | } q W| | k rR| d } | d
k rk| j d � x7t | d | d � � D]\ } }
| d
k r�| | k r�| d
k r�| j d � | j r�| | k r�| j d � n | j d | � | j |
j � } |
j
rh|
j
r q�| j d � | j d t j | f � | j d | � | j d � q�| j d | t j | f � | | k r/| d
k r�| j d � | j | � q�| j d � | j j t j d d � � | j d | j | | | | f � | j | j | j � � | j d � q�|
j
r�| j d � | j j t j d d � � | j d | j | f � | j | j | j � � | j d � q�W| d
k r�| j d � | r�| j | | � | j d � | d
k r�d } n/ | j r | j d | | f � d } n d } | j j t j d d � � | j d t j t j | j rn| j j j pqd | | j | j | j � f � | j d � d S)$NzPy_ssize_t kw_args;z1const Py_ssize_t pos_args = PyTuple_GET_SIZE(%s);zswitch (pos_args) {zdefault:r z
case %2d: z&values[%d] = PyTuple_GET_ITEM(%s, %d);zCYTHON_FALLTHROUGH;zcase 0: break;z default: r� zkw_args = PyDict_Size(%s);r z case %2d:zif (kw_args > 0) {z2PyObject* value = __Pyx_PyDict_GetItemStr(%s, %s);z-if (value) { values[%d] = value; kw_args--; }zKif (likely((values[%d] = __Pyx_PyDict_GetItemStr(%s, %s)) != 0)) kw_args--;zelse zelse {r� zFunctionArguments.cz2__Pyx_RaiseArgtupleInvalid("%s", %d, %d, %d, %d); r� z&__Pyx_RaiseKeywordRequired("%s", %s); z if (unlikely(kw_args > 0)) {rb zAconst Py_ssize_t used_pos_args = (pos_args < %d) ? pos_args : %d;Z
used_pos_argsZpos_argsZ
ParseKeywordszOif (unlikely(__Pyx_ParseOptionalKeywords(%s, %s, %s, values, %s, "%s") < 0)) %sr� r� r� )rQ r r� r/ r_ r� r� r� rj r r� r� r< rO r� re r r� r� r% �,generate_optional_kwonly_args_unpacking_coder� r0 r� r )
r r� r� r� r� r� r� rW r" Zlast_required_argr2 r� Z
pos_arg_countr& r& r' r� � s�
'
z.DefNodeWrapper.generate_keyword_unpacking_codec C sJ g } d } xM t | � D]? \ } } | j s | j r< q | sH | } | j | j � q W| rFt | � d k r� | j d | j r� d p� d t | � f � | j d � | j d | | t | � f � n | j d � | j d | � | j d t j t j
f � | j d
� t | � d k r9| j d � | j d � d S)
Nr z'if (kw_args > 0 && %s(kw_args <= %d)) {r� r� zPy_ssize_t index;z6for (index = %d; index < %d && kw_args > 0; index++) {zif (kw_args == 1) {z const Py_ssize_t index = %d;z:PyObject* value = __Pyx_PyDict_GetItemStr(%s, *%s[index]);z0if (value) { values[index] = value; kw_args--; }r� r� ) r rO r� r r< rJ rQ r0 r r� r� )r r� rW Z
optional_argsZfirst_optional_argr" r2 r& r& r' r� h s0
z;DefNodeWrapper.generate_optional_kwonly_args_unpacking_codec C s1 x* | j D] } | j r
| j | | � q
Wd S)N)rT r| �generate_arg_conversion)r rW r2 r& r& r' r` � s z0DefNodeWrapper.generate_argument_conversion_codec C s� | j } | j } | j ro | j r; | j d | j � n | j d | j � | j | | � | j d � nh | j r� | j | | � nL | j | � r� | j d | j j
| j f � n t | j d | | f � d S)Nz if (%s) {z
assert(%s); {r� z%s = %s;z+Cannot convert 1 argument from '%s' to '%s')
r{ r/ r= r� rQ r� �%generate_arg_conversion_from_pyobject�#generate_arg_conversion_to_pyobjectr r� r r r% )r r2 rW �old_type�new_typer& r& r' r� � s #z&DefNodeWrapper.generate_arg_conversionc C sX | j } | j r@ | j | j | j | j j | j | � � n t | j d | � d S)Nz2Cannot convert Python object argument to type '%s') r/ r� rQ r� r� r� r r% r )r r2 rW r� r& r& r' r� � s
z4DefNodeWrapper.generate_arg_conversion_from_pyobjectc C s~ | j } | j } | rf | j d | j j | | j | j | j j | j � f � | j | j � n t | j d | � d S)Nz%s = %s(%s); %sz5Cannot convert argument of type '%s' to Python object)
r{ Zto_py_functionrQ r� r r� r� r% Zput_var_gotrefr )r r2 rW r� r\ r& r& r' r� � s z2DefNodeWrapper.generate_arg_conversion_to_pyobjectc C sr xk | j D]` } | j r, | j | | � q
| j r
| j j sZ | j j sZ | j j r
| j | | � q
Wd S)N) rT ra r r r/ r= r� r� r! )r rW r2 r& r& r' r� � s z+DefNodeWrapper.generate_argument_type_testsc C s
| j j S)N)rv r )r r& r& r' r � s zDefNodeWrapper.error_value) rM rm rn ZdefnoderJ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r` r� r� r� r� r r& r& r& r' rw � s6 AK . U � y
rw c s� e Z d Z d Z d Z d Z d Z d Z d Z e j
d g Z
� f d d � Z � f d d � Z d d
� Z
� f d d � Z � S)
�GeneratorDefNodeTF� Generator�gbodyc s9 t | d g d d �| d <t t | � j | | � d S)Nr� r� Tr� )r� rf r� r� )r r% rt )rL r& r' r� � s zGeneratorDefNode.__init__c s9 t t | � j | � | j | j _ | j j | � d S)N)rf r� r� r� r� )r rC )rL r& r' r� � s z%GeneratorDefNode.analyse_declarationsc C s, | j j j } | j | j � } | j | j � } | j | j � } | j d � | j d | j | | j r | j j
| � n d t j | | | | j
d | j � f � | j t j t � | j rd } | j d | t j f � | j | t � | j | � | j � | j d � | j d � d S) Nr� zk__pyx_CoroutineObject *gen = __Pyx_%s_New((__pyx_coroutine_body_t) %s, %s, (PyObject *) %s, %s, %s, %s); %src �genz
gen->classobjz&%s = __Pyx_CyFunction_GetClassObj(%s);zreturn (PyObject *) gen;r� )r� r� rV r� r< �qualname� module_namerQ �
gen_type_namer� r� r r� r� r% r
r r� r� r� rG r )r rC rW Z
body_cnamer< r� r� Zclassobj_cnamer& r& r' r� � s(
!
z'GeneratorDefNode.generate_function_bodyc sb | j t j | j d � � | j j | d d �t t | � j | | � | j j | | � d S)Nz Coroutine.c�protoT) re r r� r� r� r� rf r� r� )r rC rW )rL r& r' r� � s z.GeneratorDefNode.generate_function_definitions)rM rm rn r� � is_coroutine�is_iterable_coroutine� is_asyncgenr� r� rB r� r� r� r� r� r& r& )rL r' r� � s r� c @ s e Z d Z d Z d Z d S)� AsyncDefNode� CoroutineTN)rM rm rn r� r� r& r& r& r' r� s r� c @ s e Z d Z d Z d Z d S)�IterableAsyncDefNodeZIterableCoroutineTN)rM rm rn r� r� r& r& r& r' r� s r� c @ s e Z d Z d Z d Z d S)� AsyncGenNodeZAsyncGenTN)rM rm rn r� r� r& r& r& r' r� s r� c sy e Z d Z d Z d Z d Z d Z d d d d � f d d � Z d d � Z d d � Z d d
d � Z
d d
� Z � S)�GeneratorBodyDefNodeTFNc sG t t | � j d | d | d | d | d d d g d d d d � d S) Nr% r� r< �is_async_gen_bodyr- rT r/ r0 )rf r� r� )r r% r< r� r� )rL r& r' r� s z GeneratorBodyDefNode.__init__c C s~ | j | j � } | j d � } t j | | } | j d t | j d | d d �} | | _ t | j � | _
| | _ d S)N� generatorr r� rQ ) r� r� r Zgenbody_prefixr, r r% rV r r< r r� )r rC r� r< r r� r& r& r' �declare_generator_body! s z+GeneratorBodyDefNode.declare_generator_bodyc C s | j | � | j | � d S)N)rr r� )r rC r& r& r' r� + s
z)GeneratorBodyDefNode.analyse_declarationsc C sT d | j j t j t j t j f } | r? | j d | � n | j d | � d S)Nz]static PyObject *%s(__pyx_CoroutineObject *%s, CYTHON_UNUSED PyThreadState *%s, PyObject *%s)z%s; /* proto */z%s /* generator body */
{)r� rV r �generator_cnameZlocal_tstate_cname�sent_value_cnamerQ )r rW r� r[ r& r& r' r� / s
z-GeneratorBodyDefNode.generate_function_headerc C s< | j } | j j | | � | j | � | j � | _ | j | j � | j | | � | j d � | j
| � | j � } | j d t j
� | j � } | j � | j | j j p� | j j � | j j d } | j j d } | s� | r7| j � d | j _ | j r$| j j | � n d } | j | � | j j | j j j � | j � } | j d � }
| j |
� | j |
� | j d | j! t j" | j � � | j# r`| j$ d k r`| j$ } | t% j& k r�d } n@ | t% j' k r�d } n( | t% j( k rd
} n t) d | � � | j d t j
| | j! t j
| j � f � | j* t j
� | j+ | | � | j j j j, r�| j d
| j j j- t j. � | j j j/ d t j0 � f � | j d t j. � | s�| r�d | j _ | j | j � | j d � | j d � | j# rs| j j1 rs| j2 rS| j j3 t4 j5 d d � � | j d | j2 rkd n d � | j6 | j7 � rp| j j1 s�| j8 | j9 � | j | j7 � | j# r�| j$ d k r�| j: t j
t; � t< j= | j> � j? j@ k r-| j j3 t4 j5 d d � � | j d tA | j2 � � x- | j jB � D] \ }
} | jC |
| � q=W| jD | j j � | j | j9 � | j# r�| jE t j
� n | j: t j
t; � | j d t j0 � | j d t j0 � | j d t j0 � | s�| r| jF t j
d | j jG �| jH � | j d t j
� | j d � | jI | j � | s[| r�| jJ | j j | j d | j jG �| j d t j0 � | j d |
� x. | jK D]# \ } } | j d! | | f � q�W| j d" � | s�| r
| jF d# d | j jG �| jH � | j d$ � | j d � | jL � d S)%Nr� zPyObject *%s = NULL;r� r� TZ first_runz%sz
PyList_New(0)zPySet_New(NULL)z PyDict_New()z)invalid type of inlined comprehension: %sz %s = %s; %sz%s = %s;z %s->closurez CYTHON_MAYBE_UNUSED_VAR(%s);Fz/* function exit code */�StopAsyncIterationz Coroutine.czPyErr_SetNone(%s);Z __Pyx_PyExc_StopAsyncIterationZPyExc_StopIterationZpep479z*__Pyx_Generator_Replace_StopIteration(%d);z+__Pyx_Coroutine_ResetAndClearException(%s);z%s->resume_label = -1;z%__Pyx_Coroutine_clear((PyObject*)%s);r z
return %s;r� zswitch (%s->resume_label) {zcase 0: goto %s;zcase %d: goto %s;z2default: /* CPython raises the right error here */r� z return NULL;)Mr� r� r� r� r� r� r� r% r� rQ r� r� r r� r� r� r� r< r r� r� r� r� r� r� r� r� Zinit_closure_tempsr� r/ r) Z use_labelr� r� r� �
is_inlined�inlined_comprehension_typer Z list_type�set_type� dict_typer r� r� r� r� r� rC r� r� r� re r r� r� r� r� r� r� r r �generator_stopr� �contextZfuture_directivesr� r� r� r r r r� r r r� Z yield_labelsr )r rC rW r� Zclosure_init_coder r� r� r� Z resume_codeZfirst_run_label� target_typeZ comp_initr r/ r" �labelr& r& r' r� : s�
!
z2GeneratorBodyDefNode.generate_function_definitions) rM rm rn r- r� r� r� r� r� r� r� r� r& r& )rL r' r� s
r� c @ s7 e Z d Z d g Z d Z d d � Z d d � Z d S)rF r� Nc s | j � _ � j j r! d } n d } d d l m � � j � j t � � _ � j
� j d � j d � � f d d � � j | d � D� �} | j j s� | j j
r� t � j d t � j d | �t � j d
d �g �� _ n t � j d
| �� _ � j j | � � _ � S) Nr r )r� rI rT c s+ g | ]! } � j � j d | j �� q S)r< )r� r% r< )r� r2 )r� r r& r' r� � s z9OverrideCheckNode.analyse_expressions.<locals>.<listcomp>r� �exprr9 )r� rT r� r� r� r� �RawCNameExprNoder% r � func_noderK r r� �
is_returncoder� � ExprStatNoderL r� rq )r rC Z first_argZ call_noder& )r� r r' rq � s" !z%OverrideCheckNode.analyse_expressionsc C s� | j | j j j � } | j j r4 d t j } n d | j d j } | j d � | j d t j
� | j d � | j j r� | j d � n | j d | � | j j t
d d �} | j j | � | j j t j d
d � � | j | | j � } | j d | | | | f � | j | � d
| } d | | j j j f } | j d | | f � | j j | � | j d � | j | t j
� | j j | � | j d � d S)Nz((PyObject *)%s)r z /* Check if called by wrapper */zif (unlikely(%s)) ;z#/* Check if overridden in Python */zelse {z5else if (unlikely(Py_TYPE(%s)->tp_dictoffset != 0)) {r� TZPyObjectGetAttrStrzObjectHandling.cz*%s = __Pyx_PyObject_GetAttrStr(%s, %s); %szPyCFunction_Check(%s)z1(PyCFunction_GET_FUNCTION(%s) != (PyCFunction)%s)zif (!%s || %s) {r� ) r� r� r� r< r� r � module_cnamerT r rQ rT r� r� r r� � set_cnamer� re r r� r� r% r� rV r� r� r� r
r� )r rW Zinterned_attr_cnamer� Zfunc_node_temp�errZ is_builtin_function_or_methodZ
is_overriddenr& r& r' r� � s6
z)OverrideCheckNode.generate_execution_code)rM rm rn r� r� rq r� r& r& r& r' rF � s
rF c @ s e Z d Z d S)� ClassDefNodeN)rM rm rn r& r& r& r' r� s r� c @ s� e Z d Z d d d d d d d d d g Z d
Z d
Z d Z d
Z d
Z d
d
d d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d
S)�PyClassDefNoder� rd � metaclass�mkwri � class_resultrJ �
class_cellr� NFc C s t j | | � | | _ | | _ | | _ | | _ | | _ d d l m } | j r� t j
r� t | j | j � } | j
| d | �}
n d }
| } | rpd } d | _ | j r^| j rFxw t t | j � � d d d � D]S \ } }
|
j j d k r� | j d k r t |
j d � |
j | _ | j | =q� W| | _ qp| j d k spt � n | j | � | _ | s�| j s�| j s�| j r�| j d k r| r�| j r�| j } n d } | r�| j j r�| j j r�n! | j | d | d | j �| _ d } n d } | j | d
| d |
d | j d | j d | j �| _ | j | d
| d | j d | j d |
d | j d | j d
| d | �| _ nE | j! | d g �| _ | j" | d
| d | d | j d |
�| _ | j# | d
| �| _$ | j% | j � | _& d S)Nr )r� r9 FTr� z2keyword argument 'metaclass' passed multiple timesr ri r< r- rd Zcalculate_metaclass�allow_py2_metaclassr8 r� )'r� r� r< r- r� r� ri r� r� r r� r. r% Z
StringNode�is_py3_style_classr7 r8 r� r r� r9 r� r r �AssertionError� ProxyNodeZis_sequence_constructorrT ZPyClassMetaclassNodeZPyClassNamespaceNoderd Z Py3ClassNode�classobj�DictNodeZ ClassNoder� rJ ZClassCellInjectorNoder )r r% r< ri r- r� r� rl Zforce_py3_semanticsr� Zdoc_noder r" r� ZmkdictZneeds_metaclass_calculationr& r& r' r� % sl / !
! zPyClassDefNode.__init__c C s� | j r t | j j d � d Sd d l m } t | j d d d d d | j d | j j pr | j | j d
g �d | j
d | j d
d d | j �S)zO
Return this node as if it were declared as an extension class
z7Python3 style class could not be represented as C classNr )r� r� rQ r� �
class_nameri rT r� r� r� Fr- )
r r r r% r� r� �
CClassDefNoder< ri r� r� r� r- )r r� r& r& r' � as_cclasso s $ zPyClassDefNode.as_cclassc C sK | } x | j s | j r' | j } q Wt d | j d | � } | _ | S)Nr< r� )r� r r� r r< r) )r rC r� r r& r& r' � create_scope� s
zPyClassDefNode.create_scopec C s� | j } | j rm d d l m } x? | j d d d � D]' } | | j d | j d | g �} q9 Wd | _ | | _ | j j | � | j j | � | j
| � } | j | _ | j j j
| _ | j j | � d S)Nr )rK rI rT r� )r r� r� rK r% r? r r� rJ �analyse_target_declarationr r� r� r �class_obj_cnamer� )r rC r rK r? r r& r& r' r� � s z#PyClassDefNode.analyse_declarationsc C s� | j r | j j | � | _ | j r< | j j | � | _ | j rZ | j j | � | _ | j j | � | _ | j j | � | _ | j } | j j | � | _ | j j | | j
� | j j | � | _ | S)N) ri rq r� r rd r r) r� rJ �analyse_target_expressionr r )r rC r r& r& r' rq � s z"PyClassDefNode.analyse_expressionsc C s- | j | j | � | j j | j | � d S)N)r� r) r� r� )r rC rW r& r& r' r� � s z,PyClassDefNode.generate_function_definitionsc C s� | j | j � | j j | � | j } | j rB | j j | � | j r[ | j j | � | j rt | j j | � | j j | � | j j
� | _ | _ | j
j | � | j j | � | j j | � | j
j | | j j
� � | j
j | � | j j
� | _ | _ | j j | j | � | j j | � | j j | � | j rq| j j | � | j j | � | j r�| j j | � | j j | � | j r�| j j | � | j j | � | j j � d S)N)r� r% Z
pyclass_stackr r) ri rB r r� rd r� �namespace_cnamer r r� r� r Zgenerate_injection_code�generate_disposal_coder rJ rL rI �pop)r rW r r& r& r' r� � s> z&PyClassDefNode.generate_execution_code)rM rm rn r� r� r r r� r r� r r r� rq r� r� r& r& r& r' r�
s I r� c @ s� e Z d Z d g Z d Z d Z d Z d Z d Z d Z d Z
d Z d d � Z d d � Z
d d � Z d
d � Z d d
� Z d d � Z e d d d � � Z d d � Z d S)r
r� NFc C sb t | d � s[ d d l m } | j rR | j | j | g | j d d �| _ n d | _ | j S)N�_buffer_defaultsr )rn Z
need_completeF)r= r� rn �buffer_defaults_nodero �buffer_defaults_posr )r rC rn r& r& r' rp � s z CClassDefNode.buffer_defaultsc C s� | j rL | j d k rL | j j d � } | j | | j � } | sR d Sn | } | j d | j d | j d d d d d | j d d d
| j d | j d | j d
| j d | j
d | j | � d | j �
| _
d S)Nr �.r< r% r5 r � implementingr� rE �objstruct_cname�
typeobj_cnamer� r* r} rp �shadow)r� r� �splitrY r% �declare_c_classr �objstruct_name� typeobj_namer* r} rp r r� )r rC rW �
home_scoper& r& r' r� � s( zCClassDefNode.declarec s� � j r# � j r# t � j d � � j rC t � j d � j � d � _ � j r� d � _ x, � j D]! } | j
� j k rh | � _ qh W� j d k r� t � j d � j � � _ d � j _
� j � j � � j j r�� j j d } | j � � } | t j t j t j f k r4� j | j � � j } | d k rZt | j d � j � n� | t j k rrd } n� | j r�| j o�| j r�t | j d | � n� | j � s�t | j d | j
� j f � ny | j r| j j r| j rt | j d | � j f � n; | j rE| j
d, k rEt | j d | j
� n | � _ � j j! d
d � d k r�| t j k r�t" � j d d � � j# d k } | r�� j r�� j j r�� j j$ j% � � f d d � � d S� j r(� j& d k r(� j j' d � } � j( | � j � } | s.d Sn � } � j& d k r�� j d k r�� j t) j* k r�� j+ d d � d k r�t" � j d � j d � | j, d � j d � j d | o�� j- d | o�� j- d � j d � j d � j d � j. d � j& d � j/ d! � j0 d" � j1 � � d# � j2 �
� _3 � j2 rI� j3 | j � j � _4 | � k r�� j& d k r�� j5 � j � j3 � j � � j3 j j � _ } | d k r�� j | _ � j6 r�t7 j8 r�t9 � j � j6 � | _6 | rr� j# j: | � � j j; d$ � }
|
rT|
j< rT| j= rT| j> rT� j j? d% � |
_@ � j jA d$ |
j6 |
j � � j- rid | _= n d | _> tB � j j � d k rd| s�� j- r�t � j j d j d& � x= � j j d d � D]% } | j � � r�t | j d' � q�Wd � j3 j _C d d( lD mE } | jF � j d) | jG � j d* � j �� j | jH � j d+ g �g �� _I nN � j r�� j jJ p�� j jC � j3 j _C d � _I n d � j3 j _C d � _I � jK � j3 � x � j3 j j$ D]
}
|
� q�Wd S)-NzTObject struct name specification required for C class defined in 'extern from' blockz:Decorators not allowed on cdef classes (used on type '%s')r r z+First base of '%s' is not an extension typez '%s' is not an extension typez*Base class '%s' of type '%s' is incompletez%Base class '%s' of type '%s' is finalr� �str�byteszGinheritance from PyVarObject types like '%s' is not currently supported�freelistzIfreelists cannot be used on subtypes, only the base class can manage themc s
� j � � S)N)r� r& )rC r r& r' �<lambda>I s z4CClassDefNode.analyse_declarations.<locals>.<lambda>r r Z __builtin__� zcpython.z %s already a builtin Cython typer< r% r5 r r� rE r r r� r* r} rp r r} Z__dict__getterz'Only declare first base in declaration.z+Only one extension type base class allowed.)r� rT r9 r8 )r� r r! )Lr� r r r% r� r rE r� �moduleZcimported_modulesr< r r� Zhas_extern_class�add_imported_moduleri rT r: r
r? r> r@ rX Z
sign_and_namer/ r r} r r r� r) r� r� r* r r� Zdefered_declarationsr r� r rY r Z
builtin_typesr r r� r r* r} rp r r� rC �add_imported_entryr- r r� r. r� rh Z is_variableZdefinedZ implementedZmangle_internalZ getter_cname�declare_propertyrJ �
early_initr� r� r� ZIdentifierStringNoder �type_init_argsZ is_externalZallocate_vtable_names)r rC r% r� rE Zbase_class_scopeZhas_bodyrW r r) Z
dict_entryZ
other_baser� Zthunkr& )rC r r' r� s�
* " # $ ! z"CClassDefNode.analyse_declarationsc C sJ | j r- | j j j } | j j | � | _ | j rF | j j | � | S)N)r� r� r/ r) rq r* )r rC r) r& r& r' rq � s z!CClassDefNode.analyse_expressionsc C s6 | j r2 | j | j | � | j j | j | � d S)N)r� r� r) r� )r rC rW r& r& r' r� � s z+CClassDefNode.generate_function_definitionsc C s� | j | j � | j r) | j j | � | j j j s�| j r�| j j | � d | j j � } d | } | j
j t j
d � } | j d | | j j � f � | j | j | | j � � | j | � | j d | | f � | j d � | j d | | f � | j | j | j � � | j d � | j
j | � | j | t j
� | j | � | j d | j j j | f � | j | t j
� | j j | � | j j | � | j | j | d � d S)
NzPyTuple_GET_ITEM(%s, 1)z(((PyTypeObject*)PyTuple_GET_ITEM(%s, 0))Tz0%s = PyType_Type.tp_new(&PyType_Type, %s, NULL);z*if (((PyTypeObject*) %s)->tp_base != %s) {zPPyErr_Format(PyExc_TypeError, "best base '%s' must be equal to first base '%s'",zB ((PyTypeObject*) %s)->tp_base->tp_name, %s->tp_name);r� z%s.tp_bases = %s;)r� r% r� r� r� r/ r) r* rB r� r� r� r
r rQ r� r� r� r� r� rG r r� r rI �generate_type_ready_code)r rW ri Z
first_baseZ
trial_typer& r& r' r� � s8
z%CClassDefNode.generate_execution_codec C s� | j } | j } | j } | s% d S| j d k r�x! t j D] } | j | | � q> W| r� | j j t j
d d � � d } n d } | j d | | | j | j
� f � | j d | � t j | d � } t j | d � } | d k rw| d k rw| j rd
}
d } n d }
d
} | j j t j
| d � � | j d | | f � | j d | |
f � | j d � xZ| j j j D]I} | j d- k }
| j r�t j r�| j r�|
r�t j j | j � } | r�| j � n d } | r�| j | � | j d � | j d � | j d | | j | j d | j
� f � | j d � | j d | j � | j d | j | j f � | j d | j � | j d � | j d � | j d � | r�| j d � q�W| j rg| j j t j
d d � � | j d | | j | j | j
� f � | rg| j j t j
d d � � | j d! | | j | j
� f � | j j r�| j j d" r�| j d# t j | j! | | j | j
� f � | j" s�| j# d$ � n d } | rF| j t$ k r6d% | } | j% r| j& } n
d& | j& } | j d' | | | | j' f � n t( | j
d( � | j" s^| j# d) � n d r�| j j t j
d* d � � | j d+ | | j | j
� f � | j r�| j d, | j) | j f � d S).Nr Z PyType_ReadyzExtensionTypes.cZ__Pyx_PyType_Readyzif (%s(&%s) < 0) %sz%s.tp_print = 0;Z tp_getattroZ
tp_dictoffsetrb Z#__Pyx_PyObject_GenericGetAttrNoDictZ PyObject_GenericGetAttrNoDictZ __Pyx_PyObject_GenericGetAttrZPyObject_GenericGetAttrzObjectHandling.cz�if ((CYTHON_USE_TYPE_SLOTS && CYTHON_USE_PYTYPE_LOOKUP) && likely(!%s.tp_dictoffset && %s.tp_getattro == PyObject_GenericGetAttr)) {z%s.tp_getattro = %s;r� r� r� z #if CYTHON_COMPILING_IN_CPYTHONr� zEPyObject *wrapper = PyObject_GetAttrString((PyObject *)&%s, "%s"); %s�wrapperz/if (Py_TYPE(wrapper) == &PyWrapperDescr_Type) {z0%s = *((PyWrapperDescrObject *)wrapper)->d_base;z %s.doc = %s;z0((PyWrapperDescrObject *)wrapper)->d_base = &%s;z#endifZ SetVTablezImportExport.cz+if (__Pyx_SetVtable(%s.tp_dict, %s) < 0) %sZ MergeVTablesz#if (__Pyx_MergeVtables(&%s) < 0) %s�internalz=if (PyObject_SetAttrString(%s, "%s", (PyObject *)&%s) < 0) %s� __weakref__z%s.tp_weaklistoffsetz struct %sz#if (%s == 0) %s = offsetof(%s, %s);z)__weakref__ slot must be of type 'object'Z__reduce_cython__Z SetupReducez.if (__Pyx_setup_reduce((PyObject*)&%s) < 0) %sz %s = &%s;)r� r� )*r/ r r) r� r Z
slot_tableZgenerate_dynamic_init_coder� re r r� rQ r� r% Zget_slot_code_by_namer� Zpyfunc_entriesr< r
r r� r� r) r* r+ r� r� Z vtable_cnameZ
vtabptr_cnameZ is_internalr� r r� r Zis_closure_class_scoperh r r* r r r r� )r� rW Zheap_type_basesr/ r r) r, Z readyfuncZgetattr_slot_funcZdictoffset_slot_funcZpy_cfuncZ utility_funcr\ r� r Z
weakref_entryZtp_weaklistoffsetZ objstructr& r& r' r+ � s�
z&CClassDefNode.generate_type_ready_codec C s6 | j r | j j | � | j r2 | j j | � d S)N)r* r� r� )r rW r& r& r' r� P s zCClassDefNode.annotate)rM rm rn r� r r r* r} r r r� r rp r� r� rq r� r� r� r+ r� r& r& r& r' r
� s$ � #�r
c @ sU e Z d Z d g Z d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
� PropertyNoder� c C sM | j | j | j | j � | _ | j | j j _ | j j | j j � d S)N) r( r< r- r% r� r� r) r� r� )r rC r& r& r' r� a s !z!PropertyNode.analyse_declarationsc C s | j j | � | _ | S)N)r� rq )r rC r& r& r' rq f s z PropertyNode.analyse_expressionsc C s | j j | | � d S)N)r� r� )r rC rW r& r& r' r� j s z*PropertyNode.generate_function_definitionsc C s d S)Nr& )r rW r& r& r' r� m s z$PropertyNode.generate_execution_codec C s | j j | � d S)N)r� r� )r rW r& r& r' r� p s zPropertyNode.annotateN) rM rm rn r� r� rq r� r� r� r& r& r& r' r/ W s r/ c @ s: e Z d Z g Z d d � Z d d � Z d d � Z d S)�
GlobalNodec C s+ x$ | j D] } | j | | j � q
Wd S)N)�namesZdeclare_globalr% )r rC r< r& r& r' r� { s z GlobalNode.analyse_declarationsc C s | S)Nr& )r rC r& r& r' rq s z GlobalNode.analyse_expressionsc C s d S)Nr& )r rW r& r& r' r� � s z"GlobalNode.generate_execution_codeN)rM rm rn r� r� rq r� r& r& r& r' r0 t s r0 c @ s: e Z d Z g Z d d � Z d d � Z d d � Z d S)� NonlocalNodec C s+ x$ | j D] } | j | | j � q
Wd S)N)r1 Zdeclare_nonlocalr% )r rC r< r& r& r' r� � s z!NonlocalNode.analyse_declarationsc C s | S)Nr& )r rC r& r& r' rq � s z NonlocalNode.analyse_expressionsc C s d S)Nr& )r rW r& r& r' r� � s z$NonlocalNode.generate_execution_codeN)rM rm rn r� r� rq r� r& r& r& r' r2 � s r2 c @ sg e Z d Z d g Z d d � Z d d � Z d d � Z d Z d d
� Z d d � Z d
d � Z
d S)r� r� c
C s@ d d l m } | j } t | | j � r� | j j � } | d k r<| j � \ } } t | � rt t | j
d � xd | j D]Y \ } } | j | � } | d k r� t | j
d � q~ | j
| j | | j
d d �q~ Wt | _ nU t | d d � d k r<| j r!| j | � t | _ n | j s3| j r<t | _ d S) Nr )r� r� z!Variable names must be specified.z Unknown typer TrB )r� r� r� rK ZGeneralCallNoderI �as_cython_attribute�explicit_args_kwdsrJ r r% r8 r: r, r9 � PassStatNoderL r� r; Zdeclare_from_annotation� is_attribute� is_subscript)
r rC r� r� r\ rT rU �varr$ r/ r& r& r' r� � s( #
z!ExprStatNode.analyse_declarationsc C s1 d | j _ | j j | � | _ d | j _ | S)NF)r� �result_is_usedrq )r rC r& r& r' rq � s z ExprStatNode.analyse_expressionsc C s) | j j j r% | j j r% | j � d S)N)r� r/ r= �is_tempr� )r rC r& r& r' r� � s zExprStatNode.nogil_checkz Discarding owned Python objectc C s� | j | j � d | j _ | j j | � | j j r� | j j � r� | j j � } | j j j sp d | } | j d | � | j j
| � | j j | � d S)NFz
(void)(%s)z%s;) r� r% r� r9 rB r: r� r/ r� rQ r rI )r rW r� r& r& r' r� � s
z$ExprStatNode.generate_execution_codec C s | j j | | � d S)N)r� r� )r rC rW r& r& r' r� � s z*ExprStatNode.generate_function_definitionsc C s | j j | � d S)N)r� r� )r rW r& r& r' r� � s zExprStatNode.annotateN) rM rm rn r� r� rq r� r� r� r� r� r& r& r& r' r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�AssignmentNodec C s` | j | � } t | t � r\ t | t � r\ | j j j r\ | j j � r\ t | j d � | S)Nz9Storing unsafe C derivative of temporary Python reference)
rp rK r; �ParallelAssignmentNode�rhsr/ r� Z is_ephemeralr r% )r rC rV r& r& r' rq � s
z"AssignmentNode.analyse_expressionsc C s. | j | j � | j | � | j | � d S)N)r� r% � generate_rhs_evaluation_coderL )r rW r& r& r' r� � s
z&AssignmentNode.generate_execution_codeN)rM rm rn rq r� r& r& r& r' r; � s r; c @ s� e Z d Z d d g Z d Z d Z d Z d d � Z d d d � Z d d
� Z d d � Z
d
d � Z d d � Z d d � Z
d d d � Z d d � Z d d � Z d S)�SingleAssignmentNoder r= Fc C s- d d l m } t | j | j � r | j j j � } | r | j j � \ } } | d k rHt | � d k r� t | d j
d � d S| d k r� | j d � } | d k s� d | k r� t | j
d � d S| d } n d } | d
j | � } | d k rt | d
j
d � d S| j
} | d k r�t | | j � rX| j | j
f g }
n? t | | j � r�d d
� | j D� }
n t | j
d � d Sx3 |
D]+ \ } } | j | | | d d d | �q�Wt | � d k r�| d | _ qEd | _ q d | _ t | | j � s&t | j
d � | j | j | | j
d d �n�| d k r�d | _ t | � d
k s{| d k r�t | j j
d � d Sg }
xa | j D]V \ } } | j | � } | d k r�t | j
d � q�|
j | j | | j
f � q�Wt |
� t | j � k rd St | j
| j � sCt | j
j
d � | j
j } t | � } | j | | | d | j j
� x� |
D]" \ } } } | j | | | � q�Wnb | d k r d | _ | r�t | j j j
d � t | j j
d | j
j d | �} | j | � | j rd S| j
j | � d S) Nr )r� r� �typedefr" z Invalid positional argument.r� zInvalid keyword argument.rQ r z Unknown typec S s" g | ] } | j | j f � q Sr& )r< r% )r� r8 r& r& r' r� s z=SingleAssignmentNode.analyse_declarations.<locals>.<listcomp>zInvalid declarationr TzInvalid declaration.r( �unionz.Struct or union members must be given by name.F�
fused_typez*fused_type does not take keyword argumentsr< rb )r� r@ )r( rA ) r� r� rK r= ZCallNoderI r3 r4 rJ r r% �compile_time_valuer: r r� r< r� rT r, �declaration_onlyry r8 r r9 r r. rx r� r
)r rC r� � func_namerT rU Zkwdictr� r/ r �varsr8 r% �members�memberr$ r< r) Z fusednoder& r& r' r� s�
# "
z)SingleAssignmentNode.analyse_declarationsr c C sp d d l m } | j j | � | _ | j | � } | r> | S| j j | � | _ | j j | � | j | � } | r| | St | j | j
� r� | j j | j � | j j | j � | _ nf | j j
j r%t | j | j � s%| j | j j d | j d d d d �| _ | j j | � | _ | j j
j r�| j | j d | j j
| j j
g � } | r�| j } d | _ | j
j | _ | j
j | _ | j d k r�| j d k r�| j t j d d
� � q| j j | j j
| � } n | j j | j j
| � } | s6| j s6| j rH| j rH| j
j rH| j | � } n | j
j rc| j | � } | | _ | S) Nr )r� r� rY �stop�=Tr ZCppExceptionConversionzCppSupport.cpp) r� r� r= rp �
unroll_rhsr rr �gil_assignment_check�
unroll_lhsrK ZMemoryViewIndexNodeZanalyse_broadcast_operationZ$analyse_as_memview_scalar_assignmentr/ r� �SliceIndexNoder% r� Zlookup_operator_for_types�is_overloaded_assignmentr
r� re r r� r r6 r; r� r= �coerce_to_temp�coerce_to_simple)r rC �use_tempr� Zunrolled_assignment�opr= r& r& r' rp W sH -* z"SingleAssignmentNode.analyse_typesc C sy d d l m } m } | } d } } } }
| j j rI | j j } n�| j j sa | j j r8x6 t | | j � r� | j
p� | j r� | j } } qd Wt | | j � r| j } | j
} | r� | j
t j | � } | j } | r | j
t j | � } n} | j j ri| j j ri| j | j d t | j j � d t | j j t � rZ| j j n | j �} n t | j d � d Sd } | r�| j
t j | � } d d � } y- | | d � | | d � | | d � } Wq5t k
r�t | j d � d SYq5Xq<| j j r1| j j } t | t � s5d Sq<d Sn d S| | k rft | j d
| | f � d Sg }
| j | � } | g } | r�| j r�| j | � } | j | � | r�| j r�| j | � } | j | � | r| j r| j | � } | j | � xat | � D]S} | j | j d t | � d | d t j �} | d k r�| j � r�| j | j } | j | j d t | � d | �} n$ | j | j d d
d | d | �} | d k r:| j � r| j � r| j | j } | j | j d t | � d | �} n$ | j | j d d d | d | �} |
j | j | j d | d | j | � �� qW|
| |
f S)Nr )r� � UtilNodesr9 r~ z*C array iteration requires known end indexc S s3 | d k r | S| j � r# | j St d � � d S)NzNot a constant.)�has_constant_resultr~ r� )rV Z
none_valuer& r& r' � get_const� s
z.SingleAssignmentNode.unroll.<locals>.get_constr z5C array assignment currently requires known endpointsz=Assignment to/from slice of wrong length, expected %s, got %sr/ �operatorr� Zoperand1Zoperand2r r� r� )!r� r� rT r/ � is_ctupler� r� r� rK rN rY rI r� r r
r� ZIntNoder% r r Zconstant_value_not_setr r� Z
LetRefNoder� r r_ rU r~ ZMulNodeZAddNodeZ IndexNoderp )r rV Z target_sizerC r� rT r� Z
start_nodeZ stop_nodeZ step_node�
check_nodeZ
slice_sizerV re �refsr� Zix_nodeZ
step_valueZ index_valuer& r& r' �unroll� s� ( ! -
- '$ ' 2zSingleAssignmentNode.unrollc
C s� d d l m } g } xH t | | � D]7 \ } } | j t | j d | d | d | j �� q& Wt d | j d | � j | � }
| r� t d | j d | |
g � }
x- | d d d � D] } | j
| |
� }
q� W|
S) Nr )rT r r= �firstr% r� r� ) r� rT r8 r r? r% r\ r< rq r� ZLetNode) r rZ rY �lhs_listZrhs_listrC rT � assignmentsr r= rV r� r& r& r' �unroll_assignments� s /! z'SingleAssignmentNode.unroll_assignmentsc C s� d d l m } t | j | j � s) d St d d � | j j D� � rL d S| j | j t | j j � | � } | sz d S| \ } } } | j
| | | j j | | � S)Nr )r� c s s | ] } | j Vq d S)N)Z
is_starred)r� r2 r& r& r' r� � s z2SingleAssignmentNode.unroll_rhs.<locals>.<genexpr>) r� r� rK r r� �anyrT r[ r= rJ r_ )r rC r� �unrolledrY rZ r= r& r& r' rK � s $z SingleAssignmentNode.unroll_rhsc C s� | j j j r d Sd d l m } t | j | j � s< d S| j | j t | j j
� | � } | sj d S| \ } } } | j | | | | j j
| � S)Nr )r� ) r r/ rX r� r� rK r= r� r[ rJ rT r_ )r rC r� ra rY rZ r r& r& r' rM � s $z SingleAssignmentNode.unroll_lhsc C s | j j | � d S)N)r= rB )r rW r& r& r' r> s z1SingleAssignmentNode.generate_rhs_evaluation_codec C sW | j r= | j j | j | d | j d | j d | j �n | j j | j | � d S)NrK r
r� )rO r rL r= r
r� )r rW rK r& r& r' rL s
z-SingleAssignmentNode.generate_assignment_codec C s | j j | | � d S)N)r= r� )r rC rW r& r& r' r� s z2SingleAssignmentNode.generate_function_definitionsc C s$ | j j | � | j j | � d S)N)r r� r= )r rW r& r& r' r� s z SingleAssignmentNode.annotateN)rM rm rn r� r\ rO rD r� rp r[ r_ rK rM r> rL r� r� r& r& r& r' r? � s T0 ^
r? c @ s� e Z d Z d d d d g Z d Z d Z d Z d d � Z d d d
� Z d d � Z d
d d � Z
d d � Z d d � Z d S)�CascadedAssignmentNoder] r= �coerced_values�
cloned_valuesNc C s% x | j D] } | j | � q
Wd S)N)r] r
)r rC r r& r& r' r� 5 s z+CascadedAssignmentNode.analyse_declarationsr c C s� d d l m } m } t � } xT t | j � D]C \ } } | j | � } | j | <| j | � | j | j � q/ W| j
j | � } t | � d k rt
t | � � j r� | j d | | j
g � } | s| j | j � | � } n | j | j � | � } | j rD| j rD| s2| j s2| j j rD| j | � } n | j | � } | j rh| | � n | | _
g | _ i }
g | _ x� | j D]� } | j j o�| j d | | j
g � } | j j | � | j |
k r�| j | j k r�| | j
� } | s| j | j | � } | j j | � | |
| j <q�Wg | _ x? | j D]4 } |
j | j | j
� } | j j | | � � qOW| S)Nr )� CloneNoder rJ ) r� re r rz r r] rr rL r� r/ r= rp rJ �next�iterr� Zlookup_operatorr r r; r� r6 r= rP rQ r: rc �assignment_overloadsr rd r* ) r rC rR re r Z lhs_typesr" r r= rS rc Z
overloadedr& r& r' rp 9 sF
'! z$CascadedAssignmentNode.analyse_typesc C s | j j | � d S)N)r= rB )r rW r& r& r' r> j s z3CascadedAssignmentNode.generate_rhs_evaluation_codeFc C s� x | j D] } | j | � q
WxO t | j | j | j � D]2 \ } } } | j | � | j | | d | �q= Wx+ | j D] } | j | � | j | � q} W| j j | � | j j | � d S)NrK )
rc rB r8 r] rd rh rL r rI r= )r rW rK r= r Zoverload� rhs_valuer& r& r' rL m s +
z/CascadedAssignmentNode.generate_assignment_codec C s | j j | | � d S)N)r= r� )r rC rW r& r& r' r� | s z4CascadedAssignmentNode.generate_function_definitionsc C su x | j D] } | j | � q
Wx= t | j | j � D]& \ } } | j | � | j | � q7 W| j j | � d S)N)rc r� r8 r] rd r= )r rW r= r r& r& r' r� s "
z CascadedAssignmentNode.annotate)
rM rm rn r� rd rc rh r� rp r> rL r� r� r& r& r& r' rb " s 1 rb c @ sU e Z d Z d g Z d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
r< r� c C s% x | j D] } | j | � q
Wd S)N)r� r� )r rC r� r& r& r' r� � s z+ParallelAssignmentNode.analyse_declarationsc s# � f d d � | j D� | _ | S)Nc s% g | ] } | j � d d �� q S)rR r )rp )r� r� )rC r& r' r� � s z>ParallelAssignmentNode.analyse_expressions.<locals>.<listcomp>)r� )r rC r& )rC r' rq � s z*ParallelAssignmentNode.analyse_expressionsc C sV | j | j � x | j D] } | j | � q Wx | j D] } | j | � q; Wd S)N)r� r% r� r> rL )r rW r� r& r& r' r� � s
z.ParallelAssignmentNode.generate_execution_codec C s( x! | j D] } | j | | � q
Wd S)N)r� r� )r rC rW r� r& r& r' r� � s z4ParallelAssignmentNode.generate_function_definitionsc C s% x | j D] } | j | � q
Wd S)N)r� r� )r rW r� r& r& r' r� � s z ParallelAssignmentNode.annotateN) rM rm rn r� r� rq r� r� r� r& r& r& r' r< � s r< c @ sX e Z d Z d d g Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�InPlaceAssignmentNoder r= c C s | j j | � d S)N)r r
)r rC r& r& r' r� � s z*InPlaceAssignmentNode.analyse_declarationsc C s� | j j | � | _ | j j | � | _ | j j sB | j j rc | j j | j j | � | _ n9 | j j j r� | j d k r� | j j t
j | � | _ | S)Nz+-) r= rp r rr �is_memview_index�is_buffer_accessr r/ � is_stringrW r
r� )r rC r& r& r' rp � s ! z#InPlaceAssignmentNode.analyse_typesc C sn | j | j � | j | j } } | j | � | j | � | j } | d k r[ d } n | d k rw t | j d � | j s� | j r� | j
j r� t | j d � | d k r� | j
j r� | j
j d r� t | j d � | j | | | � nB | j rt | j d � n& | j d
| j � | | j � f � | j | � | j | � | j | � | j | � d S) Nz//�/z**zNo C inplace power operatorzAIn-place operators not allowed on object buffers in this release.�%� cdivisionz;In-place non-c divide operators not allowed on int buffers.z4Inplace operators not supported on memoryview slicesz
%s %s= %s;)rn ro )r� r% r r= rB Z generate_subexpr_evaluation_coderW r rl rk r/ r= r� r� r� Z generate_buffer_setitem_codeZis_memview_slicerQ r� Z generate_subexpr_disposal_codeZfree_subexpr_tempsr rI )r rW r r= Zc_opr& r& r' r� � s,
) &
z-InPlaceAssignmentNode.generate_execution_codec C s$ | j j | � | j j | � d S)N)r r� r= )r rW r& r& r' r� � s z InPlaceAssignmentNode.annotatec C s2 d d l m } | j | j | j | j | j � S)Nr )r� )r� r� Z
binop_noder% rW r r= )r r� r& r& r' �create_binop_node� s z'InPlaceAssignmentNode.create_binop_nodeN) rM rm rn r� r� rp r� r� rq r& r& r& r' rj � s rj c @ s[ e Z d Z d d g Z d d � Z e j Z d Z d d � Z d d � Z
d
d � Z d S)
�
PrintStatNode� arg_tuplerO c C s� | j r- | j j | � } | j | � | _ | j j | � } | j | � | _ | j t � t | j j � d k r� | j r� | j t � | S)Nr )
rO rq �coerce_to_pyobjectrs re �printing_utility_coderJ rT �append_newline�printing_one_utility_code)r rC rO rs r& r& r' rq s
!
z!PrintStatNode.analyse_expressionszPython print statementc C sZ | j | j � | j r; | j j | � | j j � } n d } t | j j � d k r� | j r� | j j d } | j | � | j d | | j � | j
| j � f � | j | � | j | � ne | j j | � | j d | | j j � | j | j
| j � f � | j j | � | j j | � | j rV| j j | � | j j | � d S)Nrb r r z"if (__Pyx_PrintOne(%s, %s) < 0) %sz#if (__Pyx_Print(%s, %s, %d) < 0) %s)
r� r% rO rB � py_resultrJ rs rT rv rQ r� r rI )r rW Z
stream_resultr2 r& r& r' r� s6 !
z%PrintStatNode.generate_execution_codec C s3 | j r | j j | | � | j j | | � d S)N)rO r� rs )r rC rW r& r& r' r� 6 s z+PrintStatNode.generate_function_definitionsc C s- | j r | j j | � | j j | � d S)N)rO r� rs )r rW r& r& r' r� ; s zPrintStatNode.annotateN) rM rm rn r� rq r| r� r� r� r� r� r� r& r& r& r' rr � s ! rr c @ sL e Z d Z d g Z d d � Z e j Z d Z d d � Z d d � Z
d S)
� ExecStatNoderT c C s_ xH t | j � D]7 \ } } | j | � } | j | � } | | j | <q W| j t j � | S)N)r rT rq rt re r Zpyexec_utility_code)r rC r" r2 r& r& r' rq H s z ExecStatNode.analyse_expressionszPython exec statementc C s" | j | j � g } x1 | j D]& } | j | � | j | j � � q Wt | d d g d d t | � � � } | j j t
j d d �} | j d | f | � x+ | j D] } | j
| � | j | � q� W| j | j | | j � � | j | � | j | t � | j j | � d S)Nrb r$ r� Tz %s = __Pyx_PyExec3(%s, %s, %s);)r� r% rT rB r rx r� rJ r� r� r
r rQ r rI r� r� r� r� )r rW rT r2 Z temp_resultr& r& r' r� S s
*
z$ExecStatNode.generate_execution_codec C s% x | j D] } | j | � q
Wd S)N)rT r� )r rW r2 r& r& r' r� e s zExecStatNode.annotateN) rM rm rn r� rq r| r� r� r� r� r� r& r& r& r' ry A s ry c @ sa e Z d Z d g Z d Z d d � Z d d � Z d d � Z d Z d
d � Z d d
� Z
d S)� DelStatNoderT Fc C s% x | j D] } | j | � q
Wd S)N)rT r
)r rC r2 r& r& r' r� r s z DelStatNode.analyse_declarationsc C s x� t | j � D]� \ } } | j | d � } | j | <| j j sZ | j r� | j j r� | j r� | j j r� t | j
d � q | j j r� | j j j
r� | j | � q | j j
r� t | j
d � q | j r� | j j t j k r� q t | j
d � q W| S)Nz Deletion of global C variablez Deletion of non-heap C++ objectz&Deletion of non-Python, non-C++ object)r rT r r/ r= r; r� r� Z
is_cglobalr r% r� rE r� r� r7 r� r �bytearray_type)r rC r" r2 r& r& r' rq v s ! z DelStatNode.analyse_expressionsc C s. x' | j D] } | j j r
| j � q
Wd S)N)rT r/ r= r� )r rC r2 r& r& r' r� � s zDelStatNode.nogil_checkzDeleting Python objectc C s� | j | j � x� | j D]� } | j j sV | j j sV | j ro | j j t j k ro | j
| d | j �q | j j r | j j
j r | j | � | j d | j � � | j | � q Wd S)N�ignore_nonexistingz
delete %s;)r� r% rT r/ r= r� r7 r� r r{ Zgenerate_deletion_coder| r� rE r� rB rQ r� r )r rW r2 r& r& r' r� � s
z#DelStatNode.generate_execution_codec C s% x | j D] } | j | � q
Wd S)N)rT r� )r rW r2 r& r& r' r� � s zDelStatNode.annotateN) rM rm rn r� r| r� rq r� r� r� r� r& r& r& r' rz j s rz c @ s. e Z d Z g Z d d � Z d d � Z d S)r5 c C s | S)Nr& )r rC r& r& r' rq � s z PassStatNode.analyse_expressionsc C s d S)Nr& )r rW r& r& r' r� � s z$PassStatNode.generate_execution_codeN)rM rm rn r� rq r� r& r& r& r' r5 � s r5 c s( e Z d Z d Z � f d d � Z � S)�IndirectionNodez�
This adds an indirection so that the node can be shared and a subtree can
be removed at any time by clearing self.stats.
c s' t t | � j | d j d | �d S)Nr r� )rf r} r� r% )r r� )rL r& r' r� � s zIndirectionNode.__init__)rM rm rn ry r� r& r& )rL r' r} � s r} c @ s4 e Z d Z g Z d Z d d � Z d d � Z d S)�
BreakStatNodeTc C s | S)Nr& )r rC r& r& r' rq � s z!BreakStatNode.analyse_expressionsc C s@ | j | j � | j s, t | j d � n | j | j � d S)Nz break statement not inside loop)r� r% � break_labelr r� )r rW r& r& r' r� � s z%BreakStatNode.generate_execution_codeN)rM rm rn r� r� rq r� r& r& r& r' r~ � s r~ c @ s4 e Z d Z g Z d Z d d � Z d d � Z d S)�ContinueStatNodeTc C s | S)Nr& )r rC r& r& r' rq � s z$ContinueStatNode.analyse_expressionsc C sA | j s t | j d � d S| j | j � | j | j � d S)Nz"continue statement not inside loop)�continue_labelr r% r� r� )r rW r& r& r' r� � s
z(ContinueStatNode.generate_execution_codeN)rM rm rn r� r� rq r� r& r& r& r' r� � s r� c @ s e Z d Z d g Z d Z d Z d Z d Z d d � Z d d � Z d Z
d d
� Z d d � Z d
d � Z
d d � Z d S)rL r9 TFc C s� | j } | | _ | s, t | j d � | S| j r� | j rN t | j d � | j j | � | _ | j su | j r� t | j j d � q� | j j | j | � | _ n. | j r� | j r� | j r� t | j d � | S)Nz!Return not inside a function bodyz$Return with value in async generatorz"Return with value in void functionzReturn value required)
r r r% r9 � in_async_genrp r� r� r r= )r rC r r& r& r' rq � s"
z"ReturnStatNode.analyse_expressionsc C s | j j r | j � d S)N)r r= r� )r rC r& r& r' r� � s zReturnStatNode.nogil_checkzReturning Python objectc
C sK | j | j � | j s d S| j } | j j r] | j t j | j � | r] | j r] d } | rn| j | � | j j
r� d d l m } | j
d t j d | j d | j d | d | d | j � n� | j r | j j t j d d
� � | j d t j | j � f � | j | � n3 | j | � | j d t j | j | j � f � | j | � | j | � n� | j j r�| j r�| j r�| j j t j d
d
� � | j d � | j d t j � q| j t j | j � n" | j j r| j | | j j � x- | j! j" � D] \ } } | j# | | � qW| j$ | j% � d S)Nr )ra Z lhs_cnameZlhs_typeZlhs_posr= rW r� ZReturnWithStopIterationz Coroutine.cz-%s = NULL; __Pyx_ReturnWithStopIteration(%s);z%s = %s;r� z/PyErr_SetNone(__Pyx_PyExc_StopAsyncIteration); z
%s = NULL;)&r� r% r r9 r= r� r r� r� rB r� r� ra Zput_acquire_memoryviewsliceZin_nogil_context� in_generatorr� re r r� rQ rx r rE rF rH rI r� r� r� r� �
put_returnrA r� Ztemps_holding_referencer� r� r� )r rW r9 ra r r/ r& r& r' r� sZ
z&ReturnStatNode.generate_execution_codec C s4 | j r | j d � | j d t j | f � d S)Nz%#pragma omp critical(__pyx_returning)z%s = %s;)� in_parallel� putln_openmprQ r r� )r rW r9 r& r& r' r� ; s
zReturnStatNode.put_returnc C s& | j d k r" | j j | | � d S)N)r9 r� )r rC rW r& r& r' r� @ s z,ReturnStatNode.generate_function_definitionsc C s | j r | j j | � d S)N)r9 r� )r rW r& r& r' r� D s zReturnStatNode.annotateN)rM rm rn r� r� r� r� r� rq r� r� r� r� r� r� r& r& r& r' rL � s 9 rL c @ sg e Z d Z d d d d g Z d Z d d � Z e j Z d Z d d
� Z
d d � Z d
d � Z d S)�
RaiseStatNode�exc_type� exc_value�exc_tb�causeTc C sp | j r- | j j | � } | j | � | _ | j rZ | j j | � } | j | � | _ | j r� | j j | � } | j | � | _ | j r� | j j | � } | j | � | _ d | _ | j rl| j rl| j rl| j } d d l m } t | | j
� r3| j p&| j d k o&| j j r3| j
} | j rl| j j rl| j | _ | j d k rld | _ | S)Nr )r� � MemoryError)r� rp rt r� r� r� �builtin_exc_namer� r� rK rK rT rs rI r; r� �
is_builtinr< )r rC r� r� r� r� �excr� r& r& r' rq T s0 % z!RaiseStatNode.analyse_expressionszRaising exceptionc C s� | j | j � | j d k r@ | j d | j | j � � d S| j r� | j j | � | j j � } | j j r� | j j
| j j � n d } | j r� | j j | � | j j � } n d } | j
r� | j
j | � | j
j � } n d } | j r | j j | � | j j � } n d } | j j t � | j d | | | | f � xF | j | j | j
| j f D]& } | rr| j | � | j | � qrW| j | j | j � � d S)Nr� zPyErr_NoMemory(); %srb z __Pyx_Raise(%s, %s, %s, %s);)r� r% r� rQ r� r� rB rx r; r� Zuse_entry_utility_coder� r� r� r� re �raise_utility_coder rI )r rW Z type_codeZ
value_codeZtb_codeZ
cause_coder" r& r& r' r� r sF %
z%RaiseStatNode.generate_execution_codec C s� | j d k r" | j j | | � | j d k rD | j j | | � | j d k rf | j j | | � | j d k r� | j j | | � d S)N)r� r� r� r� r� )r rC rW r& r& r' r� � s z+RaiseStatNode.generate_function_definitionsc C sh | j r | j j | � | j r2 | j j | � | j rK | j j | � | j rd | j j | � d S)N)r� r� r� r� r� )r rW r& r& r' r� � s zRaiseStatNode.annotateN)
rM rm rn r� r� rq r| r� r� r� r� r� r� r& r& r& r' r� I s *
r� c @ sC e Z d Z g Z d Z d d � Z e j Z d Z d d � Z
d S)�ReraiseStatNodeTc C s | S)Nr& )r rC r& r& r' rq � s z#ReraiseStatNode.analyse_expressionszRaising exceptionc C s | j | j � | j j } | r� | j j t � | j | d � | j | d � | j | d � | j d t
| � � x | D] } | j d | � q� W| j � | j | j | j � � n9 | j j t
j d d � � | j d | j | j � � d S) Nr r r" z&__Pyx_ErrRestoreWithState(%s, %s, %s);z%s = 0; ZReRaiseExceptionz Exceptions.cz __Pyx_ReraiseException(); %s)r� r% r� �exc_varsr� re r rG r rQ r� r� r� r r� )r rW rF Zvarnamer& r& r' r� � s
z'ReraiseStatNode.generate_execution_codeN) rM rm rn r� r� rq r| r� r� r� r� r& r& r& r' r� � s r� c @ s[ e Z d Z d d g Z d d � Z e j Z d Z d d � Z d d � Z
d
d � Z d S)
�AssertStatNoder r9 c C s� | j j | � | _ | j r� | j j | � } | j t j k sO | j j r� d d l m } | | j
d | g d d �} | j | d d �j | � | _ n | j | � | _ | S)Nr )r� rT ZslowTZ
skip_children) r Zanalyse_boolean_expressionr9 rp r/ r Z
tuple_typer r� r� r% rt )r rC r9 r� r& r& r' rq � s $z"AssertStatNode.analyse_expressionszRaising exceptionc C s | j d � | j d � | j | j � | j j | � | j d | j j � � | j r� | j j | � | j d | j j � � | j j | � | j j | � n
| j d � | j | j
| j � � | j d � | j j | � | j j | � | j d � | j d � d S)Nz!#ifndef CYTHON_WITHOUT_ASSERTIONSz!if (unlikely(!Py_OptimizeFlag)) {zif (unlikely(!%s)) {z*PyErr_SetObject(PyExc_AssertionError, %s);z$PyErr_SetNone(PyExc_AssertionError);r� z#endif) rQ r� r% r rB r� r9 rx r rI r� )r rW r& r& r' r� � s.
z&AssertStatNode.generate_execution_codec C s9 | j j | | � | j d k r5 | j j | | � d S)N)r r� r9 )r rC rW r& r& r' r� s z,AssertStatNode.generate_function_definitionsc C s- | j j | � | j r) | j j | � d S)N)r r� r9 )r rW r& r& r' r� s zAssertStatNode.annotateN) rM rm rn r� rq r| r� r� r� r� r� r� r& r& r& r' r� � s
r� c @ sg e Z d Z d d g Z d d � Z d d � Z d d � Z d d
d � Z d d
� Z d d � Z d S)�
IfStatNode�
if_clauses� else_clausec C s> x | j D] } | j | � q
W| j r: | j j | � d S)N)r� r� r� )r rC � if_clauser& r& r' r� s z IfStatNode.analyse_declarationsc sA � f d d � | j D� | _ | j r= | j j � � | _ | S)Nc s g | ] } | j � � � q Sr& )rq )r� r� )rC r& r' r� s z2IfStatNode.analyse_expressions.<locals>.<listcomp>)r� r� rq )r rC r& )rC r' rq s z IfStatNode.analyse_expressionsc C s | j | j � | j � } t | j � } | j rW | j | j d | j d d �n
| d 8} xL t | j � D]; \ } } | j | | j � | j | | d | | k �qq W| j r� | j | j j � | j
d � | j j | � | j
d � | j | � d S)Nr �inverseT�is_lastz
/*else*/ {r� r� ) r� r% r� rJ r� r� �_set_branch_hintr r� r� rQ r� )r rW r� �lastr" r� r& r& r' r� s #
z"IfStatNode.generate_execution_codeFc C s� | j s
d St | t � s' | j r+ d St | j d t t f � r� t | j � d k r� t t t t
t f } x. | j d d � D] } t | | � s� d Sq� W| r� d n d | _ d S)Nr r� r r� r� )
r� rK r� r� r� r� rJ r� r; rz r0 r2 � branch_hint)r �clauseZstatements_noder� Znon_branch_nodesrV r& r& r' r� 3 s zIfStatNode._set_branch_hintc C sJ x! | j D] } | j | | � q
W| j d k rF | j j | | � d S)N)r� r� r� )r rC rW r� r& r& r' r� B s z(IfStatNode.generate_function_definitionsc C s> x | j D] } | j | � q
W| j r: | j j | � d S)N)r� r� r� )r rW r� r& r& r' r� H s zIfStatNode.annotateN)
rM rm rn r� r� rq r� r� r� r� r& r& r& r' r� s r� c @ s^ e Z d Z d d g Z d Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d S)� IfClauseNode� conditionr� Nc C s | j j | � d S)N)r� r� )r rC r& r& r' r� X s z!IfClauseNode.analyse_declarationsc C s. | j j | � | _ | j j | � | _ | S)N)r� � analyse_temp_boolean_expressionr� rq )r rC r& r& r' rq [ s z IfClauseNode.analyse_expressionsc C s� | j j | � | j | j � | j j � } | j rK d | j | f } | j d | � | j j | � | j j | � | j j
| � | j | j d d �| p� | j j s� | j | � | j d � d S)Nz%s(%s)z if (%s) {r� Fr� )
r� rB r� r% r� r� rQ r rI r� r� r� r� )r rW r� r� r� r& r& r' r� ` s
z$IfClauseNode.generate_execution_codec C s* | j j | | � | j j | | � d S)N)r� r� r� )r rC rW r& r& r' r� o s z*IfClauseNode.generate_function_definitionsc C s$ | j j | � | j j | � d S)N)r� r� r� )r rW r& r& r' r� s s zIfClauseNode.annotate)
rM rm rn r� r� r� rq r� r� r� r& r& r& r' r� O s r� c @ s@ e Z d Z d d g Z d d � Z d d � Z d d � Z d S)
�SwitchCaseNode�
conditionsr� c C s xE | j D]: } | j | j � | j | � | j d | j � � q
W| j j | � | j | j d d �| j d � d S)Nzcase %s:r� Fzbreak;)r� r� r% rB rQ r� r� r� )r rW r r& r& r' r� � s
z&SwitchCaseNode.generate_execution_codec C s; x! | j D] } | j | | � q
W| j j | | � d S)N)r� r� r� )r rC rW r r& r& r' r� � s z,SwitchCaseNode.generate_function_definitionsc C s5 x | j D] } | j | � q
W| j j | � d S)N)r� r� r� )r rW r r& r& r' r� � s zSwitchCaseNode.annotateN)rM rm rn r� r� r� r� r& r& r& r' r� x s r� c @ sC e Z d Z d d d g Z d d � Z d d � Z d d � Z d
S) �SwitchStatNode�test�casesr� c C s� | j j | � | j | j � | j d | j j � � x | j D] } | j | � qD W| j d k r� | j d � | j j | � | j d � n
| j d � | j d � d S)Nz
switch (%s) {zdefault:zbreak;zdefault: break;r� ) r� rB r� r% rQ r� r� r� r� )r rW �caser& r& r' r� � s
z&SwitchStatNode.generate_execution_codec C s] | j j | | � x! | j D] } | j | | � q W| j d k rY | j j | | � d S)N)r� r� r� r� )r rC rW r� r& r& r' r� � s
z,SwitchStatNode.generate_function_definitionsc C sT | j j | � x | j D] } | j | � q W| j d k rP | j j | � d S)N)r� r� r� r� )r rW r� r& r& r' r� � s
zSwitchStatNode.annotateN)rM rm rn r� r� r� r� r& r& r& r' r� � s r� c @ s e Z d Z d S)�LoopNodeN)rM rm rn r& r& r& r' r� � s r� c @ s[ e Z d Z d d d g Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d S)�
WhileStatNoder� r� r� c C s- | j j | � | j r) | j j | � d S)N)r� r� r� )r rC r& r& r' r� � s z"WhileStatNode.analyse_declarationsc C sU | j r | j j | � | _ | j j | � | _ | j rQ | j j | � | _ | S)N)r� r� r� rq r� )r rC r& r& r' rq � s z!WhileStatNode.analyse_expressionsc C s | j | j � | j � } | j d � | j r| | j j | � | j j | � | j d | j j � � | j j | � | j j
| � | j | j � | j d � | j
} | j | � | j r| j | j j � | j d � | j j
| � | j d � | j | � d S)Nz while (1) {zif (!%s) break;r� z
/*else*/ {)r� r% �new_loop_labelsrQ r� rB r r� rI r� r� r� r� r �set_loop_labelsr� )r rW �old_loop_labelsr r& r& r' r� � s*
z%WhileStatNode.generate_execution_codec C sU | j r | j j | | � | j j | | � | j d k rQ | j j | | � d S)N)r� r� r� r� )r rC rW r& r& r' r� � s
z+WhileStatNode.generate_function_definitionsc C sF | j r | j j | � | j j | � | j rB | j j | � d S)N)r� r� r� r� )r rW r& r& r' r� � s
zWhileStatNode.annotateN) rM rm rn r� r� rq r� r� r� r& r& r& r' r� � s r� c
@ s� e Z d Z d d d d d d d d d d
g
Z d Z Z d Z Z d Z Z d d
� Z
d d � Z d d � Z d d � Z
d S)�DictIterationNextNode�dict_obj�
expected_size�
pos_index_var�coerced_key_var�coerced_value_var�coerced_tuple_var�
key_target� value_target� tuple_target� is_dict_flagNc C sP t j | | j d | d | d | d | d | d | d | d d d
t j � d S) Nr� r� r� r� r� r� r� r: Tr/ )r| r� r% r
� c_bint_type)r r� r� r� r� r� r� r� r& r& r' r� s z DictIterationNextNode.__init__c C s� d d l m } | j j | � | _ | j j | � | _ | j rX | j j | � | _ | j r� | j j | � | _ | j | j j t
j � | _ | j j
| j j | � | _ | j r| j j | � | _ | j | j j d t
j �| _ | j j
| j j | � | _ | j ri| j j | � | _ | j | j j t
j � | _ | j j
| j j | � | _ | j j | � | _ | S)Nr )r� r/ )r� r� r� rp r� r� r� rr �TempNoder% r
r �key_refr r/ r� r� � value_refr� r� � tuple_refr� r� )r rC r� r& r& r' rq s& ! z)DictIterationNextNode.analyse_expressionsc C s | j j | | � d S)N)r� r� )r rC rW r& r& r' r� , s z3DictIterationNextNode.generate_function_definitionsc
C s' | j j t j d d � � | j j | � g } g } x� | j | j | j f | j | j
| j f | j | j
| j f g D]d \ } } } | d k r� d } n3 | j | | | f � | j | � d | j � } | j | � q~ W| j j t j d � } | j d | | j j � | j j � | j j � | d | d | d | j j � f � | j d
| � | j | j d | | j � � | j j | � x* | D]" \ } } } | j | j � � q�Wx$ | D] \ } } } | j | � q�Wx4 | D], \ } } } | j | | � | j | � q�Wd S) NZ dict_iterz
Optimize.crc z&%sFz7%s = __Pyx_dict_iter_next(%s, %s, &%s, %s, %s, %s, %s);r r r" z if (unlikely(%s == 0)) break;z%s == -1)!r� re r r� r� rB r� r� r� r� r� r� r� r� r� r �allocater� r� r� r
r? rQ rx r� r� r� �
error_goto_ifr% r� r� rL �release) r rW r^ Ztemp_addressesr8 r� rJ �addr� result_tempr& r& r' r� / sB (
z-DictIterationNextNode.generate_execution_code)rM rm rn r� r� r� r� r� r� r� r� rq r� r� r& r& r& r' r� � s
r� c @ sb e Z d Z d d d d d d g Z d Z Z d d � Z d
d � Z d d
� Z d d � Z d S)�SetIterationNextNode�set_objr� r� r� r� � is_set_flagNc C sD t j | | j d | d | d | d | d | d d d t j �d S) Nr� r� r� r� r� r: Tr/ )r| r� r% r
r� )r r� r� r� r� r� r& r& r' r� a s z SetIterationNextNode.__init__c C s� d d l m } | j j | � | _ | j j | � | _ | j j | � | _ | j j | � | _ | j | j j d t
j �| _ | j j
| j j | � | _ | j j | � | _ | S)Nr )r� r/ )r� r� r� rp r� r� r� rr r� r% r
r r� r r/ r� r� )r rC r� r& r& r' rq l s ! z(SetIterationNextNode.analyse_expressionsc C s | j j | | � d S)N)r� r� )r rC rW r& r& r' r� w s z2SetIterationNextNode.generate_function_definitionsc C s2 | j j t j d d � � | j j | � | j } | j | � | j j t
j d � } | j d | | j j
� | j j � | j j � | j � | j j � f � | j d | � | j | j d | | j � � | j j | � | j | j � � | j j | � | j j | j | � | j | � d S)NZset_iterz
Optimize.cFz/%s = __Pyx_set_iter_next(%s, %s, &%s, &%s, %s);z if (unlikely(%s == 0)) break;z%s == -1)r� re r r� r� rB r� r� r� r� r
r? rQ rx r� r� r� r� r� r% r� r� r� r� rL r� )r rW r� r� r& r& r' r� z s&
z,SetIterationNextNode.generate_execution_code)
rM rm rn r� r� r� r� rq r� r� r& r& r& r' r� Y s
r� c K sG d | k r6 | d j r& t | | � St | | � Sn
t | | � Sd S)N�iterator)�is_async�AsyncForStatNode�
ForInStatNode�ForFromStatNode)r% r� r& r& r' � ForStatNode� s
r� c @ sy e Z d Z d d d d d g Z d Z d Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d S)�_ForInStatNoderJ r� r� r� r� NFc C s t d � � d S)Nz!must be implemented by subclasses)�NotImplementedError)r r& r& r' �_create_item_node� s z _ForInStatNode._create_item_nodec C sG | j j | � | j j | � | j r9 | j j | � | j � d S)N)rJ r
r� r� r� r� )r rC r& r& r' r� � s
z#_ForInStatNode.analyse_declarationsc C s� | j j | � | _ | j j | � | _ | j � | j j | � | _ | j r� | j j j sq | j j j r� | j j j
| j j � r� n | j j | j j | � | _ | j j | � | _ | j
r� | j
j | � | _
| S)N)rJ rr r� rq r� r� r� r/ r� r� r r r� r� )r rC r& r& r' rq � s
z"_ForInStatNode.analyse_expressionsc C s� | j | j � | j � } | j j | � | j d � | j j | � | j j | j | � | j j
| � | j | j � | j | j � | j d � | j
} | j | � | j ri| j } | j d � | _ | j d � | j j
| � | j d � | j | j � r\| j | � | j | j � | j | j � | j j | � | j | � | j | � | j | j � | j | � r�| j | � | j j | � | j j | � d S)Nz
for (;;) {r� Zouter_continuez
/*else*/ {)r� r% r� r� rB rQ r� rJ rL r� r� r� r� r r� r� r� r� r� r rI )r rW r� r Zorig_continue_labelr& r& r' r� � s<
z&_ForInStatNode.generate_execution_codec C s_ | j j | | � | j j | | � | j j | | � | j d k r[ | j j | | � d S)N)rJ r� r� r� r� )r rC rW r& r& r' r� � s
z,_ForInStatNode.generate_function_definitionsc C s] | j j | � | j j | � | j j | � | j rI | j j | � | j j | � d S)N)rJ r� r� r� r� r� )r rW r& r& r' r� � s z_ForInStatNode.annotate) rM rm rn r� r� r� r� r� rq r� r� r� r& r& r& r' r� � s
( r� c @ s" e Z d Z d Z d d � Z d S)r� Fc C s& d d l m } | | j � | _ d S)Nr )�NextNode)r� r� r� r� )r r� r& r& r' r� s z ForInStatNode._create_item_nodeN)rM rm rn r� r� r& r& r& r' r� s r� c @ s. e Z d Z d Z d d � Z d d � Z d S)r� Tc K sY d | k s t � d d l m } | j | d j d d �| d <t j | | | � d S)Nr� r )r� r� r2 )r r� r� ZAwaitIterNextExprNoder% r� r� )r r% r� r� r& r& r' r� s zAsyncForStatNode.__init__c C s, d d l m } | j | j � | j _ d S)Nr )r� )r� r� Z
AsyncNextNoder� r� r2 )r r� r& r& r' r� s z"AsyncForStatNode._create_item_nodeN)rM rm rn r� r� r� r& r& r& r' r�
s r� c @ s� e Z d Z d d d d d d g Z d Z d Z d Z d Z d Z d
d � Z d d
� Z
d d � Z d d � Z d d � Z
d d! d d" d d# d d$ i Z d d � Z d d � Z d S)%r� rJ �bound1�bound2�stepr� r� FNz&For-loop using object bounds or targetc C s= x6 | j | j | j f D] } | j j r | j � q Wd S)N)rJ r� r� r/ r= r� )r rC r� r& r& r' r� 9 s zForFromStatNode.nogil_checkc C s= | j j | � | j j | � | j r9 | j j | � d S)N)rJ r
r� r� r� )r rC r& r& r' r� > s z$ForFromStatNode.analyse_declarationsc C s d d l m } | j j | � | _ | j j | � | _ | j j | � | _ | j d k r� t | j | j � r� t
| j j d d � | j j | � | _ | j | � | j j
} | j p� | j s� t | j j d � | j j | � | _ | j r| j j | � | _ | S)Nr )r� zdProbable infinite loop in for-from-by statement. Consider switching the directions of the relations.r" z>for-from loop variable must be c numeric type or Python object)r� r� rJ rr r� rp r� r� rK ZUnaryMinusNoder r% � set_up_loopr/ r= r[ r r� rq r� )r rC r� r� r& r& r' rq D s"
z#ForFromStatNode.analyse_expressionsc C s& d d l m } | j j } | j r. | } n� | j rJ t | j j d � | j r\ t j
n t j } | j j j s� t j
| | j j � } | j j j s� t j
| | j j � } | j d k r� | j j j r� t j
| | j j � } | j j | | � | _ | j j | | � | _ | j j s;| j j | � | _ | j d k r�| j j | | � | _ | j j s�| j j | � | _ | j s�| j r�d | _ t | j | j � r�t | j d � � | j | _ d | _ nE d | _ | j | j | | � } | | _ | j | � j | � | _ d S)Nr )r� zWInteger loops over enum values are fragile. Please cast to a safe integer type instead.FzEBuffer or memoryview slicing/indexing not allowed as for-loop target.T) r� r� rJ r/ r[ �is_enumr r% r= r
r> r? r� �widest_numeric_typer� r� r r� rP � is_py_targetrK ZBufferIndexNoder � loopvar_node�py_loopvar_noder� re rt )r rC r� r� Z loop_typeZc_loopvar_noder& r& r' r� Y s@ zForFromStatNode.set_up_loopc C s� | j | j � | j � } | j } | j j | � | j j | � | j | j \ } } | j d k r� | j j | � | j j
� } d | d | f } n d } d d l m } t
| j | j � r� | j j | � t
| j | j � r | j j | � | j j j rt j n | j j } | rL| j rL| j j | d � } n | j j
� } | j r�| j r�| j d d k r�| j d | | j j
� | | | | j | j j
� | | | f
� n> | j d | | j j
� | | | j | j j
� | | f � | j }
|
d k r<| r<| j | j j | | � }
|
d k rh|
j | � | j j |
| � | j j! | � | j" | j# � | r| j r| j j$ j% r�| j& | j j d � } | j | � | j' | j j$ j( � } | j j$ j) j* r| j+ j, t- j. d
d � � d }
n2 | j+ j, t- j. d
d � � d | j j$ j) j/ }
| j d | j
� |
| | j0 | j
� | j j � f � | j1 | j
� � n | j } | j2 | j j | | j j$ j) � } | | _3 | j4 | � | j j$ j% r| j5 | j
� | j � | j6 | � | j d � | rZ| j rZ| j j | � | j j | j | � | rz| j rz| j j7 | � | j8 } | j9 | � | j: r�| j d � | j: j! | � | j d � | j" | � | j j; | � | j j< | � | j j; | � | j j< | � t
| j | j � r5| j j6 | � t
| j | j � rZ| j j6 | � | j d k r�| j j; | � | j j< | � d S)Nz%s=%sr �1r )r� F�>z-for (%s = %s%s + %s; %s %s %s + %s; ) { %s%s;z!for (%s = %s%s; %s %s %s; %s%s) {ZGetModuleGlobalNamezObjectHandling.cz __Pyx_GetModuleGlobalName(%s)ZGetNameInClassz __Pyx_GetNameInClass(%s, %%s)z %s = %s; %sr� z
/*else*/ {)=r� r% r� �
from_ranger� rB r� �relation_tableZ relation1r� r� r� r� rK r� r� r� r� rJ r/ r� r
r> r� r� r� r� rV Z relation2rQ r� rL r� r� r� r� r� Z is_pyglobalZ
PyTempNoder� r< r) r� r� re r r� r r� r� ZCoerceFromPyTypeNodeZ temp_code�generate_result_coder
r� r� r r� r� r rI )r rW r� r� �offsetZincopr� r� Z loopvar_typeZ loopvar_nameZcoerced_loopvar_nodeZ target_nodeZinterned_cnameZ lookup_funcZ from_py_noder r& r& r' r� s� $&
#
z'ForFromStatNode.generate_execution_codez<=r� �++r� �+1z>=�--r� �-1c C s� | j j | | � | j j | | � | j j | | � | j d k r[ | j j | | � | j j | | � | j d k r� | j j | | � d S)N)rJ r� r� r� r� r� r� )r rC rW r& r& r' r� � s z-ForFromStatNode.generate_function_definitionsc C sv | j j | � | j j | � | j j | � | j rI | j j | � | j j | � | j rr | j j | � d S)N)rJ r� r� r� r� r� r� )r rW r& r& r' r� s zForFromStatNode.annotate)r� r� )r� r� )r� r� )r� r� )rM rm rn r� r� r� r� r� r� r� r� rq r� r� r� r� r� r& r& r& r' r� s" & s
r� c @ sd e Z d Z d Z d d d d g Z d Z d Z d d � Z d d
� Z d d � Z d
d � Z
d S)� WithStatNodea
Represents a Python with statement.
Implemented by the WithTransform as follows:
MGR = EXPR
EXIT = MGR.__exit__
VALUE = MGR.__enter__()
EXC = True
try:
try:
TARGET = VALUE # optional
BODY
except:
EXC = False
if not EXIT(*EXCINFO):
raise
finally:
if EXC:
EXIT(None, None, None)
MGR = EXIT = VALUE = None
�manager�
enter_callrJ r� Nc C s4 | j j | � | j j | � | j j | � d S)N)r� r� r� r� )r rC r& r& r' r� / s z!WithStatNode.analyse_declarationsc C sz | j j | � | _ | j j | � | _ | j ra d d l m } | | j j | j j � | _ | j j
| � | _ | S)Nr )r� ) r� rp r� rJ r� r� r% r/ � target_tempr� rq )r rC r� r& r& r' rq 4 s z WithStatNode.analyse_expressionsc C s= | j j | | � | j j | | � | j j | | � d S)N)r� r� r� r� )r rC rW r& r& r' r� > s z*WithStatNode.generate_function_definitionsc C s6 | j | j � | j d � | j j | � | j j t d d �| _ | j j
t j d d � � | j d | j | j j
� | j t | j r� d n d � � | j | j | j � f � | j | j � | j � } | j } | j j | � | j rW| j j | � | j j | � | j d | j j � | j j � f � | j j | � n | j j | � | j j | � | j j | � | j j | � | | _ | j j | � | j | � r| j! � } | j" | � | j# | � | j$ | j t � | j" | � | j# | � | j j% | j � | j d
� d S) Nz /*with:*/ {r� FZPyObjectLookupSpecialzObjectHandling.cz-%s = __Pyx_PyObject_LookupSpecial(%s, %s); %sZ __aexit__�__exit__z%s = %s;r� )&r� r% rQ r� rB r� r� r Zexit_varr� re r r� rx r� r r� r� r� r� r� r� rJ r� r� rE r� rH r rI r� r� r� r� r� r� r� r� )r rW r� Zintermediate_error_labelZstep_over_labelr& r& r' r� C sF
! )
z$WithStatNode.generate_execution_code) rM rm rn ry r� r� r� r� rq r� r� r& r& r& r' r�
s
r� c @ sX e Z d Z d d g Z d Z d Z d d � Z d d � Z d d � Z d
d � Z d S) � WithTargetAssignmentStatNoder= r Nc C s | j j | � d S)N)r r
)r rC r& r& r' r� � s z1WithTargetAssignmentStatNode.analyse_declarationsc C sJ | j j | � | _ | j j | � | j j j | j j | � | _ | S)N)r rr rL � with_noder� r r/ r= )r rC r& r& r' rq � s !z0WithTargetAssignmentStatNode.analyse_expressionsc C s= | j j | � | j j | j | � | j j j | � d S)N)r= rB r rL r� r� r� )r rW r& r& r' r� � s z4WithTargetAssignmentStatNode.generate_execution_codec C s$ | j j | � | j j | � d S)N)r r� r= )r rW r& r& r' r� � s z%WithTargetAssignmentStatNode.annotate)
rM rm rn r� r� r= r� rq r� r� r& r& r& r' r� t s
r� c @ sp e Z d Z d d d g Z d Z d d � Z d d � Z e j Z d Z
d
d � Z d d
� Z d d � Z
d S)�TryExceptStatNoder� �except_clausesr� Fc C sN | j j | � x | j D] } | j | � q W| j rJ | j j | � d S)N)r� r� r� r� )r rC �
except_clauser& r& r' r� � s
z&TryExceptStatNode.analyse_declarationsc C s� | j j | � | _ d } x\ t | j � D]K \ } } | j | � } | j | <| rg t | j d � | j s+ d } q+ W| | _ | j r� | j j | � | _ | S)Nr z default 'except:' must be lastr ) r� rq r r� r r% �pattern�has_default_clauser� )r rC Zdefault_clause_seenr" r� r& r& r' rq � s
z%TryExceptStatNode.analyse_expressionszTry-except statementc s� � j } � j } � j } � j � } � j } � j d � } � j d � } � j d � } � j d � }
| r� � j d � n d } | r� � j d � n d } � j d � }
� f d d � t d
� D� � � j | j � � j d � � j
� } � j d � |
� _ | � _ | � _ | j j � � � j | j d
d �� j d � � j
j � } � j | � } | r� j j t � | j s�| j d � | j d � | j d d j d d � � D� � � x � D] } | j | � q�W� � f d d � } n9 d d � � D� } | j d d j | � � d d � } | � _ | � _ | j j } | j r�� j | j j � � j d � | j j � � � j d � | s�| j j } | r�| s�x � D] } � j | t � q�W� j |
� � j | � x$ | D] \ } } � j | | � qW� j
j } | � j
_ x! | j D] } | j � | � qGW| � j
_ | j s�� j | � x� | | f | | f | | f |
| f | | f g D]| \ } } � j | � r�| r�� j |
� r�� j |
� � j | � � j | j d
d �| r&| � � j | � q�W� j | � r�| rj� j |
� rj� j |
� � j | � | r�| � � j |
� r�� j |
� � j d � x � D] } � j
j | � q�W| � _ | � _ | � _ | � _ d S) NZexception_handledZ except_errorZ
except_returnZ
try_returnZ try_breakZ try_continueZtry_endc s% g | ] } � j j t d � � q S)F)r� r� r )r� ru )rW r& r' r� � s z=TryExceptStatNode.generate_execution_code.<locals>.<listcomp>r$ r� z
/*try:*/ {r� Fr� r� r� z__Pyx_ExceptionSave(%s);z, c S s g | ] } d | � q S)z&%sr& )r� r8 r& r& r' r� � s c s< x � D] } � j | � q W� j d d j � � � d S)Nz__Pyx_ExceptionReset(%s);z, )r rQ r� )r< )rW �
exc_save_varsr& r' �restore_saved_exception� s
zJTryExceptStatNode.generate_execution_code.<locals>.restore_saved_exceptionc S s g | ] } d | � q S)z (void)%s;r& )r� r8 r& r& r' r� � s z%s /* mark used */rH c S s d S)Nr& r& r& r& r' r� � s z /*else:*/ {)!r� r r� r� r� r� r_ r� r% rQ r� r� r� r� �all_free_managed_tempsr� r� re � reset_exception_utility_coder� r� � put_xgotrefr� r� r� r r� r� Zcurrent_exceptr� �generate_handling_coder� r� ) r rW �old_return_label�old_break_label�old_continue_labelr� r� Zexcept_end_labelZexcept_error_labelZexcept_return_labelZtry_return_labelZtry_break_labelZtry_continue_labelZ
try_end_labelZsave_exc�temps_to_clean_upZ can_raiser8 r� Zmark_vars_usedZnormal_case_terminates� temp_name� temp_typeZ outer_exceptr� Z
exit_label� old_labelr r& )rW r� r' r� � s�
z)TryExceptStatNode.generate_execution_codec C s] | j j | | � x! | j D] } | j | | � q W| j d k rY | j j | | � d S)N)r� r� r� r� )r rC rW r� r& r& r' r� / s
z/TryExceptStatNode.generate_function_definitionsc C sN | j j | � x | j D] } | j | � q W| j rJ | j j | � d S)N)r� r� r� r� )r rW Z except_noder& r& r' r� 6 s
zTryExceptStatNode.annotateN)rM rm rn r� r� r� rq r| r� r� r� r� r� r� r& r& r& r' r� � s x r� c @ sp e Z d Z d d d d g Z d Z d Z d Z d d � Z d d
� Z d d � Z d
d � Z
d d � Z d S)�ExceptClauseNoder� rJ r� r� NFc C s- | j r | j j | � | j j | � d S)N)rJ r
r� r� )r rC r& r& r' r� U s z%ExceptClauseNode.analyse_declarationsc C s� | j | _ | j rZ xB t | j � D]1 \ } } | j | � } | j | � | j | <q% W| j r� d d l m } | j | j
� | _ | j j | | j � | _ | j
j | � | _
| S)Nr )r� )r �
function_namer� r rq rt rJ r� r� Z ExcValueNoder% r� r r� )r rC r" r� r� r& r& r' rq Z s z$ExceptClauseNode.analyse_expressionsc s � j | j � | j rt d d � | j D� � } | r� � f d d � t d � D� } � j j t j d d � � � j d t
| � � � j j t j d d
� � d | d } n( f } � j j t j d
d � � d } g } x5 | j D]* } | j � � | j | | j
� � q� W� j j t j d d �} � j d | d j | � f � x+ | j D] } | j � � | j � � qfW| r�� j d t
| � � � j d j d d � | D� � � x | D] } � j j | � q�W� j d | � � j j | � n
� j d � t | j d d � r�| j d k r�| j d k r�� j j t j d d � � � j d � � j | � � j d � d S� f d d � t d � D� } � j | j � � j j t � d t
| � }
� j d |
� j | j � f � x | D] } � j | � qW| j rj| j j! | d � | j j � � | j j" | j � � | j d k r�x0 t# | | j j$ � D] \ } }
|
j! | � q�W� j% � j& } } � j' d � � _% � j' d! � � _& � j j( } | � j _( | j j) � � | � j _( | j j* sQx | D] } � j+ | t, � q*W� j | � xv � j% | f � j& | f g D]V \ } } � j- | � rp� j. | � x | D] } � j+ | t, � q�W� j | � qpW| � _% | � _& x | D] } � j j | � q�W� j d � d S)"Nc s s. | ]$ } | j p% | j � o% | j Vq d S)N)r� � is_simpler: )r� r� r& r& r' r� o s z:ExceptClauseNode.generate_handling_code.<locals>.<genexpr>c s( g | ] } � j j t d d �� q S)r� T)r� r� r )r� ru )rW r& r' r� t s z;ExceptClauseNode.generate_handling_code.<locals>.<listcomp>r$ �PyErrFetchRestorez Exceptions.cz __Pyx_ErrFetch(&%s, &%s, &%s);ZFastTypeCheckszModuleSetupCode.cz*__Pyx_PyErr_GivenExceptionMatches(%s, %%s)r ZPyErrExceptionMatchesz __Pyx_PyErr_ExceptionMatches(%s)r� Fz%s = %s;z || z __Pyx_ErrRestore(%s, %s, %s);rH c S s g | ] } d | � q S)z%s = 0;r& )r� r8 r& r& r' r� � s z if (%s) {z
/*except:*/ {r� Tz__Pyx_ErrRestore(0,0,0);r� c s( g | ] } � j j t d d �� q S)r� T)r� r� r )r� ru )rW r& r' r� � s z
&%s, &%s, &%sz"if (__Pyx_GetException(%s) < 0) %sr Z except_breakZexcept_continue)/r� r% r� r� r_ r� re r r� rQ r� rB r rx r� r� r
r? r� r rI r� r� r� �excinfo_targetrJ r� r r� �get_exception_utility_coder� r� r� Zset_varrL r8 rT r r� r� r� r� r� r� r r� r� )r rW r� Zhas_non_literalsr� Z
exc_test_funcZ exc_testsr� Z
match_flagr� Zexc_argsr8 ZtempvarrV r� r� � old_exc_varsr� r� r& )rW r' r� j s�
#
"
z'ExceptClauseNode.generate_handling_codec C s9 | j d k r" | j j | | � | j j | | � d S)N)rJ r� r� )r rC rW r& r& r' r� � s z.ExceptClauseNode.generate_function_definitionsc C sW | j r* x | j D] } | j | � q W| j rC | j j | � | j j | � d S)N)r� r� rJ r� )r rW r� r& r& r' r� � s zExceptClauseNode.annotate) rM rm rn r� r� r� Z is_except_asr� rq r� r� r� r& r& r& r' r� > s j r� c @ s� e Z d Z d d d g Z d Z d Z d Z d Z d Z d Z e
d d � � Z d
d � Z d d
� Z
e j Z d Z d d � Z d d � Z d d d d � Z d d d d � Z d d � Z d d � Z d S)�TryFinallyStatNoder� �finally_clause�finally_except_clauser TNFc C s t | d | d | �} | S)Nr� r )r )r% rC r� r rV r& r& r' r� � s z"TryFinallyStatNode.create_analysedc C sI | j j | � t j | j � | _ | j j | � | j j | � d S)N)r� r� r �deepcopyr r )r rC r& r& r' r� s z'TryFinallyStatNode.analyse_declarationsc C se | j j | � | _ | j j | � | _ | j j | � | _ | j ra | j j ra | j | _ | S)N)r� rq r r r r� �func_return_type)r rC r& r& r' rq s z&TryFinallyStatNode.analyse_expressionszTry-finally statementc s� � j � j � � j } � j � } � j � } � j } � j sL | � _ � j � } � j d � � j j } d � j _ � j
j � � | � j _ � j d � � j | � � j j
� } � j � j j � � j d � � j g � f d d � } � j o� j | � }
� j j } � j
j sq� j d � | � j � � � j j sd� j | � � j d � |
r�� j | � � j d � � j s�� j d � � j r�� j � | rt � f d
d � t d � D� � } � j j t j t j t j � � d
d �}
n
d } }
t � f d d � t d � D� � } � j � | | | |
� � j � } � j d � � j j } | d d � � j _ � j j � � | � j _ � j d � | r'� j! � | | |
� | rx | D] } � j j" | � q�W|
r� j j" |
� � j | � x_ t# � j � | � D]H \ } } � j | � s[q=� j | � � j$ � | � � j | � q=Wx | D] } � j j" | � q�W� j d � � j | � � j% } f } x�t& t# | | � � D]�\ } \ } } � j | � s
q�| | k r"|
r"q�� j d | � d } | | k r
� j r�t � f d d � t d � D� � } � j � g | � � j j s
� j' r
� j r
t( � j t) � r
� j j � j' d
d �} � j d | t* j+ f � � j' j, r
� j d t* j+ � | � j � � | | k r�| r|� j d t* j+ | f � � j' j, rf� j d | � � j j" | � d } � j r�� j! � | � � j j s�� j | � � j d � q�W� j | � � j d � d S)Nz
/*try:*/ {r r� z/*finally:*/ {c s? | d } t j | � } | � j k r5 | | d <n | } | S)Nr )r r r )Z_nextrV Z node_copy)r r& r' �fresh_finally_clause* s
zHTryFinallyStatNode.generate_execution_code.<locals>.fresh_finally_clausez/*normal exit:*/{z/*exception exit:*/{r� c s+ g | ]! } � j j t j d d �� q S)r� F)r� r� r
r? )r� ru )rW r& r' r� G s z>TryFinallyStatNode.generate_execution_code.<locals>.<listcomp>r" r� Fc s( g | ] } � j j t d d �� q S)r� F)r� r� r )r� ru )rW r& r' r� O s � r� r$ z%s: {c s( g | ] } � j j t d d �� q S)r� F)r� r� r )r� ru )rW r& r' r� � s z%s = %s;z%s = 0;)-r� r% r� Zall_new_labels�get_all_labels�handle_error_caser� rQ r� Zin_try_finallyr� r� �set_all_labelsr� r �preserve_exceptionr� r� r� r� r� �is_try_finally_in_nogilr� r� r_ r� r
ZCPtrTyper4 Z c_char_type�put_error_catcherr� r �put_error_uncatcherr� r8 �put_error_cleanerr� r r rK � GILExitNoder r� r= )r rW r� Z
old_labelsZ
new_labelsr� Z catch_labelZwas_in_try_finallyr� r Zpreserve_errorZneeds_success_cleanup�exc_lineno_cnames�exc_filename_cnamer� Zfinally_old_labelsr r r� r� r� r" Zret_tempr& )rW r r' r� s�
"
(
z*TryFinallyStatNode.generate_execution_codec C s* | j j | | � | j j | | � d S)N)r� r� r )r rC rW r& r& r' r� � s z0TryFinallyStatNode.generate_function_definitionsc C sI | j j t � | j j t � | j j t � | j rI | j d d � | j d � | j d j d d � | D� � � x$ | D] \ } } | j | | � q� W| j d | d d � � | j d | d d � d
� x | D] } | j
| � q� W| r2| j d | d t j | d
t j
| t j f � | j rE| j � d S)Nr� Fr� rH c S s g | ] } d | � q S)z%s = 0;r& )r� r8 r& r& r' r� � s z8TryFinallyStatNode.put_error_catcher.<locals>.<listcomp>z>if (PY_MAJOR_VERSION >= 3) __Pyx_ExceptionSwap(&%s, &%s, &%s);r$ zmif ((PY_MAJOR_VERSION < 3) || unlikely(__Pyx_GetException(&%s, &%s, &%s) < 0)) __Pyx_ErrFetch(&%s, &%s, &%s);r" z%s = %s; %s = %s; %s = %s;r r )r� re r r� �swap_exception_utility_coder r� rQ r� r� r� r � lineno_cname�
clineno_cname�filename_cnamer� ) r rW r� r� r r r� r/ r8 r& r& r' r
� s,
#
z$TryFinallyStatNode.put_error_catcherc C sM | j j t � | j j t � | j r9 | j d d � | j d � x% | d d � D] } | j | � qW W| j d | d d � � | j d � x% | d d � D] } | j | � q� W| j d | d d � � | j r� | j � | j d j d d
� | D� � � | rI| j d t
j | d t
j | d
t
j
| f � d S)Nr� Fz if (PY_MAJOR_VERSION >= 3) {r$ z!__Pyx_ExceptionReset(%s, %s, %s);r� z __Pyx_ErrRestore(%s, %s, %s);rH c S s g | ] } d | � q S)z%s = 0;r& )r� r8 r& r& r' r� � s z:TryFinallyStatNode.put_error_uncatcher.<locals>.<listcomp>z%s = %s; %s = %s; %s = %s;r r )r� re r r� r r� rQ r r� r� r r r r )r rW r� r r r8 r& r& r' r � s(
#
z&TryFinallyStatNode.put_error_uncatcherc C s� | j j t � | j r) | j d d � | j d � x% | d d � D] } | j | � qG W| j d | d d � � | j d � x( | d d � D] } | j | t � q� W| j r� | j � | j d j
d g d � | d d � � d S) Nr� Fz if (PY_MAJOR_VERSION >= 3) {r$ z!__Pyx_ExceptionReset(%s, %s, %s);r� rH z%s = 0;) r� re r� r r� rQ r r� r r� r� )r rW r� r8 r& r& r' r � s
z$TryFinallyStatNode.put_error_cleanerc C s$ | j j | � | j j | � d S)N)r� r� r )r rW r& r& r' r� � s zTryFinallyStatNode.annotate)rM rm rn r� r r r r r r� r� r� r� rq r| r� r� r� r� r� r
r r r� r& r& r& r' r � s$ � ! r c @ s" e Z d Z d Z d Z d Z d S)�NogilTryFinallyStatNodezJ
A try/finally statement that may be used in nogil code sections.
FN)rM rm rn ry r r� r& r& r& r' r � s r c sX e Z d Z d Z d d � Z d d � Z � f d d � Z d d � Z d
d � Z � S) � GILStatNodeNc C sQ | | _ | j | | | � t j | | d | d t | d | d | j ��d S)Nr� r �state�
state_temp)r �create_state_temp_if_neededr r� r r )r r% r r� r& r& r' r� s zGILStatNode.__init__c C s} d d l m } | � } | j | � | j s3 d S| d k rK t j } n t j } d d l m } | j | | � | _
d S)Nr )�YieldNodeCollector�gil)r� ) ZParseTreeTransformsr Z
visitchildrenZyieldsr
Zc_gilstate_typeZc_threadstate_ptr_typer� r� r� r )r r% r r� r Z collectorr� r� r& r& r' r s
z'GILStatNode.create_state_temp_if_neededc s@ | j d k | _ | j d k r* d | _ t t | � j | � S)Nr T)r Z_in_with_gil_blockr� rf r r� )r rC )rL r& r' r� # s z GILStatNode.analyse_declarationsc C sS | j t j d d � � | j } | j d k | _ t j | | � } | | _ | S)NZForceInitThreadszModuleSetupCode.cr )re r r� r r r rq )r rC � was_nogilrV r& r& r' rq * s z GILStatNode.analyse_expressionsc C s� | j | j � | j � | j rE | j j | � | j j � } n d } | j j } | j d k r� | j d | � d | j _ n | j
d | � d | j _ t j | | � | j r� | j j
| � | | j _ | j � d S)Nr �variableTF)r� r% � begin_blockr r� r� r� r� r r� Zput_release_gilr r� r� � end_block)r rW r Zold_gil_configr& r& r' r� 3 s"
z#GILStatNode.generate_execution_code) rM rm rn r r� r r� rq r� r& r& )rL r' r s r c @ s: e Z d Z d Z g Z d Z d d � Z d d � Z d S)r z]
Used as the 'finally' block in a GILStatNode
state string 'gil' or 'nogil'
Nc C s | S)Nr& )r rC r& r& r' rq W s z GILExitNode.analyse_expressionsc C sQ | j r | j j � } n d } | j d k r@ | j | � n
| j | � d S)Nr )r r� r r� Zput_acquire_gil)r rW r r& r& r' r� Z s z#GILExitNode.generate_execution_code)rM rm rn ry r� r rq r� r& r& r& r' r M s
r c @ s" e Z d Z d Z d d � Z d S)�
EnsureGILNodezI
Ensure the GIL in nogil functions for cleanup before returning.
c C s | j d d � d S)Nr� F)r� )r rW r& r& r' r� k s z%EnsureGILNode.generate_execution_codeN)rM rm rn ry r� r& r& r& r' r" f s r" c C s d d l m } | j S)Nr )ra )r� ra rf )ra r& r& r' �cython_view_utility_codeo s r# z
cpython.arrayc C s t j d d � S)N�ArrayAPIz arrayarray.h)r r� r& r& r& r' r# w s r# zcpython.array.arrayc C s t j d d � S)Nr$ z arrayarray.h)r r� r& r& r& r' r# x s z cython.viewZasyncioZ__Pyx_patch_asyncioZ PatchAsyncIOz Coroutine.c�inspectZ__Pyx_patch_inspectZ PatchInspectc @ s@ e Z d Z g Z d Z d d � Z d d � Z d d � Z d S) �CImportStatNodeFc C s} | j s t | j d � d S| j | j | j d | j rA d n d �} d | j k r+d d � | j j d � D� } | d } | j j | � } | } xC | d d � D]1 } | j | � } | j | | | j � | } q� W| j
r| j | j
| | j � qS| j | � | j | | | j � n( | j
p:| j } | j | | | j � | j t k ry| j
t | j � � d S) Nz$cimport only allowed at module level�relative_levelr r r c S s g | ] } t | � � q Sr& )r )r� r< r& r& r' r� � s z8CImportStatNode.analyse_declarations.<locals>.<listcomp>r� )r� r r% � find_moduler� � is_absoluter r� Zfind_submodule�declare_module�as_namer& �utility_code_for_cimportsre )r rC � module_scoper1 Ztop_nameZtop_module_scoper< �submodule_scoper& r& r' r� � s, '
z$CImportStatNode.analyse_declarationsc C s | S)Nr& )r rC r& r& r' rq � s z#CImportStatNode.analyse_expressionsc C s d S)Nr& )r rW r& r& r' r� � s z'CImportStatNode.generate_execution_codeN)rM rm rn r� r) r� rq r� r& r& r& r' r& � s
r& c @ sX e Z d Z g Z d Z d Z d Z d d � Z d d � Z d d � Z d d � Z
d S)
�FromCImportStatNodeNc
C s� | j s t | j d � d S| j rU | j | j j d � k rU t | j d � d S| j | j | j d | j �} | j } | j | � x�| j D]�\ } } } } | d k r� x}t
| j j � � D] \ } } | j
| | | � q� Wq� | j | � } | r7| r+| j | | � r+| j | � d | _ n� | d k sO| d k ry| j | d | d
d d d d
| �} n� | d k r�| j | d
| d | �} nl | j j | d | d
| j d d �}
|
j | k r�| j | p�| |
| j � n t | d | | f � | r� | p | } | j
| | | � q� W| j d � sV| j d � r�| t k rv| j t | � � xM | j D]B \ } } } } d | | f } | t k r�| j t | � � q�Wd S)Nz$cimport only allowed at module levelr z3relative cimport beyond main package is not allowedr' r� r r( rA r' r) r* r r% �classr� Z relative_toZabsolute_fallbackFz%Name '%s' not declared in module '%s'Zcpython�cythonz%s.%s)r� r r% r' r �countr( r� r& �imported_namesr� r� re r' rX �declaration_matchesZ
redeclaredr0 r. r r� �
parent_moduler* r� r, re )
r rC r- r� r% r< r+ r' Z
local_namer� r. ru Zfqnamer& r& r' r� � sL $!
"
$ z(FromCImportStatNode.analyse_declarationsc C sV | j s
d S| j } | d k r2 | j sR d Sn | j s? d S| | j k rR d Sd S)Nr r0 r )ri r/ r} Zis_struct_or_unionr' )r r� r' r/ r& r& r' r4 � s z'FromCImportStatNode.declaration_matchesc C s | S)Nr& )r rC r& r& r' rq � s z'FromCImportStatNode.analyse_expressionsc C s d S)Nr& )r rW r& r& r' r� � s z+FromCImportStatNode.generate_execution_code) rM rm rn r� r� r' r3 r� r4 rq r� r& r& r& r' r/ � s . r/ c @ sC e Z d Z d g Z d Z d d � Z d d � Z d d � Z d S)
�FromImportStatNoder% r c C si xb | j D]W \ } } | d k rT | j s? t | j d � d Sd | _ d | _ q
| j | � q
Wd S)Nr� z%import * only allowed at module levelr )re r� r r% Zhas_import_star� import_starr
)r rC r< rJ r& r& r' r�
s z'FromImportStatNode.analyse_declarationsc
C s� d d l m } | j j | � | _ | j | j t � | _ g | _ x�| j D]�\ } } | d k r� xf| j
j � D]< \ } } | j rx | j j
rx | j t j d d � � Pqx WqP | j | j � } | j rv| j j | k rvt | j d � rv| j j | j j j k rqP yI | j | j j j d | j d | j j �} | j j | j k r`wP Wn t k
ruYn X| j | d � } | j t k r�d } n | j j | j | � } | j j | | | f � qP W| S) Nr )r� r� Z ExtTypeTestzObjectHandling.cr� r% r' ) r� r� r% rq r� r% r r� �interned_itemsre r� ri r/ r} re r r� rX r< r= r� r9 r( r� r r� r r r ) r rC r� r< rJ ru r� r% � coerced_itemr& r& r' rq s8
z&FromImportStatNode.analyse_expressionsc C s� | j | j � | j j | � | j r[ | j d t j | j j � | j | j � f � | j j
t d d �} | j j
| � | j r� | j j t j d d � � x� | j D]� \ } } } | j d | | j j � | j | � | j | | j � f � | j | � | d k r.| j | j | � n* | j | � | j | � | j | | � | j | t � q� W| j j | � | j j | � | j j | � d S)Nzif (%s(%s) < 0) %s;r� T�
ImportFromzImportExport.cz!%s = __Pyx_ImportFrom(%s, %s); %s) r� r% r% rB r7 rQ r rx r� r� r� r r� r� r8 r� re r r� r� r� r� rL Zallocate_temp_resultr� r� r� r rI )r rW Z item_tempr< rJ r9 r& r& r' r� 8 s<
z*FromImportStatNode.generate_execution_codeN)rM rm rn r� r7 r� rq r� r& r& r& r' r6 � s
#r6 c @ s e Z d Z d Z d Z d S)� ParallelNodez4
Base class for cython.parallel constructs.
N)rM rm rn ry r� r& r& r& r' r; Z s r; c s� e Z d Z d Z d d g Z d Z d Z d Z d Z d Z d Z
e j e j
e j f Z e j e j e j f Z e j e j e j f Z d Z � f d d � Z d d
� Z d d � Z d
d � Z d d � Z d d � Z d d d � Z d d � Z d d � Z! d d � Z" d d � Z# f d d � Z$ d d � Z% d! d" � Z& d# d$ � Z' d% d& � Z( d d' d( � Z) d) d* � Z* d+ d, � Z+ d- d. � Z, d/ d0 � Z- d d d d1 d2 � Z. d3 Z/ d4 Z0 d5 e/ e0 f Z1 d6 d7 � Z2 d8 d9 � Z3 � S):�ParallelStatNodea
Base class for 'with cython.parallel.parallel():' and 'for i in prange():'.
assignments { Entry(var) : (var.pos, inplace_operator_or_None) }
assignments to variables in this parallel section
parent parent ParallelStatNode or None
is_parallel indicates whether this node is OpenMP parallel
(true for #pragma omp parallel for and
#pragma omp parallel)
is_parallel is true for:
#pragma omp parallel
#pragma omp parallel for
sections, but NOT for
#pragma omp for
We need this to determine the sharing attributes.
privatization_insertion_point a code insertion point used to make temps
private (esp. the "nsteps" temp)
args tuple the arguments passed to the parallel construct
kwargs DictNode the keyword arguments passed to the parallel
construct (replaced by its compile time value)
r� � num_threadsNFr c sS t t | � j | | � | j d � p+ i | _ t � | _ i | _ g | _ d S)Nr^ ) rf r< r� r* r^ rz �seen_closure_vars�privates�assigned_nodes)r r% rt )rL r& r' r� � s
zParallelStatNode.__init__c C s� | j j | � d | _ | j rZg } t � } x� | j j D]� } | j j | k rp t | j d | j j � | j
| j j � | j j d k r� | j j s� | j | _ q> | j r� | j j d k r� | j j s� | j | _
q> | j | � q> W| | j _ y | j j | � | _ Wqct k
rV} z t | j j d � WYd d } ~ XqcXn i | _ xS | j j � D]B \ } } | | j k r�t | j d | � qst | | | � qsWd S)Nz$Duplicate keyword argument found: %sr= � chunksizez=Only compile-time values may be supplied as keyword argumentsz Invalid keyword argument: %s)r� r� r= rt rz r8 r� r9 r r% r� r� � is_prangerA r rC � Exceptionre �valid_keyword_argumentsr� )r rC �pairs�seenZdictitemrd r� r r& r& r' r� � s4 ) z%ParallelStatNode.analyse_declarationsc C s[ | j r | j j | � | _ | j r< | j j | � | _ | j j | � | _ | j | � | j d k rW| j r� | j j d k r� | j j r� t | j d � nl | j r� | j j r� t | j d � nC | j j j
r| j j r| j j | � d k rt | j d � | j j
� s3| j j j rW| j j t j | � j | � | _ | S)Nz-num_threads already declared in outer sectionz;num_threads must be declared in the parent parallel sectionr z.argument to num_threads must be greater than 0)r= rq rA r� �analyse_sharing_attributes�parentrB r r% r/ r� r� rC r� r= r r
r? rP )r rC r& r& r' rq � s&
( z$ParallelStatNode.analyse_expressionsc C s� x� | j j � D]� \ } \ } } | j rW | j rW | | j j k rW t | d � q | j rw | rw t | d � q d } | j | | | | � q Wd S)z�
Analyse the privates for this block and set them in self.privates.
This should be called in a post-order fashion during the
analyse_expressions phase
z0Cannot assign to private of outer parallel blockz*Reductions not allowed for parallel blocksTN)r^ re rB � is_parallelrH r �propagate_var_privatization)r rC r� r% rS � lastprivater& r& r' rG � s "
z+ParallelStatNode.analyse_sharing_attributesc C s� | | f | j | <| j j r0 t | d � d S| j r� | j rd | | j j k rd | j j } n | j } | r� | s | r� | j | | | | � d S)a
Propagate the sharing attributes of a variable. If the privatization is
determined by a parent scope, done propagate further.
If we are a prange, we propagate our sharing attributes outwards to
other pranges. If we are a prange in parallel block and the parallel
block does not determine the variable private, we propagate to the
parent of the parent. Recursion stops at parallel blocks, as they have
no concept of lastprivate or reduction.
So the following cases propagate:
sum is a reduction for all loops:
for i in prange(n):
for j in prange(n):
for k in prange(n):
sum += i * j * k
sum is a reduction for both loops, local_var is private to the
parallel with block:
for i in prange(n):
with parallel:
local_var = ... # private to the parallel
for j in prange(n):
sum += i * j
Nested with parallel blocks are disallowed, because they wouldn't
allow you to propagate lastprivates or reductions:
#pragma omp parallel for lastprivate(i)
for i in prange(n):
sum = 0
#pragma omp parallel private(j, sum)
with parallel:
#pragma omp parallel
with parallel:
#pragma omp for lastprivate(j) reduction(+:sum)
for j in prange(n):
sum += i
# sum and j are well-defined here
# sum and j are undefined here
# sum and j are undefined here
z9Memoryview slices can only be shared in parallel sectionsN) r? r/ r� r rB rI rH r^ rJ )r r� r% rS rK rH r& r& r' rJ s 5
z,ParallelStatNode.propagate_var_privatizationc C s� | j r | j j | | � S| j | j k r5 | j S| j j | j d � } | j j | j � | j j | � | j j | | j f � | j
d | | j f � | | _ d S)zo
Helper function that allocate a temporary for a closure variable that
is assigned to.
Tz%s = %s;N) rH �_allocate_closure_tempr r> r� r� r/ r� �modified_entriesr rQ )r rW r� r r& r& r' rL R s z'ParallelStatNode._allocate_closure_tempc C s� d } x� t | j j � � D]� \ } \ } } | r | sH | | k r | j j � } | r | rv | j d � d } | j d | j | j j | � f � q Wd S)NTz4/* Initialize private variables to invalid values */Fz%s = %s;)�sortedr? re r/ �
invalid_valuerQ r rC )r rW �excluder\ r� rS rK rO r& r& r' �initialize_privates_to_nanh s (
z+ParallelStatNode.initialize_privates_to_nanc C sD | j } | j j } | | j _ | j | � | | j _ | j � S)N)�1begin_of_parallel_control_block_point_after_declsr� �ownerrB r� )r rW r� r� rS r& r& r' �evaluate_before_blockw s
z&ParallelStatNode.evaluate_before_blockc C s3 | j d k r/ | j d | j | | j � � d S)zS
Write self.num_threads if set as the num_threads OpenMP directive
Nz num_threads(%s))r= r� rT )r rW r& r& r' �put_num_threads� s z ParallelStatNode.put_num_threadsc C sI g | _ x9 t | j � D]( } | j s1 | j r | j | | � q Wd S)a
If a variable is in a scope object, we need to allocate a temp and
assign the value from the temp to the variable in the scope object
after the parallel section. This kind of copying should be done only
in the outermost parallel section.
N)rM rN r^ Z from_closurer� rL )r rW r� r& r& r' �declare_closure_privates� s z)ParallelStatNode.declare_closure_privatesc C sT xM | j D]B \ } } | j d | | j f � | j j | j � | | _ q
Wd S)z�
Release any temps used for variables in scope objects. As this is the
outermost parallel block, we don't need to delete the cnames from
self.seen_closure_vars.
z%s = %s;N)rM rQ r r� r� )r rW r� r\ r& r& r' �release_closure_privates� s z)ParallelStatNode.release_closure_privatesc
C s! | j r | j } | j j � | _ } g g } } xI t | � D]; \ } } | j s` | j rp | j | � qB | j | � qB W| r� | j d d j
| � � | r� | j d d j
| � � | j r t j
g } | j r| j | j � | j d | j � | j d d j
| � � d S)z�
Make any used temporaries private. Before the relevant code block
code.start_collecting_temps() should have been called.
z private(%s)z, z firstprivate(%s)z private(%s, %s, %s)z shared(%s)N)rI � privatization_insertion_pointr� Zstop_collecting_tempsr� rN r= r� r r� r� �breaking_label_usedr � parallel_why�error_label_used�extend� parallel_exc�pos_info)
r rW Z
exclude_tempsr� r� r? Z
firstprivatesr� r/ Z shared_varsr& r& r' �privatize_temps� s$
z ParallelStatNode.privatize_tempsc C s� | j r� | j r� | j d � xf t | j � D]U \ } } | j r[ | j | d d �q0 | j r0 | j | | � | j d | � q0 Wd S)Nz /* Clean up any temporaries */r� Fz
%s = NULL;) rI �is_nested_prangerQ rN r� r� r r= r� )r rW r� r/ r& r& r' �
cleanup_temps� s
z ParallelStatNode.cleanup_tempsc C sx | j � | _ | j � | _ | j | _ | j d d � | _ | j � | j � | _ | j � | _
| j | � d S)a�
Sets up a block that surrounds the parallel block to determine
how the parallel section was exited. Any kind of return is
trapped (break, continue, return, exceptions). This is the idea:
{
int why = 0;
#pragma omp parallel
{
return # -> goto new_return_label;
goto end_parallel;
new_return_label:
why = 3;
goto end_parallel;
end_parallel:;
#pragma omp flush(why) # we need to flush for every iteration
}
if (why == 3)
goto old_return_label;
}
r< �returnN) r� r� r� r� r� r� r� r r� �%begin_of_parallel_control_block_pointrR �"undef_builtin_expect_apple_gcc_bug)r rW r& r& r' �!setup_parallel_control_flow_block� s
z2ParallelStatNode.setup_parallel_control_flow_blockc C s | j � | _ d S)a�
Each OpenMP thread in a parallel section that contains a with gil block
must have the thread-state initialized. The call to
PyGILState_Release() then deallocates our threadstate. If we wouldn't
do this, each with gil block would allocate and deallocate one, thereby
losing exception information before it can be saved before leaving the
parallel section.
N)r� �begin_of_parallel_block)r rW r& r& r' �begin_parallel_block� s z%ParallelStatNode.begin_parallel_blockc C s� | j r� | j } | } | j d � | j d d � | j d � | j d � | j d � | j d � | j d � | j d � | j � | j d � | j | � | j � | j d � | j d
� | j d � d S) a�
To ensure all OpenMP threads have thread states, we ensure the GIL
in each thread (which creates a thread state if it doesn't exist),
after which we release the GIL.
On exit, reacquire the GIL and release the thread state.
If compiled without OpenMP support (at the C level), then we still have
to acquire the GIL to decref any object temporaries.
z#ifdef _OPENMPr� TZPy_BEGIN_ALLOW_THREADSz#endif /* _OPENMP */ZPy_END_ALLOW_THREADSz#elsez{
z#ifndef _OPENMPz}
N)r[ rf rQ r� Zput_safera r� )r rW Z
begin_codeZend_coder& r& r' �end_parallel_block� s$
z#ParallelStatNode.end_parallel_blockc C s� | j � } | j � } d | _ d | _ d | _ g | _ | j � } xA | D]9 } | j | � rO | j py | | j k | _ d | _ qO W| j r� | j | � x� t | � D]� \ } } | j | � s� q� | | j k } | j
| � | o� | s8| | j k rd | _ | j | � | j
d t j | d f � | j ra| j ra| ra| j | � q� | j | � q� W| j r�| j r�| j r�| j
| � | j | � | j
| � | r�| j r�| j d t j � d S)a�
Trap any kind of return inside a parallel construct. 'should_flush'
indicates whether the variable should be flushed, which is needed by
prange to skip the loop. It also indicates whether we need to register
a continue (we need this for parallel blocks, but not for prange
loops, as it is a direct jump there).
It uses the same mechanism as try/finally:
1 continue
2 break
3 return
4 error
FTz%s = %d;r z#pragma omp flush(%s)N)r� �any_label_usedrY r[ �parallel_private_tempsr r� r� r� r r� r� �fetch_parallel_exceptionrQ r rZ rB �save_parallel_varsr� ) r rW � should_flushZsave_lastprivates_labelZdont_return_labelZ
all_labelsr� r" Zis_continue_labelr& r& r' �trap_parallel_exit! sF
z#ParallelStatNode.trap_parallel_exitc
C s= d | j } | j d | � t j d 7_ | j � | j } d } x� t | j j � � D]� \ } \ } } | s\ | j j r� q\ | j j
� } d | } | j }
| d 7} | j j � } | r� d | j j
| � } n d } | j d | | | f � | j d | |
f � | j j | |
f � q\ W| j � d
S) a�
The following shenanigans are instated when we break, return or
propagate errors from a prange. In this case we cannot rely on
lastprivate() to do its job, as no iterations may have executed yet
in the last thread, leaving the values undefined. It is most likely
that the breaking thread has well-defined values of the lastprivate
variables, so we keep those values.
z __pyx_parallel_lastprivates%dz#pragma omp critical(%s)r r z__pyx_parallel_temp%dz = r� z%s %s%s;z%s = %s;N)�critical_section_counterr� r< r rc rN r? re r/ r= r< r rO rC rQ rj r r! )
r rW � section_namer� Z
temp_countr� rS rK Z type_decl�
temp_cname�
private_cnamerO r r& r& r' rl ]! s*
(
z#ParallelStatNode.save_parallel_varsc C s� | j � | j d d � | j d t j � | j d t j � | j d | j � t t | j | j
� � } d | j _ | j d t
| � � | j t j � | j d � | j � | j � d S) a^
As each OpenMP thread may raise an exception, we need to fetch that
exception from the threadstate and save it for after the parallel
section where it can be re-raised in the master thread.
Although it would seem that __pyx_filename, __pyx_lineno and
__pyx_clineno are only assigned to under exception conditions (i.e.,
when we have the GIL), and thus should be allowed to be shared without
any race condition, they are in fact subject to the same race
conditions that they were previously when they were global variables
and functions were allowed to release the GIL:
thread A thread B
acquire
set lineno
release
acquire
set lineno
release
acquire
fetch exception
release
skip the fetch
deallocate threadstate deallocate threadstate
r� Tz#pragma omp flush(%s)z
if (!%s) {z'__Pyx_ErrFetchWithState(&%s, &%s, &%s);z%s = %s; %s = %s; %s = %s;r� N)r r� r� r �parallel_exc_typerQ r] r r8 �parallel_pos_infor^ r� Zuses_error_indicatorr� r� r� r! )r rW r^ r& r& r' rk �! s
z)ParallelStatNode.fetch_parallel_exceptionc C s� | j � | j d d � | j t j � | j d | j � t t | j | j
� � } | j d t | � � | j � | j
� d S)z Re-raise a parallel exceptionr� Tz&__Pyx_ErrRestoreWithState(%s, %s, %s);z%s = %s; %s = %s; %s = %s;N)r r� rG r rs rQ r] r r8 r^ rt r� r� r! )r rW r^ r& r& r' �restore_parallel_exception�! s
z+ParallelStatNode.restore_parallel_exceptionc C s$ | j | j | j | j f � d S)z�
Restore all old labels. Call this before the 'else' clause to for
loops and always before ending the parallel control flow block.
N)r
r� r� r� )r rW r& r& r' �restore_labels�! s z ParallelStatNode.restore_labelsc C s� | j } | j r| | j d | j � | j d | j � | j d t j � | j d � | j d t j � | j d � | r� | j } n | j } | r�| j d t j � | j d t j � | j d t j � x. | j
D]# \ } } | j d | | f � q� W| j d
t j � | rA| j d � | j | j
� | rd| j d � | j | j � | r�| j d
� | j | j � | j r�| j j t � | j d � | j | � | j | j � | j d � | j d � | j � | j | � d S)a�
This ends the parallel control flow block and based on how the parallel
section was exited, takes the corresponding action. The break_ and
continue_ parameters indicate whether these should be propagated
outwards:
for i in prange(...):
with cython.parallel.parallel():
continue
Here break should be trapped in the parallel block, and propagated to
the for loop.
z*const char *%s = NULL; int %s = 0, %s = 0;z,PyObject *%s = NULL, *%s = NULL, *%s = NULL;z if (%s) {zg/* This may have been overridden by a continue, break or return in another thread. Prefer the error. */z%s = 4;r� zint %s;z%s = 0;z%s = %s;z
switch (%s) {z case 1: z case 2: z case 3: z case 4:N)rc r[ rQ rt r] r rs rZ ri rY rj r� r� r� r r� r� re r ru r� r! �"redef_builtin_expect_apple_gcc_bug) r rW �break_� continue_�return_r� ri rq rr r& r& r' � end_parallel_control_flow_block�! sP
z0ParallelStatNode.end_parallel_control_flow_blockz((defined(__APPLE__) || defined(__OSX__))zQ(defined(__GNUC__) && (__GNUC__ > 2 || (__GNUC__ == 2 && (__GNUC_MINOR__ > 95))))z
(%s && %s)c C s | j s | j | j � d S)z]
A bug on OS X Lion disallows __builtin_expect macros. This code avoids them
N)rH Zundef_builtin_expect�redef_condition)r rW r& r& r' rd " s z3ParallelStatNode.undef_builtin_expect_apple_gcc_bugc C s | j s | j | j � d S)N)rH Zredef_builtin_expectr| )r rW r& r& r' rw " s z3ParallelStatNode.redef_builtin_expect_apple_gcc_bug)4rM rm rn ry r� r� rB r` r[ r= rA r rs Zparallel_exc_valueZparallel_exc_tbr] Zparallel_filenameZparallel_linenoZparallel_clinenort r r r r^ ro r� r� rq rG rJ rL rQ rT rU rV rW r_ ra re rg rh rn rl rk ru rv r{ Z buggy_platform_macro_conditionZhave_expect_conditionr| rd rw r& r& )rL r' r< b sZ & G
% C + .
H r< c sC e Z d Z d Z d g Z d Z � f d d � Z d d � Z � S)�ParallelWithBlockNodezI
This node represents a 'with cython.parallel.parallel():' block
r= Nc s3 t t | � j | � | j r/ t | j d � d S)Nz=cython.parallel.parallel() does not take positional arguments)rf r} r� rT r r% )r rC )rL r& r' r� ." s z*ParallelWithBlockNode.analyse_declarationsc C s� | j | � | j | � | j d � | j d � | j rs d d � | j D� } | j d d j t | � � � | j � | _ | j | � | j d � | j d � | j
� | j | � | j | � | j
j � | j j | � | j | � | j | � | j | � | j � | j | j � } | j | j � } | j | j � } | j | � | j | d | d
| d | �| j | � d S) Nz#ifdef _OPENMPz#pragma omp parallel c S s% g | ] } | j j s | j � q Sr& )r/ r= r )r� rd r& r& r' r� <" s zAParallelWithBlockNode.generate_execution_code.<locals>.<listcomp>z private(%s)z, r� z#endif /* _OPENMP */rx ry rz ) rV re rQ r� r? r� rN r� rX rU r rg rQ r� �start_collecting_tempsr� r� rn r_ rh r! r� r� r r� rv r{ rW )r rW r? ry rx rz r& r& r' r� 4" s8
z-ParallelWithBlockNode.generate_execution_code)rM rm rn ry rD r= r� r� r& r& )rL r' r} %" s
r} c s� e Z d Z d Z d d d d d d g Z d Z Z Z Z d Z Z
Z d Z d Z
d Z d
d d d g Z � f d d
� Z � f d d � Z � f d d � Z d d � Z d d � Z d d � Z � S)�ParallelRangeNodez�
This node represents a 'for i in cython.parallel.prange():' construct.
target NameNode the target iteration variable
else_clause Node or None the else clause of this loop
r� rJ r� rT r= rA NT�scheduler c s, t t | � j | | � t | � | _ d S)N)rf r r� r5 r� )r r% rU )rL r& r' r� q" s zParallelRangeNode.__init__c s; t t | � j | � | j j | � | j d k rE | j j | � | j sd t | j � d k rx t | j d � d St | j � d k r� | j \ | _
nH t | j � d k r� | j \ | _ | _
n | j \ | _ | _
| _ t
| j d � r| j j d � | _ | j d k r7t | j d | j f � d S)
Nr$ z0Invalid number of positional arguments to pranger r" �decode�ascii�static�dynamic�guided�runtimez'Invalid schedule argument to prange: %s)Nr� r� r� r� )rf r r� rJ r
r� rT rJ r r% rI rY r� r= r� r� )r rC )rL r& r' r� v" s z&ParallelRangeNode.analyse_declarationsc sS | j } | j r d | _ | j d k r> t | j d � | S| j j | � | _ | j j j s� | j j j s� t | j j d | j j � t j | _
n4 | j j | _
| j
j s� t | j j d d d �d | _
| j | j | j f } x� t | | j
� D]� \ } } | d k r| j | � | j j sMt | j d
| � q| j su| j | � } t | | | � t j | j
| j � | _
qW| j d k r�| j j | � | _ t | j d � r�| j j d f | j | j j <t t | � j | � } | j r�| j s)t | j j d � nk | j d
k rNt | j j d � nF | j j j r�| j j r�| j j! | � d k r�t | j j d � | j j" t j# | � j | � | _ | j r�| | _ | j$ o�| j$ j% | _& | j& rO| } x" | j$ r| j$ j% r| j$ } q�W| j j' | j � | j( j' | j( � | j) j* | j) � | S)NTz/prange() can only be used as part of a for loopz Must be of numeric type, not %sz1Unsigned index type not allowed before OpenMP 3.0r� r" rY rI r� z%s argument must be numericr� z$Must provide schedule with chunksizer� z,Chunksize not valid for the schedule runtimer z Chunksize must not be negative)rY rI r� )+r rJ r r% rr r/ r[ r= r
r� �
index_typerV r r1 rY rI r� r8 rp r� rP r� r� r� rq r= r^ r� rf r rA r� r� rC r r? rH rB r` r~ r? r@ r\ )r rC r �start_stop_steprV r< rH )rL r& r' rq �" sv
z%ParallelRangeNode.analyse_expressionsc C st d } | j | j | j | j f } xI t | | � D]8 \ } } | d k r4 | j j r4 t | j d | � q4 Wd S)NrY rI r� rJ z6%s may not be a Python object as we don't have the GIL)rY rI r� rJ ) rY rI r� rJ r8 r/ r= r r% )r rC r1 Znodesr< rV r& r& r' r� �" s z ParallelRangeNode.nogil_checkc C sm | j | � | j j j } d | d | j j j � i } | j | j | j f } d } xv t | | j
| � D]_ \ } } } | d k r� | } n1 | j r� | j � } n | j
| � | j � } | | | <qn W| j j | j d � | d <| j j | j d � | d <| j d | � | j | � | j � | _ | j d
| � | j d | � | j � | j | | � | j � | j | � | j r�| j r�| j d t j � | j � | j d
� | j j | � | j � | j | � xD | | j! | j" f D], }
|
d k r|
j# | � |
j$ | � qW| j j% | d � | j j% | d � | j& | � d S)a,
Generate code in the following steps
1) copy any closure variables determined thread-private
into temporaries
2) allocate temps for start, stop and step
3) generate a loop that calculates the total number of steps,
which then computes the target iteration variable for every step:
for i in prange(start, stop, step):
...
becomes
nsteps = (stop - start) / step;
i = start;
#pragma omp parallel for lastprivate(i)
for (temp = 0; temp < nsteps; temp++) {
i = start + step * temp;
...
}
Note that accumulation of 'i' would have a data dependency
between iterations.
Also, you can't do this
for (i = start; i < stop; i += step)
...
as the '<' operator should become '>' for descending loops.
'for i from x < i < y:' does not suffer from this problem
as the relational operator is known at compile time!
4) release our temps and write back any private closure variables
rJ r� rb r� NFr" Znstepszif (%(step)s == 0) abort();zS%(nsteps)s = (%(stop)s - %(start)s + %(step)s - %(step)s/abs(%(step)s)) / %(step)s;zif (%(nsteps)s > 0)z if (%s < 2)z
/* else */)rb rb r� )'rV rJ r� r r/ r< rY rI r� r8 r1 r� r� rB r� r� r� r� rQ re r� Zcontrol_flow_var_code_pointr �
generate_loopr! rv r� rY r� r rZ r� r{ rA r= r rI r� rW ) r rW Ztarget_index_cname�fmt_dictr� �defaultsrV r< r� r� r� r& r& r' r� �" sP (
%
z)ParallelRangeNode.generate_execution_codec
C sR | j r | j d � n
| j d � | j sZ | j d � | j � | _ | j j } n� | j d � | j � | _ | j } | j d � | j d � | j � | j | � | j r� | j d � n
| j d � | j d � xt | j
j � � D]� \ } \ } } | rk| d k rk| | j j
k rk| j j rNt | j d � q�| j d | | j f � q� | | j j
k r�| j d
| j � | j d | j � q� | j j s� | r�d } n d
} | j d | | j f � q� W| j r5| j rd | j | | j � } n d } | j d | j | f � | j | � | j d � | j d � | j d | � | j � | j � } | j � | j d | � | j | d | j j
�| j r�| j j � | j j | � | j | d d �| j | � | j r| j d t j � | j! � | j! � | j rN| j" | � | j! � d S)Nz#if 0z#ifdef _OPENMPz#pragma omp forz#pragma omp parallelr� z#endif /* _OPENMP */z+*-&^|z#Python objects cannot be reductionsz reduction(%s:%s)z firstprivate(%s)z lastprivate(%s)rK rQ z %s(%s)z, %sz schedule(%s%s)z,for (%(i)s = 0; %(i)s < %(nsteps)s; %(i)s++)z=%(target)s = (%(target_type)s)(%(start)s + %(step)s * %(i)s);rP rm Tz if (%s < 2))#r` rQ rI r� r� rX rH r rg rN r? re rJ r� r/ r= r r% r r� rA rT rU rQ r� r~ r� r� rn r_ rY r rZ r! rh )
r rW r� Zreduction_codepointr� rS rK rQ rA Zguard_around_body_codepointr& r& r' r� c# sv
($
z ParallelRangeNode.generate_loop)rM rm rn ry r� r� rJ r� rT rY rI r� rB r r� rD r� r� rq r� r� r� r& r& )rL r' r [" s Y ur c @ s[ e Z d Z d Z d g Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�CnameDecoratorNodea?
This node is for the cname decorator in CythonUtilityCode:
@cname('the_cname')
cdef func(...):
...
In case of a cdef class the cname specifies the objstruct_cname.
node the node to which the cname decorator is applied
cname the cname the node should get
rV c C s� | j j | � | j } t | t � r8 | j j d } t | t � | _ t | t t f � } | j
} | j r� | j | _ | j | _ d | _
| j r�d | j k r�| j | j � | _ n| r� | j | _ | j _ n� | j } | j | _ | j d | j _ t j | j | j _ | j d | j _ | j j | j j _ | j j | j _ | j d | _ xY | j j � D]H \ } } | j r�| j | j � | _ | j ry| j | j � | _ qyWd S)Nr Tr �_objZ_typeru ) rV r� rK r� r� r� r� � is_functionr� r� r� r rV r0 r� r- r/ r) r r Ztypeobj_prefixr r� r rC r� r� re )r rC rV Zis_struct_or_enumrd r) r< r� r& r& r' r� �# s8 z'CnameDecoratorNode.analyse_declarationsc C s0 d | k r | j d � d } d | j | f S)Nr r z%s_%sr� )r r )r r r& r& r' r- �# s zCnameDecoratorNode.manglec C s | j j | � | _ | S)N)rV rq )r rC r& r& r' rq $ s z&CnameDecoratorNode.analyse_expressionsc C s� | j r� | j r� | j d } t | j t � rP | j j | d d d d �nY d d l m } | j j } | j
} | j | _
| j | | j
� | d d �| | _
| j j | | � d S)
z=Ensure a prototype for every @cname method in the right placeZutility_code_protor� Fr� Tr )�
ModuleNodeZ
definitionN)r� r r� rK rV rB r� r� r� r� r rV Z generate_cfunction_declarationr� r� )r rC rW Zh_coder� r� r r& r& r' r� $ s
z0CnameDecoratorNode.generate_function_definitionsc C s | j j | � d S)N)rV r� )r rW r& r& r' r� $ s z*CnameDecoratorNode.generate_execution_codeN)
rM rm rn ry r� r� r- rq r� r� r& r& r& r' r� �# s & r� aC
/* Test for GCC > 2.95 */
#if defined(__GNUC__) && (__GNUC__ > 2 || (__GNUC__ == 2 && (__GNUC_MINOR__ > 95)))
#define likely(x) __builtin_expect(!!(x), 1)
#define unlikely(x) __builtin_expect(!!(x), 0)
#else /* !__GNUC__ or GCC < 2.95 */
#define likely(x) (x)
#define unlikely(x) (x)
#endif /* __GNUC__ */
z1
#define likely(x) (x)
#define unlikely(x) (x)
ZPrintz
Printing.cZPrintOner� z Exceptions.cZRaiseExceptionZ GetExceptionZ
SwapExceptionZSaveResetExceptionZ AddTracebackr� zR
static PyObject *__Pyx_GetExceptionTuple(PyThreadState *__pyx_tstate); /*proto*/
�impla1
static PyObject *__Pyx_GetExceptionTuple(CYTHON_UNUSED PyThreadState *__pyx_tstate) {
PyObject *type = NULL, *value = NULL, *tb = NULL;
if (__Pyx_GetException(&type, &value, &tb) == 0) {
PyObject* exc_info = PyTuple_New(3);
if (exc_info) {
Py_INCREF(type);
Py_INCREF(value);
Py_INCREF(tb);
PyTuple_SET_ITEM(exc_info, 0, type);
PyTuple_SET_ITEM(exc_info, 1, value);
PyTuple_SET_ITEM(exc_info, 2, tb);
return exc_info;
}
}
return NULL;
}
�requires)��
__future__r r1 r� rU r r r � itertoolsr r� r ZErrorsr r r r r r
r r r ZSymtabr r
r r r r r rc r ZStringEncodingr r r r ZPythranr r r! ZUtilsr# rq r3 r r4 r( r. rG r^ r/ r_ ro r{ r| r� r� r� r� r� r� r� r� r� r� r� r3 r6 rS rA rT r] rg r> rt rv rx r{ r� r� r� r� r� r� r� r� rf rA rB rw r� r� r� r� r� rF r� r� r
r/ r0 r2 r� r; r? rb r< rj rr ry rz r5 r} r~ r� rL r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r" r# r, Zutility_code_for_importsr& r/ r6 r; r< r} r r� Zgcc_branch_hintsZbranch_prediction_macrosr� ru rw r r� r� r r� Ztraceback_utility_codeZ get_exception_tuple_utility_coder& r& r& r' �<module> sH
$"4 /
�*
%#,�
s ^!Q '
j-X1!� � �� �� � � � � -8�E�� � @� 1f/GC)7
rh =C) )<\<
a
�g!��� H +O\� � �6� kc