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 deZe ZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd d eZG d d d eZeeZd d ZeeZG d!d" d"eZ G d#d$ d$eZ e ed%dZ 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<d0d1Z$d=d2d3Z%e Z&e' Z(d4d5 Z)d6d7 Z*d8d9 Z+d:d; Z,dS )> N )
allocate_lock) CDefErrorVerificationErrorVerificationMissing c C s@ | t @ rd| }| t@ r(d| }| t@ r<d| }|S )Nz const z
volatile z __restrict )Q_CONSTlstrip
Q_VOLATILE
Q_RESTRICT)quals replace_with r /model.pyqualify s r c @ sP e Zd ZdZdZdddZdd Zd d
Zd d Zdd
dZ dd Z
dd Z dS )BaseTypeByIdentityF a C filer c C sx | j }| }|rB|dr.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_markerstrip
startswithr replacer _get_c_name)selfr contextr
resultr r r
get_c_name s
z BaseTypeByIdentity.get_c_namec C s | j ddS )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_btypesKeyErrorbuild_backend_type
setdefault)r ffi
finishlist can_delayBTypeZBType2r 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 |fqS r )getattr).0namer# 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_typeis_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| fS 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 | |dS )NZ
new_void_type global_cacher r- r. r r r r+ [ s zVoidType.build_backend_typeNr9 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 ZdZdddddddddddddddddddddddddddddddddddddddddddddddddddd3Zdd Zd d
Zd d Zd
d Zdd Z dd Z
dS )
PrimitiveTyper5 cifj)3charZshortintlongz long longz signed charz
unsigned charzunsigned short unsigned int
unsigned longzunsigned long longfloatZdoublez 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| jS )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 compilationNotImplementedErrorr5 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 ZdZdddZdS )BaseFunctionType)argsr! ellipsisabiNc C s || _ || _|| _|| _dd | j D }| jr8|d |p@dg}| jd|f }|d k r||d d | d |dd }| jjd|| _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_patternjoinr r )r rh r! ri rj Zreprargsr r r r rI s
zBaseFunctionType.__init__)Nr9 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| jS 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_STDCALLAttributeErrorrK rE )r r- r. r! rh tpZabi_argsr r r r+ s$
z"FunctionPtrType.build_backend_typec C s t | j| j| j| jS 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 ZdZdddZdd ZdS ) PointerType)totyper
r c C s> || _ || _t|d}|jr*d| f }|jd|| _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 | tS r( )rx r )ry r r r ConstPointerType s r{ c @ s e Zd ZdZdddZdS )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)itemlengthTc C sF || _ || _|d krd}n|dkr(d}nd| }| j jd|| _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 ) forcenamekindr5 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 )NrH 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 ZdZdZdZdZdddZdd Zdd d
Z d d Z
dd
dZ dd Z dd Z
dd Zdd ZdS )
StructOrUnionNr Fc C s* || _ || _|| _|| _|| _| d S r( )r5 fldnamesfldtypes
fldbitsizefldqualsr )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|tr|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|trh|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 namestypesZbitsizesr 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 krBn| jd kr fdd| jD }tt| j|| j }d}| j
r| j
dkrd}n
d| j
f} j j ||| d d f| nbg }| j\}}} }
t
t| jD ]} || } | j| }
t|
tr|
jd
kr|
j }t| |\}}|dkrT| d | j| j| pLd f |
|}
| jd | |
f | j| d d | _|
}t|
tr|
jd krn8 |}|| kr| d
| j| j| pd || f || qtt| 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 fixedlayoutlistr r r packedrt Zcomplete_struct_or_unionranger r r} r r~ r1 divmodsizeof_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 | | dS )Nz new_%s_typekey)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 )
StructTypestructNr9 r: r; r r r r r r s r c @ s e Zd ZdZdS ) UnionTypeunionNr r r r r r s r c @ sB e Zd ZdZdZdZdddZdd Zdd Zd
d Z d d
Z
dS )EnumTypeenumFNc 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|| dS )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 rd}t d}t d}nd}t d}t d}|||} |||}
| | } | |
} |dd | d > kr|dd | | > k r| S |dd | d > kr6|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 minmaxwarningsZ__warningregistry__clear NameErrorwarnr 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 originr5 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 | tjrtS t4 tt| ds2t t| _ t| j W 5 Q R S Q R X d S )N __typecache)
r r
ModuleType_typecache_cffi_backend global_lockhasattrr weakrefWeakValueDictionaryr )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 resecacheZres1r 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