Implemented the stackstack

1 files changed, 57 insertions, 0 deletions

diff --git a/src/stackStack.nim b/src/stackStack.nim
new file mode 100644
index 0000000..6d32c87
--- /dev/null
+++ b/src/stackStack.nim
@@ -0,0 +1,57 @@
+import stack
+
+type
+  StackStack* = object
+    height: int
+    head: ref Stack
+
+func NewStackStack*(): ref StackStack =
+  result = new(StackStack)
+  result.head = NewStack()
+  result.height = 1
+
+func Pop*(ss: StackStack): int =
+  return ss.head[].Pop()
+
+func Push*(ss: StackStack, v: int) =
+  ss.head[].Push(v)
+
+func PopVector*(ss: StackStack): (int, int) =
+  return ss.head[].PopVector()
+
+func PushVector*(ss: var StackStack, v: tuple[x, y: int]) =
+  ss.head[].PushVector(v)
+
+func Clear*(ss: var StackStack) =
+  ss.head[].Clear()
+
+func Begin*(ss: var StackStack, v: tuple[x, y: int]) =
+  inc ss.height
+  let soss = ss.head
+  let n = soss[].Pop()
+  ss.head = NewStack(size = abs(n), next = soss)
+  let toss = ss.head
+  if n > 0:
+    toss[].Transfert(soss[], n)
+  elif n < 0:
+    for i in 0 ..< -n:
+      soss[].Push(0)
+  soss[].PushVector(v)
+
+func End*(ss: var StackStack, v: var tuple[x, y: int]): bool =
+  ## Implements the '}' command behaviour which pops a stack from the stack stack
+  ## returns true if a reflect should happen
+  if ss.height == 1:
+    return true
+  let toss = ss.head
+  let soss = toss[].Next()
+  let n = toss[].Pop()
+  v.y = soss[].Pop()
+  v.x = soss[].Pop()
+  if n > 0:
+    soss[].Transfert(toss[], n)
+  else:
+    soss[].Discard(-n)
+  dec ss.height
+  ss.head = soss
+  return false