[sinks] update sinks reader for ramses-tine

This commit is contained in:
Noe Brucy
2021-11-02 15:49:52 +01:00
parent 7af9ae020a
commit d51be7504b
+14 -4
View File
@@ -1354,10 +1354,21 @@ class SnapshotProcessor(HDF5Container):
return alpha_g
def _sinks(self):
csv_name = f"{self.path}/output_{self.num:05}/sink_{self.num:05}.csv"
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",
"M",
"dmf",
"id",
"msink",
"dmfsink",
"x",
"y",
"z",
@@ -1374,7 +1385,6 @@ class SnapshotProcessor(HDF5Container):
"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)
return {key: df[key].values for key in df}