diff --git a/olaflow/processing/animate.py b/olaflow/processing/animate.py index e1a79a4..60c5519 100644 --- a/olaflow/processing/animate.py +++ b/olaflow/processing/animate.py @@ -61,7 +61,7 @@ AW[:, iz0[idz0], ix0[idx0]] = model.fields["alpha.water"] U = np.full((model.t.size, *X.shape), np.nan) U[:, iz0[idz0], ix0[idx0]] = np.linalg.norm(model.fields["U"], axis=1) -fig = plt.figure(figsize=(19.2, 10.8), dpi=100) +fig = plt.figure() gs = GridSpec(3, 1, figure=fig, height_ratios=[1, 0.05, 0.05]) ax = fig.add_subplot(gs[0]) cax1 = fig.add_subplot(gs[1]) @@ -92,7 +92,7 @@ ax.set(xlabel="x (m)", ylabel="z (m)", aspect="equal", facecolor="#000000") ax.grid(c="k", alpha=0.2) -figU = plt.figure(figsize=(19.2, 10.8), dpi=100) +figU = plt.figure() gsU = GridSpec( 2 if args.max else 3, 1, @@ -138,6 +138,9 @@ if args.max: fig.savefig(out.joinpath("max_aw.pdf")) figU.savefig(out.joinpath("max_U.pdf")) else: + fig.set(figwidth=19.2, figheight=10.8, dpi=100) + figU.set(figwidth=19.2, figheight=10.8, dpi=100) + figU.colorbar(ur_m, label=r"$U_a$", cax=caxu2, shrink=0.6, orientation="horizontal") tit = ax.text(