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

use borsh::{BorshDeserialize, BorshSerialize};
use core::fmt;

#[derive(PartialEq, Eq, PartialOrd, Ord, Clone, Copy, BorshDeserialize, BorshSerialize)]
pub enum StackUsize {
    U2,
    U3,
    U4,
}

impl Default for StackUsize {
    fn default() -> Self {
        StackUsize::U2
    }
}

impl fmt::Debug for StackUsize {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(
            f,
            "{}",
            match self {
                StackUsize::U2 => "StackUsize::U2",
                StackUsize::U3 => "StackUsize::U3",
                StackUsize::U4 => "StackUsize::U4",
            }
        )
    }
}

pub const fn init_storage_stack(bar: &StackUsize) -> usize {
    match bar {
        StackUsize::U2 => 2,
        StackUsize::U3 => 3,
        StackUsize::U4 => 4,
    }
}