provide access to number of local data rows for fixes

2020-03-07 10:58:35 -05:00
parent 6f1d913e7e
commit 475b7dc4f4
2 changed files with 5 additions and 0 deletions
--- a/python/lammps.py
+++ b/python/lammps.py
@ -510,6 +510,10 @@ class lammps(object):
      result = ptr[0]
      self.lib.lammps_free(ptr)
      return result
+    elif (style == 2) and (type == 0):
+      self.lib.lammps_extract_fix.restype = POINTER(c_int)
+      ptr = self.lib.lammps_extract_fix(self.lmp,id,style,type,i,j)
+      return ptr[0]
    elif (style == 1) or (style == 2):
      if type == 1:
        self.lib.lammps_extract_fix.restype = POINTER(c_double)