noe/pipeline
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[galaxy] load only needed data

Browse Source
This commit is contained in:
Noe Brucy
2021-06-15 10:16:23 +02:00
parent b334c89f61
commit 13929cf99d
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
galaxy.py
+3 -5
View File
@@ -6,8 +6,8 @@ from plotter import U
def get_gas_dm_stars(pp): def get_gas_dm_stars(pp):
# Load arrays # Load arrays
pp.load_parts() pp.load_parts(keys=["pos", "vel", "mass", "epoch"])
pp.load_cells() pp.load_cells(keys=["pos", "vel", "dx", "rho"])
cells = pp.cells cells = pp.cells
parts = pp.parts parts = pp.parts
@@ -81,9 +81,7 @@ def get_sfr(pp, stars):
mass = stars["mass"].copy() mass = stars["mass"].copy()
mass *= pp.info["unit_mass"].express(U.Msun) mass *= pp.info["unit_mass"].express(U.Msun)
mask = epoch > 0 mask = epoch > 0
masstot, time = np.histogram( masstot, time = np.histogram(epoch[mask], weights=mass[mask], bins=200)
epoch[mask], weights=mass[mask], bins=200
)
dtime = np.diff(time) dtime = np.diff(time)
sfr = masstot[-1] / dtime[-1] sfr = masstot[-1] / dtime[-1]
except KeyError: except KeyError: