
    wg?                         d dl mZmZmZ d dlmZ ej                  Zej                  Zd Zd Z	d Z
	 	 ddZ	 ddZdd	Zd
 Zg i fdZd Zy)    )Permutation
_af_invert_af_rmul)isprimec                     t        |       }dg|z  }t        |      D ]
  }||| |   <    |}t        |      D ]  }|| vs|||<   |dz  } |S )a  
    Order `\{0, 1, \dots, n-1\}` so that base points come first and in order.

    Parameters
    ==========

    base : the base
    degree : the degree of the associated permutation group

    Returns
    =======

    A list ``base_ordering`` such that ``base_ordering[point]`` is the
    number of ``point`` in the ordering.

    Examples
    ========

    >>> from sympy.combinatorics import SymmetricGroup
    >>> from sympy.combinatorics.util import _base_ordering
    >>> S = SymmetricGroup(4)
    >>> S.schreier_sims()
    >>> _base_ordering(S.base, S.degree)
    [0, 1, 2, 3]

    Notes
    =====

    This is used in backtrack searches, when we define a relation `\ll` on
    the underlying set for a permutation group of degree `n`,
    `\{0, 1, \dots, n-1\}`, so that if `(b_1, b_2, \dots, b_k)` is a base we
    have `b_i \ll b_j` whenever `i<j` and `b_i \ll a` for all
    `i\in\{1,2, \dots, k\}` and `a` is not in the base. The idea is developed
    and applied to backtracking algorithms in [1], pp.108-132. The points
    that are not in the base are taken in increasing order.

    References
    ==========

    .. [1] Holt, D., Eick, B., O'Brien, E.
           "Handbook of computational group theory"

    r      )lenrange)basedegreebase_lenorderingicurrents         ]/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/sympy/combinatorics/util.py_base_orderingr      sr    X 4yHs6zH8_ aG6] D=!HQKqLG O    c                 b   | j                   }| j                  }d}d}t               }t        |dz        D ]x  }||vs||dz  |z
  k  sd}|j	                  |       |}||   |k7  r$|dz  }||   }|j	                  |       ||   |k7  r$||z  }||dz  kD  sc||dz
  k  slt        |      sx y y)a  
    Checks for cycles of prime length p with n/2 < p < n-2.

    Explanation
    ===========

    Here `n` is the degree of the permutation. This is a helper function for
    the function is_alt_sym from sympy.combinatorics.perm_groups.

    Examples
    ========

    >>> from sympy.combinatorics.util import _check_cycles_alt_sym
    >>> from sympy.combinatorics import Permutation
    >>> a = Permutation([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], [11, 12]])
    >>> _check_cycles_alt_sym(a)
    False
    >>> b = Permutation([[0, 1, 2, 3, 4, 5, 6], [7, 8, 9, 10]])
    >>> _check_cycles_alt_sym(b)
    True

    See Also
    ========

    sympy.combinatorics.perm_groups.PermutationGroup.is_alt_sym

    r      r   TF)size
array_formsetr
   addr   )permnafcurrent_len	total_lenusedr   js           r   _check_cycles_alt_symr!   F   s    8 			A	BKI5D1a4[ D=QA	!11KHHQKAQ%1*q qE Q%1* $IQT!kAE&9gk>R r   c           
         t        |       }|d   j                  }t        |      D cg c]  }g  }}d}|D ]x  }d}||dz
  k  r>|j                  | |      | |   k(  r&|dz  }||dz
  k  r|j                  | |      | |   k(  r&||kD  r|}t        |dz         D ]  }	||	   j	                  |        z t        |dz   |      D ]1  }
