U
    ZÊ¹jµ  ã                   @   s¬   d dl mZmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
mZ d dlmZ d dlmZ d dlmZ d d	lmZmZmZmZmZ d d
lmZ G dd„ deƒZdd„ ZdS )é    )ÚBasicÚdiff)ÚS)Údefault_sort_key)ÚMatrix)ÚIntegralÚ	integrate)ÚGeometryEntity)Úsimplify)Útopological_sort)Ú
CoordSys3DÚVectorÚParametricRegionÚparametric_region_listÚImplicitRegion)Ú_get_coord_systemsc                       sD   e Zd ZdZ‡ fdd„Zedd„ ƒZedd„ ƒZedd	„ ƒZ	‡  Z
S )
ÚParametricIntegrala3  
    Represents integral of a scalar or vector field
    over a Parametric Region

    Examples
    ========

    >>> from sympy import cos, sin, pi
    >>> from sympy.vector import CoordSys3D, ParametricRegion, ParametricIntegral
    >>> from sympy.abc import r, t, theta, phi

    >>> C = CoordSys3D('C')
    >>> curve = ParametricRegion((3*t - 2, t + 1), (t, 1, 2))
    >>> ParametricIntegral(C.x, curve)
    5*sqrt(10)/2
    >>> length = ParametricIntegral(1, curve)
    >>> length
    sqrt(10)
    >>> semisphere = ParametricRegion((2*sin(phi)*cos(theta), 2*sin(phi)*sin(theta), 2*cos(phi)),                            (theta, 0, 2*pi), (phi, 0, pi/2))
    >>> ParametricIntegral(C.z, semisphere)
    8*pi

    >>> ParametricIntegral(C.j + C.k, ParametricRegion((r*cos(theta), r*sin(theta)), r, theta))
    0

    c                    s|  t |ƒ}t|ƒdkrtdƒ}nt|ƒdkr0t‚ntt|ƒƒ}ˆ jdkrLtjS | 	¡ }| 
¡ }|}tj}ttˆ jƒƒD ]}	|||	 ˆ j|	  7 }qtt|ƒdkrÆttˆ jƒƒD ]}	| ||	 ˆ j|	 ¡}qªˆ jdk�rDˆ jd }
t||
ƒ}ˆ j|
 d ˆ j|
 d  }}t|tƒ�r t| |¡ƒ}nt| ¡ | ƒ}t||
||fƒ}�nˆ jdk�r|  ˆ jˆ j¡\}}t||ƒ}t||ƒ}t| |¡ƒ}t|tƒ�rž| |¡}n|| ¡  }t|ƒ}ˆ j| d ˆ j| d  }}ˆ j| d ˆ j| d  }}t||||f|||fƒ}nP|  ˆ jˆ j¡}tˆ jƒ |¡ ¡ }t|| ƒ}‡ fdd„|D ƒ}t|f|žŽ }t|tƒ�sh|S tƒ  | |ˆ ¡S d S )Nr   ÚCé   é   c                    s*   g | ]"}|ˆ j | d  ˆ j | d f‘qS )r   r   )Úlimits)Ú.0Úvar©Úparametricregion© úU/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/vector/integrals.pyÚ
<listcomp>k   s     z.ParametricIntegral.__new__.<locals>.<listcomp>) r   Úlenr   Ú
ValueErrorÚnextÚiterÚ
dimensionsr   ZZeroÚbase_vectorsÚbase_scalarsr   ÚzeroÚrangeZ
definitionÚsubsÚ
parametersr   r   Ú
isinstancer
   ÚdotZ	magnituder   Ú_bounds_caseÚcrossr   ZjacobianZdetr   ÚsuperÚ__new__)ÚclsÚfieldr   Z	coord_setZ	coord_sysr#   r$   ZparametricfieldÚrÚiZ	parameterZr_diffÚlowerÚupperZ	integrandÚresultÚuÚvZr_uZr_vZnormal_vectorZlower_uZupper_uZlower_vZupper_vÚ	variablesZcoeffÚl©Ú	__class__r   r   r.   +   sZ    





