Support JSON-encoding of NDArrays

src/appose/types.py

@@ -35,7 +35,7 @@
 
 
 def encode(data: Args) -> str:
-    return json.dumps(data, separators=(",", ":"))
+    return json.dumps(data, cls=_ApposeJSONEncoder, separators=(",", ":"))
 
 
 def decode(the_json: str) -> Args:
@@ -88,6 +88,24 @@ def ndarray(self):
             raise ImportError("NumPy is not available.")
 
 
+class _ApposeJSONEncoder(json.JSONEncoder):
+    def default(self, obj):
+        if isinstance(obj, SharedMemory):
+            return {
+                "appose_type": "shm",
+                "name": obj.name,
+                "size": obj.size,
+            }
+        if isinstance(obj, NDArray):
+            return {
+                "appose_type": "ndarray",
+                "dtype": obj.dtype,
+                "shape": obj.shape,
+                "shm": obj.shm,
+            }
+        return super().default(obj)
+
+
 def _appose_object_hook(obj: Dict):
     atype = obj.get("appose_type")
     if atype == "shm":