Always encode floats efficiently.

M-o-a-T · Oct 2, 2024 · 03ccfec · 03ccfec
1 parent 7de2e75
commit 03ccfec
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/python-ecosys/cbor2/cbor2/_encoder.py b/python-ecosys/cbor2/cbor2/_encoder.py
@@ -59,7 +59,15 @@ def encode_float(encoder, value):
     elif math.isinf(value):
         encoder.write(b"\xf9\x7c\x00" if value > 0 else b"\xf9\xfc\x00")
     else:
-        encoder.write(struct.pack(">Bd", 0xFB, value))
+        std = struct.pack(">f", value)
+        if struct.unpack(">f", std)[0] != value:
+            encoder.write(struct.pack(">Bd", 0xFB, value))
+        else:
+            half = struct.pack(">e", value)
+            if struct.unpack(">e", half)[0] != value:
+                encoder.write("\xfa" + std)
+            else:
+                encoder.write("\xf9" + half)
 
 
 def encode_int(encoder, value):

diff --git a/python-ecosys/cbor2/manifest.py b/python-ecosys/cbor2/manifest.py
@@ -1,3 +1,3 @@
-metadata(version="1.1.0", pypi="cbor2")
+metadata(version="1.1.1", pypi="cbor2")
 
 package("cbor2")