mkutils 0.1.120

Utility methods, traits, and types.
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
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
115
116
117
118
119
120
121
122

use crate::utils::Utils;
use derive_more::{Add, Constructor};
use mkutils_macros::{ConstAssoc, SaturatingAdd as MkutilsSaturatingAdd};
use num::{Zero, traits::SaturatingAdd};
use std::ops::Add;
use zed_sum_tree::ContextLessSummary;

macro_rules! dimension_type_impls {
    ($dimension_type:ident, $dimension_field:ident) => {
        #[derive(
            ::derive_more::Add,
            ::derive_more::Constructor,
            ::mkutils_macros::ConstAssoc,
            ::mkutils_macros::SaturatingAdd,
            ::std::clone::Clone,
        )]
        #[const_assoc(pub ZERO: Self = Self::new(0))]
        // #[const_assoc(pub ONE: Self = Self::new(1))]
        pub struct $dimension_type(usize);

        impl $dimension_type {
            pub const fn get(&self) -> usize {
                self.0
            }
        }

        impl<'a> ::zed_sum_tree::Dimension<'a, TextSummary> for $dimension_type {
            fn zero(_context: <TextSummary as ::zed_sum_tree::Summary>::Context<'a>) -> Self {
                Self::ZERO
            }

            fn add_summary(
                &mut self,
                text_summary: &'a TextSummary,
                _context: <TextSummary as ::zed_sum_tree::Summary>::Context<'a>,
            ) {
                crate::utils::Utils::saturating_add_assign(&mut self.0, &text_summary.length.$dimension_field)
            }
        }
    };
}

#[derive(Add, Clone, ConstAssoc, Constructor, MkutilsSaturatingAdd)]
#[const_assoc(pub ZERO: Self = Self::new(0, 0))]
pub struct Length {
    pub newlines: usize,
    pub extended_graphemes: usize,
}

#[derive(Clone, ConstAssoc, Constructor)]
#[const_assoc(pub ZERO: Self = Self::uniform(0))]
pub struct LineLengthSet {
    pub first: usize,
    pub last: usize,
    pub max: usize,
}

impl LineLengthSet {
    pub const fn uniform(len: usize) -> Self {
        Self::new(len, len, len)
    }
}

#[derive(Clone, ConstAssoc, Constructor)]
#[const_assoc(pub ZERO: Self = Self::new(Length::ZERO, LineLengthSet::ZERO))]
pub struct TextSummary {
    pub length: Length,
    pub line_lengths: LineLengthSet,
}

impl TextSummary {
    fn add_ref(&self, other: &Self) -> Self {
        let length = self.length.saturating_add(&other.length);
        let first = if self.length.newlines.is_zero() {
            self.line_lengths.last.saturating_add(other.line_lengths.first)
        } else {
            self.line_lengths.first
        };
        let last = if other.length.newlines.is_zero() {
            self.line_lengths.last.saturating_add(other.line_lengths.first)
        } else {
            other.line_lengths.last
        };
        let max = crate::max!(self.line_lengths.max, other.line_lengths.max, first, last);
        let line_lengths = LineLengthSet::new(first, last, max);

        Self::new(length, line_lengths)
    }

    pub fn add_to(self, other: &mut Self) -> Self {
        other.saturating_add_assign(self.ref_immut());

        self
    }
}

impl Add<Self> for TextSummary {
    type Output = Self;

    fn add(self, rhs: Self) -> Self::Output {
        self.add_ref(&rhs)
    }
}

impl SaturatingAdd for TextSummary {
    fn saturating_add(&self, v: &Self) -> Self {
        self.add_ref(v)
    }
}

impl ContextLessSummary for TextSummary {
    fn zero() -> Self {
        Self::ZERO
    }

    fn add_summary(&mut self, other: &Self) {
        self.saturating_add_assign(other);
    }
}

dimension_type_impls!(LengthNewlines, newlines);
dimension_type_impls!(LengthExtendedGraphemes, extended_graphemes);