Skip to content

Commit b1abd94

Browse files
committed
feat: Add stellar temperature
1 parent b1951c7 commit b1abd94

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

src/radiosim/ppdisks/simulation.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ def get_default_sampling_config():
3636
"binary_period": [6.04800e5, 3e9], # Seconds (logarithmic sampling)
3737
"binary_eccentricity": [0.0, 0.4], # 0 = Circle, 0 < e < 1 = Ellipse
3838
"stellar_mass": [0.5, 5], # Solar Masses
39+
"stellar_temperature": [2000.0, 3000.0], # Kelvin
3940
"num_planets": [1, 5],
4041
"planet_mass": [1.0e-6, 5.0e-3], # Solar Masses
4142
"planet_orbit_radius": [7.0, 30.0], # Astronomical Units
@@ -264,6 +265,11 @@ def simulate(
264265

265266
distances = self._planet_config.get_distances()
266267

268+
# If binary: distance -> binary eccentricity || binary period != distance
269+
# for the two stars in the planet file
270+
if planet_parameters["binary_system"]:
271+
distances = distances[2:]
272+
267273
param_config["mesh_parameters.ymin"] = (
268274
(np.min([mesh_parameters["y_min"], distances.min()]) * un.AU)
269275
.to(self._unit_system.length)
@@ -593,6 +599,12 @@ def sample_dict(read_dict):
593599
size=num_stars,
594600
)
595601

602+
planet_parameters["stellar_temperature"] = rng.uniform(
603+
low=planet_sampling["stellar_temperature"][0],
604+
high=planet_sampling["stellar_temperature"][1],
605+
size=num_stars,
606+
)
607+
596608
num_planets = rng.integers(
597609
low=planet_sampling["num_planets"][0],
598610
high=planet_sampling["num_planets"][1],

0 commit comments

Comments
 (0)