Bugfix osi trace (#438)

* add test for OSITrace

Signed-off-by: SergeM <sbmatyunin@gmail.com>

* bugfix in OSITrace

probably a typo

Signed-off-by: SergeM <sbmatyunin@gmail.com>

* Fixed relative import in the test

Signed-off-by: SergeM <sbmatyunin@gmail.com>

* input data for the test is generated on fly

Signed-off-by: SergeM <sbmatyunin@gmail.com>

* install the module on travis in order to get the dependencies

Signed-off-by: SergeM <sbmatyunin@gmail.com>

* Install the module on travis for deploy stage in order to get the dependencies; add python3-wheel to remove warnings

Signed-off-by: SergeM <sbmatyunin@gmail.com>
Signed-off-by: Habedank Clemens <qxs2704@europe.bmw.corp>

Author: serge-m

.travis.yml

@@ -61,6 +61,8 @@ addons:
         - doxygen-gui
         - graphviz
         - python3-setuptools
+        - python3-pip
+        - python3-wheel
 
 
 jobs:
@@ -78,6 +80,8 @@ jobs:
           - cd ..
           - python3 setup.py build
           - python3 setup.py sdist
+          - python3 -m pip install -U pip
+          - python3 -m pip install .
           - cd build
           - cmake --build . --target install
           - cd ..
@@ -103,6 +107,8 @@ jobs:
           - cd ..
           - python3 setup.py build
           - python3 setup.py sdist
+          - python3 -m pip install -U pip
+          - python3 -m pip install .
           - cd build
           - cmake --build . --target install
           - cd ..
@@ -128,6 +134,8 @@ jobs:
           - cd ..
           - python3 setup.py build
           - python3 setup.py sdist
+          - python3 -m pip install -U pip
+          - python3 -m pip install .
           - cd build
           - cmake --build . --target install
           - cd ..

format/OSITrace.py

@@ -236,7 +236,7 @@ def make_readable(self, name, interval=None, index=None):
 
             if interval is None and index is not None:
                 if type(index) == int:
-                    f.write(str(scenario.get_message_by_index(0)))
+                    f.write(str(self.get_message_by_index(0)))
                 else:
                     raise Exception("Argument 'index' needs to be of type 'int'")
 

tests/test_osi_trace.py

@@ -0,0 +1,64 @@
+import os
+import tempfile
+import unittest
+
+from format.OSITrace import OSITrace
+from osi3.osi_sensorview_pb2 import SensorView
+import struct
+
+
+class TestOSITrace(unittest.TestCase):
+    def test_osi_trace(self):
+        with tempfile.TemporaryDirectory() as tmpdirname:
+            path_output = os.path.join(tmpdirname, "output.txth")
+            path_input = os.path.join(tmpdirname, "input.osi")
+            create_sample(path_input)
+
+            trace = OSITrace()
+            trace.from_file(path=path_input)
+            trace.make_readable(path_output, index=1)
+            trace.scenario_file.close()
+
+            self.assertTrue(os.path.exists(path_output))
+
+
+def create_sample(path):
+    f = open(path, "ab")
+    sensorview = SensorView()
+
+    sv_ground_truth = sensorview.global_ground_truth
+    sv_ground_truth.version.version_major = 3
+    sv_ground_truth.version.version_minor = 0
+    sv_ground_truth.version.version_patch = 0
+
+    sv_ground_truth.timestamp.seconds = 0
+    sv_ground_truth.timestamp.nanos = 0
+
+    moving_object = sv_ground_truth.moving_object.add()
+    moving_object.id.value = 114
+
+    # Generate 10 OSI messages for 9 seconds
+    for i in range(10):
+        # Increment the time
+        sv_ground_truth.timestamp.seconds += 1
+        sv_ground_truth.timestamp.nanos += 100000
+
+        moving_object.vehicle_classification.type = 2
+
+        moving_object.base.dimension.length = 5
+        moving_object.base.dimension.width = 2
+        moving_object.base.dimension.height = 1
+
+        moving_object.base.position.x = 0.0 + i
+        moving_object.base.position.y = 0.0
+        moving_object.base.position.z = 0.0
+
+        moving_object.base.orientation.roll = 0.0
+        moving_object.base.orientation.pitch = 0.0
+        moving_object.base.orientation.yaw = 0.0
+
+        """Serialize"""
+        bytes_buffer = sensorview.SerializeToString()
+        f.write(struct.pack("<L", len(bytes_buffer)) + bytes_buffer)
+
+    f.close()