about summary refs log tree commit diff
path: root/src/block/extension/graphiccontrol.rs
blob: 830dff401104541c362e9345a7e6b0d2fde029ce (plain)
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
use std::{convert::TryInto, fmt};

#[derive(Clone, Debug)]
pub struct GraphicControl {
    pub(crate) packed: u8,
    pub(crate) delay_time: u16,
    pub(crate) transparency_index: u8,
}

impl GraphicControl {
    pub fn new(
        disposal_method: DisposalMethod,
        user_input_flag: bool,
        transparency_flag: bool,
        delay_time: u16,
        transparency_index: u8,
    ) -> Self {
        let mut ret = Self {
            packed: 0,
            delay_time,
            transparency_index,
        };

        ret.set_disposal_method(disposal_method);
        ret.user_input(user_input_flag);
        ret.transparency(transparency_flag);

        ret
    }

    pub fn disposal_method(&self) -> Option<DisposalMethod> {
        match self.packed & 0b000_111_00 {
            0b000_000_00 => Some(DisposalMethod::NoAction),
            0b000_100_00 => Some(DisposalMethod::DoNotDispose),
            0b000_010_00 => Some(DisposalMethod::RestoreBackground),
            0b000_110_00 => Some(DisposalMethod::RestorePrevious),
            _ => None,
        }
    }

    pub fn set_disposal_method(&mut self, method: DisposalMethod) {
        match method {
            DisposalMethod::NoAction => self.packed &= 0b111_000_1_1,
            DisposalMethod::DoNotDispose => self.packed |= 0b000_100_0_0,
            DisposalMethod::RestoreBackground => self.packed |= 0b000_010_0_0,
            DisposalMethod::RestorePrevious => self.packed |= 0b000_110_0_0,
        };
    }

    pub fn transparency_index(&self) -> u8 {
        self.transparency_index
    }

    pub fn user_input(&mut self, flag: bool) {
        if flag {
            self.packed |= 0b000_000_1_0;
        } else {
            self.packed &= 0b111_111_0_1;
        }
    }

    pub fn transparency(&mut self, flag: bool) {
        if flag {
            self.packed |= 0b000_000_0_1;
        } else {
            self.packed &= 0b111_111_1_0;
        }
    }

    pub fn delay_time(&self) -> u16 {
        self.delay_time
    }

    pub fn delay_time_mut(&mut self) -> &mut u16 {
        &mut self.delay_time
    }

    //TODO: Transparency index setter
}

impl From<[u8; 4]> for GraphicControl {
    fn from(arr: [u8; 4]) -> Self {
        let packed = arr[0];
        let delay_time = u16::from_le_bytes(arr[1..3].try_into().unwrap());
        let transparency_index = arr[3];

        Self {
            packed,
            delay_time,
            transparency_index,
        }
    }
}

#[derive(Clone, Copy, Debug, PartialEq)]
pub enum DisposalMethod {
    NoAction,
    DoNotDispose,
    RestoreBackground,
    RestorePrevious,
}

impl fmt::Display for DisposalMethod {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let st = match self {
            DisposalMethod::NoAction => "Dispose as Normal",
            DisposalMethod::DoNotDispose => "No Dispose",
            DisposalMethod::RestoreBackground => "Restore to background",
            DisposalMethod::RestorePrevious => "Restore previous image",
        };

        write!(f, "{}", st)
    }
}