stack-push.py
1.47 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
from codanim.data import Heap, Struct, Value, Box
from codanim.flow import FUNC, BLOCK, ENV, WS, DECL, EXPR, STMT
heap = Heap()
stack = heap.new(Struct({"val": Value(1), "next": None}))
stack = heap.new(Struct({"val": Value(2), "next": stack}))
s = Value(stack, nodeid="s")
t = Value(None, nodeid="t", value={"xshift": "-25mm"})
push = FUNC(BLOCK(ENV("s", s),
ENV("u", Value(3)),
ENV("top", t),
ENV("new", heap.new),
ENV("Struct", Struct),
ENV("Value", Value),
WS("\n "),
DECL("top", EXPR("new(Struct({'val': None, 'next': None}))",
src="malloc(sizeof(StackCell))"),
src="Stack {name} = {init};"),
WS("\n "),
STMT("top.val = u", src="top->val = u;"),
WS("\n "),
STMT("top.next = s", src="top->next = *s;"),
WS("\n "),
STMT("s = top", src="*s = top;"),
WS("\n")),
src="void stackPush (Stack* s, uint u) {{{body}}}\n")
# run code and save its animation
push.IP += 1
push()
with open("out.code", "w") as out :
out.write(push.tex())
# save data and animation
b = Box(heap, Box(t, s, grow="right"), grow="above")
with open("out.tikz", "w") as out :
out.write(b.tex(tail=r"""
\node[above] at (s.north) {\texttt{s}};
\node[above] at (t.north) {\texttt{top}};
"""))