o
    jgÇ   ã                   @   s¾   d Z ddl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mZ ddlmZ edd	„ ƒZeddd„ƒZeddd„ƒZeddd„ƒZdd„ Zedd„ ƒZdd„ Zdd„ Zd
S )zÓ
Known facts in assumptions module.

This module defines the facts between unary predicates in ``get_known_facts()``,
and supports functions to generate the contents in
``sympy.assumptions.ask_generated`` file.
é    )ÚQ)ÚAppliedPredicate)Úcacheit)ÚSymbol)Úto_cnfÚAndÚNotÚImpliesÚ
EquivalentÚ	Exclusive)Úsatisfiablec                   C   sÜ   t jt jt jB t jB t jt jt jB t jt jt jB t j	t jt jB t j
t jt jB t jt jt jB t jB t jB t jB t jt jt jB t jt jt jB t jt jt jB t jB t jB t jt jt jB t jB t jt jt jB t jB t jt jt jB iS ©N)r   ÚrealÚnegativeÚzeroÚpositiveÚintegerÚevenÚoddÚnonpositiveÚnonzeroÚnonnegativeÚextended_realÚnegative_infiniteÚpositive_infiniteÚextended_positiveÚextended_negativeÚextended_nonzeroÚextended_nonpositiveÚextended_nonnegativeÚcomplexÚ	algebraicÚtranscendental© r#   r#   úO/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/assumptions/facts.pyÚget_composite_predicates   s    ôr%   Nc                 C   s&   | du rt dƒ} tt| ƒt| ƒƒ}|S )z÷
    Facts between unary predicates.

    Parameters
    ==========

    x : Symbol, optional
        Placeholder symbol for unary facts. Default is ``Symbol('x')``.

    Returns
    =======

    fact : Known facts in conjugated normal form.

    NÚx)r   r   Úget_number_factsÚget_matrix_facts©r&   Úfactr#   r#   r$   Úget_known_facts&   s   þr+   c                 C   s  | du rt dƒ} ttt | ¡t | ¡t | ¡t | ¡t | ¡ƒtt 	| ¡t 
| ¡ƒtt 	| ¡t 
| ¡B t | ¡ƒtt | ¡t | ¡ƒtt 	| ¡t | ¡t | ¡B ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡B t | ¡t | ¡@ ƒtt | ¡t | ¡@ t | ¡ @ t | ¡ƒtt 	| ¡t | ¡ƒtt 
| ¡t | ¡ƒtt | ¡t | ¡t | ¡B ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡B t | ¡ƒtt | ¡t | ¡B t | ¡ƒƒ}|S )zþ
    Facts between unary number predicates.

    Parameters
    ==========

    x : Symbol, optional
        Placeholder symbol for unary facts. Default is ``Symbol('x')``.

    Returns
    =======

    fact : Known facts in conjugated normal form.

    Nr&   )r   r   r   r   r   r   r   r   r   r   Ú	imaginaryr	   r    r"   r!   r
   ÚrationalÚ
irrationalr   r   r   Ú	compositeÚprimeÚ	hermitianÚantihermitianÚinfiniteÚfiniteÚcommutativer)   r#   r#   r$   r'   A   s6   ÿ(*Þ$r'   c                 C   s  | du rt dƒ} ttt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡@ t | ¡ƒtt | ¡t | ¡ƒtt | ¡t 	| ¡ƒtt | ¡t 
| ¡ƒtt | ¡t | ¡ƒtt | ¡t 	| ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡t | ¡B ƒtt | ¡t | ¡@ t | ¡ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒtt 	| ¡t | ¡ƒtt 	| ¡t 
| ¡ƒtt | ¡t 
| ¡ƒtt | ¡t 
| ¡@ t 	| ¡ƒtt 	| ¡t | ¡ ƒtt | ¡t | ¡ƒtt | ¡t | ¡ƒƒ}|S )zþ
    Facts between unary matrix predicates.

    Parameters
    ==========

    x : Symbol, optional
        Placeholder symbol for unary facts. Default is ``Symbol('x')``.

    Returns
    =======

    fact : Known facts in conjugated normal form.

    Nr&   )r   r   r	   r   Ú
