From 7bb863a46ca2663b70f4f4f1531b54a89f4195bd Mon Sep 17 00:00:00 2001 From: Richard Berger Date: Fri, 10 Sep 2021 14:22:08 -0400 Subject: [PATCH] Return None in case of null pointer --- python/lammps/numpy_wrapper.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/python/lammps/numpy_wrapper.py b/python/lammps/numpy_wrapper.py index b29503a271..e883b6e5c5 100644 --- a/python/lammps/numpy_wrapper.py +++ b/python/lammps/numpy_wrapper.py @@ -386,6 +386,9 @@ class numpy_wrapper: # ------------------------------------------------------------------------- def iarray(self, c_int_type, raw_ptr, nelem, dim=1): + if raw_ptr is None: + return None + import numpy as np np_int_type = self._ctype_to_numpy_int(c_int_type) @@ -405,7 +408,11 @@ class numpy_wrapper: # ------------------------------------------------------------------------- def darray(self, raw_ptr, nelem, dim=1): + if raw_ptr is None: + return None + import numpy as np + if dim == 1: ptr = cast(raw_ptr, POINTER(c_double * nelem)) else: