nstack 0.16.0

Arity 4 stack for the kelvin merkle toolkit
Documentation 
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
//
// Copyright (c) DUSK NETWORK. All rights reserved.

use crate::annotation::Cardinality;
use crate::NStack;

use core::borrow::Borrow;

use microkelvin::{Branch, BranchMut, Child, Step, Walk, Walker};
use ranno::Annotation;

impl<T, A> NStack<T, A>
where
    A: Annotation<Self> + Borrow<Cardinality>,
{
    /// Construct a [`Branch`] pointing to the `nth` element, if any
    pub fn nth(&self, index: u64) -> Option<Branch<Self, A>> {
        Branch::walk(self, Index(index))
    }

    /// Construct a [`BranchMut`] pointing to the `nth` element, if any
    pub fn nth_mut(&mut self, index: u64) -> Option<BranchMut<Self, A>> {
        BranchMut::walk(self, Index(index))
    }
}

struct Index(u64);

impl<T, A> Walker<NStack<T, A>, A> for Index
where
    A: Annotation<NStack<T, A>> + Borrow<Cardinality>,
{
    fn walk(&mut self, walk: Walk<NStack<T, A>, A>) -> Step {
        for i in 0.. {
            match walk.child(i) {
                Child::Leaf(_) => {
                    if self.0 == 0 {
                        return Step::Found(i);
                    } else {
                        self.0 -= 1
                    }
                }
                Child::Node(node) => {
                    let anno = node.anno();
                    let c = (*anno).borrow();

                    let c = **c;

                    if self.0 < c {
                        return Step::Into(i);
                    }

                    self.0 -= c;
                }
                Child::Empty => (),
                Child::EndOfNode => return Step::Abort,
            }
        }
        unreachable!()
    }
}