def _parse_parameter_file(self):
"""
Parses the parameter file and establishes the various
dictionaries.
"""
self.unit_base = {}
# Let's read the file
lines = open(self.parameter_filename).readlines()
comments = ["%", ";"]
for line in (l.strip() for l in lines):
for comment in comments:
if comment in line: line = line[0:line.find(comment)]
if len(line) < 2: continue
param, vals = (i.strip() for i in line.split(None, 1))
# First we try to decipher what type of value it is.
vals = vals.split()
# Special case approaching.
if "(do" in vals: vals = vals[:1]
if len(vals) == 0:
pcast = str # Assume NULL output
else:
v = vals[0]
# Figure out if it's castable to floating point:
try:
float(v)
except ValueError:
pcast = str
else:
if any("." in v or "e" in v for v in vals):
pcast = float
elif v == "inf":
pcast = str
else:
pcast = int
# Now we figure out what to do with it.
if param.startswith("Unit"):
self.unit_base[param] = float(vals[0])
if len(vals) == 0:
vals = ""
elif len(vals) == 1:
vals = pcast(vals[0])
else:
vals = np.array([pcast(i) for i in vals])
self.parameters[param] = vals
# Domain dimensions for Gadget datasets are always 2x2x2 for octree
self.domain_dimensions = np.array([2,2,2])
if self.parameters["ComovingIntegrationOn"]:
cosmo_attr = {"box_size": "BoxSize",
"omega_lambda": "OmegaLambda",
"omega_matter": "Omega0",
"hubble_constant": "HubbleParam"}
self.initial_redshift = 1.0 / self.parameters["TimeBegin"] - 1.0
self.final_redshift = 1.0 / self.parameters["TimeMax"] - 1.0
self.cosmological_simulation = 1
for a, v in cosmo_attr.items():
if v not in self.parameters:
raise MissingParameter(self.parameter_filename, v)
setattr(self, a, self.parameters[v])
self.domain_left_edge = np.array([0., 0., 0.])
self.domain_right_edge = np.array([1., 1., 1.]) * self.parameters['BoxSize']
else:
self.cosmological_simulation = 0
self.omega_lambda = self.omega_matter = \
self.hubble_constant = 0.0
def _parse_parameter_file(self):
"""
Parses the parameter file and establishes the various
dictionaries.
"""
self.conversion_factors = {}
redshift_outputs = []
# Let's read the file
lines = open(self.parameter_filename).readlines()
for line in (l.strip() for l in lines):
if "#" in line:
line = line[0 : line.find("#")]
if "//" in line:
line = line[0 : line.find("//")]
if len(line) < 2:
continue
param, vals = (i.strip() for i in line.split("=", 1))
# First we try to decipher what type of value it is.
vals = vals.split()
# Special case approaching.
if "(do" in vals:
vals = vals[:1]
if len(vals) == 0:
pcast = str # Assume NULL output
else:
v = vals[0]
# Figure out if it's castable to floating point:
try:
float(v)
except ValueError:
pcast = str
else:
if any("." in v or "e" in v for v in vals):
pcast = float
elif v == "inf":
pcast = str
else:
pcast = int
# Now we figure out what to do with it.
if param.endswith("Units") and not param.startswith("Temperature"):
dataType = param[:-5]
# This one better be a float.
self.conversion_factors[dataType] = float(vals[0])
if param.startswith("CosmologyOutputRedshift["):
index = param[param.find("[") + 1 : param.find("]")]
redshift_outputs.append(
{"index": int(index), "redshift": float(vals[0])}
)
elif len(vals) == 0:
vals = ""
elif len(vals) == 1:
vals = pcast(vals[0])
else:
vals = np.array([pcast(i) for i in vals if i != "-99999"])
self.parameters[param] = vals
self.refine_by = self.parameters["RefineBy"]
self.dimensionality = self.parameters["TopGridRank"]
if self.dimensionality > 1:
self.domain_dimensions = self.parameters["TopGridDimensions"]
if len(self.domain_dimensions) < 3:
tmp = self.domain_dimensions.tolist()
tmp.append(1)
self.domain_dimensions = np.array(tmp)
self.domain_left_edge = np.array(
self.parameters["DomainLeftEdge"], "float64"
).copy()
self.domain_right_edge = np.array(
self.parameters["DomainRightEdge"], "float64"
).copy()
else:
self.domain_left_edge = np.array(
self.parameters["DomainLeftEdge"], "float64"
)
self.domain_right_edge = np.array(
self.parameters["DomainRightEdge"], "float64"
)
self.domain_dimensions = np.array(
[self.parameters["TopGridDimensions"], 1, 1]
)
if self.parameters["ComovingCoordinates"]:
cosmo_attr = {
"box_size": "CosmologyComovingBoxSize",
"omega_lambda": "CosmologyOmegaLambdaNow",
"omega_matter": "CosmologyOmegaMatterNow",
"omega_radiation": "CosmologyOmegaRadiationNow",
"hubble_constant": "CosmologyHubbleConstantNow",
"initial_redshift": "CosmologyInitialRedshift",
"final_redshift": "CosmologyFinalRedshift",
}
self.cosmological_simulation = 1
for a, v in cosmo_attr.items():
if v not in self.parameters:
raise MissingParameter(self.parameter_filename, v)
setattr(self, a, self.parameters[v])
else:
self.cosmological_simulation = 0
self.omega_lambda = self.omega_matter = self.hubble_constant = 0.0
# make list of redshift outputs
self.all_redshift_outputs = []
if not self.cosmological_simulation:
return
for output in redshift_outputs:
output["filename"] = os.path.join(
self.parameters["GlobalDir"],
"%s%04d" % (self.parameters["RedshiftDumpDir"], output["index"]),
"%s%04d" % (self.parameters["RedshiftDumpName"], output["index"]),
)
del output["index"]
self.all_redshift_outputs = redshift_outputs