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

[sinks] update sinks reader for ramses-tine

Browse Source
This commit is contained in:
Noe Brucy
2021-11-02 15:49:52 +01:00
parent 7af9ae020a
commit d51be7504b
1 changed files with 31 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
snapshotprocessor.py
+31 -21
View File
@@ -1354,28 +1354,38 @@ class SnapshotProcessor(HDF5Container):
return alpha_g
def _sinks(self):
header = [
"Id",
"M",
"dmf",
"x",
"y",
"z",
"vx",
"vy",
"vz",
"rot_period",
"lx",
"ly",
"lz",
"acc_rate",
"acc_lum",
"age",
"int_lum",
"Teff",
]
csv_name = f"{self.path}/output_{self.num:05}/sink_{self.num:05}.csv"
df = pd.read_csv(csv_name, header=None, names=header)
f = open(csv_name)
first_line = f.readline()
f.close()
if first_line[1] == "#":
header = first_line[3:-1].split(",")
df = pd.read_csv(csv_name, header=None, names=header, skiprows=2)
else:
header = [
"id",
"msink",
"dmfsink",
"x",
"y",
"z",
"vx",
"vy",
"vz",
"rot_period",
"lx",
"ly",
"lz",
"acc_rate",
"acc_lum",
"age",
"int_lum",
"Teff",
]
df = pd.read_csv(csv_name, header=None, names=header)
return {key: df[key].values for key in df}
def _pspec(self, **kwargs):