Implement Display for Function

SalsaGal · Oct 7, 2024 · 0c89581 · 0c89581
1 parent b733594
commit 0c89581
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/sf2panlaw/src/main.rs b/sf2panlaw/src/main.rs
@@ -1,4 +1,5 @@
 use std::{
+    fmt::Display,
     fs::File,
     io::Write,
     path::{Path, PathBuf},
@@ -21,6 +22,15 @@ impl Function {
     }
 }
 
+impl Display for Function {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            Self::Amplify => f.write_str("amplify"),
+            Self::Attenuate => f.write_str("attenuate"),
+        }
+    }
+}
+
 #[derive(Parser)]
 #[clap(version)]
 struct Args {
@@ -96,13 +106,8 @@ fn convert(path: &Path, function: Function) {
     }
 
     let mut output = File::create(format!(
-        "{}_{}.{}",
+        "{}_{function}.{}",
         path.with_extension("").to_string_lossy(),
-        if function == Function::Amplify {
-            "amplified"
-        } else {
-            "attenuated"
-        },
         path.extension().unwrap().to_string_lossy(),
     ))
     .unwrap();