pye57 0.4.17__tar.gz → 0.4.18__tar.gz

{pye57-0.4.17/src/pye57.egg-info → pye57-0.4.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pye57
-Version: 0.4.17
+Version: 0.4.18
 Summary: Python .e57 files reader/writer
 Home-page: https://www.github.com/davidcaron/pye57
 Author: David Caron

pye57-0.4.18/src/pye57/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.4.18"

{pye57-0.4.17 → pye57-0.4.18}/src/pye57/scan_header.py

@@ -28,7 +28,11 @@ class ScanHeader:
 
     @property
     def rotation_matrix(self) -> np.array:
-        q = Quaternion([e.value() for e in self.node["pose"]["rotation"]])
+        try:
+            rotation = self.node["pose"]["rotation"]
+            q = Quaternion([e.value() for e in rotation])
+        except libe57.E57Exception:
+            q = Quaternion()
         return q.rotation_matrix
 
     @property
@@ -36,13 +40,17 @@ class ScanHeader:
         try:
             rotation = self.node["pose"]["rotation"]
             q = Quaternion([e.value() for e in rotation])
-        except KeyError:
+        except libe57.E57Exception:
             q = Quaternion()
         return q.elements
 
     @property
     def translation(self):
-        return np.array([e.value() for e in self.node["pose"]["translation"]])
+        try:
+            translation_values = [e.value() for e in self.node["pose"]["translation"]]
+        except libe57.E57Exception:
+            translation_values = [0] * 3
+        return np.array(translation_values)
 
     def pretty_print(self, node=None, indent=""):
         if node is None:

{pye57-0.4.17 → pye57-0.4.18/src/pye57.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pye57
-Version: 0.4.17
+Version: 0.4.18
 Summary: Python .e57 files reader/writer
 Home-page: https://www.github.com/davidcaron/pye57
 Author: David Caron

{pye57-0.4.17 → pye57-0.4.18}/tests/test_main.py

@@ -54,6 +54,12 @@ def e57_with_normals_path():
     # created using Python's open3d and CloudCompare
     return sample_data("testWithNormals.e57")
 
+@pytest.fixture
+def e57_translation_without_rotation_path():
+    # this e57 was generated using write_scan_raw
+    # however, the line to set the rotation was temporarily commented out
+    return sample_data("translationWithoutRotation.e57")
+
 @pytest.fixture
 def temp_e57_write(request):
     path = sample_data("test_write.e57")
@@ -323,6 +329,20 @@ def test_scan_position(e57_path):
     e57 = pye57.E57(e57_path)
     assert np.allclose(e57.scan_position(3), np.array([[3.01323456e+05, 5.04260184e+06, 1.56040279e+01]]))
 
+def test_translation_without_rotation(e57_translation_without_rotation_path):
+    import numpy as np
+    e57 = pye57.E57(e57_translation_without_rotation_path)
+    header = e57.get_header(0)
+    # translation is defined but not rotation:
+    assert header["pose"].isDefined("translation")
+    assert not header["pose"].isDefined("rotation")
+    # an arbitrary translation was defined in the e57:
+    assert np.array_equal(header.translation, np.array([4,5,6]))
+    # when rotation is not defined, the identity is used:
+    assert np.array_equal(header.rotation, np.array([1,0,0,0]))
+    assert np.array_equal(header.rotation_matrix, np.eye(3))
+    # the scan can be read when translation is defined but not rotation:
+    e57.read_scan(0, ignore_missing_fields=True)
 
 BUFFER_TYPES = {
     libe57.FloatNode: 'd',

pye57-0.4.17/src/pye57/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.4.17"