| Server IP : 93.86.61.54 / Your IP : 216.73.216.206 Web Server : Apache/2.4.62 (Ubuntu) System : Linux rasin.ddns.net 6.8.0-124-generic #124~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Tue May 26 21:05:19 UTC x86_64 User : www-data ( 33) PHP Version : 8.4.22 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /usr/lib/python3/dist-packages/sympy/simplify/__pycache__/ |
Upload File : |
o
�8Va�� � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
d dlmZ d dlmZ d d lmZmZ d d
lmZ d d lmZ d d lmZ d d
lm Z d dl m Z m Z m Z m!Z!m"Z"m#Z#m$Z$ d dl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z- d dl.m/Z/ d dl0m1Z1 d dl2m3Z3 d dl4m5Z5 d dl6m7Z7m8Z8 d dl9m:Z: dd� Z;dd� Z<dd� Z=ded d �Z>d d � Z?d!d"� Z@d#d$� ZAdfd&d'�ZBdfd(d)�ZCd*d+� ZDdgd-d.�ZEd/d0� ZFdgd1d2�ZGd3d4� ZHdhd6d7�ZId8d9� ZJdgd:d;�ZKd<d=� ZLd>d?� ZMd@dA� ZNdgdBdC�ZOdfdDdE�ZPdfdFdG�ZQdHdI� ZRdfdJdK�ZSdLdM� ZTdNdO� ZUe:�rMeVeWe8e;e<e=e?e@eBeCeDeEeFeGeIeKeMe>eNeOePeQeLeReSf��\Z;Z<Z=Z?Z@ZBZCZDZEZFZGZIZKZMZ>ZNZOZPZQZLZRZSeBe;feCe;fe7gZXeIeBe;feCe;fe;gfZYeNeEe;feNeEeHe;fe7gZZe@eHfe7gZ[e@e?e@eKe@eMe@e;fZ\e@e?eGe@e?eIfeBeDeIe@feZeXeFeYe@eFeFe[fe7gZ]dPdQ� fdRdS�Z^didTdU�Z_dV�`� ZaebeVeceaeVeWed� jeea�����ZfdWdX� Zgd5ahdedYdZ�Zid[d\� Zjd]d^� Zkd_d`� Zldadb� Zmdcdd� Znd5S )j� )� defaultdict��Add)�S)�ordered)�Expr)�Factors� gcd_terms� factor_terms��
expand_mul)�Mul)�pi�I)�Pow��Dummy)�sympify)�binomial)�cosh�sinh�tanh�coth�sech�csch�HyperbolicFunction)�cos�sin�tan�cot�sec�csc�sqrt�TrigonometricFunction)�
perfect_power��factor�� bottom_up)�greedy)�identity�debug)� SYMPY_DEBUGc C s | � � �� �� S )zSimplification of rational polynomials, trying to simplify
the expression, e.g. combine things like 3*x + 2*x, etc....
)�normalr& �expand��rv� r1 �3/usr/lib/python3/dist-packages/sympy/simplify/fu.py�TR0 s r3 c C � dd� }t | |�S )z�Replace sec, csc with 1/cos, 1/sin
Examples
========
>>> from sympy.simplify.fu import TR1, sec, csc
>>> from sympy.abc import x
>>> TR1(2*csc(x) + sec(x))
1/cos(x) + 2/sin(x)
c S sH t | t�r| jd }tjt|� S t | t�r"| jd }tjt|� S | S �Nr )�
isinstancer �argsr �Oner r! r �r0 �ar1 r1 r2 �f4 s
zTR1.<locals>.fr' �r0 r; r1 r1 r2 �TR1( s
r= c C r4 )a@ Replace tan and cot with sin/cos and cos/sin
Examples
========
>>> from sympy.simplify.fu import TR2
>>> from sympy.abc import x
>>> from sympy import tan, cot, sin, cos
>>> TR2(tan(x))
sin(x)/cos(x)
>>> TR2(cot(x))
cos(x)/sin(x)
>>> TR2(tan(tan(x) - sin(x)/cos(x)))
0
c S sL t | t�r| jd }t|�t|� S t | t�r$| jd }t|�t|� S | S r5 )r6 r r7 r r r r9 r1 r1 r2 r; R s
zTR2.<locals>.fr' r< r1 r1 r2 �TR2@ s
r> Fc � � fdd�}t | |�S )a� Converts ratios involving sin and cos as follows::
sin(x)/cos(x) -> tan(x)
sin(x)/(cos(x) + 1) -> tan(x/2) if half=True
Examples
========
>>> from sympy.simplify.fu import TR2i
>>> from sympy.abc import x, a
>>> from sympy import sin, cos
>>> TR2i(sin(x)/cos(x))
tan(x)
Powers of the numerator and denominator are also recognized
>>> TR2i(sin(x)**2/(cos(x) + 1)**2, half=True)
tan(x/2)**2
The transformation does not take place unless assumptions allow
(i.e. the base must be positive or the exponent must be an integer
for both numerator and denominator)
>>> TR2i(sin(x)**a/(cos(x) + 1)**a)
sin(x)**a/(cos(x) + 1)**a
c s� | j s| S | �� \�� �js� jr| S �fdd����� ���fdd�t��� �D �}�s/| S � �� � � �fdd�t� �� �D �}� sE| S ��fdd�}|�|� |� |� g }�D ]�}t|t�r�t|j d d d
�}|� v r�� | �| kr�|�
t |j d ��| � d �|< � |< qZ�r�d | }|� v r�� | �| kr�|�
t |j d d ��| � d �|< � |< qZt|t�r�t|j d d d
�}|� v r� | �| kr�|�
t |j d ��| � d �|< � |< qZ��r@|j �r@|j d t
ju �r@t|j d t��r@t|j d j d d d
�}|� v �r@� | �| k�r@� | j�s'|j�r@|�
t |j d d ��| � d �|< � |< qZ|�rpt|d
d� ��� D � � tdd� � �� D �� } | tdd� |D �� tdd� |D �� 9 } | S )Nc sF |j s| jo"| jttfv p"� o"| jo"t| j�dko"tdd� | jD ��S )N� c s s( � | ]}t d d� t�|�D ��V qdS )c s s* � | ]}t |t�p|jo|jtu V qd S �N)r6 r �is_Pow�base)�.0�air1 r1 r2 � <genexpr>� s � "�z8TR2i.<locals>.f.<locals>.ok.<locals>.<genexpr>.<genexpr>N)�anyr
� make_args�rD r: r1 r1 r2 rF � s � ��z.TR2i.<locals>.f.<locals>.ok.<locals>.<genexpr>) �
is_integer� is_positive�funcr r �is_Add�lenr7 rG )�k�e��halfr1 r2 �ok� s � ���zTR2i.<locals>.f.<locals>.okc �( g | ]}�|� | �s|� � |�f�qS r1 ��pop�rD rO )�nrS r1 r2 �
<listcomp>� � ( z#TR2i.<locals>.f.<locals>.<listcomp>c rT r1 rU rW )�drS r1 r2 rY � rZ c s� g }| D ]!}|j r%t|j�dkr%� rt|�nt|�}||kr%|�||f� q|rat|�D ]
\}\}}| |= |||< q,t|� �� }|D ]}| | || }�||�rV|| |< qB|�||f� qB~d S d S �N� ) rM rN r7 r& r
�append� enumerater
�as_powers_dict)r[ �ddone�newkrO �knew�i�v)rR rS r1 r2 � factorize� s& �
�z"TR2i.<locals>.f.<locals>.factorizer F��evaluater] r@ c S � g | ]
\}}|r|| �qS r1 r1 �rD �brP r1 r1 r2 rY � � c S ri r1 r1 rj r1 r1 r2 rY � rl c S s g | ]\}}|| �qS r1 r1 rj r1 r1 r2 rY � � )�is_Mul�as_numer_denom�is_Atomr` �list�keysr6 r r r7 r^ r rM r r8 rJ rK r
�items)r0 �ndonera rf �trO r: �a1rQ )r[ rX rS r2 r; z sh
�
� �(�"��(zTR2i.<locals>.fr' )r0 rR r; r1 rQ r2 �TR2i^ s
Urw c �" ddl m� � fdd�}t| |�S )aR Induced formula: example sin(-a) = -sin(a)
Examples
========
>>> from sympy.simplify.fu import TR3
>>> from sympy.abc import x, y
>>> from sympy import pi
>>> from sympy import cos
>>> TR3(cos(y - x*(y - x)))
cos(x*(x - y) + y)
>>> cos(pi/2 + x)
-sin(x)
>>> cos(30*pi/2 + x)
-cos(x)
r ��signsimpc s� t | t�s| S | �� | jd ��} t | t�s| S | jd tjd jtjd | jd j u r4du rTn | S ttttt t
t
t t t t t i}|| j tjd | jd �} | S )Nr � r@ T)
r6 r# rL r7 r �PirK r r r r r r! )r0 �fmapry r1 r2 r; � s
: � zTR3.<locals>.f)�sympy.simplify.simplifyrz r( r< r1 ry r2 �TR3� s
r c C s | S )a� Identify values of special angles.
a= 0 pi/6 pi/4 pi/3 pi/2
----------------------------------------------------
sin(a) 0 1/2 sqrt(2)/2 sqrt(3)/2 1
cos(a) 1 sqrt(3)/2 sqrt(2)/2 1/2 0
tan(a) 0 sqt(3)/3 1 sqrt(3) --
Examples
========
>>> from sympy import pi
>>> from sympy import cos, sin, tan, cot
>>> for s in (0, pi/6, pi/4, pi/3, pi/2):
... print('%s %s %s %s' % (cos(s), sin(s), tan(s), cot(s)))
...
1 0 0 zoo
sqrt(3)/2 1/2 sqrt(3)/3 sqrt(3)
sqrt(2)/2 sqrt(2)/2 1 1
1/2 sqrt(3)/2 sqrt(3) sqrt(3)/3
0 1 zoo 0
r1 r/ r1 r1 r2 �TR4� s r� c s � ����fdd�}t | |�S )a+ Helper for TR5 and TR6 to replace f**2 with h(g**2)
Options
=======
max : controls size of exponent that can appear on f
e.g. if max=4 then f**4 will be changed to h(g**2)**2.
pow : controls whether the exponent must be a perfect power of 2
e.g. if pow=True (and max >= 6) then f**6 will not be changed
but f**8 will be changed to h(g**2)**4
>>> from sympy.simplify.fu import _TR56 as T
>>> from sympy.abc import x
>>> from sympy import sin, cos
>>> h = lambda x: 1 - x
>>> T(sin(x)**3, sin, cos, h, 4, False)
(1 - cos(x)**2)*sin(x)
>>> T(sin(x)**6, sin, cos, h, 6, False)
(1 - cos(x)**2)**3
>>> T(sin(x)**6, sin, cos, h, 6, True)
sin(x)**6
>>> T(sin(x)**8, sin, cos, h, 10, True)
(1 - cos(x)**2)**4
c s | j r | jj� ks | S | jjs| S | jdk dkr| S | j�kdkr#| S | jdkr*| S | jdkr;��| jjd �d �S | jd dkr]| jd }� | jjd ���| jjd �d �| S | jdkred}n �st| jd rn| S | jd }nt| j�}|s}| S | jd }��| jjd �d �| S )Nr Tr] r@ r{ )rB rC rL �exp�is_realr7 r$ )r0 rP �p�r; �g�h�max�powr1 r2 �_f1 s4
,
z_TR56.<locals>._fr' )r0 r; r� r� r� r� r� r1 r� r2 �_TR56 s
#r� r{ c C � t | ttdd� ||d�S )a� Replacement of sin**2 with 1 - cos(x)**2.
See _TR56 docstring for advanced use of ``max`` and ``pow``.
Examples
========
>>> from sympy.simplify.fu import TR5
>>> from sympy.abc import x
>>> from sympy import sin
>>> TR5(sin(x)**2)
1 - cos(x)**2
>>> TR5(sin(x)**-2) # unchanged
sin(x)**(-2)
>>> TR5(sin(x)**4)
(1 - cos(x)**2)**2
c S � d| S r\ r1 ��xr1 r1 r2 �<lambda>i � zTR5.<locals>.<lambda>�r� r� )r� r r �r0 r� r� r1 r1 r2 �TR5W � r� c C r� )a� Replacement of cos**2 with 1 - sin(x)**2.
See _TR56 docstring for advanced use of ``max`` and ``pow``.
Examples
========
>>> from sympy.simplify.fu import TR6
>>> from sympy.abc import x
>>> from sympy import cos
>>> TR6(cos(x)**2)
1 - sin(x)**2
>>> TR6(cos(x)**-2) #unchanged
cos(x)**(-2)
>>> TR6(cos(x)**4)
(1 - sin(x)**2)**2
c S r� r\ r1 r� r1 r1 r2 r� ~ r� zTR6.<locals>.<lambda>r� )r� r r r� r1 r1 r2 �TR6l r� r� c C r4 )a Lowering the degree of cos(x)**2.
Examples
========
>>> from sympy.simplify.fu import TR7
>>> from sympy.abc import x
>>> from sympy import cos
>>> TR7(cos(x)**2)
cos(2*x)/2 + 1/2
>>> TR7(cos(x)**2 + 1)
cos(2*x)/2 + 3/2
c S s< | j r| jjtkr| jdks| S dtd| jjd � d S )Nr@ r] r )rB rC rL r r� r7 r/ r1 r1 r2 r; � s zTR7.<locals>.fr' r< r1 r1 r2 �TR7� s
r� Tc r? )aq Converting products of ``cos`` and/or ``sin`` to a sum or
difference of ``cos`` and or ``sin`` terms.
Examples
========
>>> from sympy.simplify.fu import TR8
>>> from sympy import cos, sin
>>> TR8(cos(2)*cos(3))
cos(5)/2 + cos(1)/2
>>> TR8(cos(2)*sin(3))
sin(5)/2 + sin(1)/2
>>> TR8(sin(2)*sin(3))
-cos(5)/2 + cos(1)/2
c s� | j s| jr| jjttfv r| jjs| jjs| S � r]dd� | � � D �\}}t
|dd�}t
|dd�}||ks9||kr[t || �} | j r[| j d j
r[t| j �dkr[| j d jr[t| �� � } | S tg tg d g i}tt�| ��D ]B}|jttfv r�||j �|j d � ql|jr�|jjr�|jdkr�|jjttfv r�||jj �|jj d g|j � ql|d �|� ql|t }|t }|r�|s�t|�dks�t|�dks�| S |d }tt|�t|��}t|�D ]} |�� }
|�� } |�t|
| �t|
| � d � q�t|�dk�r |�� }
|�� } |�t|
| �t|
| � d � t|�dks�|�r(|�t|�� �� t|�dk�rP|�� }
|�� } |�t|
| � t|
| � d � t|�dk�s/|�r\|�t|�� �� t
tt|� ��S )Nc S � g | ]}t |��qS r1 r �rD rd r1 r1 r2 rY � � z"TR8.<locals>.f.<locals>.<listcomp>F��firstr r@ r] )rn rB rC rL r r r� rJ rK ro �TR8r r7 � is_RationalrN rM r
� as_coeff_Mulr rH r^ �
is_Integer�extend�min�rangerV r ) r0 rX r[ �newn�newdr7 r: �c�srd rv �a2r� r1 r2 r; � sj ���� $ $" �$�zTR8.<locals>.fr' �r0 r� r; r1 r� r2 r� � s
7r� c C r4 )ab Sum of ``cos`` or ``sin`` terms as a product of ``cos`` or ``sin``.
Examples
========
>>> from sympy.simplify.fu import TR9
>>> from sympy import cos, sin
>>> TR9(cos(1) + cos(2))
2*cos(1/2)*cos(3/2)
>>> TR9(cos(1) + 2*sin(1) + 2*sin(2))
cos(1) + 4*sin(3/2)*cos(1/2)
If no change is made by TR9, no re-arrangement of the
expression will be made. For example, though factoring
of common term is attempted, if the factored expression
wasn't changed, the original expression will be returned:
>>> TR9(cos(3) + cos(3)*cos(2))
cos(3) + cos(2)*cos(3)
c s" | j s| S d� fdd� � t| � �S )NTc s� | j s| S tt| j��}t|�dkrgd}tt|��D ]8}|| }|d u r%qt|d t|��D ]#}|| }|d u r9q.|| }� |�} | |krQ| ||< d ||< d} nq.q|retdd� |D �� } | j re� | �} | S t|� }
|
so| S |
\} } }
}}}|r�| |
kr�| | d t|| d � t|| d � S | dk r�||}}d| t || d � t || d � S | |
kr�| | d t || d � t|| d � S | dk r�||}}d| t|| d � t || d � S ) Nr@ Fr] Tc S � g | ]}|r|�qS r1 r1 �rD r� r1 r1 r2 rY # r� z.TR9.<locals>.f.<locals>.do.<locals>.<listcomp>r ����
rM rq r r7 rN r� r �
trig_splitr r )r0 r� r7 �hitrd rE �j�aj�was�new�split�gcd�n1�n2r: rk �iscos��dor1 r2 r� � sT ��,
(,
(zTR9.<locals>.f.<locals>.do�T)rM �process_common_addendsr/ r1 r� r2 r; � s
>zTR9.<locals>.fr' r< r1 r1 r2 �TR9� s
Dr� c r? )a� Separate sums in ``cos`` and ``sin``.
Examples
========
>>> from sympy.simplify.fu import TR10
>>> from sympy.abc import a, b, c
>>> from sympy import cos, sin
>>> TR10(cos(a + b))
-sin(a)*sin(b) + cos(a)*cos(b)
>>> TR10(sin(a + b))
sin(a)*cos(b) + sin(b)*cos(a)
>>> TR10(sin(a + b + c))
(-sin(a)*sin(b) + cos(a)*cos(b))*sin(c) + (sin(a)*cos(b) + sin(b)*cos(a))*cos(c)
c s | j ttfvr | S | j }| jd }|jr�� r tt|j��}nt|j�}|�� }t� |�}|jrc|tkrKt|�t
t|�dd� t|�t
t|�dd� S t|�t
t|�dd� t|�t
t|�dd� S |tkrwt|�t|� t|�t|� S t|�t|� t|�t|� S | S )Nr Fr� ) rL r r r7 rM rq r rV r �
_from_args�TR10)r0 r; �argr7 r: rk r� r1 r2 r; T s,
�� zTR10.<locals>.fr' r� r1 r� r2 r� B s
r� c C s t du rt� dd� }t| |�S )a� Sum of products to function of sum.
Examples
========
>>> from sympy.simplify.fu import TR10i
>>> from sympy import cos, sin, sqrt
>>> from sympy.abc import x
>>> TR10i(cos(1)*cos(3) + sin(1)*sin(3))
cos(2)
>>> TR10i(cos(1)*sin(3) + sin(1)*cos(3) + cos(3))
cos(3) + sin(4)
>>> TR10i(sqrt(2)*cos(x)*x + sqrt(6)*sin(x)*x)
2*sqrt(2)*x*sin(x + pi/6)
Nc s� | j s| S d
� fdd� � t| � dd� �} | j r�tt�}| jD ]/}d}|jrC|jD ]}|jrB|jtj u rB|j
j rB|| � |� d} nq(|sM|tj
� |� q g }|D ]`}t| tfD ]W}||v r�tt|| ��D ]H}|| | d u rsqhtt|| ��D ]4}|| | d u r�q{t|| | || | �} � | �}
|
| kr�|� |
� d || |< d || |< nq{qhqZqR|r�t|dd � |�� D � � } n� | �} | S | j s| S ) NTc sX | j s| S tt| j��}t|�dkrgd}tt|��D ]8}|| }|d u r%qt|d t|��D ]#}|| }|d u r9q.|| }� |�} | |krQ| ||< d ||< d} nq.q|retdd� |D �� } | j re� | �} | S t|ddi�}
|
sr| S |
\} } }
}}}|r�| | } | |
kr�| t|| � S | t|| � S | | } | |
kr�| t || � S | t || � S )Nr@ Fr] Tc S r� r1 r1 r� r1 r1 r2 rY � r� z0TR10i.<locals>.f.<locals>.do.<locals>.<listcomp>�twor� )r0 r� r7 r� rd rE r� r� r� r� r� r� r� r� r: rk �samer� r1 r2 r� � sP
��z TR10i.<locals>.f.<locals>.doc S s t t| j��S rA )�tupler � free_symbolsr� r1 r1 r2 r� � s z"TR10i.<locals>.f.<locals>.<lambda>r r] c S s g | ] }t d d� |D �� �qS )c S r� r1 r1 r� r1 r1 r2 rY � r� z/TR10i.<locals>.f.<locals>.<listcomp>.<listcomp>r )rD re r1 r1 r2 rY � s �z$TR10i.<locals>.f.<locals>.<listcomp>r� )rM r� r rq r7 rn rB r� r �HalfrC r� r^ r8 �_ROOT3� _invROOT3r� rN r �values) r0 �byradr: r� rE r7 rk rd r� r� r� r1 r� r2 r; � sf 8
�
���
���
��(zTR10i.<locals>.f)�_ROOT2�_rootsr( r< r1 r1 r2 �TR10ir s
kr� Nc r? )an Function of double angle to product. The ``base`` argument can be used
to indicate what is the un-doubled argument, e.g. if 3*pi/7 is the base
then cosine and sine functions with argument 6*pi/7 will be replaced.
Examples
========
>>> from sympy.simplify.fu import TR11
>>> from sympy import cos, sin, pi
>>> from sympy.abc import x
>>> TR11(sin(2*x))
2*sin(x)*cos(x)
>>> TR11(cos(2*x))
-sin(x)**2 + cos(x)**2
>>> TR11(sin(4*x))
4*(-sin(x)**2 + cos(x)**2)*sin(x)*cos(x)
>>> TR11(sin(4*x/3))
4*(-sin(x/3)**2 + cos(x/3)**2)*sin(x/3)*cos(x/3)
If the arguments are simply integers, no change is made
unless a base is provided:
>>> TR11(cos(2))
cos(2)
>>> TR11(cos(4), 2)
-sin(2)**2 + cos(2)**2
There is a subtle issue here in that autosimplification will convert
some higher angles to lower angles
>>> cos(6*pi/7) + cos(3*pi/7)
-cos(pi/7) + cos(3*pi/7)
The 6*pi/7 angle is now pi/7 but can be targeted with TR11 by supplying
the 3*pi/7 base:
>>> TR11(_, 3*pi/7)
-sin(3*pi/7)**2 + cos(3*pi/7)**2 + cos(3*pi/7)
c s0 | j ttfvr | S � rS| j }|� d �}tj}|jr |�� \}}|j ttfvr)| S | jd |jd krQt� �}t� �}|tu rI|d |d | S d| | | S | S | jd js�| jd jdd�\}}|j d dkr�|j d | |j
}t t|��}t t|��}| j tkr�d| | } | S |d |d } | S )Nr@ r T)�rational) rL r r r r8 rn r� r7 � is_Numberr� �q�TR11)r0 r; ru �cor� r� �mr� �rC r1 r2 r; s8
�zTR11.<locals>.fr' )r0 rC r; r1 r� r2 r� � s *
#r� c C r4 )a�
Helper for TR11 to find half-arguments for sin in factors of
num/den that appear in cos or sin factors in the den/num.
Examples
========
>>> from sympy.simplify.fu import TR11, _TR11
>>> from sympy import cos, sin
>>> from sympy.abc import x
>>> TR11(sin(x/3)/(cos(x/6)))
sin(x/3)/cos(x/6)
>>> _TR11(sin(x/3)/(cos(x/6)))
2*sin(x/6)
>>> TR11(sin(x/6)/(sin(x/3)))
sin(x/6)/sin(x/3)
>>> _TR11(sin(x/6)/(sin(x/3)))
1/(2*cos(x/6))
c S sL t | t�s| S dd� }t|| �� �\}}dd� }|| ||�} || ||�} | S )Nc S sZ t t�}t�| �D ]!}|�� \}}|jr*|dkr*|jttfv r*||j � |j
d � q |S r5 ) r �setr
rH � as_base_expr� rL r r �addr7 )�flatr7 �firk rP r1 r1 r2 � sincos_args_ s �z%_TR11.<locals>.f.<locals>.sincos_argsc S sX |t D ]%}|d }||t v rt}n
||t v r t }nqt| |�} || �|� q| S �Nr@ )r r r� �remove)r0 �num_args�den_args�nargrR rL r1 r1 r2 � handle_matchk s
z&_TR11.<locals>.f.<locals>.handle_match)r6 r �mapro )r0 r� r� r� r� r1 r1 r2 r; [ s
z_TR11.<locals>.fr' r< r1 r1 r2 �_TR11F s
%r� c r? )z�Separate sums in ``tan``.
Examples
========
>>> from sympy.abc import x, y
>>> from sympy import tan
>>> from sympy.simplify.fu import TR12
>>> TR12(tan(x + y))
(tan(x) + tan(y))/(-tan(x)*tan(y) + 1)
c s� | j tks| S | jd }|jrE� rtt|j��}nt|j�}|�� }t�|�}|jr3t t|�dd�}nt|�}t|�| dt|�| S | S )Nr Fr� r] )
rL r r7 rM rq r rV r r� �TR12)r0 r� r7 r: rk �tbr� r1 r2 r; � s
zTR12.<locals>.fr' r� r1 r� r2 r� � s
r� c rx )aK Combine tan arguments as
(tan(y) + tan(x))/(tan(x)*tan(y) - 1) -> -tan(x + y).
Examples
========
>>> from sympy.simplify.fu import TR12i
>>> from sympy import tan
>>> from sympy.abc import a, b, c
>>> ta, tb, tc = [tan(i) for i in (a, b, c)]
>>> TR12i((ta + tb)/(-ta*tb + 1))
tan(a + b)
>>> TR12i((ta + tb)/(ta*tb - 1))
-tan(a + b)
>>> TR12i((-ta - tb)/(ta*tb - 1))
tan(a + b)
>>> eq = (ta + tb)/(-ta*tb + 1)**2*(-3*ta - 3*tc)/(2*(ta*tc - 1))
>>> TR12i(eq.expand())
-3*tan(a + b)*tan(a + c)/(2*(tan(a) + tan(b) - 1))
r r% c s� | j s | js | js | S | �� \}}|jr|js| S i }dd� }tt�|��}t|�D ]w\}}||�}|rL|\} }
t dd� |
jD �� } t
j || < | ||< q*|j rb� |�}|jra|� |j� t
j ||< q*|jr�|j
jsm|jjr�||j�}|r�|\} }
t dd� |
jD �� } |j
|| < | |j
||< q*� |�}|jr�|� |j� t
j ||< q*|s�| S dd� }tt�t|���} d}
t| �D ]�\}}||�}|�s|| �}|r�t
j| |< nK|j r� |�}|jr�| � |j� t
j | |< q�|j�r|j
js�|jj�r||j�}|�rt
j | |< n� |�}|j�r| � |j� t
j | |< q�q�t
j | |< d}
t d d� |D �� } || }|�t
j �}|d u�rC|�r>||| < n|�| � | | t| � 9 < q�|
�ret| � t|� td
d� |�� D �� } | S ) Nc S sd t | �}|r(|\}}}|tju r*|jr,t|j�dkr.tdd� |jD ��r0||fS d S d S d S d S d S )Nr@ c s s � | ]}t |t�V qd S rA )r6 r )rD r� r1 r1 r2 rF � � � z/TR12i.<locals>.f.<locals>.ok.<locals>.<genexpr>)� as_f_sign_1r � NegativeOnern rN r7 �all)�dir� r� r; r� r1 r1 r2 rS � s
��z TR12i.<locals>.f.<locals>.okc S � g | ]}|j d �qS �r �r7 �rD �_r1 r1 r2 rY � � z$TR12i.<locals>.f.<locals>.<listcomp>c S r� r� r� r� r1 r1 r2 rY � r� c S sJ | j r t| j�dkr | j\}}t|t�r!t|t�r#||fS d S d S d S d S r� )rM rN r7 r6 r )�nir: rk r1 r1 r2 rS � s
�FTc S r� r� r� r� r1 r1 r2 rY r� c S s, g | ]\}}t d d� |jD �� d | �qS )c S r� r1 )r rI r1 r1 r2 rY s �z/TR12i.<locals>.f.<locals>.<listcomp>.<listcomp>r] )r r7 )rD rd rP r1 r1 r2 rY s ����)rM rn rB ro r7 rq r
rH r_ r r r8 r� r� rJ rC rK r
r� �extract_additivelyrV r rs )r0 rX r[ �dokrS �d_argsrd r� r� r� ru r� �n_argsr� r� �ed�newedr% r1 r2 r; � s�
�
�
��
�zTR12i.<locals>.f)�sympyr& r( r< r1 r% r2 �TR12i� s
cr� c C r4 )a Change products of ``tan`` or ``cot``.
Examples
========
>>> from sympy.simplify.fu import TR13
>>> from sympy import tan, cot
>>> TR13(tan(3)*tan(2))
-tan(2)/tan(5) - tan(3)/tan(5) + 1
>>> TR13(cot(3)*cot(2))
cot(2)*cot(5) + 1 + cot(3)*cot(5)
c S s~ | j s| S tg tg d g i}tt�| ��D ] }|jttfv r)||j �|jd � q|d �|� q|t }|t }t |�dk rGt |�dk rG| S |d }t |�dkrx|�
� }|�
� }|�dt|�t|| � t|�t|| � � t |�dksQ|r�|�t|�
� �� t |�dkr�|�
� }|�
� }|�dt|�t|| � t|�t|| � � t |�dks�|r�|�t|�
� �� t|� S )Nr r@ r] ) rn r r r r
rH rL r^ r7 rN rV )r0 r7 r: ru r� �t1�t2r1 r1 r2 r; 1 s6 2 � 2 �zTR13.<locals>.fr' r< r1 r1 r2 �TR13# s
r� c s d� fdd� � t | � �S )a� Returns cos(x)*cos(2*x)*...*cos(2**(k-1)*x) -> sin(2**k*x)/(2**k*sin(x))
Examples
========
>>> from sympy.simplify.fu import TRmorrie, TR8, TR3
>>> from sympy.abc import x
>>> from sympy import Mul, cos, pi
>>> TRmorrie(cos(x)*cos(2*x))
sin(4*x)/(4*sin(x))
>>> TRmorrie(7*Mul(*[cos(x) for x in range(10)]))
7*sin(12)*sin(16)*cos(5)*cos(7)*cos(9)/(64*sin(1)*sin(3))
Sometimes autosimplification will cause a power to be
not recognized. e.g. in the following, cos(4*pi/7) automatically
simplifies to -cos(3*pi/7) so only 2 of the 3 terms are
recognized:
>>> TRmorrie(cos(pi/7)*cos(2*pi/7)*cos(4*pi/7))
-sin(3*pi/7)*cos(3*pi/7)/(4*sin(pi/7))
A touch by TR8 resolves the expression to a Rational
>>> TR8(_)
-1/8
In this case, if eq is unsimplified, the answer is obtained
directly:
>>> eq = cos(pi/9)*cos(2*pi/9)*cos(3*pi/9)*cos(4*pi/9)
>>> TRmorrie(eq)
1/16
But if angles are made canonical with TR3 then the answer
is not simplified without further work:
>>> TR3(eq)
sin(pi/18)*cos(pi/9)*cos(2*pi/9)/2
>>> TRmorrie(_)
sin(pi/18)*sin(4*pi/9)/(8*sin(pi/9))
>>> TR8(_)
cos(7*pi/18)/(16*sin(pi/9))
>>> TR3(_)
1/16
The original expression would have resolve to 1/16 directly with TR8,
however:
>>> TR8(eq)
1/16
References
==========
.. [1] https://en.wikipedia.org/wiki/Morrie%27s_law
Tc s | j s| S |r| �� \}}�|d��|d� S tt�� i }g }| jD ]*}|�� \}}|jrGt|t�rG|jd � � \} }
� |
�
| � |||< q"|�
|� q"g } � D ]�}
� |
}|� � |r�d} |d }
}|
|v ru| d7 } |
d9 }
|
|v si| dkr�t d| | |
�d| t ||
� }d }g }t
| �D ] }|
d }
t|
|
dd�}|�
|
� t|| |p�|| �}q�t
| �D ] }|�� }
t|
|
dd�}|| |8 < || s�|�|
� q�| �
|| � nt|�d�|
�}|�
||| � |s]qQ| �rt| | � fdd�� D � � } | S )Nr r] r@ Frg c s* g | ]}� | D ]
}t || d d��qqS )Frg )r )rD r: rO r� r1 r2 rY � s
��z'TRmorrie.<locals>.f.<locals>.<listcomp>)rn ro r rq r7 r� r� r6 r r� r^ �sortr r� r� rV r� r
)r0 r� rX r[ �coss�otherr� rk rP r� r: r� rO �cc�ci�newarg�take�ccsrd �key�r; r� r2 r; � sh
�(
���
�zTRmorrie.<locals>.fNr� r' r/ r1 r r2 �TRmorrieR s ;
9r c r? )a Convert factored powers of sin and cos identities into simpler
expressions.
Examples
========
>>> from sympy.simplify.fu import TR14
>>> from sympy.abc import x, y
>>> from sympy import cos, sin
>>> TR14((cos(x) - 1)*(cos(x) + 1))
-sin(x)**2
>>> TR14((sin(x) - 1)*(sin(x) + 1))
-cos(x)**2
>>> p1 = (cos(x) + 1)*(cos(x) - 1)
>>> p2 = (cos(y) - 1)*2*(cos(y) + 1)
>>> p3 = (3*(cos(y) - 1))*(3*(cos(y) + 1))
>>> TR14(p1*p2*p3*(x - 1))
-18*(x - 1)*sin(x)**2*sin(y)**4
c s@ | j s| S �r,| �� \}}|tjur,t|dd�}t|dd�}||ks&||kr*|| } | S g }g }| jD ]P}|jrM|�� \}} | jsJ|j sJ|�
|� q3|}ntj} t |�}
|
r_|
d j t
tfvrr| tju rj|�
|� n|�
|| � q3|
\} } }
|�
| | j| | |
|f� q3tt|��}t|�}ttd�� }\} }} } }
}|�r�|�d�� |�r�|d �� | j�r;�| j�r;� | �| k�r:� |
�|
k�r:|�d��t� | �| �}�| |kr��fdd�|D �}|| |8 < |�d|� n � | |k�r� fdd�|D �}|| |8 < |�d|� t� | t
��r t}nt
}|�
� | �| |� | jd �d | � q�nK� | �| k�r�� | �| k�r�� |
�|
k�r�|�d��� | }t� | t
��rjt}nt
}|�
� | �| |� | jd �d | � q�|�
� | � | � |s�t|�|k�r�t|� } | S )
NFr� r] � r c � g | ]}� | �qS r1 r1 r� )�Br1 r2 rY r� z#TR14.<locals>.f.<locals>.<listcomp>c r r1 r1 r� )�Ar1 r2 rY r� r@ )rn ro r r8 �TR14r7 rB r� rJ rK r^ r� rL r r r� rq r rN r� rV r� �insertr6 r
)r0 rX r[ r� r� r� �processr: rk rP r� r� r; �si�notherrr ru r �remr� )r r r2 r; � s�
2�
2�.zTR14.<locals>.fr' r� r1 r� r2 r � s
`r c � � �fdd�}t | |�S )a Convert sin(x)**-2 to 1 + cot(x)**2.
See _TR56 docstring for advanced use of ``max`` and ``pow``.
Examples
========
>>> from sympy.simplify.fu import TR15
>>> from sympy.abc import x
>>> from sympy import sin
>>> TR15(1 - 1/sin(x)**2)
-cot(x)**2
c �t t | t�r t | jt�s
| S | j}|d dkr"t| j|d �| j S d| }t|ttdd� � �d�}||kr8|} | S )Nr@ r] c S � d| S r\ r1 r� r1 r1 r2 r� [ r� z!TR15.<locals>.f.<locals>.<lambda>r� )r6 r rC r r� �TR15r� r �r0 rP �iar: r� r1 r2 r; R � zTR15.<locals>.fr' �r0 r� r� r; r1 r� r2 r B �
r c r )a Convert cos(x)**-2 to 1 + tan(x)**2.
See _TR56 docstring for advanced use of ``max`` and ``pow``.
Examples
========
>>> from sympy.simplify.fu import TR16
>>> from sympy.abc import x
>>> from sympy import cos
>>> TR16(1 - 1/cos(x)**2)
-tan(x)**2
c r )Nr@ r] c S r r\ r1 r� r1 r1 r2 r� | r� z!TR16.<locals>.f.<locals>.<lambda>r� )r6 r rC r r� r r� r r r� r1 r2 r; s r zTR16.<locals>.fr' r r1 r� r2 �TR16c r r c C r4 )aD Convert f(x)**-i to g(x)**i where either ``i`` is an integer
or the base is positive and f, g are: tan, cot; sin, csc; or cos, sec.
Examples
========
>>> from sympy.simplify.fu import TR111
>>> from sympy.abc import x
>>> from sympy import tan
>>> TR111(1 - 1/tan(x)**2)
1 - cot(x)**2
c S s� t | t�r| jjs| jjr| jjs| S t | jt�r%t| jj d �| j S t | jt
�r7t | jj d �| j S t | jt �rIt
| jj d �| j S | S r5 )r6 r rC rK r� rJ � is_negativer r r7 r r! r r r/ r1 r1 r2 r; � s ���� zTR111.<locals>.fr' r< r1 r1 r2 �TR111� s
r c r )ah Convert tan(x)**2 to sec(x)**2 - 1 and cot(x)**2 to csc(x)**2 - 1.
See _TR56 docstring for advanced use of ``max`` and ``pow``.
Examples
========
>>> from sympy.simplify.fu import TR22
>>> from sympy.abc import x
>>> from sympy import tan, cot
>>> TR22(1 + tan(x)**2)
sec(x)**2
>>> TR22(1 + cot(x)**2)
csc(x)**2
c sR t | t�r
| jjttfv s| S t| ttdd� � �d�} t| ttdd� � �d�} | S )Nc S � | d S r\ r1 r� r1 r1 r2 r� � r� z!TR22.<locals>.f.<locals>.<lambda>r� c S r r\ r1 r� r1 r1 r2 r� � r� ) r6 r rC rL r r r� r r! r/ r� r1 r2 r; � s
zTR22.<locals>.fr' r r1 r� r2 �TR22� s
r c C r4 )a Convert sin(x)**n and cos(x)**n with positive n to sums.
Examples
========
>>> from sympy.simplify.fu import TRpower
>>> from sympy.abc import x
>>> from sympy import cos, sin
>>> TRpower(sin(x)**6)
-15*cos(2*x)/32 + 3*cos(4*x)/16 - cos(6*x)/32 + 5/16
>>> TRpower(sin(x)**3*cos(2*x)**4)
(3*sin(x)/4 - sin(3*x)/4)*(cos(4*x)/2 + cos(8*x)/8 + 3/8)
References
==========
.. [1] https://en.wikipedia.org/wiki/List_of_trigonometric_identities#Power-reduction_formulae
c s� t | t�r
t | jttf�s| S | �� \}� |jd �� jr�� jr�� j rAt |t�rAdd� t
� �fdd�t � d d �D �� } nl� j rjt |t�rjdd� d� d d t
� �fdd�t � d d �D �� } nC� j r�t |t�r�dd� t
� �fdd�t � d �D �� } n$� j r�t |t�r�dd� d� d t
� �fd d�t � d �D �� } � j r�| d� t
� � d � 7 } | S )
Nr r@ r] c �* g | ]}t � |�t� d | � � �qS �r@ �r r rW �rX r� r1 r2 rY � � "�z&TRpower.<locals>.f.<locals>.<listcomp>���c �2 g | ]}t � |�d | t� d| � � �qS �r" r@ )r r rW r r1 r2 rY � � ���c r r r rW r r1 r2 rY � r! c r# r$ r rW r r1 r2 rY � r% )r6 r rC r r r� r7 r� rK �is_oddr r� �is_evenr )r0 rk r1 r r2 r; � s0
�( �
�$
� zTRpower.<locals>.fr' r< r1 r1 r2 �TRpower� s
r( c C s t | �t��S )z�Return count of trigonometric functions in expression.
Examples
========
>>> from sympy.simplify.fu import L
>>> from sympy.abc import x
>>> from sympy import cos, sin
>>> L(cos(x)+sin(x))
2
)r �countr# r/ r1 r1 r2 �L� s r* c C s t | �| �� fS rA )r* � count_opsr� r1 r1 r2 r� # s r� c s� t t� �}t t� �}| }t| �} t| t�s"| j� fdd�| jD �� S t| �} | � t
t �rD|| �}� |�� | �k r:|} | � t
t �rDt | �} | � t
t�r^|| �}tt|��}t|| ||g� d�} tt| �| � d�S )aH Attempt to simplify expression by using transformation rules given
in the algorithm by Fu et al.
:func:`fu` will try to minimize the objective function ``measure``.
By default this first minimizes the number of trig terms and then minimizes
the number of total operations.
Examples
========
>>> from sympy.simplify.fu import fu
>>> from sympy import cos, sin, tan, pi, S, sqrt
>>> from sympy.abc import x, y, a, b
>>> fu(sin(50)**2 + cos(50)**2 + sin(pi/6))
3/2
>>> fu(sqrt(6)*cos(x) + sqrt(2)*sin(x))
2*sqrt(2)*sin(x + pi/3)
CTR1 example
>>> eq = sin(x)**4 - cos(y)**2 + sin(y)**2 + 2*cos(x)**2
>>> fu(eq)
cos(x)**4 - 2*cos(y)**2 + 2
CTR2 example
>>> fu(S.Half - cos(2*x)/2)
sin(x)**2
CTR3 example
>>> fu(sin(a)*(cos(b) - sin(b)) + cos(a)*(sin(b) + cos(b)))
sqrt(2)*sin(a + b + pi/4)
CTR4 example
>>> fu(sqrt(3)*cos(x)/2 + sin(x)/2)
sin(x + pi/3)
Example 1
>>> fu(1-sin(2*x)**2/4-sin(y)**2-cos(x)**4)
-cos(x)**2 + cos(y)**2
Example 2
>>> fu(cos(4*pi/9))
sin(pi/18)
>>> fu(cos(pi/9)*cos(2*pi/9)*cos(3*pi/9)*cos(4*pi/9))
1/16
Example 3
>>> fu(tan(7*pi/18)+tan(5*pi/18)-sqrt(3)*tan(5*pi/18)*tan(7*pi/18))
-sqrt(3)
Objective function example
>>> fu(sin(x)/cos(x)) # default objective function
tan(x)
>>> fu(sin(x)/cos(x), measure=lambda x: -x.count_ops()) # maximize op count
sin(x)/cos(x)
References
==========
.. [1] https://citeseerx.ist.psu.edu/viewdoc/download?doi=10.1.1.657.2478&rep=rep1&type=pdf
c s g | ]}t |� d ��qS )��measure)�furI r, r1 r2 rY o rm zfu.<locals>.<listcomp>)r )r) �RL1�RL2r r6 r rL r7 r= �hasr r r> r r r� r r� rw )r0 r- �fRL1�fRL2r� �rv1�rv2r1 r, r2 r. # s$
F
r. c C s t t�}|r,| jD ]!}|�� \}}|dk r| }| }|||r#||�ndf �|� q n|rA| jD ]}|tj||�f �|� q1ntd��g }d}|D ]4} || }
| \}} t|
�dkrvt |
ddi�} || �}
|
| krn|
} d}|�|| � qK|�||
d � qK|r�t |� } | S )a Apply ``do`` to addends of ``rv`` that (if ``key1=True``) share at least
a common absolute value of their coefficient and the value of ``key2`` when
applied to the argument. If ``key1`` is False ``key2`` must be supplied and
will be the only key applied.
r r] zmust have at least one keyFrh T)
r rq r7 r� r^ r r8 �
ValueErrorrN r )r0 r� �key2�key1�abscr: r� r7 r� rO re r� rP r� r1 r1 r2 r� ~ s<
�
� r� z~
TR0 TR1 TR2 TR3 TR4 TR5 TR6 TR7 TR8 TR9 TR10 TR10i TR11
TR12 TR13 L TR2i TRmorrie TR12i
TR14 TR15 TR16 TR111 TR22c C s t d�t d�aadt ad S )Nr@ � r] )r" r� r� r� r1 r1 r1 r2 r� � s r� c s t du rt� dd� | |fD �\} }| �|�\}}| �|��� }d }}tj|jv r4|�tj�}| }ntj|jv rC|�tj�}| }dd� ||fD �\} }dd� }|| |�} | du r]dS | \}
} } |||�} | du rmdS | \}
}}| sv|s}| r�t | t
�r�|
|||
| | f\}
} } }
}}||}}|s�| p�| }|p�|}t ||j �s�dS ||||j d |j d t |t
�fS |
s�|
s�| r�|r�| r�|r�t | | j �t ||j �ur�dS d d
� | | fD �� t� fd d �||fD ��s�dS |||| j d | j d t | | j �fS | r�| �s|�r|�s|�r| du �r| du �s|du �r|du �rdS | �p | }|�p$|}|j |j k�r.dS |
�s4tj}
|
�s:tj}
|
|
u �rP|t 9 }||||j d td
dfS |
|
tk�rj|d|
9 }||||j d td dfS |
|
tk�r�|d|
9 }||||j d td dfS dS )a) Return the gcd, s1, s2, a1, a2, bool where
If two is False (default) then::
a + b = gcd*(s1*f(a1) + s2*f(a2)) where f = cos if bool else sin
else:
if bool, a + b was +/- cos(a1)*cos(a2) +/- sin(a1)*sin(a2) and equals
n1*gcd*cos(a - b) if n1 == n2 else
n1*gcd*cos(a + b)
else a + b was +/- cos(a1)*sin(a2) +/- sin(a1)*cos(a2) and equals
n1*gcd*sin(a + b) if n1 = n2 else
n1*gcd*sin(b - a)
Examples
========
>>> from sympy.simplify.fu import trig_split
>>> from sympy.abc import x, y, z
>>> from sympy import cos, sin, sqrt
>>> trig_split(cos(x), cos(y))
(1, 1, 1, x, y, True)
>>> trig_split(2*cos(x), -2*cos(y))
(2, 1, -1, x, y, True)
>>> trig_split(cos(x)*sin(y), cos(y)*sin(y))
(sin(y), 1, 1, x, y, True)
>>> trig_split(cos(x), -sqrt(3)*sin(x), two=True)
(2, 1, -1, x, pi/6, False)
>>> trig_split(cos(x), sin(x), two=True)
(sqrt(2), 1, 1, x, pi/4, False)
>>> trig_split(cos(x), -sin(x), two=True)
(sqrt(2), 1, -1, x, pi/4, False)
>>> trig_split(sqrt(2)*cos(x), -sqrt(6)*sin(x), two=True)
(2*sqrt(2), 1, -1, x, pi/6, False)
>>> trig_split(-sqrt(6)*cos(x), -sqrt(2)*sin(x), two=True)
(-2*sqrt(2), 1, 1, x, pi/3, False)
>>> trig_split(cos(x)/sqrt(6), sin(x)/sqrt(2), two=True)
(sqrt(6)/3, 1, 1, x, pi/6, False)
>>> trig_split(-sqrt(6)*cos(x)*sin(y), -sqrt(2)*sin(x)*sin(y), two=True)
(-2*sqrt(2)*sin(y), 1, 1, x, pi/3, False)
>>> trig_split(cos(x), sin(x))
>>> trig_split(cos(x), sin(z))
>>> trig_split(2*cos(x), -sin(x))
>>> trig_split(cos(x), -sqrt(3)*sin(x))
>>> trig_split(cos(x)*cos(y), sin(x)*sin(z))
>>> trig_split(cos(x)*cos(y), sin(x)*sin(y))
>>> trig_split(-sqrt(6)*cos(x), sqrt(2)*sin(x)*sin(y), two=True)
Nc S r� r1 �r r� r1 r1 r2 rY � r� z trig_split.<locals>.<listcomp>r] c S � g | ]}|� � �qS r1 ��as_exprr� r1 r1 r2 rY � r� c S sd d }}t j}| jr�| �� \}} t| j�dks|sdS | jr$t| j�}n| g}|�d�} t| t �r4| }nt| t
�r<| }n| j rJ| j t j
u rJ|| 9 }ndS |r||d }t|t �r_|r\|}n |}n t|t
�rl|ri|}n|}n|j rz|j t j
u rz||9 }ndS |t jur�|||fS d||fS t| t �r�| }nt| t
�r�| }|du r�|du r�dS |t jur�|nd}|||fS )a� Return ``a`` as a tuple (r, c, s) such that
``a = (r or 1)*(c or 1)*(s or 1)``.
Three arguments are returned (radical, c-factor, s-factor) as
long as the conditions set by ``two`` are met; otherwise None is
returned. If ``two`` is True there will be one or two non-None
values in the tuple: c and s or c and r or s and r or s or c with c
being a cosine function (if possible) else a sine, and s being a sine
function (if possible) else oosine. If ``two`` is False then there
will only be a c or s term in the tuple.
``two`` also require that either two cos and/or sin be present (with
the condition that if the functions are the same the arguments are
different or vice versa) or that a single cosine or a single sine
be present with an optional radical.
If the above conditions dictated by ``two`` are not met then None
is returned.
Nr@ r )r r8 rn r� rN r7 rq rV r6 r r rB r� r� )r: r� r� r� r� r7 rk r1 r1 r2 � pow_cos_sin� sN
z trig_split.<locals>.pow_cos_sinr c S s h | ]}|j �qS r1 r� )rD r� r1 r1 r2 � <setcomp>R s z trig_split.<locals>.<setcomp>c 3 s � | ]}|j � v V qd S rA r� r� r� r1 r2 rF S r� z trig_split.<locals>.<genexpr>r{ Fr@ r: r )r� r� r- r� r> r r� �factors�quor6 r rL r7 r r� r8 r r� r� )r: rk r� �ua�ubr� r� r� r? r� �coa�ca�sa�cob�cb�sbr� r� r1 r� r2 r� � s| 3
B
"$�(
�r� c C sR | j r
t| j�dkr dS | j\}}|tjtjfv r;tj}|jr6|jd jr6|jd dk r6| | }}| }|||fS dd� | jD �\}}|�|�\}}|� |��
� }tj|j v rd|� tj�}d}d}ntj|j v ru|� tj�}d}d}nd }}dd� ||fD �\}}|tju r�||}}||}}|dkr�| }| }|tju r�|||fS dS ) a� If ``e`` is a sum that can be written as ``g*(a + s)`` where
``s`` is ``+/-1``, return ``g``, ``a``, and ``s`` where ``a`` does
not have a leading negative coefficient.
Examples
========
>>> from sympy.simplify.fu import as_f_sign_1
>>> from sympy.abc import x
>>> as_f_sign_1(x + 1)
(1, x, 1)
>>> as_f_sign_1(x - 1)
(1, x, -1)
>>> as_f_sign_1(-x + 1)
(-1, x, -1)
>>> as_f_sign_1(-x - 1)
(-1, x, 1)
>>> as_f_sign_1(2*x + 2)
(2, x, 1)
r@ Nr c S r� r1 r; r� r1 r1 r2 rY � r� z as_f_sign_1.<locals>.<listcomp>r" r] c S r< r1 r= r� r1 r1 r2 rY � r� )
rM rN r7 r r� r8 rn r� r- r� r> rA rB ) rP r: rk r� rC rD r� r� r� r1 r1 r2 r� l s>
�r� c r? )a2 Replace all hyperbolic functions with trig functions using
the Osborne rule.
Notes
=====
``d`` is a dummy variable to prevent automatic evaluation
of trigonometric/hyperbolic functions.
References
==========
.. [1] https://en.wikipedia.org/wiki/Hyperbolic_function
c s� t | t�s| S | jd }|js|� n t�� fdd�|jD ��}t | t�r+tt|� S t | t �r4t
|�S t | t �r?tt |� S t | t
�rJt|�t S t | t�rSt|�S t | t�r^t|�t S td| j ��)Nr c s g | ]}|� �qS r1 r1 r� �r[ r1 r2 rY � r� z'_osborne.<locals>.f.<locals>.<listcomp>� unhandled %s)r6 r r7 rM r r� r r r r r r r r r r r r r! �NotImplementedErrorrL r9 rK r1 r2 r; � s"
(
z_osborne.<locals>.fr' �rP r[ r; r1 rK r2 �_osborne� s
rO c r? )a1 Replace all trig functions with hyperbolic functions using
the Osborne rule.
Notes
=====
``d`` is a dummy variable to prevent automatic evaluation
of trigonometric/hyperbolic functions.
References
==========
.. [1] https://en.wikipedia.org/wiki/Hyperbolic_function
c s� t | t�s| S | jd j� dd�\}}|�� tji�|t }t | t�r*t |�t S t | t
�r3t |�S t | t �r>t
|�t S t | t�rIt|�t S t | t�rRt|�S t | t�r]t|�t S td| j ��)Nr T)�as_AddrL )r6 r# r7 �as_independent�xreplacer r8 r r r r r r r r r r r r! r rM rL )r0 �constr� r: rK r1 r2 r; � s"
z_osbornei.<locals>.fr' rN r1 rK r2 � _osbornei� s
rT c sn ddl m� ddlm� | �t�}dd� |D ��| �t���}dd� �D ��t� �t |��� ���fdd�fS ) a� Return an expression containing hyperbolic functions in terms
of trigonometric functions. Any trigonometric functions initially
present are replaced with Dummy symbols and the function to undo
the masking and the conversion back to hyperbolics is also returned. It
should always be true that::
t, f = hyper_as_trig(expr)
expr == f(t)
Examples
========
>>> from sympy.simplify.fu import hyper_as_trig, fu
>>> from sympy.abc import x
>>> from sympy import cosh, sinh
>>> eq = sinh(x)**2 + cosh(x)**2
>>> t, f = hyper_as_trig(eq)
>>> f(fu(t))
cosh(2*x)
References
==========
.. [1] https://en.wikipedia.org/wiki/Hyperbolic_function
r ry )�collectc S s g | ]}|t � f�qS r1 r )rD ru r1 r1 r2 rY r� z!hyper_as_trig.<locals>.<listcomp>c S s g | ]\}}||f�qS r1 r1 )rD rO re r1 r1 r2 rY rm c s � �t | ���t����tj�S rA )rT rR �dictr �
ImaginaryUnitr� �rU r[ �repsrz r1 r2 r� s
��z hyper_as_trig.<locals>.<lambda>)
r~ rz �sympy.simplify.radsimprU �atomsr# rR rV r rO )r0 �trigs�maskedr1 rX r2 �
hyper_as_trig� s
r^ c C s | � tt�s| S ttt| ���S )a� Convert products and powers of sin and cos to sums.
Explanation
===========
Applied power reduction TRpower first, then expands products, and
converts products to sums with TR8.
Examples
========
>>> from sympy.simplify.fu import sincos_to_sum
>>> from sympy.abc import x
>>> from sympy import cos, sin
>>> sincos_to_sum(16*sin(x)**3*cos(2*x)**2)
7*sin(x) - 5*sin(3*x) + 3*sin(5*x) - sin(7*x)
)r1 r r r� r r( )�exprr1 r1 r2 �
sincos_to_sum s r` )F)r{ Fr� rA )NT)o� collectionsr �sympy.core.addr �sympy.core.basicr �sympy.core.compatibilityr �sympy.core.exprr �sympy.core.exprtoolsr r r
�sympy.core.functionr �sympy.core.mulr
�sympy.core.numbersr r �sympy.core.powerr �sympy.core.symbolr �sympy.core.sympifyr �(sympy.functions.combinatorial.factorialsr �%sympy.functions.elementary.hyperbolicr r r r r r r �(sympy.functions.elementary.trigonometricr r r r r r! r" r# �sympy.ntheory.factor_r$ �sympy.polys.polytoolsr& r~ r( �sympy.strategies.treer) �sympy.strategies.corer* r+ r� r, r3 r= r> rw r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r( r* rq r� �CTR1�CTR2�CTR3�CTR4r/ r0 r. r� r� �fufuncsrV �zip�locals�get�FUr� r� r� r� rO rT r^ r` r1 r1 r1 r2 �<module> s� $(
t*
@
K
^0
P
=#}/
w
y
!!
. �� �
�
[*�
79(' +