Merge pull request #109 from matt-frey/add-plot-parcel-number

add plot_parcel_number plot

python-scripts/plot-diagnostics.py

@@ -4,7 +4,8 @@
     plot_rms_volume_error,  \
     plot_max_volume_error,  \
     plot_aspect_ratio,      \
-    plot_parcel_volume
+    plot_parcel_volume,     \
+    plot_parcel_number
 import os
 import sys
 
@@ -19,7 +20,8 @@
         'rms-volume-error',
         'max-volume-error',
         'aspect-ratio',
-        'parcel-volume'
+        'parcel-volume',
+        'parcel-number'
     ]
 
 
@@ -66,6 +68,9 @@
     elif args.kind == kinds[3]:
         for fname in args.filenames:
             plot_parcel_volume(fname, show=args.show, fmt=args.fmt)
+    elif args.kind == kinds[4]:
+        for fname in args.filenames:
+            plot_parcel_number(fname, show=args.show, fmt=args.fmt)
     else:
         raise ValuError("Plot '" + args.kind + "' not supported!")
 

python-scripts/tools/plots.py

@@ -321,6 +321,38 @@ def plot_aspect_ratio(fname, show=False, fmt="png"):
     plt.close()
 
 
+def plot_parcel_number(fname, show=False, fmt="png"):
+    """
+    Plot the number of parcels in simulation.
+    """
+    h5reader = H5Reader()
+    h5reader.open(fname)
+
+    nsteps = h5reader.get_num_steps()
+
+    nparcels = np.zeros(nsteps)
+
+    for step in range(nsteps):
+        nparcels[step] = h5reader.get_num_parcels(step)
+
+    h5reader.close()
+
+    plt.figure()
+    plt.plot(nparcels, color='blue')
+    plt.grid(linestyle='dashed', zorder=-1)
+
+    plt.xlabel(r'number of iterations')
+    plt.ylabel(r'parcel count')
+    plt.tight_layout()
+
+    if show:
+        plt.show()
+    else:
+        prefix = os.path.splitext(fname)[0]
+        plt.savefig(prefix + '_parcel_number_profile.' + fmt, bbox_inches='tight')
+    plt.close()
+
+
 def plot_parcel_volume(fname, show=False, fmt="png"):
     """
     Plot the mean and standard deviation of the parcel volume