summaryrefslogtreecommitdiff
path: root/src/gallium/frontends/rusticl/util/math.rs
blob: f6e8d96e9ae93d90d7f71e15f8afbccd280e405c (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
use std::ops::Add;
use std::ops::Rem;
use std::ops::Sub;

pub fn gcd<T>(mut a: T, mut b: T) -> T
where
    T: Copy + Default + PartialEq,
    T: Rem<Output = T>,
{
    let mut c = a % b;
    while c != T::default() {
        a = b;
        b = c;
        c = a % b;
    }

    b
}

pub fn align<T>(val: T, a: T) -> T
where
    T: Add<Output = T>,
    T: Copy,
    T: Default,
    T: PartialEq,
    T: Rem<Output = T>,
    T: Sub<Output = T>,
{
    let tmp = val % a;
    if tmp == T::default() {
        val
    } else {
        val + (a - tmp)
    }
}