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/src/library.cpp
+++ b/src/library.cpp
@ -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;
    }