zParametricIntegral.__new__c                    sz   t | ¡ ƒ}g }|D ]H‰|ˆ d ‰ |ˆ d ‰ˆ  ¡ ‰ ˆ ¡ ‰| ‡ ‡‡fdd„|D ƒ¡ q|sf|S t||ftd�S d S )Nr   r   c                 3   s6   | ].}ˆ|krˆ   |h¡s&ˆ  |h¡rˆ|fV  qd S )N)Ú
issuperset)r   Úq©Zlower_pÚpZupper_pr   r   Ú	<genexpr>   s
      ÿz2ParametricIntegral._bounds_case.<locals>.<genexpr>)Úkey)ÚlistÚkeysZatomsÚextendr   r   )r/   r(   r   ÚVÚEr   r>   r   r+   s   s    zParametricIntegral._bounds_casec                 C   s
   | j d S )Nr   ©Úargs©Úselfr   r   r   r0   ‡   s    zParametricIntegral.fieldc                 C   s
   | j d S )Nr   rG   rI   r   r   r   r   ‹   s    z#ParametricIntegral.parametricregion)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r.   Úclassmethodr+   Úpropertyr0   r   Ú__classcell__r   r   r:   r   r      s   H

r   c                 G   s˜   t |ƒdkrŠt|d tƒr(t| |d ƒS t|d tƒrPt|d ƒd }t| |ƒS t|d tƒrŠt|d ƒ}d}|D ]}|t| |ƒ7 }qr|S t| f|žŽ S )a˜  
    Compute the integral of a vector/scalar field
    over a a region or a set of parameters.

    Examples
    ========
    >>> from sympy.vector import CoordSys3D, ParametricRegion, vector_integrate
    >>> from sympy.abc import x, y, t
    >>> C = CoordSys3D('C')

    >>> region = ParametricRegion((t, t**2), (t, 1, 5))
    >>> vector_integrate(C.x*C.i, region)
    12

    Integrals over some objects of geometry module can also be calculated.

    >>> from sympy.geometry import Point, Circle, Triangle
    >>> c = Circle(Point(0, 2), 5)
    >>> vector_integrate(C.x**2 + C.y**2, c)
    290*pi
    >>> triangle = Triangle(Point(-2, 3), Point(2, 3), Point(0, 5))
    >>> vector_integrate(3*C.x**2*C.y*C.i + C.j, triangle)
    -8

    Integrals over some simple implicit regions can be computed. But in most cases,
    it takes too long to compute over them. This is due to the expressions of parametric
    representation becoming large.

    >>> from sympy.vector import ImplicitRegion
    >>> c2 = ImplicitRegion((x, y), (x - 2)**2 + (y - 1)**2 - 9)
    >>> vector_integrate(1, c2)
    6*pi

    Integral of fields with respect to base scalars:

    >>> vector_integrate(12*C.y**3, (C.y, 1, 3))
    240
    >>> vector_integrate(C.x**2*C.z, C.x)
    C.x**3*C.z/3
    >>> vector_integrate(C.x*C.i - C.y*C.k, C.x)
    (Integral(C.x, C.x))*C.i + (Integral(-C.y, C.x))*C.k
    >>> _.doit()
    C.x**2/2*C.i + (-C.x*C.y)*C.k

    r   r   )	r   r)   r   r   r   r   Úvector_integrater	   r   )r0   ÚregionZregions_listr5   Úregr   r   r   rR   �   s    .
rR   N)Z
sympy.corer   r   Zsympy.core.singletonr   Zsympy.core.sortingr   Zsympy.matricesr   Zsympy.integralsr   r   Zsympy.geometry.entityr	   Zsympy.simplify.simplifyr
   Zsympy.utilities.iterablesr   Zsympy.vectorr   r   r   r   r   Zsympy.vector.operatorsr   r   rR   r   r   r   r   Ú<module>   s    