ffmpeg_the_third/util/
rational.rs

1use std::cmp::Ordering;
2use std::fmt;
3use std::ops::{Add, Div, Mul, Sub};
4
5use crate::ffi::*;
6use libc::c_int;
7
8#[derive(Copy, Clone)]
9pub struct Rational(pub i32, pub i32);
10
11impl Rational {
12    #[inline]
13    pub fn new(numerator: i32, denominator: i32) -> Self {
14        Rational(numerator, denominator)
15    }
16
17    #[inline]
18    pub fn numerator(&self) -> i32 {
19        self.0
20    }
21
22    #[inline]
23    pub fn denominator(&self) -> i32 {
24        self.1
25    }
26
27    #[inline]
28    pub fn reduce(&self) -> Rational {
29        match self.reduce_with_limit(i32::MAX) {
30            Ok(r) => r,
31            Err(r) => r,
32        }
33    }
34
35    #[inline]
36    pub fn reduce_with_limit(&self, max: i32) -> Result<Rational, Rational> {
37        unsafe {
38            let mut dst_num: c_int = 0;
39            let mut dst_den: c_int = 0;
40
41            let exact = av_reduce(
42                &mut dst_num,
43                &mut dst_den,
44                i64::from(self.numerator()),
45                i64::from(self.denominator()),
46                i64::from(max),
47            );
48
49            if exact == 1 {
50                Ok(Rational(dst_num, dst_den))
51            } else {
52                Err(Rational(dst_num, dst_den))
53            }
54        }
55    }
56
57    #[inline]
58    pub fn invert(&self) -> Rational {
59        Rational::from(av_inv_q((*self).into()))
60    }
61}
62
63impl From<AVRational> for Rational {
64    #[inline]
65    fn from(value: AVRational) -> Rational {
66        Rational(value.num, value.den)
67    }
68}
69
70impl From<Rational> for AVRational {
71    #[inline]
72    fn from(value: Rational) -> AVRational {
73        AVRational {
74            num: value.0,
75            den: value.1,
76        }
77    }
78}
79
80impl From<f64> for Rational {
81    #[inline]
82    fn from(value: f64) -> Rational {
83        unsafe { Rational::from(av_d2q(value, c_int::MAX)) }
84    }
85}
86
87impl From<Rational> for f64 {
88    #[inline]
89    fn from(value: Rational) -> f64 {
90        av_q2d(value.into())
91    }
92}
93
94impl From<Rational> for u32 {
95    #[inline]
96    fn from(value: Rational) -> u32 {
97        unsafe { av_q2intfloat(value.into()) }
98    }
99}
100
101impl From<(i32, i32)> for Rational {
102    fn from((num, den): (i32, i32)) -> Rational {
103        Rational::new(num, den)
104    }
105}
106
107impl PartialEq for Rational {
108    fn eq(&self, other: &Rational) -> bool {
109        if self.0 == other.0 && self.1 == other.1 {
110            return true;
111        }
112
113        let a = self.reduce();
114        let b = other.reduce();
115
116        if a.0 == b.0 && a.1 == b.1 {
117            return true;
118        }
119
120        false
121    }
122}
123
124impl Eq for Rational {}
125
126impl PartialOrd for Rational {
127    #[inline]
128    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
129        match av_cmp_q((*self).into(), (*other).into()) {
130            0 => Some(Ordering::Equal),
131            1 => Some(Ordering::Greater),
132            -1 => Some(Ordering::Less),
133
134            _ => None,
135        }
136    }
137}
138
139impl Add for Rational {
140    type Output = Rational;
141
142    #[inline]
143    fn add(self, other: Rational) -> Rational {
144        unsafe { Rational::from(av_add_q(self.into(), other.into())) }
145    }
146}
147
148impl Sub for Rational {
149    type Output = Rational;
150
151    #[inline]
152    fn sub(self, other: Rational) -> Rational {
153        unsafe { Rational::from(av_sub_q(self.into(), other.into())) }
154    }
155}
156
157impl Mul for Rational {
158    type Output = Rational;
159
160    #[inline]
161    fn mul(self, other: Rational) -> Rational {
162        unsafe { Rational::from(av_mul_q(self.into(), other.into())) }
163    }
164}
165
166impl Div for Rational {
167    type Output = Rational;
168
169    #[inline]
170    fn div(self, other: Rational) -> Rational {
171        unsafe { Rational::from(av_div_q(self.into(), other.into())) }
172    }
173}
174
175impl fmt::Display for Rational {
176    fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
177        f.write_str(&format!("{}/{}", self.numerator(), self.denominator()))
178    }
179}
180
181impl fmt::Debug for Rational {
182    fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
183        f.write_str(&format!(
184            "Rational({}/{})",
185            self.numerator(),
186            self.denominator()
187        ))
188    }
189}
190
191#[inline]
192pub fn nearer(q: Rational, q1: Rational, q2: Rational) -> Ordering {
193    unsafe {
194        match av_nearer_q(q.into(), q1.into(), q2.into()) {
195            1 => Ordering::Greater,
196            -1 => Ordering::Less,
197            _ => Ordering::Equal,
198        }
199    }
200}
ffmpeg_the_third/util/rational.rs

ffmpeg_the_third/util/
rational.rs
