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 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184
use std::marker; use std::string::ToString; pub trait FrameRate { const FPS: u32; } #[derive(Debug, PartialEq)] pub struct FrameRate2400; impl FrameRate for FrameRate2400 { const FPS: u32 = 24; } #[derive(Debug, PartialEq)] pub struct FrameRate2500; impl FrameRate for FrameRate2500 { const FPS: u32 = 25; } #[derive(Debug, PartialEq)] pub struct FrameRate3000; impl FrameRate for FrameRate3000 { const FPS: u32 = 30; } #[derive(Debug, PartialEq)] pub struct FrameRate5000; impl FrameRate for FrameRate5000 { const FPS: u32 = 50; } #[derive(Debug, PartialEq)] pub struct FrameRate6000; impl FrameRate for FrameRate6000 { const FPS: u32 = 60; } #[derive(Debug, Clone, PartialEq, Serialize)] pub struct Timecode<FrameRate> { pub hours: u8, pub minutes: u8, pub seconds: u8, pub frame: u8, pub drop_frame: bool, pub color_frame: bool, frame_rate: marker::PhantomData::<FrameRate>, } impl<FrameRate> ToString for Timecode<FrameRate> { fn to_string(&self) -> String { let separator = match self.drop_frame { true => ';', false => ':', }; format!( "{:02}:{:02}:{:02}{}{:02}", self.hours, self.minutes, self.seconds, separator, self.frame ) } } impl<FrameRate> From<u32> for Timecode<FrameRate> { fn from(frames: u32) -> Self { let fps = 24; let hours = frames / (60 * 60 * fps); let minutes = (frames / (60 * fps)) - hours * 60; let seconds = (frames / fps) - minutes * 60 - hours * 60 * 60; let frame = frames - seconds * fps - minutes * 60 * fps - hours * 60 * 60 * fps; let drop_frame = false; let color_frame = false; log::trace!("{} {} {} {}", hours, minutes, seconds, frame); Timecode { hours: hours as u8, minutes: minutes as u8, seconds: seconds as u8, frame: frame as u8, drop_frame, color_frame, frame_rate: marker::PhantomData } } } impl<FrameRate> Timecode<FrameRate> { pub fn parse_smpte_331m(data: &[u8]) -> Option<Self> { if data.len() != 17 { return None; } match data[0] { 0x81 => Timecode::<FrameRate>::parse_smpte_12m(&data[1..]), _ => None, } } pub fn parse_smpte_12m(data: &[u8]) -> Option<Self> { if data.len() < 4 { return None; } let mask_tens_2 = 0b0011_0000; let mask_tens_3 = 0b0111_0000; let frame = Timecode::<FrameRate>::get_number(data[0], mask_tens_2); let seconds = Timecode::<FrameRate>::get_number(data[1], mask_tens_3); let minutes = Timecode::<FrameRate>::get_number(data[2], mask_tens_3); let hours = Timecode::<FrameRate>::get_number(data[3], mask_tens_2); let color_frame = (data[0] & 0b1000_0000) != 0; let drop_frame = (data[0] & 0b0100_0000) != 0; Some(Timecode::<FrameRate> { hours, minutes, seconds, frame, drop_frame, color_frame, frame_rate: marker::PhantomData::<FrameRate>, }) } fn get_number(data: u8, mask_tens: u8) -> u8 { let mask_unit = 0x0F; let tens = (data & mask_tens) >> 4; let unit = data & mask_unit; (10 * tens) + unit } } #[test] fn timecode_from_frame() { let t1 = Timecode::<FrameRate2400>::from(24 * 60 * 60 * 10); assert_eq!(t1.hours, 10); assert_eq!(t1.minutes, 0); assert_eq!(t1.seconds, 0); assert_eq!(t1.frame, 0); let t1 = Timecode::<FrameRate2400>::from(24 * 60 * 60); assert_eq!(t1.hours, 1); assert_eq!(t1.minutes, 0); assert_eq!(t1.seconds, 0); assert_eq!(t1.frame, 0); let t2 = Timecode::<FrameRate2400>::from(24 * 60 * 60 - 1); assert_eq!(t2.hours, 0); assert_eq!(t2.minutes, 59); assert_eq!(t2.seconds, 59); assert_eq!(t2.frame, 23); let t2 = Timecode::<FrameRate2400>::from(24 * 60); assert_eq!(t2.hours, 0); assert_eq!(t2.minutes, 1); assert_eq!(t2.seconds, 0); assert_eq!(t2.frame, 0); let t2 = Timecode::<FrameRate2400>::from(24 * 60 - 1); assert_eq!(t2.hours, 0); assert_eq!(t2.minutes, 0); assert_eq!(t2.seconds, 59); assert_eq!(t2.frame, 23); let t3 = Timecode::<FrameRate2400>::from(24); assert_eq!(t3.hours, 0); assert_eq!(t3.minutes, 0); assert_eq!(t3.seconds, 1); assert_eq!(t3.frame, 0); let t3 = Timecode::<FrameRate2400>::from(23); assert_eq!(t3.hours, 0); assert_eq!(t3.minutes, 0); assert_eq!(t3.seconds, 0); assert_eq!(t3.frame, 23); }