||
   j	                  t        t        t        |                         3 |S c c}w )aL  
    Distribute the group elements ``gens`` by membership in basic stabilizers.

    Explanation
    ===========

    Notice that for a base `(b_1, b_2, \dots, b_k)`, the basic stabilizers
    are defined as `G^{(i)} = G_{b_1, \dots, b_{i-1}}` for
    `i \in\{1, 2, \dots, k\}`.

    Parameters
    ==========

    base : a sequence of points in `\{0, 1, \dots, n-1\}`
    gens : a list of elements of a permutation group of degree `n`.

    Returns
    =======
    list
        List of length `k`, where `k` is the length of *base*. The `i`-th entry
        contains those elements in *gens* which fix the first `i` elements of
        *base* (so that the `0`-th entry is equal to *gens* itself). If no
        element fixes the first `i` elements of *base*, the `i`-th element is
        set to a list containing the identity element.

    Examples
    ========

    >>> from sympy.combinatorics.named_groups import DihedralGroup
    >>> from sympy.combinatorics.util import _distribute_gens_by_base
    >>> D = DihedralGroup(3)
    >>> D.schreier_sims()
    >>> D.strong_gens
    [(0 1 2), (0 2), (1 2)]
    >>> D.base
    [0, 1]
    >>> _distribute_gens_by_base(D.base, D.strong_gens)
    [[(0 1 2), (0 2), (1 2)],
     [(1 2)]]

    See Also
    ========

    _strong_gens_from_distr, _orbits_transversals_from_bsgs,
    _handle_precomputed_bsgs

    r   r   )r	   r   r
   _array_formappend_af_newlist)r   gensr   r   _stabsmax_stab_indexgenr    kr   s              r   _distribute_gens_by_baser-   v   s   ` 4yH!W\\Fx)AR)E)N !(Q,3??47#;tAw#FFA (Q,3??47#;tAw#F~Nq1u 	!A!HOOC 	!! >A%x0 6aU6] 3456L *s   	C7Nc                     |t        | |      }|!|t        | |      \  }}nQt        | |d      }nB|@t        |       }dg|z  }t        |      D ]!  }t	        ||   j                               ||<   # |||fS )a_  
    Calculate BSGS-related structures from those present.

    Explanation
    ===========

    The base and strong generating set must be provided; if any of the
    transversals, basic orbits or distributed strong generators are not
    provided, they will be calculated from the base and strong generating set.

    Parameters
    ==========

    base : the base
    strong_gens : the strong generators
    transversals : basic transversals
    basic_orbits : basic orbits
    strong_gens_distr : strong generators distributed by membership in basic stabilizers

    Returns
    =======

    (transversals, basic_orbits, strong_gens_distr)
        where *transversals* are the basic transversals, *basic_orbits* are the
        basic orbits, and *strong_gens_distr* are the strong generators distributed
        by membership in basic stabilizers.

    Examples
    ========

    >>> from sympy.combinatorics.named_groups import DihedralGroup
    >>> from sympy.combinatorics.util import _handle_precomputed_bsgs
    >>> D = DihedralGroup(3)
    >>> D.schreier_sims()
    >>> _handle_precomputed_bsgs(D.base, D.strong_gens,
    ... basic_orbits=D.basic_orbits)
    ([{0: (2), 1: (0 1 2), 2: (0 2)}, {1: (2), 2: (1 2)}], [[0, 1, 2], [1, 2]], [[(0 1 2), (0 2), (1 2)], [(1 2)]])

    See Also
    ========

    _orbits_transversals_from_bsgs, _distribute_gens_by_base

    NT)transversals_only)r-   _orbits_transversals_from_bsgsr	   r
   r&   keys)r   strong_genstransversalsbasic_orbitsstrong_gens_distrr   r   s          r   _handle_precomputed_bsgsr6      s    \  4T;G.t5FG 'L, /t5F=AC  4yH 6(?L8_ ?"&|A';';'=">Q?'888r   c                 T   ddl m} t        |       }|d   d   j                  }dg|z  }dg|z  }|du rdg|z  }	t	        |      D ]S  }
 ||||
   | |
   dd      \  ||
<   ||
<   t        ||
         ||
<   |du s5t        ||
   j                               	|
<   U |r|S |s	|fS 	||fS )aX  
    Compute basic orbits and transversals from a base and strong generating set.

    Explanation
    ===========

    The generators are provided as distributed across the basic stabilizers.
    If the optional argument ``transversals_only`` is set to True, only the
    transversals are returned.

    Parameters
    ==========

    base : The base.
    strong_gens_distr : Strong generators distributed by membership in basic stabilizers.
    transversals_only : bool, default: False
        A flag switching between returning only the
        transversals and both orbits and transversals.
    slp : bool, default: False
        If ``True``, return a list of dictionaries containing the
        generator presentations of the elements of the transversals,
        i.e. the list of indices of generators from ``strong_gens_distr[i]``
        such that their product is the relevant transversal element.

    Examples
    ========

    >>> from sympy.combinatorics import SymmetricGroup
    >>> from sympy.combinatorics.util import _distribute_gens_by_base
    >>> S = SymmetricGroup(3)
    >>> S.schreier_sims()
    >>> strong_gens_distr = _distribute_gens_by_base(S.base, S.strong_gens)
    >>> (S.base, strong_gens_distr)
    ([0, 1], [[(0 1 2), (2)(0 1), (1 2)], [(1 2)]])

    See Also
    ========

    _distribute_gens_by_base, _handle_precomputed_bsgs

    r   )_orbit_transversalNFT)pairsslp)sympy.combinatorics.perm_groupsr8   r	   r   r
   dictr&   r1   )r   r5   r/   r:   r8   r   r   r3   slpsr4   r   s              r   r0   r0      s    V C4yHq!!$))F6(?L6(?DE!vh8_ ;#5f>OPQ>R!%a$$@ Qa|A/Q%"<?#7#7#9:LO; --\4//r   c                    ddl m} t        |       }|d   j                  }|t	        | |      }|3g }t        |      D ]#  } ||||   | |         }|j                  |       % |j                  g        |dd }	t        |dz
  dd      D ]s  }||   dd }
||   D ]a  }|||dz      vs|
dd }|j                  |       |g k(  r* |||| |         }|||   k(  s@|
j                  |       |	j                  |       c u |	S )a  
    Remove redundant generators from a strong generating set.

    Parameters
    ==========

    base : a base
    strong_gens : a strong generating set relative to *base*
    basic_orbits : basic orbits
    strong_gens_distr : strong generators distributed by membership in basic stabilizers

    Returns
    =======

    A strong generating set with respect to ``base`` which is a subset of
    ``strong_gens``.

    Examples
    ========

    >>> from sympy.combinatorics import SymmetricGroup
    >>> from sympy.combinatorics.util import _remove_gens
    >>> from sympy.combinatorics.testutil import _verify_bsgs
    >>> S = SymmetricGroup(15)
    >>> base, strong_gens = S.schreier_sims_incremental()
    >>> new_gens = _remove_gens(base, strong_gens)
    >>> len(new_gens)
    14
    >>> _verify_bsgs(S, base, new_gens)
    True

    Notes
    =====

    This procedure is outlined in [1],p.95.

    References
    ==========

    .. [1] Holt, D., Eick, B., O'Brien, E.
           "Handbook of computational group theory"

    r   )_orbitNr   )r;   r?   r	   r   r-   r
   r$   remove)r   r2   r4   r5   r?   r   r   r   basic_orbitres	gens_copyr+   	temp_gens
temp_orbits                 r   _remove_gensrG   8  s@   X 74yH^  F 4T;Gx 	-A ):1)=tAwGK,	- R 
a.C8a<R( $%a(+	$Q' 		$C+AE22%aL	  %?#FItAw?
a0$$S)JJsO		$$ Jr   c                    | j                   }t        |      }t        |      D ]S  }|||      }|||   k(  r|||   vrt        |      |dz   fc S ||   |   j                   }t	        t        |      |      }U t        |      |dz   fS )a!  
    Attempt to decompose a permutation using a (possibly partial) BSGS
    structure.

    Explanation
    ===========

    This is done by treating the sequence ``base`` as an actual base, and
    the orbits ``orbits`` and transversals ``transversals`` as basic orbits and
    transversals relative to it.

    This process is called "sifting". A sift is unsuccessful when a certain
    orbit element is not found or when after the sift the decomposition
    does not end with the identity element.

    The argument ``transversals`` is a list of dictionaries that provides
    transversal elements for the orbits ``orbits``.

    Parameters
    ==========

    g : permutation to be decomposed
    base : sequence of points
    orbits : list
        A list in which the ``i``-th entry is an orbit of ``base[i]``
        under some subgroup of the pointwise stabilizer of `
        `base[0], base[1], ..., base[i - 1]``. The groups themselves are implicit
        in this function since the only information we need is encoded in the orbits
        and transversals
    transversals : list
        A list of orbit transversals associated with the orbits *orbits*.

    Examples
    ========

    >>> from sympy.combinatorics import Permutation, SymmetricGroup
    >>> from sympy.combinatorics.util import _strip
    >>> S = SymmetricGroup(5)
    >>> S.schreier_sims()
    >>> g = Permutation([0, 2, 3, 1, 4])
    >>> _strip(g, S.base, S.basic_orbits, S.basic_transversals)
    ((4), 5)

    Notes
    =====

    The algorithm is described in [1],pp.89-90. The reason for returning
    both the current state of the element being decomposed and the level
    at which the sifting ends is that they provide important information for
    the randomized version of the Schreier-Sims algorithm.

    References
    ==========

    .. [1] Holt, D., Eick, B., O'Brien, E."Handbook of computational group theory"

    See Also
    ========

    sympy.combinatorics.perm_groups.PermutationGroup.schreier_sims
    sympy.combinatorics.perm_groups.PermutationGroup.schreier_sims_random

    r   )r#   r	   r
   r%   r   r   )	gr   orbitsr3   hr   r   betaus	            r   _striprN     s    @ 	
A4yH8_ 'az47?vay 1:q1u$$OD!--Z]A&' 1:x!|##r   c                    t        |      }t        |dz   |      D ]  }| ||      }	|	||   k(  r|	||   vr|s	| |dz   fc S | |dz   |fc S ||   |	   }
| |
k(  r|s	d|dz   fc S d|dz   |fc S t        t        |
      |       } |sj||   |	   dd }|j	                          |D cg c]  }||ff	 }}||z   } |s| |dz   fS | |dz   |fS c c}w )z
    optimized _strip, with h, transversals and result in array form
    if the stripped elements is the identity, it returns False, base_len + 1

    j    h[base[i]] == base[i] for i <= j

    r   FN)r	   r
   r   r   reverse)rK   r   rJ   r3   r    r:   r=   r   r   rL   rM   u_slprI   s                r   	_strip_afrR     s    4yH1Q3! az47?vay !a%xa!eS= OD!6hl**(Q,++Z]A&GDM!$EMMO(-.1a!Y.E.#+C%& (Q,hlC	 /s   $C
c                 |    t        |       dk(  r| d   dd S | d   }| d   D ]  }||vs|j                  |        |S )a%  
    Retrieve strong generating set from generators of basic stabilizers.

    This is just the union of the generators of the first and second basic
    stabilizers.

    Parameters
    ==========

    strong_gens_distr : strong generators distributed by membership in basic stabilizers

    Examples
    ========

    >>> from sympy.combinatorics import SymmetricGroup
    >>> from sympy.combinatorics.util import (_strong_gens_from_distr,
    ... _distribute_gens_by_base)
    >>> S = SymmetricGroup(3)
    >>> S.schreier_sims()
    >>> S.strong_gens
    [(0 1 2), (2)(0 1), (1 2)]
    >>> strong_gens_distr = _distribute_gens_by_base(S.base, S.strong_gens)
    >>> _strong_gens_from_distr(strong_gens_distr)
    [(0 1 2), (2)(0 1), (1 2)]

    See Also
    ========

    _distribute_gens_by_base

    r   r   N)r	   r$   )r5   resultr+   s      r   _strong_gens_from_distrrU     s\    @ " #A&&"1%$Q' 	#C& c"	# r   )NNN)FF)NN) sympy.combinatorics.permutationsr   r   r   sympy.ntheoryr   rmulr%   r   r!   r-   r6   r0   rG   rN   rR   rU    r   r   <module>rZ      sr    N N !


5p-`>B >BBF>9D AF=0@DNJ$Z 57R  B'r   