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

Add missing argtypes for lammps_extract_global

Browse Source
This commit is contained in:
Richard Berger
2020-08-28 13:54:06 -04:00
parent 167f12a4a4
commit caeb0af0d1
1 changed files with 1 additions and 1 deletions
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.argtypes = [c_void_p, c_char_p, c_int]
self.lib.lammps_get_last_error_message.restype = 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] 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 # 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 style in (LMP_STYLE_GLOBAL, LMP_STYLE_LOCAL):
if datatype == LMP_TYPE_VECTOR: if datatype == LMP_TYPE_VECTOR:
nrows = self.lmp.extract_compute(cid, style, LMP_SIZE_VECTOR) nrows = self.lmp.extract_compute(cid, style, LMP_SIZE_VECTOR)
print("NROWS", nrows)
return self.darray(value, nrows) return self.darray(value, nrows)
elif datatype == LMP_TYPE_ARRAY: elif datatype == LMP_TYPE_ARRAY:
nrows = self.lmp.extract_compute(cid, style, LMP_SIZE_ROWS) nrows = self.lmp.extract_compute(cid, style, LMP_SIZE_ROWS)