diff options
Diffstat (limited to 'fg21sim/galactic/freefree.py')
| -rw-r--r-- | fg21sim/galactic/freefree.py | 19 |
1 files changed, 16 insertions, 3 deletions
diff --git a/fg21sim/galactic/freefree.py b/fg21sim/galactic/freefree.py index 9c9424f..a1fcf6d 100644 --- a/fg21sim/galactic/freefree.py +++ b/fg21sim/galactic/freefree.py @@ -86,6 +86,7 @@ class FreeFree: self.use_float = self.configs.getn("output/use_float") self.checksum = self.configs.getn("output/checksum") self.clobber = self.configs.getn("output/clobber") + self.frequencies = self.configs.frequencies # [MHz] self.freq_unit = au.Unit(self.configs.getn("frequency/unit")) # logger.info("Loaded and set up configurations") @@ -292,10 +293,17 @@ class FreeFree: filepath = None return (skymap_f, filepath) - def simulate(self, frequencies): + def simulate(self, frequencies=None): """ Simulate the synchrotron map at the specified frequencies. + Parameters + ---------- + frequencies : float, or list[float] + The frequencies where to simulate the foreground map. + Unit: [MHz] + Default: None (i.e., use ``self.frequencies``) + Returns ------- skymaps : list[1D `~numpy.ndarray`] @@ -303,10 +311,15 @@ class FreeFree: paths : list[str] List of (absolute) path to the output sky maps. """ + if frequencies is not None: + frequencies = np.array(frequencies, ndmin=1) + else: + frequencies = self.frequencies + skymaps = [] paths = [] - for f in np.array(frequencies, ndmin=1): - skymap_f, outfile = self.simulate_frequency(f) + for freq in frequencies: + skymap_f, outfile = self.simulate_frequency(freq) skymaps.append(skymap_f) paths.append(outfile) return (skymaps, paths) |