Update remaining variable names after refactor

This commit is contained in:
Richard Berger
2021-05-11 21:22:29 -04:00
committed by Axel Kohlmeyer
parent 4ccb17ab62
commit 0ee7bc6e01

View File

@ -911,14 +911,14 @@ class lammps(object):
else: return None else: return None
if ctype == LMP_TYPE_SCALAR: if ctype == LMP_TYPE_SCALAR:
if style == LMP_STYLE_GLOBAL: if cstyle == LMP_STYLE_GLOBAL:
self.lib.lammps_extract_compute.restype = POINTER(c_double) self.lib.lammps_extract_compute.restype = POINTER(c_double)
with ExceptionCheck(self): with ExceptionCheck(self):
ptr = self.lib.lammps_extract_compute(self.lmp,cid,cstyle,ctype) ptr = self.lib.lammps_extract_compute(self.lmp,cid,cstyle,ctype)
return ptr[0] return ptr[0]
elif style == LMP_STYLE_ATOM: elif cstyle == LMP_STYLE_ATOM:
return None return None
elif style == LMP_STYLE_LOCAL: elif cstyle == LMP_STYLE_LOCAL:
self.lib.lammps_extract_compute.restype = POINTER(c_int) self.lib.lammps_extract_compute.restype = POINTER(c_int)
with ExceptionCheck(self): with ExceptionCheck(self):
ptr = self.lib.lammps_extract_compute(self.lmp,cid,cstyle,ctype) ptr = self.lib.lammps_extract_compute(self.lmp,cid,cstyle,ctype)