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 s | j }|ddkst| }|rT|dr@d|kr@d| }n|d dkrTd| }t||}|d|}d |krtd
| |f |S ) N&r *z&[(%s)r z[( $z-cannot generate '%s' in %s: unknown type name) c_name_with_markercountAssertionErrorstrip
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 sN z|j | }W n: tk
rH | ||}|j | |}||ksDtY nX |S N)_cached_btypesKeyErrorbuild_backend_type
setdefaultr )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) r3 r4 r: r r r r r s
r c @ s$ e Zd Zdd Zdd Zdd ZdS )BaseTypec C s | j |j ko| | kS r* ) __class__r: r! otherr r r __eq__J s zBaseType.__eq__c C s
| |k S r* r rB r r r __ne__N s zBaseType.__ne__c C s t | jt| fS r* )hashrA tupler: r% r r r __hash__Q s zBaseType.__hash__N)r; r< r= rD rE rH 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/ r0 r r r r- [ s zVoidType.build_backend_typeNr; r< r= r9 rJ r- r r r r rI U s rI 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)r; r< r= rP r r r r rO a s rO c 4 @ s e Zd ZdZdddddddddddddddddddddddddddddddddddddddddddddddddddd3Zdd Zd d
Zd d Zd
d Zdd Z dd Z
dS )
PrimitiveTyper7 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" || j kst|| _|d | _d S Nr )ALL_PRIMITIVE_TYPESr r7 r r! r7 r r r rJ s zPrimitiveType.__init__c C s | j | j dkS )NrS r^ r7 r% r r r is_char_type s zPrimitiveType.is_char_typec C s | j | j dkS )NrT r` r% r r r r) s z PrimitiveType.is_integer_typec C s | j | j dkS )NrU r` r% r r r
is_float_type s zPrimitiveType.is_float_typec C s | j | j dkS )NrV r` r% r r r rP s z PrimitiveType.is_complex_typec C s t | |d| jS )NZnew_primitive_type)rL r7 rM r r r r- s z PrimitiveType.build_backend_typeN) r; r< r= r9 r^ rJ ra r) rb rP r- r r r r rQ f sv 7rQ c @ s( e Zd ZdZdd Zdd Zdd ZdS ) UnknownIntegerTyperR c C s || _ |d | _d S r] r7 r r_ r r r rJ 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 compilationNotImplementedErrorr7 rM r r r r- s z%UnknownIntegerType.build_backend_typeN)r; r< r= r9 rJ r) r- r r r r rc s rc c @ s e Zd ZdZdd Zdd ZdS )UnknownFloatTyperR c C s || _ |d | _d S r] rd r_ r r r rJ s zUnknownFloatType.__init__c C s t d| j d S )Nz2float type '%s' can only be used after compilationre rM r r r r- s z#UnknownFloatType.build_backend_typeNrN r r r r rg s rg 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& )r6 argr r r r8 s z-BaseFunctionType.__init__.<locals>.<listcomp>...Zvoidz, r r r ) ri r# rj rk append
_base_patternjoinr r )r! ri r# rj rk Zreprargsr r r r rJ s
zBaseFunctionType.__init__)Nr; r< r= r9 rJ r r r r rh s rh 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 rM r r r r- s z"RawFunctionType.build_backend_typec C s t | j| j| j| jS r* )FunctionPtrTyperi r# rj rk r% r r r as_function_pointer s z#RawFunctionType.as_function_pointerN)r; r< r= ro r? r- rt r r r r rr s rr c @ s e Zd ZdZdd Zdd ZdS )rs 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# r3 ri rn rk rj _backendZ FFI_STDCALLAttributeErrorrL rG )r! r/ r0 r# ri tpZabi_argsr r r r- s$
z"FunctionPtrType.build_backend_typec C s t | j| j| j| jS r* )rr ri r# rj rk r% r r r as_raw_function s z FunctionPtrType.as_raw_functionN)r; r< r= ro r- rx r r r r rs s rs c @ s" e Zd ZdZdddZdd ZdS ) PointerType)totyper
r c C s> || _ || _t|d}|jr*d| f }|jd|| _d S )Nz *&r r )rz r
r r> r
r r )r! rz r
Zextrar r r rJ s
zPointerType.__init__c C s | j j||dd}t| |d|S )NT)r1 new_pointer_type)rz r3 rL )r! r/ r0 ZBItemr r r r- s z PointerType.build_backend_typeN)r rN r r r r ry s
ry c C s
t | tS r* )ry r )rz r r r ConstPointerType s r| c @ s e Zd ZdZdddZdS )NamedPointerType)rz r7 r c C s" t | || || _|d | _d S r] )ry rJ r7 r )r! rz r7 r
r r r rJ s zNamedPointerType.__init__N)r rq 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&[]rm z
&[/*...*/]z&[%s]r )r r r r )r! r r Zbracketsr r r rJ ) 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 )Nrm z)cannot render the type %r: unknown lengthZnew_array_type)r r r r3 ry rL )r! r/ r0 ZBPtrItemr r r r- 9 s
z ArrayType.build_backend_typeN)r; r< r= r9 r> rJ r r- r r r r r~ % s
r~ rW c @ s, e Zd ZdZdZdd Zdd Zdd ZdS ) StructOrUnionOrEnumrR Nc C s$ | j pd| j| jf }|d | _d S )Nz%s %sr ) forcenamekindr7 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 dst| j d d S )Nr )r endswithr r% r r r get_official_nameP s z%StructOrUnionOrEnum.get_official_name)r; r< r= r9 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* )r7 fldnamesfldtypes
fldbitsizefldqualsr )r! r7 r r r r r r r rJ [ s zStructOrUnion.__init__c c s> | j d k r:t| j| j D ] \}}|dkrt|tr|V qd S )Nr )r zipr
isinstancer )r! r7 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 r7 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 rn rG r r r r ) r! namestypesZbitsizesr r7 r r r
r r r
force_flattenw s
zStructOrUnion.force_flattenc C s$ t | |||}|s | || |S r* )r r3 finish_backend_type)r! r/ r0 r1 r2 r r r r3 s
z StructOrUnion.get_cached_btypec s2 | 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| npg }| 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 kr| dkstn8 |}|| 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 )r3 )r6 rw r/ r0 r r r8 s z5StructOrUnion.finish_backend_type.<locals>.<listcomp>r ) r r rm z field '%s.%s' has a bogus size?z{}z=field '%s.%s' is declared as %d bytes, but is really %d bytes) completedrf r7 r+ r fixedlayoutlistr r r packedru Zcomplete_struct_or_unionranger r r~ r r r3 divmodsizeof_verification_errorr r rn )r! r/ r0 r2 r Zlst extra_flagsZfieldofsZ fieldsizeZ totalsizeZtotalalignmentrT 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 rn rL r r rM r r r r- s
z StructOrUnion.build_backend_type)N)T)F)r; r< r= r r r r rJ r r r r3 r r r r- r r r r r U s
?r c @ s e Zd ZdZdS )
StructTypestructNr; 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* )r7 enumerators
enumvalues baseinttyper )r! r7 r r r r r r rJ s
zEnumType.__init__c C s4 t | | | jd kr0| }d|dd | _d S )Nr r _)r r r r r )r! r r7 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_baseinttyperL r r r )r! r/ r0 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 rX rY rZ r[ r r z=%s values don't all fit into either 'long' or 'unsigned long')r r3 r minmaxwarningsZ__warningregistry__clear NameErrorwarnr rQ r r )
r! r/ r0 Zsmallest_valueZ
largest_valuer ZsignZ
candidate1Z
candidate2Zbtype1Zbtype2Zsize1Zsize2r r r r sH
zEnumType.build_baseinttype)N) r; r< r= r r r rJ 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 originr7 Z
structnamerw 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}|rtz |j| W S tk
r8 Y nX zt|j|| }W n4 tk
r } 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) popr Z
_typecacher, r5 ru rf r get)
Zsrctyper/ funcnameri kwdsr resecacheZres1r r r rL J s" $
rL c C s t d| d|S )N?r{ rK )r/ r2 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 )ri r str)r r7 r r r attach_exception_infod s r )N)N)-r r lockr errorr r r r r r r objectr r@ rI Z void_typerO rQ rc rg rh rr rs ry Z
voidp_typer| Zconst_voidp_typer} r~ Zchar_array_typer r r r r r r r r r r rL r r r r r r <module> sN
/
L
G