m2cce/internship
m2cce/internship
1
Fork 0
Code Issues Activity

Add options to diff

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-05-06 11:14:05 +02:00
parent 56c5e5c230
commit 71a897cf58
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
1 changed files with 49 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

62
olaflow/processing/diff.py
View file

@ -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"
)
)