File: //usr/local/lib/python3.10/turtledemo/__pycache__/sorting_animate.cpython-310.pyc
o
�i� � @ s� d Z ddlT ddlZG dd� de�ZG dd� de�Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z d%dd�Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"Zd#Zed$kr_e� Ze� dS dS )&a�
sorting_animation.py
A minimal sorting algorithm animation:
Sorts a shelf of 10 blocks using insertion
sort, selection sort and quicksort.
Shelfs are implemented using builtin lists.
Blocks are turtles with shape "square", but
stretched to rectangles by shapesize()
---------------------------------------
To exit press space button
---------------------------------------
� )�*Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�Blockc C sF || _ tj| ddd� | �� | �|d dd� | �d� | �� d S )NZsquareF)�shapeZvisibleg �?� �black)�size�Turtle�__init__Zpu� shapesize� fillcolor�st)�selfr � r �7/usr/local/lib/python3.10/turtledemo/sorting_animate.pyr s
zBlock.__init__c C � | � d� d S )NZred�r �r
r r r �glow � z
Block.glowc C r )Nr r r r r r �unglow# r z Block.unglowc C s d� | j�S )NzBlock size: {0})�formatr r r r r �__repr__& s zBlock.__repr__N)�__name__�
__module__� __qualname__r r r r r r r r r s
r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�Shelfc C s || _ d| _dS )z.create a shelf. y is y-position of first blockij���N)�y�x)r
r r r r r , s
zShelf.__init__c C sP |� � \}}}|d d }|�| j| � |�| jdt| � � | �|� d S )Nr � �" )r
�setyr �setxr �len�append)r
�d�width�_�y_offsetr r r �push1 s
z
Shelf.pushc C s0 | |d � D ]}|� � \}}|�|d � qd S �Nr ��posr! �r
�i�bZxposr& r r r �_close_gap_from_i9 � �zShelf._close_gap_from_ic C s0 | |d � D ]}|� � \}}|�|d � qd S r) r* r, r r r �_open_gap_from_i> r0 zShelf._open_gap_from_ic C s, t �| |�}|�� |�d� | �|� |S )N�� )�list�popr r r/ )r
�keyr. r r r r4 C s
z Shelf.popc C sb | � |� t�| ||� |�| jd| � |�� \}}}|d d }|�| j| � |�� d S )Nr r r ) r1 r3 �insertr! r r
r r r )r
r5 r. r% r& r' r r r r6 J s
z Shelf.insertN) r r r r r( r/ r1 r4 r6 r r r r r * s r c C sz t | �}td|�D ]1}|}|dkr1| | j| |d jk r1|d }|dkr1| | j| |d jk s | �|| �|�� q d S )N� r �r" �ranger r6 r4 )�shelf�lengthr- Zholer r r �isortT s �r< c C sj t | �}td|d �D ]'}|}t|d |�D ]}| | j| | jk r$|}q||kr2| �|| �|�� q d S )Nr r7 r8 )r: r; �jZiminr- r r r �ssort] s ���r> c C sn | | }| � || �|�� |}t||�D ]}| | j|jk r+| � || �|�� |d }q| � || �|�� |S �Nr7 )r6 r4 r9 r )r: �left�right� pivot_indexZpivotZ store_indexr- r r r � partitiong s �rC c C sB ||k r |}t | |||�}t| ||d � t| |d |� d S d S r? )rC �qsort)r: r@ rA rB Zpivot_new_indexr r r rD r s �rD c C s� t � t� ttd��} t�| � t| �D ] \}}t|tt��D ]}t| j |d kr4t�
|t� |�� q qt t
� t tdd� t� d S )N�
r7 ��line)� disable_keys�clearr3 r9 �random�shuffle� enumerater" �sr r6 r4 � show_text�
instructions1�
instructions2� enable_keys)�targetr- �tr= r r r � randomizey s
��
rT c C s( d| }t dd| � t| ddd� d S )Nr r i����center)ZCourier� Zbold)ZalignZfont)Zgoto�write)�textrG r r r rN � s rN c C �@ t � t� td� tt� t� tt� ttdd� t� d S )NzSelection Sortr7 rF )rH rI rN r> rM rO rP rQ r r r r � start_ssort� �
rZ c C rY )NzInsertion Sortr7 rF )rH rI rN r< rM rO rP rQ r r r r � start_isort� r[ r\ c C sL t � t� td� ttdtt�d � t� tt� ttdd� t� d S )NZ Quicksortr r7 rF ) rH rI rN rD rM r" rO rP rQ r r r r � start_qsort� s
r] c C s( t d�ad} | D ] }t�t|�� qd S )Ni8���)
� r � � r7 � rE � � � )r rM r( r )Zvalsr- r r r �
init_shelf� s
�re c C s, t d d� t d d� t d d� t d d� d S )NrM r- �q�r)�onkeyr r r r rH � s
rH c C s6 t td� t td� t td� t td� t td� d S )Nr- rM rf rg �space)rh r\ rZ r] rT Zbyer r r r rQ � s
rQ c C s@ t � �� t� t� t� tt� ttdd� t� t � dS )Nr7 rF Z EVENTLOOP)
Z getscreenZ clearscreenZhtZpenupre rN rO rP rQ �listenr r r r �main� s
rk zApress i for insertion sort, s for selection sort, q for quicksortz spacebar to quit, r to randomize�__main__)r )�__doc__ZturtlerJ r r r3 r r< r> rC rD rT rN rZ r\ r] re rH rQ rk rO rP r �msgZmainloopr r r r �<module> s0 *
