jianmu
/
lammps
forked from lijiext/lammps
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add missing argtypes for lammps_extract_global

This commit is contained in:
Richard Berger 2020-08-28 13:54:06 -04:00
parent 167f12a4a4
commit caeb0af0d1
No known key found for this signature in database
GPG Key ID: A9E83994E0BA0CAB
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/lammps.py
View File

@ -281,6 +281,7 @@ class lammps(object):
self.lib.lammps_get_last_error_message.argtypes = [c_void_p, c_char_p, c_int]
self.lib.lammps_get_last_error_message.restype = c_int
self.lib.lammps_extract_global.argtypes = [c_void_p, c_char_p]
self.lib.lammps_extract_compute.argtypes = [c_void_p, c_char_p, c_int, c_int]
# detect if Python is using version of mpi4py that can pass a communicator
@ -443,7 +444,6 @@ class lammps(object):
if style in (LMP_STYLE_GLOBAL, LMP_STYLE_LOCAL):
if datatype == LMP_TYPE_VECTOR:
nrows = self.lmp.extract_compute(cid, style, LMP_SIZE_VECTOR)
print("NROWS", nrows)
return self.darray(value, nrows)
elif datatype == LMP_TYPE_ARRAY:
nrows = self.lmp.extract_compute(cid, style, LMP_SIZE_ROWS)