Move version check into cython

cta-observatory · Aug 31, 2023 · 93ba61a · 93ba61a
1 parent cd998c7
commit 93ba61a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/eventio/simtel/objects.py b/src/eventio/simtel/objects.py
@@ -1413,10 +1413,8 @@ def __str__(self):
         )
 
     def parse(self):
-        ''' '''
-        assert_version_in(self, (1, 2))
+        ''''''
         self.seek(0)
-
         d = MCEvent.parse_mc_event(self.read(), self.header.version)
         d['event_id'] = self.header.id
         return d

diff --git a/src/eventio/simtel/parsing.pyx b/src/eventio/simtel/parsing.pyx
@@ -54,6 +54,12 @@ cpdef dict parse_mc_event(
     const uint8_t[:] data,
     uint32_t version
 ):
+    if version > 2:
+        raise NotImplementedError(
+            'Unsupported version of MCEvent:'
+            ' only versions up to 2 supported,'
+            f' got: {version} '
+        )
 
     cdef uint64_t pos = 0
     cdef float xcore, ycore, aweight