ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Return None in case of null pointer

Browse Source
This commit is contained in:
Richard Berger
2021-09-10 14:22:08 -04:00
parent e10d89d8c4
commit 7bb863a46c
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/lammps/numpy_wrapper.py
View File

@ -386,6 +386,9 @@ class numpy_wrapper:
# ------------------------------------------------------------------------- # -------------------------------------------------------------------------
def iarray(self, c_int_type, raw_ptr, nelem, dim=1): def iarray(self, c_int_type, raw_ptr, nelem, dim=1):
if raw_ptr is None:
return None
import numpy as np import numpy as np
np_int_type = self._ctype_to_numpy_int(c_int_type) 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): def darray(self, raw_ptr, nelem, dim=1):
if raw_ptr is None:
return None
import numpy as np import numpy as np
if dim == 1: if dim == 1:
ptr = cast(raw_ptr, POINTER(c_double * nelem)) ptr = cast(raw_ptr, POINTER(c_double * nelem))
else: else: