TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

vm: add a TUCK implementation with tests

Browse source 
Also fix stack's InsertAt for the edge case of inserting into a position that
currently is empty (the first good TUCK test illustrates this).
This commit is contained in:
Roman Khimov 2019-09-05 16:34:35 +03:00
parent eb224aeebe
commit 974f5db6bc
3 changed files with 53 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pkg/vm/stack.go
View file

@ -132,23 +132,15 @@ func (s *Stack) insert(e, at *Element) *Element {
return e
}
// InsertBefore will insert the element before the mark on the stack.
func (s *Stack) InsertBefore(e, mark *Element) *Element {
if mark == nil {
return nil
}
return s.insert(e, mark.prev)
}
// InsertAt will insert the given item (n) deep on the stack.
// Be very careful using it and _always_ check both e and n before invocation
// as it will silently do wrong things otherwise.
func (s *Stack) InsertAt(e *Element, n int) *Element {
before := s.Peek(n)
before := s.Peek(n - 1)
if before == nil {
return nil
}
return s.InsertBefore(e, before)
return s.insert(e, before)
}
// Push pushes the given element on the stack.