Add Dmap description method.

1 files changed, 13 insertions, 2 deletions

diff --git a/pyecsca/sca/re/tree.py b/pyecsca/sca/re/tree.py
index f19efac..2ae7945 100644
--- a/pyecsca/sca/re/tree.py
+++ b/pyecsca/sca/re/tree.py
@@ -192,6 +192,17 @@ class Map:
         # Finally, adjust the codomain
         self.codomain.update(other.codomain)
 
+    def describe(self) -> str:
+        return "\n".join(
+            (
+                "Total configs: {len(self.cfg_map)}, ({self.cfg_map.memory_usage(index=True).sum():_} bytes)",
+                "Rows: {len(self.mapping)}, ({self.mapping.memory_usage(index=True).sum():_} bytes)",
+                "Inputs: {len(self.domain)}",
+                "Codomain: {len(self.codomain)}",
+                "None in codomain: {None in self.codomain}"
+            )
+        )
+
 
 @public
 class Node(NodeMixin):
@@ -278,8 +289,8 @@ class Tree:
                 f"Size: {self.size}",
                 f"Leaves: {len(leaf_sizes)}",
                 f"Leaf sizes: {sorted(leaf_sizes)}",
-                f"Average leaf size: {np.mean(leaf_sizes):.3}",
-                f"Mean result size: {np.mean(leafs):.3}",
+                f"Average leaf size: {np.mean(leaf_sizes):.3f}",
+                f"Mean result size: {np.mean(leafs):.3f}",
             )
         )