m2cce/internship
m2cce/internship
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Processing update to reduce memory usage

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-03-15 10:11:21 +01:00
parent 45d07f8de0
commit bb9dfe8603
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
2 changed files with 23 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

21
swash/processing/sws_npz.py
View file

@ -27,15 +27,14 @@ rsws = ReadSwash()
rsws.read_time(sws_out.joinpath("tsec.dat"))
rsws.read_x(sws_out.joinpath("xp.dat"))
log.info("Reading 'dep'")
rsws.read_scalar(sws_out.joinpath("dep.dat"))
log.info("Reading 'botl'")
rsws.read_scalar(sws_out.joinpath("botl.dat"), const=True)
log.info("Reading 'watl'")
rsws.read_scalar(sws_out.joinpath("watl.dat"))
log.info("Reading 'vel'")
rsws.read_vector(sws_out.joinpath("vel.dat"))
log.info(f"Writing npz file in '{inp}'")
inp.mkdir(exist_ok=True)
np.savez_compressed(inp.joinpath("sws"), t=rsws.t, x=rsws.x, **rsws.data)
log.info(f"Wrinting output in '{inp}'")
log.info("Reading 'dep'")
np.save(inp.joinpath("dep"), rsws.read_scalar(sws_out.joinpath("dep.dat")))
log.info("Reading 'botl'")
np.save(inp.joinpath("botl"), rsws.read_scalar(sws_out.joinpath("botl.dat"),
const=True))
log.info("Reading 'watl'")
np.save(inp.joinpath("watl"), rsws.read_scalar(sws_out.joinpath("watl.dat")))
log.info("Reading 'vel'")
np.save(inp.joinpath("vel"), rsws.read_vector(sws_out.joinpath("vel.dat")))