fix: add writer_info

Signed-off-by: Henry Schreiner <henryschreineriii@gmail.com>

Author: henryiii

src/boost_histogram/serialization/_axis.py

@@ -42,6 +42,8 @@ def _(ax: axis.Regular | axis.Integer, /) -> dict[str, Any]:
             "overflow": ax.traits.overflow,
             "circular": ax.traits.circular,
         }
+    if isinstance(ax, axis.Integer):
+        data["writer_info"] = {"boost-histogram": {"orig_type": "Integer"}}
     if ax.metadata is not None:
         data["metadata"] = ax.metadata
 

src/boost_histogram/serialization/_storage.py

@@ -26,11 +26,12 @@ def _storage_to_dict(_storage: Any, /, data: Any) -> dict[str, Any]:  # noqa: AR
 @_storage_to_dict.register(storage.Int64)
 @_storage_to_dict.register(storage.Unlimited)
 def _(
-    _storage: storage.AtomicInt64 | storage.Double | storage.Int64 | storage.Unlimited,
+    storage_: storage.AtomicInt64 | storage.Double | storage.Int64 | storage.Unlimited,
     /,
     data: Any,
 ) -> dict[str, Any]:
     return {
+        "writer_info": {"boost-histogram": {"orig_type": type(storage_).__name__}},
         "type": "int" if np.issubdtype(data.dtype, np.integer) else "double",
         "values": data,
     }

src/boost_histogram/serialization/generic.py

@@ -2,7 +2,7 @@
 
 from typing import Any
 
-from .. import Histogram
+from .. import Histogram, __version__
 from ._axis import _axis_from_dict, _axis_to_dict
 from ._storage import _data_from_dict, _storage_from_dict, _storage_to_dict
 
@@ -18,6 +18,7 @@ def to_dict(h: Histogram, /) -> dict[str, Any]:
 
     # Convert the histogram to a dictionary
     data = {
+        "writer_info": {"boost-histogram": {"version": __version__}},
         "axes": [_axis_to_dict(axis) for axis in h.axes],
         "storage": _storage_to_dict(h.storage_type(), h.view(flow=True)),
     }