Unverified Commit 475b7dc4 authored by Axel Kohlmeyer's avatar Axel Kohlmeyer
Browse files

provide access to number of local data rows for fixes

parent 6f1d913e
Loading
Loading
Loading
Loading
+4 −0
Original line number Diff line number Diff line
@@ -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)
+1 −0
Original line number Diff line number Diff line
@@ -621,6 +621,7 @@ void *lammps_extract_fix(void *ptr, char *id, int style, int type,

    if (style == 2) {
      if (!fix->local_flag) return NULL;
      if (type == 0) return (void *) &fix->size_local_rows;
      if (type == 1) return (void *) fix->vector_local;
      if (type == 2) return (void *) fix->array_local;
    }