Add support for more SI suffixes

williamyang98 · Jul 23, 2024 · 8dc6f2b · 8dc6f2b
1 parent f1011e4
commit 8dc6f2b
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/ffmpeg.rs b/src/ffmpeg.rs
@@ -6,8 +6,11 @@ use thiserror::Error;
 enum SizeBytes {
     Byte,
     KiB,
+    KB,
     MiB,
+    MB,
     GiB,
+    GB,
 }
 
 impl TryFrom<&str> for SizeBytes {
@@ -16,8 +19,11 @@ impl TryFrom<&str> for SizeBytes {
         match v {
             "B"   => Ok(Self::Byte),
             "KiB" => Ok(Self::KiB),
+            "kB" => Ok(Self::KB),
             "MiB" => Ok(Self::MiB),
+            "MB" => Ok(Self::MB),
             "GiB" => Ok(Self::GiB),
+            "GB" => Ok(Self::GB),
             _ => Err("Unknown unit"),
         }
     }
@@ -27,9 +33,12 @@ impl SizeBytes {
     fn to_bytes(self) -> usize {
         match self {
             Self::Byte => 1,
-            Self::KiB => 1_000,
-            Self::MiB => 1_000_000,
-            Self::GiB => 1_000_000_000,
+            Self::KiB => 1024,
+            Self::MiB => 1024*1024,
+            Self::GiB => 1024*1024*1024,
+            Self::KB => 1000,
+            Self::MB => 1000*1000,
+            Self::GB => 1000*1000*1000,
         }
     }
 }