2019-02-12 18:56:05 +01:00
|
|
|
class Simulation:
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
self.runid = None # type:int
|
|
|
|
self.vcode = None
|
|
|
|
self.alphacode = None
|
|
|
|
self.mcode = None
|
|
|
|
self.gammacode = None
|
|
|
|
self.wtcode = None
|
|
|
|
self.wpcode = None
|
|
|
|
self.v = None # v/v_esc
|
|
|
|
self.alpha = None # impact angle
|
|
|
|
self.total_mass = None # m
|
|
|
|
self.projectile_mass = None # mp
|
|
|
|
self.target_mass = None # mt
|
|
|
|
self.projectile_water_fraction = None # wp
|
|
|
|
self.target_water_fraction = None # wt
|
|
|
|
self.largest_aggregate_mass = None # mS1
|
|
|
|
self.largest_aggregate_water_fraction = None # wmfS1
|
|
|
|
self.second_largest_aggregate_mass = None # mS2
|
|
|
|
self.second_largest_aggregate_water_fraction = None # wmfS2
|
|
|
|
self.rel_velocity = None # vrel
|
|
|
|
self.rel_velocity_per_esc_velocity = None # vrel_over_vesc
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
def from_dict(cls, data: dict):
|
|
|
|
sim = cls()
|
|
|
|
for key in data:
|
|
|
|
setattr(sim, key, data[key])
|
|
|
|
return sim
|
|
|
|
|
|
|
|
@property
|
|
|
|
def gamma(self) -> float:
|
|
|
|
return self.projectile_mass / self.target_mass
|
|
|
|
|
|
|
|
@property
|
|
|
|
def relative_projectile_mass(self) -> float:
|
|
|
|
return self.projectile_mass / self.total_mass
|
|
|
|
|
|
|
|
@property
|
|
|
|
def relative_target_mass(self) -> float:
|
|
|
|
return self.target_mass / self.total_mass
|
|
|
|
|
|
|
|
@property
|
|
|
|
def largest_aggregate_relative_mass(self) -> float:
|
|
|
|
"""
|
|
|
|
p['mS1_over_mt'] = p['mS1'] / p['mt']
|
|
|
|
"""
|
|
|
|
return self.largest_aggregate_mass / self.target_mass
|
|
|
|
|
|
|
|
@property
|
|
|
|
def second_largest_aggregate_relative_mass(self) -> float:
|
|
|
|
"""
|
|
|
|
p['mS2_over_mp'] = p['mS2'] / p['mp']
|
|
|
|
"""
|
|
|
|
return self.second_largest_aggregate_mass / self.projectile_mass
|
|
|
|
|
|
|
|
@property
|
|
|
|
def initial_water_mass(self) -> float:
|
|
|
|
return self.projectile_mass * self.projectile_water_fraction + self.target_mass * self.target_water_fraction
|
|
|
|
|
|
|
|
@property
|
|
|
|
def water_retention_both(self) -> float:
|
|
|
|
"""
|
|
|
|
p['wretentionB'] = (p['mS1'] * p['wmfS1'] + p['mS2'] * p['wmfS2']) / (p['mp'] * p['wp'] + p['mt'] * p['wt'])
|
|
|
|
"""
|
|
|
|
return (
|
|
|
|
self.largest_aggregate_mass * self.largest_aggregate_water_fraction
|
|
|
|
+ self.second_largest_aggregate_mass * self.second_largest_aggregate_water_fraction
|
|
|
|
) / self.initial_water_mass
|
|
|
|
|
|
|
|
@property
|
|
|
|
def water_retention_main(self) -> float:
|
|
|
|
"""
|
|
|
|
p['wretention1'] = p['mS1'] * p['wmfS1'] / (p['mp'] * p['wp'] + p['mt'] * p['wt'])
|
|
|
|
"""
|
|
|
|
return self.largest_aggregate_mass * self.largest_aggregate_water_fraction / self.initial_water_mass
|
|
|
|
|
2019-02-13 12:53:17 +01:00
|
|
|
@property
|
|
|
|
def simulation_key(self):
|
|
|
|
return "id{:04d}_v{:.1f}_a{:.0f}_m{:.0f}_g{:.1f}_wt{:.1f}_wp{:.1f}".format(
|
|
|
|
self.runid, self.vcode, self.alphacode, self.mcode, self.gammacode, self.wtcode, self.wpcode
|
|
|
|
)
|
|
|
|
|
|
|
|
def __repr__(self):
|
|
|
|
return f"<Simulation '{self.simulation_key}'>"
|
|
|
|
|
2019-02-12 18:56:05 +01:00
|
|
|
def load_params_from_dirname(self, dirname: str) -> None:
|
|
|
|
params = dirname.split("_")
|
|
|
|
self.runid = int(params[0][2:])
|
|
|
|
self.vcode = float(params[1][1:])
|
|
|
|
self.alphacode = float(params[2][1:])
|
|
|
|
self.mcode = float(params[3][1:])
|
|
|
|
self.gammacode = float(params[4][1:])
|
|
|
|
self.wtcode = float(params[5][2:])
|
|
|
|
self.wpcode = float(params[6][2:])
|
2019-02-13 12:53:17 +01:00
|
|
|
assert dirname == self.simulation_key
|
2019-02-12 18:56:05 +01:00
|
|
|
|
|
|
|
def load_params_from_spheres_ini_log(self, filename: str) -> None:
|
|
|
|
with open(filename) as f:
|
|
|
|
lines = [line.rstrip("\n") for line in f]
|
|
|
|
for i in range(len(lines)):
|
|
|
|
line = lines[i]
|
|
|
|
if "Geometry:" in line:
|
|
|
|
self.v = float(lines[i + 2].split(" = ")[-1])
|
|
|
|
self.alpha = float(lines[i + 3].split(" = ")[-1][:-1])
|
|
|
|
if "Masses:" in line:
|
|
|
|
self.total_mass = float(lines[i + 2].split()[3])
|
|
|
|
self.projectile_mass = float(lines[i + 4].split()[3])
|
|
|
|
self.target_mass = float(lines[i + 6].split()[3])
|
|
|
|
if "Mantle/shell mass fractions:" in line:
|
|
|
|
self.projectile_water_fraction = float(lines[i + 1].split()[7])
|
|
|
|
self.target_water_fraction = float(lines[i + 3].split()[7])
|
|
|
|
|
|
|
|
def load_params_from_aggregates_txt(self, filename: str) -> None:
|
|
|
|
with open(filename) as f:
|
|
|
|
lines = [line.rstrip("\n") for line in f]
|
|
|
|
for i in range(len(lines)):
|
|
|
|
line = lines[i]
|
|
|
|
if "#largest aggregate" in line:
|
|
|
|
self.largest_aggregate_mass = float(lines[i + 2].split()[0])
|
|
|
|
self.largest_aggregate_water_fraction = float(lines[i + 2].split()[2])
|
|
|
|
if "#2nd largest aggregate" in line:
|
|
|
|
self.second_largest_aggregate_mass = float(lines[i + 2].split()[0])
|
|
|
|
self.second_largest_aggregate_water_fraction = float(lines[i + 2].split()[2])
|
|
|
|
if "# distance" in line:
|
|
|
|
self.distance = float(lines[i + 1].split()[0])
|
|
|
|
self.rel_velocity = float(lines[i + 1].split()[1])
|
|
|
|
self.rel_velocity_per_esc_velocity = float(lines[i + 1].split()[2])
|
|
|
|
|
|
|
|
def assert_all_loaded(self) -> None:
|
|
|
|
for key, value in self.__dict__.items():
|
|
|
|
assert value is not None
|