From be6d0b39b0b4ffa7381e7a2ae8d6cd0091efed0e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=27=C3=A9lectron=20rare?=
 <108685187+electron-rare@users.noreply.github.com>
Date: Wed, 13 May 2026 13:14:04 +0200
Subject: [PATCH] perf(smplx): numpy tobytes serialization
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Replace 10475-iteration struct.pack loop with
np.ascontiguousarray(...).tobytes() — one C call per field.
SMPLXPerson fields now typed as np.ndarray; worker stops
converting to Python tuples. Wire format byte-identical.
---
 data_only_viz/smplx_osc_sender.py | 29 +++++++++++++++++++----------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/data_only_viz/smplx_osc_sender.py b/data_only_viz/smplx_osc_sender.py
index 68a9be4..86070c5 100644
--- a/data_only_viz/smplx_osc_sender.py
+++ b/data_only_viz/smplx_osc_sender.py
@@ -76,6 +76,16 @@ class SMPLXTCPSender:
                 pass
             self._sock = None
 
+    @staticmethod
+    def _pad10(arr: "np.ndarray") -> "np.ndarray":
+        """Return a contiguous float32 array of length 10, zero-padded if shorter."""
+        flat = np.ascontiguousarray(arr, dtype="<f4").ravel()
+        if len(flat) >= 10:
+            return flat[:10]
+        out = np.zeros(10, dtype="<f4")
+        out[:len(flat)] = flat
+        return out
+
     @staticmethod
     def _serialize_persons(persons: Sequence[SMPLXPerson]) -> bytes:
         buf = bytearray()
@@ -84,16 +94,15 @@ class SMPLXTCPSender:
         for p in persons:
             buf += struct.pack("<i", p.pid)
             buf += struct.pack("<f", float(p.confidence))
-            tx, ty, tz = (p.translation + (0.0, 0.0, 0.0))[:3]
-            buf += struct.pack("<fff", float(tx), float(ty), float(tz))
-            betas = list(p.betas[:10]) + [0.0] * max(0, 10 - len(p.betas))
-            for b in betas[:10]:
-                buf += struct.pack("<f", float(b))
-            expr = list(p.expression[:10]) + [0.0] * max(0, 10 - len(p.expression))
-            for e in expr[:10]:
-                buf += struct.pack("<f", float(e))
-            for vx, vy, vz in p.vertices_3d:
-                buf += struct.pack("<fff", float(vx), float(vy), float(vz))
+            trans = np.ascontiguousarray(p.translation, dtype="<f4").ravel()
+            if len(trans) < 3:
+                t3 = np.zeros(3, dtype="<f4")
+                t3[:len(trans)] = trans
+                trans = t3
+            buf += trans[:3].tobytes()
+            buf += SMPLXTCPSender._pad10(p.betas).tobytes()
+            buf += SMPLXTCPSender._pad10(p.expression).tobytes()
+            buf += np.ascontiguousarray(p.vertices_3d, dtype="<f4").tobytes()
         return bytes(buf)
 
     def _run(self) -> None: