guilhem_lavaux/cosmotool
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rearrange cleanup for NFS stale handles

Browse Source
This commit is contained in:
Guilhem Lavaux 2018-01-05 22:17:26 +01:00
parent 531084dbc1
commit 945166c1f9
1 changed files with 31 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
python/cosmotool/smooth.py
View File

@ -11,7 +11,7 @@ def smooth_particle_density(
velocities=None, velocities=None,
radius=1e6, radius=1e6,
boxsize=None, boxsize=None,
resolution=128,i resolution=128,
center=None, tmpprefix=None ): center=None, tmpprefix=None ):
"""Use adaptive smoothing to produce density and momentum fields. """Use adaptive smoothing to produce density and momentum fields.
The algorithm is originally described in [1]. The algorithm is originally described in [1].
@ -68,8 +68,8 @@ def smooth_particle_density(
raise ValueError("Need a boxsize") raise ValueError("Need a boxsize")
cx,cy,cz=center cx,cy,cz=center
with TemporaryDirectory(prefix=tmpprefix) as tmpdir: tmpdir = TemporaryDirectory(prefix=tmpprefix)
h5_file = os.path.join(tmpdir, 'particles.h5') h5_file = os.path.join(tmpdir.name, 'particles.h5')
with h5.File(h5_file, mode="w") as f: with h5.File(h5_file, mode="w") as f:
data = f.create_dataset('particles', shape=(position.shape[0],7), dtype=np.float32) data = f.create_dataset('particles', shape=(position.shape[0],7), dtype=np.float32)
data[:,:3] = position[:,:3] data[:,:3] = position[:,:3]
@ -88,16 +88,18 @@ def smooth_particle_density(
str(boxsize), str(boxsize),
str(resolution), str(resolution),
str(cx), str(cy), str(cz) str(cx), str(cy), str(cz)
], cwd=tmpdir) ], cwd=tmpdir.name)
f0 = h5.File(os.path.join(tmpdir,'fields.h5'), mode="r") f0 = h5.File(os.path.join(tmpdir.name,'fields.h5'), mode="r")
def cleanup_f0(): def cleanup_f0():
f0.close() f0.close()
tmpdir.cleanup()
class Dict(dict): class Dict(dict):
pass pass
t = Dict(rho=f0['density'], p=[f0['p0'], f0['p1'], f0['p2']]) t = Dict(rho=f0['density'], p=[f0['p0'], f0['p1'], f0['p2']])
t._tmpdir_=tmpdir
weakref.finalize(t, cleanup_f0) weakref.finalize(t, cleanup_f0)
return t return t