fix scale in the quiver plot

caiostringari · Jul 14, 2021 · 2f8272f · 2f8272f
1 parent a3c5c84
commit 2f8272f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/doc/flow.png b/doc/flow.png
diff --git a/src/post/plot_averaged_flow.py b/src/post/plot_averaged_flow.py
@@ -56,6 +56,13 @@
                         required=False,
                         help="Maximun speed value.")
 
+    parser.add_argument("--scale", "-qs",
+                        action="store",
+                        dest="scale",
+                        default=100,
+                        required=False,
+                        help="Scale for the arrows.")
+
     parser.add_argument("--output", "-o",
                         action="store",
                         dest="output",
@@ -74,6 +81,7 @@
 
     CUT = float(args.cut)
     step = int(args.step)
+    scale = float(args.scale)
 
     # try to read the average image
     try:
@@ -126,7 +134,7 @@
                   grid_y[::step, ::step],
                   u_mean[::step, ::step],
                   v_mean[::step, ::step], mag[::step, ::step],
-                  cmap="viridis", norm=norm)
+                  cmap="viridis", norm=norm, scale=scale)
 
     if has_avg:
         extent = [grid_x.min(), grid_x.max(), grid_y.min(), grid_y.max()]