Add options to diff
This commit is contained in:
parent
56c5e5c230
commit
71a897cf58
1 changed files with 49 additions and 13 deletions
|
@ -24,6 +24,20 @@ parser.add_argument(
|
||||||
help="Post-processing directory",
|
help="Post-processing directory",
|
||||||
required=True,
|
required=True,
|
||||||
)
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"-t",
|
||||||
|
"--timestep",
|
||||||
|
type=float,
|
||||||
|
help="Time-step to compare",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"-f",
|
||||||
|
"--func",
|
||||||
|
type=str,
|
||||||
|
help="Post-process function to compare",
|
||||||
|
default="graphUniform",
|
||||||
|
)
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
logging.basicConfig(level=max((10, 20 - 10 * args.verbose)))
|
logging.basicConfig(level=max((10, 20 - 10 * args.verbose)))
|
||||||
|
@ -43,19 +57,41 @@ def get_pickle(out):
|
||||||
|
|
||||||
models = list(map(get_pickle, args.output))
|
models = list(map(get_pickle, args.output))
|
||||||
|
|
||||||
fig, ax = plt.subplots(len(models), constrained_layout=True)
|
if args.timestep is None:
|
||||||
for i, (_ax, _model) in enumerate(zip(ax, models)):
|
fig, ax = plt.subplots(len(models), constrained_layout=True)
|
||||||
_ax.contour(
|
for i, (_ax, _model) in enumerate(zip(ax, models)):
|
||||||
_model.t,
|
_ax.contour(
|
||||||
_model.post_fields["graphUniform"]["x_alpha.water"],
|
_model.t,
|
||||||
_model.post_fields["graphUniform"]["alpha.water"].T,
|
_model.post_fields[args.func]["x_alpha.water"],
|
||||||
(0.5,),
|
_model.post_fields[args.func]["alpha.water"].T,
|
||||||
colors="k",
|
(0.5,),
|
||||||
|
colors="k",
|
||||||
|
)
|
||||||
|
|
||||||
|
_ax.set(xlabel="t (s)", ylabel="z (m)", title=f"Case {i}")
|
||||||
|
_ax.grid()
|
||||||
|
|
||||||
|
fig.savefig(
|
||||||
|
args.output[0].joinpath(
|
||||||
|
f"diff_{args.func}_{'_'.join([o.name for o in args.output])}.pdf"
|
||||||
|
)
|
||||||
)
|
)
|
||||||
|
else:
|
||||||
|
fig, ax = plt.subplots(len(models), constrained_layout=True)
|
||||||
|
for i, (_ax, _model) in enumerate(zip(ax, models)):
|
||||||
|
_ax.tricontour(
|
||||||
|
_model.x,
|
||||||
|
_model.z,
|
||||||
|
_model.fields["alpha.water"][np.where(_model.t == args.timestep)[0]][0],
|
||||||
|
levels=(0.5,),
|
||||||
|
colors="k",
|
||||||
|
)
|
||||||
|
|
||||||
_ax.set(xlabel="t (s)", ylabel="z (m)", title=f"Case {i}")
|
_ax.set(xlabel="x (m)", ylabel="z (m)", title=f"Case {i}")
|
||||||
_ax.grid()
|
_ax.grid()
|
||||||
|
|
||||||
fig.savefig(
|
fig.savefig(
|
||||||
args.output[0].joinpath(f"diff_{'_'.join([o.name for o in args.output])}.pdf")
|
args.output[0].joinpath(
|
||||||
)
|
f"diff_t{args.timestep}_{'_'.join([o.name for o in args.output])}.pdf"
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
Loading…
Reference in a new issue