U
    9%eHE                     @   s@  d dl Z d dlZd dlZd dlZ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Zd dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z  d dl!m"Z" ddl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 dd	l8m9Z9 ej:j;< rJd d
lm=Z= ddl#m>Z>m?Z? d dl@mAZA G dd dZBG dd dZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd  ZKd!d" ZLd#d$ ZMe
ejN d%d&d'ZOd(d) ZPd*d+ ZQdEd,d-ZRdFd.d/ZSd0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdddddd d"d$d-d/gZTdS )G    N)dedent)Type)AnyType	AwaitTypeBoolTypeComplexTypeDeviceObjTypeDictTypeEnumType	FloatType
FutureTypeInterfaceTypeIntTypeListTypeNoneType
NumberTypeOptionalTypeStreamObjType
StringType
TensorType	TupleType	UnionType)get_source_lines_and_file   )_Await_qualified_nameAnyBroadcastingList1BroadcastingList2BroadcastingList3DictFutureis_awaitis_dict	is_futureis_ignored_fnis_listis_optionalis_tupleis_unionListOptionalTupleUnion   )_get_script_class)RRefType)is_rrefRRef)OpOverloadPacketc                   @   s   e Zd Zdd Zdd ZdS )Modulec                 C   s   || _ || _d S N)namemembers)selfr6   r7    r9   T/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/jit/annotations.py__init__E   s    zModule.__init__c                 C   s>   z| j | W S  tk
r8   td| j d| d Y nX d S )NzModule z has no member called )r7   KeyErrorRuntimeErrorr6   r8   r6   r9   r9   r:   __getattr__I   s    zModule.__getattr__N)__name__
__module____qualname__r;   r?   r9   r9   r9   r:   r4   D   s   r4   c                   @   sN   e Zd Zeddejiejeddeieeee	e
eed
Zdd Zdd	 Zd
S )EvalEnvtorchTensortypingr,   )
rD   rE   rF   r,   r*   r    r+   r-   r!   Awaitc                 C   s    || _ tjj rt| jd< d S )Nr2   )rcbrD   distributedrpcis_availabler2   env)r8   rH   r9   r9   r:   r;   `   s    zEvalEnv.__init__c                 C   s4   || j kr| j | S | jd k	r(| |S tt|d S r5   )rL   rH   getattrbuiltinsr>   r9   r9   r:   __getitem__e   s
    



zEvalEnv.__getitem__N)r@   rA   rB   r4   rD   rE   r,   r*   r    r+   r-   r!   r   rL   r;   rO   r9   r9   r9   r:   rC   R   s   rC   c           	      C   s   t | trt| j|}n
t| |}|d k	rJ|rJ|\}}|dd  }||f}|d krd\}}z"tdt| d }t|}W n tk
r   Y nX |d k	rt	|||}|S )Nr.   )NN r   )

isinstancer3   try_real_annotationsopr   joinr   get_type_line	TypeErrorparse_type_line)	fnrH   locZ	is_method	signatureZparam_typesreturn_type	type_linesourcer9   r9   r:   get_signaturem   s"    

r^   c                 C   s   t | pt | S r5   )inspect
isfunctionismethodZthe_callabler9   r9   r:   is_function_or_method   s    rc   c                 C   s6   t | st| r| j} t | r.t| jd k	S dS d S )NF)rc   callable__call__r_   getfullargspecvarargsrb   r9   r9   r:   	is_vararg   s
    rh   c                 C   sl   t | tr| j} t| s0t| r0t| jr0| j} t| rVt| rJt| } t	| j
S dd t|D S d S )Nc                 S   s   g | ]}t |qS r9   )str).0ir9   r9   r:   
<listcomp>   s     z#get_param_names.<locals>.<listcomp>)rQ   r3   rS   rc   rd   re   r%   r_   unwraprf   argsrange)rX   Zn_argsr9   r9   r:   get_param_names   s    

rp   c              	   C   s   zt dt| d }W n ttfk
r4   Y d S X |d krBd S t|}t|jdkrt	|jd tj
rtjj|d|jd j dt|jdkst	|jd tjstjj|dd S )NrP   r   r.   zCannot instantiate class 'z' in a script functionz$Expected a single top-level function)r   rT   r   OSErrorrV   astparselenbodyrQ   ClassDefrD   jitZfrontendZFrontendErrorr6   FunctionDef)rX   rY   r]   Zpy_astr9   r9   r:   check_fn   s"    
   ry   c                 C   sD   t | ddd}t|D ]}d|jkrtd|  dqt|||S )zKEvaluate statement as long as it does not contain any method/function callsrP   eval)modeCALLz/Type annotation should not contain calls, but 'z' does)compiledisget_instructionsopnamer=   rz   )stmtglobrY   bytecodeZinsnr9   r9   r:   _eval_no_call   s    

r   c           	   
      s   t | \}}zt|i t|}W n0 ttfk
rP } ztd|W 5 d}~X Y nX t|tsb|f}zt|i t|}W n0 ttfk
r } ztd|W 5 d}~X Y nX  fdd|D }|t| fS )zParses a type annotation specified as a comment.

    Example inputs:
        # type: (Tensor, torch.Tensor) -> Tuple[Tensor]
        # type: (Tensor, Tuple[Tensor, Tensor]) -> Tensor
    z6Failed to parse the argument list of a type annotationNz4Failed to parse the return type of a type annotationc                    s   g | ]}t | qS r9   )ann_to_typerj   annrY   r9   r:   rl      s     z#parse_type_line.<locals>.<listcomp>)	split_type_liner   rC   	NameErrorSyntaxErrorr=   rQ   tupler   )	r\   rH   rY   Zarg_ann_strZret_ann_strZarg_anneZret_ann	arg_typesr9   r   r:   rW      s*    
rW   c                    sn  d |  d}dd t|D }tt fdd|}tdttfdd|}t|d	krtd
ttfdd|}t|d	krtdt|d	 d	  d d d dS t|dkr|d	 d 	 S d}g }|D ]4\}}d|kr ||f} qqވ |kr|
| q|dkr>tdddd |D  d  fdd}t||}	d|	}
|d d|
S )zETries to find the line containing a comment with the type annotation.# type:
c                 S   s   g | ]\}}||fqS r9   r9   )rj   line_numliner9   r9   r:   rl      s     z!get_type_line.<locals>.<listcomp>c                    s    | d kS Nr.   r9   r   type_commentr9   r:   <lambda>       zget_type_line.<locals>.<lambda>z!# type:\ ignore(\[[a-zA-Z-]+\])?$c                    s     | d  S r   searchr   )type_patternr9   r:   r     r   r   z&#[	 ]*type[	 ]*(?!: ignore(\[.*\])?$):c                    s     | d S r   r   r   )wrong_type_patternr9   r:   r     r   zThe annotation prefix in line z* is probably invalid.
It must be '# type:'zl
See PEP 484 (https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)z
for examplesNr.   z# type: (...) -> z_Return type line '# type: (...) -> ...' not found on multiline type annotation
for type lines:
c                 S   s   g | ]}|d  qS )r.   r9   )rj   r   r9   r9   r:   rl   %  s     zl
(See PEP 484 https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)c                    s"   | |   t  d  }| S r5   )findrt   strip)r   Z	item_typer   r9   r:   get_parameter_type)  s    z)get_type_line.<locals>.get_parameter_typez, z...)split	enumeratelistfilterrer}   rt   r=   ri   r   appendrT   mapreplace)r]   linesZ
type_linesZwrong_type_linesZreturn_lineZparameter_type_linesr   r   r   typesZparameter_typesr9   )r   r   r   r:   rU      s\    






rU   c                 C   sZ   t d}z| d}W n tk
r4   tddY nX | ||  | |d d  fS )a  Splits the comment with the type annotation into parts for argument and return types.

    For example, for an input of:
        # type: (Tensor, torch.Tensor) -> Tuple[Tensor, Tensor]

    This function will return:
        ("(Tensor, torch.Tensor)", "Tuple[Tensor, Tensor]")

    r   z->z3Syntax error in type annotation (cound't find `->`)Nr   )rt   index
ValueErrorr=   r   )r\   Zstart_offsetZ	arrow_posr9   r9   r:   r   3  s    
r   c                    s   zt | W n tk
r$   Y dS X jgdd j D  }tfdd|D r\dS  fddj D }tj }||fS )z:Tries to use the Py3.5+ annotation syntax to get the type.Nc                 S   s   g | ]
}|j qS r9   )
annotationrj   pr9   r9   r:   rl   Q  s    z(try_real_annotations.<locals>.<listcomp>c                 3   s   | ]}| j kV  qd S r5   )emptyr   )sigr9   r:   	<genexpr>T  s     z'try_real_annotations.<locals>.<genexpr>c                    s   g | ]}t |j qS r9   )r   r   r   r   r9   r:   rl   W  s     )r_   rZ   r   return_annotation
parametersvaluesallr   )rX   rY   Z
all_annotsr   r[   r9   )rY   r   r:   rR   G  s    rR   )r   c                    sZ   t | }|std| j ddd |D } fdd|D }tj|}|sVt S |S )NzNo enum values defined for: ''c                 S   s   h | ]}t |jqS r9   )typevalue)rj   vr9   r9   r:   	<setcomp>c  s     z&get_enum_value_type.<locals>.<setcomp>c                    s   g | ]}t | qS r9   try_ann_to_type)rj   tr   r9   r:   rl   d  s     z'get_enum_value_type.<locals>.<listcomp>)r   r   	__class__rD   _CZunify_type_listr   get)r   rY   Zenum_valuesr   Zir_typesresr9   r   r:   get_enum_value_type^  s    r   c                 C   sP   t | tjrdS t | tjtjtjtjtjtjtj	tj
tjf	rLtd dS dS )NTzTorchScript will treat type annotations of Tensor dtype-specific subtypes as if they are normal Tensors. dtype constraints are not enforced in compilation either.F)
issubclassrD   rE   Z
LongTensorZDoubleTensorZFloatTensorZ	IntTensorZShortTensorZ
HalfTensorZ
CharTensorZ
ByteTensorZ
BoolTensorwarningswarn)r   r9   r9   r:   	is_tensorp  s(    r   c                 C   s   d S r5   r9   )Zinpr9   r9   r:   	_fake_rcb  s    r   c                    sL  | t jjkrt S | d kr$t S t | r>t| r>t S t	| rt
| jdkrj| jd dkrjtg S t fdd| jD S t| rt| jd  }|rt|S t| r(t| jd  }t| jd  }|d krtd| jd  d   |d krtd| jd  d   t||S t| rt| jd td rT| jd }n
| jd }t| }d}|st|t| t|t t|S t| r.t| jttthkrt  S g }	| jD ]Z}
|
d kr|	!t  t|
 }d}|st|t| t|t |	!| qt"|	S t#j$j%& rZt'| rZt(t| jd  S t)| rxt*t| jd  S t+| rt,| d	rt| jd  nt- }t.|S | tkrt/ S | tkrt0 S | tkrt1 S | t2krt3 S | t4krt5 S | t6krt- S | td kr0t S t | rRt,| d
rRt7| j8S | t#j9krft: S | t#j;krzt< S | t#j=krt1 S t | rt| t>j?rt@| d krt#jAjBC|  }|D }ntE| }tF|tG|  tH| S t | r.t@| }|d k	r|S t#jIJ| r.t#jAjBC|  S |d kr<tK}t#jLM|  |S )Nr.   r   r9   c                    s   g | ]}t | qS r9   r   )rj   ar   r9   r:   rl     s     z#try_ann_to_type.<locals>.<listcomp>Unknown type annotation: '' at zWUnsupported annotation {} could not be resolved because {} could not be resolved. At
{}__args____torch_script_interface__)Nr_   	Signaturer   r   ZgetInferredr   r   isclassr   r(   rt   r   r   r&   r   r   r#   r   	highlightr	   r'   r   r   AssertionErrorformatreprr   r)   setintfloatcomplexr   r   r   rD   rI   rJ   rK   r1   r0   r$   r   r"   hasattrr   r   r   r   r   ri   r   boolr   r   r   r   Zdevicer   ZStreamr   ZdtypeenumEnumr/   rw   _scriptZ_recursive_compile_classZqualified_namer   r
   r   r   _jit_internalZcan_compile_classr   r   Z_resolve_type_from_object)r   rY   rH   Z	elem_typekeyr   	containedZ
valid_typemsginnerr   Z
maybe_typeZelementTypeZscripted_classr6   Zmaybe_script_classr9   r   r:   r     s    





$



$











r   c                 C   s4   t | ||}|d k	r|S td|  d|  d S )Nr   r   )r   r   r   )r   rY   rH   Zthe_typer9   r9   r:   r     s    r   r   r*   r   r   r   r,   r(   r&   r    r#   r'   r)   r   r   r   r   r   r   r   r	   r   )N)N)Urr   rN   r~   r   r_   r   r   textwrapr   rF   r   rD   Ztorch._Cr   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   Ztorch._sourcesr   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   _stater/   rI   rJ   rK   r0   r1   r2   Z
torch._opsr3   r4   rC   r^   rc   rh   rp   ry   r   rW   rU   r   rR   r   r   r   r   r   r   __all__r9   r9   r9   r:   <module>   s   XXC
k
