@@ -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