U
    Jºcd  ã                   @   sn   d dl mZmZ d dlmZ dd„ Zdd„ Zdd„ Zd	d
„ Zeee	ƒdd„ ƒZ
dd„ Zeeee	ƒdd„ ƒZdS )é   )ÚunifyÚreify)Údispatchc                 C   s&   t  | | tft¡ t | tft¡ | S )aÐ   Register standard unify and reify operations on class
    This uses the type and __dict__ or __slots__ attributes to define the
    nature of the term
    See Also:
    >>> class A(object):
    ...     def __init__(self, a, b):
    ...         self.a = a
    ...         self.b = b
    >>> # xdoctest: +SKIP
    >>> unifiable(A)
    <class 'unification.more.A'>
    >>> x = var('x')
    >>> a = A(1, 2)
    >>> b = A(1, x)
    >>> unify(a, b, {})
    {~x: 2}
    )Ú_unifyÚaddÚdictÚunify_objectÚ_reifyÚreify_object)Úcls© r   úJ/tmp/pip-unpacked-wheel-gikjz4vx/torch/fx/experimental/unification/more.pyÚ	unifiable   s    r   c                 C   s"   t | dƒrt| |ƒS t| |ƒS dS )aŸ   Reify a Python object with a substitution
    >>> class Foo(object):
    ...     def __init__(self, a, b):
    ...         self.a = a
    ...         self.b = b
    ...     def __str__(self):
    ...         return "Foo(%s, %s)"%(str(self.a), str(self.b))
    >>> # xdoctest: +SKIP
    >>> x = var('x')
    >>> f = Foo(1, x)
    >>> print(f)
    Foo(1, ~x)
    >>> print(reify_object(f, {x: 2}))
    Foo(1, 2)
    Ú	__slots__N)ÚhasattrÚ_reify_object_slotsÚ_reify_object_dict©ÚoÚsr   r   r   r
   "   s    

r
   c                 C   s8   t  t| ƒ¡}t| j|ƒ}|| jkr(| S |j |¡ |S )N)ÚobjectÚ__new__Útyper   Ú__dict__Úupdate)r   r   ÚobjÚdr   r   r   r   8   s    
r   c                    sb   ‡ fdd„ˆ j D ƒ}t||ƒ}||kr*ˆ S t tˆ ƒ¡}tˆ j |ƒD ]\}}t|||ƒ qD|S d S )Nc                    s   g | ]}t ˆ |ƒ‘qS r   ©Úgetattr)Ú.0Úattr©r   r   r   Ú
<listcomp>B   s     z'_reify_object_slots.<locals>.<listcomp>)r   r   r   r   r   ÚzipÚsetattr)r   r   ÚattrsZ	new_attrsZnewobjÚslotr    r   r!   r   r   A   s    
r   c                 C   s   t t| j| j| jf|ƒŽ S )z! Reify a Python ``slice`` object )Úslicer   ÚstartÚstopÚstepr   r   r   r   r	   M   s    r	   c                    s^   t ˆ ƒt ˆƒkrdS tˆ dƒrJt‡ fdd„ˆ jD ƒ‡fdd„ˆjD ƒ|ƒS tˆ jˆj|ƒS dS )a­   Unify two Python objects
    Unifies their type and ``__dict__`` attributes
    >>> class Foo(object):
    ...     def __init__(self, a, b):
    ...         self.a = a
    ...         self.b = b
    ...     def __str__(self):
    ...         return "Foo(%s, %s)"%(str(self.a), str(self.b))
    >>> # xdoctest: +SKIP
    >>> x = var('x')
    >>> f = Foo(1, x)
    >>> g = Foo(1, 2)
    >>> unify_object(f, g, {})
    {~x: 2}
    Fr   c                    s   g | ]}t ˆ |ƒ‘qS r   r   ©r   r&   )Úur   r   r"   k   s     z unify_object.<locals>.<listcomp>c                    s   g | ]}t ˆ |ƒ‘qS r   r   r+   )Úvr   r   r"   l   s     N)r   r   r   r   r   ©r,   r-   r   r   )r,   r-   r   r   X   s    
þr   c                 C   s$   t | j| j| jf|j|j|jf|ƒS )z! Unify a Python ``slice`` object )r   r(   r)   r*   r.   r   r   r   r   q   s    r   N)Úcorer   r   r   r   r
   r   r   r'   r   r	   r   r   r   r   r   r   Ú<module>   s   	


