QuantConfig implements display

Author: JordanMaples

diskann-benchmark/src/inputs/bftree.rs

@@ -293,6 +293,36 @@ impl QuantConfig {
     }
 }
 
+impl std::fmt::Display for QuantConfig {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            Self::Spherical {
+                seed,
+                transform_kind,
+                num_bits,
+                pre_scale,
+                quant_store_config,
+            } => {
+                write_field!(f, "quantization", "spherical")?;
+                write_field!(f, "num_bits", num_bits)?;
+                write_field!(f, "seed", seed)?;
+                write_field!(f, "transform_kind", transform_kind)?;
+                if let Some(pre_scale) = pre_scale {
+                    write_field!(f, "pre_scale", pre_scale)?;
+                }
+                if let Some(cfg) = quant_store_config {
+                    writeln!(f, "\n  Quant Store:")?;
+                    write!(f, "{}", cfg)?;
+                }
+            }
+            Self::None => {
+                write_field!(f, "quantization", "none")?;
+            }
+        }
+        Ok(())
+    }
+}
+
 #[derive(Debug, Serialize, Deserialize)]
 pub(crate) struct BfTreeBuild {
     build: IndexBuild,
@@ -388,20 +418,7 @@ impl std::fmt::Display for BfTreeBuild {
         }
         write_field!(f, "tag", Self::tag())?;
 
-        if let QuantConfig::Spherical {
-            seed,
-            transform_kind,
-            num_bits,
-            ..
-        } = &self.quantization
-        {
-            write_field!(f, "quantization", "spherical")?;
-            write_field!(f, "num_bits", num_bits)?;
-            write_field!(f, "seed", seed)?;
-            write_field!(f, "transform_kind", transform_kind)?;
-        } else {
-            write_field!(f, "quantization", "none")?;
-        }
+        write!(f, "{}", self.quantization)?;
 
         writeln!(f)?;
         self.build.summarize_fields(f)?;
@@ -414,14 +431,6 @@ impl std::fmt::Display for BfTreeBuild {
             writeln!(f, "\n  Neighbor Store:")?;
             write!(f, "{}", cfg)?;
         }
-        if let QuantConfig::Spherical {
-            quant_store_config: Some(ref cfg),
-            ..
-        } = self.quantization
-        {
-            writeln!(f, "\n  Quant Store:")?;
-            write!(f, "{}", cfg)?;
-        }
 
         Ok(())
     }
@@ -530,20 +539,7 @@ impl std::fmt::Display for BfTreeStreamingRun {
         }
         write_field!(f, "tag", Self::tag())?;
 
-        if let QuantConfig::Spherical {
-            seed,
-            transform_kind,
-            num_bits,
-            ..
-        } = &self.quantization
-        {
-            write_field!(f, "quantization", "spherical")?;
-            write_field!(f, "num_bits", num_bits)?;
-            write_field!(f, "seed", seed)?;
-            write_field!(f, "transform_kind", transform_kind)?;
-        } else {
-            write_field!(f, "quantization", "none")?;
-        }
+        write!(f, "{}", self.quantization)?;
 
         writeln!(f)?;
         self.build.summarize_fields(f)?;
@@ -556,14 +552,6 @@ impl std::fmt::Display for BfTreeStreamingRun {
             writeln!(f, "\n  Neighbor Store:")?;
             write!(f, "{}", cfg)?;
         }
-        if let QuantConfig::Spherical {
-            quant_store_config: Some(ref cfg),
-            ..
-        } = self.quantization
-        {
-            writeln!(f, "\n  Quant Store:")?;
-            write!(f, "{}", cfg)?;
-        }
 
         Ok(())
     }