U
.e @ s d Z ddlT ddlZG dd deZG dd deZdd Zd
d Zd d
Z dd Z
dd Z d%ddZ dd Z
dd Zdd Zdd Zd d Zd d Zd d! Zd"Zd#Zed$kre Ze 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)sizeTurtle__init__Zpu shapesize fillcolorst)selfr r 2/usr/lib64/python3.8/turtledemo/sorting_animate.pyr s
zBlock.__init__c C s | d d S )NZredr r
r r r glow s z
Block.glowc C s | d d S )Nr r r r r r unglow" s z Block.unglowc C s d | jS )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 blockijN)yx)r
r r r r r + s zShelf.__init__c C sP | \}}}|d d }|| j| || jdt| | | d S )Nr " )r
setyr setxr lenappend)r
dwidth_y_offsetr r r push0 s
z
Shelf.pushc C s0 | |d D ] }| \}}||d q d S Nr posr r
ibZxposr$ r r r _close_gap_from_i8 s zShelf._close_gap_from_ic C s0 | |d D ] }| \}}||d q d S r' r( r* r r r _open_gap_from_i= s zShelf._open_gap_from_ic C s, t | |}| |d | | |S )N )listpopr r r- )r
keyr, r r r r1 B s
z Shelf.popc C sb | | t| || || jd| | \}}}|d d }|| j| | d S )Nr r r ) r. r0 insertr r r
r r r )r
r2 r, r# r$ r% r r r r3 I s
z Shelf.insertN) r r r r r& r- r. r1 r3 r r r r r ) s r c C s\ t | }td|D ]D}|}|dkrD| | j| |d jk rD|d }q| || | qd S )N r r ranger r3 r1 )shelflengthr+ Zholer r r isortS s
r9 c C sj t | }td|d D ]N}|}t|d |D ] }| | j| | jk r,|}q,||kr| || | qd S )Nr r4 r5 )r7 r8 jZiminr+ r r r ssort\ s r; c C sn | | }| || | |}t||D ].}| | j|jk r(| || | |d }q(| || | |S Nr4 )r3 r1 r6 r )r7 leftright pivot_indexZpivotZ store_indexr+ r r r partitionf s
r@ c C s> ||k r:|}t | |||}t| ||d t| |d | d S r< )r@ qsort)r7 r= r> r? Zpivot_new_indexr r r rA q s
rA c C s t t ttd} t| t| D ]@\}}t|ttD ](}t| j |d kr@t
|t | q@q*t t
t tdd t d S )N
r4 line) disable_keysclearr0 r6 randomZshuffle enumerater sr r3 r1 show_text
instructions1
instructions2 enable_keys)targetr+ tr: r r r randomizex s
rP c C s( d| }t dd| t| ddd d S )Nr r icenter)ZCourier Zbold)ZalignZfont)Zgotowrite)textrD r r r rJ s rJ c C s@ t t td tt t tt ttdd t d S )NzSelection Sortr4 rC )rE rF rJ r; rI rK rL rM r r r r start_ssort s rU c C s@ t t td tt t tt ttdd t d S )NzInsertion Sortr4 rC )rE rF rJ r9 rI rK rL rM r r r r start_isort s rV c C sL t t td ttdttd t tt ttdd t d S )NZ Quicksortr r4 rC ) rE rF rJ rA rI r rK rL rM r r r r start_qsort s rW c C s( t dad} | D ]}tt| qd S )Ni8)
r r4 rB )r rI r& r )Zvalsr+ r r r
init_shelf s r_ c C s, t d d t d d t d d t d d d S )NrI r+ qr)onkeyr r r r rE s
rE c C s6 t td t td t td t td t td d S )Nr+ rI r` ra Zspace)rb rV rU rW rP Zbyer r r r rM s
rM c C s@ t t t t tt ttdd t t dS )Nr4 rC Z EVENTLOOP)
Z getscreenZ clearscreenZhtZpenupr_ rJ rK rL rM Zlistenr r r r main s
rc zApress i for insertion sort, s for selection sort, q for quicksortz spacebar to quit, r to randomize__main__)r )__doc__ZturtlerG r r r0 r r9 r; r@ rA rP rJ rU rV rW r_ rE rM rc rK rL r msgZmainloopr r r r s, *