֦i d Z ddl ddlZ G d de Z G d de Zd Zd Zd
Z d Z
d Z dd
Z d Z
d Zd Zd Zd Zd Zd ZdZdZedk( r e Z e yy)a
sorting_animation.py
A minimal sorting algorithm animation:
Sorts a shelf of 10 blocks using insertion
sort, selection sort and quicksort.
Shelves are implemented using builtin lists.
Blocks are turtles with shape "square", but
stretched to rectangles by shapesize()
---------------------------------------
To exit press space button
---------------------------------------
)*Nc $ e Zd Zd Zd Zd Zd Zy)Blockc || _ t j | dd | j | j |dz dd | j d | j
y )NsquareF)shapevisibleg ? black)sizeTurtle__init__pu shapesize fillcolorst)selfr s 3/usr/lib64/python3.12/turtledemo/sorting_animate.pyr zBlock.__init__ sM He< tcz3* w c & | j d y )Nredr r s r glowz
Block.glow s u r c & | j d y )Nr r r s r unglowz Block.unglow# s w r c 8 dj | j S )NzBlock size: {0})formatr r s r __repr__zBlock.__repr__& s '' 22r N)__name__
__module__ __qualname__r r r r r r r r s 3r r c 0 e Zd Zd Zd Zd Zd Zd Zd Zy)Shelfc || _ d| _ y)z.create a shelf. y is y-position of first blockijN)yx)r r' s r r zShelf.__init__, s r c |j \ }}}|dz dz }|j | j |z |j | j dt | z z | j
| y )Nr
" )r setyr' setxr( lenappend)r dwidth_y_offsets r pushz
Shelf.push1 s] kkm q!19r> tvv ! tvvSY&' Ar c f | |d D ]) }|j \ }}|j |dz
+ y Nr+ posr- r ibxposr2 s r _close_gap_from_izShelf._close_gap_from_i9 1 abAeegGD!
FF4"9 r c f | |d D ]) }|j \ }}|j |dz + y r6 r7 r9 s r _open_gap_from_izShelf._open_gap_from_i> r> r c t j | | }|j |j d | j | |S )N )listpopr r, r= )r keyr; s r rD z Shelf.popC s9 HHT3 s s#r c * | j | t j | || |j | j d|z z |j \ }}}|dz dz }|j
| j |z |j y )Nr+ r
r* ) r@ rC insertr- r( r r, r' r )r rE r; r1 r2 r3 s r rG z Shelf.insertJ su c" D#q! tvvS !kkm q!19r> tvv !
r N) r r! r" r r4 r= r@ rD rG r# r r r% r% * s
r r% c * t | }t d| D ]y }|}|dkD rO| | j | |dz
j k r-|dz
}|dkD r#| | j | |dz
j k r-| j || j | { y )N r r. ranger rG rD )shelflengthr: holes r isortrO T s
ZF
1f
Qh58==5?+?+??!8D Qh58==5?+?+??
T599Q<(
r c t | }t d|dz
D ]a }|}t |dz | D ]$ }| | j | | j k s#|}& ||k7 sA| j || j | c y )Nr rI rJ )rL rM jiminr: s r ssortrS ] sx
ZF
1fqj
!q1uf%AQx}}uT{ / // & 19 LLEIIdO ,
"r c @ | | }| j || j | |}t || D ]E }| | j |j k s | j || j | |dz }G | j || j | |S NrI )rG rD rK r )rL leftright pivot_indexpivot store_indexr: s r partitionr[ g s + E LL +./K
4
8==5:: % LLeiil 3%/K
LLeii ./ r c p ||k r1|}t | ||| }t | ||dz
t | |dz | y y rU )r[ qsort)rL rV rW rX pivot_new_indexs r r] r] r sE e| #E4 D
eT?Q./
e_q(%0 r c t t t t d } t j
| t
| D ]f \ }}t |t t D ]E }t | j |dz k( s t j |t j | G h t t t t d t y )N
rI line) disable_keysclearrC rK randomshuffle enumerater. sr rG rD show_text
instructions1
instructions2 enable_keys)targetr: trQ s r randomizero y s N G
%)_F
NN6&!1q#a&!AtyyAE!AEE!H% " " m
m!$Mr c H d|z }t dd|z
t | dd y )Nr* r icenter)Courier bold)alignfont)gotowrite)textrb s r ri ri s&
9D4$; $h%<=r c t t t d t t t t t
t t d t y )NzSelection SortrI ra )rc rd ri rS rh rj rk rl r# r r start_ssortr{ 8 N G
!H G
m
m!$Mr c t t t d t t t t t
t t d t y )NzInsertion SortrI ra )rc rd ri rO rh rj rk rl r# r r start_isortr~ r| r c t t t d t t dt t dz
t t t t t d t y )N Quicksortr rI ra ) rc rd ri r] rh r. rj rk rl r# r r start_qsortr sD N G
k !QA
G
m
m!$Mr c h t d ad} | D ] }t j t | " y )Ni8)
r
rI r` )r% rh r4 r )valsr: s r
init_shelfr s*
d A *D
uQx r c d t d d t d d t d d t d d y )Nrh r: qr)onkeyr# r r rc rc s* $ $ $ $r c t t d t t d t t d t t d t t
d y )Nr: rh r r space)r r~ r{ r ro byer# r r rl rl s4 +s +s +s )S #wr c t j t t t t t t t d t t y)NrI ra EVENTLOOP)
getscreen clearscreenhtpenupr ri rj rk rl listenr# r r mainr s=
K D%'L
m
m!$M
H r zApress i for insertion sort, s for selection sort, q for quicksortz spacebar to quit, r to randomize__main__)r )__doc__turtlere r
r rC r% rO rS r[ r] ro ri r{ r~ r r rc rl r rj rk r msgmainloopr# r r r s
3F 3((D (T - 1 >
T
2
Z
&C J r