orthogonalÚpositive_definiteÚunitaryÚreal_elementsÚnormalÚ
invertibleÚsquareÚdiagonalÚupper_triangularÚlower_triangularÚ
triangularÚ	symmetricÚunit_triangularÚfullrankr
   ÚsingularÚinteger_elementsÚcomplex_elementsr)   r#   r#   r$   r(   |   s8   èr(   c                 C   sˆ   t |ƒ}t| |ƒ}i }| ¡ D ]2\}}tƒ }tƒ }|D ]}	t|	tƒr)| |	j¡ qt|	tƒr9|	j	d }
| |
j¡ q||f||j< q|S )a  
    Computes and returns a dictionary which contains the relations between
    unary predicates.

    Each key is a predicate, and item is two groups of predicates.
    First group contains the predicates which are implied by the key, and
    second group contains the predicates which are rejected by the key.

    All predicates in *keys* and *fact* must be unary and have same placeholder
    symbol.

    Parameters
    ==========

    keys : list of AppliedPredicate instances.

    fact : Fact between predicates in conjugated normal form.

    Examples
    ========

    >>> from sympy import Q, And, Implies
    >>> from sympy.assumptions.facts import generate_known_facts_dict
    >>> from sympy.abc import x
    >>> keys = [Q.even(x), Q.odd(x), Q.zero(x)]
    >>> fact = And(Implies(Q.even(x), ~Q.odd(x)),
    ...     Implies(Q.zero(x), Q.even(x)))
    >>> generate_known_facts_dict(keys, fact)
    {Q.even: ({Q.even}, {Q.odd}),
     Q.odd: ({Q.odd}, {Q.even, Q.zero}),
     Q.zero: ({Q.even, Q.zero}, {Q.odd})}
    r   )
r   Úsingle_fact_lookupÚitemsÚsetÚ
isinstancer   ÚaddÚfunctionr   Úargs)Úkeysr*   Úfact_cnfÚmappingÚretÚkeyÚvalueÚimpliedÚrejectedÚexprÚpredr#   r#   r$   Úgenerate_known_facts_dict®   s   !



€rX   c                  C   s\   t jt jt jt jt jt jh} g }t jjD ]}| 	d¡rqt
t |ƒ}|| v r&q| |¡ q|S )z”
    Return every unary predicates registered to ``Q``.

    This function is used to generate the keys for
    ``generate_known_facts_dict``.

    Ú__)r   ÚeqÚneÚgtÚltÚgeÚleÚ	__class__Ú__dict__Ú
startswithÚgetattrÚappend)ÚexcludeÚresultÚattrrW   r#   r#   r$   Úget_known_facts_keysà   s   


rh   c                 C   sf   i }| D ],}|h||< | D ]"}||kr/t |||ƒr ||  |¡ t | ||ƒr/||  | ¡ qq|S r   )Úask_full_inferencerK   )Úknown_facts_keysÚknown_facts_cnfrP   rR   Ú	other_keyr#   r#   r$   rG   ÷   s   
€ûrG   c                 C   s0   t t||| ƒƒs
dS t t||t| ƒƒƒsdS dS )z9
    Method for inferring properties about objects.

    FTN)r   r   r   )ÚpropositionÚassumptionsrk   r#   r#   r$   ri     s
   ri   r   )Ú__doc__Úsympy.assumptions.askr   Úsympy.assumptions.assumer   Úsympy.core.cacher   Úsympy.core.symbolr   Úsympy.logic.boolalgr   r   r   r	   r
   r   Úsympy.logic.inferencer   r%   r+   r'   r(   rX   rh   rG   ri   r#   r#   r#   r$   Ú<module>   s(     
:12
