U
ل=^�T � @ s� d dl Z d dlZddlmZ ddlmZmZmZ dZdZ dZ
dd� Z G d d
� d
e �Z
G d d � d e
�ZG d
d� de�Ze� ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�Zee�Zd d � Zee�ZG d!d"� d"e�Z G d#d$� d$e�Z e ed%�d�Z G d&d'� d'e
�Z G d(d)� d)e �Z G d*d+� d+e �Z!G d,d-� d-e �Z"G d.d/� d/e �Z#d<d0d1�Z$d=d2d3�Z%e� Z&e�'� Z(d4d5� Z)d6d7� Z*d8d9� Z+d:d;� Z,dS )>� N� )�
allocate_lock)� CDefError�VerificationError�VerificationMissing� � c C s@ | t @ rd|�� }| t@ r(d|�� }| t@ r<d|�� }|S )Nz const z
volatile z __restrict )�Q_CONST�lstrip�
Q_VOLATILE�
Q_RESTRICT)�quals� replace_with� r � /model.py�qualify s r c @ sP e Zd ZdZdZddd�Zdd� Zd d
� Zd d � Zdd
d�Z dd� Z
dd� Z dS )�BaseTypeByIdentityF� �a C filer c C sx | j }|�� }|rB|�d�r.d|kr.d| }n|d dkrBd| }t||�}|�d|�}d|krttd | �� |f ��|S )
N�*z&[�(%s)r z[(� �&�$z-cannot generate '%s' in %s: unknown type name)�c_name_with_marker�strip�
startswithr �replacer � _get_c_name)�selfr �contextr
�resultr r r �
get_c_name s
��z BaseTypeByIdentity.get_c_namec C s | j �dd�S )Nr r )r r �r r r r r / s z BaseTypeByIdentity._get_c_namec C s d| � � kS )Nr �r r# r r r �
has_c_name2 s z BaseTypeByIdentity.has_c_namec C s dS �NFr r# r r r �is_integer_type5 s z"BaseTypeByIdentity.is_integer_typec C sB z|j | }W n. tk
r< | �||�}|j �| |�}Y nX |S �N)�_cached_btypes�KeyError�build_backend_type�
setdefault)r �ffi�
finishlist� can_delay�BTypeZBType2r r r �get_cached_btype8 s z#BaseTypeByIdentity.get_cached_btypec C s d| � � f S )Nz<%s>r$ r# r r r �__repr__A s zBaseTypeByIdentity.__repr__c s � fdd�� j D �S )Nc s g | ]}|t � |�f�qS r )�getattr)�.0�namer# r r �
<listcomp>E s z1BaseTypeByIdentity._get_items.<locals>.<listcomp>)�_attrs_r# r r# r �
_get_itemsD s z BaseTypeByIdentity._get_itemsN)r r r )F) �__name__�
__module__� __qualname__�
is_array_type�is_raw_functionr" r r% r' r1 r2 r8 r r r r r s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�BaseTypec C s | j |j ko| �� |�� kS r( )� __class__r8 �r �otherr r r �__eq__J s �zBaseType.__eq__c C s
| |k S r( r r@ r r r �__ne__N s zBaseType.__ne__c C s t | jt| �� �f�S r( )�hashr? �tupler8 r# r r r �__hash__Q s zBaseType.__hash__N)r9 r: r; rB rC rF r r r r r> H s r> c @ s e Zd ZdZdd� Zdd� ZdS )�VoidTyper c C s
d| _ d S )Nzvoid&�r r# r r r �__init__X s zVoidType.__init__c C s t | |d�S )NZ
new_void_type�� global_cache�r r- r. r r r r+ [ s zVoidType.build_backend_typeN�r9 r: r; r7 rI r+ r r r r rG U s rG c @ s e Zd Zdd� ZdS )�BasePrimitiveTypec C s dS r& r r# r r r �is_complex_typeb s z!BasePrimitiveType.is_complex_typeN)r9 r: r; rO r r r r rN a s rN c 4 @ s� e Zd ZdZdddddddddddddddddddddddddddddddddddddddddddddddddddd�3Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dS )�
PrimitiveType�r5 �c�i�f�j)3�charZshort�int�longz long longz signed charz
unsigned charzunsigned short� unsigned int�
unsigned longzunsigned long long�floatZdoublez long doublezfloat _Complexzdouble _ComplexZ_BoolZwchar_tZchar16_tZchar32_tZint8_tZuint8_tZint16_tZuint16_tZint32_tZuint32_tZint64_tZuint64_tZ int_least8_tZ
uint_least8_tZ
int_least16_tZuint_least16_tZ
int_least32_tZuint_least32_tZ
int_least64_tZuint_least64_tZ int_fast8_tZ uint_fast8_tZ int_fast16_tZ
uint_fast16_tZ int_fast32_tZ
uint_fast32_tZ int_fast64_tZ
uint_fast64_tZintptr_tZ uintptr_tZintmax_tZ uintmax_tZ ptrdiff_tZsize_tZssize_tc C s || _ |d | _d S �Nr �r5 r �r r5 r r r rI � s zPrimitiveType.__init__c C s | j | j dkS )NrR ��ALL_PRIMITIVE_TYPESr5 r# r r r � is_char_type� s zPrimitiveType.is_char_typec C s | j | j dkS )NrS r_ r# r r r r' � s z PrimitiveType.is_integer_typec C s | j | j dkS )NrT r_ r# r r r �
is_float_type� s zPrimitiveType.is_float_typec C s | j | j dkS )NrU r_ r# r r r rO � s z PrimitiveType.is_complex_typec C s t | |d| j�S )NZnew_primitive_type)rK r5 rL r r r r+ � s z PrimitiveType.build_backend_typeN) r9 r: r; r7 r` rI ra r' rb rO r+ r r r r rP f sv �7rP c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �UnknownIntegerTyperQ c C s || _ |d | _d S r\ r] r^ r r r rI � s zUnknownIntegerType.__init__c C s dS )NTr r# r r r r' � s z"UnknownIntegerType.is_integer_typec C s t d| j ��d S )Nz4integer type '%s' can only be used after compilation��NotImplementedErrorr5 rL r r r r+ � s �z%UnknownIntegerType.build_backend_typeN)r9 r: r; r7 rI r' r+ r r r r rc � s rc c @ s e Zd ZdZdd� Zdd� ZdS )�UnknownFloatTyperQ c C s || _ |d | _d S r\ r] r^ r r r rI � s zUnknownFloatType.__init__c C s t d| j ��d S )Nz2float type '%s' can only be used after compilationrd rL r r r r+ � s �z#UnknownFloatType.build_backend_typeNrM r r r r rf � s rf c @ s e Zd ZdZddd�ZdS )�BaseFunctionType)�argsr! �ellipsis�abiNc C s� || _ || _|| _|| _dd� | j D �}| jr8|�d� |p@dg}| jd�|�f }|d k r||d d� | d |dd � }| jj�d|�| _d S ) Nc S s g | ] }|� � �qS r r$ )r4 �argr r r r6 � s z-BaseFunctionType.__init__.<locals>.<listcomp>�...Zvoidz, r r r ) rh r! ri rj �append�
_base_pattern�joinr r )r rh r! ri rj Zreprargsr r r r rI � s
�zBaseFunctionType.__init__)N�r9 r: r; r7 rI r r r r rg � s rg c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�RawFunctionTypez(&)(%s)Tc C s t d| f ��d S )NzPcannot render the type %r: it is a function type, not a pointer-to-function type)r rL r r r r+ � s �z"RawFunctionType.build_backend_typec C s t | j| j| j| j�S r( )�FunctionPtrTyperh r! ri rj r# r r r �as_function_pointer� s z#RawFunctionType.as_function_pointerN)r9 r: r; rn r= r+ rs r r r r rq � s rq c @ s e Zd ZdZdd� Zdd� ZdS )rr z(*&)(%s)c C s� | j �||�}g }| jD ]}|�|�||�� qd}| jdkrh| jshz|jjf}W n tk
rf Y nX t | |dt
|�|| jf|�� S )Nr Z __stdcallZnew_function_type) r! r1 rh rm rj ri �_backendZ FFI_STDCALL�AttributeErrorrK rE )r r- r. r! rh �tpZabi_argsr r r r+ � s$
��z"FunctionPtrType.build_backend_typec C s t | j| j| j| j�S r( )rq rh r! ri rj r# r r r �as_raw_function s z FunctionPtrType.as_raw_functionN)r9 r: r; rn r+ rw r r r r rr � s rr c @ s" e Zd ZdZddd�Zdd� ZdS ) � PointerType)�totyper
r c C s> || _ || _t|d�}|jr*d|�� f }|j�d|�| _d S )Nz *&r r )ry r
r r< r
r r )r ry r
Zextrar r r rI s
zPointerType.__init__c C s | j j||dd�}t| |d|�S )NT)r/ �new_pointer_type)ry r1 rK )r r- r. ZBItemr r r r+ s z PointerType.build_backend_typeN)r rM r r r r rx s
rx c C s
t | t�S r( )rx r )ry r r r �ConstPointerType s r{ c @ s e Zd ZdZddd�ZdS )�NamedPointerType)ry r5 r c C s" t �| ||� || _|d | _d S r\ )rx rI r5 r )r ry r5 r
r r r rI s zNamedPointerType.__init__N)r rp r r r r r| s r| c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
� ArrayType)�item�lengthTc C sF || _ || _|d krd}n|dkr(d}nd| }| j j�d|�| _d S )Nz&[]rl z
&[/*...*/]z&[%s]r )r~ r r r )r r~ r Zbracketsr r r rI ) s �zArrayType.__init__c C s t | j|�S r( )r} r~ )r Z newlengthr r r �resolve_length6 s zArrayType.resolve_lengthc C sJ | j dkrtd| f ��| j�||� t| j��||�}t| |d|| j �S )Nrl z)cannot render the type %r: unknown lengthZnew_array_type)r r r~ r1 rx rK )r r- r. ZBPtrItemr r r r+ 9 s
�z ArrayType.build_backend_typeN)r9 r: r; r7 r< rI r� r+ r r r r r} % s
r} rV c @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) �StructOrUnionOrEnumrQ Nc C s$ | j pd| j| jf }|d | _d S )Nz%s %sr )� forcename�kindr5 r r^ r r r �build_c_name_with_markerH s z,StructOrUnionOrEnum.build_c_name_with_markerc C s || _ | �� d S r( )r� r� )r r� r r r �force_the_nameL s z"StructOrUnionOrEnum.force_the_namec C s | j d d� S )N���rH r# r r r �get_official_nameP s z%StructOrUnionOrEnum.get_official_name)r9 r: r; r7 r� r� r� r� r r r r r� D s
r� c @ sj e Zd ZdZdZdZdZddd�Zdd� Zdd d
�Z d d � Z
dd
d�Z dd� Z dd� Z
dd� Zdd� ZdS )�
StructOrUnionNr Fc C s* || _ || _|| _|| _|| _| �� d S r( )r5 �fldnames�fldtypes�
fldbitsize�fldqualsr� )r r5 r� r� r� r� r r r rI [ s zStructOrUnion.__init__c c s> | j d k r:t| j| j �D ] \}}|dkrt|t�r|V qd S )Nr )r� �zipr� �
isinstancer� )r r5 �typer r r �anonymous_struct_fieldsc s
z%StructOrUnion.anonymous_struct_fieldsTc c s| | j }|d kr dt| j� }t| j| j| j|�D ]F\}}}}|dkrht|t�rh|rh|�� D ]
}|V qZq0||||fV q0d S )N)r r ) r� �lenr� r� r� r� r� r� �
enumfields)r Z expand_anonymous_struct_unionr� r5 r� �bitsizer
r! r r r r� i s
��
zStructOrUnion.enumfieldsc C sz g }g }g }g }| � � D ]4\}}}}|�|� |�|� |�|� |�|� qt|�| _t|�| _t|�| _t|�| _d S r( )r� rm rE r� r� r� r� ) r �names�typesZbitsizesr� r5 r� r� r
r r r �
force_flattenw s
zStructOrUnion.force_flattenc C s$ t �| |||�}|s | �||� |S r( )r� r1 �finish_backend_type)r r- r. r/ r0 r r r r1 � s
� z StructOrUnion.get_cached_btypec s$ | j r$| j dkr td| jf ��d S � j| }d| _ | jd krB�n�| jd kr�� �fdd�| jD �}tt| j|| j ��}d}| j
r�| j
dkr�d}n
d| j
f}� j j ||| d d f|�� �nbg }| j\}}} }
t
t| j��D �]} || } | j| }
t|
t��r�|
jd
k�r�|
j�� ��}t| � �|��\}}|dk�rT| �d | j| j| �pLd f � |
�|�}
| jd | � |
f | j| d d � | _|
�� ��}t|
t��r�|
jd k�r�n8� �|�}|| k�r�| �d
| j| j| �p�d || f � |�|� q�tt| j|| j |��}� j � ||| | |
� d| _ d S )Nr z(recursive structure declaration for '%s'r c s g | ]}|� � ���qS r )r1 )r4 rv �r- r. r r r6 � s �z5StructOrUnion.finish_backend_type.<locals>.<listcomp>r )� r r� rl z field '%s.%s' has a bogus size?z{}z=field '%s.%s' is declared as %d bytes, but is really %d bytes)� completedre r5 r) r� � fixedlayout�listr� r� r� �packedrt Zcomplete_struct_or_union�ranger� r� r} r r~ r1 �divmod�sizeof�_verification_errorr� rm )r r- r. r0 r� Zlst� extra_flagsZfieldofsZ fieldsizeZ totalsizeZtotalalignmentrS ZfsizeZftypeZ BItemTypeZnlenZnrestZ
BFieldTypeZ bitemsizer r� r r� � s�
�
�
��
��
�
��� �z!StructOrUnion.finish_backend_typec C s t |��d S r( )r )r �msgr r r r� � s z!StructOrUnion._verification_errorc C s | j r | jd kr t| �� ��d S r( )�partialr� r r r# r r r �check_not_partial� s z StructOrUnion.check_not_partialc C s. | � � |�| � t| |d| j | �� | d�S )Nz new_%s_type��key)r� rm rK r� r� rL r r r r+ � s
�z StructOrUnion.build_backend_type)N)T)F)r9 r: r; r� r� r� r� rI r� r� r� r1 r� r� r� r+ r r r r r� U s
?r� c @ s e Zd ZdZdS )�
StructType�structN�r9 r: r; r� r r r r r� � s r� c @ s e Zd ZdZdS )� UnionType�unionNr� r r r r r� � s r� c @ sB e Zd ZdZdZdZddd�Zdd� Zdd � Zd
d � Z d d
� Z
dS )�EnumType�enumFNc C s$ || _ || _|| _|| _| �� d S r( )r5 � enumerators�
enumvalues� baseinttyper� )r r5 r� r� r� r r r rI � s
zEnumType.__init__c C s4 t �| |� | jd kr0| �� }d|�dd� | _d S )Nr r �_)r� r� r� r� r )r r� r5 r r r r� � s
zEnumType.force_the_namec C s | j r| jst| �� ��d S r( )r� �partial_resolvedr r r# r r r r� � s zEnumType.check_not_partialc
C s4 | � � | �||�}t| |d| �� | j| j|| d�S )NZ
new_enum_typer� )r� �build_baseinttyperK r� r� r� )r r- r. Z
base_btyper r r r+ � s �zEnumType.build_backend_typec
C sJ | j d k r| j �||�S | jr4t| j�}t| j�}nDdd l}z t�� W n tk
r\ Y nX |� d| �
� � d }}|dk r�d}t d�}t d�}nd}t d�}t d�}|�||�} |�||�}
|� | �} |� |
�} |dd | d > k�r|dd | | > k �r| S |dd | d > k�r6|dd | | > k �r6|
S t
d
| �
� ��d S ) Nr zU%r has no values explicitly defined; guessing that it is equivalent to 'unsigned int'r rW rX rY rZ r� r� z=%s values don't all fit into either 'long' or 'unsigned long')r� r1 r� �min�max�warningsZ__warningregistry__�clear� NameError�warnr rP r� r )
r r- r. Zsmallest_valueZ
largest_valuer� ZsignZ
candidate1Z
candidate2Zbtype1Zbtype2Zsize1Zsize2r r r r� sH
�
���zEnumType.build_baseinttype)N) r9 r: r; r� r� r� rI r� r� r+ r� r r r r r� � s
r� c C s2 |d krd| }t |d d d �}|�| � d|_|S )Nz$%s� unknown_type)r� r� �origin�r5 Z
structnamerv r r r r� - s
r� c C s( |d krd| }t |d d d �}t|| �S )Nz$$%s)r� r| r� r r r �unknown_ptr_type5 s r� c
C sR t | tj�rtS t�4 tt| �d�s2t�� t| �_ t| �j W 5 Q R � S Q R X d S )N� __typecache)
r� r� �
ModuleType�_typecache_cffi_backend� global_lock�hasattrr� �weakref�WeakValueDictionaryr� )Zbackendr r r �
get_typecache? s r� c
O s� |� d||f�}z |j| W S tk
r0 Y nX zt|j|�|� }W n4 tk
rz } ztd|| |f ��W 5 d }~X Y nX |j}t�@ |�|�} | d kr�|||< |W 5 Q R � S | W 5 Q R � S W 5 Q R X d S )Nr� z
%s: %r: %s)�popZ
_typecacher* r3 rt re r� �get)
Zsrctyper- �funcnamerh �kwdsr� �res�e�cacheZres1r r r rK J s $
rK c C s t d| d|�S )N�?rz rJ )r- r0 r r r �
pointer_cachea s r� c C s@ | j r<t| j d �tkr<d|| j d f f| j dd � | _ d S )Nr z%s: %sr )rh r� �str)r� r5 r r r �attach_exception_infod s r� )N)N)-r� r� �lockr �errorr r r r r r r �objectr r> rG Z void_typerN rP rc rf rg rq rr rx Z
voidp_typer{ Zconst_voidp_typer| r} Zchar_array_typer� r� r� r� r� r� r� r� r� r� r� rK r� r� r r r r �<module> sN
/
L
G